MC_Efficiency_Analyzer.cc

Go to the documentation of this file.

/* \class MCEfficiencyAnalyzer
 *
 * Muon reconstruction efficiency from MC truth, 
 * for global muons, tracks and standalone. Take as input the output of the
 * standard EWK skim: zToMuMu
 *
 * Produces in output the efficency number
 *
 * \author Michele de Gruttola, INFN Naples
*/

#include "FWCore/Framework/interface/EDAnalyzer.h"
#include "DataFormats/Math/interface/deltaR.h"
#include "FWCore/Utilities/interface/InputTag.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "DataFormats/Candidate/interface/Particle.h"
#include "DataFormats/Candidate/interface/Candidate.h"
#include "DataFormats/HepMCCandidate/interface/GenParticle.h"
#include "DataFormats/Candidate/interface/CandMatchMap.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "DataFormats/Common/interface/Ref.h"
#include <iostream>
#include <iterator>
using namespace edm;
using namespace std;
using namespace reco;

class MCEfficiencyAnalyzer : public edm::EDAnalyzer 
{
public:
  MCEfficiencyAnalyzer(const edm::ParameterSet& pset) : 
    zMuMu_( pset.getParameter<InputTag>( "zMuMu" ) ),
    Muons_( pset.getParameter<InputTag>( "Muons" ) ),
    MuonsMap_( pset.getParameter<InputTag>( "MuonsMap" ) ),
    Tracks_( pset.getParameter<InputTag>( "Tracks" ) ),
    TracksMap_( pset.getParameter<InputTag>( "TracksMap" ) ),
    genParticles_( pset.getParameter<InputTag>( "genParticles" ) ),
    StandAlone_( pset.getParameter<InputTag>( "StandAlone" ) ),
    StandAloneMap_( pset.getParameter<InputTag>( "StandAloneMap" ) ),
    etacut_( pset.getParameter<double>( "etacut" ) ),
    ptcut_( pset.getParameter<double>( "ptcut" ) ),    
    deltaRStacut_( pset.getParameter<double>( "deltaRStacut" ) )

  { 
    nMuMC =0; nMureco =0; nTrk=0; nSta=0; nNotMuMatching =0 ;  
  }
  
  virtual void analyze(const edm::Event& event, const edm::EventSetup& setup) 
  {
    
    Handle<CandidateCollection> zMuMu;
    event.getByLabel(zMuMu_, zMuMu);
    
    
    Handle<CandidateCollection> Muons;
    event.getByLabel(Muons_, Muons);
    
    Handle<CandidateCollection> Tracks;
    event.getByLabel(Tracks_, Tracks);
    
    Handle<CandidateCollection> StandAlone;
    event.getByLabel(StandAlone_, StandAlone);
    
    Handle<CandMatchMap> MuonsMap;
    event.getByLabel(MuonsMap_, MuonsMap);
    
    Handle<CandMatchMap> TracksMap;
    event.getByLabel(TracksMap_, TracksMap);
    
    Handle<CandMatchMap> StandAloneMap;
    event.getByLabel(StandAloneMap_, StandAloneMap);
    
    Handle<GenParticleCollection> genParticles;
    event.getByLabel(genParticles_, genParticles);

    //Getting muons from Z MC  
    for( unsigned int k = 0; k < genParticles->size(); k++ ) 
      {
    const Candidate & ZCand = (*genParticles)[ k ];
    int status = ZCand.status(); 
    
    if (ZCand.pdgId()==23&& status==3 )
      {
        // positive muons 
        const Candidate  *  muCand1 =   ZCand.daughter(0);
        if (muCand1->status()==3)   
          {
        for (unsigned int d =0 ; d< muCand1->numberOfDaughters(); d++)
          { 
            const Candidate * muCandidate = muCand1->daughter(d);
            if (muCandidate->pdgId() == muCand1->pdgId() )
              {
            muCand1 = muCand1->daughter(d);
              }
          } 
          }
        // negative muons
        const Candidate  *  muCand2 =   ZCand.daughter(1);
        if (muCand2->status()==3) 
          {     
        for (unsigned int e =0 ; e< muCand2->numberOfDaughters(); e++)
          { 
            const Candidate * muCandidate = muCand2->daughter(e);
            if (muCandidate->pdgId() == muCand2->pdgId() )
              {
            muCand2 = muCand2->daughter(e);
              }
          }
          }
        
        double deltaR_Mu_Sta =0;
        int nMurecoTemp = nMureco; 
        // getting mu matched
        CandMatchMap::const_iterator i;  
        for(i = MuonsMap->begin(); i != MuonsMap->end(); i++ )  
          {
        const Candidate/* & reco = * i -> key,*/ &  mc =  * i -> val;
        if ((muCand1 == &mc) && (mc.pt()>ptcut_) && (abs(mc.eta()<etacut_)))
          {
            nMuMC++;
            nMureco++;
            break;
          }
          }
        if (nMureco == nMurecoTemp ) // I.E. MU RECO NOT FOUND!!!!  
          {
                int nTrkTemp = nTrk;                
        int nStaTemp = nSta;
 
        // getting tracks matched and doing the same, CONTROLLING IF MU IS RECONSTRUCTED AS A TRACK AND NOT AS A MU
        CandMatchMap::const_iterator l;         
        for(l = TracksMap->begin(); l != TracksMap->end(); l++ )
          {
            const Candidate /* & Trkreco = * l -> key, */ &  Trkmc =  * l -> val;
            if (( muCand1 == & Trkmc) && (Trkmc.pt()>ptcut_) && (abs(Trkmc.eta()<etacut_))) 
              {
                nMuMC++;
            nTrk++;  
            break;
              }  
          }
        // the same for standalone  
        CandMatchMap::const_iterator n;  
        for(n = StandAloneMap->begin(); n != StandAloneMap->end(); n++ )                  
          {
            const Candidate & Stareco = * n -> key, &  Stamc =  * n -> val;
            if ((muCand1 == &Stamc ) && (Stamc.pt()>ptcut_) && (abs(Stamc.eta()<etacut_)))  
              {
            nMuMC++;
            nSta++;
            deltaR_Mu_Sta = deltaR(Stareco, *muCand1);
                        
            //  cout<<"Ho trovato un sta reco "<<endl;
            break;
              }
          }
        // controlling if sta and trk are reconstrucetd both, and if so the get the deltaR beetween muon MC and reco sta, to controll correlation to this happening  
        if ((nSta == nStaTemp + 1) && (nTrk == nTrkTemp + 1 ) ) 
          { 
            nNotMuMatching ++; 
            if ((deltaR_Mu_Sta< deltaRStacut_))
              {
            v_.push_back(deltaR_Mu_Sta) ;
                   cout << "Not matching from trk and sta matched to MC mu, to reconstruct a recoMU" << endl;
              }
          }
          }
      }
      }
  }

  
  virtual void endJob() 
  {
    
    
    cout <<"--- nMuMC == "<<nMuMC<<endl;
    cout <<"--- nMureco == "<<nMureco<<endl;
    cout <<"--- nSta == "<<nSta<<endl;
    cout <<"--- nTrk == "<<nTrk<<endl;
    cout <<"--- nNotMuMatching from a trk and sta matched to a Mu MC == "<<nNotMuMatching<<endl; 
    if (nMuMC!=0 )
      {
    cout<<" effMu == "<<(double) nMureco/nMuMC<<endl;
    cout<<" effTrk == "<< (double)(nTrk + nMureco) /nMuMC<<endl;  
    cout<<" effSta == "<< (double)(nSta + nMureco) / nMuMC<<endl;
      }   
     
vector< int >::const_iterator p2;
    for (unsigned int i =0 ; i < v_.size(); ++i ) 
      {
    cout<<" delta R Mu Sta == "<< v_[i]<<endl;
        }
  }
  
  
  
  
  
private:
  
  InputTag zMuMu_, Muons_, MuonsMap_, Tracks_, TracksMap_, genParticles_, StandAlone_,StandAloneMap_;
  double etacut_, ptcut_, deltaRStacut_;
  int nMuMC, nMureco, nTrk, nSta, nNotMuMatching;
  vector<double> v_;
};


DEFINE_FWK_MODULE(MCEfficiencyAnalyzer);