CMS 3D CMS Logo

List of all members | Public Member Functions | Static Public Attributes | Private Attributes
upgradeWorkflowComponents.PatatrackWorkflow Class Reference
Inheritance diagram for upgradeWorkflowComponents.PatatrackWorkflow:
upgradeWorkflowComponents.UpgradeWorkflow

Public Member Functions

def __init__ (self, digi={}, reco={}, mini={}, harvest={}, kwargs)
 
def condition (self, fragment, stepList, key, hasHarvest)
 
def setup_ (self, step, stepName, stepDict, k, properties)
 
- 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

 digi
 
 harvest
 
 offset
 
 reco
 
 suffix
 

Private Attributes

 __digi
 
 __harvest
 
 __mini
 
 __reco
 

Additional Inherited Members

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

Detailed Description

Definition at line 871 of file upgradeWorkflowComponents.py.

Constructor & Destructor Documentation

◆ __init__()

def upgradeWorkflowComponents.PatatrackWorkflow.__init__ (   self,
  digi = {},
  reco = {},
  mini = {},
  harvest = {},
  kwargs 
)

Definition at line 872 of file upgradeWorkflowComponents.py.

872  def __init__(self, digi = {}, reco = {}, mini = {}, harvest = {}, **kwargs):
873  # adapt the parameters for the UpgradeWorkflow init method
874  super(PatatrackWorkflow, self).__init__(
875  steps = [
876  'Digi',
877  'HLTOnly',
878  'DigiTrigger',
879  'Reco',
880  'RecoFakeHLT',
881  'HARVEST',
882  'HARVESTFakeHLT',
883  'RecoGlobal',
884  'HARVESTGlobal',
885  'RecoNano',
886  'RecoNanoFakeHLT',
887  'HARVESTNano',
888  'HARVESTNanoFakeHLT',
889  'MiniAOD',
890  'Nano',
891  'ALCA',
892  'ALCAPhase2'
893  ],
894  PU = [
895  'Digi',
896  'HLTOnly',
897  'DigiTrigger',
898  'Reco',
899  'RecoFakeHLT',
900  'HARVEST',
901  'HARVESTFakeHLT',
902  'RecoGlobal',
903  'HARVESTGlobal',
904  'RecoNano',
905  'RecoNanoFakeHLT',
906  'HARVESTNano',
907  'HARVESTNanoFakeHLT',
908  'MiniAOD',
909  'Nano',
910  'ALCA',
911  'ALCAPhase2'
912  ],
913  **kwargs)
914  self.__digi = digi
915  self.__reco = reco
916  if 'DQM' in self.__reco:
917  self.__reco.update({
918  '--datatier': 'GEN-SIM-RECO,DQMIO',
919  '--eventcontent': 'RECOSIM,DQM'
920  })
921  self.__mini = mini
922  self.__harvest = harvest
923 
def __init__(self, dataset, job_number, job_id, job_name, isDA, isMC, applyBOWS, applyEXTRACOND, extraconditions, runboundary, lumilist, intlumi, maxevents, gt, allFromGT, alignmentDB, alignmentTAG, apeDB, apeTAG, bowDB, bowTAG, vertextype, tracktype, refittertype, ttrhtype, applyruncontrol, ptcut, CMSSW_dir, the_dir)
#define update(a, b)

Member Function Documentation

◆ condition()

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

Definition at line 924 of file upgradeWorkflowComponents.py.

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

924  def condition(self, fragment, stepList, key, hasHarvest):
925  # select only a subset of the workflows
926  years = ['2021','2023','2024','2026']
927  fragments = ["TTbar_14","ZMM_14","ZEE_14","ZTT_14","NuGun","SingleMu","QCD_Pt15To7000_Flat"]
928  selected = [
929  (any(y in key for y in years) and ('FS' not in key) and any( f in fragment for f in fragments)),
930  (('HI' in key) and ('Hydjet' in fragment) and ("PixelOnly" in self.suffix) )
931  ]
932  result = any(selected) and hasHarvest
933 
934  return result
935 
bool any(const std::vector< T > &v, const T &what)
Definition: ECalSD.cc:37

◆ setup_()

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

Definition at line 936 of file upgradeWorkflowComponents.py.

References upgradeWorkflowComponents.PatatrackWorkflow.__digi, upgradeWorkflowComponents.UpgradeWorkflow_weightedVertex.__harvest, upgradeWorkflowComponents.PatatrackWorkflow.__harvest, upgradeWorkflowComponents.PatatrackWorkflow.__mini, upgradeWorkflowComponents.UpgradeWorkflow_weightedVertex.__reco, upgradeWorkflowComponents.PatatrackWorkflow.__reco, and python.rootplot.root2matplotlib.replace().

936  def setup_(self, step, stepName, stepDict, k, properties):
937  # skip ALCA and Nano steps (but not RecoNano or HARVESTNano for Run3)
938  if 'ALCA' in step or 'Nano'==step:
939  stepDict[stepName][k] = None
940  elif ('Digi' in step and "NoHLT" not in step) or 'HLTOnly' in step:
941  if self.__digi is None:
942  stepDict[stepName][k] = None
943  else:
944  stepDict[stepName][k] = merge([self.__digi, stepDict[step][k]])
945  elif 'Reco' in step:
946  if self.__reco is None:
947  stepDict[stepName][k] = None
948  else:
949  stepDict[stepName][k] = merge([self.__reco, stepDict[step][k]])
950  if 'Phase2' in stepDict[stepName][k]['--era']:
951  if 'DQM:@standardDQM+@ExtraHLT' in stepDict[stepName][k]['-s']:
952  stepDict[stepName][k]['-s'] = stepDict[stepName][k]['-s'].replace('DQM:@standardDQM+@ExtraHLT','DQM:@phase2')
953  if 'VALIDATION:@standardValidation' in stepDict[stepName][k]['-s']:
954  stepDict[stepName][k]['-s'] = stepDict[stepName][k]['-s'].replace('VALIDATION:@standardValidation','VALIDATION:@phase2Validation')
955 
956 
957  elif 'MiniAOD' in step:
958  if self.__mini is None:
959  stepDict[stepName][k] = None
960  else:
961  stepDict[stepName][k] = merge([self.__mini, stepDict[step][k]])
962  elif 'HARVEST' in step:
963  if self.__harvest is None:
964  stepDict[stepName][k] = None
965  else:
966  stepDict[stepName][k] = merge([self.__harvest, stepDict[step][k]])
967 
968 # Pixel-only quadruplets workflow running on CPU
969 # - HLT on CPU
970 # - Pixel-only reconstruction on CPU, with DQM and validation
971 # - harvesting
972 
973 upgradeWFs['PatatrackPixelOnlyCPU'] = PatatrackWorkflow(
Definition: merge.py:1
def replace(string, replacements)

Member Data Documentation

◆ __digi

upgradeWorkflowComponents.PatatrackWorkflow.__digi
private

◆ __harvest

upgradeWorkflowComponents.PatatrackWorkflow.__harvest
private

◆ __mini

upgradeWorkflowComponents.PatatrackWorkflow.__mini
private

◆ __reco

upgradeWorkflowComponents.PatatrackWorkflow.__reco
private

◆ digi

upgradeWorkflowComponents.PatatrackWorkflow.digi
static

Definition at line 974 of file upgradeWorkflowComponents.py.

◆ harvest

upgradeWorkflowComponents.PatatrackWorkflow.harvest
static

Definition at line 981 of file upgradeWorkflowComponents.py.

◆ offset

upgradeWorkflowComponents.PatatrackWorkflow.offset
static

Definition at line 985 of file upgradeWorkflowComponents.py.

◆ reco

upgradeWorkflowComponents.PatatrackWorkflow.reco
static

Definition at line 977 of file upgradeWorkflowComponents.py.

◆ suffix

upgradeWorkflowComponents.PatatrackWorkflow.suffix
static

Definition at line 984 of file upgradeWorkflowComponents.py.