Inheritance diagram for PFTauPrimaryVertexProducer:

Classes
struct	DiscCutPair

Public Types
enum	Alg { useInputPV =0, useFontPV }

typedef std::vector< DiscCutPair * >	DiscCutPairVec

Public Types inherited from edm::stream::EDProducer<>
typedef CacheContexts< T... >	CacheTypes

typedef CacheTypes::GlobalCache	GlobalCache

typedef AbilityChecker< T... >	HasAbility

typedef CacheTypes::LuminosityBlockCache	LuminosityBlockCache

typedef LuminosityBlockContextT< LuminosityBlockCache, RunCache, GlobalCache >	LuminosityBlockContext

typedef CacheTypes::LuminosityBlockSummaryCache	LuminosityBlockSummaryCache

typedef CacheTypes::RunCache	RunCache

typedef RunContextT< RunCache, GlobalCache >	RunContext

typedef CacheTypes::RunSummaryCache	RunSummaryCache

Public Member Functions
	PFTauPrimaryVertexProducer (const edm::ParameterSet &iConfig)

void	produce (edm::Event &, const edm::EventSetup &) override

	~PFTauPrimaryVertexProducer () override

Public Member Functions inherited from edm::stream::EDProducer<>
	EDProducer ()=default

bool	hasAbilityToProduceInLumis () const final

bool	hasAbilityToProduceInRuns () const final

Private Attributes
int	Algorithm_

edm::InputTag	beamSpotTag_

edm::EDGetTokenT< reco::BeamSpot >	beamSpotToken_

std::unique_ptr< StringCutObjectSelector< reco::PFTau > >	cut_

DiscCutPairVec	discriminators_

edm::InputTag	ElectronTag_

edm::EDGetTokenT< std::vector< reco::Electron > >	ElectronToken_

edm::InputTag	MuonTag_

edm::EDGetTokenT< std::vector< reco::Muon > >	MuonToken_

edm::InputTag	PFTauTag_

edm::EDGetTokenT< std::vector< reco::PFTau > >	PFTauToken_

edm::InputTag	PVTag_

edm::EDGetTokenT< reco::VertexCollection >	PVToken_

edm::ParameterSet	qualityCutsPSet_

bool	RemoveElectronTracks_

bool	RemoveMuonTracks_

bool	useBeamSpot_

bool	useSelectedTaus_

std::unique_ptr< tau::RecoTauVertexAssociator >	vertexAssociator_

Detailed Description

Definition at line 58 of file PFTauPrimaryVertexProducer.cc.

Member Typedef Documentation

typedef std::vector<DiscCutPair*> PFTauPrimaryVertexProducer::DiscCutPairVec

Definition at line 70 of file PFTauPrimaryVertexProducer.cc.

Member Enumeration Documentation

enum PFTauPrimaryVertexProducer::Alg

Enumerator
useInputPV
useFontPV

Definition at line 60 of file PFTauPrimaryVertexProducer.cc.

60 {useInputPV=0, useFontPV};

PFTauPrimaryVertexProducer::useFontPV

Definition: PFTauPrimaryVertexProducer.cc:60

PFTauPrimaryVertexProducer::useInputPV

Definition: PFTauPrimaryVertexProducer.cc:60

Constructor & Destructor Documentation

PFTauPrimaryVertexProducer::PFTauPrimaryVertexProducer ( const edm::ParameterSet & iConfig )

explicit

Definition at line 98 of file PFTauPrimaryVertexProducer.cc.

References PFTauPrimaryVertexProducer::DiscCutPair::cut_, cut_, PFTauPrimaryVertexProducer::DiscCutPair::cutFormula_, pfDeepCMVADiscriminatorsJetTags_cfi::discriminators, discriminators_, edm::ParameterSet::exists(), edm::ParameterSet::getParameter(), PFTauPrimaryVertexProducer::DiscCutPair::inputToken_, muonDTDigis_cfi::pset, qualityCutsPSet_, AlCaHLTBitMon_QueryRunRegistry::string, and vertexAssociator_.

                                                                                     :
   PFTauTag_(iConfig.getParameter<edm::InputTag>("PFTauTag")),
   PFTauToken_(consumes<std::vector<reco::PFTau> >(PFTauTag_)),
   ElectronTag_(iConfig.getParameter<edm::InputTag>("ElectronTag")),
   ElectronToken_(consumes<std::vector<reco::Electron> >(ElectronTag_)),
   MuonTag_(iConfig.getParameter<edm::InputTag>("MuonTag")),
   MuonToken_(consumes<std::vector<reco::Muon> >(MuonTag_)),
   PVTag_(iConfig.getParameter<edm::InputTag>("PVTag")),
   PVToken_(consumes<reco::VertexCollection>(PVTag_)),
   beamSpotTag_(iConfig.getParameter<edm::InputTag>("beamSpot")),
   beamSpotToken_(consumes<reco::BeamSpot>(beamSpotTag_)),
   Algorithm_(iConfig.getParameter<int>("Algorithm")),
   qualityCutsPSet_(iConfig.getParameter<edm::ParameterSet>("qualityCuts")),
   useBeamSpot_(iConfig.getParameter<bool>("useBeamSpot")),
   useSelectedTaus_(iConfig.getParameter<bool>("useSelectedTaus")),
   RemoveMuonTracks_(iConfig.getParameter<bool>("RemoveMuonTracks")),
   RemoveElectronTracks_(iConfig.getParameter<bool>("RemoveElectronTracks"))
 {
   std::vector<edm::ParameterSet> discriminators =iConfig.getParameter<std::vector<edm::ParameterSet> >("discriminators");
   // Build each of our cuts
   for(auto const& pset : discriminators) {
     DiscCutPair* newCut = new DiscCutPair();
     newCut->inputToken_ =consumes<reco::PFTauDiscriminator>(pset.getParameter<edm::InputTag>("discriminator"));
 
     if ( pset.existsAs<std::string>("selectionCut") ) newCut->cutFormula_ = new TFormula("selectionCut", pset.getParameter<std::string>("selectionCut").data());
     else newCut->cut_ = pset.getParameter<double>("selectionCut");
     discriminators_.push_back(newCut);
   }
   // Build a string cut if desired
   if (iConfig.exists("cut")) cut_.reset(new StringCutObjectSelector<reco::PFTau>(iConfig.getParameter<std::string>( "cut" )));
   produces<edm::AssociationVector<PFTauRefProd, std::vector<reco::VertexRef> > >();
   produces<VertexCollection>("PFTauPrimaryVertices"); 
 
   vertexAssociator_.reset(new tau::RecoTauVertexAssociator(qualityCutsPSet_,consumesCollector()));
 }

PFTauPrimaryVertexProducer::~PFTauPrimaryVertexProducer ( )

override

Definition at line 136 of file PFTauPrimaryVertexProducer.cc.

136 {}

Member Function Documentation

void PFTauPrimaryVertexProducer::produce	(	edm::Event &	iEvent,
		const edm::EventSetup &	iSetup
	)

override

Definition at line 138 of file PFTauPrimaryVertexProducer.cc.

References Algorithm_, ecalDrivenElectronSeedsParameters_cff::beamSpot, beamSpotToken_, TransientTrackBuilder::build(), egammaForCoreTracking_cff::cands, cut_, DEFINE_FWK_MODULE, TtSemiLepEvtBuilder_cfi::disc, listHistos::discr, discriminators_, nanoDQM_cff::Electron, ElectronToken_, edm::EventSetup::get(), edm::Event::getByToken(), edm::Event::getRefBeforePut(), edm::Ref< C, T, F >::isNonnull(), edm::HandleBase::isValid(), electrons_cff::matched, eostools::move(), Mu, MuonToken_, TriggerAnalyzer::passed, PFTauToken_, edm::Event::put(), nanoDQM_cfi::PV, PVToken_, RemoveElectronTracks_, RemoveMuonTracks_, AdaptiveVertexFitter::setWeightThreshold(), metsig::tau, reco::Vertex::tracks_begin(), reco::Vertex::tracks_end(), useBeamSpot_, useFontPV, useInputPV, useSelectedTaus_, AdaptiveVertexFitter::vertex(), and vertexAssociator_.

                                                                                     {
   // Obtain Collections
   edm::ESHandle<TransientTrackBuilder> transTrackBuilder;
   iSetup.get<TransientTrackRecord>().get("TransientTrackBuilder",transTrackBuilder);
   
   edm::Handle<std::vector<reco::PFTau> > Tau;
   iEvent.getByToken(PFTauToken_,Tau);
 
   edm::Handle<std::vector<reco::Electron> > Electron;
   iEvent.getByToken(ElectronToken_,Electron);
 
   edm::Handle<std::vector<reco::Muon> > Mu;
   iEvent.getByToken(MuonToken_,Mu);
 
   edm::Handle<reco::VertexCollection > PV;
   iEvent.getByToken(PVToken_,PV);
 
   edm::Handle<reco::BeamSpot> beamSpot;
   iEvent.getByToken(beamSpotToken_,beamSpot);
 
   // Set Association Map
   auto AVPFTauPV = std::make_unique<edm::AssociationVector<PFTauRefProd, std::vector<reco::VertexRef>>>(PFTauRefProd(Tau));
   auto VertexCollection_out = std::make_unique<VertexCollection>();
   reco::VertexRefProd VertexRefProd_out = iEvent.getRefBeforePut<reco::VertexCollection>("PFTauPrimaryVertices");
 
   // Load each discriminator
   for(auto& disc : discriminators_) {
     edm::Handle<reco::PFTauDiscriminator> discr;
     iEvent.getByToken(disc->inputToken_, discr);
     disc->discr_ = &(*discr);
   }
 
   // Set event for VerexAssociator if needed
   if(useInputPV==Algorithm_)
     vertexAssociator_->setEvent(iEvent);
 
   // For each Tau Run Algorithim 
   if(Tau.isValid()){
     for(reco::PFTauCollection::size_type iPFTau = 0; iPFTau < Tau->size(); iPFTau++) {
       reco::PFTauRef tau(Tau, iPFTau);
       reco::Vertex thePV;
       if(useInputPV==Algorithm_){
     thePV =(*( vertexAssociator_->associatedVertex(*tau)));
       }
       else if(useFontPV==Algorithm_){
     thePV=PV->front();
       }
       // Check if it passed all the discrimiantors
       bool passed(true); 
       for(auto const& disc : discriminators_) {
         // Check this discriminator passes
     bool passedDisc = true;
     if ( disc->cutFormula_ )passedDisc = (disc->cutFormula_->Eval((*disc->discr_)[tau]) > 0.5);
     else passedDisc = ((*disc->discr_)[tau] > disc->cut_);
         if ( !passedDisc ){passed = false; break;}
       }
       if (passed && cut_.get()){passed = (*cut_)(*tau);}
       if (passed){
     std::vector<reco::TrackBaseRef> SignalTracks;
     for(reco::PFTauCollection::size_type jPFTau = 0; jPFTau < Tau->size(); jPFTau++) {
       if(useSelectedTaus_ || iPFTau==jPFTau){
         reco::PFTauRef RefPFTau(Tau, jPFTau);
         // Get tracks from PFTau daugthers
         const std::vector<edm::Ptr<reco::PFCandidate> > cands = RefPFTau->signalPFChargedHadrCands(); 
         for (std::vector<edm::Ptr<reco::PFCandidate> >::const_iterator iter = cands.begin(); iter!=cands.end(); iter++){
           if(iter->get()->trackRef().isNonnull()) SignalTracks.push_back(reco::TrackBaseRef(iter->get()->trackRef()));
           else if(iter->get()->gsfTrackRef().isNonnull()){SignalTracks.push_back(reco::TrackBaseRef(((iter)->get()->gsfTrackRef())));}
         }
       }
     }
     // Get Muon tracks
     if(RemoveMuonTracks_){
 
       if(Mu.isValid()) {
         for(reco::MuonCollection::size_type iMuon = 0; iMuon< Mu->size(); iMuon++){
           reco::MuonRef RefMuon(Mu, iMuon);
           if(RefMuon->track().isNonnull()) SignalTracks.push_back(reco::TrackBaseRef(RefMuon->track()));
         }
       }
     }
     // Get Electron Tracks
     if(RemoveElectronTracks_){
       if(Electron.isValid()) {
         for(reco::ElectronCollection::size_type iElectron = 0; iElectron<Electron->size(); iElectron++){
           reco::ElectronRef RefElectron(Electron, iElectron);
           if(RefElectron->track().isNonnull()) SignalTracks.push_back(reco::TrackBaseRef(RefElectron->track()));
         }
       }
     }
     // Get Non-Tau tracks
     reco::TrackCollection nonTauTracks;
     for(std::vector<reco::TrackBaseRef>::const_iterator vtxTrkRef=thePV.tracks_begin();vtxTrkRef<thePV.tracks_end();vtxTrkRef++){
       bool matched = false;
       for (unsigned int sigTrk = 0; sigTrk < SignalTracks.size(); sigTrk++) {
         if ( (*vtxTrkRef) == SignalTracks[sigTrk] ) {
           matched = true;
         }
       }
       if ( !matched ) nonTauTracks.push_back(**vtxTrkRef);
     }   
     // Refit the vertex
     TransientVertex transVtx;
     std::vector<reco::TransientTrack> transTracks;
     for (reco::TrackCollection::iterator iter=nonTauTracks.begin(); iter!=nonTauTracks.end(); ++iter){
       transTracks.push_back(transTrackBuilder->build(*iter));
     }
     bool FitOk(true);
     if ( transTracks.size() >= 2 ) {
       AdaptiveVertexFitter avf;
       avf.setWeightThreshold(0.1); //weight per track. allow almost every fit, else --> exception
       try {
         if ( !useBeamSpot_ ){
           transVtx = avf.vertex(transTracks);
         } else {
           transVtx = avf.vertex(transTracks, *beamSpot);
         }
       } catch (...) {
         FitOk = false;
       }
     } else FitOk = false;
     if ( FitOk ) thePV = transVtx;
       }
       VertexRef VRef = reco::VertexRef(VertexRefProd_out, VertexCollection_out->size());
       VertexCollection_out->push_back(thePV);
       AVPFTauPV->setValue(iPFTau, VRef);
     }
   }
   iEvent.put(std::move(VertexCollection_out),"PFTauPrimaryVertices");
   iEvent.put(std::move(AVPFTauPV));
 }