PHPy6ForwardElectronTrig.C

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file PHPy6ForwardElectronTrig.C

#include "PHPy6GenTrigger.h"
#include "PHPy6ForwardElectronTrig.h"
#include <phool/PHCompositeNode.h>
#include <phool/phool.h>
#include <phool/getClass.h>

#include <phhepmc/PHHepMCGenEvent.h>
#include <HepMC/GenEvent.h>

#include <cstdlib>
#include <iostream>
using namespace std;

//___________________________________________________________________________
PHPy6ForwardElectronTrig::PHPy6ForwardElectronTrig(const std::string &name): PHPy6GenTrigger(name) 
{
  ntriggered_forward_electron = 0;
  nconsidered_forward_electron = 0;
  
  n_em_required = 1; 
  n_ep_required = 1; 
  n_comb_required = 1; 
  pt_required = 0.5; 
  eta_low = 1.0; 
  eta_high = 5.0; 

  RequireElectron = false;
  RequirePositron = false; 
  RequireOR = false; 
  RequireAND = false; 
  RequireCOMBO = true; 

}

void PHPy6ForwardElectronTrig::PrintConfig()
{
  cout << endl; 
  cout << "PHPy6ForwardElectronTrig Configuration: " << endl; 
  cout << " >=" << n_ep_required << " e+ required" << endl; 
  cout << " >=" << n_em_required << " e- required" << endl; 
  cout << " >=" << n_comb_required << " combined required" << endl; 
  cout << " Electron transverse momentum > " << pt_required << " GeV required" << endl; 
  cout << " " << eta_low << " < eta < " << eta_high << endl; 

  if(RequireElectron) cout << " RequireElectron is set" << endl;
  if(RequirePositron) cout << " RequirePositron is set" << endl;
  if(RequireOR) cout << " RequireOR is set" << endl;
  if(RequireAND) cout << " RequireAND is set" << endl;
  if(RequireCOMBO) cout << " RequireCOMBINED is set" << endl;

  cout << endl; 
}
        
bool PHPy6ForwardElectronTrig::Apply( const HepMC::GenEvent* evt )
{

  // increment counter
  ++nconsidered_forward_electron;
        
  // Print Out Trigger Information Once, for Posterity
  static int trig_info_printed = 0;
  if ( trig_info_printed==0 )
    {
      PrintConfig(); 
      trig_info_printed = 1;
    }

  // Check the HepMC particle list - 
        
  unsigned int n_em_found = 0; 
  unsigned int n_ep_found = 0; 

  for ( HepMC::GenEvent::particle_const_iterator p 
          = evt->particles_begin(); p != evt->particles_end(); ++p ){
    if ( (abs((*p)->pdg_id()) == 11) && ((*p)->status()==1) && 
         ((*p)->momentum().pseudoRapidity() > eta_low) && ((*p)->momentum().pseudoRapidity() < eta_high) && 
         (sqrt(pow((*p)->momentum().px(),2) + pow((*p)->momentum().py(),2))>pt_required) ) {
      if(((*p)->pdg_id()) == 11) n_em_found++;
      if(((*p)->pdg_id()) == -11) n_ep_found++;
    }
  }

  if( (RequireOR && ((n_em_found>=n_em_required)||(n_ep_found>=n_ep_required)) ) ||
      (RequireElectron && (n_em_found>=n_em_required)) ||
      (RequirePositron && (n_ep_found>=n_ep_required)) ||
      (RequireAND && (n_em_found>=n_em_required) && (n_ep_found>=n_ep_required)) ||
      (RequireCOMBO && (n_em_found+n_ep_found)>=n_comb_required) ) 
  {
    ++ntriggered_forward_electron;
    return true; 
  }

  return false;
  
}