#include <PFTrackProducer.h>

Inheritance diagram for PFTrackProducer:

Public Member Functions
	PFTrackProducer (const edm::ParameterSet &)
	Constructor. More...

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

bool	hasAbilityToProduceInBeginLumis () const final

bool	hasAbilityToProduceInBeginProcessBlocks () const final

bool	hasAbilityToProduceInBeginRuns () const final

bool	hasAbilityToProduceInEndLumis () const final

bool	hasAbilityToProduceInEndProcessBlocks () const final

bool	hasAbilityToProduceInEndRuns () const final

Private Member Functions
void	beginRun (const edm::Run &, const edm::EventSetup &) override

void	endRun (const edm::Run &, const edm::EventSetup &) override

void	produce (edm::Event &, const edm::EventSetup &) override
	Produce the PFRecTrack collection. More...

Private Attributes
bool	gsfinev_

edm::EDGetTokenT< reco::GsfTrackCollection >	gsfTrackLabel_

edm::EDGetTokenT< reco::MuonCollection >	muonColl_

std::unique_ptr< PFTrackTransformer >	pfTransformer_
	PFTrackTransformer. More...

reco::TrackBase::TrackQuality	trackQuality_

std::vector< edm::EDGetTokenT< reco::TrackCollection > >	tracksContainers_

std::vector< edm::EDGetTokenT< std::vector< Trajectory > > >	trajContainers_

bool	trajinev_

bool	useQuality_
	TRACK QUALITY. More...

edm::EDGetTokenT< reco::VertexCollection >	vtx_h

Additional Inherited Members
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

Detailed Description

Definition at line 28 of file PFTrackProducer.h.

Constructor & Destructor Documentation

◆ PFTrackProducer()

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

explicit

Constructor.

Definition at line 23 of file PFTrackProducer.cc.

                                                             : pfTransformer_() {
   produces<reco::PFRecTrackCollection>();
  
   std::vector<InputTag> tags = iConfig.getParameter<vector<InputTag> >("TkColList");
   trajinev_ = iConfig.getParameter<bool>("TrajInEvents");
   tracksContainers_.reserve(tags.size());
   if (trajinev_) {
     trajContainers_.reserve(tags.size());
   }
   for (auto const& tag : tags) {
     tracksContainers_.push_back(consumes<reco::TrackCollection>(tag));
     if (trajinev_) {
       trajContainers_.push_back(consumes<std::vector<Trajectory> >(tag));
     }
   }
  
   useQuality_ = iConfig.getParameter<bool>("UseQuality");
  
   gsfinev_ = iConfig.getParameter<bool>("GsfTracksInEvents");
   if (gsfinev_) {
     gsfTrackLabel_ = consumes<reco::GsfTrackCollection>(iConfig.getParameter<InputTag>("GsfTrackModuleLabel"));
   }
  
   trackQuality_ = reco::TrackBase::qualityByName(iConfig.getParameter<std::string>("TrackQuality"));
  
   muonColl_ = consumes<reco::MuonCollection>(iConfig.getParameter<InputTag>("MuColl"));
  
   vtx_h = consumes<reco::VertexCollection>(iConfig.getParameter<edm::InputTag>("PrimaryVertexLabel"));
 }

References edm::ParameterSet::getParameter(), gsfinev_, gsfTrackLabel_, muonColl_, reco::TrackBase::qualityByName(), AlCaHLTBitMon_QueryRunRegistry::string, GlobalPosition_Frontier_DevDB_cff::tag, triggerMatcherToHLTDebug_cfi::tags, trackQuality_, tracksContainers_, trajContainers_, trajinev_, useQuality_, and vtx_h.

Member Function Documentation

◆ beginRun()

void PFTrackProducer::beginRun	(	const edm::Run &	run,
		const edm::EventSetup &	iSetup
	)

overrideprivate

Definition at line 204 of file PFTrackProducer.cc.

                                                                           {
   ESHandle<MagneticField> magneticField;
   iSetup.get<IdealMagneticFieldRecord>().get(magneticField);
   pfTransformer_ = std::make_unique<PFTrackTransformer>(math::XYZVector(magneticField->inTesla(GlobalPoint(0, 0, 0))));
   if (!trajinev_)
     pfTransformer_->OnlyProp();
 }

References edm::EventSetup::get(), get, HLT_FULL_cff::magneticField, pfTransformer_, and trajinev_.

◆ endRun()

void PFTrackProducer::endRun	(	const edm::Run &	run,
		const edm::EventSetup &	iSetup
	)

overrideprivate

Definition at line 213 of file PFTrackProducer.cc.

213 { pfTransformer_.reset(); }

References pfTransformer_.

◆ produce()

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

overrideprivate

Produce the PFRecTrack collection.

Definition at line 53 of file PFTrackProducer.cc.

                                                                      {
   //create the empty collections
   auto PfTrColl = std::make_unique<reco::PFRecTrackCollection>();
  
   //read track collection
   Handle<GsfTrackCollection> gsftrackcoll;
   bool foundgsf = false;
   if (gsfinev_) {
     foundgsf = iEvent.getByToken(gsfTrackLabel_, gsftrackcoll);
   }
  
   //Get PV for STIP calculation, if there is none then take the dummy
   Handle<reco::VertexCollection> vertex;
   iEvent.getByToken(vtx_h, vertex);
   reco::Vertex dummy;
   const reco::Vertex* pv = &dummy;
   if (vertex.isValid()) {
     pv = &*vertex->begin();
   } else {  // create a dummy PV
     reco::Vertex::Error e;
     e(0, 0) = 0.0015 * 0.0015;
     e(1, 1) = 0.0015 * 0.0015;
     e(2, 2) = 15. * 15.;
     reco::Vertex::Point p(0, 0, 0);
     dummy = reco::Vertex(p, e, 0, 0, 0);
   }
  
   edm::ESHandle<TransientTrackBuilder> builder;
   iSetup.get<TransientTrackRecord>().get("TransientTrackBuilder", builder);
   TransientTrackBuilder thebuilder = *(builder.product());
  
   // read muon collection
   Handle<reco::MuonCollection> recMuons;
   iEvent.getByToken(muonColl_, recMuons);
  
   //default value for when trajinev_ is false
   const vector<Trajectory> dummyTj(0);
  
   for (unsigned int istr = 0; istr < tracksContainers_.size(); istr++) {
     //Track collection
     Handle<reco::TrackCollection> tkRefCollection;
     iEvent.getByToken(tracksContainers_[istr], tkRefCollection);
     reco::TrackCollection Tk = *(tkRefCollection.product());
  
     //Use a pointer to aoid unnecessary copying of the collection
     const vector<Trajectory>* Tj = &dummyTj;
     if (trajinev_) {
       //Trajectory collection
       Handle<vector<Trajectory> > tjCollection;
       iEvent.getByToken(trajContainers_[istr], tjCollection);
  
       Tj = tjCollection.product();
     }
  
     for (unsigned int i = 0; i < Tk.size(); i++) {
       reco::TrackRef trackRef(tkRefCollection, i);
  
       if (useQuality_ && (!(Tk[i].quality(trackQuality_)))) {
         bool isMuCandidate = false;
  
         //TrackRef trackRef(tkRefCollection, i);
  
         if (recMuons.isValid()) {
           for (unsigned j = 0; j < recMuons->size(); j++) {
             reco::MuonRef muonref(recMuons, j);
             if (muonref->track().isNonnull())
               if (muonref->track() == trackRef && muonref->isGlobalMuon()) {
                 isMuCandidate = true;
                 //cout<<" SAVING TRACK "<<endl;
                 break;
               }
           }
         }
         if (!isMuCandidate) {
           continue;
         }
       }
  
       // find the pre-id kf track
       bool preId = false;
       if (foundgsf) {
         //NOTE: foundgsf is only true if gsftrackcoll is valid
         for (auto const& gsfTrack : *gsftrackcoll) {
           if (gsfTrack.seedRef().isNull())
             continue;
           auto const& seed = *(gsfTrack.extra()->seedRef());
           auto const& ElSeed = dynamic_cast<ElectronSeed const&>(seed);
           if (ElSeed.ctfTrack().isNonnull()) {
             if (ElSeed.ctfTrack() == trackRef) {
               preId = true;
               break;
             }
           }
         }
       }
       if (preId) {
         // Set PFRecTrack of type KF_ElCAND
         reco::PFRecTrack pftrack(trackRef->charge(), reco::PFRecTrack::KF_ELCAND, i, trackRef);
  
         bool valid = false;
         if (trajinev_) {
           valid = pfTransformer_->addPoints(pftrack, *trackRef, (*Tj)[i]);
         } else {
           Trajectory FakeTraj;
           valid = pfTransformer_->addPoints(pftrack, *trackRef, FakeTraj);
         }
         if (valid) {
           //calculate STIP
           double stip = -999;
           const reco::PFTrajectoryPoint& atECAL = pftrack.extrapolatedPoint(reco::PFTrajectoryPoint::ECALEntrance);
           if (atECAL.isValid())  //if track extrapolates to ECAL
           {
             GlobalVector direction(pftrack.extrapolatedPoint(reco::PFTrajectoryPoint::ECALEntrance).position().x(),
                                    pftrack.extrapolatedPoint(reco::PFTrajectoryPoint::ECALEntrance).position().y(),
                                    pftrack.extrapolatedPoint(reco::PFTrajectoryPoint::ECALEntrance).position().z());
             stip = IPTools::signedTransverseImpactParameter(thebuilder.build(*trackRef), direction, *pv)
                        .second.significance();
           }
           pftrack.setSTIP(stip);
           PfTrColl->push_back(pftrack);
         }
       } else {
         reco::PFRecTrack pftrack(trackRef->charge(), reco::PFRecTrack::KF, i, trackRef);
         bool valid = false;
         if (trajinev_) {
           valid = pfTransformer_->addPoints(pftrack, *trackRef, (*Tj)[i]);
         } else {
           Trajectory FakeTraj;
           valid = pfTransformer_->addPoints(pftrack, *trackRef, FakeTraj);
         }
  
         if (valid) {
           double stip = -999;
           const reco::PFTrajectoryPoint& atECAL = pftrack.extrapolatedPoint(reco::PFTrajectoryPoint::ECALEntrance);
           if (atECAL.isValid()) {
             GlobalVector direction(pftrack.extrapolatedPoint(reco::PFTrajectoryPoint::ECALEntrance).position().x(),
                                    pftrack.extrapolatedPoint(reco::PFTrajectoryPoint::ECALEntrance).position().y(),
                                    pftrack.extrapolatedPoint(reco::PFTrajectoryPoint::ECALEntrance).position().z());
             stip = IPTools::signedTransverseImpactParameter(thebuilder.build(*trackRef), direction, *pv)
                        .second.significance();
           }
           pftrack.setSTIP(stip);
           PfTrColl->push_back(pftrack);
         }
       }
     }
   }
   iEvent.put(std::move(PfTrColl));
 }