Inheritance diagram for GenParticleProducer:

Public Member Functions
bool	convertParticle (reco::GenParticle &cand, const HepMC::GenParticle *part, const IDto3Charge &id2Charge) const

bool	fillDaughters (reco::GenParticleCollection &cand, const HepMC::GenParticle *part, reco::GenParticleRefProd const &ref, size_t index, std::unordered_map< int, size_t > &barcodes) const

bool	fillIndices (const HepMC::GenEvent mc, std::vector< const HepMC::GenParticle > &particles, std::vector< int > &barCodeVector, int offset, std::unordered_map< int, size_t > &barcodes) const

	GenParticleProducer (const edm::ParameterSet &)
	constructor More...

virtual std::shared_ptr< IDto3Charge >	globalBeginRun (const edm::Run &, const edm::EventSetup &) const override

virtual void	globalEndRun (edm::Run const &, edm::EventSetup const &) const override

virtual void	produce (edm::StreamID, edm::Event &e, const edm::EventSetup &) const override
	process one event More...

	~GenParticleProducer ()
	destructor More...

Public Member Functions inherited from edm::global::EDProducer< edm::RunCache< IDto3Charge > >
	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< 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 &	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

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 Attributes
bool	abortOnUnknownPDGCode_
	unknown code treatment flag More...

bool	doSubEvent_
	input & output modes More...

MCTruthHelper< HepMC::GenParticle >	mcTruthHelper_

MCTruthHelper< reco::GenParticle >	mcTruthHelperGenParts_

edm::EDGetTokenT< CrossingFrame< edm::HepMCProduct > >	mixToken_

bool	saveBarCodes_
	save bar-codes More...

edm::EDGetTokenT< edm::HepMCProduct >	srcToken_
	source collection name More...

bool	useCF_

std::vector< edm::EDGetTokenT< edm::HepMCProduct > >	vectorSrcTokens_

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 79 of file GenParticleProducer.cc.

Constructor & Destructor Documentation

GenParticleProducer::GenParticleProducer ( const edm::ParameterSet & cfg )

constructor

Definition at line 138 of file GenParticleProducer.cc.

References egammaCTFFinalFitWithMaterial_cff::alias, edm::ParameterSet::getParameter(), mixToken_, saveBarCodes_, srcToken_, AlCaHLTBitMon_QueryRunRegistry::string, and useCF_.

                                                                    :
   abortOnUnknownPDGCode_( cfg.getUntrackedParameter<bool>( "abortOnUnknownPDGCode", true ) ),
   saveBarCodes_( cfg.getUntrackedParameter<bool>( "saveBarCodes", false ) ),
   doSubEvent_(cfg.getUntrackedParameter<bool>( "doSubEvent", false )),
   useCF_(cfg.getUntrackedParameter<bool>( "useCrossingFrame", false ))
 {
   produces<GenParticleCollection>();
   produces<math::XYZPointF>("xyz0");
   produces<float>("t0");
   if( saveBarCodes_ ) {
     std::string alias( cfg.getParameter<std::string>( "@module_label" ) );
     produces<vector<int> >().setBranchAlias( alias + "BarCodes" );
   }
 
   if(useCF_) mixToken_ = mayConsume<CrossingFrame<HepMCProduct> >(InputTag(cfg.getParameter<std::string>( "mix" ),"generatorSmeared"));
   else srcToken_ = mayConsume<HepMCProduct>(cfg.getParameter<InputTag>( "src" ));
 }

GenParticleProducer::~GenParticleProducer ( )

destructor

Definition at line 156 of file GenParticleProducer.cc.

156 {

157 }

Member Function Documentation

bool GenParticleProducer::convertParticle	(	reco::GenParticle &	cand,
		const HepMC::GenParticle *	part,
		const IDto3Charge &	id2Charge
	)		const

Definition at line 297 of file GenParticleProducer.cc.

References MCTruthHelper< P >::fillGenStatusFlags(), mcTruthHelper_, mmToCm, p4, cosmictrackingParticleSelector_cfi::pdgId, reco::GenParticle::setCollisionId(), reco::LeafCandidate::setP4(), reco::LeafCandidate::setPdgId(), reco::LeafCandidate::setStatus(), reco::LeafCandidate::setThreeCharge(), reco::LeafCandidate::setVertex(), reco::GenParticle::statusFlags(), findQualityFiles::v, and badGlobalMuonTaggersAOD_cff::vtx.

Referenced by produce().

                                                                                                                                   {
    Candidate::LorentzVector p4( part->momentum() );
    int pdgId = part->pdg_id();
    cand.setThreeCharge( id2Charge.chargeTimesThree( pdgId ) );
    cand.setPdgId( pdgId );
    cand.setStatus( part->status() );
    cand.setP4( p4 );
    cand.setCollisionId(0);
    const GenVertex * v = part->production_vertex();
    if ( v != 0 ) {
       ThreeVector vtx = v->point3d();
       Candidate::Point vertex( vtx.x() * mmToCm, vtx.y() * mmToCm, vtx.z() * mmToCm );
       cand.setVertex( vertex );
    } else {
       cand.setVertex( Candidate::Point( 0, 0, 0 ) );
    }
    mcTruthHelper_.fillGenStatusFlags(*part, cand.statusFlags());
    return true;
 }

bool GenParticleProducer::fillDaughters	(	reco::GenParticleCollection &	cand,
		const HepMC::GenParticle *	part,
		reco::GenParticleRefProd const &	ref,
		size_t	index,
		std::unordered_map< int, size_t > &	barcodes
	)		const

Definition at line 317 of file GenParticleProducer.cc.

References GenParticle::GenParticle, diffTreeTool::index, funct::m, and edm::second().

Referenced by produce().

                                                                                                                                                                                                        {
 
    const GenVertex * productionVertex = part->production_vertex();
    size_t numberOfMothers = productionVertex->particles_in_size();
    if ( numberOfMothers > 0 ) {
       GenVertex::particles_in_const_iterator motherIt = productionVertex->particles_in_const_begin();
       for( ; motherIt != productionVertex->particles_in_const_end(); motherIt++) {
      const HepMC::GenParticle * mother = * motherIt;
      size_t m = barcodes.find( mother->barcode() )->second;
      cands[ m ].addDaughter( GenParticleRef( ref, index ) );
      cands[ index ].addMother( GenParticleRef( ref, m ) );
       }
    }
 
    return true;
 }

bool GenParticleProducer::fillIndices	(	const HepMC::GenEvent *	mc,
		std::vector< const HepMC::GenParticle * > &	particles,
		std::vector< int > &	barCodeVector,
		int	offset,
		std::unordered_map< int, size_t > &	barcodes
	)		const

Definition at line 334 of file GenParticleProducer.cc.

References begin, DEFINE_FWK_MODULE, end, GenParticle::GenParticle, training_settings::idx, PFRecoTauDiscriminationByIsolation_cfi::offset, and AlCaHLTBitMon_ParallelJobs::p.

Referenced by produce().

                                                                                                                                                                                                  {
    size_t idx = offset;
    HepMC::GenEvent::particle_const_iterator begin = mc->particles_begin(), end = mc->particles_end();
    for( HepMC::GenEvent::particle_const_iterator p = begin; p != end; ++ p ) {
       const HepMC::GenParticle * particle = * p;
       size_t barCode_this_event = particle->barcode();
       size_t barCode = barCode_this_event + offset;
       if( barcodes.find(barCode) != barcodes.end() )
      throw cms::Exception( "WrongReference" )
         << "barcodes are duplicated! " << endl;
       particles[idx] = particle;
       barCodeVector[idx] = barCode;
       barcodes.insert( make_pair(barCode_this_event, idx ++) );
    }
       return true;
 }

std::shared_ptr< IDto3Charge > GenParticleProducer::globalBeginRun	(	const edm::Run &	,
		const edm::EventSetup &	es
	)		const

overridevirtual

Definition at line 159 of file GenParticleProducer.cc.

References abortOnUnknownPDGCode_, and edm::EventSetup::getData().

                                                                                                     {
   ESHandle<HepPDT::ParticleDataTable> pdt;
   es.getData( pdt );
   return std::make_shared<IDto3Charge>(*pdt, abortOnUnknownPDGCode_);
 }

virtual void GenParticleProducer::globalEndRun	(	edm::Run const &	,
		edm::EventSetup const &
	)		const

inlineoverridevirtual

Definition at line 89 of file GenParticleProducer.cc.

References GenParticle::GenParticle, diffTreeTool::index, CaloTowersParam_cfi::mc, PFRecoTauDiscriminationByIsolation_cfi::offset, and HadronAndPartonSelector_cfi::particles.

89 {};

void GenParticleProducer::produce	(	edm::StreamID	,
		edm::Event &	e,
		const edm::EventSetup &	es
	)		const

overridevirtual

process one event

fill indices

Definition at line 165 of file GenParticleProducer.cc.

References convertParticle(), edmIntegrityCheck::d, doSubEvent_, Exception, f, fillDaughters(), fillIndices(), GenParticle::GenParticle, edm::Event::getByToken(), edm::HepMCProduct::GetEvent(), MixCollection< T >::getObject(), edm::Event::getRefBeforePut(), edm::Event::getRun(), mps_fire::i, edm::RefProd< C >::id(), edm::Run::index(), edm::errors::InvalidReference, LogDebug, CaloTowersParam_cfi::mc, mixToken_, mmToCm, mmToNs, eostools::move(), PFRecoTauDiscriminationByIsolation_cfi::offset, HadronAndPartonSelector_cfi::particles, position, edm::Handle< T >::product(), edm::Event::put(), reco::CompositeRefCandidateT< D >::resetDaughters(), saveBarCodes_, MixCollection< T >::size(), findQualityFiles::size, srcToken_, and useCF_.

                                                                                     {
 
    std::unordered_map<int, size_t> barcodes;
 
    size_t totalSize = 0;
    const GenEvent * mc = 0;
    MixCollection<HepMCProduct>* cfhepmcprod = 0;
    size_t npiles = 1;
 
    if(useCF_){
       Handle<CrossingFrame<HepMCProduct> > cf;
       evt.getByToken(mixToken_,cf);
       cfhepmcprod = new MixCollection<HepMCProduct>(cf.product());
       npiles = cfhepmcprod->size();
       LogDebug("GenParticleProducer")<<"npiles : "<<npiles<<endl;
       for(unsigned int icf = 0; icf < npiles; ++icf){
      LogDebug("GenParticleProducer")<<"subSize : "<<cfhepmcprod->getObject(icf).GetEvent()->particles_size()<<endl;
      totalSize += cfhepmcprod->getObject(icf).GetEvent()->particles_size();
       }
       LogDebug("GenParticleProducer")<<"totalSize : "<<totalSize<<endl;
    }else{
       Handle<HepMCProduct> mcp;
       evt.getByToken( srcToken_, mcp );
       mc = mcp->GetEvent();
       if( mc == 0 )
      throw edm::Exception( edm::errors::InvalidReference )
         << "HepMC has null pointer to GenEvent" << endl;
       totalSize  = mc->particles_size();
    }
 
    // initialise containers
    const size_t size = totalSize;
   vector<const HepMC::GenParticle *> particles( size );
   auto candsPtr = std::make_unique<GenParticleCollection>(size);
   auto barCodeVector = std::make_unique<vector<int>>(size);
   std::unique_ptr<math::XYZPointF> xyz0Ptr(new math::XYZPointF(0.,0.,0.));
   std::unique_ptr<float> t0Ptr(new float(0.f));
   reco::GenParticleRefProd ref = evt.getRefBeforePut<GenParticleCollection>();
   GenParticleCollection & cands = * candsPtr;
   size_t offset = 0;
   size_t suboffset = 0;
 
   IDto3Charge const& id2Charge = *runCache(evt.getRun().index());
   if(doSubEvent_ || useCF_){
      for(size_t ipile = 0; ipile < npiles; ++ipile){
     LogDebug("GenParticleProducer")<<"mixed object ipile : "<<ipile<<endl;
     barcodes.clear();
     if(useCF_) mc = cfhepmcprod->getObject(ipile).GetEvent();
 
     //Look whether heavy ion/signal event
     bool isHI = false;
     const HepMC::HeavyIon * hi = mc->heavy_ion();
     if(hi && hi->Ncoll_hard() > 1) isHI = true;
     size_t num_particles = mc->particles_size();
     LogDebug("GenParticleProducer")<<"num_particles : "<<num_particles<<endl;
         if (ipile == 0) {
             auto origin = (*mc->vertices_begin())->position();
             xyz0Ptr->SetXYZ(origin.x() * mmToCm, origin.y() * mmToCm, origin.z() * mmToCm);
             *t0Ptr = origin.t() * mmToNs;
         }
     fillIndices(mc, particles, *barCodeVector, offset,barcodes);
     // fill output collection and save association
     for( size_t ipar = offset; ipar < offset + num_particles; ++ ipar ) {
        const HepMC::GenParticle * part = particles[ ipar ];
        reco::GenParticle & cand = cands[ ipar ];
        // convert HepMC::GenParticle to new reco::GenParticle
        convertParticle(cand, part,id2Charge);
        cand.resetDaughters( ref.id() );
     }
 
     for( size_t d = offset; d < offset + num_particles; ++ d ) {
        const HepMC::GenParticle * part = particles[ d ];
        const GenVertex * productionVertex = part->production_vertex();
        int sub_id = 0;
        if ( productionVertex != 0 ) {
           sub_id = productionVertex->id();
           if(!isHI) sub_id = 0;
           // search barcode map and attach daughters
           fillDaughters(cands,part,ref,d,barcodes);
        }else{
           const GenVertex * endVertex = part->end_vertex();
           if(endVertex != 0) sub_id = endVertex->id();
           else throw cms::Exception( "SubEventID" )<<"SubEvent not determined. Particle has no production and no end vertex!"<<endl;
        }
        if(sub_id < 0) sub_id = 0;
        int new_id = sub_id + suboffset;
        GenParticleRef dref( ref, d );
        cands[d].setCollisionId(new_id); // For new GenParticle
        LogDebug("VertexId")<<"SubEvent offset 3 : "<<suboffset;
     }
     int nsub = -2;
     if(isHI){
        nsub = hi->Ncoll_hard()+1;
        suboffset += nsub;
     }else{
        suboffset += 1;
     }
     offset += num_particles;
      }
   }else{
      auto origin = (*mc->vertices_begin())->position();
      xyz0Ptr->SetXYZ(origin.x() * mmToCm, origin.y() * mmToCm, origin.z() * mmToCm);
      *t0Ptr = origin.t() * mmToNs;
      fillIndices(mc, particles, *barCodeVector, 0,barcodes);
 
      // fill output collection and save association
      for( size_t i = 0; i < particles.size(); ++ i ) {
     const HepMC::GenParticle * part = particles[ i ];
     reco::GenParticle & cand = cands[ i ];
     // convert HepMC::GenParticle to new reco::GenParticle
     convertParticle(cand, part,id2Charge);
     cand.resetDaughters( ref.id() );
      }
 
      // fill references to daughters
      for( size_t d = 0; d < cands.size(); ++ d ) {
     const HepMC::GenParticle * part = particles[ d ];
     const GenVertex * productionVertex = part->production_vertex();
     // search barcode map and attach daughters
     if ( productionVertex != 0 ) fillDaughters(cands,part,ref,d,barcodes);
     cands[d].setCollisionId(0);
      }
   }
   
   evt.put(std::move(candsPtr));
   if(saveBarCodes_) evt.put(std::move(barCodeVector));
   if(cfhepmcprod) delete cfhepmcprod;
   evt.put(std::move(xyz0Ptr),"xyz0");
   evt.put(std::move(t0Ptr),"t0");
 }