ViewVC Help
View File | Revision Log | Show Annotations | Root Listing
root/cvsroot/UserCode/CITCommon/CommonData/interface/ElectronTree.h
Revision: 1.1
Committed: Wed Jul 18 19:29:54 2012 UTC (12 years, 9 months ago) by sixie
Content type: text/plain
Branch: MAIN
Log Message:
common data formats

File Contents

# User Rev Content
1 sixie 1.1 #ifndef ElectronTree_H
2     #define ElectronTree_H
3    
4     #include "TFile.h"
5     #include "TTree.h"
6     #include "TError.h"
7     #include <cmath>
8     #include "assert.h"
9    
10     class ElectronTree {
11    
12     public:
13    
14     /// bit map
15     /// DON'T CHANGE ORDER
16    
17     //*******************************************
18     //=== ElectronTriggerBits ====
19     //*******************************************
20     enum ElectronTriggerBits { kEleTrigger_Ele = 0x000001,
21     kEleTrigger_Ele_CaloIdL_CaloIsoVL = 0x000002,
22     kEleTrigger_Ele_CaloIdT_TrkIdVL_CaloIsoVL_TrkIsoVL = 0x000004
23     };
24    
25     /// variables
26     Float_t fWeight;
27     UInt_t fRunNumber;
28     UInt_t fLumiSectionNumber;
29     UInt_t fEventNumber;
30     Bool_t fEleEventNumberParity;
31     Float_t fElePt;
32     Float_t fEleEta;
33     Float_t fElePhi;
34     Float_t fEleSCEt;
35     Float_t fEleSCEta;
36     Float_t fEleSCPhi;
37     Float_t fEleEcalEnergy;
38     Bool_t fEleIsEcalDriven;
39     UInt_t fEleTriggerBit;
40     Float_t fRho;
41     UInt_t fNVertices;
42    
43     // Conversion and IP
44     Float_t fEleD0;
45     Float_t fEleDZ;
46     Float_t fEleIP3d;
47     Float_t fEleIP3dSig;
48     Bool_t fEleMatchedConversion;
49     Float_t fEleConvDCot;
50     Float_t fEleConvDist;
51     UInt_t fEleNMissHits;
52    
53     // E/P variables
54     Float_t fEleNBrem;
55     Float_t fEleFBrem;
56     Float_t fEleEOverP;
57     Float_t fEleESeedClusterOverPIn;
58     Float_t fEleESeedClusterOverPout;
59     Float_t fEleEEleClusterOverPout;
60     Float_t fEleOneOverEMinusOneOverP;
61    
62     // track cluster matching
63     Float_t fEleDEtaIn;
64     Float_t fEleDPhiIn;
65     Float_t fEledEtaCalo;
66     Float_t fEledPhiCalo;
67    
68     //shower shape
69     Float_t fEleSigmaIEtaIEta;
70     Float_t fEleSigmaIPhiIPhi;
71     Float_t fEleSigmaIEtaIPhi;
72     Float_t fEleSCEtaWidth;
73     Float_t fEleSCPhiWidth;
74     Float_t fEleR9;
75     Float_t fElePreShowerOverRaw;
76     Float_t fEleHoverE;
77    
78     //track quality
79     Float_t fEleGsfTrackChi2OverNdof;
80     Float_t fEleKFTrackChi2OverNDoF;
81     Int_t fEleKFTrackNHits;
82     Int_t fEleKFTrackNLayersWithMeasurement;
83     Float_t fEleOneMinusSeedE1x5OverE5x5;
84    
85     //Isolation Variables
86     Float_t fElePFMVA;
87     Float_t fEleTrkIso03;
88     Float_t fEleEMIso03;
89     Float_t fEleHadIso03;
90     Float_t fEleTrkIso04;
91     Float_t fEleEMIso04;
92     Float_t fEleHadIso04;
93     Float_t fElePFIso04;
94     Float_t fChargedIso_DR0p0To0p1;
95     Float_t fChargedIso_DR0p1To0p2;
96     Float_t fChargedIso_DR0p2To0p3;
97     Float_t fChargedIso_DR0p3To0p4;
98     Float_t fChargedIso_DR0p4To0p5;
99     Float_t fGammaIso_DR0p0To0p1;
100     Float_t fGammaIso_DR0p1To0p2;
101     Float_t fGammaIso_DR0p2To0p3;
102     Float_t fGammaIso_DR0p3To0p4;
103     Float_t fGammaIso_DR0p4To0p5;
104     Float_t fNeutralHadronIso_DR0p0To0p1;
105     Float_t fNeutralHadronIso_DR0p1To0p2;
106     Float_t fNeutralHadronIso_DR0p2To0p3;
107     Float_t fNeutralHadronIso_DR0p3To0p4;
108     Float_t fNeutralHadronIso_DR0p4To0p5;
109    
110     Bool_t fElePassTriggerDenominator;
111    
112     public:
113     /// this is the main element
114     TTree *tree_;
115     TFile *f_;
116    
117     /// hold the names of variables to facilitate things (filled during Init)
118     std::vector<std::string> variables_;
119    
120     /// default constructor
121     ElectronTree() {};
122     /// default destructor
123     ~ElectronTree(){
124     if (f_) f_->Close();
125     };
126    
127     /// initialize varibles and fill list of available variables
128     void InitVariables() {
129     fWeight = 0.0;
130     fRunNumber = 0.0;
131     fLumiSectionNumber = 0.0;
132     fEventNumber = 0.0;
133     fEleEventNumberParity = 0.0;
134     fElePt = 0.0;
135     fEleEta = 0.0;
136     fElePhi = 0.0;
137     fEleSCEt = 0.0;
138     fEleSCEta = 0.0;
139     fEleSCPhi = 0.0;
140     fEleEcalEnergy = 0.0;
141     fEleIsEcalDriven = 0.0;
142     fEleTriggerBit = 0.0;
143     fRho = 0.0;
144     fNVertices = 0.0;
145     fEleD0 = 0.0;
146     fEleDZ = 0.0;
147     fEleIP3d = 0.0;
148     fEleIP3dSig = 0.0;
149     fEleMatchedConversion = 0.0;
150     fEleConvDCot = 0.0;
151     fEleConvDist = 0.0;
152     fEleNMissHits = 0.0;
153     fEleNBrem = 0.0;
154     fEleFBrem = 0.0;
155     fEleEOverP = 0.0;
156     fEleESeedClusterOverPIn = 0.0;
157     fEleESeedClusterOverPout = 0.0;
158     fEleEEleClusterOverPout = 0.0;
159     fEleOneOverEMinusOneOverP = 0.0;
160     fEleDEtaIn = 0.0;
161     fEleDPhiIn = 0.0;
162     fEledEtaCalo = 0.0;
163     fEledPhiCalo = 0.0;
164     fEleSigmaIEtaIEta = 0.0;
165     fEleSigmaIPhiIPhi = 0.0;
166     fEleSigmaIEtaIPhi = 0.0;
167     fEleSCEtaWidth = 0.0;
168     fEleSCPhiWidth = 0.0;
169     fEleR9 = 0.0;
170     fElePreShowerOverRaw = 0.0;
171     fEleHoverE = 0.0;
172     fEleGsfTrackChi2OverNdof = 0.0;
173     fEleKFTrackChi2OverNDoF = 0.0;
174     fEleKFTrackNHits = 0.0;
175     fEleKFTrackNLayersWithMeasurement = 0.0;
176     fEleOneMinusSeedE1x5OverE5x5 = 0.0;
177     fElePFMVA = 0.0;
178     fEleTrkIso03 = 0.0;
179     fEleEMIso03 = 0.0;
180     fEleHadIso03 = 0.0;
181     fEleTrkIso04 = 0.0;
182     fEleEMIso04 = 0.0;
183     fEleHadIso04 = 0.0;
184     fElePFIso04 = 0.0;
185     fChargedIso_DR0p0To0p1 = 0.0;
186     fChargedIso_DR0p1To0p2 = 0.0;
187     fChargedIso_DR0p2To0p3 = 0.0;
188     fChargedIso_DR0p3To0p4 = 0.0;
189     fChargedIso_DR0p4To0p5 = 0.0;
190     fGammaIso_DR0p0To0p1 = 0.0;
191     fGammaIso_DR0p1To0p2 = 0.0;
192     fGammaIso_DR0p2To0p3 = 0.0;
193     fGammaIso_DR0p3To0p4 = 0.0;
194     fGammaIso_DR0p4To0p5 = 0.0;
195     fNeutralHadronIso_DR0p0To0p1 = 0.0;
196     fNeutralHadronIso_DR0p1To0p2 = 0.0;
197     fNeutralHadronIso_DR0p2To0p3 = 0.0;
198     fNeutralHadronIso_DR0p3To0p4 = 0.0;
199     fNeutralHadronIso_DR0p4To0p5 = 0.0;
200     fElePassTriggerDenominator = 0.0;
201     }
202    
203     /// load a ElectronTree
204     void LoadTree(const char* file){
205     f_ = TFile::Open(file);
206     assert(f_);
207     tree_ = dynamic_cast<TTree*>(f_->Get("Electrons"));
208     assert(tree_);
209     }
210    
211     /// create a ElectronTree
212     void CreateTree(){
213     tree_ = new TTree("Electrons","Electrons");
214     f_ = 0;
215    
216     //book the branches
217     tree_->Branch("weight",&fWeight,"weight/F");
218     tree_->Branch("run",&fRunNumber,"run/i");
219     tree_->Branch("lumi",&fLumiSectionNumber,"lumi/i");
220     tree_->Branch("event",&fEventNumber,"event/i");
221     tree_->Branch("EventNumberParity",&fEleEventNumberParity,"EventNumberParity/O");
222     tree_->Branch("pt",&fElePt,"pt/F");
223     tree_->Branch("eta",&fEleEta,"eta/F");
224     tree_->Branch("phi",&fElePhi,"phi/F");
225     tree_->Branch("scEt",&fEleSCEt,"scEt/F");
226     tree_->Branch("scEta",&fEleSCEta,"scEta/F");
227     tree_->Branch("scPhi",&fEleSCPhi,"scPhi/F");
228     tree_->Branch("ecalenergy",&fEleEcalEnergy,"ecalenergy/F");
229     tree_->Branch("ecaldriven",&fEleIsEcalDriven,"ecaldriven/O");
230     tree_->Branch("triggerBit",&fEleTriggerBit,"triggerBit/i");
231     tree_->Branch("rho",&fRho,"rho/F");
232     tree_->Branch("vertices",&fNVertices,"vertices/i");
233     tree_->Branch("d0",&fEleD0,"d0/F");
234     tree_->Branch("dz",&fEleDZ,"dz/F");
235     tree_->Branch("ip3d",&fEleIP3d,"ip3d/F");
236     tree_->Branch("ip3ds",&fEleIP3dSig,"ip3ds/F");
237     tree_->Branch("matchConv",&fEleMatchedConversion,"matchConv/O");
238     tree_->Branch("dcot",&fEleConvDCot,"dcot/F");
239     tree_->Branch("dist",&fEleConvDist,"dist/F");
240     tree_->Branch("missHits",&fEleNMissHits,"NMissHits/i");
241     tree_->Branch("nbrems",&fEleNBrem,"nbrems/F");
242     tree_->Branch("fbrem",&fEleFBrem,"fbrem/F");
243     tree_->Branch("EoP",&fEleEOverP,"EoP/F");
244     tree_->Branch("EoPin",&fEleESeedClusterOverPIn,"EoPin/F");
245     tree_->Branch("ESeedoPout",&fEleESeedClusterOverPout,"ESeedoPout/F");
246     tree_->Branch("EEleoPout",&fEleEEleClusterOverPout,"EEleoPout/F");
247     // tree_->Branch("eleEoPout",&fEleEEleClusterOverPout,"eleEoPout/F");
248     tree_->Branch("IoEmIoP",&fEleOneOverEMinusOneOverP,"IoEmIoP/F");
249     tree_->Branch("deta",&fEleDEtaIn,"deta/F");
250     tree_->Branch("dphi",&fEleDPhiIn,"dphi/F");
251     tree_->Branch("detacalo",&fEledEtaCalo,"detacalo/F");
252     tree_->Branch("dphicalo",&fEledPhiCalo,"dphicalo/F");
253     tree_->Branch("see",&fEleSigmaIEtaIEta,"see/F");
254     tree_->Branch("spp",&fEleSigmaIPhiIPhi,"spp/F");
255     tree_->Branch("sep",&fEleSigmaIEtaIPhi,"sep/F");
256     tree_->Branch("etawidth",&fEleSCEtaWidth,"etawidth/F");
257     tree_->Branch("phiwidth",&fEleSCPhiWidth,"phiwidth/F");
258     tree_->Branch("R9",&fEleR9,"R9/F");
259     tree_->Branch("PreShowerOverRaw",&fElePreShowerOverRaw,"PreShowerOverRaw/F");
260     tree_->Branch("HoE",&fEleHoverE,"HoE/F");
261     tree_->Branch("gsfchi2",&fEleGsfTrackChi2OverNdof,"gsfchi2/F");
262     tree_->Branch("kfchi2",&fEleKFTrackChi2OverNDoF,"kfchi2/F");
263     tree_->Branch("kfhits",&fEleKFTrackNHits,"kfhits/I");
264     tree_->Branch("kflayers",&fEleKFTrackNLayersWithMeasurement,"kflayers/I");
265     tree_->Branch("OneMinusSeedE1x5OverE5x5",&fEleOneMinusSeedE1x5OverE5x5,"OneMinusSeedE1x5OverE5x5/F");
266     tree_->Branch("PFMVA",&fElePFMVA,"PFMVA/F");
267     tree_->Branch("trkIso03",&fEleTrkIso03,"trkIso03/F");
268     tree_->Branch("ecalIso03",&fEleEMIso03,"ecalIso03/F");
269     tree_->Branch("hcalIso03",&fEleHadIso03,"hcalIso03/F");
270     tree_->Branch("trkIso04",&fEleTrkIso04,"trkIso04/F");
271     tree_->Branch("ecalIso04",&fEleEMIso04,"ecalIso04/F");
272     tree_->Branch("hcalIso04",&fEleHadIso04,"hcalIso04/F");
273     tree_->Branch("pfIso04",&fElePFIso04,"pfIso04/F");
274     tree_->Branch("ChargedIso_DR0p0To0p1",&fChargedIso_DR0p0To0p1,"ChargedIso_DR0p0To0p1/F");
275     tree_->Branch("ChargedIso_DR0p1To0p2",&fChargedIso_DR0p1To0p2,"ChargedIso_DR0p1To0p2/F");
276     tree_->Branch("ChargedIso_DR0p2To0p3",&fChargedIso_DR0p2To0p3,"ChargedIso_DR0p2To0p3/F");
277     tree_->Branch("ChargedIso_DR0p3To0p4",&fChargedIso_DR0p3To0p4,"ChargedIso_DR0p3To0p4/F");
278     tree_->Branch("ChargedIso_DR0p4To0p5",&fChargedIso_DR0p4To0p5,"ChargedIso_DR0p4To0p5/F");
279     tree_->Branch("GammaIso_DR0p0To0p1",&fGammaIso_DR0p0To0p1,"GammaIso_DR0p0To0p1/F");
280     tree_->Branch("GammaIso_DR0p1To0p2",&fGammaIso_DR0p1To0p2,"GammaIso_DR0p1To0p2/F");
281     tree_->Branch("GammaIso_DR0p2To0p3",&fGammaIso_DR0p2To0p3,"GammaIso_DR0p2To0p3/F");
282     tree_->Branch("GammaIso_DR0p3To0p4",&fGammaIso_DR0p3To0p4,"GammaIso_DR0p3To0p4/F");
283     tree_->Branch("GammaIso_DR0p4To0p5",&fGammaIso_DR0p4To0p5,"GammaIso_DR0p4To0p5/F");
284     tree_->Branch("NeutralHadronIso_DR0p0To0p1",&fNeutralHadronIso_DR0p0To0p1,"NeutralHadronIso_DR0p0To0p1/F");
285     tree_->Branch("NeutralHadronIso_DR0p1To0p2",&fNeutralHadronIso_DR0p1To0p2,"NeutralHadronIso_DR0p1To0p2/F");
286     tree_->Branch("NeutralHadronIso_DR0p2To0p3",&fNeutralHadronIso_DR0p2To0p3,"NeutralHadronIso_DR0p2To0p3/F");
287     tree_->Branch("NeutralHadronIso_DR0p3To0p4",&fNeutralHadronIso_DR0p3To0p4,"NeutralHadronIso_DR0p3To0p4/F");
288     tree_->Branch("NeutralHadronIso_DR0p4To0p5",&fNeutralHadronIso_DR0p4To0p5,"NeutralHadronIso_DR0p4To0p5/F");
289     tree_->Branch("PassTriggerDenominator",&fElePassTriggerDenominator,"PassTriggerDenominator/O");
290    
291    
292     }
293    
294     // initialze a ElectronTree
295     void InitTree(){
296     assert(tree_);
297     // don't forget to set pointers to zero before you set address
298     // or you will fully appreciate that "ROOT sucks" :)
299     InitVariables();
300     //Set branch address
301     Int_t currentState = gErrorIgnoreLevel;
302    
303     tree_->SetBranchAddress("weight",&fWeight);
304     tree_->SetBranchAddress("run",&fRunNumber);
305     tree_->SetBranchAddress("lumi",&fLumiSectionNumber);
306     tree_->SetBranchAddress("event",&fEventNumber);
307     tree_->SetBranchAddress("EventNumberParity",&fEleEventNumberParity);
308     tree_->SetBranchAddress("pt",&fElePt);
309     tree_->SetBranchAddress("eta",&fEleEta);
310     tree_->SetBranchAddress("phi",&fElePhi);
311     tree_->SetBranchAddress("scEt",&fEleSCEt);
312     tree_->SetBranchAddress("scEta",&fEleSCEta);
313     tree_->SetBranchAddress("scPhi",&fEleSCPhi);
314     tree_->SetBranchAddress("ecalenergy",&fEleEcalEnergy);
315     tree_->SetBranchAddress("ecaldriven",&fEleIsEcalDriven);
316     tree_->SetBranchAddress("triggerBit",&fEleTriggerBit);
317     tree_->SetBranchAddress("rho",&fRho);
318     tree_->SetBranchAddress("vertices",&fNVertices);
319     tree_->SetBranchAddress("d0",&fEleD0);
320     tree_->SetBranchAddress("dz",&fEleDZ);
321     tree_->SetBranchAddress("ip3d",&fEleIP3d);
322     tree_->SetBranchAddress("ip3ds",&fEleIP3dSig);
323     tree_->SetBranchAddress("matchConv",&fEleMatchedConversion);
324     tree_->SetBranchAddress("dcot",&fEleConvDCot);
325     tree_->SetBranchAddress("dist",&fEleConvDist);
326     tree_->SetBranchAddress("missHits",&fEleNMissHits);
327     tree_->SetBranchAddress("nbrems",&fEleNBrem);
328     tree_->SetBranchAddress("fbrem",&fEleFBrem);
329     tree_->SetBranchAddress("EoP",&fEleEOverP);
330     tree_->SetBranchAddress("EoPin",&fEleESeedClusterOverPIn);
331     tree_->SetBranchAddress("ESeedoPout",&fEleESeedClusterOverPout);
332     tree_->SetBranchAddress("EEleoPout",&fEleEEleClusterOverPout);
333     tree_->SetBranchAddress("IoEmIoP",&fEleOneOverEMinusOneOverP);
334     tree_->SetBranchAddress("deta",&fEleDEtaIn);
335     tree_->SetBranchAddress("dphi",&fEleDPhiIn);
336     tree_->SetBranchAddress("detacalo",&fEledEtaCalo);
337     tree_->SetBranchAddress("dphicalo",&fEledPhiCalo);
338     tree_->SetBranchAddress("see",&fEleSigmaIEtaIEta);
339     tree_->SetBranchAddress("spp",&fEleSigmaIPhiIPhi);
340     tree_->SetBranchAddress("sep",&fEleSigmaIEtaIPhi);
341     tree_->SetBranchAddress("etawidth",&fEleSCEtaWidth);
342     tree_->SetBranchAddress("phiwidth",&fEleSCPhiWidth);
343     tree_->SetBranchAddress("R9",&fEleR9);
344     tree_->SetBranchAddress("PreShowerOverRaw",&fElePreShowerOverRaw);
345     tree_->SetBranchAddress("HoE",&fEleHoverE);
346     tree_->SetBranchAddress("gsfchi2",&fEleGsfTrackChi2OverNdof);
347     tree_->SetBranchAddress("kfchi2",&fEleKFTrackChi2OverNDoF);
348     tree_->SetBranchAddress("kfhits",&fEleKFTrackNHits);
349     tree_->SetBranchAddress("kflayers",&fEleKFTrackNLayersWithMeasurement);
350     tree_->SetBranchAddress("OneMinusSeedE1x5OverE5x5",&fEleOneMinusSeedE1x5OverE5x5);
351     tree_->SetBranchAddress("PFMVA",&fElePFMVA);
352     tree_->SetBranchAddress("trkIso03",&fEleTrkIso03);
353     tree_->SetBranchAddress("ecalIso03",&fEleEMIso03);
354     tree_->SetBranchAddress("hcalIso03",&fEleHadIso03);
355     tree_->SetBranchAddress("trkIso04",&fEleTrkIso04);
356     tree_->SetBranchAddress("ecalIso04",&fEleEMIso04);
357     tree_->SetBranchAddress("hcalIso04",&fEleHadIso04);
358     tree_->SetBranchAddress("pfIso04",&fElePFIso04);
359     tree_->SetBranchAddress("ChargedIso_DR0p0To0p1",&fChargedIso_DR0p0To0p1);
360     tree_->SetBranchAddress("ChargedIso_DR0p1To0p2",&fChargedIso_DR0p1To0p2);
361     tree_->SetBranchAddress("ChargedIso_DR0p2To0p3",&fChargedIso_DR0p2To0p3);
362     tree_->SetBranchAddress("ChargedIso_DR0p3To0p4",&fChargedIso_DR0p3To0p4);
363     tree_->SetBranchAddress("ChargedIso_DR0p4To0p5",&fChargedIso_DR0p4To0p5);
364     tree_->SetBranchAddress("GammaIso_DR0p0To0p1",&fGammaIso_DR0p0To0p1);
365     tree_->SetBranchAddress("GammaIso_DR0p1To0p2",&fGammaIso_DR0p1To0p2);
366     tree_->SetBranchAddress("GammaIso_DR0p2To0p3",&fGammaIso_DR0p2To0p3);
367     tree_->SetBranchAddress("GammaIso_DR0p3To0p4",&fGammaIso_DR0p3To0p4);
368     tree_->SetBranchAddress("GammaIso_DR0p4To0p5",&fGammaIso_DR0p4To0p5);
369     tree_->SetBranchAddress("NeutralHadronIso_DR0p0To0p1",&fNeutralHadronIso_DR0p0To0p1);
370     tree_->SetBranchAddress("NeutralHadronIso_DR0p1To0p2",&fNeutralHadronIso_DR0p1To0p2);
371     tree_->SetBranchAddress("NeutralHadronIso_DR0p2To0p3",&fNeutralHadronIso_DR0p2To0p3);
372     tree_->SetBranchAddress("NeutralHadronIso_DR0p3To0p4",&fNeutralHadronIso_DR0p3To0p4);
373     tree_->SetBranchAddress("NeutralHadronIso_DR0p4To0p5",&fNeutralHadronIso_DR0p4To0p5);
374     tree_->SetBranchAddress("PassTriggerDenominator",&fElePassTriggerDenominator);
375    
376     gErrorIgnoreLevel = currentState;
377     }
378    
379     };
380    
381    
382    
383     #endif