#include <MuScleFitMuonSelector.h>

Public Member Functions
const reco::Candidate *	getStatus1Muon (const reco::Candidate *status3Muon)

const reco::Candidate *	getStatus3Muon (const reco::Candidate *status3Muon)

	MuScleFitMuonSelector (const edm::InputTag &muonLabel, const int muonType, const bool PATmuons, const std::vector< int > &resfind, const bool speedup, const std::string &genParticlesName, const bool compareToSimTracks, const edm::InputTag &simTracksCollectionName, const bool sherpa, const bool debug)

void	selectMuons (const edm::Event &event, std::vector< MuScleFitMuon > &muons, std::vector< GenMuonPair > &genPair, std::vector< std::pair< lorentzVector, lorentzVector > > &simPair, MuScleFitPlotter *plotter)
	Main method used to select muons of type specified by muonType_ from the collection specified by muonLabel_ and PATmuons_. More...

	~MuScleFitMuonSelector ()

Protected Member Functions
template<typename T >
std::vector< MuScleFitMuon >	fillMuonCollection (const std::vector< T > &tracks)
	Template function used to convert the muon collection to a vector of reco::LeafCandidate. More...

GenMuonPair	findGenMuFromRes (const reco::GenParticleCollection *genParticles)

GenMuonPair	findGenMuFromRes (const edm::HepMCProduct *evtMC)

std::pair< lorentzVector, lorentzVector >	findSimMuFromRes (const edm::Handle< edm::HepMCProduct > &evtMC, const edm::Handle< edm::SimTrackContainer > &simTracks)

void	selectGeneratedMuons (const edm::Handle< pat::CompositeCandidateCollection > &collAll, const std::vector< const pat::Muon * > &collMuSel, std::vector< GenMuonPair > &genPair, MuScleFitPlotter *plotter)
	For PATmuons the generator information is read directly from the PAT object. More...

void	selectGenSimMuons (const edm::Event &event, std::vector< GenMuonPair > &genPair, std::vector< std::pair< lorentzVector, lorentzVector > > &simPair, MuScleFitPlotter *plotter)

void	selectSimulatedMuons (const edm::Event &event, const bool ifHepMC, edm::Handle< edm::HepMCProduct > evtMC, std::vector< std::pair< lorentzVector, lorentzVector > > &simPair, MuScleFitPlotter *plotter)

bool	selGlobalMuon (const pat::Muon *aMuon)
	Apply the Onia cuts to select globalMuons. More...

bool	selTrackerMuon (const pat::Muon *aMuon)
	Apply the Onia cuts to select trackerMuons. More...

template<typename T >
void	takeSelectedMuonType (const T &muon, std::vector< reco::Track > &tracks)
	Template function used to extract the selected muon type from the muon collection. More...

Protected Attributes
const bool	compareToSimTracks_

const bool	debug_

const std::string	genParticlesName_

const edm::InputTag	muonLabel_

const int	muonType_

const bool	PATmuons_

const std::vector< int >	resfind_

const bool	sherpa_

const edm::InputTag	simTracksCollectionName_

const bool	speedup_

Static Protected Attributes
static const double	mMu2 = 0.011163612

static const unsigned int	motherPdgIdArray [6] = {23, 100553, 100553, 553, 100443, 443}

Detailed Description

Definition at line 29 of file MuScleFitMuonSelector.h.

Constructor & Destructor Documentation

MuScleFitMuonSelector::MuScleFitMuonSelector	(	const edm::InputTag &	muonLabel,
		const int	muonType,
		const bool	PATmuons,
		const std::vector< int > &	resfind,
		const bool	speedup,
		const std::string &	genParticlesName,
		const bool	compareToSimTracks,
		const edm::InputTag &	simTracksCollectionName,
		const bool	sherpa,
		const bool	debug
	)

inline

Definition at line 32 of file MuScleFitMuonSelector.h.

                                                  :
     muonLabel_(muonLabel),
     muonType_(muonType),
     PATmuons_(PATmuons),
     resfind_(resfind),
     speedup_(speedup),
     genParticlesName_(genParticlesName),
     compareToSimTracks_(compareToSimTracks),
     simTracksCollectionName_(simTracksCollectionName),
     sherpa_(sherpa),
     debug_(debug)
   {}

MuScleFitMuonSelector::~MuScleFitMuonSelector ( )

inline

Definition at line 48 of file MuScleFitMuonSelector.h.

References findGenMuFromRes(), findSimMuFromRes(), GenHFHadronMatcher_cfi::genParticles, getStatus1Muon(), getStatus3Muon(), extraflags_cff::muons, trackingPlots::plotter, selectGeneratedMuons(), selectGenSimMuons(), selectMuons(), selectSimulatedMuons(), selGlobalMuon(), selTrackerMuon(), and tkConvValidator_cfi::simTracks.

48 {}

Member Function Documentation

template<typename T >

std::vector<MuScleFitMuon> MuScleFitMuonSelector::fillMuonCollection ( const std::vector< T > & tracks )

inlineprotected

Template function used to convert the muon collection to a vector of reco::LeafCandidate.

Definition at line 90 of file MuScleFitMuonSelector.h.

References gather_cfg::cout, debug_, mMu2, RPCpg::mu, metsig::muon, extraflags_cff::muons, mathSSE::sqrt(), and HiIsolationCommonParameters_cff::track.

Referenced by selectMuons().

   {
     std::vector<MuScleFitMuon> muons;
     typename std::vector<T>::const_iterator track;
     for( track = tracks.begin(); track != tracks.end(); ++track ) {
       reco::Particle::LorentzVector mu;
       mu = reco::Particle::LorentzVector(track->px(),track->py(),track->pz(),
                      sqrt(track->p()*track->p() + mMu2));
       
       Double_t hitsTk(0), hitsMuon(0), ptError(0);
       if ( const reco::Muon* myMu = dynamic_cast<const reco::Muon*>(&(*track))  ){
     hitsTk =   myMu->innerTrack()->hitPattern().numberOfValidTrackerHits();
     hitsMuon = myMu->innerTrack()->hitPattern().numberOfValidMuonHits();
     ptError =  myMu->innerTrack()->ptError();
       }
       else if ( const pat::Muon* myMu = dynamic_cast<const pat::Muon*>(&(*track)) ) {
     hitsTk =   myMu->innerTrack()->hitPattern().numberOfValidTrackerHits();
     hitsMuon = myMu->innerTrack()->hitPattern().numberOfValidMuonHits();
     ptError =  myMu->innerTrack()->ptError();
       }
       else if (const reco::Track* myMu = dynamic_cast<const reco::Track*>(&(*track))){
     hitsTk =   myMu->hitPattern().numberOfValidTrackerHits();
     hitsMuon = myMu->hitPattern().numberOfValidMuonHits();
     ptError =  myMu->ptError();
       }
       
       MuScleFitMuon muon(mu,track->charge(),ptError,hitsTk,hitsMuon);
 
     if (debug_>0) {
       std::cout<<"[MuScleFitMuonSelector::fillMuonCollection] after MuScleFitMuon initialization"<<std::endl;
       std::cout<<"  muon = "<<muon<<std::endl;
     }
 
     muons.push_back(muon);
     }
     return muons;
   }

GenMuonPair MuScleFitMuonSelector::findGenMuFromRes ( const reco::GenParticleCollection * genParticles )

protected

Definition at line 420 of file MuScleFitMuonSelector.cc.

References funct::abs(), gather_cfg::cout, debug_, getStatus1Muon(), ires, GenMuonPair::motherId, motherPdgIdArray, GenMuonPair::mu1, GenMuonPair::mu2, reco::Candidate::p4(), and resfind_.

Referenced by selectGenSimMuons(), and ~MuScleFitMuonSelector().

 {
   // std::pair<lorentzVector,lorentzVector> muFromRes;
   GenMuonPair muFromRes;
 
   //Loop on generated particles
   if( debug_>0 ) std::cout << "Starting loop on " << genParticles->size() << " genParticles" << std::endl;
   for( reco::GenParticleCollection::const_iterator part=genParticles->begin(); part!=genParticles->end(); ++part ) {
     if (debug_>0) std::cout<<"genParticle has pdgId = "<<std::abs(part->pdgId())<<" and status = "<<part->status()<<std::endl;
     if (std::abs(part->pdgId())==13){// && part->status()==3) {
       bool fromRes = false;
       unsigned int motherPdgId = part->mother()->pdgId();
       if( debug_>0 ) {
     std::cout << "Found a muon with mother: " << motherPdgId << std::endl;
       }
       for( int ires = 0; ires < 6; ++ires ) {
     // if( motherPdgId == motherPdgIdArray[ires] && resfind_[ires] ) fromRes = true; // changed by EM 2015.07.30
     // begin of comment  
     // the list of resonances motherPdgIdArray does not contain the photon (PdgId = 21) while ~1% of the 
     // mu+mu- events in the range [50,120] GeV has a photon as the mother.
     // It needs to be fixed without spoiling the logic of the selection of different resonances
     // e.g. mixing up onia etc.
     // end of comment
     if( ( motherPdgId == motherPdgIdArray[ires] && resfind_[ires] ) || motherPdgId == 21 ) fromRes = true;
       }
       if(fromRes){
     if (debug_>0) std::cout<<"fromRes = true, motherPdgId = "<<motherPdgId<<std::endl;
     const reco::Candidate* status3Muon = &(*part);
     const reco::Candidate* status1Muon = getStatus1Muon(status3Muon);
     if(part->pdgId()==13) {
       if (status1Muon->p4().pt()!=0) muFromRes.mu1 = MuScleFitMuon(status1Muon->p4(),-1);
       else muFromRes.mu1 = MuScleFitMuon(status3Muon->p4(),-1);
       if( debug_>0 ) std::cout << "Found a genMuon - : " << muFromRes.mu1 << std::endl;
       //      muFromRes.first = (lorentzVector(status1Muon->p4().px(),status1Muon->p4().py(),
       //                       status1Muon->p4().pz(),status1Muon->p4().e()));
     }
     else {
       if (status1Muon->p4().pt()!=0) muFromRes.mu2 = MuScleFitMuon(status1Muon->p4(),+1);
       else muFromRes.mu2 = MuScleFitMuon(status3Muon->p4(),+1);
       if( debug_>0 ) std::cout << "Found a genMuon + : " << muFromRes.mu2 << std::endl;
       //      muFromRes.second = (lorentzVector(status1Muon->p4().px(),status1Muon->p4().py(),
       //                        status1Muon->p4().pz(),status1Muon->p4().e()));
     }
     muFromRes.motherId = motherPdgId; 
       }
     }
   }
   return muFromRes;
 }

GenMuonPair MuScleFitMuonSelector::findGenMuFromRes ( const edm::HepMCProduct * evtMC )

protected

Definition at line 377 of file MuScleFitMuonSelector.cc.

References funct::abs(), edm::HepMCProduct::GetEvent(), ires, GenMuonPair::motherId, motherPdgIdArray, GenMuonPair::mu1, GenMuonPair::mu2, resfind_, and sherpa_.

 {
   const HepMC::GenEvent* Evt = evtMC->GetEvent();
   GenMuonPair muFromRes;
   //Loop on generated particles
   for (HepMC::GenEvent::particle_const_iterator part=Evt->particles_begin();
        part!=Evt->particles_end(); part++) {
     if (std::abs((*part)->pdg_id())==13 && (*part)->status()==1) {
       bool fromRes = false;
       unsigned int motherPdgId = 0;
       for (HepMC::GenVertex::particle_iterator mother = (*part)->production_vertex()->particles_begin(HepMC::ancestors);
        mother != (*part)->production_vertex()->particles_end(HepMC::ancestors); ++mother) {
         motherPdgId = (*mother)->pdg_id();
 
         // For sherpa the resonance is not saved. The muons from the resonance can be identified
         // by having as mother a muon of status 3.
         if( sherpa_ ) {
           if( motherPdgId == 13 && (*mother)->status() == 3 ) fromRes = true;
         }
         else {
           for( int ires = 0; ires < 6; ++ires ) {
         if( motherPdgId == motherPdgIdArray[ires] && resfind_[ires] ) fromRes = true;
           }
         }
       }
       if(fromRes){
     if((*part)->pdg_id()==13) {
       //   muFromRes.first = (*part)->momentum();
       muFromRes.mu1 = (lorentzVector((*part)->momentum().px(),(*part)->momentum().py(),
                      (*part)->momentum().pz(),(*part)->momentum().e()));
     }
     else {
       muFromRes.mu2 = (lorentzVector((*part)->momentum().px(),(*part)->momentum().py(),
                      (*part)->momentum().pz(),(*part)->momentum().e()));
     }
     muFromRes.motherId = motherPdgId; 
       }
     }
   }
   return muFromRes;
 }

std::pair< lorentzVector, lorentzVector > MuScleFitMuonSelector::findSimMuFromRes	(	const edm::Handle< edm::HepMCProduct > &	evtMC,
		const edm::Handle< edm::SimTrackContainer > &	simTracks
	)

protected

Definition at line 470 of file MuScleFitMuonSelector.cc.

References funct::abs(), GenParticle::GenParticle, edm::HepMCProduct::GetEvent(), runTauDisplay::gp, ires, motherPdgIdArray, resfind_, and muonSimHitMatcherPSet::simTrack.

Referenced by selectSimulatedMuons(), and ~MuScleFitMuonSelector().

 {
   //Loop on simulated tracks
   std::pair<lorentzVector, lorentzVector> simMuFromRes;
   for( edm::SimTrackContainer::const_iterator simTrack=simTracks->begin(); simTrack!=simTracks->end(); ++simTrack ) {
     //Chose muons
     if (std::abs((*simTrack).type())==13) {
       //If tracks from IP than find mother
       if ((*simTrack).genpartIndex()>0) {
     HepMC::GenParticle* gp = evtMC->GetEvent()->barcode_to_particle ((*simTrack).genpartIndex());
         if( gp != nullptr ) {
 
           for (HepMC::GenVertex::particle_iterator mother = gp->production_vertex()->particles_begin(HepMC::ancestors);
                mother!=gp->production_vertex()->particles_end(HepMC::ancestors); ++mother) {
 
             bool fromRes = false;
             unsigned int motherPdgId = (*mother)->pdg_id();
             for( int ires = 0; ires < 6; ++ires ) {
               if( ( motherPdgId == motherPdgIdArray[ires] && resfind_[ires] ) || motherPdgId == 21 ) fromRes = true;
             }
             if( fromRes ) {
               if(gp->pdg_id() == 13)
                 simMuFromRes.first = lorentzVector(simTrack->momentum().px(),simTrack->momentum().py(),
                                                    simTrack->momentum().pz(),simTrack->momentum().e());
               else
                 simMuFromRes.second = lorentzVector(simTrack->momentum().px(),simTrack->momentum().py(),
                                                     simTrack->momentum().pz(),simTrack->momentum().e());
             }
           }
         }
         // else LogDebug("MuScleFitUtils") << "WARNING: no matching genParticle found for simTrack" << std::endl;
       }
     }
   }
   return simMuFromRes;
 }

const reco::Candidate * MuScleFitMuonSelector::getStatus1Muon ( const reco::Candidate * status3Muon )

Definition at line 10 of file MuScleFitMuonSelector.cc.

References reco::Candidate::daughter(), mps_fire::i, reco::Candidate::numberOfDaughters(), and reco::Candidate::pdgId().

Referenced by findGenMuFromRes(), and ~MuScleFitMuonSelector().

                                                                      {
   const reco::Candidate* tempMuon = status3Muon;
   //  bool lastCopy = (static_cast<const reco::GenParticle*>(tempMuon))->isLastCopy();                      //  isLastCopy() likely not enough robust
   bool isPromptFinalState = static_cast<const reco::GenParticle*>(tempMuon)->isPromptFinalState();        //  pre-CMSSW_74X code: int status = tempStatus1Muon->status();
   while(tempMuon == nullptr || tempMuon->numberOfDaughters()!=0){
     if ( isPromptFinalState ) break;                                                     //  pre-CMSSW_74X code: if (status == 1) break;
     //std::vector<const reco::Candidate*> daughters;
     for (unsigned int i=0; i<tempMuon->numberOfDaughters(); ++i){
       if ( tempMuon->daughter(i)->pdgId()==tempMuon->pdgId() ){
     tempMuon = tempMuon->daughter(i);
     isPromptFinalState = static_cast<const reco::GenParticle*>(tempMuon)->isPromptFinalState();      //  pre-CMSSW_74X code: status = tempStatus1Muon->status();
     break;
       }else continue;
     }//for loop
   }//while loop
   
   return tempMuon;
 }

const reco::Candidate * MuScleFitMuonSelector::getStatus3Muon ( const reco::Candidate * status3Muon )

Definition at line 30 of file MuScleFitMuonSelector.cc.

References reco::Candidate::daughter(), mps_fire::i, reco::Candidate::numberOfDaughters(), and reco::Candidate::pdgId().

Referenced by ~MuScleFitMuonSelector().

                                                                      {
   const reco::Candidate* tempMuon = status3Muon;
   bool lastCopy = static_cast<const reco::GenParticle*>(tempMuon)->isLastCopyBeforeFSR();        //  pre-CMSSW_74X code: int status = tempStatus1Muon->status();
   while(tempMuon == nullptr || tempMuon->numberOfDaughters()!=0){
     if ( lastCopy ) break;                                                      //  pre-CMSSW_74X code: if (status == 3) break;
     //std::vector<const reco::Candidate*> daughters;
     for (unsigned int i=0; i<tempMuon->numberOfDaughters(); ++i){
       if ( tempMuon->daughter(i)->pdgId()==tempMuon->pdgId() ){
     tempMuon = tempMuon->daughter(i);
     lastCopy = static_cast<const reco::GenParticle*>(tempMuon)->isLastCopyBeforeFSR();  //  pre-CMSSW_74X code: status = tempStatus1Muon->status();
     break;
       }else continue;
     }//for loop
   }//while loop
   
   return tempMuon;
 }

void MuScleFitMuonSelector::selectGeneratedMuons	(	const edm::Handle< pat::CompositeCandidateCollection > &	collAll,
		const std::vector< const pat::Muon * > &	collMuSel,
		std::vector< GenMuonPair > &	genPair,
		MuScleFitPlotter *	plotter
	)

protected

For PATmuons the generator information is read directly from the PAT object.

Definition at line 254 of file MuScleFitMuonSelector.cc.

References gather_cfg::cout, debug_, MuScleFitPlotter::fillGen(), edm::Ref< C, T, F >::get(), edm::Ref< C, T, F >::isAvailable(), trackerHitRTTI::isMatched(), objects.autophobj::motherId, and nullptr.

Referenced by ~MuScleFitMuonSelector().

 {
   reco::GenParticleCollection genPatParticles{};
 
   //explicitly for JPsi but can be adapted!!!!!
   for(std::vector<pat::CompositeCandidate>::const_iterator it=collAll->begin();
       it!=collAll->end();++it) {
     reco::GenParticleRef genJpsi = it->genParticleRef();
     bool isMatched = (genJpsi.isAvailable() && genJpsi->pdgId() == 443);  
     if (isMatched){
       genPatParticles.push_back(*genJpsi.get());
     }
   }
 
   if(collMuSel.size()==2) {
     reco::GenParticleRef genMu1 = collMuSel[0]->genParticleRef();
     reco::GenParticleRef genMu2 = collMuSel[1]->genParticleRef();
     bool isMuMatched = (genMu1.isAvailable() && genMu2.isAvailable() && 
             genMu1->pdgId()*genMu2->pdgId() == -169);
     if (isMuMatched) {
       genPatParticles.push_back(*genMu1.get());
       genPatParticles.push_back(*genMu2.get());
 
       unsigned int motherId = 0;
       if( genMu1->mother() != nullptr ) {
      motherId = genMu1->mother()->pdgId();
       }
       if(genMu1->pdgId()==13)
     genPair.push_back(GenMuonPair(genMu1.get()->p4(), genMu2.get()->p4(), motherId));
       else
     // genPair.push_back(std::make_pair(genMu2.get()->p4(),genMu1.get()->p4()) );
     genPair.push_back(GenMuonPair(genMu2.get()->p4(), genMu1.get()->p4(), motherId));
 
       plotter->fillGen(genPatParticles, true);
 
       if (debug_>0) std::cout << "Found genParticles in PAT" << std::endl;
     }
     else {
       std::cout << "No recomuon selected so no access to generated info"<<std::endl;
       // Fill it in any case, otherwise it will not be in sync with the event number
       // genPair.push_back( std::make_pair( lorentzVector(0.,0.,0.,0.), lorentzVector(0.,0.,0.,0.) ) );    
       genPair.push_back( GenMuonPair(lorentzVector(0.,0.,0.,0.), lorentzVector(0.,0.,0.,0.), 0 ) );    
     }
   }
   else{
     std::cout << "No recomuon selected so no access to generated info"<<std::endl;
     // Fill it in any case, otherwise it will not be in sync with the event number
     // genPair.push_back( std::make_pair( lorentzVector(0.,0.,0.,0.), lorentzVector(0.,0.,0.,0.) ) );
     genPair.push_back( GenMuonPair(lorentzVector(0.,0.,0.,0.), lorentzVector(0.,0.,0.,0.), 0 ) );    
   }
   if(debug_>0) {
     std::cout << "genParticles:" << std::endl;
     // std::cout << genPair.back().first << " , " << genPair.back().second << std::endl;
     std::cout << genPair.back().mu1 << " , " << genPair.back().mu2 << std::endl;
   }
 }

void MuScleFitMuonSelector::selectGenSimMuons	(	const edm::Event &	event,
		std::vector< GenMuonPair > &	genPair,
		std::vector< std::pair< lorentzVector, lorentzVector > > &	simPair,
		MuScleFitPlotter *	plotter
	)

protected

Definition at line 314 of file MuScleFitMuonSelector.cc.

References gather_cfg::cout, debug_, MuScleFitPlotter::fillGen(), findGenMuFromRes(), GenHFHadronMatcher_cfi::genParticles, genParticlesName_, edm::HandleBase::isValid(), edm::Handle< T >::product(), selectSimulatedMuons(), and sherpa_.

Referenced by selectMuons(), and ~MuScleFitMuonSelector().

 {  
   // Find and store in histograms the generated and simulated resonance and muons
   // ----------------------------------------------------------------------------
   edm::Handle<edm::HepMCProduct> evtMC;
   edm::Handle<reco::GenParticleCollection> genParticles;
 
   // Fill gen information only in the first loop
   bool ifHepMC=false;
 
   event.getByLabel( genParticlesName_, evtMC );
   event.getByLabel( genParticlesName_, genParticles );
   if( evtMC.isValid() ) {
     genPair.push_back( findGenMuFromRes(evtMC.product()) );
     plotter->fillGen(*evtMC, sherpa_);
     ifHepMC = true;
     if (debug_>0) std::cout << "Found hepMC" << std::endl;
   }
   else if( genParticles.isValid() ) {
     genPair.push_back( findGenMuFromRes(genParticles.product()) );
     plotter->fillGen(*genParticles);
     if (debug_>0) std::cout << "Found genParticles" << std::endl;
   }
   else {
     std::cout<<"ERROR "<<"non generation info and speedup true!!!!!!!!!!!!"<<std::endl;
     // Fill it in any case, otherwise it will not be in sync with the event number
     // genPair.push_back( std::make_pair( lorentzVector(0.,0.,0.,0.), lorentzVector(0.,0.,0.,0.) ) );
     genPair.push_back( GenMuonPair( lorentzVector(0.,0.,0.,0.), lorentzVector(0.,0.,0.,0.), 0 ) );
   }
   if(debug_>0) {
     std::cout << "genParticles:" << std::endl;
     std::cout << genPair.back().mu1 << " , " << genPair.back().mu2 << std::endl;
   }
   selectSimulatedMuons(event, ifHepMC, evtMC, simPair, plotter);
 }

void MuScleFitMuonSelector::selectMuons	(	const edm::Event &	event,
		std::vector< MuScleFitMuon > &	muons,
		std::vector< GenMuonPair > &	genPair,
		std::vector< std::pair< lorentzVector, lorentzVector > > &	simPair,
		MuScleFitPlotter *	plotter
	)

Main method used to select muons of type specified by muonType_ from the collection specified by muonLabel_ and PATmuons_.

Definition at line 90 of file MuScleFitMuonSelector.cc.

References allMuons_cfi::allMuons, mergedMuons_cfi::caloMuons, reco::LeafCandidate::charge(), gather_cfg::cout, reco::CompositeCandidate::daughter(), fillMuonCollection(), MuScleFitPlotter::fillRec(), pat::Muon::innerTrack(), reco::Muon::isGlobalMuon(), reco::Muon::isTrackerMuon(), edm::HandleBase::isValid(), metsig::muon, muonLabel_, muonType_, PATmuons_, selectGenSimMuons(), selGlobalMuon(), selTrackerMuon(), speedup_, takeSelectedMuonType(), l1t::tracks, and pat::PATObject< ObjectType >::userFloat().

Referenced by ~MuScleFitMuonSelector().

 {
   edm::Handle<pat::CompositeCandidateCollection > collAll;
   event.getByLabel("onia2MuMuPatTrkTrk", collAll);
   if(!collAll.isValid()) {
     edm::LogWarning("MuScleFitUtils") << "J/psi not present in event!";
   }
   std::vector<const pat::Muon*> collMuSel;
 
   //================onia cuts===========================/
 
   if (muonType_ <= -1 && PATmuons_) {
     std::vector<const pat::CompositeCandidate*> collSelGG;
     std::vector<const pat::CompositeCandidate*> collSelGT;
     std::vector<const pat::CompositeCandidate*> collSelTT;
     if (collAll.isValid()) {
 
       for (std::vector<pat::CompositeCandidate>::const_iterator it=collAll->begin();
         it!=collAll->end(); ++it) {
 
         const pat::CompositeCandidate* cand = &(*it);
         // cout << "Now checking candidate of type " << theJpsiCat << " with pt = " << cand->pt() << endl;
         const pat::Muon* muon1 = dynamic_cast<const pat::Muon*>(cand->daughter("muon1"));
         const pat::Muon* muon2 = dynamic_cast<const pat::Muon*>(cand->daughter("muon2"));
 
         if ((muon1->charge() * muon2->charge())>0)
           continue;
         // global + global?
         if (muon1->isGlobalMuon() && muon2->isGlobalMuon()) {
           if (selGlobalMuon(muon1) && selGlobalMuon(muon2) && cand->userFloat("vProb") > 0.001) {
             collSelGG.push_back(cand);
             continue;
           }
         }
         // global + tracker? (x2)    
         if (muon1->isGlobalMuon() && muon2->isTrackerMuon()) {
           if (selGlobalMuon(muon1) &&  selTrackerMuon(muon2) && cand->userFloat("vProb") > 0.001) {
             collSelGT.push_back(cand);
             continue;
           }
         }
         if (muon2->isGlobalMuon() && muon1->isTrackerMuon()) {
           if (selGlobalMuon(muon2) && selTrackerMuon(muon1) && cand->userFloat("vProb") > 0.001) {
             collSelGT.push_back(cand);
             continue;
           }
         }
         // tracker + tracker?  
         if (muon1->isTrackerMuon() && muon2->isTrackerMuon()) {
           if (selTrackerMuon(muon1) && selTrackerMuon(muon2) && cand->userFloat("vProb") > 0.001) {
             collSelTT.push_back(cand);
             continue;
           }
         }
       }
     }
     // Split them in independent collections if using muonType_ == -2, -3 or -4. Take them all if muonType_ == -1.
     std::vector<reco::Track> tracks;
     if (!collSelGG.empty()){
       //CHECK THAT THEY ARE ORDERED BY PT !!!!!!!!!!!!!!!!!!!!!!!
       const pat::Muon* muon1 = dynamic_cast<const pat::Muon*>(collSelGG[0]->daughter("muon1"));
       const pat::Muon* muon2 = dynamic_cast<const pat::Muon*>(collSelGG[0]->daughter("muon2"));
       if (muonType_ == -1 || muonType_ == -2) {
         tracks.push_back(*(muon1->innerTrack()));
         tracks.push_back(*(muon2->innerTrack()));
         collMuSel.push_back(muon1);
         collMuSel.push_back(muon2);
       }
     }
     else if (collSelGG.empty() && !collSelGT.empty()){
       //CHECK THAT THEY ARE ORDERED BY PT !!!!!!!!!!!!!!!!!!!!!!!
       const pat::Muon* muon1 = dynamic_cast<const pat::Muon*>(collSelGT[0]->daughter("muon1"));
       const pat::Muon* muon2 = dynamic_cast<const pat::Muon*>(collSelGT[0]->daughter("muon2"));
       if (muonType_ == -1 || muonType_ == -3) {
         tracks.push_back(*(muon1->innerTrack()));
         tracks.push_back(*(muon2->innerTrack()));
         collMuSel.push_back(muon1);
         collMuSel.push_back(muon2);
       }
     }
     else if (collSelGG.empty() && collSelGT.empty() && !collSelTT.empty()){
       //CHECK THAT THEY ARE ORDERED BY PT !!!!!!!!!!!!!!!!!!!!!!!
       const pat::Muon* muon1 = dynamic_cast<const pat::Muon*>(collSelTT[0]->daughter("muon1"));
       const pat::Muon* muon2 = dynamic_cast<const pat::Muon*>(collSelTT[0]->daughter("muon2"));
       if (muonType_ == -1 || muonType_ == -4) {
         tracks.push_back(*(muon1->innerTrack()));
         tracks.push_back(*(muon2->innerTrack()));
         collMuSel.push_back(muon1);
         collMuSel.push_back(muon2);
       }
     }
     if (tracks.size() != 2 && !tracks.empty()){
       std::cout<<"ERROR strange number of muons selected by onia cuts!"<<std::endl;
       abort();
     }
     muons = fillMuonCollection(tracks);
   }
   else if ((muonType_<4 && muonType_>=0) || muonType_>=10) { // Muons (glb,sta,trk)
     std::vector<reco::Track> tracks;
     if (PATmuons_ == true) {
       edm::Handle<pat::MuonCollection> allMuons;
       event.getByLabel(muonLabel_, allMuons);
       if (muonType_ == 0) {
         // Take directly the muon
         muons = fillMuonCollection(*allMuons);
       }
       else {
         for (std::vector<pat::Muon>::const_iterator muon = allMuons->begin(); muon != allMuons->end(); ++muon) {
           //std::cout<<"pat muon is global "<<muon->isGlobalMuon()<<std::endl;
           takeSelectedMuonType(muon, tracks);
         }
         muons = fillMuonCollection(tracks);
       }
     }
     else {
       edm::Handle<reco::MuonCollection> allMuons;
       event.getByLabel(muonLabel_, allMuons);
       if (muonType_ == 0) {
         // Take directly the muon
         muons = fillMuonCollection(*allMuons);
       }
       else {
         for (std::vector<reco::Muon>::const_iterator muon = allMuons->begin(); muon != allMuons->end(); ++muon) {
           takeSelectedMuonType(muon, tracks);
         }
         muons = fillMuonCollection(tracks);
       }
     }
   }
   else if (muonType_==4){  //CaloMuons
     edm::Handle<reco::CaloMuonCollection> caloMuons;
     event.getByLabel(muonLabel_, caloMuons);
     std::vector<reco::Track> tracks;
     for (std::vector<reco::CaloMuon>::const_iterator muon = caloMuons->begin(); muon != caloMuons->end(); ++muon){
       tracks.push_back(*(muon->track()));
     }
     muons = fillMuonCollection(tracks);
   }
 
   else if (muonType_==5) { // Inner tracker tracks
     edm::Handle<reco::TrackCollection> tracks;
     event.getByLabel(muonLabel_, tracks);
     muons = fillMuonCollection(*tracks);
   }
   plotter->fillRec(muons);
 
   // Generation and simulation part
   if (speedup_ == false)
   {
     if (PATmuons_) {
       // EM 2015.04.02 temporary fix to run on MINIAODSIM (which contains PAT muons) but not the "onia2MuMuPatTrkTrk" collection   
       // selectGeneratedMuons(collAll, collMuSel, genPair, plotter);
       selectGenSimMuons(event, genPair, simPair, plotter);
     }
     else {
       selectGenSimMuons(event, genPair, simPair, plotter);
     }
   }
 }

void MuScleFitMuonSelector::selectSimulatedMuons	(	const edm::Event &	event,
		const bool	ifHepMC,
		edm::Handle< edm::HepMCProduct >	evtMC,
		std::vector< std::pair< lorentzVector, lorentzVector > > &	simPair,
		MuScleFitPlotter *	plotter
	)

protected

Definition at line 353 of file MuScleFitMuonSelector.cc.

References gather_cfg::cout, MuScleFitPlotter::fillGenSim(), MuScleFitPlotter::fillSim(), findSimMuFromRes(), edm::HandleBase::isValid(), tkConvValidator_cfi::simTracks, and simTracksCollectionName_.

Referenced by selectGenSimMuons(), and ~MuScleFitMuonSelector().

 {
   edm::Handle<edm::SimTrackContainer> simTracks;
   bool simTracksFound = false;
   event.getByLabel(simTracksCollectionName_, simTracks);
   if( simTracks.isValid() ) {
     plotter->fillSim(simTracks);
     if(ifHepMC) {
       simPair.push_back( findSimMuFromRes(evtMC, simTracks) );
       simTracksFound = true;
       plotter->fillGenSim(evtMC,simTracks);
     }
   }
   else {
     std::cout << "SimTracks not existent" << std::endl;
   }
   if( !simTracksFound ) {
     simPair.push_back( std::make_pair( lorentzVector(0.,0.,0.,0.), lorentzVector(0.,0.,0.,0.) ) );
   }
 }

bool MuScleFitMuonSelector::selGlobalMuon ( const pat::Muon * aMuon )

protected

Apply the Onia cuts to select globalMuons.

Definition at line 50 of file MuScleFitMuonSelector.cc.

References funct::abs(), pat::Muon::globalTrack(), pat::Muon::innerTrack(), pat::Muon::muonID(), reco::HitPattern::numberOfValidMuonHits(), AlCaHLTBitMon_ParallelJobs::p, reco::HitPattern::pixelLayersWithMeasurement(), and lumiQueryAPI::q.

Referenced by selectMuons(), and ~MuScleFitMuonSelector().

 {
   reco::TrackRef iTrack = aMuon->innerTrack();
   const reco::HitPattern& p = iTrack->hitPattern();
 
   reco::TrackRef gTrack = aMuon->globalTrack();
   const reco::HitPattern& q = gTrack->hitPattern();
 
   return (//isMuonInAccept(aMuon) &&// no acceptance cuts!
     iTrack->found() > 11 &&
     gTrack->chi2()/gTrack->ndof() < 20.0 &&
     q.numberOfValidMuonHits() > 0 &&
     iTrack->chi2()/iTrack->ndof() < 4.0 &&
     aMuon->muonID("TrackerMuonArbitrated") &&
     aMuon->muonID("TMLastStationAngTight") &&
     p.pixelLayersWithMeasurement() > 1 &&
     std::abs(iTrack->dxy()) < 3.0 &&  //should be done w.r.t. PV!
     std::abs(iTrack->dz()) < 15.0 //should be done w.r.t. PV!
   );
 }

bool MuScleFitMuonSelector::selTrackerMuon ( const pat::Muon * aMuon )

protected

Apply the Onia cuts to select trackerMuons.

Definition at line 71 of file MuScleFitMuonSelector.cc.

References funct::abs(), pat::Muon::innerTrack(), pat::Muon::muonID(), AlCaHLTBitMon_ParallelJobs::p, and reco::HitPattern::pixelLayersWithMeasurement().

Referenced by selectMuons(), and ~MuScleFitMuonSelector().

 {
   reco::TrackRef iTrack = aMuon->innerTrack();
   const reco::HitPattern& p = iTrack->hitPattern();
 
   return (//isMuonInAccept(aMuon) // no acceptance cuts!
     iTrack->found() > 11 &&
     iTrack->chi2()/iTrack->ndof() < 4.0 &&
     aMuon->muonID("TrackerMuonArbitrated") &&
     aMuon->muonID("TMLastStationAngTight") &&
     p.pixelLayersWithMeasurement() > 1 &&
     std::abs(iTrack->dxy()) < 3.0 && //should be done w.r.t. PV!
     std::abs(iTrack->dz()) < 15.0 //should be done w.r.t. PV!
   );
 }

template<typename T >

void MuScleFitMuonSelector::takeSelectedMuonType	(	const T &	muon,
		std::vector< reco::Track > &	tracks
	)

inlineprotected

Template function used to extract the selected muon type from the muon collection.

Definition at line 130 of file MuScleFitMuonSelector.h.

References muonType_.

Referenced by selectMuons().

   {
     // std::cout<<"muon "<<muon->isGlobalMuon()<<muon->isStandAloneMuon()<<muon->isTrackerMuon()<<std::endl;
     //NNBB: one muon can be of many kinds at once but with the muonType_ we are sure
     // to avoid double counting of the same muon
     if(muon->isGlobalMuon() && muonType_==1)
       tracks.push_back(*(muon->globalTrack()));
     else if(muon->isStandAloneMuon() && muonType_==2)
       tracks.push_back(*(muon->outerTrack()));
     else if(muon->isTrackerMuon() && muonType_==3)
       tracks.push_back(*(muon->innerTrack()));
 
     else if( muonType_ == 10 && !(muon->isStandAloneMuon()) ) //particular case!!
       tracks.push_back(*(muon->innerTrack()));
     else if( muonType_ == 11 && muon->isGlobalMuon() )
       tracks.push_back(*(muon->innerTrack()));
     else if( muonType_ == 13 && muon->isTrackerMuon() )
       tracks.push_back(*(muon->innerTrack()));
   }