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

Member Function Documentation

◆ condition()

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

Definition at line 2591 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 2490 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 2493 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):
         # fastsim version
         if 'FS' in k:
             # setup for stage 1 fastsim
             if "Gen" in stepName:
                 stepNamePmx = stepName.replace('Gen','Premix')
                 if not stepNamePmx in stepDict: stepDict[stepNamePmx] = {}
                 stepDict[stepNamePmx][k] = merge([
                     {
                         '-s': 'GEN,SIM,RECOBEFMIX,DIGI:pdigi_valid',
                         '--fast':'',
                         '--datatier': 'PREMIX',
                         '--eventcontent': 'PREMIX',
                         '--procModifiers': 'premix_stage1'
                     },
                     stepDict[stepName][k]
                 ])
                 if "ProdLike" in self.suffix:
                     # todo
                     pass
             # setup for stage 2 fastsim
             elif "FastSimRun3" in step:
                 # go back to non-PU step version
                 d = merge([stepDict[self.getStepName(step)][k]])
                 if d is None: return
                 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"},
                            d])
                 if "--procModifiers" in d:
                     d["--procModifiers"] += ",premix_stage2"
                 else:
                     d["--procModifiers"] = "premix_stage2"
                 # for combined stage1+stage2
                 if "_PMXS1S2" in self.suffix:
                     d = merge([digiPremixLocalPileup, d])
                 stepDict[stepName][k] = d
             elif "HARVESTFastRun3" in step:
                 # increment input step number
                 stepDict[stepName][k] = merge([{'--filein':'file:step3_inDQM.root'},stepDict[stepName][k]])
         else:
             # 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
             # separate nano step now only used in ProdLike workflows for Run3/Phase2
             elif "Nano"==step:
                 # go back to non-PU step version
                 d = merge([stepDict[self.getStepName(step)][k]])
                 if "_PMXS1S2" in self.suffix and "--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 2599 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(