CMS 3D CMS Logo

Functions | Variables
iterativeTkConfig Namespace Reference

Functions

def _classifier (iteration, oldStyle=False, oldStyleQualityMasks=False)
 
def _clusterRemover (iteration)
 
def _modulePrefix (iteration)
 
def _seedOrTrackProducers (postfix, typ)
 
def _tracks (iteration)
 
def allEras ()
 
def clusterRemoverForIter (iteration, eraName="", postfix="", module=None)
 
def createEarlyTask (eraName, postfix, modDict)
 
def iterationAlgos (postfix, includeSequenceName=False)
 
def nonDefaultEras ()
 
def seedProducers (postfix)
 
def trackProducers (postfix)
 

Variables

 _allEras
 
 _defaultEra
 
 _defaultEraName
 
 _iterations
 
 _iterations_muonSeeded
 
 _iterations_muonSeeded_trackingPhase1
 
 _iterations_muonSeeded_trackingPhase2PU140
 
 _iterations_trackingLowPU
 
 _iterations_trackingPhase1
 
 _iterations_trackingPhase2PU140
 
 _multipleSeedProducers
 
 _multipleSeedProducers_trackingLowPU
 
 _multipleSeedProducers_trackingPhase1
 
 _multipleSeedProducers_trackingPhase2PU140
 
 _nonDefaultEraNames
 
 _nonDefaultEras
 
 _oldStyleHasSelector
 
 _trackClusterRemoverBase
 
 _trackClusterRemoverBase_trackingPhase2PU140
 
 func
 
 maxChi2
 
 minNumberOfLayersWithMeasBeforeFiltering
 
 phase2OTClusters
 
 phase2pixelClusters
 
 pixelClusters
 
 stripClusters
 
 TrackQuality
 

Function Documentation

◆ _classifier()

def iterativeTkConfig._classifier (   iteration,
  oldStyle = False,
  oldStyleQualityMasks = False 
)
private

Definition at line 142 of file iterativeTkConfig.py.

References _modulePrefix().

Referenced by clusterRemoverForIter().

142 def _classifier(iteration, oldStyle=False, oldStyleQualityMasks=False):
143  pre = _modulePrefix(iteration)
144  if oldStyle:
145  if iteration in _oldStyleHasSelector:
146  return pre+"Selector:" + ("QualityMasks" if oldStyleQualityMasks else pre)
147  else:
148  return pre
149  else:
150  return pre+":QualityMasks"
151 
def _classifier(iteration, oldStyle=False, oldStyleQualityMasks=False)
def _modulePrefix(iteration)

◆ _clusterRemover()

def iterativeTkConfig._clusterRemover (   iteration)
private

Definition at line 136 of file iterativeTkConfig.py.

References _modulePrefix().

Referenced by clusterRemoverForIter().

136 def _clusterRemover(iteration):
137  return _modulePrefix(iteration)+"Clusters"
138 
def _clusterRemover(iteration)
def _modulePrefix(iteration)

◆ _modulePrefix()

def iterativeTkConfig._modulePrefix (   iteration)
private

Definition at line 133 of file iterativeTkConfig.py.

Referenced by _classifier(), _clusterRemover(), _seedOrTrackProducers(), _tracks(), and iterationAlgos().

133 def _modulePrefix(iteration):
134  return iteration[0].lower()+iteration[1:]
135 
def _modulePrefix(iteration)

◆ _seedOrTrackProducers()

def iterativeTkConfig._seedOrTrackProducers (   postfix,
  typ 
)
private

Definition at line 173 of file iterativeTkConfig.py.

References _modulePrefix(), and python.rootplot.root2matplotlib.replace().

Referenced by seedProducers(), and trackProducers().

173 def _seedOrTrackProducers(postfix, typ):
174  ret = []
175  iters = globals()["_iterations"+postfix]
176  if typ == "Seeds":
177  multipleSeedProducers = globals()["_multipleSeedProducers"+postfix]
178  else:
179  multipleSeedProducers = None
180  for i in iters:
181  seeder = _modulePrefix(i)+typ
182  if multipleSeedProducers is not None and i in multipleSeedProducers:
183  ret.extend([seeder+m for m in multipleSeedProducers[i]])
184  else:
185  ret.append(seeder)
186 
187  for i in globals().get("_iterations_muonSeeded"+postfix, _iterations_muonSeeded):
188  if _modulePrefix(i).endswith("Step"):
189  ret.append(_modulePrefix(i)+typ)
190  else:
191  ret.append(_modulePrefix(i).replace("Step", typ))
192 
193  return ret
194 
def _seedOrTrackProducers(postfix, typ)
def replace(string, replacements)
def _modulePrefix(iteration)

◆ _tracks()

def iterativeTkConfig._tracks (   iteration)
private

Definition at line 139 of file iterativeTkConfig.py.

References _modulePrefix().

Referenced by clusterRemoverForIter().

139 def _tracks(iteration):
140  return _modulePrefix(iteration)+"Tracks"
141 
def _modulePrefix(iteration)
def _tracks(iteration)

◆ allEras()

def iterativeTkConfig.allEras ( )

Definition at line 152 of file iterativeTkConfig.py.

152 def allEras():
153  return _allEras
154 

◆ clusterRemoverForIter()

def iterativeTkConfig.clusterRemoverForIter (   iteration,
  eraName = "",
  postfix = "",
  module = None 
)

Definition at line 201 of file iterativeTkConfig.py.

References _classifier(), _clusterRemover(), and _tracks().

201 def clusterRemoverForIter(iteration, eraName="", postfix="", module=None):
202  if module is None:
203  module = _trackClusterRemoverBase.clone()
204  if eraName == "trackingPhase2PU140":
205  module = globals().get("_trackClusterRemoverBase"+postfix, _trackClusterRemoverBase)
206 
207  iters = globals()["_iterations"+postfix]
208  try:
209  # DisplacedRegionalStep is a special case because it comes after the
210  # usual muon-seeded steps
211  ind = iters.index(iteration) if iteration != "DisplacedRegionalStep" else len(iters)
212  except ValueError:
213  # if the iteration is not active in era, just return the same
214  return module
215 
216  if ind == 0:
217  raise Exception("Iteration %s is the first iteration in era %s, asking cluster remover configuration does not make sense" % (iteration, eraName))
218  prevIter = iters[ind-1]
219  # JetCoreRegionalStep uses all clusters, so if that is the previous
220  # iteration, use the one before that for cluster removal
221  if prevIter == "JetCoreRegionalStep":
222  prevIter = iters[ind-2]
223 
224  customize = dict(
225  trajectories = _tracks(prevIter),
226  oldClusterRemovalInfo = _clusterRemover(prevIter) if ind >= 2 else "", # 1st iteration does not have cluster remover
227  )
228  if eraName in ["trackingPhase2PU140"]:
229  customize["overrideTrkQuals"] = _classifier(prevIter, oldStyle=True) # old-style selector
230  elif eraName == "trackingLowPU":
231  customize["overrideTrkQuals"] = _classifier(prevIter, oldStyle=True, oldStyleQualityMasks=True) # old-style selector with 'QualityMasks' instance label
232  else:
233  customize["trackClassifier"] = _classifier(prevIter)
234 
235  return module.clone(**customize)
236 
def clusterRemoverForIter(iteration, eraName="", postfix="", module=None)
def _clusterRemover(iteration)
def _classifier(iteration, oldStyle=False, oldStyleQualityMasks=False)
def _tracks(iteration)

◆ createEarlyTask()

def iterativeTkConfig.createEarlyTask (   eraName,
  postfix,
  modDict 
)

Definition at line 158 of file iterativeTkConfig.py.

158 def createEarlyTask(eraName, postfix, modDict):
159  task = cms.Task()
160  for it in globals()["_iterations"+postfix]:
161  task.add(modDict[it+'Task'])
162  return task
163 
def createEarlyTask(eraName, postfix, modDict)

◆ iterationAlgos()

def iterativeTkConfig.iterationAlgos (   postfix,
  includeSequenceName = False 
)

Definition at line 164 of file iterativeTkConfig.py.

References _modulePrefix().

164 def iterationAlgos(postfix, includeSequenceName=False):
165  muonVariable = "_iterations_muonSeeded"+postfix
166  iterations = globals()["_iterations"+postfix] + globals().get(muonVariable, _iterations_muonSeeded)
167 
168  if includeSequenceName:
169  return [(_modulePrefix(i), i) for i in iterations]
170  else:
171  return [_modulePrefix(i) for i in iterations]
172 
def iterationAlgos(postfix, includeSequenceName=False)
def _modulePrefix(iteration)

◆ nonDefaultEras()

def iterativeTkConfig.nonDefaultEras ( )

Definition at line 155 of file iterativeTkConfig.py.

155 def nonDefaultEras():
156  return _nonDefaultEras
157 

◆ seedProducers()

def iterativeTkConfig.seedProducers (   postfix)

Definition at line 195 of file iterativeTkConfig.py.

References _seedOrTrackProducers().

195 def seedProducers(postfix):
196  return _seedOrTrackProducers(postfix, "Seeds")
197 
def _seedOrTrackProducers(postfix, typ)
def seedProducers(postfix)

◆ trackProducers()

def iterativeTkConfig.trackProducers (   postfix)

Definition at line 198 of file iterativeTkConfig.py.

References _seedOrTrackProducers().

198 def trackProducers(postfix):
199  return _seedOrTrackProducers(postfix, "Tracks")
200 
def _seedOrTrackProducers(postfix, typ)
def trackProducers(postfix)

Variable Documentation

◆ _allEras

iterativeTkConfig._allEras
private

Definition at line 16 of file iterativeTkConfig.py.

◆ _defaultEra

iterativeTkConfig._defaultEra
private

Definition at line 10 of file iterativeTkConfig.py.

◆ _defaultEraName

iterativeTkConfig._defaultEraName
private

Definition at line 6 of file iterativeTkConfig.py.

◆ _iterations

iterativeTkConfig._iterations
private

Definition at line 19 of file iterativeTkConfig.py.

◆ _iterations_muonSeeded

iterativeTkConfig._iterations_muonSeeded
private

Definition at line 66 of file iterativeTkConfig.py.

◆ _iterations_muonSeeded_trackingPhase1

iterativeTkConfig._iterations_muonSeeded_trackingPhase1
private

Definition at line 70 of file iterativeTkConfig.py.

◆ _iterations_muonSeeded_trackingPhase2PU140

iterativeTkConfig._iterations_muonSeeded_trackingPhase2PU140
private

Definition at line 75 of file iterativeTkConfig.py.

◆ _iterations_trackingLowPU

iterativeTkConfig._iterations_trackingLowPU
private

Definition at line 29 of file iterativeTkConfig.py.

◆ _iterations_trackingPhase1

iterativeTkConfig._iterations_trackingPhase1
private

Definition at line 38 of file iterativeTkConfig.py.

◆ _iterations_trackingPhase2PU140

iterativeTkConfig._iterations_trackingPhase2PU140
private

Definition at line 56 of file iterativeTkConfig.py.

◆ _multipleSeedProducers

iterativeTkConfig._multipleSeedProducers
private

Definition at line 79 of file iterativeTkConfig.py.

◆ _multipleSeedProducers_trackingLowPU

iterativeTkConfig._multipleSeedProducers_trackingLowPU
private

Definition at line 83 of file iterativeTkConfig.py.

◆ _multipleSeedProducers_trackingPhase1

iterativeTkConfig._multipleSeedProducers_trackingPhase1
private

Definition at line 86 of file iterativeTkConfig.py.

◆ _multipleSeedProducers_trackingPhase2PU140

iterativeTkConfig._multipleSeedProducers_trackingPhase2PU140
private

Definition at line 95 of file iterativeTkConfig.py.

◆ _nonDefaultEraNames

iterativeTkConfig._nonDefaultEraNames
private

Definition at line 7 of file iterativeTkConfig.py.

◆ _nonDefaultEras

iterativeTkConfig._nonDefaultEras
private

Definition at line 11 of file iterativeTkConfig.py.

◆ _oldStyleHasSelector

iterativeTkConfig._oldStyleHasSelector
private

Definition at line 96 of file iterativeTkConfig.py.

◆ _trackClusterRemoverBase

iterativeTkConfig._trackClusterRemoverBase
private

Definition at line 111 of file iterativeTkConfig.py.

◆ _trackClusterRemoverBase_trackingPhase2PU140

iterativeTkConfig._trackClusterRemoverBase_trackingPhase2PU140
private

Definition at line 125 of file iterativeTkConfig.py.

◆ func

iterativeTkConfig.func

Definition at line 52 of file iterativeTkConfig.py.

◆ maxChi2

iterativeTkConfig.maxChi2

Definition at line 112 of file iterativeTkConfig.py.

◆ minNumberOfLayersWithMeasBeforeFiltering

iterativeTkConfig.minNumberOfLayersWithMeasBeforeFiltering

Definition at line 116 of file iterativeTkConfig.py.

◆ phase2OTClusters

iterativeTkConfig.phase2OTClusters

Definition at line 128 of file iterativeTkConfig.py.

◆ phase2pixelClusters

iterativeTkConfig.phase2pixelClusters

Definition at line 127 of file iterativeTkConfig.py.

◆ pixelClusters

iterativeTkConfig.pixelClusters

Definition at line 113 of file iterativeTkConfig.py.

◆ stripClusters

iterativeTkConfig.stripClusters

Definition at line 114 of file iterativeTkConfig.py.

◆ TrackQuality

iterativeTkConfig.TrackQuality

Definition at line 115 of file iterativeTkConfig.py.