analysis/TreeMaker/TreeMaker.h

0001 #ifndef __TREEMAKER_H__
0002 #define __TREEMAKER_H__
0003
0004 // --- need to check all these includes...
0005 #include <fun4all/SubsysReco.h>
0006
0007 #include "TTree.h"
0008 #include "TFile.h"
0009
0010 class PHCompositeNode;
0011
0012 class TreeMaker: public SubsysReco
0013 {
0014
0015  public:
0016
0017   TreeMaker(const std::string &name="TreeMaker.root");
0018   ~TreeMaker();
0019
0020   int Init(PHCompositeNode*);
0021   int InitRun(PHCompositeNode*);
0022   int process_event(PHCompositeNode*);
0023   int End(PHCompositeNode*);
0024   int CreateNode(PHCompositeNode*);
0025
0026   int InitializeTree();
0027   int InitializeCounters();
0028
0029   int GetTruthJets(PHCompositeNode*);
0030   int GetRecoJets(PHCompositeNode*);
0031   int GetClusters(PHCompositeNode*);
0032   int CopyAndMakeJets(PHCompositeNode*);
0033   int CopyAndMakeClusters(PHCompositeNode*);
0034   int UseFastJet(PHCompositeNode*);
0035
0036  private:
0037
0038   TFile *outfile;
0039
0040   TTree *tree;
0041
0042   std::string foutname;
0043
0044   int b_tower_n;
0045   int b_tower_layer[10000];
0046   float b_tower_e[10000];
0047   float b_tower_eta[10000];
0048   float b_tower_phi[10000];
0049
0050   float b_tower_total_0;
0051   float b_tower_total_1;
0052   float b_tower_total_2;
0053
0054   int b_jet5_n;
0055   float b_jet5_e[200];
0056   float b_jet5_pt[200];
0057   float b_jet5_eta[200];
0058   float b_jet5_phi[200];
0059
0060   int b_jet4_n;
0061   float b_jet4_e[200];
0062   float b_jet4_pt[200];
0063   float b_jet4_eta[200];
0064   float b_jet4_phi[200];
0065
0066   int b_jet3_n;
0067   float b_jet3_e[200];
0068   float b_jet3_pt[200];
0069   float b_jet3_eta[200];
0070   float b_jet3_phi[200];
0071
0072   int b_jet2_n;
0073   float b_jet2_e[200];
0074   float b_jet2_pt[200];
0075   float b_jet2_eta[200];
0076   float b_jet2_phi[200];
0077
0078   int b_modjet2_n;
0079   float b_modjet2_e[200];
0080   float b_modjet2_pt[200];
0081   float b_modjet2_eta[200];
0082   float b_modjet2_phi[200];
0083
0084   int b_truthjet5_n;
0085   float b_truthjet5_e[200];
0086   float b_truthjet5_pt[200];
0087   float b_truthjet5_eta[200];
0088   float b_truthjet5_phi[200];
0089
0090   int b_truthjet4_n;
0091   float b_truthjet4_e[200];
0092   float b_truthjet4_pt[200];
0093   float b_truthjet4_eta[200];
0094   float b_truthjet4_phi[200];
0095
0096   int b_truthjet3_n;
0097   float b_truthjet3_e[200];
0098   float b_truthjet3_pt[200];
0099   float b_truthjet3_eta[200];
0100   float b_truthjet3_phi[200];
0101
0102   int b_truthjet2_n;
0103   float b_truthjet2_e[200];
0104   float b_truthjet2_pt[200];
0105   float b_truthjet2_eta[200];
0106   float b_truthjet2_phi[200];
0107
0108 };
0109
0110 #endif // __TREEMAKER_H__
0111