20def createJetContextFlags():
21 flags = AthConfigFlags()
22
23
24 run3context=ldict(
25 Tracks = "InDetTrackParticles",
26 JetTracks = "JetSelectedTracks",
27 JetTracksQualityCuts = "JetSelectedTracks_trackSelOpt",
28 Vertices = "PrimaryVertices",
29 TVA = "JetTrackVtxAssoc",
30 GhostTracks = "PseudoJetGhostTrack",
31 GhostTracksLabel = "GhostTrack",
32 EventDensity = "EventDensity",
33 GhostTrackCutLevel = 'NoCut',
34
35
36
37 trackSelOptions = ldict( CutLevel = "Loose", minPt=500, maxAbsEta=2.5 ),
38 )
39
40 flags.addFlag("Jet.Context.Run3" , run3context)
41
42 flags.addFlag("Jet.Context.Run4" , run3context.clone(
43 trackSelOptions = run3context["trackSelOptions"].clone(maxAbsEta=4.0)
44 ))
45 flags.addFlag("Jet.Context.HL_LHC", flags.Jet.Context.Run4)
46
47
48
49 def _defaultFlag(prevFlags):
50 try:
51 run = prevFlags.GeoModel.Run
52 except (ValueError, RuntimeError) :
53
54 return {}
55 return prevFlags.Jet.Context.Run3 if run <= LHCPeriod.Run3 else prevFlags.Jet.Context.Run4
56 flags.addFlag("Jet.Context.default", _defaultFlag)
57
58
59
60
61
62
63 flags.addFlag("Jet.Context.EleRM" , run3context.clone(
64 Tracks = "InDetTrackParticles_EleRM",
65 TVA = "JetTrackVtxAssoc_EleRM",
66 JetTracks = "JetSelectedTracks_EleRM",
67 GhostTracks = "PseudoJetGhostTrack_EleRM",
68 EventDensity = "EleRM_EventDensity",
69 ))
70
71
72
73
74
75 flags.addFlag("Jet.Context.CommonTrackKeys",["Tracks", "Vertices", "TVA", "GhostTracks", "GhostTracksLabel", "JetTracks", "JetTracksQualityCuts"],)
76
77
78
79
80 if moduleExists("TriggerMenuMT"):
81 from TriggerMenuMT.HLT.Jet.JetRecoCommon import addJetContextFlags
82 addJetContextFlags(flags)
83
84
85
86 if moduleExists("DerivationFrameworkHiggs"):
87 from DerivationFrameworkHiggs.HIGG1D1CustomJetsConfig import addJetContextFlags
88 addJetContextFlags(flags)
89
90
91
92 if moduleExists("DerivationFrameworkPhys"):
93 from DerivationFrameworkPhys.GNNVertexConfig import addJetContextFlags
94 addJetContextFlags(flags)
95
96 return flags
97
98
99