#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	hasAbilityToProduceInLumis () const final

bool	hasAbilityToProduceInRuns () 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
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 12 of file PFDisplacedTrackerVertexProducer.h.

Constructor & Destructor Documentation

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

explicit

Constructor.

Definition at line 10 of file PFDisplacedTrackerVertexProducer.cc.

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

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

PFDisplacedTrackerVertexProducer::~PFDisplacedTrackerVertexProducer ( )

override

Destructor.

Definition at line 24 of file PFDisplacedTrackerVertexProducer.cc.

References pfTransformer_.

 {
   delete pfTransformer_;
 }

Member Function Documentation

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

overrideprivate

Definition at line 98 of file PFDisplacedTrackerVertexProducer.cc.

References edm::EventSetup::get(), MagneticField::inTesla(), seedCreatorFromRegionConsecutiveHitsEDProducer_cff::magneticField, PFTrackTransformer::OnlyProp(), and pfTransformer_.

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

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

overrideprivate

Definition at line 109 of file PFDisplacedTrackerVertexProducer.cc.

References pfTransformer_.

                                                {
   delete pfTransformer_;
   pfTransformer_=nullptr;
 }

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

overrideprivate

Produce the PFRecTrack collection.

Definition at line 30 of file PFDisplacedTrackerVertexProducer.cc.

References PFTrackTransformer::addPoints(), reco::TrackBase::charge(), edm::Event::getByToken(), edm::Event::getRefBeforePut(), icoll, training_settings::idx, edm::RefToBase< T >::key(), reco::PFRecTrack::KF, eostools::move(), pfDisplacedVertexContainer_, pfTrackContainer_, pfTransformer_, edm::RefVector< C, T, F >::push_back(), edm::Event::put(), and pfDisplacedTrackerVertex_cfi::trackColl.

 {
 
   //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));
 }