ShallowSimTracksProducer Class Reference

#include <ShallowSimTracksProducer.h>

Inheritance diagram for ShallowSimTracksProducer:

Public Member Functions
	ShallowSimTracksProducer (const edm::ParameterSet &)
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 EDProducer &	operator= (const EDProducer &)=delete

Private Member Functions
void	produce (edm::Event &iEvent, const edm::EventSetup &iSetup) override

Private Attributes
const edm::EDGetTokenT < reco::TrackToTrackingParticleAssociator >	associator_token_
const std::string	Prefix
const std::string	Suffix
const edm::EDGetTokenT < TrackingParticleCollection >	trackingParticles_token_
const edm::EDGetTokenT < edm::View< reco::Track > >	tracks_token_

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 12 of file ShallowSimTracksProducer.h.

Constructor & Destructor Documentation

ShallowSimTracksProducer::ShallowSimTracksProducer ( const edm::ParameterSet &  conf )

explicit

Definition at line 10 of file ShallowSimTracksProducer.cc.

References Prefix, and Suffix.

    : Prefix(conf.getParameter<std::string>("Prefix")),
      Suffix(conf.getParameter<std::string>("Suffix")),
      trackingParticles_token_(
          consumes<TrackingParticleCollection>(conf.getParameter<edm::InputTag>("TrackingParticles"))),
      associator_token_(
          consumes<reco::TrackToTrackingParticleAssociator>(conf.getParameter<edm::InputTag>("Associator"))),
      tracks_token_(consumes<edm::View<reco::Track>>(conf.getParameter<edm::InputTag>("Tracks"))) {
  produces<std::vector<unsigned>>(Prefix + "multi" + Suffix);
  produces<std::vector<int>>(Prefix + "type" + Suffix);
  produces<std::vector<float>>(Prefix + "charge" + Suffix);
  produces<std::vector<float>>(Prefix + "momentum" + Suffix);
  produces<std::vector<float>>(Prefix + "pt" + Suffix);
  produces<std::vector<double>>(Prefix + "theta" + Suffix);
  produces<std::vector<double>>(Prefix + "phi" + Suffix);
  produces<std::vector<double>>(Prefix + "eta" + Suffix);
  produces<std::vector<double>>(Prefix + "qoverp" + Suffix);
  produces<std::vector<double>>(Prefix + "vx" + Suffix);
  produces<std::vector<double>>(Prefix + "vy" + Suffix);
  produces<std::vector<double>>(Prefix + "vz" + Suffix);
}

Member Function Documentation

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

overrideprivate

Definition at line 32 of file ShallowSimTracksProducer.cc.

References associator_token_, RecoTauCleanerPlugins::charge, TrackingParticle::charge(), edm::AssociationMap< edm::OneToManyWithQualityGeneric< edm::View< reco::Track >, TrackingParticleCollection, double > >::const_iterator, PVValHelper::dxy, PVValHelper::eta, TrackingParticle::eta(), shallow::findTrackIndex(), edm::Ref< C, T, F >::get(), mps_fire::i, eostools::move(), trackerHitRTTI::multi, TrackingParticle::p(), TrackingParticle::parentVertex(), TrackingParticle::pdgId(), phi, TrackingParticle::phi(), TrackingVertex::position(), Prefix, DiDispStaMuonMonitor_cfi::pt, TrackingParticle::pt(), findQualityFiles::size, Suffix, theta(), TrackingParticle::theta(), HLT_FULL_cff::track, trackingTruthProducer_cfi::trackingParticles, trackingParticles_token_, tracks, and tracks_token_.

                                                                                   {
  edm::Handle<edm::View<reco::Track>> tracks;
  event.getByToken(tracks_token_, tracks);
  edm::Handle<TrackingParticleCollection> trackingParticles;
  event.getByToken(trackingParticles_token_, trackingParticles);
  edm::Handle<reco::TrackToTrackingParticleAssociator> associator;
  event.getByToken(associator_token_, associator);

  unsigned size = tracks->size();
  auto multi = std::make_unique<std::vector<unsigned>>(size, 0);
  auto type = std::make_unique<std::vector<int>>(size, 0);
  auto charge = std::make_unique<std::vector<float>>(size, 0);
  auto momentum = std::make_unique<std::vector<float>>(size, -1);
  auto pt = std::make_unique<std::vector<float>>(size, -1);
  auto theta = std::make_unique<std::vector<double>>(size, -1000);
  auto phi = std::make_unique<std::vector<double>>(size, -1000);
  auto eta = std::make_unique<std::vector<double>>(size, -1000);
  auto dxy = std::make_unique<std::vector<double>>(size, -1000);
  auto dsz = std::make_unique<std::vector<double>>(size, -1000);
  auto qoverp = std::make_unique<std::vector<double>>(size, -1000);
  auto vx = std::make_unique<std::vector<double>>(size, -1000);
  auto vy = std::make_unique<std::vector<double>>(size, -1000);
  auto vz = std::make_unique<std::vector<double>>(size, -1000);

  reco::RecoToSimCollection associations = associator->associateRecoToSim(tracks, trackingParticles);

  for (reco::RecoToSimCollection::const_iterator association = associations.begin(); association != associations.end();
       association++) {
    const reco::Track* track = association->key.get();
    const int matches = association->val.size();
    if (matches > 0) {
      const TrackingParticle* tparticle = association->val[0].first.get();
      unsigned i = shallow::findTrackIndex(tracks, track);

      multi->at(i) = matches;
      type->at(i) = tparticle->pdgId();
      charge->at(i) = tparticle->charge();
      momentum->at(i) = tparticle->p();
      pt->at(i) = tparticle->pt();
      theta->at(i) = tparticle->theta();
      phi->at(i) = tparticle->phi();
      eta->at(i) = tparticle->eta();
      qoverp->at(i) = tparticle->charge() / tparticle->p();

      const TrackingVertex* tvertex = tparticle->parentVertex().get();
      vx->at(i) = tvertex->position().x();
      vy->at(i) = tvertex->position().y();
      vz->at(i) = tvertex->position().z();
    }
  }

  event.put(std::move(multi), Prefix + "multi" + Suffix);
  event.put(std::move(type), Prefix + "type" + Suffix);
  event.put(std::move(charge), Prefix + "charge" + Suffix);
  event.put(std::move(momentum), Prefix + "momentum" + Suffix);
  event.put(std::move(pt), Prefix + "pt" + Suffix);
  event.put(std::move(theta), Prefix + "theta" + Suffix);
  event.put(std::move(phi), Prefix + "phi" + Suffix);
  event.put(std::move(eta), Prefix + "eta" + Suffix);
  event.put(std::move(qoverp), Prefix + "qoverp" + Suffix);
  event.put(std::move(vx), Prefix + "vx" + Suffix);
  event.put(std::move(vy), Prefix + "vy" + Suffix);
  event.put(std::move(vz), Prefix + "vz" + Suffix);
}

Member Data Documentation

const edm::EDGetTokenT<reco::TrackToTrackingParticleAssociator> ShallowSimTracksProducer::associator_token_

private

Definition at line 20 of file ShallowSimTracksProducer.h.

Referenced by produce().

const std::string ShallowSimTracksProducer::Prefix

private

Definition at line 17 of file ShallowSimTracksProducer.h.

Referenced by produce(), and ShallowSimTracksProducer().

const std::string ShallowSimTracksProducer::Suffix

private

Definition at line 18 of file ShallowSimTracksProducer.h.

Referenced by produce(), and ShallowSimTracksProducer().

const edm::EDGetTokenT<TrackingParticleCollection> ShallowSimTracksProducer::trackingParticles_token_

private

Definition at line 19 of file ShallowSimTracksProducer.h.

Referenced by produce().

const edm::EDGetTokenT<edm::View<reco::Track> > ShallowSimTracksProducer::tracks_token_

private

Definition at line 21 of file ShallowSimTracksProducer.h.

Referenced by produce().