-
Notifications
You must be signed in to change notification settings - Fork 0
/
BTagUtils.h
144 lines (100 loc) · 5.28 KB
/
BTagUtils.h
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
#ifndef __BTAGUTILS__
#define __BTAGUTILS__
#include "TH1F.h"
#include "TH2F.h"
#include "TF1.h"
#include "TFile.h"
#include "TLorentzVector.h"
#include <iostream>
#include <iomanip>
#include <string>
#include <sstream>
#include <ctime>
#include <map>
#include <algorithm>
#include <math.h>
#include <vector>
class JetFlavourReader
{
public:
JetFlavourReader(const std::string name_JetFlavourFile);
virtual ~JetFlavourReader();
int getJetFlavour(const int& lumis, const int& event, const TLorentzVector* jetP4 );
protected:
std::string name_JetFlavourFile_;
std::map<int,std::map<int,std::vector<TLorentzVector> > > AODjet_p4_;
std::map<int,std::map<int,std::vector<int> > > AODjet_Flav_;
std::vector<float> dR_;
std::map<float,int> dR_map_;
};
class BtagSFReader
{
public:
BtagSFReader(const std::string name_btagSFFile);
virtual ~BtagSFReader();
float getSF(const TLorentzVector* jetP4,const float& flavour, std::string WP);
float getSFErrorUp(const TLorentzVector* jetP4,const float& flavour, std::string WP);
float getSFErrorDown(const TLorentzVector* jetP4,const float& flavour, std::string WP);
protected:
std::string name_btagSFFile_;
TF1* SFb_CSVL_;
TH1F* h1_SFb_CSVL_;
TF1* SFb_CSVM_;
TH1F* h1_SFb_CSVM_;
TF1* SFb_CSVT_;
TH1F* h1_SFb_CSVT_;
TF1* SFudsg_CSVL_00_05_max_;
TF1* SFudsg_CSVL_00_05_mean_;
TF1* SFudsg_CSVL_00_05_min_;
TF1* SFudsg_CSVL_05_10_max_;
TF1* SFudsg_CSVL_05_10_mean_;
TF1* SFudsg_CSVL_05_10_min_;
TF1* SFudsg_CSVL_10_15_max_;
TF1* SFudsg_CSVL_10_15_mean_;
TF1* SFudsg_CSVL_10_15_min_;
TF1* SFudsg_CSVL_15_24_max_;
TF1* SFudsg_CSVL_15_24_mean_;
TF1* SFudsg_CSVL_15_24_min_;
TF1* SFudsg_CSVM_00_08_max_;
TF1* SFudsg_CSVM_00_08_mean_;
TF1* SFudsg_CSVM_00_08_min_;
TF1* SFudsg_CSVM_08_16_max_;
TF1* SFudsg_CSVM_08_16_mean_;
TF1* SFudsg_CSVM_08_16_min_;
TF1* SFudsg_CSVM_16_24_max_;
TF1* SFudsg_CSVM_16_24_mean_;
TF1* SFudsg_CSVM_16_24_min_;
TF1* SFudsg_CSVT_00_24_max_;
TF1* SFudsg_CSVT_00_24_mean_;
TF1* SFudsg_CSVT_00_24_min_;
};
class BtagEfficiencyReader
{
public:
BtagEfficiencyReader();
void Init(const std::string name_btagEfficienciesFile);
virtual ~BtagEfficiencyReader();
float getBtagEfficiency(const TLorentzVector* jetP4, std::string WP, const int& jet_flavour);
float getBtagEfficiencyError(const TLorentzVector* jetP4, std::string WP, const int& jet_flavour);
protected:
std::string name_btagEfficienciesFile_;
TH2F* h2_BTaggingEff_b_L_;
TH2F* h2_BTaggingEff_b_M_;
TH2F* h2_BTaggingEff_b_T_;
TH2F* h2_BTaggingEff_c_L_;
TH2F* h2_BTaggingEff_c_M_;
TH2F* h2_BTaggingEff_c_T_;
TH2F* h2_BTaggingEff_udsg_L_;
TH2F* h2_BTaggingEff_udsg_M_;
TH2F* h2_BTaggingEff_udsg_T_;
};
// Functions to calculate the final event weight and error
float jetWeight(std::string WP, const float jet_SF, const float jet_eff, const float& jet_csvBtag);
float jetWeight_err(std::string WP, const float jet_SF,const float jet_SF_err, const float jet_eff,const float jet_eff_err, const float& jet_csvBtag);
float eventWeight_2jets(std::string WP, const float& j1_SF, const float& j2_SF, const float& j1_eff, const float& j2_eff, const float& j1_csvBtag, const float& j2_csvBtag);
float eventWeight_error_2jets(std::string WP, const float& j1_SF, const float& j1_SF_error, const float& j2_SF, const float& j2_SF_error, const float& j1_eff, const float& j1_eff_error, const float& j2_eff, const float& j2_eff_error, const float& j1_flavour, const float& j2_flavour, const float& j1_csvBtag, const float& j2_csvBtag);
float eventWeight_3jets(std::string WP, const float& j1_SF, const float& j2_SF, const float& j3_SF, const float& j1_eff, const float& j2_eff, const float& j3_eff, const float& j1_csvBtag, const float& j2_csvBtag, const float& j3_csvBtag);
float eventWeight_error_3jets(std::string WP, const float& j1_SF, const float& j1_SF_error, const float& j2_SF, const float& j2_SF_error, const float& j3_SF, const float& j3_SF_error, const float& j1_eff, const float& j1_eff_error, const float& j2_eff, const float& j2_eff_error, const float& j3_eff, const float& j3_eff_error, const float& j1_flavour, const float& j2_flavour, const float& j3_flavour, const float& j1_csvBtag, const float& j2_csvBtag, const float& j3_csvBtag);
float eventWeight_4jets(std::string WP, const float& j1_SF, const float& j2_SF, const float& j3_SF, const float& j4_SF, const float& j1_eff, const float& j2_eff, const float& j3_eff, const float& j4_eff, const float& j1_csvBtag, const float& j2_csvBtag, const float& j3_csvBtag, const float& j4_csvBtag);
float eventWeight_error_4jets(std::string WP, const float& j1_SF, const float& j1_SF_error, const float& j2_SF, const float& j2_SF_error, const float& j3_SF, const float& j3_SF_error, const float& j4_SF, const float& j4_SF_error, const float& j1_eff, const float& j1_eff_error, const float& j2_eff, const float& j2_eff_error, const float& j3_eff, const float& j3_eff_error, const float& j4_eff, const float& j4_eff_error, const float& j1_flavour, const float& j2_flavour, const float& j3_flavour, const float& j4_flavour, const float& j1_csvBtag, const float& j2_csvBtag, const float& j3_csvBtag, const float& j4_csvBtag);
#endif