df/db2/AnaTutorialECCE_8h_source.html

#ifndef ANATUTORIALECCE_H__

#define ANATUTORIALECCE_H__


#include <fun4all/SubsysReco.h>


class Fun4AllHistoManager;

class PHCompositeNode;

class TFile;

class TTree;

class TH1;

class TH2;

class PHCompositeNode;

class RawClusterContainer;

class RawCluster;

class SvtxTrackMap;

class JetMap;

class GlobalVertex;

class PHHepMCGenEventMap;

class JetEvalStack;

class JetRecoEval;

class PHG4TruthInfoContainer;

class PHHepMCGenEvent;

class CaloTriggerInfo;

class JetTruthEval;

class JetEvalStack;


class AnaTutorialECCE : public SubsysReco

{

 public:

  AnaTutorialECCE(const std::string &name = "AnaTutorialECCE",

              const std::string &fname = "AnaTutorialECCE.root");


  // Destructor

  virtual ~AnaTutorialECCE();


  int Init(PHCompositeNode *);


  int process_event(PHCompositeNode *);


  int End(PHCompositeNode *);


  void setMinJetPt(float minjetpt) { m_minjetpt = minjetpt; }


  void setMinClusPt(float mincluspt) { m_mincluspt = mincluspt; }


  void analyzeTracks(bool analyzeTracks) { m_analyzeTracks = analyzeTracks; }

  void analyzeClusters(bool analyzeClusters) { m_analyzeClusters = analyzeClusters; }

  void analyzeJets(bool analyzeJets) { m_analyzeJets = analyzeJets; }

  void analyzeTruth(bool analyzeTruth) { m_analyzeTruth = analyzeTruth; }


 private:

  std::string m_outfilename;


  Fun4AllHistoManager *m_hm;


  float m_minjetpt;


  float m_mincluspt;


  bool m_analyzeTracks;


  bool m_analyzeClusters;


  bool m_analyzeJets;


  bool m_analyzeTruth;


  TFile *m_outfile;

  TTree *m_clustertree;

  TTree *m_tracktree;

  TTree *m_hepmctree;

  TTree *m_truthtree;

  TTree *m_recojettree;

  TTree *m_truthjettree;

  TH1 *m_phi_h;

  TH2 *m_eta_phi_h;


  JetEvalStack *m_jetEvalStack = nullptr;


  void getTracks(PHCompositeNode *topNode);

  void getTruthJets(PHCompositeNode *topNode);

  void getReconstructedJets(PHCompositeNode *topNode);

  void getEMCalClusters(PHCompositeNode *topNode);

  void getHEPMCTruth(PHCompositeNode *topNode);

  void getPHG4Truth(PHCompositeNode *topNode);


  void initializeVariables();

  void initializeTrees();


  int m_partid1;

  int m_partid2;

  double m_x1;

  double m_x2;

  int m_mpi;

  int m_process_id;

  double m_truthenergy;

  double m_trutheta;

  double m_truthphi;

  double m_truthpx;

  double m_truthpy;

  double m_truthpz;

  double m_truthpt;

  double m_truthp;

  int m_numparticlesinevent;

  int m_truthpid;


  double m_tr_px;

  double m_tr_py;

  double m_tr_pz;

  double m_tr_p;

  double m_tr_pt;

  double m_tr_phi;

  double m_tr_eta;

  int m_charge;

  double m_chisq;

  int m_ndf;

  double m_dca;

  double m_tr_x;

  double m_tr_y;

  double m_tr_z;

  float m_tr_pion_loglikelihood;

  float m_tr_kaon_loglikelihood;

  float m_tr_proton_loglikelihood;

  int m_truth_is_primary;

  double m_truthtrackpx;

  double m_truthtrackpy;

  double m_truthtrackpz;

  double m_truthtrackp;

  double m_truthtracke;

  double m_truthtrackpt;

  double m_truthtrackphi;

  double m_truthtracketa;

  int m_truthtrackpid;


  double m_recojetpt;

  int m_recojetid;

  double m_recojetpx;

  double m_recojetpy;

  double m_recojetpz;

  double m_recojetphi;

  double m_recojetp;

  double m_recojetenergy;

  double m_recojeteta;

  int m_truthjetid;

  double m_truthjetp;

  double m_truthjetphi;

  double m_truthjeteta;

  double m_truthjetpt;

  double m_truthjetenergy;

  double m_truthjetpx;

  double m_truthjetpy;

  double m_truthjetpz;

  double m_dR;


  double m_clusenergy;

  double m_cluseta;

  double m_clustheta;

  double m_cluspt;

  double m_clusphi;

  double m_cluspx;

  double m_cluspy;

  double m_cluspz;

  double m_E_4x4;

};


#endif