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

Constructor & Destructor Documentation

◆ __init__()

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

Definition at line 794 of file upgradeWorkflowComponents.py.

794  def __init__(self, digi = {}, reco = {}, mini = {}, harvest = {}, **kwargs):
795  # adapt the parameters for the UpgradeWorkflow init method
796  super(PatatrackWorkflow, self).__init__(
797  steps = [
798  'Digi',
799  'DigiTrigger',
800  'Reco',
801  'RecoFakeHLT',
802  'HARVEST',
803  'HARVESTFakeHLT',
804  'RecoGlobal',
805  'HARVESTGlobal',
806  'RecoNano',
807  'RecoNanoFakeHLT',
808  'HARVESTNano',
809  'HARVESTNanoFakeHLT',
810  'MiniAOD',
811  'Nano',
812  'ALCA',
813  'ALCAPhase2'
814  ],
815  PU = [
816  'Digi',
817  'DigiTrigger',
818  'Reco',
819  'RecoFakeHLT',
820  'HARVEST',
821  'HARVESTFakeHLT',
822  'RecoGlobal',
823  'HARVESTGlobal',
824  'RecoNano',
825  'RecoNanoFakeHLT',
826  'HARVESTNano',
827  'HARVESTNanoFakeHLT',
828  'MiniAOD',
829  'Nano',
830  'ALCA',
831  'ALCAPhase2'
832  ],
833  **kwargs)
834  self.__digi = digi
835  self.__reco = reco
836  if 'DQM' in self.__reco:
837  self.__reco.update({
838  '--datatier': 'GEN-SIM-RECO,DQMIO',
839  '--eventcontent': 'RECOSIM,DQM'
840  })
841  self.__mini = mini
842  self.__harvest = harvest
843 
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 844 of file upgradeWorkflowComponents.py.

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

844  def condition(self, fragment, stepList, key, hasHarvest):
845  # select only a subset of the workflows
846  years = ['2021','2023','2024','2026']
847  fragments = ["TTbar_14","ZMM_14","ZEE_14","ZTT_14","NuGun","SingleMu","QCD_Pt15To7000_Flat"]
848  selected = [
849  (any(y in key for y in years) and ('FS' not in key) and any( f in fragment for f in fragments)),
850  (('HI' in key) and ('Hydjet' in fragment) and ("PixelOnly" in self.suffix) )
851  ]
852  result = any(selected) and hasHarvest
853 
854  return result
855 
bool any(const std::vector< T > &v, const T &what)
Definition: ECalSD.cc:36

◆ setup_()

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

Definition at line 856 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().

856  def setup_(self, step, stepName, stepDict, k, properties):
857  # skip ALCA and Nano steps (but not RecoNano or HARVESTNano for Run3)
858  if 'ALCA' in step or 'Nano'==step:
859  stepDict[stepName][k] = None
860  elif 'Digi' in step:
861  if self.__digi is None:
862  stepDict[stepName][k] = None
863  else:
864  stepDict[stepName][k] = merge([self.__digi, stepDict[step][k]])
865  elif 'Reco' in step:
866  if self.__reco is None:
867  stepDict[stepName][k] = None
868  else:
869  stepDict[stepName][k] = merge([self.__reco, stepDict[step][k]])
870  if 'Phase2' in stepDict[stepName][k]['--era']:
871  if 'DQM:@standardDQM+@ExtraHLT' in stepDict[stepName][k]['-s']:
872  stepDict[stepName][k]['-s'] = stepDict[stepName][k]['-s'].replace('DQM:@standardDQM+@ExtraHLT','DQM:@phase2')
873  if 'VALIDATION:@standardValidation' in stepDict[stepName][k]['-s']:
874  stepDict[stepName][k]['-s'] = stepDict[stepName][k]['-s'].replace('VALIDATION:@standardValidation','VALIDATION:@phase2Validation')
875 
876 
877  elif 'MiniAOD' in step:
878  if self.__mini is None:
879  stepDict[stepName][k] = None
880  else:
881  stepDict[stepName][k] = merge([self.__mini, stepDict[step][k]])
882  elif 'HARVEST' in step:
883  if self.__harvest is None:
884  stepDict[stepName][k] = None
885  else:
886  stepDict[stepName][k] = merge([self.__harvest, stepDict[step][k]])
887 
888 # Pixel-only quadruplets workflow running on CPU
889 # - HLT on CPU
890 # - Pixel-only reconstruction on CPU, with DQM and validation
891 # - harvesting
892 
893 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 894 of file upgradeWorkflowComponents.py.

◆ harvest

upgradeWorkflowComponents.PatatrackWorkflow.harvest
static

Definition at line 901 of file upgradeWorkflowComponents.py.

◆ offset

upgradeWorkflowComponents.PatatrackWorkflow.offset
static

Definition at line 905 of file upgradeWorkflowComponents.py.

◆ reco

upgradeWorkflowComponents.PatatrackWorkflow.reco
static

Definition at line 897 of file upgradeWorkflowComponents.py.

◆ suffix

upgradeWorkflowComponents.PatatrackWorkflow.suffix
static

Definition at line 904 of file upgradeWorkflowComponents.py.