/afs/cern.ch/work/a/aaltunda/public/www/CMSSW_5_3_14/src/ElectroWeakAnalysis/ZMuMu/plugins/ZMuMuAnalyzer.cc

Go to the documentation of this file.

/* \class ZMuMuAnalyzer
 *
 * Z->mu+m- standard analysis for cross section 
 * measurements. Take as input the output of the
 * standard EWK skim: zToMuMu
 * 
 * Produces mass spectra and other histograms for
 * the samples in input:
 *
 *  + Z -> mu+mu-, both muons are "global" muons
 *  + Z -> mu+mu-, one muons is "global" muons, one unmatched tracks
 *  + Z -> mu+mu-, one muons is "global" muons, one unmatched stand-alone muon
 * 
 *
 * \author Michele de Gruttola, INFN Naples
 *
 * \id $Id: ZMuMuAnalyzer.cc,v 1.7 2010/02/19 02:46:28 wmtan Exp $
 *
 */
#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 "DataFormats/Candidate/interface/OverlapChecker.h"
#include "DataFormats/Candidate/interface/Particle.h"
#include "DataFormats/Candidate/interface/Candidate.h"
#include "DataFormats/Candidate/interface/CandMatchMap.h"
#include "FWCore/ServiceRegistry/interface/Service.h"
#include "CommonTools/UtilAlgos/interface/TFileService.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "DataFormats/Common/interface/AssociationVector.h"
#include "TH1.h"
#include <iostream>
#include <iterator>
using namespace edm;
using namespace std;
using namespace reco;

typedef edm::AssociationVector<reco::CandidateRefProd, std::vector<double> > IsolationCollection;

class ZMuMuAnalyzer : public edm::EDAnalyzer {
public:
  ZMuMuAnalyzer(const edm::ParameterSet& pset);
private:
  virtual void analyze(const edm::Event& event, const edm::EventSetup& setup);
  virtual void endJob();
  
  OverlapChecker overlap_;
  InputTag zMuMu_;
  InputTag zMuTrack_;
  InputTag zMuStandAlone_;
  InputTag  muIso_, trackIso_, standAloneIso_;
  InputTag  zMuMuMap_ ,zMuTrackMap_, zMuStandAloneMap_;
  double isocut_, etacut_, ptcut_,ptSTAcut_,  minZmass_, maxZmass_;
  TH1D * h_zMuMu_mass_, * h_zMuSingleTrack_mass_, * h_zMuSingleStandAlone_mass_,* h_zMuSingleStandAloneOverlap_mass_,
    * h_zMuMuMatched_mass_,
 *h_zMuSingleTrackMatched_mass_,
    * h_zMuSingleStandAloneMatched_mass_, 
    * h_zMuSingleStandAloneOverlapMatched_mass_;
};

ZMuMuAnalyzer::ZMuMuAnalyzer(const edm::ParameterSet& pset) : 
  zMuMu_( pset.getParameter<InputTag>( "zMuMu" ) ),
  zMuTrack_( pset.getParameter<InputTag>( "zMuTrack" ) ),
  zMuStandAlone_( pset.getParameter<InputTag>( "zMuStandAlone" ) ),
  muIso_( pset.getParameter<InputTag>( "muIso" ) ),
  trackIso_( pset.getParameter<InputTag>( "trackIso" ) ),
  standAloneIso_( pset.getParameter<InputTag>( "standAloneIso" ) ),
  zMuMuMap_( pset.getParameter<InputTag>( "zMuMuMap" ) ),
  zMuTrackMap_( pset.getParameter<InputTag>( "zMuTrackMap" ) ),
  zMuStandAloneMap_( pset.getParameter<InputTag>( "zMuStandAloneMap" ) ),
  isocut_( pset.getParameter<double>( "isocut" ) ),
  etacut_( pset.getParameter<double>( "etacut" ) ),
  ptcut_( pset.getParameter<double>( "ptcut" ) ),
  ptSTAcut_( pset.getParameter<double>( "ptSTAcut" ) ),
  
  minZmass_( pset.getParameter<double>( "minZmass" )),
  maxZmass_( pset.getParameter<double>( "maxZmass" )) {
  
  Service<TFileService> fs;
  h_zMuMu_mass_ = fs->make<TH1D>( "ZMuMumass", "ZMuMu mass(GeV)", 200,  0., 200. );
  h_zMuSingleTrack_mass_ = fs->make<TH1D>( "ZMuSingleTrackmass", "ZMuSingleTrack mass(GeV)", 100,  0., 200. );
  h_zMuSingleStandAlone_mass_ = fs->make<TH1D>( "ZMuSingleStandAlonemass", "ZMuSingleStandAlone mass(GeV)", 50,  0., 200. );
  h_zMuSingleStandAloneOverlap_mass_ = fs->make<TH1D>( "ZMuSingleStandAloneOverlapmass", "ZMuSingleStandAloneOverlap  mass(GeV)", 50,  0., 200. );
  
  
  h_zMuMuMatched_mass_ = fs->make<TH1D>( "ZMuMuMatchedmass", "ZMuMu Matched  mass(GeV)", 200,  0., 200. );
  h_zMuSingleTrackMatched_mass_ = fs->make<TH1D>( "ZMuSingleTrackmassMatched", "ZMuSingleTrackMatched mass(GeV)", 100,  0., 200. );
  h_zMuSingleStandAloneMatched_mass_ = fs->make<TH1D>( "ZMuSingleStandAlonemassMatched", "ZMuSingleStandAloneMatched mass(GeV)", 50,  0., 200. );
  h_zMuSingleStandAloneOverlapMatched_mass_ = fs->make<TH1D>( "ZMuSingleStandAloneOverlapmassMatched", "ZMuSingleStandAloneMatched Overlap  mass(GeV)", 50,  0., 200. );
}

void ZMuMuAnalyzer::analyze(const edm::Event& event, const edm::EventSetup& setup) {
  Handle<CandidateCollection> zMuMu;
  event.getByLabel(zMuMu_, zMuMu);
  Handle<CandidateCollection> zMuTrack;
  event.getByLabel( zMuTrack_, zMuTrack );
  Handle<CandidateCollection> zMuStandAlone;
  event.getByLabel( zMuStandAlone_, zMuStandAlone );  

  unsigned int nZMuMu = zMuMu->size();
  unsigned int nZTrackMu = zMuTrack->size();
  unsigned int nZStandAloneMu = zMuStandAlone->size();
  static const double zMass = 91.1876; // PDG Z mass

  //  cout << "nZMuMu = " << nZMuMu << endl;
  //  cout << "nZTrackMu = " << nZTrackMu << endl;
  //  cout << "nZStandAloneMu = " << nZStandAloneMu << endl;
 
  Handle<CandMatchMap> zMuMuMap;
  if( nZMuMu > 0 ) {
    event.getByLabel(zMuMuMap_, zMuMuMap);
  }

  Handle<CandMatchMap> zMuTrackMap;
  if( nZTrackMu > 0 ) {
    event.getByLabel( zMuTrackMap_, zMuTrackMap );
  }

  Handle<CandMatchMap> zMuStandAloneMap;
  if( nZStandAloneMu > 0 ) {
    event.getByLabel( zMuStandAloneMap_, zMuStandAloneMap );  
  }    

  Handle<IsolationCollection> muIso;
  event.getByLabel(muIso_, muIso);
  ProductID muIsoId = muIso->keyProduct().id();
  Handle<IsolationCollection> trackIso;
  event.getByLabel(trackIso_, trackIso);
  ProductID trackIsoId = trackIso->keyProduct().id();
  
  Handle<IsolationCollection> standAloneIso;
  event.getByLabel(standAloneIso_, standAloneIso);
  ProductID standAloneIsoId = standAloneIso->keyProduct().id();
  
  if (nZMuMu > 0) {
    double mass = 1000000.;
    for( unsigned int i = 0; i < nZMuMu; i++ ) {
      const Candidate & zmmCand = (*zMuMu)[ i ];
      CandidateRef CandRef(zMuMu,i);
      CandidateRef lep1 = zmmCand.daughter( 0 )->masterClone().castTo<CandidateRef>();
      CandidateRef lep2 = zmmCand.daughter( 1 )->masterClone().castTo<CandidateRef>();
      
      const  double iso1 = muIso->value( lep1.key() );  
      const  double iso2 = muIso->value( lep2.key() );  
      
      double m = zmmCand.mass();
      if (lep1->pt()>ptcut_ && lep2->pt()>ptcut_ &&    
          fabs(lep1->eta())<etacut_ && fabs(lep2->eta())<etacut_ &&
          m>minZmass_ && m<maxZmass_ && iso1 < isocut_ && iso2 <isocut_) {
        if ( fabs( mass - zMass ) > fabs( m - zMass ) ) {
          mass = m;
        }
        
        h_zMuMu_mass_->Fill( mass );      
        CandMatchMap::const_iterator m0 = zMuMuMap->find(CandRef);
        if( m0 != zMuMuMap->end()) {
            h_zMuMuMatched_mass_->Fill( mass );
        }
      }
    }
  }
  
  //ZmuSingleTRack
  if (nZMuMu ==0 && nZTrackMu>0) {
    for( unsigned int j = 0; j < nZTrackMu; j++ ) {
      const Candidate & ztmCand = (*zMuTrack)[ j ];
      CandidateRef CandRef(zMuTrack,j);
      CandidateRef lep1 = ztmCand.daughter( 0 )->masterClone().castTo<CandidateRef>();
      CandidateRef lep2 = ztmCand.daughter( 1 )->masterClone().castTo<CandidateRef>();
      
      ProductID id1 = lep1.id();
      ProductID id2 = lep2.id();
      double iso1 = -1;
      double iso2 = -1;
      
      if( id1 == muIsoId ) 
        iso1 = muIso->value( lep1.key() );      
      else if ( id1 == trackIsoId )
        iso1 = trackIso->value( lep1.key() );   
      
      if( id2 == muIsoId ) 
              iso2 = muIso->value( lep2.key() );        
      else if ( id2 == trackIsoId )
        iso2 = trackIso->value( lep2.key() );   
 
      double mt = ztmCand.mass();
      if (lep1->pt()>ptcut_ && lep2->pt()>ptcut_ &&    
          fabs(lep1->eta())<etacut_ && fabs(lep2->eta())<etacut_ &&
          mt>minZmass_ && mt<maxZmass_ && iso1<isocut_ && iso2 <isocut_) {
        h_zMuSingleTrack_mass_->Fill( mt );
        CandMatchMap::const_iterator m0 = zMuTrackMap->find(CandRef);
        if( m0 != zMuTrackMap->end()) {
          h_zMuSingleTrackMatched_mass_->Fill( mt );
        }
      }
    }
  }   
  
  //ZmuSingleStandAlone
  if (nZMuMu ==0 && nZStandAloneMu>0) {
    //      unsigned int index = 1000;
    for( unsigned int j = 0; j < nZStandAloneMu; j++ ) {
      const Candidate & zsmCand = (*zMuStandAlone)[ j ];
      CandidateRef CandRef(zMuStandAlone,j);
      CandidateRef lep1 = zsmCand.daughter( 0 )->masterClone().castTo<CandidateRef>();
      CandidateRef lep2 = zsmCand.daughter( 1 )->masterClone().castTo<CandidateRef>();
      
      ProductID id1 = lep1.id();
      ProductID id2 = lep2.id();
      double iso1 = -1;
      double iso2 = -1;
      
      if( id1 == muIsoId ) 
        iso1 = muIso->value( lep1.key() );      
      else if ( id1 == standAloneIsoId )
        iso1 = standAloneIso->value( lep1.key() );      
      
      if( id2 == muIsoId ) 
        iso2 = muIso->value( lep2.key() );      
      else if ( id2 == standAloneIsoId )
        iso2 = standAloneIso->value( lep2.key() );      
      
      double ms = zsmCand.mass();
      if (lep1->pt()>ptSTAcut_ && lep2->pt()>ptSTAcut_ &&    
          fabs(lep1->eta())<etacut_ && fabs(lep2->eta())<etacut_ &&
          ms>minZmass_ && ms<maxZmass_ && iso1<isocut_ && iso2 <isocut_) {
        h_zMuSingleStandAlone_mass_->Fill( ms );
        CandMatchMap::const_iterator m0 = zMuStandAloneMap->find(CandRef);
        if( m0 != zMuStandAloneMap->end()) {
          h_zMuSingleStandAloneMatched_mass_->Fill( ms );
        }
        
        bool noOverlap = true;
        for( unsigned int j = 0; j < zMuTrack->size(); j++ ) {
          const Candidate & ztmCand = (*zMuTrack)[ j ];
          CandidateRef CandReft(zMuTrack,j);
          
          CandidateRef lep1 = ztmCand.daughter( 0 )->masterClone().castTo<CandidateRef>();
          CandidateRef lep2 = ztmCand.daughter( 1 )->masterClone().castTo<CandidateRef>();
          
          ProductID id1 = lep1.id();
          ProductID id2 = lep2.id();
          double iso1 = -1;
          double iso2 = -1;
          
          if( id1 == muIsoId ) 
            iso1 = muIso->value( lep1.key() );  
          else if ( id1 == trackIsoId )
            iso1 = trackIso->value( lep1.key() );       
          
          if( id2 == muIsoId ) 
            iso2 = muIso->value( lep2.key() );  
          else if ( id2 == trackIsoId )
            iso2 = trackIso->value( lep2.key() );       
          
          double mt = ztmCand.mass();
          if (lep1->pt()>ptcut_ && lep2->pt()>ptcut_ &&    
              fabs(lep1->eta())<etacut_ && fabs(lep2->eta())<etacut_ &&
              mt>minZmass_ && mt<maxZmass_ && iso1<isocut_ && iso2 <isocut_) {
            
            if ( overlap_( ztmCand, zsmCand ) ) {
              noOverlap = false; 
              break;
            }
            if (!noOverlap ) {
              h_zMuSingleStandAloneOverlap_mass_->Fill( ms ); 
              CandMatchMap::const_iterator m1 = zMuTrackMap->find(CandReft);
              CandMatchMap::const_iterator m2 = zMuStandAloneMap->find(CandRef);
              
              if( m1 != zMuTrackMap->end()  && m2 != zMuStandAloneMap->end()  ) {
                h_zMuSingleStandAloneOverlapMatched_mass_->Fill( ms );
              }
            }
          }         
        }
      }
    }
  }
}  
   
void ZMuMuAnalyzer::endJob() {
  double Nzmm = h_zMuMu_mass_->GetEntries() ;
  double Nzsm = h_zMuSingleStandAlone_mass_->GetEntries()  ;
  double Nzsnom = h_zMuSingleStandAloneOverlap_mass_->GetEntries()  ;
  double Nztm = h_zMuSingleTrack_mass_->GetEntries();
  
  double NzmmMatch = h_zMuMuMatched_mass_->GetEntries() ;
  double NzsmMatch = h_zMuSingleStandAloneMatched_mass_->GetEntries()  ;
  double NzsnomMatch = h_zMuSingleStandAloneOverlapMatched_mass_->GetEntries()  ;
  double NztmMatch = h_zMuSingleTrackMatched_mass_->GetEntries();

  cout<<"-- N SingleTrackMu = "<<Nztm<<endl;
  cout<<"-----N SinglStandAloneMu = "<<Nzsm<<endl;
  cout<<"-----N SingleStandAloneOverlapMu = "<<Nzsnom<<endl;
  cout<<"------- N MuMu = "<<Nzmm<<endl;
  
  cout<<"-- N SingleTrackMuMatched = "<<NztmMatch<<endl;
  cout<<"-----N SinglStandAloneMuMatched = "<<NzsmMatch<<endl;
  cout<<"-----N SingleStandAloneOverlapMuMatched  = "<<NzsnomMatch<<endl;
  cout<<"------- N MuMu Matched  = "<<NzmmMatch<<endl;
}
  
#include "FWCore/Framework/interface/MakerMacros.h"

DEFINE_FWK_MODULE(ZMuMuAnalyzer);