ViewVC Help
View File | Revision Log | Show Annotations | Root Listing
root/cvsroot/UserCode/VHbbAnalysis/VHbbDataFormats/interface/TriggerWeight.h
Revision: 1.15
Committed: Wed Jun 6 13:25:49 2012 UTC (12 years, 10 months ago) by arizzi
Content type: text/plain
Branch: MAIN
CVS Tags: Step2ForV33_v1, EdmV33Jun12v2, EdmV33Jun12v1, Step2ForV32_v2, Step2ForV32_v1
Changes since 1.14: +6 -0 lines
Log Message:
fix missing line loading mu leg

File Contents

# User Rev Content
1 arizzi 1.1 #ifndef TRIGGERWEIGHT_H
2     #define TRIGGERWEIGHT_H
3    
4     #include "FWCore/ParameterSet/interface/ProcessDesc.h"
5     #include "FWCore/PythonParameterSet/interface/PythonProcessDesc.h"
6 arizzi 1.11 //#include "VHbbAnalysis/VHbbDataFormats/interface/TriggerZnunuCurve.h"
7 arizzi 1.4 #include "VHbbAnalysis/VHbbDataFormats/interface/MultiThresholdEfficiency.h"
8 arizzi 1.1 #include <TH1F.h>
9 arizzi 1.3 #include <TF1.h>
10 arizzi 1.1 #include <TFile.h>
11     #include <TTree.h>
12     #include <iostream>
13     class TriggerWeight
14     {
15     public:
16 arizzi 1.11 TriggerWeight(const edm::ParameterSet& ana) : combiner2Thr(2), combiner1Thr(1)
17 arizzi 1.1 {
18 arizzi 1.4 tscaleHLTmu=openFile(ana,"hltMuFileName");
19     tscaleIDmu=openFile(ana,"idMuFileName");
20     tscaleHLTele1=openFile(ana,"hltEle1FileName");
21     tscaleHLTele2=openFile(ana,"hltEle2FileName");
22 arizzi 1.9 tscaleHLTele1Aug=openFile(ana,"hltEle1AugFileName");
23     tscaleHLTele2Aug=openFile(ana,"hltEle2AugFileName");
24 arizzi 1.5 tscaleID80Ele=openFile(ana,"idEle80FileName");
25     tscaleID95Ele=openFile(ana,"idEle95FileName");
26 arizzi 1.4 tscaleHLTeleJet1=openFile(ana,"hltJetEle1FileName");
27     tscaleHLTeleJet2=openFile(ana,"hltJetEle2FileName");
28 arizzi 1.5 tscaleRecoEle=openFile(ana,"recoEleFileName");
29     // tscalePFMHTele=openFile(ana,"hltPFMHTEleFileName");
30     tscaleSingleEleMay=openFile(ana,"hltSingleEleMayFileName");
31     tscaleSingleEleV4=openFile(ana,"hltSingleEleV4FileName");
32 arizzi 1.6 tscaleHLTmuOr30=openFile(ana,"hltMuOr30FileName");
33 arizzi 1.4
34 arizzi 1.12 tscaleSingleEle2012Awp95=openFile(ana,"hltSingleEle2012Awp95");
35     tscaleSingleEle2012Awp80=openFile(ana,"hltSingleEle2012Awp80");
36     tscaleSingleMuon2012A=openFile(ana,"hltSingleMuon2012A");
37     tscaleDoubleEle2012A_leg8=openFile(ana,"hltDoubleEle2012A_leg8");
38     tscaleDoubleEle2012A_leg17=openFile(ana,"hltDoubleEle2012A_leg17");
39     tscaleDoubleMuon2012A_leg8=openFile(ana,"hltDoubleMuon2012A_leg8");
40 arizzi 1.15 tscaleDoubleMuon2012A_leg17=openFile(ana,"hltDoubleMuon2012A_leg17");
41 arizzi 1.13
42     tscaleDoubleMuon2012A_dZ=openFile(ana,"hltDoubleMuon2012A_dZ");
43     tscaleDoubleEle2012A_dZ=openFile(ana,"hltDoubleEle2012A_dZ");
44 arizzi 1.12
45     tscaleMuPlusWCandPt2012A_legMu=openFile(ana,"hltMuPlusWCandPt2012A_legMu");
46     tscaleMuPlusWCandPt2012A_legW=openFile(ana,"hltMuPlusWCandPt2012A_legW");
47    
48 arizzi 1.1 if(tscaleHLTmu == 0 || tscaleIDmu == 0)
49     {
50     std::cout << "ERROR: cannot load Muon Trigger efficiencies" << std::endl;
51     }
52 arizzi 1.4
53    
54 arizzi 1.1
55     }
56 arizzi 1.4
57 arizzi 1.10 static TTree * openFile(const edm::ParameterSet& ana, const char * name)
58 arizzi 1.4 {
59 arizzi 1.5 TFile *hltMuFile = new TFile (ana.getParameter<std::string> (name).c_str(),"read");
60 arizzi 1.4 if(hltMuFile) return (TTree*) hltMuFile->Get("tree");
61     else return 0;
62     }
63    
64 arizzi 1.10 static std::pair<float,float> efficiencyFromPtEta(float pt1, float eta1, TTree *t)
65 arizzi 1.4 {
66 arizzi 1.15 // std::cout << "here " << t << " pt1 " << pt1 << " eta1 " << eta1 << std::endl;
67 arizzi 1.4 float s1 = 1.,err=1.;
68     std::pair<float,float> r(s1,err);
69     if(!t) return r;
70     float ptMin,ptMax,etaMin,etaMax,scale,error;
71     int count = 0;
72     t->SetBranchAddress("ptMin",&ptMin);
73     t->SetBranchAddress("ptMax",&ptMax);
74     t->SetBranchAddress("etaMin",&etaMin);
75     t->SetBranchAddress("etaMax",&etaMax);
76     t->SetBranchAddress("scale",&scale);
77     t->SetBranchAddress("error",&error);
78 arizzi 1.7 float lastPtBin = 200;
79     /* for(int jentry = 0; jentry < t->GetEntries(); jentry++)
80     {
81     t->GetEntry(jentry);
82     if(ptMax >= lastPtBin) lastPtBin =ptMax;
83     }*/
84 arizzi 1.4 for(int jentry = 0; jentry < t->GetEntries(); jentry++)
85     {
86     t->GetEntry(jentry);
87 arizzi 1.7 if(ptMax==lastPtBin) ptMax=1e99;
88 arizzi 1.4 if((pt1 > ptMin) && (pt1 < ptMax) && (eta1 > etaMin) && (eta1 < etaMax))
89     {
90     s1 = scale;
91     err=error;
92     count++;
93     }
94     }
95    
96     if(count == 0 || s1 == 0)
97     {
98     return r;
99     }
100    
101     r.first=s1;
102     r.second = err;
103     return (r);
104     }
105 arizzi 1.1
106     float scaleMuIsoHLT(float pt1, float eta1)
107     {
108 arizzi 1.4 return efficiencyFromPtEta(pt1,eta1,tscaleHLTmu).first;
109 arizzi 1.1 }
110    
111    
112    
113     float scaleMuID(float pt1, float eta1)
114     {
115 arizzi 1.4 return efficiencyFromPtEta(pt1,eta1,tscaleIDmu).first;
116 arizzi 1.1 }
117    
118 arizzi 1.3
119 arizzi 1.9 double scaleDoubleEle17Ele8Aug( std::vector<float> pt, std::vector<float> eta )
120     {
121     std::vector< std::vector<float> > allEleWithEffs;
122     for(unsigned int j=0; j< pt.size(); j++)
123     {
124     std::vector<float> thisEleEffs;
125     thisEleEffs.push_back(efficiencyFromPtEta(pt[j],eta[j],tscaleHLTele1Aug).first);
126     thisEleEffs.push_back(efficiencyFromPtEta(pt[j],eta[j],tscaleHLTele2Aug).first);
127    
128     allEleWithEffs.push_back(thisEleEffs);
129     }
130    
131     return combiner2Thr.weight<Trigger1High2Loose>(allEleWithEffs);
132    
133     }
134    
135    
136 arizzi 1.5 double scaleDoubleEle17Ele8( std::vector<float> pt, std::vector<float> eta )
137     {
138     std::vector< std::vector<float> > allEleWithEffs;
139     for(unsigned int j=0; j< pt.size(); j++)
140     {
141     std::vector<float> thisEleEffs;
142     thisEleEffs.push_back(efficiencyFromPtEta(pt[j],eta[j],tscaleHLTele1).first);
143     thisEleEffs.push_back(efficiencyFromPtEta(pt[j],eta[j],tscaleHLTele2).first);
144 arizzi 1.9
145 arizzi 1.5 allEleWithEffs.push_back(thisEleEffs);
146     }
147    
148     return combiner2Thr.weight<Trigger1High2Loose>(allEleWithEffs);
149    
150     }
151    
152     double scaleSingleEleMay( float pt, float eta){ return efficiencyFromPtEta(pt,eta,tscaleSingleEleMay).first;}
153     double scaleSingleEleV4( float pt, float eta){ return efficiencyFromPtEta(pt,eta,tscaleSingleEleV4).first; }
154     double scaleID80Ele( float pt, float eta) { return efficiencyFromPtEta(pt,eta,tscaleID80Ele).first; }
155     double scaleID95Ele( float pt, float eta) { return efficiencyFromPtEta(pt,eta,tscaleID95Ele).first; }
156     double scaleRecoEle( float pt, float eta){ return efficiencyFromPtEta(pt,eta,tscaleRecoEle).first; }
157     double scalePFMHTEle( float MetPFPt){
158     double weightPFMHTrigger=0.;
159    
160     //FIXME: read from file
161 dlopes 1.8 if(MetPFPt>0. && MetPFPt<5.) weightPFMHTrigger=0.305;
162     if(MetPFPt>5. && MetPFPt<10.) weightPFMHTrigger=0.351;
163     if(MetPFPt>10. && MetPFPt<15.) weightPFMHTrigger=0.461;
164     if(MetPFPt>15. && MetPFPt<20.) weightPFMHTrigger=0.572;
165     if(MetPFPt>20. && MetPFPt<25.) weightPFMHTrigger=0.713;
166     if(MetPFPt>25. && MetPFPt<30.) weightPFMHTrigger=0.844;
167     if(MetPFPt>30. && MetPFPt<35.) weightPFMHTrigger=0.914;
168     if(MetPFPt>35. && MetPFPt<40.) weightPFMHTrigger=0.939;
169     if(MetPFPt>40. && MetPFPt<45.) weightPFMHTrigger=0.981;
170     if(MetPFPt>45. && MetPFPt<50.) weightPFMHTrigger=0.982;
171     if(MetPFPt>50. && MetPFPt<60.) weightPFMHTrigger=0.993;
172     if(MetPFPt>60. && MetPFPt<70.) weightPFMHTrigger=0.995;
173     if(MetPFPt>70. && MetPFPt<100.) weightPFMHTrigger=0.995;
174     if(MetPFPt>100.) weightPFMHTrigger=1.;
175 arizzi 1.5 return weightPFMHTrigger;
176     }
177    
178 arizzi 1.11
179 arizzi 1.5 double scaleJet30Jet25( std::vector<float> pt, std::vector<float> eta)
180 arizzi 1.4 {
181 arizzi 1.5
182     std::vector< std::vector<float> > allJetsWithEffs;
183     for(unsigned int j=0; j< pt.size(); j++)
184     {
185     std::vector<float> thisJetEffs;
186     thisJetEffs.push_back(efficiencyFromPtEta(pt[j],eta[j],tscaleHLTeleJet1).first);
187     thisJetEffs.push_back(efficiencyFromPtEta(pt[j],eta[j],tscaleHLTeleJet2).first);
188 arizzi 1.11 // std::cout << " jet pt " << pt[j] << " eta " << eta[j] << " eff1 " << thisJetEffs[0] << " eff2 " << thisJetEffs[1] << std::endl;
189 arizzi 1.5 allJetsWithEffs.push_back(thisJetEffs);
190 arizzi 1.11
191     }
192     float res = combiner2Thr.weight<Trigger1High2Loose>(allJetsWithEffs);
193     // std::cout << "Result is " << res << std::endl;
194     return res;
195     // return combiner2Thr.weight<Trigger1High2Loose>(allJetsWithEffs);
196    
197     }
198     /*
199     TF1 fpt("f","1-exp(-0.157*(x-19.3))", 0., 9999999.);
200    
201     MET80:
202     TF1 fmet80("f","1/ (1 + exp( -0.0709 * (x - 100.7)))", 0., 9999999.);
203    
204     MET100:
205     TF1 fmet100("f","1/ (1 + exp( -0.0679 * (x - 128.8)))", 0., 9999999.);
206     */
207    
208     //LP curve used for MET
209     double scaleMetHLT( double met){
210     return 1. / (1. + ( exp( 0.059486 * ( 123.27 - met ))));
211     }
212    
213     //MET80 component of the factorized JET+MET trigger
214     double scaleMET80(double et)
215     {
216     return 1. / (1. + exp( -0.0709 * (et - 100.7)));
217     }
218    
219     //MET100 component
220     double scaleMET100(double et)
221     {
222     return 1. / (1. + exp( -0.0679 * (et - 128.8)));
223     }
224    
225     //Single jet20 efficiency for MET+2CJet20
226     double jet20efficiency( double pt)
227     {
228     if(pt < 10 ) return 0;
229     return 1. - exp(-0.157*(pt-19.3));
230     }
231    
232     //combined 2 jets efficiency out of N jets, using jet20 efficiency curve
233     double scale2CentralJet( std::vector<float> pt, std::vector<float> eta)
234     {
235    
236     std::vector< std::vector<float> > allJetsWithEffs;
237     for(unsigned int j=0; j< pt.size(); j++)
238     {
239     if(fabs(eta[j]) < 2.5)
240     {
241     std::vector<float> thisJetEffs;
242     thisJetEffs.push_back(jet20efficiency(pt[j]));
243     allJetsWithEffs.push_back(thisJetEffs);
244     }
245    
246 arizzi 1.5 }
247    
248 arizzi 1.11 return combiner1Thr.weight<Trigger2SingleThr>(allJetsWithEffs);
249     }
250    
251     //New MET 150
252     double scaleMET150(double et)
253     {
254     return 1./ (1. + exp( -0.129226 * (et - 156.699)));
255 arizzi 1.4 }
256 arizzi 1.3
257 arizzi 1.6 float scaleMuOr30IsoHLT(float pt1, float eta1)
258     {
259     return efficiencyFromPtEta(pt1,eta1,tscaleHLTmuOr30).first;
260     }
261 arizzi 1.3
262    
263 arizzi 1.14 //For 2012A HLT_DiCentralPFJet30_PFMHT80, valid for pfMET > 110 GeV:
264    
265     double scaleDiJet30MHT80_2012A(double x)
266     {
267     if(x<110) return 0;
268     return (1e0 - exp(-0.0161*(x)));
269     }
270     //For 2012B HLT_DiCentralJetSumpT100_dPhi05_DiCentralPFJet60_25_PFMET100_HBHENoiseCleaned, valid for pfMET > 120 GeV:
271     double scaleSumpT100MET100_2012B(double x)
272     {
273     if(x<120) return 0;
274     return (1e0 - exp(-0.0130*(x)));
275     }
276     //For 2012A+B HLT_PFMET150, valid for pfMET > 150 GeV:
277     double scalePFMET150_2012AB(double x)
278     {
279     if(x<150) return 0;
280     return (1e0 - exp(-0.0671*(x-142.7)));
281     }
282    
283 arizzi 1.12 float doubleEle2012A( float pt1, float eta1, float pt2, float eta2)
284     {
285 arizzi 1.15 // std::cout << "di ele" << std::endl;
286 arizzi 1.12 float eff1_17 = efficiencyFromPtEta(pt1,eta1,tscaleDoubleEle2012A_leg17).first;
287     float eff2_17 = efficiencyFromPtEta(pt2,eta2,tscaleDoubleEle2012A_leg17).first;
288     float eff1_8 = efficiencyFromPtEta(pt1,eta1,tscaleDoubleEle2012A_leg8).first;
289     float eff2_8 = efficiencyFromPtEta(pt2,eta2,tscaleDoubleEle2012A_leg8).first;
290 arizzi 1.15 // std::cout << tscaleDoubleEle2012A_dZ << std::endl;
291 arizzi 1.13 float eff_dz = efficiencyFromPtEta(eta1,eta2,tscaleDoubleEle2012A_dZ).first; // despite the name pt,eta is actually eta1,eta2
292 arizzi 1.12
293 arizzi 1.13 return (eff1_17 * eff2_8 + eff2_17 * eff1_8 - eff1_17*eff2_17)*eff_dz;
294 arizzi 1.12
295     }
296     float doubleMuon2012A( float pt1, float eta1, float pt2, float eta2)
297     {
298 arizzi 1.15 // std::cout << "di mu" << std::endl;
299 arizzi 1.12 float eff1_17 = efficiencyFromPtEta(pt1,eta1,tscaleDoubleMuon2012A_leg17).first;
300     float eff2_17 = efficiencyFromPtEta(pt2,eta2,tscaleDoubleMuon2012A_leg17).first;
301     float eff1_8 = efficiencyFromPtEta(pt1,eta1,tscaleDoubleMuon2012A_leg8).first;
302     float eff2_8 = efficiencyFromPtEta(pt2,eta2,tscaleDoubleMuon2012A_leg8).first;
303 arizzi 1.15 // std::cout << tscaleDoubleMuon2012A_dZ << std::endl;
304 arizzi 1.13 float eff_dz = efficiencyFromPtEta(eta1,eta2,tscaleDoubleMuon2012A_dZ).first; // despite the name pt,eta is actually eta1,eta2
305 arizzi 1.12
306 arizzi 1.13 return (eff1_17 * eff2_8 + eff2_17 * eff1_8 - eff1_17*eff2_17)*eff_dz;
307 arizzi 1.12
308     }
309    
310    
311    
312     float muPlusWCandPt2012A_legW( float pt, float eta){ return efficiencyFromPtEta(pt,eta,tscaleMuPlusWCandPt2012A_legW).first;}
313     float muPlusWCandPt2012A_legMu( float pt, float eta){ return efficiencyFromPtEta(pt,eta,tscaleMuPlusWCandPt2012A_legMu).first;}
314     float singleEle2012Awp80( float pt, float eta){ return efficiencyFromPtEta(pt,eta,tscaleSingleEle2012Awp80).first;}
315     float singleEle2012Awp95( float pt, float eta){ return efficiencyFromPtEta(pt,eta,tscaleSingleEle2012Awp95).first;}
316     float singleMuon2012A( float pt, float eta){ return efficiencyFromPtEta(pt,eta,tscaleSingleMuon2012A).first;}
317    
318    
319 arizzi 1.1 private:
320 arizzi 1.4 TTree * tscaleHLTele1;
321     TTree * tscaleHLTele2;
322     TTree * tscaleHLTeleJet1;
323     TTree * tscaleHLTeleJet2;
324 arizzi 1.5 TTree * tscaleID80Ele;
325     TTree * tscaleID95Ele;
326     TTree * tscaleRecoEle;
327 arizzi 1.6 TTree * tscaleHLTmuOr30;
328    
329 arizzi 1.12 TTree * tscaleSingleEle2012Awp95;
330     TTree * tscaleSingleEle2012Awp80;
331     TTree * tscaleSingleMuon2012A;
332     TTree * tscaleDoubleEle2012A_leg8;
333     TTree * tscaleDoubleEle2012A_leg17;
334     TTree * tscaleDoubleMuon2012A_leg8;
335     TTree * tscaleDoubleMuon2012A_leg17;
336     TTree * tscaleMuPlusWCandPt2012A_legMu;
337     TTree * tscaleMuPlusWCandPt2012A_legW;
338 arizzi 1.13 TTree * tscaleDoubleEle2012A_dZ;
339     TTree * tscaleDoubleMuon2012A_dZ;
340 arizzi 1.5 // TTree * tscalePFMHTele;
341     TTree * tscaleSingleEleMay;
342     TTree * tscaleSingleEleV4;
343 arizzi 1.4
344 arizzi 1.9 TTree * tscaleHLTele1Aug;
345     TTree * tscaleHLTele2Aug;
346    
347 arizzi 1.1 TTree * tscaleHLTmu;
348     TTree * tscaleIDmu;
349 arizzi 1.4 MultiThresholdEfficiency combiner2Thr;
350 arizzi 1.11 MultiThresholdEfficiency combiner1Thr;
351 arizzi 1.1 };
352    
353     #endif