-
Notifications
You must be signed in to change notification settings - Fork 4.2k
/
bTagging_cff.py
135 lines (134 loc) · 12.1 KB
/
bTagging_cff.py
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
## list of all available btagInfos
supportedBtagInfos = [
'None'
# legacy framework (supported with RECO/AOD but not MiniAOD)
, 'impactParameterTagInfos'
, 'secondaryVertexTagInfos'
, 'secondaryVertexNegativeTagInfos'
, 'softMuonTagInfos'
, 'inclusiveSecondaryVertexFinderTagInfos'
, 'inclusiveSecondaryVertexFinderNegativeTagInfos'
, 'inclusiveSecondaryVertexFinderFilteredTagInfos'
, 'inclusiveSecondaryVertexFinderFilteredNegativeTagInfos'
# new candidate-based framework (supported with RECO/AOD/MiniAOD)
, 'pfImpactParameterTagInfos'
, 'pfImpactParameterTagInfosAK8'
, 'pfImpactParameterTagInfosCA15'
, 'pfSecondaryVertexTagInfos'
, 'pfSecondaryVertexNegativeTagInfos'
, 'pfInclusiveSecondaryVertexFinderTagInfos'
, 'pfInclusiveSecondaryVertexFinderTagInfosAK8'
, 'pfInclusiveSecondaryVertexFinderTagInfosCA15'
, 'pfInclusiveSecondaryVertexFinderNegativeTagInfos'
, 'softPFMuonsTagInfos'
, 'softPFMuonsTagInfosAK8'
, 'softPFMuonsTagInfosCA15'
, 'softPFElectronsTagInfos'
, 'softPFElectronsTagInfosAK8'
, 'softPFElectronsTagInfosCA15'
#, 'pfInclusiveSecondaryVertexFinderCtagLTagInfos'
#, 'pfInclusiveSecondaryVertexFinderCtagLNegativeTagInfos'
# TopTagInfos (unrelated to b tagging)
, 'caTopTagInfos'
]
# extend for "internal use" in PAT/MINIAOD (renaming)
supportedBtagInfos.append( 'caTopTagInfosPAT' )
## dictionary with all available btag discriminators and the btagInfos that they require
supportedBtagDiscr = {
'None' : []
# legacy framework (supported with RECO/AOD but not MiniAOD)
, 'jetBProbabilityBJetTags' : ['impactParameterTagInfos']
, 'jetProbabilityBJetTags' : ['impactParameterTagInfos']
, 'positiveOnlyJetBProbabilityBJetTags' : ['impactParameterTagInfos']
, 'positiveOnlyJetProbabilityBJetTags' : ['impactParameterTagInfos']
, 'negativeOnlyJetBProbabilityBJetTags' : ['impactParameterTagInfos']
, 'negativeOnlyJetProbabilityBJetTags' : ['impactParameterTagInfos']
, 'trackCountingHighPurBJetTags' : ['impactParameterTagInfos']
, 'trackCountingHighEffBJetTags' : ['impactParameterTagInfos']
, 'negativeTrackCountingHighPurBJetTags' : ['impactParameterTagInfos']
, 'negativeTrackCountingHighEffBJetTags' : ['impactParameterTagInfos']
, 'simpleSecondaryVertexHighEffBJetTags' : ['secondaryVertexTagInfos']
, 'simpleSecondaryVertexHighPurBJetTags' : ['secondaryVertexTagInfos']
, 'negativeSimpleSecondaryVertexHighEffBJetTags' : ['secondaryVertexNegativeTagInfos']
, 'negativeSimpleSecondaryVertexHighPurBJetTags' : ['secondaryVertexNegativeTagInfos']
, 'combinedSecondaryVertexBJetTags' : ['impactParameterTagInfos', 'secondaryVertexTagInfos']
, 'positiveCombinedSecondaryVertexBJetTags' : ['impactParameterTagInfos', 'secondaryVertexTagInfos']
, 'negativeCombinedSecondaryVertexBJetTags' : ['impactParameterTagInfos', 'secondaryVertexNegativeTagInfos']
, 'combinedSecondaryVertexV2BJetTags' : ['impactParameterTagInfos', 'secondaryVertexTagInfos']
, 'positiveCombinedSecondaryVertexV2BJetTags' : ['impactParameterTagInfos', 'secondaryVertexTagInfos']
, 'negativeCombinedSecondaryVertexV2BJetTags' : ['impactParameterTagInfos', 'secondaryVertexNegativeTagInfos']
, 'simpleInclusiveSecondaryVertexHighEffBJetTags' : ['inclusiveSecondaryVertexFinderFilteredTagInfos']
, 'simpleInclusiveSecondaryVertexHighPurBJetTags' : ['inclusiveSecondaryVertexFinderFilteredTagInfos']
, 'negativeSimpleInclusiveSecondaryVertexHighEffBJetTags' : ['inclusiveSecondaryVertexFinderFilteredNegativeTagInfos']
, 'negativeSimpleInclusiveSecondaryVertexHighPurBJetTags' : ['inclusiveSecondaryVertexFinderFilteredNegativeTagInfos']
, 'doubleSecondaryVertexHighEffBJetTags' : ['inclusiveSecondaryVertexFinderFilteredTagInfos']
, 'combinedInclusiveSecondaryVertexBJetTags' : ['impactParameterTagInfos', 'inclusiveSecondaryVertexFinderTagInfos']
, 'positiveCombinedInclusiveSecondaryVertexBJetTags' : ['impactParameterTagInfos', 'inclusiveSecondaryVertexFinderTagInfos']
, 'negativeCombinedInclusiveSecondaryVertexBJetTags' : ['impactParameterTagInfos', 'inclusiveSecondaryVertexFinderNegativeTagInfos']
, 'combinedInclusiveSecondaryVertexV2BJetTags' : ['impactParameterTagInfos', 'inclusiveSecondaryVertexFinderTagInfos']
, 'positiveCombinedInclusiveSecondaryVertexV2BJetTags' : ['impactParameterTagInfos', 'inclusiveSecondaryVertexFinderTagInfos']
, 'negativeCombinedInclusiveSecondaryVertexV2BJetTags' : ['impactParameterTagInfos', 'inclusiveSecondaryVertexFinderNegativeTagInfos']
, 'combinedMVABJetTags' : ['impactParameterTagInfos', 'inclusiveSecondaryVertexFinderTagInfos', 'softPFMuonsTagInfos', 'softPFElectronsTagInfos']
, 'positiveCombinedMVABJetTags' : ['impactParameterTagInfos', 'inclusiveSecondaryVertexFinderTagInfos', 'softPFMuonsTagInfos', 'softPFElectronsTagInfos']
, 'negativeCombinedMVABJetTags' : ['impactParameterTagInfos', 'inclusiveSecondaryVertexFinderNegativeTagInfos', 'softPFMuonsTagInfos', 'softPFElectronsTagInfos']
# new candidate-based framework (supported with RECO/AOD/MiniAOD)
, 'pfJetBProbabilityBJetTags' : ['pfImpactParameterTagInfos']
, 'pfJetProbabilityBJetTags' : ['pfImpactParameterTagInfos']
, 'pfPositiveOnlyJetBProbabilityBJetTags' : ['pfImpactParameterTagInfos']
, 'pfPositiveOnlyJetProbabilityBJetTags' : ['pfImpactParameterTagInfos']
, 'pfNegativeOnlyJetBProbabilityBJetTags' : ['pfImpactParameterTagInfos']
, 'pfNegativeOnlyJetProbabilityBJetTags' : ['pfImpactParameterTagInfos']
, 'pfTrackCountingHighPurBJetTags' : ['pfImpactParameterTagInfos']
, 'pfTrackCountingHighEffBJetTags' : ['pfImpactParameterTagInfos']
, 'pfNegativeTrackCountingHighPurBJetTags' : ['pfImpactParameterTagInfos']
, 'pfNegativeTrackCountingHighEffBJetTags' : ['pfImpactParameterTagInfos']
, 'pfSimpleSecondaryVertexHighEffBJetTags' : ['pfSecondaryVertexTagInfos']
, 'pfSimpleSecondaryVertexHighPurBJetTags' : ['pfSecondaryVertexTagInfos']
, 'pfNegativeSimpleSecondaryVertexHighEffBJetTags' : ['pfSecondaryVertexNegativeTagInfos']
, 'pfNegativeSimpleSecondaryVertexHighPurBJetTags' : ['pfSecondaryVertexNegativeTagInfos']
, 'pfCombinedSecondaryVertexBJetTags' : ['pfImpactParameterTagInfos', 'pfSecondaryVertexTagInfos']
, 'pfPositiveCombinedSecondaryVertexBJetTags' : ['pfImpactParameterTagInfos', 'pfSecondaryVertexTagInfos']
, 'pfNegativeCombinedSecondaryVertexBJetTags' : ['pfImpactParameterTagInfos', 'pfSecondaryVertexNegativeTagInfos']
, 'pfCombinedSecondaryVertexV2BJetTags' : ['pfImpactParameterTagInfos', 'pfSecondaryVertexTagInfos']
, 'pfPositiveCombinedSecondaryVertexV2BJetTags' : ['pfImpactParameterTagInfos', 'pfSecondaryVertexTagInfos']
, 'pfNegativeCombinedSecondaryVertexV2BJetTags' : ['pfImpactParameterTagInfos', 'pfSecondaryVertexNegativeTagInfos']
, 'pfCombinedInclusiveSecondaryVertexBJetTags' : ['pfImpactParameterTagInfos', 'pfInclusiveSecondaryVertexFinderTagInfos']
, 'pfPositiveCombinedInclusiveSecondaryVertexBJetTags' : ['pfImpactParameterTagInfos', 'pfInclusiveSecondaryVertexFinderTagInfos']
, 'pfNegativeCombinedInclusiveSecondaryVertexBJetTags' : ['pfImpactParameterTagInfos', 'pfInclusiveSecondaryVertexFinderNegativeTagInfos']
, 'pfCombinedInclusiveSecondaryVertexV2BJetTags' : ['pfImpactParameterTagInfos', 'pfInclusiveSecondaryVertexFinderTagInfos']
, 'pfPositiveCombinedInclusiveSecondaryVertexV2BJetTags' : ['pfImpactParameterTagInfos', 'pfInclusiveSecondaryVertexFinderTagInfos']
, 'pfNegativeCombinedInclusiveSecondaryVertexV2BJetTags' : ['pfImpactParameterTagInfos', 'pfInclusiveSecondaryVertexFinderNegativeTagInfos']
, 'softPFMuonBJetTags' : ['softPFMuonsTagInfos']
, 'softPFMuonByPtBJetTags' : ['softPFMuonsTagInfos']
, 'softPFMuonByIP3dBJetTags' : ['softPFMuonsTagInfos']
, 'softPFMuonByIP2dBJetTags' : ['softPFMuonsTagInfos']
, 'positiveSoftPFMuonBJetTags' : ['softPFMuonsTagInfos']
, 'positiveSoftPFMuonByPtBJetTags' : ['softPFMuonsTagInfos']
, 'positiveSoftPFMuonByIP3dBJetTags' : ['softPFMuonsTagInfos']
, 'positiveSoftPFMuonByIP2dBJetTags' : ['softPFMuonsTagInfos']
, 'negativeSoftPFMuonBJetTags' : ['softPFMuonsTagInfos']
, 'negativeSoftPFMuonByPtBJetTags' : ['softPFMuonsTagInfos']
, 'negativeSoftPFMuonByIP3dBJetTags' : ['softPFMuonsTagInfos']
, 'negativeSoftPFMuonByIP2dBJetTags' : ['softPFMuonsTagInfos']
, 'softPFElectronBJetTags' : ['softPFElectronsTagInfos']
, 'softPFElectronByPtBJetTags' : ['softPFElectronsTagInfos']
, 'softPFElectronByIP3dBJetTags' : ['softPFElectronsTagInfos']
, 'softPFElectronByIP2dBJetTags' : ['softPFElectronsTagInfos']
, 'positiveSoftPFElectronBJetTags' : ['softPFElectronsTagInfos']
, 'positiveSoftPFElectronByPtBJetTags' : ['softPFElectronsTagInfos']
, 'positiveSoftPFElectronByIP3dBJetTags' : ['softPFElectronsTagInfos']
, 'positiveSoftPFElectronByIP2dBJetTags' : ['softPFElectronsTagInfos']
, 'negativeSoftPFElectronBJetTags' : ['softPFElectronsTagInfos']
, 'negativeSoftPFElectronByPtBJetTags' : ['softPFElectronsTagInfos']
, 'negativeSoftPFElectronByIP3dBJetTags' : ['softPFElectronsTagInfos']
, 'negativeSoftPFElectronByIP2dBJetTags' : ['softPFElectronsTagInfos']
, 'pfCombinedMVABJetTags' : ['pfImpactParameterTagInfos', 'pfInclusiveSecondaryVertexFinderTagInfos', 'softPFMuonsTagInfos', 'softPFElectronsTagInfos']
, 'pfPositiveCombinedMVABJetTags' : ['pfImpactParameterTagInfos', 'pfInclusiveSecondaryVertexFinderTagInfos', 'softPFMuonsTagInfos', 'softPFElectronsTagInfos']
, 'pfNegativeCombinedMVABJetTags' : ['pfImpactParameterTagInfos', 'pfInclusiveSecondaryVertexFinderNegativeTagInfos', 'softPFMuonsTagInfos', 'softPFElectronsTagInfos']
, 'pfCombinedSecondaryVertexSoftLeptonBJetTags' : ['pfImpactParameterTagInfos', 'pfInclusiveSecondaryVertexFinderTagInfos', 'softPFMuonsTagInfos', 'softPFElectronsTagInfos']
, 'pfNegativeCombinedSecondaryVertexSoftLeptonBJetTags' : ['pfImpactParameterTagInfos', 'pfInclusiveSecondaryVertexFinderNegativeTagInfos', 'softPFMuonsTagInfos', 'softPFElectronsTagInfos']
#, 'pfCombinedSecondaryVertexSoftLeptonCtagLJetTags' : ['pfImpactParameterTagInfos', 'pfInclusiveSecondaryVertexFinderCtagLTagInfos', 'softPFMuonsTagInfos', 'softPFElectronsTagInfos']
#, 'pfNegativeCombinedSecondaryVertexSoftLeptonCtagLJetTags' : ['pfImpactParameterTagInfos', 'pfInclusiveSecondaryVertexFinderCtagLNegativeTagInfos', 'softPFMuonsTagInfos', 'softPFElectronsTagInfos']
, 'pfBoostedDoubleSecondaryVertexAK8BJetTags' : ['pfImpactParameterTagInfosAK8', 'pfInclusiveSecondaryVertexFinderTagInfosAK8', 'softPFMuonsTagInfosAK8', 'softPFElectronsTagInfosAK8']
, 'pfBoostedDoubleSecondaryVertexCA15BJetTags' : ['pfImpactParameterTagInfosCA15', 'pfInclusiveSecondaryVertexFinderTagInfosCA15', 'softPFMuonsTagInfosCA15', 'softPFElectronsTagInfosCA15']
}