#include <PFDisplacedTrackerVertexProducer.h>

Inheritance diagram for PFDisplacedTrackerVertexProducer:

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

	~PFDisplacedTrackerVertexProducer ()
	Destructor. More...

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

Public Member Functions inherited from edm::stream::EDProducerBase
	EDProducerBase ()

ModuleDescription const &	moduleDescription () const

virtual	~EDProducerBase ()

Public Member Functions inherited from edm::ProducerBase
void	callWhenNewProductsRegistered (std::function< void(BranchDescription const &)> const &func)

std::vector< edm::ProductResolverIndex > const &	indiciesForPutProducts (BranchType iBranchType) const

	ProducerBase ()

void	registerProducts (ProducerBase , ProductRegistry , ModuleDescription const &)

std::function< void(BranchDescription const &)>	registrationCallback () const
	used by the fwk to register list of products More...

void	resolvePutIndicies (BranchType iBranchType, std::unordered_multimap< std::string, edm::ProductResolverIndex > const &iIndicies, std::string const &moduleLabel)

virtual	~ProducerBase () noexcept(false)

Public Member Functions inherited from edm::EDConsumerBase
std::vector< ConsumesInfo >	consumesInfo () const

	EDConsumerBase ()

	EDConsumerBase (EDConsumerBase const &)=delete

	EDConsumerBase (EDConsumerBase &&)=default

ProductResolverIndexAndSkipBit	indexFrom (EDGetToken, BranchType, TypeID const &) const

void	itemsMayGet (BranchType, std::vector< ProductResolverIndexAndSkipBit > &) const

void	itemsToGet (BranchType, std::vector< ProductResolverIndexAndSkipBit > &) const

std::vector< ProductResolverIndexAndSkipBit > const &	itemsToGetFromEvent () const

void	labelsForToken (EDGetToken iToken, Labels &oLabels) const

void	modulesWhoseProductsAreConsumed (std::vector< ModuleDescription const * > &modules, ProductRegistry const &preg, std::map< std::string, ModuleDescription const * > const &labelsToDesc, std::string const &processName) const

EDConsumerBase const &	operator= (EDConsumerBase const &)=delete

EDConsumerBase &	operator= (EDConsumerBase &&)=default

bool	registeredToConsume (ProductResolverIndex, bool, BranchType) const

bool	registeredToConsumeMany (TypeID const &, BranchType) const

void	updateLookup (BranchType iBranchType, ProductResolverIndexHelper const &, bool iPrefetchMayGet)

virtual	~EDConsumerBase () noexcept(false)

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

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

virtual 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

Public Types inherited from edm::stream::EDProducerBase
typedef EDProducerAdaptorBase	ModuleType

Public Types inherited from edm::ProducerBase
typedef ProductRegistryHelper::TypeLabelList	TypeLabelList

Public Types inherited from edm::EDConsumerBase
typedef ProductLabels	Labels

Static Public Member Functions inherited from edm::stream::EDProducerBase
static const std::string &	baseType ()

static void	fillDescriptions (ConfigurationDescriptions &descriptions)

static void	prevalidate (ConfigurationDescriptions &descriptions)

Protected Member Functions inherited from edm::EDConsumerBase
template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	consumes (edm::InputTag const &tag)

EDGetToken	consumes (const TypeToGet &id, edm::InputTag const &tag)

template<BranchType B>
EDGetToken	consumes (TypeToGet const &id, edm::InputTag const &tag)

ConsumesCollector	consumesCollector ()
	Use a ConsumesCollector to gather consumes information from helper functions. More...

template<typename ProductType , BranchType B = InEvent>
void	consumesMany ()

void	consumesMany (const TypeToGet &id)

template<BranchType B>
void	consumesMany (const TypeToGet &id)

template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	mayConsume (edm::InputTag const &tag)

EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)

template<BranchType B>
EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)

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_(0)
 {
   produces<reco::PFRecTrackCollection>();
   produces<reco::PFDisplacedTrackerVertexCollection>();
 
   pfDisplacedVertexContainer_ = consumes<reco::PFDisplacedVertexCollection>(                                        iConfig.getParameter< InputTag >("displacedTrackerVertexColl"));
 
 
   pfTrackContainer_ =consumes<reco::TrackCollection>(
     iConfig.getParameter< InputTag >("trackColl"));
 
 }

PFDisplacedTrackerVertexProducer::~PFDisplacedTrackerVertexProducer ( )

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
	)

overrideprivatevirtual

Definition at line 98 of file PFDisplacedTrackerVertexProducer.cc.

References edm::EventSetup::get(), MagneticField::inTesla(), DetachedQuadStep_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
	)

overrideprivatevirtual

Definition at line 109 of file PFDisplacedTrackerVertexProducer.cc.

References pfTransformer_.

                                                {
   delete pfTransformer_;
   pfTransformer_=nullptr;
 }

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

overrideprivatevirtual

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