ShallowSimTracksProducer Class Reference

#include <ShallowSimTracksProducer.h>

Inheritance diagram for ShallowSimTracksProducer:

Public Member Functions
	ShallowSimTracksProducer (const edm::ParameterSet &)
Public Member Functions inherited from edm::EDProducer
	EDProducer ()
ModuleDescription const &	moduleDescription () const
virtual	~EDProducer ()
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
void	produce (edm::Event &, const edm::EventSetup &) 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::EDProducer
typedef EDProducer	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::EDProducer
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 ShallowSimTracksProducer.h.

Constructor & Destructor Documentation

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

explicit

Definition at line 11 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 &  event, const edm::EventSetup &  setup  )

overrideprivatevirtual

Implements edm::EDProducer.

Definition at line 34 of file ShallowSimTracksProducer.cc.

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

Referenced by JSONExport.JsonExport::export(), HTMLExport.HTMLExport::export(), and HTMLExport.HTMLExportStatic::export().

                                                     {

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

Referenced by produce().

const std::string ShallowSimTracksProducer::Prefix

private

Definition at line 20 of file ShallowSimTracksProducer.h.

Referenced by produce(), and ShallowSimTracksProducer().

const std::string ShallowSimTracksProducer::Suffix

private

Definition at line 21 of file ShallowSimTracksProducer.h.

Referenced by produce(), and ShallowSimTracksProducer().

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

private

Definition at line 22 of file ShallowSimTracksProducer.h.

Referenced by produce().

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

private

Definition at line 24 of file ShallowSimTracksProducer.h.

Referenced by produce().