MuTriggerAnalyzer.cc

Go to the documentation of this file.

#include "FWCore/Framework/interface/EDAnalyzer.h"
#include "FWCore/Utilities/interface/InputTag.h"
#include "FWCore/Framework/interface/Event.h"
#include "DataFormats/Common/interface/Handle.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "DataFormats/Candidate/interface/Candidate.h"
#include "FWCore/Common/interface/TriggerNames.h"
#include "DataFormats/Common/interface/TriggerResults.h"
#include "DataFormats/HLTReco/interface/TriggerObject.h"
#include "DataFormats/HLTReco/interface/TriggerEvent.h"
#include "DataFormats/Math/interface/deltaR.h"
#include "DataFormats/MuonReco/interface/Muon.h"
#include "FWCore/ServiceRegistry/interface/Service.h"
#include "CommonTools/UtilAlgos/interface/TFileService.h"
#include "PhysicsTools/RooStatsCms/interface/ClopperPearsonBinomialInterval.h"
#include "TGraphAsymmErrors.h"
#include "TH1.h"
#include <numeric>
#include <algorithm>
#include <string>
using namespace std;
using namespace reco;
using namespace edm;

/*
bool IsMuMatchedToHLTMu ( const reco::Candidate * dau, std::vector<reco::Particle> HLTMu , double DR, double DPtRel ) {
  unsigned int dim =  HLTMu.size();
  unsigned int nPass=0;
  if (dim==0) return false;
  for (unsigned int k =0; k< dim; k++ ) {
   if (  (deltaR(HLTMu[k], *dau) < DR)   && (fabs(HLTMu[k].pt() - dau->pt())/ HLTMu[k].pt()<DPtRel)){     nPass++ ;
    }
  }
  return (nPass>0);
}

bool IsMuMatchedToHLTSingleMu ( const reco::Candidate * dau, reco::Particle HLTMu , double DR, double DPtRel ) {
  unsigned int nPass=0;
  if (  (deltaR(HLTMu, *dau) < DR)   && (fabs(HLTMu.pt() - dau->pt())/ HLTMu.pt()<DPtRel)) {
    nPass++;
  }
  return (nPass>0);
}

*/


class MuTriggerAnalyzer : public edm::EDAnalyzer {
public:
  MuTriggerAnalyzer(const edm::ParameterSet& pset );

private:
  void analyze(const edm::Event& event, const edm::EventSetup& setup) override;
  void endJob() override;
  bool IsMuMatchedToHLTMu ( const reco::Muon & , std::vector<reco::Particle>& ,double ,double );

  edm::EDGetTokenT<TriggerResults> trigToken_;
  edm::InputTag  trigEvTag_;
  edm::EDGetTokenT< trigger::TriggerEvent >  trigEvToken_;
  edm::EDGetTokenT<View<Muon> >  muonToken_;
  double ptMuCut_, etaMuCut_;
  std::string hltPath_;
  std::string L3FilterName_;
  edm::Handle<edm::TriggerResults> triggerResults_;
  edm::TriggerNames const* trigNames_;
  edm::Handle< trigger::TriggerEvent > handleTriggerEvent_;

  double maxDPtRel_, maxDeltaR_ ;
  const int nbins_;
  const double ptMax_;
  TH1D * hTrigMuonPtNumS_;
  TH1D * hTrigMuonPtDenS_;
  TH1D * deltaR_;
  TH1D * deltaPtOverPt_;


};

bool MuTriggerAnalyzer::IsMuMatchedToHLTMu ( const reco::Muon & mu, std::vector<reco::Particle>& HLTMu , double DR, double DPtRel ) {
  size_t dim =  HLTMu.size();
  size_t nPass=0;

  // filling the denumerator;
  double muRecoPt= mu.pt();
  hTrigMuonPtDenS_-> Fill(muRecoPt);

  if (dim==0) return false;
  for (size_t k =0; k< dim; k++ ) {
    if (  (deltaR(HLTMu[k], mu) < DR)   && (fabs(HLTMu[k].pt() - mu.pt())/ HLTMu[k].pt()<DPtRel)){
      nPass++ ;
      std::cout<< " matching a muon " << std::endl;
      std::cout  << "muon reco  pt : " << muRecoPt<< std::endl;
      std::cout  << "muon reco  eta " <<  mu.eta() << std::endl;
      std::cout << "muon trigger  pt "<< HLTMu[k].pt() << std::endl;
      // filling the numerator, at the same bin as the denum.....
      hTrigMuonPtNumS_-> Fill(muRecoPt);
      deltaR_-> Fill(deltaR(HLTMu[k], mu));
      deltaPtOverPt_-> Fill(fabs(HLTMu[k].pt() - mu.pt())/ HLTMu[k].pt());


      std::cout << "muon trigger  eta : "<< HLTMu[k].eta() << std::endl;
      std::cout  <<"deltaR((HLTMu[k], mu)): "<< deltaR(HLTMu[k], mu) << std::endl;
      std::cout  <<"deltaPtOverPt: "<< fabs(HLTMu[k].pt() - mu.pt())/ HLTMu[k].pt() << std::endl;
    }
  }

  return (nPass>0);
}



MuTriggerAnalyzer::MuTriggerAnalyzer(const edm::ParameterSet& cfg ) :
  trigToken_(consumes<TriggerResults>(cfg.getParameter<edm::InputTag> ("TrigTag"))),
  trigEvTag_(cfg.getParameter<edm::InputTag> ("triggerEvent")),
  trigEvToken_(consumes< trigger::TriggerEvent >(trigEvTag_)),
  muonToken_(consumes<View<Muon> > (cfg.getUntrackedParameter<edm::InputTag>("muons"))),
  ptMuCut_(cfg.getUntrackedParameter<double>("ptMuCut")),
  etaMuCut_(cfg.getUntrackedParameter<double>("etaMuCut")),
  hltPath_(cfg.getParameter<std::string >("hltPath")),
  L3FilterName_(cfg.getParameter<std::string >("L3FilterName")),
  maxDPtRel_(cfg.getParameter<double>("maxDPtRel")),
  maxDeltaR_(cfg.getParameter<double>("maxDeltaR")),
  nbins_(cfg.getParameter<double>("ptMax_")),
  ptMax_(cfg.getParameter<double>("ptMax_")){
   Service<TFileService> fs;

 hTrigMuonPtNumS_ = fs->make<TH1D>("hTrigMuonPtNumS", "hTrigMuonPtNumS", nbins_ + 1, 0, ptMax_);
  hTrigMuonPtDenS_ = fs->make<TH1D>("hTrigMuonPtDenS", "hTrigMuonPtDenS", nbins_ +1 , 0, ptMax_);
  deltaR_ = fs->make<TH1D>("deltaR", "deltaR", nbins_+1, 0, maxDeltaR_);
  deltaPtOverPt_ = fs->make<TH1D>("deltaPtOverPt", "deltaPtOverPt", nbins_ + 1, 0, maxDPtRel_);

}

void MuTriggerAnalyzer::endJob() {
  for (int i = 0; i < nbins_+1; ++i){
    std::cout << "number of reco muon in bin " << i << " = " << hTrigMuonPtDenS_->GetBinContent(i) << std::endl;
    std::cout << "number of hlt muon in bin " << i << " = " << hTrigMuonPtNumS_->GetBinContent(i) << std::endl;


  }
}

void MuTriggerAnalyzer::analyze (const Event & ev, const EventSetup &) {
          //      bool overlap = false;

      // Trigger
      Handle<TriggerResults> triggerResults;
      TriggerNames trigNames;
      if (!ev.getByToken(trigToken_, triggerResults)) {
    LogWarning("") << ">>> TRIGGER collection does not exist !!!";
    return;
      }
      // trigNames.init(*triggerResults);
      trigNames_ = &ev.triggerNames(*triggerResults);
      //bool trigger_fired = false;
      for (unsigned int i=0; i<triggerResults->size(); i++) {


        std::string trigName = trigNames_->triggerName(i);
    if ( trigName == hltPath_ && triggerResults->accept(i)) {
      //trigger_fired = true;
    }
      }
      edm::Handle< trigger::TriggerEvent > handleTriggerEvent;
      //  LogTrace("") << ">>> Trigger bit: " << trigger_fired << " (" << hltPath_ << ")";
      if ( ! ev.getByToken( trigEvToken_, handleTriggerEvent ))  {
    LogWarning( "errorTriggerEventValid" ) << "trigger::TriggerEvent product with InputTag " << trigEvTag_.encode() << " not in event";
    return;
      }
      const trigger::TriggerObjectCollection & toc(handleTriggerEvent->getObjects());
      size_t nMuHLT =0;
      std::vector<reco::Particle>  HLTMuMatched;
      for ( size_t ia = 0; ia < handleTriggerEvent->sizeFilters(); ++ ia) {
    std::string fullname = handleTriggerEvent->filterTag(ia).encode();
    std::string name;
    size_t p = fullname.find_first_of(':');
    if ( p != std::string::npos) {
      name = fullname.substr(0, p);
    }
    else {
      name = fullname;
    }
    if ( !toc.empty() ) {
      const trigger::Keys & k = handleTriggerEvent->filterKeys(ia);
      for (trigger::Keys::const_iterator ki = k.begin(); ki !=k.end(); ++ki ) {
        if (name == L3FilterName_  ) {
          HLTMuMatched.push_back(toc[*ki].particle());
          nMuHLT++;
        }
      }
    }
      }

    //  looping on muon....
      Handle<View<Muon> >   muons;
      if (!ev.getByToken(muonToken_, muons)) {
    LogError("") << ">>> muon collection does not exist !!!";
    return;
      }
      //saving only muons with pt> ptMuCut and eta<etaMuCut
      std::vector<reco::Muon>  highPtGlbMuons;

      for (unsigned int i=0; i<muons->size(); i++ ){
        const reco::Muon & mu = muons->at(i);
    double pt = mu.pt();
    double eta = mu.eta();
    if (pt> ptMuCut_ && fabs(eta)< etaMuCut_) {
      if (mu.isGlobalMuon()) highPtGlbMuons.push_back(mu);
    }
      }
      unsigned int nHighPtGlbMu = highPtGlbMuons.size();
      std::cout << "I've got " << nHighPtGlbMu << " nHighPtGlbMu" << std::endl;
      // unsigned int nHighPtStaMu = highPtStaMuons.size();
      // stop the loop after 10 cicles....
      if (nHighPtGlbMu> 10) nHighPtGlbMu=10;
      
      for(unsigned int i =0 ; i < nHighPtGlbMu ; i++) {
    IsMuMatchedToHLTMu ( highPtGlbMuons[i],  HLTMuMatched ,maxDeltaR_, maxDPtRel_ );
      }

}


#include "FWCore/Framework/interface/MakerMacros.h"

DEFINE_FWK_MODULE( MuTriggerAnalyzer );