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 &)
 
void produce (edm::StreamID, edm::Event &, const edm::EventSetup &) const override
 
 ~PATPackedGenParticleProducer () override
 
- Public Member Functions inherited from edm::global::EDProducer<>
 EDProducer ()=default
 
- Public Member Functions inherited from edm::global::EDProducerBase
 EDProducerBase ()
 
ModuleDescription const & moduleDescription () const
 
 ~EDProducerBase () override
 
- 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 ()
 
std::vector< edm::ProductResolverIndex > const & putTokenIndexToProductResolverIndex () const
 
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, ModuleToResolverIndicies 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
using ModuleToResolverIndicies = std::unordered_multimap< std::string, std::tuple< edm::TypeID const *, const char *, edm::ProductResolverIndex >>
 
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 ( )
override

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
override

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 >::isNonnull(), 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 
76  //from prunedGenParticlesWithStatusOne to prunedGenParticles
78  iEvent.getByToken( Asso_, asso );
79 
81  iEvent.getByToken( AssoOriginal_, assoOriginal);
82 
84  iEvent.getByToken( GenOrigs_, genOrigs);
85  std::vector<int> mapping(genOrigs->size(), -1);
86 
87  //invert the value map from Orig2New to New2Orig
88  std::map< edm::Ref<reco::GenParticleCollection> , edm::Ref<reco::GenParticleCollection> > reverseMap;
89  for(unsigned int ic=0, nc = genOrigs->size(); ic < nc; ++ic)
90  {
92  edm::Ref<reco::GenParticleCollection> newRef = (*assoOriginal)[originalRef];
93  reverseMap.insert(std::pair<edm::Ref<reco::GenParticleCollection>,edm::Ref<reco::GenParticleCollection>>(newRef,originalRef));
94  }
95 
96  auto outPtrP = std::make_unique<std::vector<pat::PackedGenParticle>>();
97 
98  unsigned int packed=0;
99  for(unsigned int ic=0, nc = cands->size(); ic < nc; ++ic) {
100  const reco::GenParticle &cand=(*cands)[ic];
101  if(cand.status() ==1 && std::abs(cand.y()) < maxRapidity_)
102  {
103  // Obtain original gen particle collection reference from input reference and map
105  edm::Ref<reco::GenParticleCollection> originalRef=reverseMap[inputRef];
106  edm::Ref<reco::GenParticleCollection> finalPrunedRef=(*asso)[inputRef];
107  mapping[originalRef.key()]=packed;
108  packed++;
109  if(finalPrunedRef.isNonnull()){ //this particle exists also in the final pruned
110  outPtrP->push_back( pat::PackedGenParticle(cand,finalPrunedRef));
111  }else{
112  if(cand.numberOfMothers() > 0) {
113  edm::Ref<reco::GenParticleCollection> newRef=(*asso)[cand.motherRef(0)];
114  outPtrP->push_back( pat::PackedGenParticle(cand,newRef));
115  } else {
117  }
118  }
119 
120  }
121  }
122 
123 
125 
126  auto gp2pgp = std::make_unique<edm::Association<std::vector<pat::PackedGenParticle>>>(oh);
127  edm::Association< std::vector<pat::PackedGenParticle> >::Filler gp2pgpFiller(*gp2pgp);
128  gp2pgpFiller.insert(genOrigs, mapping.begin(), mapping.end());
129  gp2pgpFiller.fill();
130  iEvent.put(std::move(gp2pgp));
131 
132 
133 }
OrphanHandle< PROD > put(std::unique_ptr< PROD > product)
Put a new product.
Definition: Event.h:127
bool isNonnull() const
Checks for non-null.
Definition: Ref.h:253
bool getByToken(EDGetToken token, Handle< PROD > &result) const
Definition: Event.h:508
size_t numberOfMothers() const override
number of mothers
double y() const final
rapidity
key_type key() const
Accessor for product key.
Definition: Ref.h:265
const edm::EDGetTokenT< reco::GenParticleCollection > Cands_
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
int status() const final
status word
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().