ATLAS Offline Software
Loading...
Searching...
No Matches
TauConfigFlags.py
Go to the documentation of this file.
1# Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
2
3import unittest
4from AthenaConfiguration.AthConfigFlags import AthConfigFlags
5import AthenaCommon.SystemOfUnits as Units
6from AthenaConfiguration.Enums import LHCPeriod, ProductionStep
7
8def createTauConfigFlags():
9 tau_cfg = AthConfigFlags()
10
11 tau_cfg.addFlag("Tau.doTauRec", True)
12 tau_cfg.addFlag("Tau.ThinTaus", True)
13 tau_cfg.addFlag("Tau.doEarlyStopping", True)
14
15 # Switches for enabling/disabling some tools
16 tau_cfg.addFlag("Tau.doVertexCorrection", True)
17 tau_cfg.addFlag("Tau.doTJVA", True)
18 tau_cfg.addFlag("Tau.doPanTau", True)
19 tau_cfg.addFlag("Tau.doRNNTrackClass", True)
20 tau_cfg.addFlag("Tau.doTauDiscriminant", True)
21 tau_cfg.addFlag("Tau.associateLRT", False)
22 tau_cfg.addFlag("Tau.isStandalone", False)
23 # Classify Large Radius Tracks in tau track classifier
24 tau_cfg.addFlag("Tau.classifyLRT", False)
25 tau_cfg.addFlag("Tau.classifyLRTWithDedicated", False)
26
27 # Settings common to Run2 and Run3
28 tau_cfg.addFlag("Tau.SeedMinPt", 0.0*Units.GeV)
29 tau_cfg.addFlag("Tau.SeedMaxEta", lambda prevFlags: 2.5 if prevFlags.GeoModel.Run <= LHCPeriod.Run3 else 4.0)
30 tau_cfg.addFlag("Tau.MaxNTracks", -1)
31 tau_cfg.addFlag("Tau.RemoveDupeCoreTracks", True)
32 tau_cfg.addFlag("Tau.useGhostTracks", True)
33 tau_cfg.addFlag("Tau.ghostTrackDR", 0.25)
34 tau_cfg.addFlag("Tau.shotPtCut_1Photon", [430.*Units.MeV, 300.*Units.MeV, 9999999.*Units.MeV, 330.*Units.MeV, 350.*Units.MeV])
35 tau_cfg.addFlag("Tau.shotPtCut_2Photons", [10000.*Units.MeV, 10000.*Units.MeV, 9999999.*Units.MeV, 10000.*Units.MeV, 10000.*Units.MeV])
36 tau_cfg.addFlag("Tau.Pi0ScoreConfig", "TauPi0BDTWeights.root")
37 tau_cfg.addFlag("Tau.pi0EtCuts", [2800.*Units.MeV, 2700.*Units.MeV, 2500.*Units.MeV, 2900.*Units.MeV, 2700.*Units.MeV])
38 tau_cfg.addFlag("Tau.pi0MVACuts_1prong", [0.45, 0.37, 0.39, 0.40, 0.38])
39 tau_cfg.addFlag("Tau.pi0MVACuts_mprong", [0.73, 0.69, 0.58, 0.69, 0.67])
40 tau_cfg.addFlag("Tau.MinNeutralPFOPt", 100.*Units.MeV)
41
42 # Run2 settings and calibration files
43 tau_cfg.addFlag("Tau.tauRecToolsCVMFSPath", "tauRecTools/R22_preprod")
44 tau_cfg.addFlag("Tau.tauRNNTrackClassConfig", lambda prevFlags: "RNNTrackClassifier_2021-07-19_14-25-14_90_25_30.json" if prevFlags.GeoModel.Run <= LHCPeriod.Run3 else "Run4/RNNTrackClassifier_v1.json")
45 tau_cfg.addFlag("Tau.CalibrateLCConfig", "CaloTES_R22_Round2.5.root")
46 tau_cfg.addFlag("Tau.CombinedTESConfig", lambda prevFlags: "CombinedTES_R22_Round2.5_v2.root" if prevFlags.Common.ProductionStep is ProductionStep.Derivation else "CombinedTES_R22_Round2.5.root")
47 tau_cfg.addFlag("Tau.MvaTESConfig0p", "MvaTES_0p_R23.root")
48 tau_cfg.addFlag("Tau.MvaTESConfig", "MvaTES_R23.root")
49 tau_cfg.addFlag("Tau.MinPt0p", 9.25*Units.GeV)
50 tau_cfg.addFlag("Tau.MinPt", 6.75*Units.GeV)
51 tau_cfg.addFlag("Tau.MinPtDAOD", 13*Units.GeV)
52 tau_cfg.addFlag("Tau.MinPtLITE", 20*Units.GeV)
53 tau_cfg.addFlag("Tau.MaxTracksDAOD", 5)
54 tau_cfg.addFlag("Tau.TauJetRNNConfig", ["tauid_rnn_1p_R22_v1.json", "tauid_rnn_2p_R22_v1.json", "tauid_rnn_3p_R22_v1.json"])
55 tau_cfg.addFlag("Tau.TauJetRNNWPConfig", ["tauid_rnnWP_1p_R22_v0.root", "tauid_rnnWP_2p_R22_v0.root", "tauid_rnnWP_3p_R22_v0.root"])
56 tau_cfg.addFlag("Tau.TauEleRNNConfig", ["taueveto_rnn_config_1P_r22.json", "taueveto_rnn_config_3P_r22.json"])
57 tau_cfg.addFlag("Tau.TauEleRNNWPConfig", ["rnneveto_mc16d_flat_1p_fix.root", "rnneveto_mc16d_flat_3p_fix.root"])
58 tau_cfg.addFlag("Tau.DecayModeNNClassifierConfig", "NNDecayMode_R22_v1.json")
59 # GNTau ID tune file (need to add another version for noAux)
60 tau_cfg.addFlag("Tau.TauGNNConfig", ["GNTau_pruned_MC23.onnx","GNTau_trunc_MC23.onnx"])
61 tau_cfg.addFlag("Tau.TauGNNWP",
62 [
63 ["GNTauNAprune_flat_model_1p.root", "GNTauNAprune_flat_model_2p.root", "GNTauNAprune_flat_model_3p.root"],
64 ["GNTauNAtrunc_flat_model_1p.root", "GNTauNAtrunc_flat_model_2p.root", "GNTauNAtrunc_flat_model_3p.root"]
65 ])
66 tau_cfg.addFlag("Tau.TauDisplacedGNNConfig", ["GNdTau_pruned_MC23.onnx"])
67 tau_cfg.addFlag("Tau.GNTauScoreName", ["GNTauScore_v0prune","GNTauScore_v1trunc"])
68 tau_cfg.addFlag("Tau.GNTauTransScoreName", ["GNTauScoreSigTrans_v0prune","GNTauScoreSigTrans_v1trunc"])
69 tau_cfg.addFlag("Tau.GNTauMaxTracks", [30,10])
70 tau_cfg.addFlag("Tau.GNTauMaxClusters", [20,6])
71 tau_cfg.addFlag("Tau.GNTauNodeNameTau", "GN2TauNoAux_pb")
72 tau_cfg.addFlag("Tau.GNTauNodeNameJet", "GN2TauNoAux_pu")
73 tau_cfg.addFlag("Tau.GNTauDecorWPNames",
74 [
75 ["GNTauVL_v0prune", "GNTauL_v0prune", "GNTauM_v0prune", "GNTauT_v0prune"],
76 ["GNTauVL_v1trunc", "GNTauL_v1trunc", "GNTauM_v1trunc", "GNTauT_v1trunc"]
77 ])
78 # GNN eVeto config flags
79 #tau_cfg.addFlag("Tau.TauGNNeVetoConfig", ["GNTaueVeto_2025Retune.onnx"])
80 tau_cfg.addFlag("Tau.TauGNNeVetoConfig", ["GNTaueVeto_2025Retune_finitemask.onnx"])
81 tau_cfg.addFlag("Tau.TauGNNeVetoWP", ["GNTaueVeto_flat_model_1p.root", "GNTaueVeto_flat_model_1p.root", "GNTaueVeto_flat_model_1p.root"])
82 tau_cfg.addFlag("Tau.TauGNNeVetoScoreName", "TauGNNeVetoScore")
83 tau_cfg.addFlag("Tau.TauGNNeVetoTransScoreName", "TauGNNeVetoSigTrans")
84 tau_cfg.addFlag("Tau.TauGNNeVetoOutputDiscriminant_NeglogPJet_True", 3)
85 tau_cfg.addFlag("Tau.TauGNNeVetoOutputDiscriminant_PTau", 1)
86 tau_cfg.addFlag("Tau.TauGNNeVetoMaxTracks", [10])
87 tau_cfg.addFlag("Tau.TauGNNeVetoMaxClusters", [6])
88 tau_cfg.addFlag("Tau.TauGNNeVetoNodeNameTau", "TauGNNeVeto_pb")
89 tau_cfg.addFlag("Tau.TauGNNeVetoNodeNameJet", "TauGNNeVeto_pu")
90 tau_cfg.addFlag("Tau.TauGNNeVetoDecorWPNames", ["TauGNNeVeto_L", "TauGNNeVeto_M", "TauGNNeVeto_T"])
91
92
93 tau_cfg.addFlag("Tau.doTausRUs", False)
94 tau_cfg.addFlag("Tau.TausRUsModelFile", "dev/tauRecTools/tausrus_6.8.26.onnx")
95 tau_cfg.addFlag("Tau.TausRUsMinPt", lambda prevFlags: prevFlags.Tau.MinPtDAOD)
96 # Inference-as-a-service config (optional)
97 tau_cfg.addFlag("Tau.TausRUsUseTriton", False)
98 tau_cfg.addFlag("Tau.TausRUsTritonModel", "tausrus")
99 tau_cfg.addFlag("Tau.TausRUsTritonUrl", "localhost")
100 tau_cfg.addFlag("Tau.TausRUsTritonPort", 8001)
101
102 # PanTau config flags
103 from PanTauAlgs.PanTauConfigFlags import createPanTauConfigFlags
104 tau_cfg.addFlagsCategory("Tau.PanTau", createPanTauConfigFlags, prefix=True)
105
106 # create 2 flag categories, for standard taus and electron-subtracted taus
107 tau_cfg.addFlagsCategory("Tau.TauRec", createTauRecConfigFlags, prefix=True)
108 tau_cfg.addFlagsCategory("Tau.TauEleRM", createTauEleRMConfigFlags, prefix=True)
109 tau_cfg.addFlagsCategory("Tau.TauLRT", createTauLRTConfigFlags, prefix=True)
110 # define ActiveConfig in TauConfigFlags.py so it exists for client code like DerivationFramework that don't want to define it via cloneAndReplace
111 # FIXME: this looks more like a hack than good design, maybe dropping Tau.ActiveConfig and using Tau.TauRec as active config would be better?
112 tau_cfg.addFlagsCategory("Tau.ActiveConfig", createTauRecConfigFlags, prefix=True)
113
114 # e-had boosted ditaus, aka electron-subtracted taus
115 tau_cfg.addFlag("Tau.doTauEleRMRec", lambda prevFlags : True if prevFlags.GeoModel.Run <= LHCPeriod.Run3 else False)
116 # helper for derivations, TauJets_EleRM not available for AODs produced before 24.0.17
117 tau_cfg.addFlag("Tau.TauEleRM_isAvailable", lambda prevFlags : "xAOD::TauJetContainer#TauJets_EleRM" in prevFlags.Input.TypedCollections)
118 # helper for derivations, used in PHYSVAL monitoring
119 tau_cfg.addFlag("Tau.TauMuonRM_isAvailable", lambda prevFlags : "xAOD::TauJetContainer#TauJets_MuonRM" in prevFlags.Input.TypedCollections)
120
121 return tau_cfg
122
123
124def createTauRecConfigFlags():
125 from AthenaConfiguration.AthConfigFlags import AthConfigFlags
126 flags = AthConfigFlags()
127 flags.addFlag("prefix", "")
128
129 # Output containers
130 flags.addFlag("TauJets", "TauJets")
131 flags.addFlag("TauTracks", "TauTracks")
132 flags.addFlag("TauShotClusters", "TauShotClusters")
133 flags.addFlag("TauShotClustersLinks", "TauShotClusters_links")
134 flags.addFlag("TauShotPFOs", "TauShotParticleFlowObjects")
135 flags.addFlag("TauPi0Clusters", "TauPi0Clusters")
136 flags.addFlag("TauPi0ClustersLinks", "TauPi0Clusters_links")
137 flags.addFlag("TauHadronicPFOs", "TauHadronicParticleFlowObjects")
138 flags.addFlag("TauNeutralPFOs", "TauNeutralParticleFlowObjects")
139 flags.addFlag("TauChargedPFOs", "TauChargedParticleFlowObjects")
140 flags.addFlag("TauSecondaryVertices", "TauSecondaryVertices")
141 flags.addFlag("TauFinalPi0s", "TauFinalPi0s")
142
143 # Transient containers
144 flags.addFlag("TauJets_tmp", "TauJets_tmp")
145 flags.addFlag("TauCommonPi0Cells", "TauCommonPi0Cells")
146 flags.addFlag("TauPi0Clusters_tmp", "TauPi0Clusters_tmp")
147
148 # Input containers
149 flags.addFlag("VertexCollection", "PrimaryVertices")
150 flags.addFlag("TrackCollection", "InDetTrackParticles")
151 flags.addFlag("SeedJetCollection", lambda prevFlags: "AntiKt4LCTopoJets" if prevFlags.GeoModel.Run <= LHCPeriod.Run3 else "AntiKt4EMPFlowNoPtCutTauSeedJets")
152 flags.addFlag("LargeD0TrackCollection", "InDetLargeD0TrackParticles")
153 flags.addFlag("EventShapeCollection", lambda prevFlags: "Kt4LCTopoOriginEventShape" if prevFlags.GeoModel.Run <= LHCPeriod.Run3 else "Kt4EMPFlowEventShape")
154
155 # Electron-subtracted tau flags appearing in standard tau reconstruction
156 flags.addFlag("inTauEleRM", False)
157 flags.addFlag("RemoveElectronCells", False)
158 flags.addFlag("RemovedElectronClusters", "")
159
160 # Flags for LRT tau
161 flags.addFlag("inTauLRT", False)
162 return flags
163
164
165def createTauEleRMConfigFlags():
166 flags = createTauRecConfigFlags()
167 flags.prefix = "EleRM_"
168 _output_suffix = "_EleRM"
169
170 # Output containers
171 flags.TauJets = f"TauJets{_output_suffix}"
172 flags.TauTracks = f"TauTracks{_output_suffix}"
173 flags.TauShotClusters = f"TauShotClusters{_output_suffix}"
174 flags.TauShotClustersLinks = f"TauShotClusters{_output_suffix}_links"
175 flags.TauShotPFOs = f"TauShotParticleFlowObjects{_output_suffix}"
176 flags.TauPi0Clusters = f"TauPi0Clusters{_output_suffix}"
177 flags.TauPi0ClustersLinks = f"TauPi0Clusters{_output_suffix}_links"
178 flags.TauHadronicPFOs = f"TauHadronicParticleFlowObjects{_output_suffix}"
179 flags.TauNeutralPFOs = f"TauNeutralParticleFlowObjects{_output_suffix}"
180 flags.TauChargedPFOs = f"TauChargedParticleFlowObjects{_output_suffix}"
181 flags.TauSecondaryVertices = f"TauSecondaryVertices{_output_suffix}"
182 flags.TauFinalPi0s = f"TauFinalPi0s{_output_suffix}"
183
184 # Transient containers
185 flags.TauJets_tmp = f"TauJets_tmp{_output_suffix}"
186 flags.TauCommonPi0Cells = f"TauCommonPi0Cells{_output_suffix}"
187 flags.TauPi0Clusters_tmp = f"TauPi0Clusters_tmp{_output_suffix}"
188
189 # Input containers
190 flags.TrackCollection = f"InDetTrackParticles{_output_suffix}"
191 flags.SeedJetCollection = f"AntiKt4LCTopoJets{_output_suffix}"
192 flags.EventShapeCollection = "EleRM_Kt4LCTopoOriginEventShape"
193
194 # Electron-subtracted tau flags appearing in standard tau reconstruction
195 flags.inTauEleRM = True
196 flags.RemoveElectronCells = True
197 flags.RemovedElectronClusters = f"RemovedClusters{_output_suffix}"
198
199 # Electron-subtracted tau specific flags
200 flags.addFlag("EleRM_ElectronWorkingPoint", "Medium")
201 flags.addFlag("RemovedElectronTracks", f"RemovedTracks{_output_suffix}")
202 flags.addFlag("CaloCalTopoClusters_EleRM", f"CaloCalTopoClusters{_output_suffix}")
203 flags.addFlag("LCOriginTopoClusters_EleRM", f"LCOriginTopoClusters{_output_suffix}")
204 flags.addFlag("LCTopoOrigin_EleRM", f"LCTopoOrigin{_output_suffix}")
205 flags.addFlag("EleRM_CheckingConeSize", 0.6)
206
207 return flags
208
209def createTauLRTConfigFlags():
210 flags = createTauRecConfigFlags()
211 flags.prefix = "TauLRT_"
212 _output_suffix = "LRT"
213
214 # Output containers
215 flags.TauJets = f"TauJets{_output_suffix}"
216 flags.TauTracks = f"TauTracks{_output_suffix}"
217 flags.TauShotClusters = f"TauShotClusters{_output_suffix}"
218 flags.TauShotClustersLinks = f"TauShotClusters{_output_suffix}_links"
219 flags.TauShotPFOs = f"TauShotParticleFlowObjects{_output_suffix}"
220 flags.TauPi0Clusters = f"TauPi0Clusters{_output_suffix}"
221 flags.TauPi0ClustersLinks = f"TauPi0Clusters{_output_suffix}_links"
222 flags.TauHadronicPFOs = f"TauHadronicParticleFlowObjects{_output_suffix}"
223 flags.TauNeutralPFOs = f"TauNeutralParticleFlowObjects{_output_suffix}"
224 flags.TauChargedPFOs = f"TauChargedParticleFlowObjects{_output_suffix}"
225 flags.TauSecondaryVertices = f"TauSecondaryVertices{_output_suffix}"
226 flags.TauFinalPi0s = f"TauFinalPi0s{_output_suffix}"
227
228 # Transient containers
229 flags.TauJets_tmp = f"TauJets_tmp{_output_suffix}"
230 flags.TauCommonPi0Cells = f"TauCommonPi0Cells{_output_suffix}"
231 flags.TauPi0Clusters_tmp = f"TauPi0Clusters_tmp{_output_suffix}"
232
233 flags.inTauLRT = True
234
235 return flags
236
237# Self test
238
239class TestTauRecConfigFlags(unittest.TestCase):
240 def runTest(self):
241 createTauConfigFlags()
242
243
244if __name__ == "__main__":
245 unittest.main()
Generated on for ATLAS Offline Software by doxygen 1.17.0