CMS 3D CMS Logo

List of all members | Public Member Functions | Private Attributes
pat::PATPackedGenParticleProducer Class Reference
Inheritance diagram for pat::PATPackedGenParticleProducer:
edm::global::EDProducer<> edm::global::EDProducerBase edm::ProducerBase edm::EDConsumerBase edm::ProductRegistryHelper

Public Member Functions

 PATPackedGenParticleProducer (const edm::ParameterSet &)
 
virtual void produce (edm::StreamID, edm::Event &, const edm::EventSetup &) const
 
 ~PATPackedGenParticleProducer ()
 
- Public Member Functions inherited from edm::global::EDProducer<>
 EDProducer ()=default
 
- Public Member Functions inherited from edm::global::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< ConsumesInfoconsumesInfo () const
 
void convertCurrentProcessAlias (std::string const &processName)
 Convert "@currentProcess" in InputTag process names to the actual current process name. More...
 
 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 & itemsToGetFrom (BranchType iType) 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
 
EDConsumerBaseoperator= (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 Attributes

const edm::EDGetTokenT< edm::Association< reco::GenParticleCollection > > Asso_
 
const edm::EDGetTokenT< edm::Association< reco::GenParticleCollection > > AssoOriginal_
 
const edm::EDGetTokenT< reco::GenParticleCollectionCands_
 
const edm::EDGetTokenT< reco::GenParticleCollectionGenOrigs_
 
const double maxRapidity_
 

Additional Inherited Members

- Public Types inherited from edm::global::EDProducerBase
typedef EDProducerBase 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::global::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 40 of file PATPackedGenParticleProducer.cc.

Constructor & Destructor Documentation

pat::PATPackedGenParticleProducer::PATPackedGenParticleProducer ( const edm::ParameterSet iConfig)
explicit

Definition at line 56 of file PATPackedGenParticleProducer.cc.

56  :
57  Cands_(consumes<reco::GenParticleCollection>(iConfig.getParameter<edm::InputTag>("inputCollection"))),
58  GenOrigs_(consumes<reco::GenParticleCollection>(iConfig.getParameter<edm::InputTag>("inputOriginal"))),
61  maxRapidity_(iConfig.getParameter<double>("maxRapidity"))
62 {
63  produces< std::vector<pat::PackedGenParticle> > ();
64  produces< edm::Association< std::vector<pat::PackedGenParticle> > >();
65 }
T getParameter(std::string const &) const
EDGetTokenT< ProductType > consumes(edm::InputTag const &tag)
const edm::EDGetTokenT< reco::GenParticleCollection > Cands_
const edm::EDGetTokenT< edm::Association< reco::GenParticleCollection > > Asso_
const edm::EDGetTokenT< reco::GenParticleCollection > GenOrigs_
const edm::EDGetTokenT< edm::Association< reco::GenParticleCollection > > AssoOriginal_
pat::PATPackedGenParticleProducer::~PATPackedGenParticleProducer ( )

Definition at line 67 of file PATPackedGenParticleProducer.cc.

67 {}

Member Function Documentation

void pat::PATPackedGenParticleProducer::produce ( edm::StreamID  ,
edm::Event iEvent,
const edm::EventSetup iSetup 
) const
virtual

Definition at line 69 of file PATPackedGenParticleProducer.cc.

References funct::abs(), Asso_, AssoOriginal_, Cands_, DEFINE_FWK_MODULE, GenOrigs_, edm::Event::getByToken(), edm::Ref< C, T, F >::key(), HPSPFTaus_cff::mapping, maxRapidity_, reco::CompositeRefCandidateT< D >::motherRef(), eostools::move(), reco::CompositeRefCandidateT< D >::numberOfMothers(), edm::Event::put(), reco::LeafCandidate::status(), and reco::LeafCandidate::y().

69  {
70 
71 
73  iEvent.getByToken( Cands_, cands );
74  std::vector<reco::Candidate>::const_iterator cand;
75 
77  iEvent.getByToken( Asso_, asso );
78 
80  iEvent.getByToken( AssoOriginal_, assoOriginal);
81 
83  iEvent.getByToken( GenOrigs_, genOrigs);
84  std::vector<int> mapping(genOrigs->size(), -1);
85 
86  //invert the value map from Orig2New to New2Orig
87  std::map< edm::Ref<reco::GenParticleCollection> , edm::Ref<reco::GenParticleCollection> > reverseMap;
88  for(unsigned int ic=0, nc = genOrigs->size(); ic < nc; ++ic)
89  {
91  edm::Ref<reco::GenParticleCollection> newRef = (*assoOriginal)[originalRef];
92  reverseMap.insert(std::pair<edm::Ref<reco::GenParticleCollection>,edm::Ref<reco::GenParticleCollection>>(newRef,originalRef));
93  }
94 
95  auto outPtrP = std::make_unique<std::vector<pat::PackedGenParticle>>();
96 
97  unsigned int packed=0;
98  for(unsigned int ic=0, nc = cands->size(); ic < nc; ++ic) {
99  const reco::GenParticle &cand=(*cands)[ic];
100  if(cand.status() ==1 && std::abs(cand.y()) < maxRapidity_)
101  {
102  // Obtain original gen particle collection reference from input reference and map
104  edm::Ref<reco::GenParticleCollection> originalRef=reverseMap[inputRef];
105  mapping[originalRef.key()]=packed;
106  packed++;
107  if(cand.numberOfMothers() > 0) {
108  edm::Ref<reco::GenParticleCollection> newRef=(*asso)[cand.motherRef(0)];
109  outPtrP->push_back( pat::PackedGenParticle(cand,newRef));
110  } else {
112 
113  }
114 
115  }
116  }
117 
118 
120 
121  auto gp2pgp = std::make_unique<edm::Association<std::vector<pat::PackedGenParticle>>>(oh);
122  edm::Association< std::vector<pat::PackedGenParticle> >::Filler gp2pgpFiller(*gp2pgp);
123  gp2pgpFiller.insert(genOrigs, mapping.begin(), mapping.end());
124  gp2pgpFiller.fill();
125  iEvent.put(std::move(gp2pgp));
126 
127 
128 }
OrphanHandle< PROD > put(std::unique_ptr< PROD > product)
Put a new product.
Definition: Event.h:122
bool getByToken(EDGetToken token, Handle< PROD > &result) const
Definition: Event.h:460
virtual int status() const final
status word
key_type key() const
Accessor for product key.
Definition: Ref.h:265
const edm::EDGetTokenT< reco::GenParticleCollection > Cands_
virtual size_t numberOfMothers() const
number of mothers
Abs< T >::type abs(const T &t)
Definition: Abs.h:22
const edm::EDGetTokenT< edm::Association< reco::GenParticleCollection > > Asso_
daughters::value_type motherRef(size_type i=0) const
reference to mother at given position
virtual double y() const final
rapidity
const edm::EDGetTokenT< reco::GenParticleCollection > GenOrigs_
const edm::EDGetTokenT< edm::Association< reco::GenParticleCollection > > AssoOriginal_
def move(src, dest)
Definition: eostools.py:510

Member Data Documentation

const edm::EDGetTokenT<edm::Association<reco::GenParticleCollection> > pat::PATPackedGenParticleProducer::Asso_
private

Definition at line 50 of file PATPackedGenParticleProducer.cc.

Referenced by produce().

const edm::EDGetTokenT<edm::Association<reco::GenParticleCollection> > pat::PATPackedGenParticleProducer::AssoOriginal_
private

Definition at line 51 of file PATPackedGenParticleProducer.cc.

Referenced by produce().

const edm::EDGetTokenT<reco::GenParticleCollection> pat::PATPackedGenParticleProducer::Cands_
private

Definition at line 48 of file PATPackedGenParticleProducer.cc.

Referenced by produce().

const edm::EDGetTokenT<reco::GenParticleCollection> pat::PATPackedGenParticleProducer::GenOrigs_
private

Definition at line 49 of file PATPackedGenParticleProducer.cc.

Referenced by produce().

const double pat::PATPackedGenParticleProducer::maxRapidity_
private

Definition at line 52 of file PATPackedGenParticleProducer.cc.

Referenced by produce().