Back to home page

sPhenix code displayed by LXR

 
 

    


File indexing completed on 2025-08-03 08:15:56

0001 #include <TreeMaker.h>
0002 #include <TTree.h>
0003 
0004 
0005 
0006 int TreeMaker::InitializeTree(void)
0007 {
0008 
0009   tree = new TTree("ttree","sequoia");
0010 
0011   tree->Branch("jet2_n", &b_jet2_n,"jet2_n/I");
0012   tree->Branch("jet2_e", b_jet2_e,"jet2_e[jet2_n]/F");
0013   tree->Branch("jet2_pt", b_jet2_pt,"jet2_pt[jet2_n]/F");
0014   tree->Branch("jet2_eta",b_jet2_eta,"jet2_eta[jet2_n]/F");
0015   tree->Branch("jet2_phi",b_jet2_phi,"jet2_phi[jet2_n]/F");
0016 
0017   tree->Branch("jet3_n", &b_jet3_n,"jet3_n/I");
0018   tree->Branch("jet3_e", b_jet3_e,"jet3_e[jet3_n]/F");
0019   tree->Branch("jet3_pt", b_jet3_pt,"jet3_pt[jet3_n]/F");
0020   tree->Branch("jet3_eta",b_jet3_eta,"jet3_eta[jet3_n]/F");
0021   tree->Branch("jet3_phi",b_jet3_phi,"jet3_phi[jet3_n]/F");
0022 
0023   tree->Branch("jet4_n", &b_jet4_n,"jet4_n/I");
0024   tree->Branch("jet4_e", b_jet4_e,"jet4_e[jet4_n]/F");
0025   tree->Branch("jet4_pt", b_jet4_pt,"jet4_pt[jet4_n]/F");
0026   tree->Branch("jet4_eta",b_jet4_eta,"jet4_eta[jet4_n]/F");
0027   tree->Branch("jet4_phi",b_jet4_phi,"jet4_phi[jet4_n]/F");
0028 
0029   tree->Branch("jet5_n", &b_jet5_n,"jet5_n/I");
0030   tree->Branch("jet5_e", b_jet5_e,"jet5_e[jet5_n]/F");
0031   tree->Branch("jet5_pt", b_jet5_pt,"jet5_pt[jet5_n]/F");
0032   tree->Branch("jet5_eta",b_jet5_eta,"jet5_eta[jet5_n]/F");
0033   tree->Branch("jet5_phi",b_jet5_phi,"jet5_phi[jet5_n]/F");
0034 
0035   tree->Branch("modjet2_n", &b_modjet2_n,"modjet2_n/I");
0036   tree->Branch("modjet2_e", b_modjet2_e,"modjet2_e[modjet2_n]/F");
0037   tree->Branch("modjet2_pt", b_modjet2_pt,"modjet2_pt[modjet2_n]/F");
0038   tree->Branch("modjet2_eta",b_modjet2_eta,"modjet2_eta[modjet2_n]/F");
0039   tree->Branch("modjet2_phi",b_modjet2_phi,"modjet2_phi[modjet2_n]/F");
0040 
0041   tree->Branch("truthjet2_n", &b_truthjet2_n,"truthjet2_n/I");
0042   tree->Branch("truthjet2_e", b_truthjet2_e,"truthjet2_e[truthjet2_n]/F");
0043   tree->Branch("truthjet2_pt", b_truthjet2_pt,"truthjet2_pt[truthjet2_n]/F");
0044   tree->Branch("truthjet2_eta",b_truthjet2_eta,"truthjet2_eta[truthjet2_n]/F");
0045   tree->Branch("truthjet2_phi",b_truthjet2_phi,"truthjet2_phi[truthjet2_n]/F");
0046 
0047   tree->Branch("truthjet3_n", &b_truthjet3_n,"truthjet3_n/I");
0048   tree->Branch("truthjet3_e", b_truthjet3_e,"truthjet3_e[truthjet3_n]/F");
0049   tree->Branch("truthjet3_pt", b_truthjet3_pt,"truthjet3_pt[truthjet3_n]/F");
0050   tree->Branch("truthjet3_eta",b_truthjet3_eta,"truthjet3_eta[truthjet3_n]/F");
0051   tree->Branch("truthjet3_phi",b_truthjet3_phi,"truthjet3_phi[truthjet3_n]/F");
0052 
0053   tree->Branch("truthjet4_n", &b_truthjet4_n,"truthjet4_n/I");
0054   tree->Branch("truthjet4_e", b_truthjet4_e,"truthjet4_e[truthjet4_n]/F");
0055   tree->Branch("truthjet4_pt", b_truthjet4_pt,"truthjet4_pt[truthjet4_n]/F");
0056   tree->Branch("truthjet4_eta",b_truthjet4_eta,"truthjet4_eta[truthjet4_n]/F");
0057   tree->Branch("truthjet4_phi",b_truthjet4_phi,"truthjet4_phi[truthjet4_n]/F");
0058 
0059   tree->Branch("truthjet5_n", &b_truthjet5_n,"truthjet5_n/I");
0060   tree->Branch("truthjet5_e", b_truthjet5_e,"truthjet5_e[truthjet5_n]/F");
0061   tree->Branch("truthjet5_pt", b_truthjet5_pt,"truthjet5_pt[truthjet5_n]/F");
0062   tree->Branch("truthjet5_eta",b_truthjet5_eta,"truthjet5_eta[truthjet5_n]/F");
0063   tree->Branch("truthjet5_phi",b_truthjet5_phi,"truthjet5_phi[truthjet5_n]/F");
0064 
0065   return 0;
0066 
0067 }
0068 
0069 
0070 
0071 int TreeMaker::InitializeCounters()
0072 {
0073 
0074   b_tower_n = 0;
0075 
0076   b_tower_total_0 = 0;
0077   b_tower_total_1 = 0;
0078   b_tower_total_2 = 0;
0079 
0080   b_truthjet2_n = 0;
0081   b_truthjet3_n = 0;
0082   b_truthjet4_n = 0;
0083   b_truthjet5_n = 0;
0084 
0085   b_jet2_n = 0;
0086   b_jet3_n = 0;
0087   b_jet4_n = 0;
0088   b_jet5_n = 0;
0089 
0090   b_modjet2_n = 0;
0091 
0092   return 0;
0093 
0094 }
0095