PythiaMomDauFilter.cc

Go to the documentation of this file.

#include "GeneratorInterface/GenFilters/interface/PythiaMomDauFilter.h"


#include "SimDataFormats/GeneratorProducts/interface/HepMCProduct.h"
#include "HepMC/PythiaWrapper6_4.h"
#include <iostream>

using namespace edm;
using namespace std;


PythiaMomDauFilter::PythiaMomDauFilter(const edm::ParameterSet& iConfig) :
label_(iConfig.getUntrackedParameter("moduleLabel",std::string("generator"))),
particleID(iConfig.getUntrackedParameter("ParticleID", 0)),
daughterID(iConfig.getUntrackedParameter("DaughterID", 0)),
chargeconju(iConfig.getUntrackedParameter("ChargeConjugation", true)),
ndaughters(iConfig.getUntrackedParameter("NumberDaughters", 0)),
ndescendants(iConfig.getUntrackedParameter("NumberDescendants", 0)),
minptcut(iConfig.getUntrackedParameter("MinPt", 0.)),
maxptcut(iConfig.getUntrackedParameter("MaxPt", 14000.)),
minetacut(iConfig.getUntrackedParameter("MinEta", -10.)),
maxetacut(iConfig.getUntrackedParameter("MaxEta", 10.)),
mom_minptcut(iConfig.getUntrackedParameter("MomMinPt", 0.)),
mom_maxptcut(iConfig.getUntrackedParameter("MomMaxPt", 14000.)),
mom_minetacut(iConfig.getUntrackedParameter("MomMinEta", -10.)),
mom_maxetacut(iConfig.getUntrackedParameter("MomMaxEta", 10.))
{
   //now do what ever initialization is needed
   vector<int> defdauID;
   defdauID.push_back(0);
   vector<int> defdesID;
   defdesID.push_back(0);
   dauIDs = iConfig.getUntrackedParameter< vector<int> >("DaughterIDs",    defdauID);
   desIDs = iConfig.getUntrackedParameter< vector<int> >("DescendantsIDs", defdesID);
}


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

}


//
// member functions
//

// ------------ method called to produce the data  ------------
bool PythiaMomDauFilter::filter(edm::Event& iEvent, const edm::EventSetup& iSetup)
{
 using namespace edm;
 bool accepted = false;
 bool mom_accepted = false;
 Handle<HepMCProduct> evt;
 iEvent.getByLabel(label_, evt);

 const HepMC::GenEvent * myGenEvent = evt->GetEvent();
 int ndauac = 0;
 int ndau = 0;
 int ndesac = 0;
 int ndes   = 0;   
   
 for ( HepMC::GenEvent::particle_const_iterator p = myGenEvent->particles_begin();  p != myGenEvent->particles_end(); ++p ) {
     
  if( (*p)->pdg_id() != particleID ) continue ;
  if( (*p)->momentum().perp() >  mom_minptcut  && (*p)->momentum().perp() <  mom_maxptcut  && (*p)->momentum().eta()  >  mom_minetacut &&  (*p)->momentum().eta()  <  mom_maxetacut ){ 
   mom_accepted = true;
   ndauac = 0;
   ndau   = 0;
   ndesac = 0;
   ndes   = 0;     
   if ( (*p)->end_vertex() ) {  
    for ( HepMC::GenVertex::particle_iterator dau  =(*p)->end_vertex()->particles_begin(HepMC::children); dau != (*p)->end_vertex()->particles_end(HepMC::children); ++dau ) {
      ++ndau;
      for( unsigned int i=0; i<dauIDs.size(); ++i) {
       if( (*dau)->pdg_id() != dauIDs[i] ) continue ;
        ++ndauac;
        break;
     }
      if((*dau)->pdg_id() == daughterID){
       for(HepMC::GenVertex::particle_iterator des  = (*dau)->end_vertex()->particles_begin(HepMC::children); des != (*des)->end_vertex()->particles_end(HepMC::children); ++des ){
        ++ndes;        
        for( unsigned int i=0; i<desIDs.size(); ++i) {
         if( (*des)->pdg_id() != desIDs[i] ) continue ;
         if( (*des)->momentum().perp() >  minptcut  && (*des)->momentum().perp() <  maxptcut  && (*des)->momentum().eta()  >  minetacut &&  (*des)->momentum().eta()  <  maxetacut ) {
          ++ndesac;
          break;
      } 
     }
    }                    
       }
      } 
     }
    }  
    if( ndau ==  ndaughters && ndauac == ndaughters && mom_accepted && ndes == ndescendants && ndesac == ndescendants ) {
       accepted = true;
       break;
     }    
     
   }
   
   
   if( !accepted && chargeconju ) {
   
     mom_accepted = false;
     for ( HepMC::GenEvent::particle_const_iterator p = myGenEvent->particles_begin(); p != myGenEvent->particles_end(); ++p ) {
       
       if( (*p)->pdg_id() != -particleID ) continue ;
       if( (*p)->momentum().perp() >  mom_minptcut  && (*p)->momentum().perp() <  mom_maxptcut  &&  (*p)->momentum().eta()  >  mom_minetacut && (*p)->momentum().eta()  <  mom_maxetacut ){ 
       mom_accepted = true;
       ndauac = 0;
       ndau = 0;
       ndesac = 0;
       ndes   = 0;      
       if ( (*p)->end_vertex() ) {
     for ( HepMC::GenVertex::particle_iterator dau  =(*p)->end_vertex()->particles_begin(HepMC::children); dau != (*p)->end_vertex()->particles_end(HepMC::children);  ++dau ) {
       ++ndau;
       for( unsigned int i=0; i<dauIDs.size(); ++i) {
            int IDanti = -dauIDs[i];
            int pythiaCode = PYCOMP(dauIDs[i]);
            int has_antipart = pydat2.kchg[3-1][pythiaCode-1];
            if( has_antipart == 0 ) IDanti = dauIDs[i];
            if( (*dau)->pdg_id() != IDanti ) continue ;
            ++ndauac;
            break;
           }
       if((*dau)->pdg_id() == -daughterID) {
            for( HepMC::GenVertex::particle_iterator des  = (*dau)->end_vertex()->particles_begin(HepMC::children); des != (*des)->end_vertex()->particles_end(HepMC::children);  ++des ){
                 ++ndes;
        for( unsigned int i=0; i<desIDs.size(); ++i) {
         int IDanti = -desIDs[i];
         int pythiaCode = PYCOMP(desIDs[i]);
         int has_antipart = pydat2.kchg[3-1][pythiaCode-1];
         if( has_antipart == 0 ) IDanti = desIDs[i];
         if( (*des)->pdg_id() != IDanti ) continue ;
         if(   (*des)->momentum().perp() >  minptcut  && (*des)->momentum().perp() <  maxptcut  &&  (*des)->momentum().eta()  >  minetacut && (*des)->momentum().eta()  <  maxetacut ) {
           ++ndesac;
           break;
         } 
       }                     
          }  
     }
    } 
       }
      }
       if( ndau ==  ndaughters && ndauac == ndaughters && mom_accepted && ndes == ndescendants && ndesac == ndescendants ) {
     accepted = true;
     break;
       }    
     }
     
   }    

   if (accepted){
   return true; } else {return false;}

}