Analysis Software
Documentation for sPHENIX simulation software
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
ExclusiveReco.h
Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file ExclusiveReco.h
1 #ifndef __ExclusiveReco_H__
2 #define __ExclusiveReco_H__
3 
4 #include "TrackProjectorPlaneECAL.h"
5 /* Fun4All includes */
6 #include <fun4all/SubsysReco.h>
7 #include <calobase/RawTowerDefs.h>
8 
9 /* STL includes */
10 #include <math.h>
11 #include <map>
12 
13 /*HepMC include */
14 #include <phhepmc/PHHepMCGenEvent.h>
15 #include <phhepmc/PHHepMCGenEventMap.h>
16 #include <HepMC/GenEvent.h>
17 class TTree;
18 class TFile;
19 
20 class PHCompositeNode;
21 class CaloEvalStack;
22 class CaloRawTowerEval;
23 class RawCluster;
24 class RawTowerContainer;
25 class RawTowerGeomContainer;
26 //class JetMap;
27 class SvtxTrack;
28 class SvtxTrackMap;
29 class PHHepMCGenEventMap;
30 
31 
32 class ExclusiveReco : public SubsysReco
33 {
34 
35 public:
36 
37  ExclusiveReco(std::string filename);
38 
39  int
40  Init(PHCompositeNode*);
41  int
42  InitRun(PHCompositeNode*);
43  int
44  process_event(PHCompositeNode*);
45  int
46  End(PHCompositeNode*);
47 
48  void
49  set_beam_energies( float beam_electron , float beam_proton )
50  {
51  _beam_electron_ptotal = fabs( beam_electron );
52  _beam_hadron_ptotal = fabs( beam_proton );
53  }
54 
55  void
56  set_do_process_dvmp( bool select)
57  {
58  _do_process_dvmp = select;
59  }
60 
61 
62 
63 private:
64 
65  /* proton rest mass */
66  const float _mproton;
67 
68  bool _verbose;
69  bool _save_towers;
70  bool _save_tracks;
71  bool _do_process_dvmp;
72 
73  int _ievent;
74  int _total_pass;
75 
76  std::string _filename;
77  TFile *_tfile;
78 
79  /* output tree for invariant mass calculation */
80  TTree* _tree_invariant_mass;
81 
82  /* output tree for reco particles */
83  TTree* _tree_event_reco;
84 
85  /* output tree for truth particles */
86  TTree* _tree_event_truth;
87 
88  /* beam energies electron and proton */
89  float _beam_electron_ptotal;
90  float _beam_hadron_ptotal;
91 
92  /* Track Projector object */
93  TrackProjectorPlaneECAL* _trackproj;
94 
96  PHCompositeNode *_topNode;
97 
98  int AddInvariantMassInformation();
99 
100  std::vector<float> true_eta, true_phi, true_ptotal;
101  std::vector<int> true_pid;
102  std::vector<bool> is_scattered_lepton;
103 
104  std::vector<float> reco_eta, reco_phi, reco_ptotal, reco_cluster_e;
105  std::vector<int> reco_charge;
106  std::vector<bool> reco_is_scattered_lepton;
107 
108  std::vector<float> _vect1;
109  std::vector<float> _vect2;
110  std::vector<float> _vect3;
111  std::vector<float> _vect4;
112  std::vector<float> _vect5;
113  std::vector<float> _vect6;
114 
115 };
116 
117 #endif // __ExclusiveReco_H__