/cvmfs/atlas.cern.ch/repo/sw/ASG/AnalysisBase/2.4.31/TrigMuonEfficiency/TrigMuonEfficiency/MuonTriggerMatching.h

#ifndef __MuonTriggerMatching_h__
#define __MuonTriggerMatching_h__

#include <string>
#include <map>
#include <set>
#include "TTree.h"
#include "TLorentzVector.h"

#include "TrigMuonEfficiency/LeptonTriggerMatching.h"

class MuonTriggerMatching : public LeptonTriggerMatching {
  public:
    MuonTriggerMatching(const MTT__D3PDEvent *d3pd,
                        TTree *TrigConfTree=0)
      : LeptonTriggerMatching(d3pd, TrigConfTree),
        m_index(false),
        m_DimuonChainMap()
    {
      if (TrigConfTree) {
        std::vector<std::string> skip;
        skip.push_back("EF_mu");
        skip.push_back("EF_2mu");
        skip.push_back("L2_mu15");
        skip.push_back("L2_mu18");
        skip.push_back("L2_mu20");
        skip.push_back("L2_mu24");
        readTrigConfTree(TrigConfTree, skip);

      } else {
        createChainIdMapFromFile();
      }
    }
    virtual ~MuonTriggerMatching(){}


    bool match(const double eta,
               const double phi,
               const std::string& chain,
               const std::string& l1item="");


    std::pair<int, int>
    matchedTrack(const double eta,
                 const double phi,
                 const std::string& chain,
                 const std::string& l1item="");


    bool matchDimuon(const TLorentzVector& muon1,
                     const TLorentzVector& muon2,
                     const std::string& chain,
                     std::pair<bool, bool>& result1,
                     std::pair<bool, bool>& result2);


    int matchDimuon_python(const TLorentzVector& muon1,
                           const TLorentzVector& muon2,
                           const std::string& chain)
    {
      std::pair<bool, bool> result1, result2;
      matchDimuon(muon1, muon2, chain, result1, result2);
      return (result1.second << 3 | result2.first << 2 | result2.second << 1 | result1.first << 0);
    }



    // for developers
    void useIndex(bool x) {m_index = x;}


    struct DimuonChainInfo {
      public:
        std::string chain;
        std::string tightness;
        std::pair<std::string, std::string> thresholds;
        std::string l2seed;
        std::string efseed;
        bool is2011;
        bool isEFFS;
        bool isSymmetric;
        bool isValid;

        DimuonChainInfo(const std::string& chain_="") :
          chain(chain_), tightness(""), thresholds(), l2seed(""), efseed(""),
          is2011(false), isEFFS(false), isSymmetric(false), isValid(false) {}

        void dump() {
          std::cout << "DimuonChainInfo: chain = " << chain << " tightness = " << tightness
                << " thresholds.first = " << thresholds.first << " thresholds.second = " << thresholds.second
                << " l2seed = " << l2seed << " efseed = " << efseed << " is2011 = " << is2011
                << " isEFFS = " << isEFFS << " isSymmetric = " << isSymmetric << " isValid = " << isValid << "\n";
        }
    };

  private:
    bool m_index;
    std::map<std::string, DimuonChainInfo> m_DimuonChainMap; // [chain - <threshold1, threshold2>]
  
    double matchedTrackDetail_index(EFmuon& efMuonId,
                                    const EFmuon& usedEfMuonId,
                                    const double eta,
                                    const double phi,
                                    const std::string& chainForThisMuon,
                                    const std::string& l1item = std::string(),
                                    const std::string& chainEventTrigger = std::string());

    double matchedTrackDetail(EFmuon& efMuonId,
                              const EFmuon& usedEfMuonId,
                              const double eta,
                              const double phi,
                              const std::string& chainForThisMuon,
                              const std::string& l1item = std::string(),
                              const std::string& chainEventTrigger = std::string());

    std::pair<bool, bool> matchDimuon(const TLorentzVector& muon1,
                                      const TLorentzVector& muon2,
                                      const DimuonChainInfo& chain);

    void createChainIdMapFromFile(); //implementation in MuonChainIdMap.h

    bool decodeDimuonChain(DimuonChainInfo& chainInfo);

    bool checkSeedL2(const double eta,
                     const double phi,
                     const std::string& chain,
                     double l2dr=0.15,
                     double l1dr=0.20);

    bool checkSeedEF(const double eta,
                     const double phi,
                     const std::string& chain);

    bool checkSeed_index(const double eta,
                         const double phi,
                         const std::string& chain);
};


#endif //__MuonTriggerMatching_h__
// eof

---