ViewVC Help
View File | Revision Log | Show Annotations | Root Listing
root/cvsroot/UserCode/algomez/FourTop/TreeAnalyzer/interface/StoreTreeVariable.h
Revision: 1.3
Committed: Mon Jan 23 18:27:24 2012 UTC (13 years, 3 months ago) by algomez
Content type: text/plain
Branch: MAIN
Changes since 1.2: +24 -10 lines
Log Message:
include different B-tagging operating points for CSV

File Contents

# User Rev Content
1 algomez 1.1 #ifndef StoreTreeVariable_h
2     #define StoreTreeVariable_h
3    
4     #include <TFile.h>
5     #include <TTree.h>
6     #include <iostream>
7    
8     #define NMAX 100
9    
10     class StoreTreeVariable {
11    
12     public:
13    
14     StoreTreeVariable();
15    
16     virtual ~StoreTreeVariable();
17    
18     void SetMuonFalse(){StoreMuon = kFALSE;}
19     void SetElectronFalse(){StoreElectron = kFALSE;}
20     void SetJetFalse(){StoreJet = kFALSE;}
21     void SetVertexFalse(){StoreVertex = kFALSE;}
22     void SetTriggerFalse(){StoreTrigger = kFALSE;}
23     void SetMetFalse(){StoreMet = kFALSE;}
24 algomez 1.2 void SetGeneralFalse(){StoreGeneral = kFALSE;}
25 algomez 1.1
26     //typedef struct MuonVariables{
27     typedef struct {
28    
29     //Muon ID Variables
30     Int_t nummuons;
31     Double_t relisolation[NMAX];
32     Double_t d0[NMAX];
33     Int_t hits[NMAX];
34     Int_t trackhits[NMAX];
35     Double_t normalchi2[NMAX];
36     Int_t stations[NMAX];
37     Int_t pixelhits[NMAX];
38     Double_t muonvzwithPV[NMAX];
39     Int_t tightmuon[NMAX];
40     Int_t nloosemuon;
41     Int_t ntightmuon;
42    
43     //Muon Kinematic
44     Double_t pt[NMAX];
45     Double_t eta[NMAX];
46     Double_t phi[NMAX];
47     Double_t e[NMAX];
48    
49     //Muon Jet deltaR
50     Double_t deltaR[NMAX];
51    
52     } MuonVariables;
53    
54    
55     struct ElectronVariables{
56    
57     Int_t elecnum;
58     Int_t nlooseelec;
59     Int_t ntightelec;
60    
61     Double_t d0[NMAX];
62     Double_t elecvzwrtPV[NMAX];
63     Double_t relisolation[NMAX];
64     Double_t pfrelisolation[NMAX];//pfisolation
65     //Electron ID
66     Double_t sigmaIetaIeta[NMAX];
67     Double_t HoE[NMAX];
68     Double_t deltaphisc[NMAX];
69     Double_t deltaetasc[NMAX];
70    
71     //Kinematic
72     Double_t pt[NMAX];
73     Double_t eta[NMAX];
74     Double_t phi[NMAX];
75     Double_t e[NMAX];
76     //deltaR with Electron and jet
77     Double_t deltaR[NMAX];
78    
79     };
80    
81     struct JetVariables{
82    
83     Int_t numjets;
84     //Jet Kinematic
85     /*Double_t jetpt[NMAX];
86     Double_t jeteta[NMAX];
87     Double_t jetphi[NMAX];
88     Double_t jete[NMAX];
89    
90     //mc
91     Int_t jetparton[NMAX];
92    
93     //b tag
94     Double_t jettchp[NMAX];
95     Double_t jettche[NMAX];
96     Double_t jetsvhe[NMAX];
97     Double_t jetsvhp[NMAX];
98 algomez 1.3 Double_t jetcsv[NMAX]; */
99     Int_t numBjets_csvl;
100     Int_t numBjets_csvm;
101     Int_t numBjets_csvt;
102 algomez 1.1
103     //Jet Track number
104 algomez 1.3 //Int_t jettracksize[NMAX];
105    
106 algomez 1.1
107     // B discriminator
108 algomez 1.2 Double_t bdisc_1st;
109     Double_t bdisc_2nd;
110     Double_t bdisc_3rd;
111     Double_t bdisc_4th;
112 algomez 1.1
113     };
114    
115    
116     struct VertexVariables{
117    
118     Int_t numpvs;
119    
120     };
121    
122     struct TriggerVariables{
123    
124     Int_t triggername;
125     Int_t accept;
126     Int_t prescale;
127    
128     };
129    
130     struct MetVariables{
131    
132 algomez 1.3 /*
133 algomez 1.1 Double_t met;
134     Double_t metphi;
135     Double_t mht;
136     Double_t neupt;
137     Double_t rho;
138     Double_t sigmarho;
139     Double_t scalefactor;
140     Double_t eventpuweight;
141     Double_t eventpuweightdown;
142     Double_t eventpuweightup;
143     Double_t eventbtagweight; //Event B tag weight
144     Double_t event2btagweight_up; //Event B tag scale factor UP at least 2 TCHPM bjet
145     Double_t event2btagweight_down; //Event B tag scale factor DOWN
146     Double_t event1btagweight_up; //Event B tag scale factor UP at least 1 TCHPM bjet
147     Double_t event1btagweight_down;
148     Double_t eventonly1btagweight_up; //Event B tag scale factor UP at least 1 TCHPM bjet
149     Double_t eventonly1btagweight_down;
150     Double_t event1btagweight; //Event at least one TCHPM btag weight
151     Double_t eventonly1btagweight; //Only have one TCHPM btag weight
152     Double_t eventonly0btagweight; //Only have zero TCHPM btag weight
153    
154     Int_t csvbtagjetnum_up;
155     Int_t csvbtagjetnum_down;
156     Int_t flavorhistory;
157 algomez 1.3 */
158     Double_t Ht;
159     Double_t Stlep;
160     Double_t Stjet;
161 algomez 1.2 };
162    
163     struct GeneralVariables {
164 algomez 1.1
165     Int_t Run;
166     Int_t Lumi;
167     Int_t Event;
168 algomez 1.2 Double_t PUWeight;
169 algomez 1.1
170     };
171    
172     TTree* GetStoreTree(){return Top;}
173    
174     MuonVariables *GetMuonVariable(){return &VStoreMuon;}
175     ElectronVariables *GetElectronVariable(){return &VStoreElectron;}
176     JetVariables *GetJetVariable(){return &VStoreJet;}
177     VertexVariables *GetVertexVariable(){return &VStoreVertex;}
178     TriggerVariables *GetTriggerVariable(){return &VStoreTrigger;}
179     MetVariables *GetMetVariable(){return &VStoreMet;}
180 algomez 1.2 GeneralVariables *GetGeneralVariable(){return &VStoreGeneral;}
181 algomez 1.1
182     void InitialAll();
183     void InitialVertex();
184     void InitialMuon();
185     void InitialElectron();
186     void InitialJet();
187     void InitialTrigger();
188     void InitialMet();
189 algomez 1.2 void InitialGeneral();
190 algomez 1.1
191     private:
192    
193     TTree *Top;
194    
195     bool StoreMuon;
196     bool StoreVertex;
197     bool StoreElectron;
198     bool StoreJet;
199     bool StoreTrigger;
200     bool StoreMet;
201 algomez 1.2 bool StoreGeneral;
202 algomez 1.1
203     //V mean variables
204     MuonVariables VStoreMuon;
205     ElectronVariables VStoreElectron;
206     JetVariables VStoreJet;
207     VertexVariables VStoreVertex;
208     TriggerVariables VStoreTrigger;
209     MetVariables VStoreMet;
210 algomez 1.2 GeneralVariables VStoreGeneral;
211 algomez 1.1
212     };
213    
214     #endif
215    
216     StoreTreeVariable::StoreTreeVariable(){
217    
218     Top = new TTree("4Tree","4Tree");
219     Top->SetDirectory(0);
220    
221     StoreMuon = kTRUE;
222     StoreVertex = kTRUE;
223     StoreElectron = kTRUE;
224     StoreJet = kTRUE;
225     StoreTrigger = kTRUE;
226     StoreMet = kTRUE;
227 algomez 1.2 StoreGeneral = kTRUE;
228 algomez 1.1
229     }
230    
231     StoreTreeVariable::~StoreTreeVariable(){
232    
233     }
234    
235     void StoreTreeVariable::InitialMuon(){
236    
237     if(StoreMuon){
238     cout<<"We are initial the Muon Branch"<<endl;
239    
240     Top->Branch("muon_number",&VStoreMuon.nummuons,"nummuons/I");
241     Top->Branch("muon_isolation",VStoreMuon.relisolation,"reisolation[nummuons]/D");
242     Top->Branch("muon_d0",VStoreMuon.d0,"d0[nummuons]/D");
243     Top->Branch("muon_muonhist",VStoreMuon.hits,"hits[nummuons]/I");
244     Top->Branch("muon_trackhits",VStoreMuon.trackhits,"trackhits[nummuons]/I");
245     Top->Branch("muon_normalchi2",VStoreMuon.normalchi2,"normalchi2[nummuons]/D");
246     Top->Branch("muon_muonstations",VStoreMuon.stations,"stations[nummuons]/I");
247     Top->Branch("muon_pixelhits",VStoreMuon.pixelhits,"pixelhits[nummuons]/I");
248     Top->Branch("muon_muonvzwithPV",VStoreMuon.muonvzwithPV,"muonvzwithPV[nummuons]/D");
249     Top->Branch("muon_tightmuon",VStoreMuon.tightmuon,"tightmuon[nummuons]/I");
250     Top->Branch("muon_nloosemuon",&VStoreMuon.nloosemuon,"nloosemuon/I");
251     Top->Branch("muon_ntightmuon",&VStoreMuon.ntightmuon,"ntightmuon/I");
252     Top->Branch("muon_jet_deltaR",VStoreMuon.deltaR,"deltaR[nummuons]/D");
253    
254     //kinematic
255     Top->Branch("muon_pt",VStoreMuon.pt,"pt[nummuons]/D");
256     //Top->Branch("muon_pt",muonpt,"pt[nummuons]/D");
257     Top->Branch("muon_eta",VStoreMuon.eta,"eta[nummuons]/D");
258     Top->Branch("muon_phi",VStoreMuon.phi,"phi[nummuons]/D");
259     Top->Branch("muon_energy",VStoreMuon.e,"e[nummuons]/D");
260    
261     }
262    
263     }
264    
265     void StoreTreeVariable::InitialElectron(){
266    
267     if(StoreElectron){
268    
269     cout<<"We are initial the Electron Branch"<<endl;
270    
271     Top->Branch("electronnum",&VStoreElectron.elecnum,"elecnum/I");
272     //Electrom ID
273     Top->Branch("elec_d0",VStoreElectron.d0,"d0[elecnum]/D");
274     Top->Branch("elec_elecvzwrtPV",VStoreElectron.elecvzwrtPV,"elecivzwrtPV[elecnum]/D");
275     Top->Branch("elec_isolation",VStoreElectron.relisolation,"relisolation[elecnum]/D");
276     Top->Branch("elec_pfisolation",VStoreElectron.pfrelisolation,"pfrelisolation[elecnum]/D");
277     Top->Branch("elec_sigmaIetaIeta",VStoreElectron.sigmaIetaIeta,"sigmaIetaIeta[elecnum]/D");
278     Top->Branch("elec_HoE",VStoreElectron.HoE,"HoE[elecnum]/D");
279     Top->Branch("elec_deltaphisc",VStoreElectron.deltaphisc,"deltaphisc[elecnum]/D");
280     Top->Branch("elec_deltaetasc",VStoreElectron.deltaetasc,"deltaetasc[elecnum]/D");
281     //Electron kinematic
282     Top->Branch("elec_pt",VStoreElectron.pt,"pt[elecnum]/D");
283     Top->Branch("elec_eta",VStoreElectron.eta,"eta[elecnum]/D");
284     Top->Branch("elec_phi",VStoreElectron.phi,"phi[elecnum]/D");
285     Top->Branch("elec_energy",VStoreElectron.e,"e[elecnum]/D");
286     Top->Branch("elec_jet_deltaR",VStoreElectron.deltaR,"deltaR[elecnum]/D");
287    
288     }
289    
290     }
291    
292     void StoreTreeVariable::InitialJet(){
293    
294     if(StoreJet){
295    
296     cout<<"We are initial Jet Branch"<<endl;
297    
298     Top->Branch("jet_number",&VStoreJet.numjets,"numjets/I");
299     /*
300     Top->Branch("jet_pt",VStoreJet.jetpt,"jetpt[numjets]/D");
301     Top->Branch("jet_eta",VStoreJet.jeteta,"jeteta[numjets]/D");
302     Top->Branch("jet_phi",VStoreJet.jetphi,"jetphi[numjets]/D");
303     Top->Branch("jet_energy",VStoreJet.jete,"jete[numjets]/D");
304     Top->Branch("jet_partonflavor",VStoreJet.jetparton,"jetparton[numjets]/I");
305     //btag
306     Top->Branch("jet_btag_trackhighputy",VStoreJet.jettchp,"jettchp[numjets]/D");
307     Top->Branch("jet_btag_trackhigheff",VStoreJet.jettche,"jettche[numjets]/D");
308     Top->Branch("jet_btag_SVhigheff",VStoreJet.jetsvhe,"jetsvhe[numjets]/D");
309     Top->Branch("jet_btag_SVhighputy",VStoreJet.jetsvhp,"jetsvhp[numjets]/D");
310     Top->Branch("jet_btag_csv",VStoreJet.jetcsv,"jetcsv[numjets]/D");
311     */
312 algomez 1.3 Top->Branch("numBjets_csvl",&VStoreJet.numBjets_csvl,"numBjets_csvl/I");
313     Top->Branch("numBjets_csvm",&VStoreJet.numBjets_csvm,"numBjets_csvm/I");
314     Top->Branch("numBjets_csvt",&VStoreJet.numBjets_csvt,"numBjets_csvt/I");
315    
316 algomez 1.1 //b discriminant
317 algomez 1.3 Top->Branch("bdisc_1st",&VStoreJet.bdisc_1st,"bdisc_1st/D");
318     Top->Branch("bdisc_2nd",&VStoreJet.bdisc_2nd,"bdisc_2nd/D");
319     Top->Branch("bdisc_3rd",&VStoreJet.bdisc_3rd,"bdisc_3rd/D");
320     Top->Branch("bdisc_4th",&VStoreJet.bdisc_4th,"bdisc_4th/D");
321 algomez 1.1
322     //Jet track size
323     //Top->Branch("jet_tracksize",VStoreJet.jettracksize,"jettracksize[numjets]/I");
324     }
325    
326     }
327    
328     void StoreTreeVariable::InitialMet(){
329    
330     if(StoreMet){
331     cout<<"We are inital Met Branch"<<endl;
332 algomez 1.3 /*
333 algomez 1.1 Top->Branch("MET_energy",&VStoreMet.met,"met/D");
334     Top->Branch("MET_phi",&VStoreMet.metphi,"metphi/D");
335     Top->Branch("MHT_energy",&VStoreMet.mht,"mht/D");
336     Top->Branch("Neutrinuo_pt",&VStoreMet.neupt,"neupt/D");
337     Top->Branch("pileup_rho",&VStoreMet.rho,"rho/D");
338     Top->Branch("pileup_sigmarho",&VStoreMet.sigmarho,"sigmarho/D");
339     Top->Branch("scalefactor",&VStoreMet.scalefactor,"scalefactor/D");
340     Top->Branch("pileup_reweight",&VStoreMet.eventpuweight,"eventpuweight/D");
341     Top->Branch("pileup_reweightdown",&VStoreMet.eventpuweightdown,"eventpuweightdown/D");
342     Top->Branch("pileup_reweightup",&VStoreMet.eventpuweightup,"eventpuweightup/D");
343     Top->Branch("btagweight",&VStoreMet.eventbtagweight,"eventbtagweight/D");
344     Top->Branch("2btagweight_up",&VStoreMet.event2btagweight_up,"event2btagweight_up/D");
345     Top->Branch("2btagweight_down",&VStoreMet.event2btagweight_down,"event2btagweight_down/D");
346     Top->Branch("1btagweight_up",&VStoreMet.event1btagweight_up,"event1btagweight_up/D");
347     Top->Branch("1btagweight_down",&VStoreMet.event1btagweight_down,"event1btagweight_down/D");
348     Top->Branch("only1btagweight_up",&VStoreMet.eventonly1btagweight_up,"eventonly1btagweight_up/D");
349     Top->Branch("only1btagweight_down",&VStoreMet.eventonly1btagweight_down,"eventonly1btagweight_down/D");
350     Top->Branch("1btagweight",&VStoreMet.event1btagweight,"event1btagweight/D");// at least one btag weight factor
351     Top->Branch("only1btagweight",&VStoreMet.eventonly1btagweight,"eventonly1btagweight/D");// at least one btag weight factor
352     Top->Branch("only0btagweight",&VStoreMet.eventonly0btagweight,"eventonly0btagweight/D");// at least one btag weight factor
353     Top->Branch("csvbtagjetnum_up",&VStoreMet.csvbtagjetnum_up,"csvbtagjetnum_up/I");
354     Top->Branch("csvbtagjetnum_down",&VStoreMet.csvbtagjetnum_down,"csvbtagjetnum_down/I");
355     Top->Branch("flavorhistory",&VStoreMet.flavorhistory,"flavorhistory/I");
356 algomez 1.3 */
357     Top->Branch("ht",&VStoreMet.Ht,"Ht/D");
358     Top->Branch("Stlep",&VStoreMet.Stlep,"Stlep/D");
359     Top->Branch("Stjet",&VStoreMet.Stjet,"Stjet/D");
360 algomez 1.2 }
361     }
362    
363     void StoreTreeVariable::InitialGeneral(){
364    
365     if(StoreGeneral){
366     cout<<"We are inital General Branch"<<endl;
367    
368     Top->Branch("run",&VStoreGeneral.Run,"Run/I");
369     Top->Branch("lumi",&VStoreGeneral.Lumi,"Lumi/I");
370     Top->Branch("event",&VStoreGeneral.Event,"Event/I");
371     Top->Branch("PUWeight",&VStoreGeneral.PUWeight,"PUWeight/D");
372 algomez 1.1
373     }
374     }
375    
376     void StoreTreeVariable::InitialTrigger(){
377    
378     if(StoreTrigger){
379     cout<<"We are nitial Trigger Bracn"<<endl;
380    
381     Top->Branch("Trigger_Name",&VStoreTrigger.triggername,"triggername/I");
382     Top->Branch("Trigger_accept",&VStoreTrigger.accept,"accept/I");
383     Top->Branch("Trigger_prescale",&VStoreTrigger.prescale,"prescale/I");
384    
385     }
386    
387     }
388    
389     void StoreTreeVariable::InitialVertex(){
390    
391     if(StoreVertex){
392     cout<<"We are inital Vertex Branch"<<endl;
393    
394     Top->Branch("nPVs",&VStoreVertex.numpvs,"numpvs/I");
395    
396     }
397    
398     }
399    
400     void StoreTreeVariable::InitialAll(){
401    
402     InitialVertex();
403     InitialTrigger();
404     InitialMuon();
405     InitialElectron();
406     InitialJet();
407     InitialMet();
408 algomez 1.2 InitialGeneral();
409 algomez 1.1 }