GenPUProtonProducer Class Reference

Inheritance diagram for GenPUProtonProducer:

Public Member Functions
	GenPUProtonProducer (const edm::ParameterSet &)
std::shared_ptr< ConvertParticle >	globalBeginRun (const edm::Run &, const edm::EventSetup &) const override
void	globalEndRun (edm::Run const &, edm::EventSetup const &) const override
void	produce (edm::StreamID, edm::Event &e, const edm::EventSetup &) const override
	~GenPUProtonProducer () override
Public Member Functions inherited from edm::global::EDProducer< edm::RunCache< ConvertParticle > >
	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< ConsumesInfo >	consumesInfo () 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
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_
std::vector< int >	bunchList_
double	minPz_
edm::EDGetTokenT< CrossingFrame< edm::HepMCProduct > >	mixToken_
SelectProton	select_

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 132 of file GenPUProtonProducer.cc.

Constructor & Destructor Documentation

GenPUProtonProducer::GenPUProtonProducer

(

const edm::ParameterSet &

cfg

)

Definition at line 170 of file GenPUProtonProducer.cc.

References edm::ParameterSet::getParameter(), mixToken_, and AlCaHLTBitMon_QueryRunRegistry::string.

                                                                   :
  abortOnUnknownPDGCode_( cfg.getUntrackedParameter<bool>( "abortOnUnknownPDGCode", true ) ),
  bunchList_( cfg.getParameter<vector<int> >( "bunchCrossingList" ) ),
  minPz_( cfg.getParameter<double>( "minPz" ) ) 
{
  produces<GenParticleCollection>();
  mixToken_ = consumes<CrossingFrame<HepMCProduct> >( InputTag(cfg.getParameter<std::string>( "mix" ),"generatorSmeared") );
}

GenPUProtonProducer::~GenPUProtonProducer ( )

override

Definition at line 179 of file GenPUProtonProducer.cc.

{ }

Member Function Documentation

std::shared_ptr< ConvertParticle > GenPUProtonProducer::globalBeginRun ( const edm::Run &  , const edm::EventSetup &  es  ) const

override

Definition at line 181 of file GenPUProtonProducer.cc.

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

                                                                                                         {
  ESHandle<HepPDT::ParticleDataTable> pdt;
  es.getData( pdt );
  auto convert_ptr = std::make_shared<ConvertParticle>( abortOnUnknownPDGCode_ );
  if( !convert_ptr->initialized() ) convert_ptr->init( *pdt );

  return convert_ptr;
}

void GenPUProtonProducer::globalEndRun ( edm::Run const &  , edm::EventSetup const &    ) const

inlineoverride

Definition at line 139 of file GenPUProtonProducer.cc.

{};

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

override

Definition at line 190 of file GenPUProtonProducer.cc.

References MixCollection< T >::MixItr::bunch(), bunchList_, DEFINE_FWK_MODULE, spr::find(), GenParticle::GenParticle, edm::Event::getByToken(), edm::Event::getRun(), edm::Run::index(), LogDebug, minPz_, mixToken_, eostools::move(), AlCaHLTBitMon_ParallelJobs::p, edm::Handle< T >::product(), edm::Event::put(), and select_.

                                                                                    {

   size_t totalSize = 0;
   size_t npiles = 1;

   Handle<CrossingFrame<HepMCProduct> > cf;
   evt.getByToken(mixToken_,cf);
   std::unique_ptr<MixCollection<HepMCProduct> > cfhepmcprod( new MixCollection<HepMCProduct>( cf.product() ) );
   npiles = cfhepmcprod->size();

   LogDebug("GenPUProtonProducer") << " Number of pile-up events : " << npiles << endl;

   for(size_t icf = 0; icf < npiles; ++icf){
      LogDebug("GenPUProtonProducer") << "CF " << icf << " size : " << cfhepmcprod->getObject(icf).GetEvent()->particles_size() << endl;
      totalSize += cfhepmcprod->getObject(icf).GetEvent()->particles_size();
   }
   LogDebug("GenPUProtonProducer") << "Total size : " << totalSize << endl;

   // Initialise containers
   auto candsPtr = std::make_unique<GenParticleCollection>();
   GenParticleCollection& cands = *candsPtr;

   // Loop over pile-up events
   ConvertParticle const& convertParticle_ = *runCache( evt.getRun().index() );

   MixCollection<HepMCProduct>::MixItr mixHepMC_itr;
   unsigned int total_number_of_protons = 0;
   size_t idx_mix = 0;
   // Fill collection
   for( mixHepMC_itr = cfhepmcprod->begin() ; mixHepMC_itr != cfhepmcprod->end() ; ++mixHepMC_itr, ++idx_mix ){
      int bunch = mixHepMC_itr.bunch();

      if( find( bunchList_.begin(), bunchList_.end(), bunch ) != bunchList_.end() ){

     auto event = (*mixHepMC_itr).GetEvent();

     size_t num_particles = event->particles_size();

     // Fill output collection
     unsigned int number_of_protons = 0;
     for( auto p = event->particles_begin() ; p != event->particles_end() ; ++p ) {
        HepMC::GenParticle const* part = *p;  
        if( select_(part, minPz_) ) {
           reco::GenParticle cand;
           convertParticle_( cand, part );
               ++number_of_protons;
               cands.push_back( cand );
        } 
     }
     LogDebug("GenPUProtonProducer") << "Idx : " << idx_mix << " Bunch : " << bunch 
                                         << " Number of particles : " << num_particles 
                                         << " Number of protons : " << number_of_protons << endl;

         total_number_of_protons += number_of_protons;
      }
   }
   LogDebug("GenPUProtonProducer") << "Total number of protons : " << total_number_of_protons << endl;
   LogDebug("GenPUProtonProducer") << "Output collection size : " << cands.size() << endl;

   evt.put( std::move(candsPtr) );
}

Member Data Documentation

bool GenPUProtonProducer::abortOnUnknownPDGCode_

private

Definition at line 144 of file GenPUProtonProducer.cc.

Referenced by globalBeginRun().

std::vector<int> GenPUProtonProducer::bunchList_

private

Definition at line 145 of file GenPUProtonProducer.cc.

Referenced by produce().

double GenPUProtonProducer::minPz_

private

Definition at line 146 of file GenPUProtonProducer.cc.

Referenced by produce().

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

private

Definition at line 139 of file GenPUProtonProducer.cc.

Referenced by GenPUProtonProducer(), and produce().

SelectProton GenPUProtonProducer::select_

private

Definition at line 147 of file GenPUProtonProducer.cc.

Referenced by produce().