CMS 3D CMS Logo

List of all members | Public Member Functions | Static Public Attributes
upgradeWorkflowComponents.UpgradeWorkflowPremix Class Reference
Inheritance diagram for upgradeWorkflowComponents.UpgradeWorkflowPremix:
upgradeWorkflowComponents.UpgradeWorkflow upgradeWorkflowComponents.UpgradeWorkflowAdjustPU upgradeWorkflowComponents.UpgradeWorkflowPremixProdLike

Public Member Functions

def condition (self, fragment, stepList, key, hasHarvest)
 
def setup_ (self, step, stepName, stepDict, k, properties)
 
def setupPU_ (self, step, stepName, stepDict, k, properties)
 
def workflow_ (self, workflows, num, fragment, stepList, key)
 
- Public Member Functions inherited from upgradeWorkflowComponents.UpgradeWorkflow
def __init__ (self, steps, PU, suffix, offset)
 
def condition (self, fragment, stepList, key, hasHarvest)
 
def getStepName (self, step, extra="")
 
def getStepNamePU (self, step, extra="")
 
def init (self, stepDict)
 
def preventReuse (self, stepName, stepDict, k)
 
def setup (self, stepDict, k, properties)
 
def setup_ (self, step, stepName, stepDict, k, properties)
 
def setupPU (self, stepDict, k, properties)
 
def setupPU_ (self, step, stepName, stepDict, k, properties)
 
def workflow (self, workflows, num, fragment, stepList, key, hasHarvest)
 
def workflow_ (self, workflows, num, fragment, stepList, key)
 

Static Public Attributes

 offset
 
 PU
 
 steps
 
 suffix
 

Additional Inherited Members

- Public Attributes inherited from upgradeWorkflowComponents.UpgradeWorkflow
 allowReuse
 
 offset
 
 PU
 
 steps
 
 suffix
 

Detailed Description

Definition at line 2410 of file upgradeWorkflowComponents.py.

Member Function Documentation

◆ condition()

def upgradeWorkflowComponents.UpgradeWorkflowPremix.condition (   self,
  fragment,
  stepList,
  key,
  hasHarvest 
)

Definition at line 2467 of file upgradeWorkflowComponents.py.

References any(), and upgradeWorkflowComponents.UpgradeWorkflow.suffix.

2467  def condition(self, fragment, stepList, key, hasHarvest):
2468  if not 'PU' in key:
2469  return False
2470  if not any(y in key for y in ['2021', '2023', '2024', '2026']):
2471  return False
2472  if self.suffix.endswith("S1"):
2473  return "NuGun" in fragment
2474  return True
bool any(const std::vector< T > &v, const T &what)
Definition: ECalSD.cc:36

◆ setup_()

def upgradeWorkflowComponents.UpgradeWorkflowPremix.setup_ (   self,
  step,
  stepName,
  stepDict,
  k,
  properties 
)

Definition at line 2411 of file upgradeWorkflowComponents.py.

2411  def setup_(self, step, stepName, stepDict, k, properties):
2412  # just copy steps
2413  stepDict[stepName][k] = merge([stepDict[step][k]])
Definition: merge.py:1

◆ setupPU_()

def upgradeWorkflowComponents.UpgradeWorkflowPremix.setupPU_ (   self,
  step,
  stepName,
  stepDict,
  k,
  properties 
)

Definition at line 2414 of file upgradeWorkflowComponents.py.

References upgradeWorkflowComponents.UpgradeWorkflow.getStepName(), createfilelist.int, join(), submitPVValidationJobs.split(), and upgradeWorkflowComponents.UpgradeWorkflow.suffix.

2414  def setupPU_(self, step, stepName, stepDict, k, properties):
2415  # setup for stage 1
2416  if "GenSim" in stepName:
2417  stepNamePmx = stepName.replace('GenSim','Premix')
2418  if not stepNamePmx in stepDict: stepDict[stepNamePmx] = {}
2419  stepDict[stepNamePmx][k] = merge([
2420  {
2421  '-s': 'GEN,SIM,DIGI:pdigi_valid',
2422  '--datatier': 'PREMIX',
2423  '--eventcontent': 'PREMIX',
2424  '--procModifiers': 'premix_stage1'
2425  },
2426  stepDict[stepName][k]
2427  ])
2428  if "ProdLike" in self.suffix:
2429  stepDict[stepNamePmx][k] = merge([{'-s': 'GEN,SIM,DIGI'},stepDict[stepNamePmx][k]])
2430  # setup for stage 2
2431  elif "Digi" in step or "Reco" in step:
2432  # go back to non-PU step version
2433  d = merge([stepDict[self.getStepName(step)][k]])
2434  if d is None: return
2435  if "Digi" in step:
2436  tmpsteps = []
2437  for s in d["-s"].split(","):
2438  if s == "DIGI" or "DIGI:" in s:
2439  tmpsteps.extend([s, "DATAMIX"])
2440  else:
2441  tmpsteps.append(s)
2442  d = merge([{"-s" : ",".join(tmpsteps),
2443  "--datamix" : "PreMix",
2444  "--procModifiers": "premix_stage2"},
2445  d])
2446  # for combined stage1+stage2
2447  if "_PMXS1S2" in self.suffix:
2448  d = merge([digiPremixLocalPileup, d])
2449  elif "Reco" in step:
2450  if "--procModifiers" in d:
2451  d["--procModifiers"] += ",premix_stage2"
2452  else:
2453  d["--procModifiers"] = "premix_stage2"
2454  stepDict[stepName][k] = d
2455  # Increase the input file step number by one for Nano in combined stage1+stage2
2456  elif "Nano"==step:
2457  # go back to non-PU step version
2458  d = merge([stepDict[self.getStepName(step)][k]])
2459  if "--filein" in d:
2460  filein = d["--filein"]
2461  m = re.search("step(?P<ind>\d+)_", filein)
2462  if m:
2463  d["--filein"] = filein.replace(m.group(), "step%d_"%(int(m.group("ind"))+1))
2464  stepDict[stepName][k] = d
2465  # run2/3 WFs use Nano (not NanoPU) in PU WF
2466  stepDict[self.getStepName(step)][k] = merge([d])
Definition: merge.py:1
static std::string join(char **cmd)
Definition: RemoteFile.cc:19

◆ workflow_()

def upgradeWorkflowComponents.UpgradeWorkflowPremix.workflow_ (   self,
  workflows,
  num,
  fragment,
  stepList,
  key 
)

Definition at line 2475 of file upgradeWorkflowComponents.py.

References python.rootplot.root2matplotlib.replace(), and upgradeWorkflowComponents.UpgradeWorkflow.suffix.

2475  def workflow_(self, workflows, num, fragment, stepList, key):
2476  fragmentTmp = fragment
2477  if self.suffix.endswith("S1"):
2478  fragmentTmp = 'PREMIXUP' + key[2:].replace("PU", "").replace("Design", "") + '_PU25'
2479  super(UpgradeWorkflowPremix,self).workflow_(workflows, num, fragmentTmp, stepList, key)
2480 # Premix stage1
2481 upgradeWFs['PMXS1'] = UpgradeWorkflowPremix(
def replace(string, replacements)

Member Data Documentation

◆ offset

upgradeWorkflowComponents.UpgradeWorkflowPremix.offset
static

Definition at line 2490 of file upgradeWorkflowComponents.py.

◆ PU

upgradeWorkflowComponents.UpgradeWorkflowPremix.PU
static

Definition at line 2484 of file upgradeWorkflowComponents.py.

◆ steps

upgradeWorkflowComponents.UpgradeWorkflowPremix.steps
static

Definition at line 2482 of file upgradeWorkflowComponents.py.

◆ suffix

upgradeWorkflowComponents.UpgradeWorkflowPremix.suffix
static

Definition at line 2489 of file upgradeWorkflowComponents.py.