ATLAS Offline Software
Loading...
Searching...
No Matches
Classes | Functions
TauConfigFlags Namespace Reference

Classes

class  TestTauRecConfigFlags

Functions

 createTauConfigFlags ()
 createTauRecConfigFlags ()
 createTauEleRMConfigFlags ()

Function Documentation

◆ createTauConfigFlags()

TauConfigFlags.createTauConfigFlags ( )

Definition at line 8 of file TauConfigFlags.py.

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.doPi0Clus", True)
19 tau_cfg.addFlag("Tau.doPanTau", True)
20 tau_cfg.addFlag("Tau.doRNNTrackClass", True)
21 tau_cfg.addFlag("Tau.doTauDiscriminant", True)
22 tau_cfg.addFlag("Tau.associateLRT", False)
23 tau_cfg.addFlag("Tau.isStandalone", False)
24 # Classify Large Radius Tracks in tau track classifier
25 tau_cfg.addFlag("Tau.classifyLRT", 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
41 # Run2 settings and calibration files
42 tau_cfg.addFlag("Tau.tauRecToolsCVMFSPath", "tauRecTools/R22_preprod")
43 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")
44 tau_cfg.addFlag("Tau.CalibrateLCConfig", "CaloTES_R22_Round2.5.root")
45 tau_cfg.addFlag("Tau.CombinedTESConfig", "CombinedTES_R22_Round2.5.root")
46 tau_cfg.addFlag("Tau.MvaTESConfig0p", "MvaTES_0p_R23.root")
47 tau_cfg.addFlag("Tau.MvaTESConfig", "MvaTES_R23.root")
48 tau_cfg.addFlag("Tau.MinPt0p", 9.25*Units.GeV)
49 tau_cfg.addFlag("Tau.MinPt", 6.75*Units.GeV)
50 tau_cfg.addFlag("Tau.MinPtDAOD", 13*Units.GeV)
51 tau_cfg.addFlag("Tau.MaxTracksDAOD", 5)
52 tau_cfg.addFlag("Tau.TauJetRNNConfig", ["tauid_rnn_1p_R22_v1.json", "tauid_rnn_2p_R22_v1.json", "tauid_rnn_3p_R22_v1.json"])
53 tau_cfg.addFlag("Tau.TauJetRNNWPConfig", ["tauid_rnnWP_1p_R22_v0.root", "tauid_rnnWP_2p_R22_v0.root", "tauid_rnnWP_3p_R22_v0.root"])
54 tau_cfg.addFlag("Tau.TauEleRNNConfig", ["taueveto_rnn_config_1P_r22.json", "taueveto_rnn_config_3P_r22.json"])
55 tau_cfg.addFlag("Tau.TauEleRNNWPConfig", ["taueveto_rnn_flat_1P_r22.root", "taueveto_rnn_flat_3P_r22.root"])
56 tau_cfg.addFlag("Tau.DecayModeNNClassifierConfig", "NNDecayMode_R22_v1.json")
57 tau_cfg.addFlag("Tau.TauEleRNNWPfix", ["rnneveto_mc16d_flat_1p_fix.root", "rnneveto_mc16d_flat_3p_fix.root"])
58 # GNTau ID tune file (need to add another version for noAux)
59 tau_cfg.addFlag("Tau.TauGNNConfig", ["GNTau_pruned_MC23.onnx","GNTau_trunc_MC23.onnx"])
60 tau_cfg.addFlag("Tau.TauGNNWP",
61 [
62 ["GNTauNAprune_flat_model_1p.root", "GNTauNAprune_flat_model_2p.root", "GNTauNAprune_flat_model_3p.root"],
63 ["GNTauNAtrunc_flat_model_1p.root", "GNTauNAtrunc_flat_model_2p.root", "GNTauNAtrunc_flat_model_3p.root"]
64 ])
65 tau_cfg.addFlag("Tau.GNTauScoreName", ["GNTauScore_v0prune","GNTauScore_v1trunc"])
66 tau_cfg.addFlag("Tau.GNTauTransScoreName", ["GNTauScoreSigTrans_v0prune","GNTauScoreSigTrans_v1trunc"])
67 tau_cfg.addFlag("Tau.GNTauMaxTracks", [30,10])
68 tau_cfg.addFlag("Tau.GNTauMaxClusters", [20,6])
69 tau_cfg.addFlag("Tau.GNTauNodeNameTau", "GN2TauNoAux_pb")
70 tau_cfg.addFlag("Tau.GNTauNodeNameJet", "GN2TauNoAux_pu")
71 tau_cfg.addFlag("Tau.GNTauDecorWPNames",
72 [
73 ["GNTauVL_v0prune", "GNTauL_v0prune", "GNTauM_v0prune", "GNTauT_v0prune"],
74 ["GNTauVL_v1trunc", "GNTauL_v1trunc", "GNTauM_v1trunc", "GNTauT_v1trunc"]
75 ])
76
77
78 # PanTau config flags
79 from PanTauAlgs.PanTauConfigFlags import createPanTauConfigFlags
80 tau_cfg.addFlagsCategory("Tau.PanTau", createPanTauConfigFlags, prefix=True)
81
82 # create 2 flag categories, for standard taus and electron-subtracted taus
83 tau_cfg.addFlagsCategory("Tau.TauRec", createTauRecConfigFlags, prefix=True)
84 tau_cfg.addFlagsCategory("Tau.TauEleRM", createTauEleRMConfigFlags, prefix=True)
85 # define ActiveConfig in TauConfigFlags.py so it exists for client code like DerivationFramework that don't want to define it via cloneAndReplace
86 # FIXME: this looks more like a hack than good design, maybe dropping Tau.ActiveConfig and using Tau.TauRec as active config would be better?
87 tau_cfg.addFlagsCategory("Tau.ActiveConfig", createTauRecConfigFlags, prefix=True)
88
89 # e-had boosted ditaus, aka electron-subtracted taus
90 tau_cfg.addFlag("Tau.doTauEleRMRec", True)
91 # helper for derivations, TauJets_EleRM not available for AODs produced before 24.0.17
92 tau_cfg.addFlag("Tau.TauEleRM_isAvailable", lambda prevFlags : "xAOD::TauJetContainer#TauJets_EleRM" in prevFlags.Input.TypedCollections)
93 # helper for derivations, used in PHYSVAL monitoring
94 tau_cfg.addFlag("Tau.TauMuonRM_isAvailable", lambda prevFlags : "xAOD::TauJetContainer#TauJets_MuonRM" in prevFlags.Input.TypedCollections)
95
96 return tau_cfg
97
98

◆ createTauEleRMConfigFlags()

TauConfigFlags.createTauEleRMConfigFlags ( )

Definition at line 138 of file TauConfigFlags.py.

138def createTauEleRMConfigFlags():
139 flags = createTauRecConfigFlags()
140 flags.prefix = "EleRM_"
141 _output_suffix = "_EleRM"
142
143 # Output containers
144 flags.TauJets = f"TauJets{_output_suffix}"
145 flags.TauTracks = f"TauTracks{_output_suffix}"
146 flags.TauShotClusters = f"TauShotClusters{_output_suffix}"
147 flags.TauShotClustersLinks = f"TauShotClusters{_output_suffix}_links"
148 flags.TauShotPFOs = f"TauShotParticleFlowObjects{_output_suffix}"
149 flags.TauPi0Clusters = f"TauPi0Clusters{_output_suffix}"
150 flags.TauPi0ClustersLinks = f"TauPi0Clusters{_output_suffix}_links"
151 flags.TauHadronicPFOs = f"TauHadronicParticleFlowObjects{_output_suffix}"
152 flags.TauNeutralPFOs = f"TauNeutralParticleFlowObjects{_output_suffix}"
153 flags.TauChargedPFOs = f"TauChargedParticleFlowObjects{_output_suffix}"
154 flags.TauSecondaryVertices = f"TauSecondaryVertices{_output_suffix}"
155 flags.TauFinalPi0s = f"TauFinalPi0s{_output_suffix}"
156
157 # Transient containers
158 flags.TauJets_tmp = f"TauJets_tmp{_output_suffix}"
159 flags.TauCommonPi0Cells = f"TauCommonPi0Cells{_output_suffix}"
160 flags.TauPi0Clusters_tmp = f"TauPi0Clusters_tmp{_output_suffix}"
161
162 # Input containers
163 flags.TrackCollection = f"InDetTrackParticles{_output_suffix}"
164 flags.SeedJetCollection = f"AntiKt4LCTopoJets{_output_suffix}"
165 flags.EventShapeCollection = "EleRM_Kt4LCTopoOriginEventShape"
166
167 # Electron-subtracted tau flags appearing in standard tau reconstruction
168 flags.inTauEleRM = True
169 flags.RemoveElectronCells = True
170 flags.RemovedElectronClusters = f"RemovedClusters{_output_suffix}"
171
172 # Electron-subtracted tau specific flags
173 flags.addFlag("EleRM_ElectronWorkingPoint", "Medium")
174 flags.addFlag("RemovedElectronTracks", f"RemovedTracks{_output_suffix}")
175 flags.addFlag("CaloCalTopoClusters_EleRM", f"CaloCalTopoClusters{_output_suffix}")
176 flags.addFlag("LCOriginTopoClusters_EleRM", f"LCOriginTopoClusters{_output_suffix}")
177 flags.addFlag("LCTopoOrigin_EleRM", f"LCTopoOrigin{_output_suffix}")
178 flags.addFlag("EleRM_CheckingConeSize", 0.6)
179
180 return flags
181
182
183# Self test
184

◆ createTauRecConfigFlags()

TauConfigFlags.createTauRecConfigFlags ( )

Definition at line 99 of file TauConfigFlags.py.

99def createTauRecConfigFlags():
100 from AthenaConfiguration.AthConfigFlags import AthConfigFlags
101 flags = AthConfigFlags()
102 flags.addFlag("prefix", "")
103
104 # Output containers
105 flags.addFlag("TauJets", "TauJets")
106 flags.addFlag("TauTracks", "TauTracks")
107 flags.addFlag("TauShotClusters", "TauShotClusters")
108 flags.addFlag("TauShotClustersLinks", "TauShotClusters_links")
109 flags.addFlag("TauShotPFOs", "TauShotParticleFlowObjects")
110 flags.addFlag("TauPi0Clusters", "TauPi0Clusters")
111 flags.addFlag("TauPi0ClustersLinks", "TauPi0Clusters_links")
112 flags.addFlag("TauHadronicPFOs", "TauHadronicParticleFlowObjects")
113 flags.addFlag("TauNeutralPFOs", "TauNeutralParticleFlowObjects")
114 flags.addFlag("TauChargedPFOs", "TauChargedParticleFlowObjects")
115 flags.addFlag("TauSecondaryVertices", "TauSecondaryVertices")
116 flags.addFlag("TauFinalPi0s", "TauFinalPi0s")
117
118 # Transient containers
119 flags.addFlag("TauJets_tmp", "TauJets_tmp")
120 flags.addFlag("TauCommonPi0Cells", "TauCommonPi0Cells")
121 flags.addFlag("TauPi0Clusters_tmp", "TauPi0Clusters_tmp")
122
123 # Input containers
124 flags.addFlag("VertexCollection", "PrimaryVertices")
125 flags.addFlag("TrackCollection", "InDetTrackParticles")
126 flags.addFlag("SeedJetCollection", "AntiKt4LCTopoJets")
127 flags.addFlag("LargeD0TrackCollection", "InDetLargeD0TrackParticles")
128 flags.addFlag("EventShapeCollection", "Kt4LCTopoOriginEventShape")
129
130 # Electron-subtracted tau flags appearing in standard tau reconstruction
131 flags.addFlag("inTauEleRM", False)
132 flags.addFlag("RemoveElectronCells", False)
133 flags.addFlag("RemovedElectronClusters", "")
134
135 return flags
136
137
Generated on for ATLAS Offline Software by doxygen 1.14.0