@@ -1841,6 +1841,48 @@ def condition(self, fragment, stepList, key, hasHarvest):
1841
1841
offset = 0.612 ,
1842
1842
)
1843
1843
1844
+ # ECAL component
1845
+ class UpgradeWorkflow_ECalComponent (UpgradeWorkflow ):
1846
+ def __init__ (self , suffix , offset , ecalMod ,
1847
+ steps = [
1848
+ 'GenSim' ,
1849
+ 'GenSimHLBeamSpot' ,
1850
+ 'GenSimHLBeamSpot14' ,
1851
+ 'GenSimHLBeamSpotHGCALCloseBy' ,
1852
+ 'Digi' ,
1853
+ 'DigiTrigger' ,
1854
+ ],
1855
+ PU = [
1856
+ 'GenSim' ,
1857
+ 'GenSimHLBeamSpot' ,
1858
+ 'GenSimHLBeamSpot14' ,
1859
+ 'GenSimHLBeamSpotHGCALCloseBy' ,
1860
+ 'Digi' ,
1861
+ 'DigiTrigger' ,
1862
+ ]):
1863
+ super (UpgradeWorkflow_ECalComponent , self ).__init__ (steps , PU , suffix , offset )
1864
+ self .__ecalMod = ecalMod
1865
+
1866
+ def setup_ (self , step , stepName , stepDict , k , properties ):
1867
+ if 'Sim' in step or 'Digi' in step :
1868
+ if self .__ecalMod is not None :
1869
+ stepDict [stepName ][k ] = merge ([{'--procModifiers' :self .__ecalMod },stepDict [step ][k ]])
1870
+
1871
+ def condition (self , fragment , stepList , key , hasHarvest ):
1872
+ return ('2021' in key or '2023' in key or '2026' in key )
1873
+
1874
+ upgradeWFs ['ECALComponent' ] = UpgradeWorkflow_ECalComponent (
1875
+ suffix = '_ecalComponent' ,
1876
+ offset = 0.631 ,
1877
+ ecalMod = 'ecal_component' ,
1878
+ )
1879
+
1880
+ upgradeWFs ['ECALComponentFSW' ] = UpgradeWorkflow_ECalComponent (
1881
+ suffix = '_ecalComponentFSW' ,
1882
+ offset = 0.632 ,
1883
+ ecalMod = 'ecal_component_finely_sampled_waveforms' ,
1884
+ )
1885
+
1844
1886
class UpgradeWorkflow_0T (UpgradeWorkflow ):
1845
1887
def setup_ (self , step , stepName , stepDict , k , properties ):
1846
1888
myGT = stepDict [step ][k ]['--conditions' ]
0 commit comments