#include <PFDisplacedTrackerVertexProducer.h>

Inheritance diagram for PFDisplacedTrackerVertexProducer:

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

	~PFDisplacedTrackerVertexProducer () override
	Destructor. 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
const edm::ESGetToken< MagneticField, IdealMagneticFieldRecord >	magneticFieldToken_

edm::EDGetTokenT< reco::PFDisplacedVertexCollection >	pfDisplacedVertexContainer_

edm::EDGetTokenT< reco::TrackCollection >	pfTrackContainer_

PFTrackTransformer *	pfTransformer_
	PFTrackTransformer. More...

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 14 of file PFDisplacedTrackerVertexProducer.h.

Constructor & Destructor Documentation

◆ PFDisplacedTrackerVertexProducer()

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

explicit

Constructor.

Definition at line 9 of file PFDisplacedTrackerVertexProducer.cc.

     : pfTransformer_(nullptr), magneticFieldToken_(esConsumes<edm::Transition::BeginRun>()) {
   produces<reco::PFRecTrackCollection>();
   produces<reco::PFDisplacedTrackerVertexCollection>();
  
   pfDisplacedVertexContainer_ =
       consumes<reco::PFDisplacedVertexCollection>(iConfig.getParameter<InputTag>("displacedTrackerVertexColl"));
  
   pfTrackContainer_ = consumes<reco::TrackCollection>(iConfig.getParameter<InputTag>("trackColl"));
 }

References edm::BeginRun, edm::ParameterSet::getParameter(), pfDisplacedVertexContainer_, and pfTrackContainer_.

◆ ~PFDisplacedTrackerVertexProducer()

PFDisplacedTrackerVertexProducer::~PFDisplacedTrackerVertexProducer ( )

override

Destructor.

Definition at line 20 of file PFDisplacedTrackerVertexProducer.cc.

20 { delete pfTransformer_; }

References pfTransformer_.

Member Function Documentation

◆ beginRun()

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

overrideprivate

Definition at line 78 of file PFDisplacedTrackerVertexProducer.cc.

                                                                                            {
   auto const& magneticField = &iSetup.getData(magneticFieldToken_);
   pfTransformer_ = new PFTrackTransformer(math::XYZVector(magneticField->inTesla(GlobalPoint(0, 0, 0))));
   pfTransformer_->OnlyProp();
 }

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

◆ endRun()

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

overrideprivate

Definition at line 85 of file PFDisplacedTrackerVertexProducer.cc.

                                                                                          {
   delete pfTransformer_;
   pfTransformer_ = nullptr;
 }

References pfTransformer_.

◆ produce()

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

overrideprivate

Produce the PFRecTrack collection.

Definition at line 22 of file PFDisplacedTrackerVertexProducer.cc.

                                                                                       {
   //create the empty collections
   auto pfDisplacedTrackerVertexColl = std::make_unique<reco::PFDisplacedTrackerVertexCollection>();
   auto pfRecTrackColl = std::make_unique<reco::PFRecTrackCollection>();
  
   reco::PFRecTrackRefProd pfTrackRefProd = iEvent.getRefBeforePut<reco::PFRecTrackCollection>();
  
   Handle<reco::PFDisplacedVertexCollection> nuclCollH;
   iEvent.getByToken(pfDisplacedVertexContainer_, nuclCollH);
   const reco::PFDisplacedVertexCollection& nuclColl = *(nuclCollH.product());
  
   Handle<reco::TrackCollection> trackColl;
   iEvent.getByToken(pfTrackContainer_, trackColl);
  
   int idx = 0;
  
   //  cout << "Size of Displaced Vertices "
   //     <<  nuclColl.size() << endl;
  
   // loop on all NuclearInteraction
   for (unsigned int icoll = 0; icoll < nuclColl.size(); icoll++) {
     reco::PFRecTrackRefVector pfRecTkcoll;
  
     std::vector<reco::Track> refittedTracks = nuclColl[icoll].refittedTracks();
  
     // convert the secondary tracks
     for (unsigned it = 0; it < refittedTracks.size(); it++) {
       reco::TrackBaseRef trackBaseRef = nuclColl[icoll].originalTrack(refittedTracks[it]);
  
       //      cout << "track base pt = " << trackBaseRef->pt() << endl;
  
       reco::TrackRef trackRef(trackColl, trackBaseRef.key());
  
       //      cout << "track pt = " << trackRef->pt() << endl;
  
       reco::PFRecTrack pfRecTrack(trackBaseRef->charge(), reco::PFRecTrack::KF, trackBaseRef.key(), trackRef);
  
       // cout << pfRecTrack << endl;
  
       Trajectory FakeTraj;
       bool valid = pfTransformer_->addPoints(pfRecTrack, *trackBaseRef, FakeTraj);
       if (valid) {
         pfRecTkcoll.push_back(reco::PFRecTrackRef(pfTrackRefProd, idx++));
         pfRecTrackColl->push_back(pfRecTrack);
         //  cout << "after "<< pfRecTrack << endl;
       }
     }
     reco::PFDisplacedVertexRef niRef(nuclCollH, icoll);
     pfDisplacedTrackerVertexColl->push_back(reco::PFDisplacedTrackerVertex(niRef, pfRecTkcoll));
   }
  
   iEvent.put(std::move(pfRecTrackColl));
   iEvent.put(std::move(pfDisplacedTrackerVertexColl));
 }