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

Constructor & Destructor Documentation

◆ __init__()

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

Definition at line 796 of file upgradeWorkflowComponents.py.

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

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

846  def condition(self, fragment, stepList, key, hasHarvest):
847  # select only a subset of the workflows
848  years = ['2021','2023','2024','2026']
849  fragments = ["TTbar_14","ZMM_14","ZEE_14","ZTT_14","NuGun","SingleMu","QCD_Pt15To7000_Flat"]
850  selected = [
851  (any(y in key for y in years) and ('FS' not in key) and any( f in fragment for f in fragments)),
852  (('HI' in key) and ('Hydjet' in fragment) and ("PixelOnly" in self.suffix) )
853  ]
854  result = any(selected) and hasHarvest
855 
856  return result
857 
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 858 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().

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

◆ harvest

upgradeWorkflowComponents.PatatrackWorkflow.harvest
static

Definition at line 903 of file upgradeWorkflowComponents.py.

◆ offset

upgradeWorkflowComponents.PatatrackWorkflow.offset
static

Definition at line 907 of file upgradeWorkflowComponents.py.

◆ reco

upgradeWorkflowComponents.PatatrackWorkflow.reco
static

Definition at line 899 of file upgradeWorkflowComponents.py.

◆ suffix

upgradeWorkflowComponents.PatatrackWorkflow.suffix
static

Definition at line 906 of file upgradeWorkflowComponents.py.