#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_

const edm::ESGetToken< MagneticField, IdealMagneticFieldRecord >	magneticFieldToken_

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_

const edm::ESGetToken< TransientTrackBuilder, TransientTrackRecord >	transientTrackToken_

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 32 of file PFTrackProducer.h.

Constructor & Destructor Documentation

◆ PFTrackProducer()

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

explicit

Constructor.

Definition at line 19 of file PFTrackProducer.cc.

     : transientTrackToken_(esConsumes(edm::ESInputTag("", "TransientTrackBuilder"))),
       magneticFieldToken_(esConsumes<edm::Transition::BeginRun>()),
       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::BeginRun, 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 202 of file PFTrackProducer.cc.

                                                                           {
   auto const& magneticField = iSetup.getData(magneticFieldToken_);
   pfTransformer_ = std::make_unique<PFTrackTransformer>(math::XYZVector(magneticField.inTesla(GlobalPoint(0, 0, 0))));
   if (!trajinev_)
     pfTransformer_->OnlyProp();
 }

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

◆ endRun()

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

overrideprivate

Definition at line 210 of file PFTrackProducer.cc.

210 { pfTransformer_.reset(); }

References pfTransformer_.

◆ produce()

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

overrideprivate

Produce the PFRecTrack collection.

Definition at line 52 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);
   }
  
   //setup transient track builder
   TransientTrackBuilder const& thebuilder = iSetup.getData(transientTrackToken_);
  
   // 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));
 }