Inheritance diagram for GenParticleProducer:

Public Member Functions
int	chargeTimesThree (int) const

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

bool	fillDaughters (reco::GenParticleCollection &cand, const HepMC::GenParticle *part, size_t index)

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

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

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

	~GenParticleProducer ()
	destructor More...

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)

Public Attributes
std::map< int, size_t >	barcodes_

reco::GenParticleRefProd	ref_

Private Attributes
bool	abortOnUnknownPDGCode_
	unknown code treatment flag More...

std::vector< int >	chargeM_

std::map< int, int >	chargeMap_

std::vector< int >	chargeP_
	charge indices More...

bool	doSubEvent_
	input & output modes More...

bool	firstEvent_
	whether the first event was looked at 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::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 26 of file GenParticleProducer.cc.

Constructor & Destructor Documentation

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

constructor

Definition at line 91 of file GenParticleProducer.cc.

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

                                                                    :
   firstEvent_(true),
   abortOnUnknownPDGCode_( cfg.getUntrackedParameter<bool>( "abortOnUnknownPDGCode", true ) ),
   saveBarCodes_( cfg.getUntrackedParameter<bool>( "saveBarCodes", false ) ),
   chargeP_( PDGCacheMax, 0 ), chargeM_( PDGCacheMax, 0 ),
   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 111 of file GenParticleProducer.cc.

111 {

112 }

Member Function Documentation

int GenParticleProducer::chargeTimesThree ( int id ) const

Definition at line 114 of file GenParticleProducer.cc.

References abortOnUnknownPDGCode_, funct::abs(), chargeM_, chargeMap_, chargeP_, Exception, f, hcalTTPDigis_cfi::id, edm::errors::LogicError, source_particleGun_cfi::ParticleID, and PDGCacheMax.

Referenced by convertParticle().

                                                         {
   if( std::abs( id ) < PDGCacheMax )
     return id > 0 ? chargeP_[ id ] : chargeM_[ - id ];
   map<int, int>::const_iterator f = chargeMap_.find( id );
   if ( f == chargeMap_.end() )  {
     if ( abortOnUnknownPDGCode_ )
       throw edm::Exception( edm::errors::LogicError )
     << "invalid PDG id: " << id << endl;
     else
       return HepPDT::ParticleID(id).threeCharge();
   }
   return f->second;
 }

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

Definition at line 280 of file GenParticleProducer.cc.

References chargeTimesThree(), 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(), and findQualityFiles::v.

Referenced by produce().

                                                                                              {
    Candidate::LorentzVector p4( part->momentum() );
    int pdgId = part->pdg_id();
    cand.setThreeCharge( 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,
		size_t	index
	)

Definition at line 300 of file GenParticleProducer.cc.

References barcodes_, GenParticle::GenParticle, diffTreeTool::index, funct::m, ref_, 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
	)

Definition at line 317 of file GenParticleProducer.cc.

References barcodes_, 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;
 }

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

overridevirtual

process one event

fill indices

Definition at line 128 of file GenParticleProducer.cc.

References funct::abs(), barcodes_, chargeM_, chargeMap_, chargeP_, convertParticle(), edmIntegrityCheck::d, doSubEvent_, Exception, f, fillDaughters(), fillIndices(), firstEvent_, GenParticle::GenParticle, edm::Event::getByToken(), edm::EventSetup::getData(), edm::HepMCProduct::GetEvent(), MixCollection< T >::getObject(), edm::Event::getRefBeforePut(), i, edm::RefProd< C >::id(), edm::errors::InvalidReference, LogDebug, CaloTowersParam_cfi::mc, mixToken_, mmToCm, mmToNs, eostools::move(), PFRecoTauDiscriminationByIsolation_cfi::offset, AlCaHLTBitMon_ParallelJobs::p, source_particleGun_cfi::ParticleID, HadronAndPartonSelector_cfi::particles, PDGCacheMax, cosmictrackingParticleSelector_cfi::pdgId, position, edm::Handle< T >::product(), edm::Event::put(), ref_, reco::CompositeRefCandidateT< D >::resetDaughters(), saveBarCodes_, MixCollection< T >::size(), findQualityFiles::size, srcToken_, and useCF_.

                                                                     {
 
   if (firstEvent_) {
      ESHandle<HepPDT::ParticleDataTable> pdt;
      es.getData( pdt );
      for( HepPDT::ParticleDataTable::const_iterator p = pdt->begin(); p != pdt->end(); ++ p ) {
        const HepPDT::ParticleID & id = p->first;
        int pdgId = id.pid(), apdgId = std::abs( pdgId );
        int q3 = id.threeCharge();
        if ( apdgId < PDGCacheMax && pdgId > 0 ) {
      chargeP_[ apdgId ] = q3;
      chargeM_[ apdgId ] = -q3;
        } else if ( apdgId < PDGCacheMax ) {
      chargeP_[ apdgId ] = -q3;
      chargeM_[ apdgId ] = q3;
        } else {
      chargeMap_[ pdgId ] = q3;
      chargeMap_[ -pdgId ] = -q3;
        }
      }
      firstEvent_ = false;
    }
 
    barcodes_.clear();
 
    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));
   ref_ = evt.getRefBeforePut<GenParticleCollection>();
   GenParticleCollection & cands = * candsPtr;
   size_t offset = 0;
   size_t suboffset = 0;
 
   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);
     // 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);
        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,d);
        }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);
 
      // 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);
     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,d);
     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");
 }