L1MuonMatcherAlgo.cc

Go to the documentation of this file.

#include "MuonAnalysis/MuonAssociators/interface/L1MuonMatcherAlgo.h"

#include "FWCore/MessageLogger/interface/MessageLogger.h"

L1MuonMatcherAlgo::L1MuonMatcherAlgo(const edm::ParameterSet & iConfig) :
    prop_(iConfig),
    preselectionCut_(iConfig.existsAs<std::string>("preselection") ? iConfig.getParameter<std::string>("preselection") : ""),
    deltaR2_(std::pow(iConfig.getParameter<double>("maxDeltaR"),2)),
    deltaPhi_(iConfig.existsAs<double>("maxDeltaPhi") ? iConfig.getParameter<double>("maxDeltaPhi") : 10),
    deltaEta_(iConfig.existsAs<double>("maxDeltaEta") ? iConfig.getParameter<double>("maxDeltaEta") : 10),
    l1PhiOffset_(iConfig.existsAs<double>("l1PhiOffset") ? iConfig.getParameter<double>("l1PhiOffset") : 0)
{
    bool reqPhi = iConfig.existsAs<bool>("sortByDeltaPhi") && iConfig.getParameter<bool>("sortByDeltaPhi");
    bool reqEta = iConfig.existsAs<bool>("sortByDeltaEta") && iConfig.getParameter<bool>("sortByDeltaEta");
    bool reqPt  = iConfig.existsAs<bool>("sortByPt")       && iConfig.getParameter<bool>("sortByPt");
    std::string sortBy = iConfig.existsAs<std::string>("sortBy") ? iConfig.getParameter<std::string>("sortBy") : "";
    if (reqPhi + reqEta + reqPt > 1) throw cms::Exception("Configuration") << "L1MuonMatcherAlgo: Can't set more than one 'sortBy<XXX>' parameter to True.\n";
    if (sortBy == "deltaPhi") { 
        if (reqEta || reqPt) 
            throw cms::Exception("Configuration") << "L1MuonMatcherAlgo: Can't set sortBy = 'deltaPhi' and set also another 'sortBy<XXX>' parameter to True.\n";
        else reqPhi = true;
    }
    if (sortBy == "deltaEta") {
     if(reqPhi || reqPt)
            throw cms::Exception("Configuration") << "L1MuonMatcherAlgo: Can't set sortBy = 'deltaEta' and set also another 'sortBy<XXX>' parameter to True.\n";
         else reqEta = true;
    }
    if (sortBy == "pt") {
     if(reqPhi || reqEta) 
            throw cms::Exception("Configuration") << "L1MuonMatcherAlgo: Can't set sortBy = 'pt' and set also another 'sortBy<XXX>' parameter to True.\n";
        else reqPt = true;
    }
    if (sortBy == "deltaR") {
     if(reqPhi || reqEta || reqPt)
            throw cms::Exception("Configuration") << "L1MuonMatcherAlgo: Can't set sortBy = 'deltaR' and set also another 'sortBy<XXX>' parameter to True.\n";
    }
    // so, if we're here there's no ambiguity in what the user may want. either everything is false, or exactly one req is true.
    if      (reqEta) sortBy_ = SortByDeltaEta;
    else if (reqPhi) sortBy_ = SortByDeltaPhi;
    else if (reqPt)  sortBy_ = SortByPt;
    else             sortBy_ = SortByDeltaR;
}


L1MuonMatcherAlgo::~L1MuonMatcherAlgo() {}

void
L1MuonMatcherAlgo::init(const edm::EventSetup & iSetup) {
    prop_.init(iSetup);
}

bool
L1MuonMatcherAlgo::match(TrajectoryStateOnSurface & propagated, const l1extra::L1MuonParticle &l1, float &deltaR, float &deltaPhi) const {
    if (preselectionCut_(l1)) {
        GlobalPoint pos = propagated.globalPosition();
        double thisDeltaPhi = ::deltaPhi(double(pos.phi()),  l1.phi()+l1PhiOffset_);
        double thisDeltaEta = pos.eta() - l1.eta();
        double thisDeltaR2  = ::deltaR2(double(pos.eta()), double(pos.phi()), l1.eta(), l1.phi()+l1PhiOffset_);
        if ((fabs(thisDeltaPhi) < deltaPhi_) && (fabs(thisDeltaEta) < deltaEta_) && (thisDeltaR2 < deltaR2_)) {
            deltaR   = std::sqrt(thisDeltaR2);
            deltaPhi = thisDeltaPhi;
            return true;
        }
    }
    return false;
}

int
L1MuonMatcherAlgo::match(TrajectoryStateOnSurface & propagated, const std::vector<l1extra::L1MuonParticle> &l1s, float &deltaR, float &deltaPhi) const {
    return matchGeneric(propagated, l1s, preselectionCut_, deltaR, deltaPhi);
}