#include <PythiaDauVFilterMatchID.h>

Inheritance diagram for PythiaDauVFilterMatchID:

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

	PythiaDauVFilterMatchID (const edm::ParameterSet &)

	~PythiaDauVFilterMatchID () override

Public Member Functions inherited from edm::global::EDFilter<>
	EDFilter ()=default

bool	hasAbilityToProduceInLumis () const final

bool	hasAbilityToProduceInRuns () const final

bool	wantsGlobalLuminosityBlocks () const final

bool	wantsGlobalRuns () const final

bool	wantsStreamLuminosityBlocks () const final

bool	wantsStreamRuns () const final

Public Member Functions inherited from edm::global::EDFilterBase
	EDFilterBase ()

ModuleDescription const &	moduleDescription () const

	~EDFilterBase () 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 Attributes
const bool	chargeconju

std::vector< int >	dauIDs

std::unique_ptr< Pythia8::Pythia >	fLookupGen

const int	fVerbose

std::vector< double >	maxetacut

const double	maxptcut

std::vector< double >	minetacut

std::vector< double >	minptcut

const int	motherID

const int	ndaughters

const int	particleID

const edm::EDGetTokenT< edm::HepMCProduct >	token_

Additional Inherited Members
Public Types inherited from edm::global::EDFilterBase
typedef EDFilterBase	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::EDFilterBase
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 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

Description: Filter events using MotherId and ChildrenIds infos

Implementation: <Notes on="" implementation>="">

Definition at line 50 of file PythiaDauVFilterMatchID.h.

Constructor & Destructor Documentation

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

explicit

Definition at line 11 of file PythiaDauVFilterMatchID.cc.

References dauIDs, fLookupGen, edm::ParameterSet::getUntrackedParameter(), mps_fire::i, maxetacut, maxptcut, minetacut, minptcut, motherID, and particleID.

                                                                                :
   fVerbose(iConfig.getUntrackedParameter("verbose",0)),
   token_(consumes<edm::HepMCProduct>(edm::InputTag(iConfig.getUntrackedParameter("moduleLabel",std::string("generator")),"unsmeared"))),
   particleID(iConfig.getUntrackedParameter("ParticleID", 0)),
   motherID(iConfig.getUntrackedParameter("MotherID", 0)),
   chargeconju(iConfig.getUntrackedParameter("ChargeConjugation", true)),
   ndaughters(iConfig.getUntrackedParameter("NumberDaughters", 0)),
   maxptcut(iConfig.getUntrackedParameter("MaxPt", 14000.))
 {
   //now do what ever initialization is needed
   vector<int> defdauID;
   defdauID.push_back(0);
   dauIDs = iConfig.getUntrackedParameter< vector<int> >("DaughterIDs",defdauID);
   vector<double> defminptcut;
   defminptcut.push_back(0.);
   minptcut = iConfig.getUntrackedParameter< vector<double> >("MinPt",defminptcut);
   vector<double> defminetacut;
   defminetacut.push_back(-10.);
   minetacut = iConfig.getUntrackedParameter< vector<double> >("MinEta",defminetacut);
   vector<double> defmaxetacut;
   defmaxetacut.push_back(10.);
   maxetacut = iConfig.getUntrackedParameter< vector<double> >("MaxEta",defmaxetacut);
 
   edm::LogInfo("PythiaDauVFilterMatchID") << "----------------------------------------------------------------------" << endl;
   edm::LogInfo("PythiaDauVFilterMatchID") << "--- PythiaDauVFilterMatchID" << endl;
   for (unsigned int i=0; i<dauIDs.size(); ++i) {
       edm::LogInfo("PythiaDauVFilterMatchID") << "ID: " <<  dauIDs[i] << " pT > " << minptcut[i] << " " << minetacut[i] << " eta < " << maxetacut[i] << endl;
   }
   edm::LogInfo("PythiaDauVFilterMatchID") << "maxptcut   = " << maxptcut << endl;
   edm::LogInfo("PythiaDauVFilterMatchID") << "particleID = " << particleID << endl;
   edm::LogInfo("PythiaDauVFilterMatchID") << "motherID   = " << motherID << endl;
 
  // create pythia8 instance to access particle data
   edm::LogInfo("PythiaDauVFilterMatchID") << "Creating pythia8 instance for particle properties" << endl;
   if(!fLookupGen.get()) fLookupGen.reset(new Pythia());
 }

PythiaDauVFilterMatchID::~PythiaDauVFilterMatchID ( )

override

Definition at line 49 of file PythiaDauVFilterMatchID.cc.

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

Member Function Documentation

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

override

Definition at line 63 of file PythiaDauVFilterMatchID.cc.

References funct::abs(), Abs(), cms::dd::accepted(), chargeconju, class-composition::children, dauIDs, fLookupGen, fVerbose, edm::Event::getByToken(), edm::HepMCProduct::GetEvent(), mps_fire::i, decayTarget::maxEta, maxetacut, decayTarget::maxPt, maxptcut, decayTarget::minEta, minetacut, decayTarget::minPt, minptcut, motherID, ndaughters, AlCaHLTBitMon_ParallelJobs::p, particleID, decayTarget::pdgID, edmPickEvents::target, and token_.

                                                                                                     {
   using namespace edm;
   bool accepted = false;
   Handle<HepMCProduct> evt;
   iEvent.getByToken(token_, evt);
 
   int OK(1); 
   vector<int> vparticles; 
   
   HepMC::GenEvent *myGenEvent = new HepMC::GenEvent(*(evt->GetEvent()));
   
   if (fVerbose > 5) {
     edm::LogInfo("PythiaDauVFilterMatchID") << "looking for " << particleID << endl;
   }
     
   for (HepMC::GenEvent::particle_iterator p = myGenEvent->particles_begin(); p != myGenEvent->particles_end(); ++p) {
     
     if ((*p)->pdg_id() != particleID) continue ;
 
     // -- Check for mother of this particle
     if (0 != motherID) {
       OK = 0; 
       for (HepMC::GenVertex::particles_in_const_iterator des = (*p)->production_vertex()->particles_in_const_begin(); 
        des != (*p)->production_vertex()->particles_in_const_end();
        ++des) {
     if (fVerbose > 10) {
         edm::LogInfo("PythiaDauVFilterMatchID") << "mother: " << (*des)->pdg_id() << " pT: " << (*des)->momentum().perp() << " eta: " << (*des)->momentum().eta() << endl;
     }
     if (abs(motherID) == abs((*des)->pdg_id())) {
       OK = 1; 
       break;
     }
       }
     }
     if (0 == OK) continue; 
 
     //generate targets
     std::vector<decayTarget> targets;
     for (unsigned int i=0;i<dauIDs.size();i++) 
       {
     decayTarget target;
     target.pdgID = dauIDs[i];
     target.minPt = minptcut[i];
     target.maxPt = maxptcut;
     target.minEta = minetacut[i];
     target.maxEta = maxetacut[i];
     targets.push_back(target);
       }
     if (fVerbose>10) 
       {
     edm::LogInfo("PythiaDauVFilterMatchID") << "created target: ";
     for (unsigned int i=0;i<targets.size();i++) {edm::LogInfo("PythiaDauVFilterMatchID") << targets[i].pdgID << " ";}
     edm::LogInfo("PythiaDauVFilterMatchID") << endl;
       }
 
     // -- check for daugthers
     int ndau = 0;     
     if (fVerbose > 5) {
         edm::LogInfo("PythiaDauVFilterMatchID") << "found ID: " << (*p)->pdg_id() << " pT: " << (*p)->momentum().perp() << " eta: " << (*p)->momentum().eta() << endl;
     }
     vparticles.push_back((*p)->pdg_id()); 
     if ((*p)->end_vertex()) {   
       for (HepMC::GenVertex::particle_iterator des=(*p)->end_vertex()->particles_begin(HepMC::children);
        des != (*p)->end_vertex()->particles_end(HepMC::children);
        ++des) {
     if ( TMath::Abs((*des)->pdg_id()) == 22 ) {continue;}
     ++ndau;       
     if (fVerbose > 5) {
         edm::LogInfo("PythiaDauVFilterMatchID") << "ID: " << (*des)->pdg_id() << " pT: " << (*des)->momentum().perp() << " eta: " << (*des)->momentum().eta() << endl;
     }
     { // protect matchedIdx
     int matchedIdx=-1;
     for (unsigned int i=0;i<targets.size();i++)
       {
         if ( (*des)->pdg_id() != targets[i].pdgID ) {continue;}
         if (fVerbose>5) {edm::LogInfo("PythiaDauVFilterMatchID") << "i = " << i << " pT = " << (*des)->momentum().perp() << " eta = " << (*des)->momentum().eta() << endl;}
 
         if ((*des)->momentum().perp() >  targets[i].minPt  && 
         (*des)->momentum().perp() <  targets[i].maxPt  && 
         (*des)->momentum().eta()  >  targets[i].minEta && 
         (*des)->momentum().eta()  <  targets[i].maxEta ) 
           {
         vparticles.push_back((*des)->pdg_id());
         if (fVerbose > 2) {
           edm::LogInfo("PythiaDauVFilterMatchID") << "  accepted this particle " <<  (*des)->pdg_id()
                << " pT = " << (*des)->momentum().perp() << " eta = " << (*des)->momentum().eta() << endl;
         }
         matchedIdx=i;
         break;
           }
       }
     if (matchedIdx>-1) {targets.erase(targets.begin()+matchedIdx);}
     if (fVerbose>10) 
       {
         edm::LogInfo("PythiaDauVFilterMatchID") << "remaining targets: ";
         for (unsigned int i=0;i<targets.size();i++) {edm::LogInfo("PythiaDauVFilterMatchID") << targets[i].pdgID << " ";}
         edm::LogInfo("PythiaDauVFilterMatchID") << endl;
       }
     }                
       }
     }  
 
     if (ndau == ndaughters && targets.size() == 0 ) 
       {
     accepted=true;
     if (fVerbose > 0) {
       edm::LogInfo("PythiaDauVFilterMatchID") << "  accepted this decay: ";
       for (unsigned int iv = 0; iv < vparticles.size(); ++iv) edm::LogInfo("PythiaDauVFilterMatchID") << vparticles[iv] << " "; 
       edm::LogInfo("PythiaDauVFilterMatchID") << " from mother = " << motherID << endl;
     }
     break;
       }
     
   }
   
 
   int anti_particleID =  -particleID;
   if ( !(fLookupGen->particleData.isParticle( anti_particleID )) ) {
      anti_particleID = 0;
      if (fVerbose > 5) edm::LogInfo("PythiaDauVFilterMatchID") << "Particle " << particleID << " is its own anti-particle, skipping further testing " << endl;
   }
   if (!accepted && chargeconju && anti_particleID) {
     OK = 1; 
 
     for (HepMC::GenEvent::particle_iterator p = myGenEvent->particles_begin();
      p != myGenEvent->particles_end(); ++p) {
  
       if ((*p)->pdg_id() != anti_particleID) continue ;
       
       // -- Check for mother of this particle
       if (0 != motherID) {
     OK = 0; 
     for (HepMC::GenVertex::particles_in_const_iterator des = (*p)->production_vertex()->particles_in_const_begin(); 
          des != (*p)->production_vertex()->particles_in_const_end();
          ++des) {
       if (fVerbose > 10) {
           edm::LogInfo("PythiaDauVFilterMatchID") << "mother: " << (*des)->pdg_id() << " pT: " << (*des)->momentum().perp() << " eta: " << (*des)->momentum().eta() << endl;
       }
       if (abs(motherID) == abs((*des)->pdg_id())) {
         OK = 1; 
         break;
       }
     }
       }
       if (0 == OK) continue; 
 
     //generate anti targets
     std::vector<decayTarget> targets;
     for (unsigned int i=0;i<dauIDs.size();i++) 
       {
     decayTarget target;
         int IDanti =  -dauIDs[i];
         if ( !(fLookupGen->particleData.isParticle( IDanti )) ) IDanti = dauIDs[i];
         target.pdgID = IDanti;
     target.minPt = minptcut[i];
     target.maxPt = maxptcut;
     target.minEta = minetacut[i];
     target.maxEta = maxetacut[i];
     targets.push_back(target);
       }
     if (fVerbose>10) 
       {
     edm::LogInfo("PythiaDauVFilterMatchID") << "created anti target: ";
     for (unsigned int i=0;i<targets.size();i++) {edm::LogInfo("PythiaDauVFilterMatchID") << targets[i].pdgID << " ";}
     edm::LogInfo("PythiaDauVFilterMatchID") << endl;
       }
       
       if (fVerbose > 5) {
           edm::LogInfo("PythiaDauVFilterMatchID") << "found ID: " << (*p)->pdg_id() << " pT: " << (*p)->momentum().perp() << " eta: " << (*p)->momentum().eta() << endl;
       }
       vparticles.push_back((*p)->pdg_id()); 
       int ndau = 0;     
       if ((*p)->end_vertex()) {
     for (HepMC::GenVertex::particle_iterator des=(*p)->end_vertex()->particles_begin(HepMC::children);
          des != (*p)->end_vertex()->particles_end(HepMC::children);
          ++des) {
       if ( TMath::Abs((*des)->pdg_id()) == 22 ) {continue;}
       ++ndau;
       if (fVerbose > 5) {
           edm::LogInfo("PythiaDauVFilterMatchID") << "ID: " << (*des)->pdg_id() << " pT: " << (*des)->momentum().perp() << " eta: " << (*des)->momentum().eta() << endl;
       }
       { // protect matchedIdx
         int matchedIdx=-1;
         for (unsigned int i=0;i<targets.size();i++)
           {
         if ( (*des)->pdg_id() != targets[i].pdgID ) {continue;}
         if (fVerbose>5) {edm::LogInfo("PythiaDauVFilterMatchID") << "i = " << i << " pT = " << (*des)->momentum().perp() << " eta = " << (*des)->momentum().eta() << endl;}
 
         if ((*des)->momentum().perp() >  targets[i].minPt  && 
             (*des)->momentum().perp() <  targets[i].maxPt  && 
             (*des)->momentum().eta()  >  targets[i].minEta && 
             (*des)->momentum().eta()  <  targets[i].maxEta ) 
           {
             vparticles.push_back((*des)->pdg_id());
             if (fVerbose > 2) {
               edm::LogInfo("PythiaDauVFilterMatchID") << "  accepted this particle " <<  (*des)->pdg_id()
                << " pT = " << (*des)->momentum().perp() << " eta = " << (*des)->momentum().eta() << endl;
             }
             matchedIdx=i;
             break;
           }
           }
         if (matchedIdx>-1) {targets.erase(targets.begin()+matchedIdx);}
         if (fVerbose>10) 
           {
         edm::LogInfo("PythiaDauVFilterMatchID") << "remaining targets: ";
         for (unsigned int i=0;i<targets.size();i++) {edm::LogInfo("PythiaDauVFilterMatchID") << targets[i].pdgID << " ";}
         edm::LogInfo("PythiaDauVFilterMatchID") << endl;
           }
       }              
     }
       }
       if (ndau == ndaughters && targets.size() == 0 ) 
     {
       accepted=true;
       if (fVerbose > 0) {
         edm::LogInfo("PythiaDauVFilterMatchID") << "  accepted this decay: ";
         for (unsigned int iv = 0; iv < vparticles.size(); ++iv) edm::LogInfo("PythiaDauVFilterMatchID") << vparticles[iv] << " "; 
         edm::LogInfo("PythiaDauVFilterMatchID") << " from mother = " << motherID << endl;
       }
       break;
     }  
     }
     
   }    
   
   delete myGenEvent; 
   return accepted;
 
 }