ATLAS Offline Software
Functions | Variables
JetMomentToolsConfig Namespace Reference

Functions

def idTrackSelToolFromJetCtx (trkOpt, trkProperties)
 
def getEMScaleMomTool (jetdef, modspec="")
 
def getConstitFourMomTool (jetdef, modspec="")
 
def getJVFTool (jetdef, modspec)
 
def getJVTTool (jetdef, modspec)
 
def getNNJvtTool (jetdef, modspec)
 
def getTrackMomentsTool (jetdef, modspec)
 
def getTrackSumMomentsTool (jetdef, modspec)
 
def getOriginCorrVxTool (jetdef, modspec)
 
def getJetPtAssociationTool (jetdef, modspec)
 
def getQGTaggingTool (jetdef, modspec)
 
def getPFlowfJVTTool (jetdef, modspec)
 
def getPFlowbJVTTool (jetdef, modspec)
 

Variables

 jetmomentlog = Logging.logging.getLogger('JetMomentToolsConfig')
 

Function Documentation

◆ getConstitFourMomTool()

def JetMomentToolsConfig.getConstitFourMomTool (   jetdef,
  modspec = "" 
)

Definition at line 61 of file JetMomentToolsConfig.py.

61 def getConstitFourMomTool(jetdef, modspec=""):
62 
66  CaloClusterStates = {
67  "UNKNOWN" : -1,
68  "UNCALIBRATED" : 0,
69  "CALIBRATED" : 1,
70  "ALTCALIBRATED" : 2,
71  "NSTATES" : 3
72  }
73 
74  cfourmom = CompFactory.JetConstitFourMomTool("constitfourmom_{0}".format(jetdef.basename))
75  if "LCTopo" in jetdef.basename or "EMTopo" in jetdef.basename:
76  cfourmom.JetScaleNames = ["DetectorEtaPhi"]
77  if "HLT_" in jetdef.fullname():
78  cfourmom.AltConstitColls = [""]
79  cfourmom.AltConstitScales = [0]
80  cfourmom.AltJetScales = ["JetConstitScaleMomentum"]
81  else:
82  clstate = "CALIBRATED" if "LCTopo" in jetdef.basename else "UNCALIBRATED"
83  cfourmom.AltConstitColls = [jetdef.inputdef.inputname]
84  cfourmom.AltConstitScales = [CaloClusterStates[clstate]]
85  cfourmom.AltJetScales = [""]
86  # Drop the LC-calibrated four-mom for EMTopo jets as we only wanted it as a possibility
87  # in MET CST calculations but never used it
88  elif "PFlow" in jetdef.basename or "UFO" in jetdef.basename:
89  cfourmom.JetScaleNames = ["DetectorEtaPhi"]
90  cfourmom.AltConstitColls = [""]
91  cfourmom.AltConstitScales = [0]
92  cfourmom.AltJetScales = ["JetConstitScaleMomentum"]
93 
94  return cfourmom
95 
96 # Jet vertex fraction with selection.

◆ getEMScaleMomTool()

def JetMomentToolsConfig.getEMScaleMomTool (   jetdef,
  modspec = "" 
)

Definition at line 39 of file JetMomentToolsConfig.py.

39 def getEMScaleMomTool(jetdef, modspec=""):
40  # This may need updating e.g. for evolving trigger cluster container names
41  # We do the non-trivial summation over constituents unless the jets were
42  # built directly from EM-scale topoclusters, in which case we can just
43  # copy the constituent scale
44  useUncalibConstits = False
45  if jetdef.inputdef.basetype==xAODType.CaloCluster:
46  builtFromEMClusters = jetdef.inputdef.inputname in ["CaloCalTopoClusters","HLT_CaloTopoClustersFS"] and jetdef.inputdef.modifiers==["EM"]
47  useUncalibConstits = not builtFromEMClusters
48  elif (jetdef.inputdef.basetype==xAODType.ParticleFlow or jetdef.inputdef.basetype==xAODType.FlowElement):
49  useUncalibConstits = True
50  else:
51  raise ValueError("EM scale momentum not defined for input type {}".format(jetdef.inputdef.basetype))
52 
53  emscalemom = CompFactory.JetEMScaleMomTool(
54  "emscalemom_{}".format(jetdef.basename),
55  UseUncalibConstits = useUncalibConstits,
56  JetContainer = jetdef.fullname(),
57  )
58 
59  return emscalemom
60 

◆ getJetPtAssociationTool()

def JetMomentToolsConfig.getJetPtAssociationTool (   jetdef,
  modspec 
)

Definition at line 176 of file JetMomentToolsConfig.py.

176 def getJetPtAssociationTool(jetdef, modspec):
177 
178  from JetRecConfig.JetDefinition import buildJetAlgName
179 
180  truthJetAlg = buildJetAlgName(jetdef.algorithm, jetdef.radius)+'Truth'+str(modspec)+'Jets'
181 
182  jetPtAssociation = CompFactory.JetPtAssociationTool('jetPtAssociation',
183  MatchingJetContainer = truthJetAlg,
184  AssociationName = "GhostTruth")
185 
186  return jetPtAssociation
187 
188 

◆ getJVFTool()

def JetMomentToolsConfig.getJVFTool (   jetdef,
  modspec 
)

Definition at line 97 of file JetMomentToolsConfig.py.

97 def getJVFTool(jetdef, modspec):
98  trkopt = modspec or jetdef.context # allow to overide jetdef.context if modspec is specified
99  trackingKeys = jetdef._cflags.Jet.Context[trkopt]
100  jvf = CompFactory.JetVertexFractionTool(
101  "jvf",
102  VertexContainer = trackingKeys["Vertices"],
103  AssociatedTracks = trackingKeys["GhostTracksLabel"],
104  TrackVertexAssociation = trackingKeys["TVA"],
105  TrackParticleContainer = trackingKeys["Tracks"],
106  TrackSelector = idTrackSelToolFromJetCtx(trkopt, trackingKeys),
107  SuppressInputDependence = True,
108  UseOriginVertex = jetdef.byVertex
109  )
110  return jvf
111 
112 
113 # Jet vertex tagger with selection.

◆ getJVTTool()

def JetMomentToolsConfig.getJVTTool (   jetdef,
  modspec 
)

Definition at line 114 of file JetMomentToolsConfig.py.

114 def getJVTTool(jetdef, modspec):
115  jvt = CompFactory.JetVertexTaggerTool(
116  "jvt",
117  VertexContainer = jetdef._cflags.Jet.Context[modspec or jetdef.context]["Vertices"],
118  SuppressInputDependence = True,
119  UseOriginVertex = jetdef.byVertex
120  )
121  return jvt
122 
123 # Jet vertex tagger with neural network.

◆ getNNJvtTool()

def JetMomentToolsConfig.getNNJvtTool (   jetdef,
  modspec 
)

Definition at line 124 of file JetMomentToolsConfig.py.

124 def getNNJvtTool(jetdef, modspec):
125  nnjvt = CompFactory.getComp("JetPileupTag::JetVertexNNTagger")(
126  "nnjvt",
127  VertexContainer = jetdef._cflags.Jet.Context[modspec or jetdef.context]["Vertices"],
128  SuppressInputDependence = True
129  )
130  return nnjvt
131 
132 

◆ getOriginCorrVxTool()

def JetMomentToolsConfig.getOriginCorrVxTool (   jetdef,
  modspec 
)

Definition at line 166 of file JetMomentToolsConfig.py.

166 def getOriginCorrVxTool(jetdef, modspec):
167  origin_setpv = CompFactory.JetOriginCorrectionTool(
168  "jetorigin_setpv",
169  VertexContainer = jetdef._cflags.Jet.Context[modspec or jetdef.context]["Vertices"],
170  OriginCorrectedName = "",
171  OnlyAssignPV = True,
172  )
173  return origin_setpv
174 
175 

◆ getPFlowbJVTTool()

def JetMomentToolsConfig.getPFlowbJVTTool (   jetdef,
  modspec 
)

Definition at line 227 of file JetMomentToolsConfig.py.

227 def getPFlowbJVTTool(jetdef, modspec):
228 
229  from JetCalibTools import JetCalibToolsConfig
230  jetCalibrationTool = JetCalibToolsConfig.getJetCalibToolFromString(jetdef, "AnalysisLatest:mc:JetArea_Residual_EtaJES")
231 
232  wPFOTool = CompFactory.getComp('CP::WeightPFOTool')("bJVT__wPFO")
233 
234  trackingKeys = jetdef._cflags.Jet.Context[modspec or jetdef.context]
235 
236  bJVTTool = CompFactory.JetBalancePFlowJvtTool('bJVT',
237  verticesName = trackingKeys["Vertices"],
238  TrackVertexAssociation = trackingKeys["TVA"],
239  WeightPFOTool = wPFOTool,
240  JetCalibrationTool = jetCalibrationTool,
241  FEName = jetdef.inputdef.containername,
242  ORNameFE = "",
243  BjvtRawName = 'DFCommonJets_bJvt',
244  includePV = True)
245 
246  return bJVTTool

◆ getPFlowfJVTTool()

def JetMomentToolsConfig.getPFlowfJVTTool (   jetdef,
  modspec 
)

Definition at line 202 of file JetMomentToolsConfig.py.

202 def getPFlowfJVTTool(jetdef, modspec):
203 
204  from JetCalibTools import JetCalibToolsConfig
205  calibString = "AnalysisLatest:mc:JetArea_Residual_EtaJES"
206  if( modspec and modspec == "CustomVtx" ) :
207  calibString = "AnalysisLatest:mc:JetArea_Residual_EtaJES:Kt4EMPFlowCustomVtxEventShape:HggPrimaryVertices"
208  jetCalibrationTool = JetCalibToolsConfig.getJetCalibToolFromString(jetdef, calibString)
209 
210  wPFOTool = CompFactory.getComp('CP::WeightPFOTool')("fJVT__wPFO")
211 
212  trackingKeys = jetdef._cflags.Jet.Context[modspec or jetdef.context]
213 
214  fJVTTool = CompFactory.JetForwardPFlowJvtTool("fJVT",
215  verticesName = trackingKeys["Vertices"],
216  TrackVertexAssociation = trackingKeys["TVA"],
217  WeightPFOTool = wPFOTool,
218  JetCalibrationTool = jetCalibrationTool,
219  FEName = jetdef.inputdef.containername,
220  ORName = "",
221  FjvtRawName = "DFCommonJets_fJvt",
222  includePV = False)
223 
224  return fJVTTool
225 
226 

◆ getQGTaggingTool()

def JetMomentToolsConfig.getQGTaggingTool (   jetdef,
  modspec 
)

Definition at line 189 of file JetMomentToolsConfig.py.

189 def getQGTaggingTool(jetdef, modspec):
190  trkopt = modspec or jetdef.context # allow to overide jetdef.context if modspec is specified
191  trackingKeys = jetdef._cflags.Jet.Context[trkopt]
192 
193  qgtagging = CompFactory.JetQGTaggerVariableTool('qgtagging',
194  VertexContainer = trackingKeys["Vertices"],
195  TrackVertexAssociation = trackingKeys["TVA"],
196  TrackSelector = idTrackSelToolFromJetCtx(trkopt,trackingKeys),
197  )
198 
199  return qgtagging
200 
201 

◆ getTrackMomentsTool()

def JetMomentToolsConfig.getTrackMomentsTool (   jetdef,
  modspec 
)

Definition at line 133 of file JetMomentToolsConfig.py.

133 def getTrackMomentsTool(jetdef, modspec):
134  # retrieve the tracking keys to be used with modspec :
135  trkopt = modspec or jetdef.context # allow to overide jetdef.context if modspec is specified
136  trackingKeys = jetdef._cflags.Jet.Context[trkopt]
137 
138  trackmoments = CompFactory.JetTrackMomentsTool(
139  "trkmoms",
140  VertexContainer = trackingKeys["Vertices"],
141  AssociatedTracks = trackingKeys["GhostTracksLabel"],
142  TrackVertexAssociation = trackingKeys["TVA"],
143  TrackMinPtCuts = [500, 1000],
144  TrackSelector = idTrackSelToolFromJetCtx(trkopt,trackingKeys),
145  DoPFlowMoments = 'PFlow' in jetdef.fullname() or 'UFO' in jetdef.fullname() ,
146  )
147  return trackmoments
148 

◆ getTrackSumMomentsTool()

def JetMomentToolsConfig.getTrackSumMomentsTool (   jetdef,
  modspec 
)

Definition at line 149 of file JetMomentToolsConfig.py.

149 def getTrackSumMomentsTool(jetdef, modspec):
150  trkopt = modspec or jetdef.context # allow to overide jetdef.context if modspec is specified
151  trackingKeys = jetdef._cflags.Jet.Context[trkopt]
152  jettrackselloose = idTrackSelToolFromJetCtx(trkopt,trackingKeys)
153  # retrieve the tracking keys to be used with modspec :
154  tracksummoments = CompFactory.JetTrackSumMomentsTool(
155  "trksummoms",
156  VertexContainer = trackingKeys["Vertices"],
157  AssociatedTracks = trackingKeys["GhostTracksLabel"],
158  TrackVertexAssociation = trackingKeys["TVA"],
159  RequireTrackPV = True,
160  TrackSelector = jettrackselloose
161  )
162  return tracksummoments
163 
164 # This tool sets a decoration saying which the nominal HS PV was.
165 # Historically it did the origin correction, but now we do this to constituents

◆ idTrackSelToolFromJetCtx()

def JetMomentToolsConfig.idTrackSelToolFromJetCtx (   trkOpt,
  trkProperties 
) 
returns a InDetTrackSelectionTool configured with the jet context corresponding to trkOpt
  (technically wrapped inside a JetTrackSelectionTool)

Definition at line 24 of file JetMomentToolsConfig.py.

24 def idTrackSelToolFromJetCtx(trkOpt,trkProperties):
25  """returns a InDetTrackSelectionTool configured with the jet context corresponding to trkOpt
26  (technically wrapped inside a JetTrackSelectionTool)
27  """
28  # JetTrackSelectionTool is still used by trk moment tools.
29  # it should be deprecated in favor of simply the InDet tool
30  from JetRecTools.JetRecToolsConfig import getIDTrackSelectionTool #
31 
32  return CompFactory.JetTrackSelectionTool(
33  f"tracsel{trkOpt}",
34  Selector = getIDTrackSelectionTool(f"trackSel{trkOpt}", **trkProperties["trackSelOptions"])
35  )
36 
37 
38 

Variable Documentation

◆ jetmomentlog

JetMomentToolsConfig.jetmomentlog = Logging.logging.getLogger('JetMomentToolsConfig')

Definition at line 17 of file JetMomentToolsConfig.py.

JetMomentToolsConfig.getJetPtAssociationTool
def getJetPtAssociationTool(jetdef, modspec)
Definition: JetMomentToolsConfig.py:176
JetMomentToolsConfig.idTrackSelToolFromJetCtx
def idTrackSelToolFromJetCtx(trkOpt, trkProperties)
Definition: JetMomentToolsConfig.py:24
vtune_athena.format
format
Definition: vtune_athena.py:14
JetMomentToolsConfig.getConstitFourMomTool
def getConstitFourMomTool(jetdef, modspec="")
Definition: JetMomentToolsConfig.py:61
JetMomentToolsConfig.getOriginCorrVxTool
def getOriginCorrVxTool(jetdef, modspec)
Definition: JetMomentToolsConfig.py:166
python.JetDefinition.buildJetAlgName
def buildJetAlgName(finder, mainParam, variableRMassScale=-1.0, variableRMinRadius=-1.0)
Definition: JetDefinition.py:50
JetMomentToolsConfig.getPFlowbJVTTool
def getPFlowbJVTTool(jetdef, modspec)
Definition: JetMomentToolsConfig.py:227
JetCalibToolsConfig.getJetCalibToolFromString
def getJetCalibToolFromString(jetdef, modspec)
Definition: JetCalibToolsConfig.py:249
JetMomentToolsConfig.getNNJvtTool
def getNNJvtTool(jetdef, modspec)
Definition: JetMomentToolsConfig.py:124
JetMomentToolsConfig.getPFlowfJVTTool
def getPFlowfJVTTool(jetdef, modspec)
Definition: JetMomentToolsConfig.py:202
if
if(febId1==febId2)
Definition: LArRodBlockPhysicsV0.cxx:569
JetMomentToolsConfig.getJVFTool
def getJVFTool(jetdef, modspec)
Definition: JetMomentToolsConfig.py:97
str
Definition: BTagTrackIpAccessor.cxx:11
JetMomentToolsConfig.getTrackMomentsTool
def getTrackMomentsTool(jetdef, modspec)
Definition: JetMomentToolsConfig.py:133
JetMomentToolsConfig.getJVTTool
def getJVTTool(jetdef, modspec)
Definition: JetMomentToolsConfig.py:114
JetRecToolsConfig.getIDTrackSelectionTool
def getIDTrackSelectionTool(toolname, **toolProps)
Definition: JetRecToolsConfig.py:19
JetMomentToolsConfig.getQGTaggingTool
def getQGTaggingTool(jetdef, modspec)
Definition: JetMomentToolsConfig.py:189
JetMomentToolsConfig.getTrackSumMomentsTool
def getTrackSumMomentsTool(jetdef, modspec)
Definition: JetMomentToolsConfig.py:149
JetMomentToolsConfig.getEMScaleMomTool
def getEMScaleMomTool(jetdef, modspec="")
Definition: JetMomentToolsConfig.py:39
Generated on Tue Jul 2 2024 22:18:36 for ATLAS Offline Software by   doxygen 1.8.18