| 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.4 |
#include "VHbbAnalysis/VHbbDataFormats/interface/TriggerZnunuCurve.h"
|
| 7 |
|
|
#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 |
|
|
|
| 14 |
|
|
class TriggerWeight
|
| 15 |
|
|
{
|
| 16 |
|
|
public:
|
| 17 |
arizzi |
1.5 |
TriggerWeight(const edm::ParameterSet& ana) : combiner2Thr(2)
|
| 18 |
arizzi |
1.1 |
{
|
| 19 |
arizzi |
1.4 |
tscaleHLTmu=openFile(ana,"hltMuFileName");
|
| 20 |
|
|
tscaleIDmu=openFile(ana,"idMuFileName");
|
| 21 |
|
|
tscaleHLTele1=openFile(ana,"hltEle1FileName");
|
| 22 |
|
|
tscaleHLTele2=openFile(ana,"hltEle2FileName");
|
| 23 |
arizzi |
1.5 |
tscaleID80Ele=openFile(ana,"idEle80FileName");
|
| 24 |
|
|
tscaleID95Ele=openFile(ana,"idEle95FileName");
|
| 25 |
arizzi |
1.4 |
tscaleHLTeleJet1=openFile(ana,"hltJetEle1FileName");
|
| 26 |
|
|
tscaleHLTeleJet2=openFile(ana,"hltJetEle2FileName");
|
| 27 |
arizzi |
1.5 |
tscaleRecoEle=openFile(ana,"recoEleFileName");
|
| 28 |
|
|
// tscalePFMHTele=openFile(ana,"hltPFMHTEleFileName");
|
| 29 |
|
|
tscaleSingleEleMay=openFile(ana,"hltSingleEleMayFileName");
|
| 30 |
|
|
tscaleSingleEleV4=openFile(ana,"hltSingleEleV4FileName");
|
| 31 |
arizzi |
1.6 |
tscaleHLTmuOr30=openFile(ana,"hltMuOr30FileName");
|
| 32 |
arizzi |
1.4 |
|
| 33 |
arizzi |
1.1 |
if(tscaleHLTmu == 0 || tscaleIDmu == 0)
|
| 34 |
|
|
{
|
| 35 |
|
|
std::cout << "ERROR: cannot load Muon Trigger efficiencies" << std::endl;
|
| 36 |
|
|
}
|
| 37 |
arizzi |
1.4 |
|
| 38 |
|
|
|
| 39 |
arizzi |
1.1 |
|
| 40 |
|
|
}
|
| 41 |
arizzi |
1.4 |
|
| 42 |
|
|
TTree * openFile(const edm::ParameterSet& ana, const char * name)
|
| 43 |
|
|
{
|
| 44 |
arizzi |
1.5 |
TFile *hltMuFile = new TFile (ana.getParameter<std::string> (name).c_str(),"read");
|
| 45 |
arizzi |
1.4 |
if(hltMuFile) return (TTree*) hltMuFile->Get("tree");
|
| 46 |
|
|
else return 0;
|
| 47 |
|
|
}
|
| 48 |
|
|
|
| 49 |
|
|
std::pair<float,float> efficiencyFromPtEta(float pt1, float eta1, TTree *t)
|
| 50 |
|
|
{
|
| 51 |
|
|
float s1 = 1.,err=1.;
|
| 52 |
|
|
std::pair<float,float> r(s1,err);
|
| 53 |
|
|
if(!t) return r;
|
| 54 |
|
|
float ptMin,ptMax,etaMin,etaMax,scale,error;
|
| 55 |
|
|
int count = 0;
|
| 56 |
|
|
t->SetBranchAddress("ptMin",&ptMin);
|
| 57 |
|
|
t->SetBranchAddress("ptMax",&ptMax);
|
| 58 |
|
|
t->SetBranchAddress("etaMin",&etaMin);
|
| 59 |
|
|
t->SetBranchAddress("etaMax",&etaMax);
|
| 60 |
|
|
t->SetBranchAddress("scale",&scale);
|
| 61 |
|
|
t->SetBranchAddress("error",&error);
|
| 62 |
arizzi |
1.7 |
float lastPtBin = 200;
|
| 63 |
|
|
/* for(int jentry = 0; jentry < t->GetEntries(); jentry++)
|
| 64 |
|
|
{
|
| 65 |
|
|
t->GetEntry(jentry);
|
| 66 |
|
|
if(ptMax >= lastPtBin) lastPtBin =ptMax;
|
| 67 |
|
|
}*/
|
| 68 |
arizzi |
1.4 |
for(int jentry = 0; jentry < t->GetEntries(); jentry++)
|
| 69 |
|
|
{
|
| 70 |
|
|
t->GetEntry(jentry);
|
| 71 |
arizzi |
1.7 |
if(ptMax==lastPtBin) ptMax=1e99;
|
| 72 |
|
|
|
| 73 |
arizzi |
1.4 |
if((pt1 > ptMin) && (pt1 < ptMax) && (eta1 > etaMin) && (eta1 < etaMax))
|
| 74 |
|
|
{
|
| 75 |
|
|
s1 = scale;
|
| 76 |
|
|
err=error;
|
| 77 |
|
|
count++;
|
| 78 |
|
|
}
|
| 79 |
|
|
}
|
| 80 |
|
|
|
| 81 |
|
|
if(count == 0 || s1 == 0)
|
| 82 |
|
|
{
|
| 83 |
|
|
return r;
|
| 84 |
|
|
}
|
| 85 |
|
|
|
| 86 |
|
|
r.first=s1;
|
| 87 |
|
|
r.second = err;
|
| 88 |
|
|
return (r);
|
| 89 |
|
|
}
|
| 90 |
arizzi |
1.1 |
|
| 91 |
|
|
float scaleMuIsoHLT(float pt1, float eta1)
|
| 92 |
|
|
{
|
| 93 |
arizzi |
1.4 |
return efficiencyFromPtEta(pt1,eta1,tscaleHLTmu).first;
|
| 94 |
arizzi |
1.1 |
}
|
| 95 |
|
|
|
| 96 |
|
|
|
| 97 |
|
|
|
| 98 |
|
|
float scaleMuID(float pt1, float eta1)
|
| 99 |
|
|
{
|
| 100 |
arizzi |
1.4 |
return efficiencyFromPtEta(pt1,eta1,tscaleIDmu).first;
|
| 101 |
arizzi |
1.1 |
}
|
| 102 |
|
|
|
| 103 |
arizzi |
1.3 |
double scaleMetHLT( double met){
|
| 104 |
|
|
|
| 105 |
|
|
float s1 = 1;
|
| 106 |
|
|
TF1 * f = new TF1 ("f",TriggerZnunuCurve::trigMet, 0,99999, 0, "triggerZnunuCurve" );
|
| 107 |
|
|
|
| 108 |
|
|
s1 = f->Eval(met);
|
| 109 |
|
|
|
| 110 |
|
|
return (s1);
|
| 111 |
|
|
|
| 112 |
|
|
}
|
| 113 |
|
|
|
| 114 |
|
|
|
| 115 |
arizzi |
1.5 |
double scaleDoubleEle17Ele8( std::vector<float> pt, std::vector<float> eta )
|
| 116 |
|
|
{
|
| 117 |
|
|
std::vector< std::vector<float> > allEleWithEffs;
|
| 118 |
|
|
for(unsigned int j=0; j< pt.size(); j++)
|
| 119 |
|
|
{
|
| 120 |
|
|
std::vector<float> thisEleEffs;
|
| 121 |
|
|
thisEleEffs.push_back(efficiencyFromPtEta(pt[j],eta[j],tscaleHLTele1).first);
|
| 122 |
|
|
thisEleEffs.push_back(efficiencyFromPtEta(pt[j],eta[j],tscaleHLTele2).first);
|
| 123 |
|
|
allEleWithEffs.push_back(thisEleEffs);
|
| 124 |
|
|
}
|
| 125 |
|
|
|
| 126 |
|
|
return combiner2Thr.weight<Trigger1High2Loose>(allEleWithEffs);
|
| 127 |
|
|
|
| 128 |
|
|
}
|
| 129 |
|
|
|
| 130 |
|
|
double scaleSingleEleMay( float pt, float eta){ return efficiencyFromPtEta(pt,eta,tscaleSingleEleMay).first;}
|
| 131 |
|
|
double scaleSingleEleV4( float pt, float eta){ return efficiencyFromPtEta(pt,eta,tscaleSingleEleV4).first; }
|
| 132 |
|
|
double scaleID80Ele( float pt, float eta) { return efficiencyFromPtEta(pt,eta,tscaleID80Ele).first; }
|
| 133 |
|
|
double scaleID95Ele( float pt, float eta) { return efficiencyFromPtEta(pt,eta,tscaleID95Ele).first; }
|
| 134 |
|
|
double scaleRecoEle( float pt, float eta){ return efficiencyFromPtEta(pt,eta,tscaleRecoEle).first; }
|
| 135 |
|
|
double scalePFMHTEle( float MetPFPt){
|
| 136 |
|
|
double weightPFMHTrigger=0.;
|
| 137 |
|
|
|
| 138 |
|
|
//FIXME: read from file
|
| 139 |
dlopes |
1.8 |
if(MetPFPt>0. && MetPFPt<5.) weightPFMHTrigger=0.305;
|
| 140 |
|
|
if(MetPFPt>5. && MetPFPt<10.) weightPFMHTrigger=0.351;
|
| 141 |
|
|
if(MetPFPt>10. && MetPFPt<15.) weightPFMHTrigger=0.461;
|
| 142 |
|
|
if(MetPFPt>15. && MetPFPt<20.) weightPFMHTrigger=0.572;
|
| 143 |
|
|
if(MetPFPt>20. && MetPFPt<25.) weightPFMHTrigger=0.713;
|
| 144 |
|
|
if(MetPFPt>25. && MetPFPt<30.) weightPFMHTrigger=0.844;
|
| 145 |
|
|
if(MetPFPt>30. && MetPFPt<35.) weightPFMHTrigger=0.914;
|
| 146 |
|
|
if(MetPFPt>35. && MetPFPt<40.) weightPFMHTrigger=0.939;
|
| 147 |
|
|
if(MetPFPt>40. && MetPFPt<45.) weightPFMHTrigger=0.981;
|
| 148 |
|
|
if(MetPFPt>45. && MetPFPt<50.) weightPFMHTrigger=0.982;
|
| 149 |
|
|
if(MetPFPt>50. && MetPFPt<60.) weightPFMHTrigger=0.993;
|
| 150 |
|
|
if(MetPFPt>60. && MetPFPt<70.) weightPFMHTrigger=0.995;
|
| 151 |
|
|
if(MetPFPt>70. && MetPFPt<100.) weightPFMHTrigger=0.995;
|
| 152 |
|
|
if(MetPFPt>100.) weightPFMHTrigger=1.;
|
| 153 |
arizzi |
1.5 |
return weightPFMHTrigger;
|
| 154 |
|
|
}
|
| 155 |
|
|
|
| 156 |
|
|
double scaleJet30Jet25( std::vector<float> pt, std::vector<float> eta)
|
| 157 |
arizzi |
1.4 |
{
|
| 158 |
arizzi |
1.5 |
|
| 159 |
|
|
std::vector< std::vector<float> > allJetsWithEffs;
|
| 160 |
|
|
for(unsigned int j=0; j< pt.size(); j++)
|
| 161 |
|
|
{
|
| 162 |
|
|
std::vector<float> thisJetEffs;
|
| 163 |
|
|
thisJetEffs.push_back(efficiencyFromPtEta(pt[j],eta[j],tscaleHLTeleJet1).first);
|
| 164 |
|
|
thisJetEffs.push_back(efficiencyFromPtEta(pt[j],eta[j],tscaleHLTeleJet2).first);
|
| 165 |
|
|
allJetsWithEffs.push_back(thisJetEffs);
|
| 166 |
|
|
}
|
| 167 |
|
|
|
| 168 |
|
|
return combiner2Thr.weight<Trigger1High2Loose>(allJetsWithEffs);
|
| 169 |
arizzi |
1.4 |
}
|
| 170 |
arizzi |
1.3 |
|
| 171 |
arizzi |
1.6 |
float scaleMuOr30IsoHLT(float pt1, float eta1)
|
| 172 |
|
|
{
|
| 173 |
|
|
return efficiencyFromPtEta(pt1,eta1,tscaleHLTmuOr30).first;
|
| 174 |
|
|
}
|
| 175 |
arizzi |
1.3 |
|
| 176 |
|
|
|
| 177 |
arizzi |
1.1 |
private:
|
| 178 |
arizzi |
1.4 |
TTree * tscaleHLTele1;
|
| 179 |
|
|
TTree * tscaleHLTele2;
|
| 180 |
|
|
TTree * tscaleHLTeleJet1;
|
| 181 |
|
|
TTree * tscaleHLTeleJet2;
|
| 182 |
arizzi |
1.5 |
TTree * tscaleID80Ele;
|
| 183 |
|
|
TTree * tscaleID95Ele;
|
| 184 |
|
|
TTree * tscaleRecoEle;
|
| 185 |
arizzi |
1.6 |
TTree * tscaleHLTmuOr30;
|
| 186 |
|
|
|
| 187 |
arizzi |
1.5 |
// TTree * tscalePFMHTele;
|
| 188 |
|
|
TTree * tscaleSingleEleMay;
|
| 189 |
|
|
TTree * tscaleSingleEleV4;
|
| 190 |
arizzi |
1.4 |
|
| 191 |
arizzi |
1.1 |
TTree * tscaleHLTmu;
|
| 192 |
|
|
TTree * tscaleIDmu;
|
| 193 |
arizzi |
1.4 |
MultiThresholdEfficiency combiner2Thr;
|
| 194 |
arizzi |
1.1 |
};
|
| 195 |
|
|
|
| 196 |
|
|
#endif
|
