#include <PythiaFilterMultiAncestor.h>

Inheritance diagram for PythiaFilterMultiAncestor:

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

	PythiaFilterMultiAncestor (const edm::ParameterSet &)

	~PythiaFilterMultiAncestor () 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
bool	isAncestor (HepMC::GenParticle *particle, int IDtoMatch)

Private Attributes
double	betaBoost

std::vector< int >	daughterIDs

std::vector< double >	daughterMaxEtas

std::vector< double >	daughterMaxPts

std::vector< double >	daughterMinEtas

std::vector< double >	daughterMinPts

double	maxetacut

double	maxpcut

double	maxphicut

double	maxptcut

double	maxrapcut

double	minetacut

double	minpcut

double	minphicut

double	minptcut

double	minrapcut

std::vector< int >	motherIDs

int	particleID

int	processID

double	rapidity

int	status

edm::EDGetTokenT< edm::HepMCProduct >	token_

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 26 of file PythiaFilterMultiAncestor.h.

Constructor & Destructor Documentation

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

explicit

Definition at line 9 of file PythiaFilterMultiAncestor.cc.

References betaBoost, daughterIDs, daughterMaxEtas, daughterMaxPts, daughterMinEtas, daughterMinPts, and processID.

     : token_(consumes<edm::HepMCProduct>(
           edm::InputTag(iConfig.getUntrackedParameter("moduleLabel", std::string("generator")), "unsmeared"))),
       particleID(iConfig.getUntrackedParameter("ParticleID", 0)),
       minpcut(iConfig.getUntrackedParameter("MinP", 0.)),
       maxpcut(iConfig.getUntrackedParameter("MaxP", 10000.)),
       minptcut(iConfig.getUntrackedParameter("MinPt", 0.)),
       maxptcut(iConfig.getUntrackedParameter("MaxPt", 10000.)),
       minetacut(iConfig.getUntrackedParameter("MinEta", -10.)),
       maxetacut(iConfig.getUntrackedParameter("MaxEta", 10.)),
       minrapcut(iConfig.getUntrackedParameter("MinRapidity", -20.)),
       maxrapcut(iConfig.getUntrackedParameter("MaxRapidity", 20.)),
       minphicut(iConfig.getUntrackedParameter("MinPhi", -3.5)),
       maxphicut(iConfig.getUntrackedParameter("MaxPhi", 3.5)),
       status(iConfig.getUntrackedParameter("Status", 0)),
       motherIDs(iConfig.getUntrackedParameter("MotherIDs", std::vector<int>{0})),
       daughterIDs(iConfig.getUntrackedParameter("DaughterIDs", std::vector<int>{0})),
       daughterMinPts(iConfig.getUntrackedParameter("DaughterMinPts", std::vector<double>{0.})),
       daughterMaxPts(iConfig.getUntrackedParameter("DaughterMaxPts", std::vector<double>{10000.})),
       daughterMinEtas(iConfig.getUntrackedParameter("DaughterMinEtas", std::vector<double>{-10.})),
       daughterMaxEtas(iConfig.getUntrackedParameter("DaughterMaxEtas", std::vector<double>{10.})),
       processID(iConfig.getUntrackedParameter("ProcessID", 0)),
       betaBoost(iConfig.getUntrackedParameter("BetaBoost", 0.)) {
   //now do what ever initialization is needed
 }

PythiaFilterMultiAncestor::~PythiaFilterMultiAncestor ( )

override

Definition at line 35 of file PythiaFilterMultiAncestor.cc.

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

Member Function Documentation

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

override

Definition at line 64 of file PythiaFilterMultiAncestor.cc.

References funct::abs(), cms::dd::accepted(), betaBoost, class-composition::children, daughterIDs, daughterMaxEtas, daughterMaxPts, daughterMinEtas, daughterMinPts, edm::Event::getByToken(), edm::HepMCProduct::GetEvent(), mps_fire::i, isAncestor(), cmsBatch::log, maxetacut, maxpcut, maxphicut, maxptcut, maxrapcut, minetacut, minpcut, minphicut, minptcut, minrapcut, motherIDs, AlCaHLTBitMon_ParallelJobs::p, particleID, processID, rapidity, status, token_, parallelization::uint(), and MCFilterZboostHelper::zboost().

                                                                                     {
   using namespace edm;
   bool accepted = false;
   Handle<HepMCProduct> evt;
   iEvent.getByToken(token_, evt);
 
   const HepMC::GenEvent* myGenEvent = evt->GetEvent();
 
   if (processID == 0 || processID == myGenEvent->signal_process_id()) {
     for (HepMC::GenEvent::particle_const_iterator p = myGenEvent->particles_begin(); p != myGenEvent->particles_end();
          ++p) {
       HepMC::FourVector mom = MCFilterZboostHelper::zboost((*p)->momentum(), betaBoost);
       rapidity = 0.5 * log((mom.e() + mom.pz()) / (mom.e() - mom.pz()));
 
       if (abs((*p)->pdg_id()) == particleID && mom.rho() > minpcut && mom.rho() < maxpcut &&
           (*p)->momentum().perp() > minptcut && (*p)->momentum().perp() < maxptcut && mom.eta() > minetacut &&
           mom.eta() < maxetacut && rapidity > minrapcut && rapidity < maxrapcut && (*p)->momentum().phi() > minphicut &&
           (*p)->momentum().phi() < maxphicut) {
         // find the mother
         for (std::vector<int>::const_iterator motherID = motherIDs.begin(); motherID != motherIDs.end(); ++motherID) {
           // check status if no mother's pdgID is specified
           if (status == 0 && *motherID == 0) {
             accepted = true;
           }
           if (status != 0 && *motherID == 0) {
             if ((*p)->status() == status)
               accepted = true;
           }
 
           // HepMC::GenParticle* mother = (*((*p)->production_vertex()->particles_in_const_begin()));
 
           // check the mother's pdgID
           if (status == 0 && *motherID != 0) {
             // if (abs(mother->pdg_id()) == abs(*motherID)) {
             if (isAncestor(*p, *motherID)) {
               accepted = true;
             }
           }
           if (status != 0 && *motherID != 0) {
             // if ((*p)->status() == status && abs(mother->pdg_id()) == abs(*motherID)){
             if ((*p)->status() == status && isAncestor(*p, *motherID)) {
               accepted = true;
             }
           }
         }
 
         // find the daughters
         if (accepted & (!daughterIDs.empty())) {
           // if you got this far it means that the mother was found
           // now let's check the daughters
           // use a counter, if there's enough daughter  that match the pdg and kinematic
           // criteria accept the event
           uint good_dau = 0;
           for (HepMC::GenVertex::particle_iterator dau = (*p)->end_vertex()->particles_begin(HepMC::children);
                dau != (*p)->end_vertex()->particles_end(HepMC::children);
                ++dau) {
             for (unsigned int i = 0; i < daughterIDs.size(); ++i) {
               // if a daughter has its pdgID among the desired ones, apply kin cuts on it
               // if it survives, add a notch to the counter
               if ((*dau)->pdg_id() == daughterIDs[i]) {
                 if ((*dau)->momentum().perp() < daughterMinPts[i])
                   continue;
                 if ((*dau)->momentum().perp() > daughterMaxPts[i])
                   continue;
                 if ((*dau)->momentum().eta() < daughterMinEtas[i])
                   continue;
                 if ((*dau)->momentum().eta() > daughterMaxEtas[i])
                   continue;
                 ++good_dau;
               }
             }
           }
           if (good_dau < daughterIDs.size())
             accepted = false;
         }
 
         /*
       if (status == 0 && motherID != 0){
     if (abs(((*p)->mother())->pdg_id()) == abs(motherID)) {
     accepted = true;
       }
       }
     if (status != 0 && motherID != 0){
     if ((*p)->status() == status && abs(((*p)->mother())->pdg_id()) == abs(motherID)){
     accepted = true;
       }
       }
          */
       }
       // only need to satisfy the conditions _once_
       if (accepted)
         break;
     }
 
   } else {
     accepted = true;
   }
 
   if (accepted) {
     return true;
   } else {
     return false;
   }
 }

bool PythiaFilterMultiAncestor::isAncestor	(	HepMC::GenParticle *	particle,
		int	IDtoMatch
	)

private

Definition at line 45 of file PythiaFilterMultiAncestor.cc.

References funct::abs().

Referenced by filter().

                                                                                     {
   for (HepMC::GenVertex::particle_iterator ancestor = particle->production_vertex()->particles_begin(HepMC::ancestors);
        ancestor != particle->production_vertex()->particles_end(HepMC::ancestors);
        ++ancestor) {
     // std::cout << __LINE__ << "]\t particle's PDG ID " << particle->pdg_id()
     //                       << " \t particle's ancestor's PDG ID " << (*ancestor)->pdg_id()
     //                       << " \t ID to match " << IDtoMatch << std::endl;
 
     if (abs((*ancestor)->pdg_id()) == abs(IDtoMatch)) {
       //  std::cout << __LINE__ << "]\t found!" << std::endl;
       return true;
     }
   }
 
   // std::cout << __LINE__ << "]\t nope, no luck" << std::endl;
   return false;
 }