Inheritance diagram for upgradeWorkflowComponents.UpgradeWorkflowPremix:

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 1648 of file upgradeWorkflowComponents.py.

Member Function Documentation

◆ condition()

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

Definition at line 1705 of file upgradeWorkflowComponents.py.

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

     def condition(self, fragment, stepList, key, hasHarvest):
         if not 'PU' in key:
             return False
         if not any(y in key for y in ['2021', '2023', '2024', '2026']):
             return False
         if self.suffix.endswith("S1"):
             return "NuGun" in fragment
         return True

◆ setup_()

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

Definition at line 1649 of file upgradeWorkflowComponents.py.

     def setup_(self, step, stepName, stepDict, k, properties):
         # just copy steps
         stepDict[stepName][k] = merge([stepDict[step][k]])

◆ setupPU_()

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

Definition at line 1652 of file upgradeWorkflowComponents.py.

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

     def setupPU_(self, step, stepName, stepDict, k, properties):
         # setup for stage 1
         if "GenSim" in stepName:
             stepNamePmx = stepName.replace('GenSim','Premix')
             if not stepNamePmx in stepDict: stepDict[stepNamePmx] = {}
             stepDict[stepNamePmx][k] = merge([
                 {
                     '-s': 'GEN,SIM,DIGI:pdigi_valid',
                     '--datatier': 'PREMIX',
                     '--eventcontent': 'PREMIX',
                     '--procModifiers': 'premix_stage1'
                 },
                 stepDict[stepName][k]
             ])
             if "ProdLike" in self.suffix:
                 stepDict[stepNamePmx][k] = merge([{'-s': 'GEN,SIM,DIGI'},stepDict[stepNamePmx][k]])
         # setup for stage 2
         elif "Digi" in step or "Reco" in step:
             # go back to non-PU step version
             d = merge([stepDict[self.getStepName(step)][k]])
             if d is None: return
             if "Digi" in step:
                 tmpsteps = []
                 for s in d["-s"].split(","):
                     if s == "DIGI" or "DIGI:" in s:
                         tmpsteps.extend([s, "DATAMIX"])
                     else:
                         tmpsteps.append(s)
                 d = merge([{"-s"             : ",".join(tmpsteps),
                             "--datamix"      : "PreMix",
                             "--procModifiers": "premix_stage2"},
                            d])
                 # for combined stage1+stage2
                 if "_PMXS1S2" in self.suffix:
                     d = merge([digiPremixLocalPileup, d])
             elif "Reco" in step:
                 if "--procModifiers" in d:
                     d["--procModifiers"] += ",premix_stage2"
                 else:
                     d["--procModifiers"] = "premix_stage2"
             stepDict[stepName][k] = d
         # Increase the input file step number by one for Nano in combined stage1+stage2
         elif "Nano"==step:
             # go back to non-PU step version
             d = merge([stepDict[self.getStepName(step)][k]])
             if "--filein" in d:
                 filein = d["--filein"]
                 m = re.search("step(?P<ind>\d+)_", filein)
                 if m:
                     d["--filein"] = filein.replace(m.group(), "step%d_"%(int(m.group("ind"))+1))
             stepDict[stepName][k] = d
             # run2/3 WFs use Nano (not NanoPU) in PU WF
             stepDict[self.getStepName(step)][k] = merge([d])

◆ workflow_()

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

Definition at line 1713 of file upgradeWorkflowComponents.py.

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

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