StandardSequences/python/Eras.py

0001 from __future__ import print_function
0002 import FWCore.ParameterSet.Config as cms
0003 from  FWCore.ParameterSet.Config import ModifierChain,Modifier
0004
0005 class Eras (object):
0006     """
0007     Dummy container for all the cms.Modifier instances that config fragments
0008     can use to selectively configure depending on what scenario is active.
0009     """
0010     def __init__(self):
0011         allEras=['Run1_pA',
0012                  'Run1_peripheralPbPb',
0013                  'Run2_50ns',
0014                  'Run2_50ns_HIPM',
0015                  'Run2_25ns',
0016                  'Run2_25ns_HIPM',
0017                  'Run2_25ns_peripheralPbPb',
0018                  'Run2_HI',
0019                  'Run2_2016',
0020                  'Run2_2016_HIPM',
0021                  'Run2_2016_trackingLowPU',
0022                  'Run2_2016_pA',
0023                  'Run2_2017',
0024                  'Run2_2017_noMkFit',
0025                  'Run2_2017_FastSim', #new modifier for Phase1 FastSim, skips the muon GEM sequence
0026                  'Run2_2017_trackingRun2',
0027                  'Run2_2017_trackingLowPU',
0028                  'Run2_2017_pp_on_XeXe',
0029                  'Run2_2017_ppRef',
0030                  'Run2_2018',
0031                  'Run2_2018_FastSim', #new modifier for Phase1 FastSim, skips the muon GEM sequence
0032                  'Run2_2018_pp_on_AA',
0033                  'Run2_2018_pp_on_AA_noHCALmitigation',
0034                  'Run2_2018_highBetaStar',
0035                  'Run2_2018_noMkFit',
0036                  'Run3',
0037                  'Run3_2023',
0038                  'Run3_noMkFit',
0039                  'Run3_pp_on_PbPb',
0040                  'Run3_pp_on_PbPb_approxSiStripClusters',
0041                  'Run3_pp_on_PbPb_2023',
0042                  'Run3_pp_on_PbPb_approxSiStripClusters_2023',
0043                  'Run3_dd4hep',
0044                  'Run3_DDD',
0045                  'Run3_FastSim',
0046                  'Phase2',
0047                  'Phase2C9',
0048                  'Phase2C10',
0049                  'Phase2C11',
0050                  'Phase2C12',
0051                  'Phase2C9_dd4hep',
0052                  'Phase2C10_dd4hep',
0053                  'Phase2C11_dd4hep',
0054                  'Phase2C11I13',
0055                  'Phase2C12_dd4hep',
0056                  'Phase2C11M9',
0057                  'Phase2C11I13M9',
0058                  'Phase2C11I13T25M9',
0059                  'Phase2C11I13T26M9',
0060                  'Phase2C17I13M9',
0061                  'Phase2C20I13M9'
0062         ]
0063
0064         internalUseMods = ['run2_common', 'run2_25ns_specific',
0065                            'run2_50ns_specific', 'run2_HI_specific',
0066                            'stage1L1Trigger', 'fastSim',
0067                            'peripheralPbPb', 'pA_2016',
0068                            'run2_HE_2017', 'stage2L1Trigger', 'stage2L1Trigger_2017', 'stage2L1Trigger_2018', 'stage2L1Trigger_2021',
0069                            'run2_HF_2017', 'run2_HCAL_2017', 'run2_HEPlan1_2017', 'run2_HB_2018','run2_HE_2018',
0070                            'run3_HB', 'run3_HFSL', 'run3_common', 'run3_RPC',
0071                            'phase1Pixel', 'run3_GEM', 'run2_GEM_2017',
0072                            'run2_CSC_2018',
0073                            'phase2_common', 'phase2_tracker',
0074                            'phase2_muon', 'phase2_GEM', 'phase2_GE0',
0075                            'phase2_hgcal', 'phase2_timing', 'phase2_hfnose', 'phase2_hgcalV10', 'phase2_hgcalV11', 'phase2_hgcalV12',
0076                            'phase2_timing_layer', 'phase2_etlV4', 'phase2_hcal', 'phase2_ecal','phase2_ecal_devel',
0077                            'phase2_trigger',
0078                            'phase2_squarePixels', 'phase2_3DPixels',
0079                            'trackingLowPU', 'trackingPhase1',
0080                            'ctpps', 'ctpps_2016', 'ctpps_2017', 'ctpps_2018', 'ctpps_2022',
0081                            'trackingPhase2PU140','highBetaStar_2018',
0082                            'tracker_apv_vfp30_2016', 'pf_badHcalMitigationOff',
0083                            'run2_miniAOD_80XLegacy','run2_miniAOD_94XFall17',
0084                            'run2_nanoAOD_106Xv2',
0085                            'run3_nanoAOD_122', 'run3_nanoAOD_124',
0086                            'run3_ecal_devel',
0087                            'hcalHardcodeConditions', 'hcalSkipPacker',
0088                            'run2_HLTconditions_2016','run2_HLTconditions_2017','run2_HLTconditions_2018',
0089                            'bParking']
0090         internalUseModChains = ['run2_2017_noTrackingModifier', 'trackingMkFitProd']
0091
0092         self.pythonCfgLines = {}
0093
0094         for e in allEras:
0095             eObj=getattr(__import__('Configuration.Eras.Era_'+e+'_cff',globals(),locals(),[e],0),e)
0096             self.addEra(e,eObj)
0097             self.pythonCfgLines[e] = 'from Configuration.Eras.Era_'+e+'_cff import '+e
0098
0099         for e in internalUseMods:
0100             eObj=getattr(__import__('Configuration.Eras.Modifier_'+e+'_cff',globals(),locals(),[e],0),e)
0101             self.addEra(e,eObj)
0102             self.pythonCfgLines[e] = 'from Configuration.Eras.Modifier_'+e+'_cff import '+e
0103
0104         for e in internalUseModChains:
0105             eObj=getattr(__import__('Configuration.Eras.ModifierChain_'+e+'_cff',globals(),locals(),[e],0),e)
0106             self.addEra(e,eObj)
0107             self.pythonCfgLines[e] = 'from Configuration.Eras.ModifierChain_'+e+'_cff import '+e
0108
0109
0110     def addEra(self,name,obj):
0111         setattr(self,name,obj)
0112
0113     def inspectModifier(self,m,details):
0114         print('      ',m.__dict__ ['_Modifier__processModifiers'])
0115
0116     def inspectEra(self,e,details):
0117         print('\nEra:',e)
0118         print('   isChosen:',getattr(self,e)._isChosen())
0119         if details: print('   Modifiers:')
0120         nmod=0
0121         for value in getattr(self,e).__dict__['_ModifierChain__chain']:
0122             if isinstance(value, Modifier):
0123                 nmod=nmod+1
0124                 if details: self.inspectModifier(value,details)
0125         print('   ',nmod,'modifiers defined')
0126
0127     def inspect(self,name=None,onlyChosen=False,details=True):
0128         if name==None:
0129             print('Inspecting the known eras', end=' ')
0130             if onlyChosen: print(' (all active)')
0131             else: print('(all eras defined)')
0132         else:
0133             print('Inspecting the '+name+' era', end=' ')
0134
0135         allEras=[]
0136         for key, value in self.__dict__.items():
0137             if isinstance(value, ModifierChain): allEras.append(key)
0138
0139         for e in allEras:
0140             if name is not None and name==e:
0141                 self.inspectEra(e,details)
0142             if name is None:
0143                 if not onlyChosen or getattr(self,e).isChosen():
0144                     self.inspectEra(e,details)
0145
0146 eras=Eras()
0147
0148
0149 #eras.inspect()