PHPy6ForwardElectronTrig.cc

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

#include "PHPy6ForwardElectronTrig.h"
#include "PHPy6GenTrigger.h"

#include <HepMC/GenEvent.h>
#include <HepMC/GenParticle.h>     // for GenParticle
#include <HepMC/SimpleVector.h>    // for FourVector

#include <cmath>                  // for sqrt
#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;
}