PFDisplacedTrackerVertexProducer Class Reference

#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	hasAbilityToProduceInBeginRuns () const final
bool	hasAbilityToProduceInEndLumis () 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
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 21 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 79 of file PFDisplacedTrackerVertexProducer.cc.

References edm::EventSetup::get(), MagneticField::inTesla(), HLT_2018_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 87 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 23 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(), tkAlTrackRefitSequence_cff::refittedTracks, pfDisplacedTrackerVertex_cfi::trackColl, and validateGeometry_cfg::valid.

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

Member Data Documentation

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

private

Definition at line 29 of file PFDisplacedTrackerVertexProducer.h.

Referenced by PFDisplacedTrackerVertexProducer(), and produce().

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

private

Definition at line 30 of file PFDisplacedTrackerVertexProducer.h.

Referenced by PFDisplacedTrackerVertexProducer(), and produce().

PFTrackTransformer* PFDisplacedTrackerVertexProducer::pfTransformer_

private

PFTrackTransformer.

Definition at line 28 of file PFDisplacedTrackerVertexProducer.h.

Referenced by beginRun(), endRun(), produce(), and ~PFDisplacedTrackerVertexProducer().