#include <MCMultiParticleFilter.h>

Inheritance diagram for MCMultiParticleFilter:

Public Member Functions
	MCMultiParticleFilter (const edm::ParameterSet &)

	~MCMultiParticleFilter () override

Public Member Functions inherited from edm::EDFilter
	EDFilter ()

SerialTaskQueue *	globalLuminosityBlocksQueue ()

SerialTaskQueue *	globalRunsQueue ()

ModuleDescription const &	moduleDescription () const

	~EDFilter () 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)

	~ProducerBase () noexcept(false) override

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

ESProxyIndex const *	esGetTokenIndices (edm::Transition iTrans) const

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

ProductResolverIndexAndSkipBit	uncheckedIndexFrom (EDGetToken) const

void	updateLookup (BranchType iBranchType, ProductResolverIndexHelper const &, bool iPrefetchMayGet)

void	updateLookup (eventsetup::ESRecordsToProxyIndices const &)

virtual	~EDConsumerBase () noexcept(false)

Private Member Functions
void	endJob () override

bool	filter (edm::Event &, const edm::EventSetup &) override

Private Attributes
bool	acceptMore_

std::vector< double >	decayRadiusMax

std::vector< double >	decayRadiusMin

std::vector< double >	decayZMax

std::vector< double >	decayZMin

std::vector< double >	etaMax_

std::vector< int >	motherID_

int	numRequired_

std::vector< int >	particleID_

int	passedEvents_

std::vector< double >	ptMin_

edm::InputTag	src_

std::vector< int >	status_

edm::EDGetTokenT< edm::HepMCProduct >	token_

int	totalEvents_

Additional Inherited Members
Public Types inherited from edm::EDFilter
typedef EDFilter	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::EDFilter
static const std::string &	baseType ()

static void	fillDescriptions (ConfigurationDescriptions &descriptions)

static void	prevalidate (ConfigurationDescriptions &)

static bool	wantsGlobalLuminosityBlocks ()

static bool	wantsGlobalRuns ()

static bool	wantsStreamLuminosityBlocks ()

static bool	wantsStreamRuns ()

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 ESProduct , typename ESRecord , Transition Tr = Transition::Event>
auto	esConsumes ()

template<typename ESProduct , typename ESRecord , Transition Tr = Transition::Event>
auto	esConsumes (ESInputTag const &tag)

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 41 of file MCMultiParticleFilter.h.

Constructor & Destructor Documentation

MCMultiParticleFilter::MCMultiParticleFilter ( const edm::ParameterSet & iConfig )

explicit

Definition at line 4 of file MCMultiParticleFilter.cc.

References decayRadiusMax, decayRadiusMin, decayZMax, decayZMin, etaMax_, edm::ParameterSet::getUntrackedParameter(), mps_fire::i, motherID_, particleID_, ptMin_, and status_.

                                                                            :
   src_(iConfig.getUntrackedParameter<edm::InputTag>("src",edm::InputTag(std::string("generator"),"unsmeared"))),
   token_(consumes<edm::HepMCProduct>(src_)),
   numRequired_(iConfig.getParameter<int>("NumRequired")),
   acceptMore_(iConfig.getParameter<bool>("AcceptMore")),
   particleID_(iConfig.getParameter< std::vector<int> >("ParticleID")),
   ptMin_(iConfig.getParameter< std::vector<double> >("PtMin")),
   etaMax_(iConfig.getParameter< std::vector<double> >("EtaMax")),
   status_(iConfig.getParameter< std::vector<int> >("Status")),
   totalEvents_(0), passedEvents_(0)
 {
   //here do whatever other initialization is needed
 
   // default pt, eta, status cuts to "don't care"
   std::vector<double> defptmin(1, 0);
   std::vector<double> defetamax(1, 999.0);
   std::vector<int> defstat(1, 0);
   std::vector<int> defmother;
   defmother.push_back(0);
   motherID_ = iConfig.getUntrackedParameter< std::vector<int> >("MotherID", defmother);
     
   std::vector<double> defDecayRadiusmin;
   defDecayRadiusmin.push_back(-1.);
   decayRadiusMin = iConfig.getUntrackedParameter<std::vector<double> >("MinDecayRadius", defDecayRadiusmin);
 
   std::vector<double> defDecayRadiusmax;
   defDecayRadiusmax.push_back(1.e5);
   decayRadiusMax = iConfig.getUntrackedParameter<std::vector<double> >("MaxDecayRadius", defDecayRadiusmax);
 
   std::vector<double> defDecayZmin;
   defDecayZmin.push_back(-1.e5);
   decayZMin = iConfig.getUntrackedParameter<std::vector<double> >("MinDecayZ", defDecayZmin);
 
   std::vector<double> defDecayZmax;
   defDecayZmax.push_back(1.e5);
   decayZMax = iConfig.getUntrackedParameter<std::vector<double> >("MaxDecayZ", defDecayZmax);
 
   // check for same size
   if ( (ptMin_.size() > 1 &&  particleID_.size() != ptMin_.size()) 
        ||  (etaMax_.size() > 1 && particleID_.size() != etaMax_.size()) 
        ||  (status_.size() > 1 && particleID_.size() != status_.size()) 
        ||  (motherID_.size() > 1 && particleID_.size() != motherID_.size())
        ||  (decayRadiusMin.size() > 1 && particleID_.size() != decayRadiusMin.size())
        ||  (decayRadiusMax.size() > 1 && particleID_.size() != decayRadiusMax.size())
        ||  (decayZMin.size() > 1 && particleID_.size() != decayZMin.size())
        ||  (decayZMax.size() > 1 && particleID_.size() != decayZMax.size())
        ) {
     edm::LogWarning("MCMultiParticleFilter") << "WARNING: MCMultiParticleFilter: size of PtMin, EtaMax, motherID, decayRadiusMin, decayRadiusMax, decayZMin, decayZMax and/or Status does not match ParticleID size!" << std::endl;
   }
   
   // Fill arrays with defaults if necessary
   while (ptMin_.size() < particleID_.size())
     ptMin_.push_back(defptmin[0]);
   while (etaMax_.size() < particleID_.size())
     etaMax_.push_back(defetamax[0]);
   while (status_.size() < particleID_.size())
     status_.push_back(defstat[0]);
   while (motherID_.size() < particleID_.size())
     motherID_.push_back(defmother[0]);
     
   // if decayRadiusMin size smaller than particleID , fill up further with defaults
   if (particleID_.size() > decayRadiusMin.size()) {
     for (unsigned int i = decayRadiusMin.size(); i < particleID_.size(); i++) {
       decayRadiusMin.push_back(-10.);
     }
   }
   // if decayRadiusMax size smaller than particleID , fill up further with defaults
   if (particleID_.size() > decayRadiusMax.size()) {
     for (unsigned int i = decayRadiusMax.size(); i < particleID_.size(); i++) {
       decayRadiusMax.push_back(1.e5);
     }
   }
 
   // if decayZMin size smaller than particleID , fill up further with defaults
   if (particleID_.size() > decayZMin.size()) {
     for (unsigned int i = decayZMin.size(); i < particleID_.size(); i++) {
       decayZMin.push_back(-1.e5);
     }
   }
   // if decayZMax size smaller than particleID , fill up further with defaults
   if (particleID_.size() > decayZMax.size()) {
     for (unsigned int i = decayZMax.size(); i < particleID_.size(); i++) {
       decayZMax.push_back(1.e5);
     }
   }
 }

MCMultiParticleFilter::~MCMultiParticleFilter ( )

override

Definition at line 91 of file MCMultiParticleFilter.cc.

 {
  
    // do anything here that needs to be done at destruction time
    // (e.g. close files, deallocate resources etc.)
 
 }

Member Function Documentation

void MCMultiParticleFilter::endJob ( void )

overrideprivatevirtual

Reimplemented from edm::EDFilter.

Definition at line 169 of file MCMultiParticleFilter.cc.

References DEFINE_FWK_MODULE, passedEvents_, and totalEvents_.

                                    {
   edm::LogInfo("MCMultiParticleFilter") << "=== Results of MCMultiParticleFilter: passed "
                                         << passedEvents_ << "/" << totalEvents_ << " events" << std::endl;
 }

bool MCMultiParticleFilter::filter	(	edm::Event &	iEvent,
		const edm::EventSetup &	iSetup
	)

overrideprivate

Definition at line 101 of file MCMultiParticleFilter.cc.

References funct::abs(), acceptMore_, decayRadiusMax, decayRadiusMin, decayZMax, decayZMin, etaMax_, edm::Event::getByToken(), edm::HepMCProduct::GetEvent(), mps_fire::i, motherID_, numRequired_, AlCaHLTBitMon_ParallelJobs::p, particleID_, passedEvents_, ptMin_, mathSSE::sqrt(), status_, token_, and totalEvents_.

 {
   edm::Handle<edm::HepMCProduct> evt;
   iEvent.getByToken(token_, evt);
   
   totalEvents_++;
   int nFound = 0;
   
   const HepMC::GenEvent * myGenEvent = evt->GetEvent();
   
   for ( HepMC::GenEvent::particle_const_iterator p = myGenEvent->particles_begin();
     p != myGenEvent->particles_end(); ++p ) {
     
     for (unsigned int i = 0; i < particleID_.size(); ++i) {
       if ((particleID_[i] == 0 || abs(particleID_[i]) == abs((*p)->pdg_id())) &&
       (*p)->momentum().perp() > ptMin_[i] &&
       fabs((*p)->momentum().eta()) < etaMax_[i] &&
       (status_[i] == 0 || (*p)->status() == status_[i])) {
     if (!((*p)->production_vertex()))
       continue;
 
     double decx = (*p)->production_vertex()->position().x();
     double decy = (*p)->production_vertex()->position().y();
     double decrad = sqrt(decx * decx + decy * decy);
     if (decrad < decayRadiusMin[i])
       continue;
     if (decrad > decayRadiusMax[i])
       continue;
 
     double decz = (*p)->production_vertex()->position().z();
     if (decz < decayZMin[i])
       continue;
     if (decz > decayZMax[i])
       continue;
           
     if(motherID_[i] == 0 ){ // do not check for mother ID if not sepcified
       nFound++;
       break; // only match a given particle once!
     }
     else{
       bool hascorrectmother=false;
       for ( HepMC::GenVertex::particles_in_const_iterator mo = (*p)->production_vertex()->particles_in_const_begin(); mo != (*p)->production_vertex()->particles_in_const_end(); ++mo){
         if( (*mo)->pdg_id() == motherID_[i]){
           hascorrectmother = true;
           break;
         }
       }
       if(hascorrectmother){
         nFound++;
         break; // only match a given particle once! 
       }
     }
       }
     } // loop over targets
     
     if (acceptMore_ && nFound == numRequired_) break; // stop looking if we don't mind having more
   } // loop over particles
   
   if (nFound == numRequired_) {
     passedEvents_++;
     return true;
   } else {
     return false;
   }
   
 }