CMS 3D CMS Logo

List of all members | Public Member Functions | Private Member Functions | Private Attributes
PFDisplacedTrackerVertexProducer Class Reference
Inheritance diagram for PFDisplacedTrackerVertexProducer:
edm::stream::EDProducer<>

Public Member Functions

 PFDisplacedTrackerVertexProducer (const edm::ParameterSet &)
 Constructor. More...
 
 ~PFDisplacedTrackerVertexProducer () override
 Destructor. More...
 
- Public Member Functions inherited from edm::stream::EDProducer<>
 EDProducer ()=default
 
 EDProducer (const EDProducer &)=delete
 
bool hasAbilityToProduceInBeginLumis () const final
 
bool hasAbilityToProduceInBeginProcessBlocks () const final
 
bool hasAbilityToProduceInBeginRuns () const final
 
bool hasAbilityToProduceInEndLumis () const final
 
bool hasAbilityToProduceInEndProcessBlocks () const final
 
bool hasAbilityToProduceInEndRuns () const final
 
const EDProduceroperator= (const EDProducer &)=delete
 

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, IdealMagneticFieldRecordmagneticFieldToken_
 
edm::EDGetTokenT< reco::PFDisplacedVertexCollectionpfDisplacedVertexContainer_
 
edm::EDGetTokenT< reco::TrackCollectionpfTrackContainer_
 
PFTrackTransformerpfTransformer_
 PFTrackTransformer. More...
 

Additional Inherited Members

- Public Types inherited from edm::stream::EDProducer<>
using CacheTypes = CacheContexts< T... >
 
using GlobalCache = typename CacheTypes::GlobalCache
 
using HasAbility = AbilityChecker< T... >
 
using InputProcessBlockCache = typename CacheTypes::InputProcessBlockCache
 
using LuminosityBlockCache = typename CacheTypes::LuminosityBlockCache
 
using LuminosityBlockContext = LuminosityBlockContextT< LuminosityBlockCache, RunCache, GlobalCache >
 
using LuminosityBlockSummaryCache = typename CacheTypes::LuminosityBlockSummaryCache
 
using RunCache = typename CacheTypes::RunCache
 
using RunContext = RunContextT< RunCache, GlobalCache >
 
using RunSummaryCache = typename CacheTypes::RunSummaryCache
 

Detailed Description

Definition at line 11 of file PFDisplacedTrackerVertexProducer.cc.

Constructor & Destructor Documentation

◆ PFDisplacedTrackerVertexProducer()

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

Constructor.

Definition at line 39 of file PFDisplacedTrackerVertexProducer.cc.

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

40  : pfTransformer_(nullptr), magneticFieldToken_(esConsumes<edm::Transition::BeginRun>()) {
41  produces<reco::PFRecTrackCollection>();
42  produces<reco::PFDisplacedTrackerVertexCollection>();
43 
45  consumes<reco::PFDisplacedVertexCollection>(iConfig.getParameter<InputTag>("displacedTrackerVertexColl"));
46 
47  pfTrackContainer_ = consumes<reco::TrackCollection>(iConfig.getParameter<InputTag>("trackColl"));
48 }
edm::EDGetTokenT< reco::TrackCollection > pfTrackContainer_
T getParameter(std::string const &) const
Definition: ParameterSet.h:307
edm::EDGetTokenT< reco::PFDisplacedVertexCollection > pfDisplacedVertexContainer_
PFTrackTransformer * pfTransformer_
PFTrackTransformer.
const edm::ESGetToken< MagneticField, IdealMagneticFieldRecord > magneticFieldToken_

◆ ~PFDisplacedTrackerVertexProducer()

PFDisplacedTrackerVertexProducer::~PFDisplacedTrackerVertexProducer ( )
override

Destructor.

Definition at line 50 of file PFDisplacedTrackerVertexProducer.cc.

References pfTransformer_.

50 { delete pfTransformer_; }
PFTrackTransformer * pfTransformer_
PFTrackTransformer.

Member Function Documentation

◆ beginRun()

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

Definition at line 108 of file PFDisplacedTrackerVertexProducer.cc.

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

108  {
109  auto const& magneticField = &iSetup.getData(magneticFieldToken_);
112 }
T const & getData(const ESGetToken< T, R > &iToken) const noexcept(false)
Definition: EventSetup.h:119
PFTrackTransformer * pfTransformer_
PFTrackTransformer.
const edm::ESGetToken< MagneticField, IdealMagneticFieldRecord > magneticFieldToken_
XYZVectorD XYZVector
spatial vector with cartesian internal representation
Definition: Vector3D.h:31

◆ endRun()

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

Definition at line 115 of file PFDisplacedTrackerVertexProducer.cc.

References pfTransformer_.

115  {
116  delete pfTransformer_;
117  pfTransformer_ = nullptr;
118 }
PFTrackTransformer * pfTransformer_
PFTrackTransformer.

◆ produce()

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

Produce the PFRecTrack collection.

Definition at line 52 of file PFDisplacedTrackerVertexProducer.cc.

References PFTrackTransformer::addPoints(), reco::TrackBase::charge(), icoll, heavyIonCSV_trainingSettings::idx, iEvent, edm::RefToBase< T >::key(), reco::PFRecTrack::KF, eostools::move(), pfDisplacedVertexContainer_, pfTrackContainer_, pfTransformer_, edm::RefVector< C, T, F >::push_back(), tkAlTrackRefitSequence_cff::refittedTracks, pfDisplacedTrackerVertex_cfi::trackColl, and validateGeometry_cfg::valid.

52  {
53  //create the empty collections
54  auto pfDisplacedTrackerVertexColl = std::make_unique<reco::PFDisplacedTrackerVertexCollection>();
55  auto pfRecTrackColl = std::make_unique<reco::PFRecTrackCollection>();
56 
57  reco::PFRecTrackRefProd pfTrackRefProd = iEvent.getRefBeforePut<reco::PFRecTrackCollection>();
58 
60  iEvent.getByToken(pfDisplacedVertexContainer_, nuclCollH);
61  const reco::PFDisplacedVertexCollection& nuclColl = *(nuclCollH.product());
62 
64  iEvent.getByToken(pfTrackContainer_, trackColl);
65 
66  int idx = 0;
67 
68  // cout << "Size of Displaced Vertices "
69  // << nuclColl.size() << endl;
70 
71  // loop on all NuclearInteraction
72  for (unsigned int icoll = 0; icoll < nuclColl.size(); icoll++) {
73  reco::PFRecTrackRefVector pfRecTkcoll;
74 
75  std::vector<reco::Track> refittedTracks = nuclColl[icoll].refittedTracks();
76 
77  // convert the secondary tracks
78  for (unsigned it = 0; it < refittedTracks.size(); it++) {
79  reco::TrackBaseRef trackBaseRef = nuclColl[icoll].originalTrack(refittedTracks[it]);
80 
81  // cout << "track base pt = " << trackBaseRef->pt() << endl;
82 
83  reco::TrackRef trackRef(trackColl, trackBaseRef.key());
84 
85  // cout << "track pt = " << trackRef->pt() << endl;
86 
87  reco::PFRecTrack pfRecTrack(trackBaseRef->charge(), reco::PFRecTrack::KF, trackBaseRef.key(), trackRef);
88 
89  // cout << pfRecTrack << endl;
90 
91  Trajectory FakeTraj;
92  bool valid = pfTransformer_->addPoints(pfRecTrack, *trackBaseRef, FakeTraj);
93  if (valid) {
94  pfRecTkcoll.push_back(reco::PFRecTrackRef(pfTrackRefProd, idx++));
95  pfRecTrackColl->push_back(pfRecTrack);
96  // cout << "after "<< pfRecTrack << endl;
97  }
98  }
99  reco::PFDisplacedVertexRef niRef(nuclCollH, icoll);
100  pfDisplacedTrackerVertexColl->push_back(reco::PFDisplacedTrackerVertex(niRef, pfRecTkcoll));
101  }
102 
103  iEvent.put(std::move(pfRecTrackColl));
104  iEvent.put(std::move(pfDisplacedTrackerVertexColl));
105 }
edm::EDGetTokenT< reco::TrackCollection > pfTrackContainer_
reconstructed track used as an input to particle flow
Definition: PFRecTrack.h:20
std::vector< PFDisplacedVertex > PFDisplacedVertexCollection
collection of PFDisplacedVertex objects
edm::EDGetTokenT< reco::PFDisplacedVertexCollection > pfDisplacedVertexContainer_
int charge() const
track electric charge
Definition: TrackBase.h:596
int iEvent
Definition: GenABIO.cc:224
PFTrackTransformer * pfTransformer_
PFTrackTransformer.
size_t key() const
Definition: RefToBase.h:221
bool addPoints(reco::PFRecTrack &pftrack, const reco::Track &track, const Trajectory &traj, bool msgwarning=true) const
Add points to a PFTrack. return false if a TSOS is invalid.
void push_back(value_type const &ref)
Add a Ref<C, T> to the RefVector.
Definition: RefVector.h:67
std::vector< PFRecTrack > PFRecTrackCollection
collection of PFRecTrack objects
Definition: PFRecTrackFwd.h:9
def move(src, dest)
Definition: eostools.py:511
int icoll
Definition: AMPTWrapper.h:146

Member Data Documentation

◆ magneticFieldToken_

const edm::ESGetToken<MagneticField, IdealMagneticFieldRecord> PFDisplacedTrackerVertexProducer::magneticFieldToken_
private

Definition at line 31 of file PFDisplacedTrackerVertexProducer.cc.

Referenced by beginRun().

◆ pfDisplacedVertexContainer_

edm::EDGetTokenT<reco::PFDisplacedVertexCollection> PFDisplacedTrackerVertexProducer::pfDisplacedVertexContainer_
private

◆ pfTrackContainer_

edm::EDGetTokenT<reco::TrackCollection> PFDisplacedTrackerVertexProducer::pfTrackContainer_
private

◆ pfTransformer_

PFTrackTransformer* PFDisplacedTrackerVertexProducer::pfTransformer_
private