TestHepMC.h

Go to the documentation of this file.

/*
   Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef XAOD_ANALYSIS
 
#ifndef EVGENPRODTOOLS_TESTHEPMC_H
#define EVGENPRODTOOLS_TESTHEPMC_H
 
#include "GeneratorModules/GenBase.h"
#include "GeneratorModules/GenData.h"
#include "GaudiKernel/ITHistSvc.h"
#include "TFile.h"
#include "TH1.h"
#include "AtlasHepMC/GenEvent.h"
#include "AtlasHepMC/Relatives.h"
#include "TruthUtils/Loops.h"
#include <cmath>
 
#include <fstream>
 

class TestHepMC : public GenBase {
public:
 
  TestHepMC(const std::string& name, ISvcLocator* pSvcLocator);
  StatusCode initialize();
  StatusCode execute(const EventContext& ctx);
  StatusCode finalize();
 
 private:
 
  int         m_maxloops;
  int         m_pdg;
  double      m_cm_energy, m_cme_diff;
  double      m_energy_diff, m_max_energy_diff;
  bool        m_dumpEvent, m_allowMissingXSec;
  double      m_min_dist_trans, m_max_dist_trans, m_max_dist, m_min_tau, m_nonG4_energy_threshold;
  double      m_eff_warn_threshold, m_eff_fail_threshold, m_tau_eff_threshold;
  double      m_accur_margin;
  bool m_doHist;
  bool m_beamEnergyTest, m_vtxNaNTest, m_vtxDisplacedTest, m_momNaNTest, m_lifeTimeTest, m_energyG4Test;
  bool m_energyImbalanceTest, m_momImbalanceTest, m_negativeEnergyTest, m_tachyonsTest, m_unstableNoVtxTest;
  bool m_pi0NoVtxTest, m_undisplacedDaughtersTest, m_unknownPDGIDTest;
 
  std::vector<int> m_vertexStatuses;
 
  int m_nPass;
  int m_nFail;
  int m_noXSECset;
 
  int m_TotalTaus;
  int m_FastDecayedTau;
 
  int m_invalidBeamParticlesCheckRate;
  int m_beamParticleswithStatusNotFourCheckRate;
  int m_beamEnergyCheckRate;
  int m_vtxNANandINFCheckRate;
  int m_vtxDisplacedstatuscode12CheckRate;
  int m_vtxDisplacedstatuscodenot12CheckRate;
  int m_vtxDisplacedMoreThan_1m_CheckRate;
  int m_partMomentumNANandINFCheckRate;
  int m_undecayedPi0statuscode12CheckRate;
  int m_unstableNoEndVtxCheckRate;
  int m_negativeEnergyTachyonicCheckRate;
  int m_decayCheckRate;
  int m_undisplacedLLHdaughtersCheckRate;
  int m_nonZeroPhotonMassCheckRate;
  int m_energyBalanceCheckRate;
  int m_momentumBalanceCheckRate;
  int m_negativeEnergyCheckRate;
  int m_tachyonCheckRate;
  int m_stableUnstableNoParentCheckRate;
  int m_unstablePartNoDecayVtxCheckRate;
  int m_undecayedPi0CheckRate;
  int m_Status1ShortLifetime;
  int m_undisplacedDecayDaughtersOfDisplacedVtxCheckRate;
  int m_nonG4_energyCheckRate;
  int m_unknownPDGIDCheckRate;
 
  std::string m_paramFile;
  std::string m_unknownPDGIDFile;
 
  std::vector<int> m_G4pdgID_tab;
  std::vector<int> m_SusyPdgID_tab;
  std::vector<int> m_uknownPDGID_tab;

  ServiceHandle<ITHistSvc> m_thistSvc;
 
  TH1F* m_h_energy_dispVtxCheck;
  TH1F* m_h_energy_dispVtxCheck_lt10;
  TH1F* m_h_pdgid_dispVtxCheck;
  TH1F* m_h_status_dispVtxCheck;
  TH1F* m_h_px_dispVtxCheck;
  TH1F* m_h_py_dispVtxCheck;
  TH1F* m_h_pz_dispVtxCheck;
  TH1F* m_h_vx_dispVtxCheck;
  TH1F* m_h_vy_dispVtxCheck;
  TH1F* m_h_vz_dispVtxCheck;
  TH1F* m_h_vxprod_dispVtxCheck;
  TH1F* m_h_vyprod_dispVtxCheck;
  TH1F* m_h_vzprod_dispVtxCheck;
  TH1F* m_h_vtxend_dispVtxCheck;
  TH1F* m_h_vtxprod_dispVtxCheck;
 
  TH1F* m_h_photon_mass;
  TH1F* m_h_photon_energy;
  TH1F* m_h_photon_e2_p2_e2;
 
  TH1F* m_h_energyImbalance;
  TH1F* m_h_momentumImbalance_px;
  TH1F* m_h_momentumImbalance_py;
  TH1F* m_h_momentumImbalance_pz;
 
  TH1F* m_h_beamparticle1_Energy;
  TH1F* m_h_beamparticle2_Energy;
  TH1F* m_h_cmEnergyDiff;
 
  MC::Loops<HepMC::GenEvent,HepMC::ConstGenParticlePtr,HepMC::ConstGenVertexPtr> m_looper; 
 
  std::shared_ptr<GenData> m_gendata{nullptr};
 
};
 
 
#endif
 
#endif