Analysis Software
Documentation for sPHENIX simulation software
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
Photons.h
Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file Photons.h
1 #ifndef __PHOTONS_H__
2 #define __PHOTONS_H__
3 
4 #include <fun4all/SubsysReco.h>
5 #include <vector>
6 
7 #include <TFile.h>
8 #include <TH1.h>
9 #include <TH2.h>
10 #include <TMath.h>
11 #include <TTree.h>
12 
13 class PHCompositeNode;
14 class RawClusterContainer;
15 class RawCluster;
16 class SvtxTrackMap;
17 class JetMap;
18 class JetEvalStack;
19 class JetRecoEval;
20 class SvtxTrackEval;
21 class PHG4TruthInfoContainer;
22 class PHHepMCGenEvent;
23 //class RawTowerGeom;
24 //class RawTowerGeomContainer;
25 class GlobalVertex;
26 class Photons : public SubsysReco
27 {
28  public:
29  Photons(const std::string &name = "photonjet.root");
30  double isoconeradius, mincluspt;
31  float jet_cone_size;
32  float _etalow, _etahi;
33 
34  int _embed;
35  int Init(PHCompositeNode *);
36  int process_event(PHCompositeNode *);
37  int End(PHCompositeNode *);
38 
39  void set_cluspt_mincut(double pt) { mincluspt = pt; };
40  void SetFirstEventNum(int eventnum) { nevents = eventnum; };
41  void set_eta_lowhigh(float etalow, float etahi)
42  {
43  _etalow = etalow;
44  _etahi = etahi;
45  };
46  void use_embedding(int yesorno) { _embed = yesorno; };
47  private:
48  void Set_Tree_Branches();
49  void initialize_to_zero();
50 
51  TFile *file;
52  TTree *tree;
53  TTree *cluster_tree;
54  TTree *truth_g4particles;
55  TTree *inhcal_tree;
56  TTree *fcluster_tree;
57  TTree *recal_cluster_tree;
58 
59  std::string outfilename;
60 
61  int nevents;
62  TH1F *histo;
63 
64  //hcalcluster info
65  float hcal_energy;
66  float hcal_eta;
67  float hcal_phi;
68  float hcal_pt;
69  float hcal_px;
70  float hcal_py;
71  float hcal_pz;
72  float hcal_theta;
73  float hcal_x, hcal_y, hcal_z, hcal_t;
74 
75  //cluster info
76  float clus_energy;
77  float clus_eta;
78  float clus_phi;
79  float clus_pt;
80  float clus_px;
81  float clus_py;
82  float clus_pz;
83  float clus_theta;
84  float clus_x, clus_y, clus_z, clus_t;
85  float fmodphi, fmodeta;
86  float clus_ecore;
87  float clus_chi2;
88  float clus_prob;
89 
90  //recal cluster info
91  float rclus_energy;
92  float rclus_eta;
93  float rclus_phi;
94  float rclus_pt;
95  float rclus_px;
96  float rclus_py;
97  float rclus_pz;
98  float rclus_theta;
99  float rclus_x, rclus_y, rclus_z, rclus_t;
100  float rclus_ecore, rclus_chi2, rclus_prob;
101 
102  //all truth particles
103  float truthpx, truthpy, truthpz;
104  float truthp;
105  float truthphi;
106  float trutheta;
107  float truthpt;
108  float truthenergy;
109  int truthpid;
110  int numparticlesinevent;
111  int process_id;
112 
113  //clust truth variables
114  float clustruthpx;
115  float clustruthpy;
116  float clustruthpz;
117  float clustruthenergy;
118  float clustruthpt;
119  float clustruthphi;
120  float clustrutheta;
121  int clustruthpid;
122  float hclustruthpx;
123  float hclustruthpy;
124  float hclustruthpz;
125  float hclustruthenergy;
126  float hclustruthpt;
127  float hclustruthphi;
128  float hclustrutheta;
129  int hclustruthpid;
130 
131  //forward clusters
132  float fclusenergy;
133  float fclus_eta;
134  float fclus_phi;
135  float fclus_theta;
136  float fclus_pt;
137  int fclustruthpid;
138  float fclustruthpx;
139  float fclustruthpy;
140  float fclustruthpz;
141  float fclustruthenergy;
142  float fclustruthpt;
143  float fclustruthphi;
144  float fclustrutheta;
145  float fclus_px;
146  float fclus_py;
147  float fclus_pz;
148 
149  const float pi2 = -1.5707963;
150  const float threepi2 = 4.71238898;
151  const float pi = 3.1415926;
152 };
153 
154 #endif // __PHOTONS_H__