Update the scripts in SimG4CMS/Calo/test/python for Run3 and Run4 scenarios

Sunanda · Sunanda · commit be8f63267f1b · 2025-05-27T05:20:31.000+02:00
diff --git a/SimG4CMS/Calo/test/python/Gen_MinBias.py b/SimG4CMS/Calo/test/python/Gen_MinBias.py
@@ -4,7 +4,7 @@
 #
 ###############################################################################
 import FWCore.ParameterSet.Config as cms
-import os, sys, imp, re, random
+import os, sys, importlib, re, random
 import FWCore.ParameterSet.VarParsing as VarParsing
 
 ####################################################################
diff --git a/SimG4CMS/Calo/test/python/Gen_SingleMuon.py b/SimG4CMS/Calo/test/python/Gen_SingleMuon.py
@@ -4,7 +4,7 @@
 #
 ###############################################################################
 import FWCore.ParameterSet.Config as cms
-import os, sys, imp, re, random
+import os, sys, importlib, re, random
 import FWCore.ParameterSet.VarParsing as VarParsing
 
 ####################################################################
diff --git a/SimG4CMS/Calo/test/python/SimRun3_cfg.py b/SimG4CMS/Calo/test/python/SimRun3_cfg.py
@@ -8,7 +8,7 @@
 #
 ###############################################################################
 import FWCore.ParameterSet.Config as cms
-import os, sys, imp, re, random
+import os, sys, importlib, re, random
 import FWCore.ParameterSet.VarParsing as VarParsing
 
 ####################################################################
diff --git a/SimG4CMS/Calo/test/python/SimRun4_cfg.py b/SimG4CMS/Calo/test/python/SimRun4_cfg.py
@@ -1,24 +1,24 @@
 ###############################################################################
 # Way to use this:
-#   cmsRun Sim2026_cfg.py geometry=D98 type=DDD data=Muon
+#   cmsRun SimRun4_cfg.py geometry=D110 type=DDD data=Muon
 #
-#   Options for geometry: D98, D99
+#   Options for geometry: D110, D120
 #               type: DDD, DD4hep
 #               data: Muon, MinBias
 #
 ###############################################################################
 import FWCore.ParameterSet.Config as cms
-import os, sys, imp, re, random
+import os, sys, importlib, re, random
 import FWCore.ParameterSet.VarParsing as VarParsing
 
 ####################################################################
 ### SETUP OPTIONS
 options = VarParsing.VarParsing('standard')
 options.register('geometry',
-                 "D98",
+                 "D110",
                   VarParsing.VarParsing.multiplicity.singleton,
                   VarParsing.VarParsing.varType.string,
-                  "geometry of operations: D98, D99")
+                  "geometry of operations: D110, D120")
 options.register('type',
                  "DDD",
                   VarParsing.VarParsing.multiplicity.singleton,
@@ -38,25 +38,28 @@
 ####################################################################
 # Use the options
 
-from Configuration.Eras.Era_Phase2C17I13M9_cff import Phase2C17I13M9
+geomName = "Run4" + options.geometry
+import Configuration.Geometry.defaultPhase2ConditionsEra_cff as _settings
+GLOBAL_TAG, ERA = _settings.get_era_and_conditions(geomName)
+
 if (options.type == "DD4hep"):
     from Configuration.ProcessModifiers.dd4hep_cff import dd4hep
-    process = cms.Process('Sim2026',Phase2C17I13M9,dd4hep)
-    geomFile = "Configuration.Geometry.Geometry" + options.type +"Extended2026" + options.geometry + "Reco_cff"
+    process = cms.Process('SimRun4',ERA,dd4hep)
+    geomFile = "Configuration.Geometry.Geometry" + options.type +"Extended" + geomName + "Reco_cff"
 else:
-    process = cms.Process('Sim2026',Phase2C17I13M9)
-    geomFile = "Configuration.Geometry.GeometryExtended2026" + options.geometry + "Reco_cff"
+    process = cms.Process('SimRun4',ERA)
+    geomFile = "Configuration.Geometry.GeometryExtended" + geomName + "Reco_cff"
 
-globalTag = "auto:phase2_realistic_T25"
 inFile = "file:step0" + options.data + ".root"
 outFile = "file:step1" + options.type + options.geometry + options.data + ".root"
 tFile = "file:" + options.type + options.geometry + options.data + ".root"
 
-print("Geometry file: ", geomFile)
-print("Global Tag:    ", globalTag)
-print("Input file:    ", inFile)
-print("Output file:   ", outFile)
-print("Histogram file:", tFile)
+print("Geometry file:   ", geomFile)
+print("Global Tag Name: ", GLOBAL_TAG)
+print("Era Name:        ", ERA)
+print("Input file:      ", inFile)
+print("Output file:     ", outFile)
+print("Histogram file:  ", tFile)
 
 process.load(geomFile)
 process.load("SimGeneral.HepPDTESSource.pythiapdt_cfi")
@@ -67,7 +70,7 @@
 process.load("Configuration.StandardSequences.SimIdeal_cff")
 process.load("Configuration.StandardSequences.FrontierConditions_GlobalTag_cff")
 from Configuration.AlCa.GlobalTag import GlobalTag
-process.GlobalTag = GlobalTag(process.GlobalTag, globalTag, '')
+process.GlobalTag = GlobalTag(process.GlobalTag, GLOBAL_TAG, '')
 
 process.source = cms.Source("PoolSource",
     dropDescendantsOfDroppedBranches = cms.untracked.bool(False),
diff --git a/SimG4CMS/Calo/test/python/minbias.py b/SimG4CMS/Calo/test/python/minbias.py
@@ -1,11 +1,11 @@
 ###############################################################################
 # Way to use this:
 #   cmsRun minbias.py geometry=2016
-#   Options for geometry 2016, 2017, 2018, 2021, 2026, legacy
+#   Options for geometry 2016, 2017, 2018, 2021, Run4, legacy
 # 
 ###############################################################################
 import FWCore.ParameterSet.Config as cms
-import os, sys, imp, re
+import os, sys, importlib, re
 import FWCore.ParameterSet.VarParsing as VarParsing
 
 ####################################################################
@@ -15,7 +15,7 @@
                  "2021",
                   VarParsing.VarParsing.multiplicity.singleton,
                   VarParsing.VarParsing.varType.string,
-                  "geometry of operations: 2016, 2017, 2018, 2021, 2026, legacy")
+                  "geometry of operations: 2016, 2017, 2018, 2021, Run4, legacy")
 
 ### get and parse the command line arguments
  
@@ -48,7 +48,7 @@
     process = cms.Process('Sim',Run3_DDD)
     geomFile = "Configuration.Geometry.GeometryExtended" + options.geometry + "Reco_cff"
     globalTag = "auto:phase1_2022_realistic"
-elif (options.geometry == "2026"):
+elif (options.geometry == "Run4"):
     from Configuration.Eras.Era_Phase2C11M9_cff import Phase2C11M9
     process = cms.Process('Sim',Phase2C11M9)
     geomFile = "Configuration.Geometry.GeometryExtended" + options.geometry + "D110Reco_cff"
diff --git a/SimG4CMS/Calo/test/python/minbiasRun4_cfg.py b/SimG4CMS/Calo/test/python/minbiasRun4_cfg.py
@@ -2,11 +2,11 @@
 # Way to use this:
 #   cmsRun testHGCalParametersDDD_cfg.py type=V18
 #
-#   Options for type V16, V17, V17Shift, V18
+#   Options for type V16, V17, V17Shift, V18, V19
 #
 ###############################################################################
 import FWCore.ParameterSet.Config as cms
-import os, sys, imp, re
+import os, sys, importlib, re
 import FWCore.ParameterSet.VarParsing as VarParsing
 
 ####################################################################
@@ -16,23 +16,32 @@
                  "V18",
                   VarParsing.VarParsing.multiplicity.singleton,
                   VarParsing.VarParsing.varType.string,
-                  "type of operations: V16, V17, V17Shift, V18")
+                  "type of operations: V16, V17, V17Shift, V18, V19")
 
 ### get and parse the command line arguments
 options.parseArguments()
 print(options)
 
-from Configuration.Eras.Era_Phase2C17I13M9_cff import Phase2C17I13M9
-
-process = cms.Process("HGCalParametersTest",Phase2C17I13M9)
 
 if (options.type == "V18"):
+    from Configuration.Eras.Era_Phase2C22I13M9_cff import Phase2C22I13M9
+    process = cms.Process("HGCalParametersTest",Phase2C22I13M9)
+    geomFile = "Geometry.HGCalCommonData.testHGCal" + options.type + "Reco_cff"
+elif (options.type == "V19"):
+    from Configuration.Eras.Era_Phase2C22I13M9_cff import Phase2C22I13M9
+    process = cms.Process("HGCalParametersTest",Phase2C22I13M9)
     geomFile = "Geometry.HGCalCommonData.testHGCal" + options.type + "Reco_cff"
 elif (options.type == "V17Shift"):
+    from Configuration.Eras.Era_Phase2C17I13M9_cff import Phase2C17I13M9
+    process = cms.Process("HGCalParametersTest",Phase2C17I13M9)
     geomFile = "Geometry.HGCalCommonData.testHGCal" + options.type + "Reco_cff"
 elif (options.type == "V17"):
+    from Configuration.Eras.Era_Phase2C17I13M9_cff import Phase2C17I13M9
+    process = cms.Process("HGCalParametersTest",Phase2C17I13M9)
     geomFile = "Configuration.Geometry.GeometryExtendedRun4D110Reco_cff"
 else:
+    from Configuration.Eras.Era_Phase2C17I13M9_cff import Phase2C17I13M9
+    process = cms.Process("HGCalParametersTest",Phase2C17I13M9)
     geomFile = "Configuration.Geometry.GeometryExtendedRun4D100Reco_cff"
 
 outFile = "minbias" + options.type + ".root"
diff --git a/SimG4CMS/Calo/test/python/testHGCalGuardRing_cfg.py b/SimG4CMS/Calo/test/python/testHGCalGuardRing_cfg.py
@@ -1,11 +1,59 @@
+###############################################################################
+# Way to use this:
+#   cmsRun testHGCalGuardRing_cfg.py geometry=D110 type=DDD
+#
+#   Options for geometry: D104, D110, D116, D120
+#               type: DDD, DD4hep
+#
+###############################################################################
 import FWCore.ParameterSet.Config as cms
-from Configuration.Eras.Era_Phase2C17I13M9_cff import Phase2C17I13M9
+import os, sys, importlib, re, random
+import FWCore.ParameterSet.VarParsing as VarParsing
 
-process = cms.Process("HGCalGuardRingTest",Phase2C17I13M9)
+####################################################################
+### SETUP OPTIONS
+options = VarParsing.VarParsing('standard')
+options.register('geometry',
+                 "D110",
+                  VarParsing.VarParsing.multiplicity.singleton,
+                  VarParsing.VarParsing.varType.string,
+                  "geometry of operations: D104, D110, D116, D120")
+options.register('type',
+                 "DDD",
+                  VarParsing.VarParsing.multiplicity.singleton,
+                  VarParsing.VarParsing.varType.string,
+                  "type of operations: DDD, DD4hep")
+
+### get and parse the command line arguments
+options.parseArguments()
+
+print(options)
+
+####################################################################
+# Use the options
+
+geomName = "Run4" + options.geometry
+import Configuration.Geometry.defaultPhase2ConditionsEra_cff as _settings
+GLOBAL_TAG, ERA = _settings.get_era_and_conditions(geomName)
+
+if (options.type == "DD4hep"):
+    from Configuration.ProcessModifiers.dd4hep_cff import dd4hep
+    process = cms.Process('GuardRing',ERA,dd4hep)
+    geomFile = "Configuration.Geometry.Geometry" + options.type +"Extended" + geomName + "Reco_cff"
+else:
+    process = cms.Process('GuardRing',ERA)
+    geomFile = "Configuration.Geometry.GeometryExtended" + geomName + "Reco_cff"
+
+print("Geometry file:   ", geomFile)
+print("Global Tag Name: ", GLOBAL_TAG)
+print("Era Name:        ", ERA)
 
 process.load("SimGeneral.HepPDTESSource.pdt_cfi")
-process.load("Configuration.Geometry.GeometryExtendedRun4D110Reco_cff")
+process.load(geomFile)
 process.load('FWCore.MessageService.MessageLogger_cfi')
+process.load("Configuration.StandardSequences.FrontierConditions_GlobalTag_cff")
+from Configuration.AlCa.GlobalTag import GlobalTag
+process.GlobalTag = GlobalTag(process.GlobalTag, GLOBAL_TAG, '')
 
 if hasattr(process,'MessageLogger'):
     process.MessageLogger.HGCSim=dict()
diff --git a/SimG4CMS/Calo/test/python/testHGCalMouseBite_cfg.py b/SimG4CMS/Calo/test/python/testHGCalMouseBite_cfg.py
@@ -1,5 +1,5 @@
 import FWCore.ParameterSet.Config as cms
-import os, sys, imp, re
+import os, sys, importlib, re
 import FWCore.ParameterSet.VarParsing as VarParsing
 
 ####################################################################
@@ -9,25 +9,34 @@
                  "V18",
                   VarParsing.VarParsing.multiplicity.singleton,
                   VarParsing.VarParsing.varType.string,
-                  "type of operations: V16, V17, V17Shift, V18")
+                  "type of operations: V16, V17, V17Shift, V18, V19")
 
 ### get and parse the command line arguments
 options.parseArguments()
 print(options)
 
-from Configuration.Eras.Era_Phase2C17I13M9_cff import Phase2C17I13M9
-
-process = cms.Process("HGCalMouseBiteTest",Phase2C17I13M9)
 
 ####################################################################
 # Use the options
 if (options.type == "V18"):
+    from Configuration.Eras.Era_Phase2C22I13M9_cff import Phase2C22I13M9
+    process = cms.Process("HGCalMouseBiteTest",Phase2C22I13M9)
     geomFile = "Configuration.Geometry.GeometryExtendedRun4D104_cff"
+elif (options.type == "V19"):
+    from Configuration.Eras.Era_Phase2C22I13M9_cff import Phase2C22I13M9
+    process = cms.Process("HGCalMouseBiteTest",Phase2C22I13M9)
+    geomFile = "Configuration.Geometry.GeometryExtendedRun4D120_cff"
 elif (options.type == "V17Shift"):
+    from Configuration.Eras.Era_Phase2C17I13M9_cff import Phase2C17I13M9
+    process = cms.Process("HGCalMouseBiteTest",Phase2C17I13M9)
     geomFile = "Geometry.HGCalCommonData.testHGCal" + options.type + "Reco_cff"
 elif (options.type == "V16"):
+    from Configuration.Eras.Era_Phase2C17I13M9_cff import Phase2C17I13M9
+    process = cms.Process("HGCalMouseBiteTest",Phase2C17I13M9)
     geomFile = "Configuration.Geometry.GeometryExtendedRun4D100_cff"
 else:
+    from Configuration.Eras.Era_Phase2C17I13M9_cff import Phase2C17I13M9
+    process = cms.Process("HGCalMouseBiteTest",Phase2C17I13M9)
     geomFile = "Configuration.Geometry.GeometryExtendedRun4D110_cff"
 
 print("Geometry file: ", geomFile)
diff --git a/SimG4CMS/Calo/test/python/ttbar.py b/SimG4CMS/Calo/test/python/ttbar.py
@@ -1,11 +1,11 @@
 ###############################################################################
 # Way to use this:
 #   cmsRun ttbar.py geometry=2016
-#   Options for geometry 2016, 2017, 2018, 2021, 2026, legacy
+#   Options for geometry 2016, 2017, 2018, 2021, Run4, legacy
 # 
 ###############################################################################
 import FWCore.ParameterSet.Config as cms
-import os, sys, imp, re
+import os, sys, importlib, re
 import FWCore.ParameterSet.VarParsing as VarParsing
 
 ####################################################################
@@ -15,7 +15,7 @@
                  "2021",
                   VarParsing.VarParsing.multiplicity.singleton,
                   VarParsing.VarParsing.varType.string,
-                  "geometry of operations: 2016, 2017, 2018, 2021, 2026, legacy")
+                  "geometry of operations: 2016, 2017, 2018, 2021, Run4, legacy")
 
 ### get and parse the command line arguments
  
@@ -48,7 +48,7 @@
     process = cms.Process('Sim',Run3_DDD)
     geomFile = "Configuration.Geometry.GeometryExtended" + options.geometry + "Reco_cff"
     globalTag = "auto:phase1_2022_realistic"
-elif (options.geometry == "2026"):
+elif (options.geometry == "Run4"):
     from Configuration.Eras.Era_Phase2C11M9_cff import Phase2C11M9
     process = cms.Process('Sim',Phase2C11M9)
     geomFile = "Configuration.Geometry.GeometryExtended" + options.geometry + "D110Reco_cff"

Original file line number	Diff line number	Diff line change
`@@ -4,7 +4,7 @@`
`4`	`4`	`#`
`5`	`5`	`###############################################################################`
`6`	`6`	`import FWCore.ParameterSet.Config as cms`
`7`		`-import os, sys, imp, re, random`
	`7`	`+import os, sys, importlib, re, random`
`8`	`8`	`import FWCore.ParameterSet.VarParsing as VarParsing`
`9`	`9`
`10`	`10`	`####################################################################`
Original file line number	Diff line number	Diff line change
`@@ -8,7 +8,7 @@`
`8`	`8`	`#`
`9`	`9`	`###############################################################################`
`10`	`10`	`import FWCore.ParameterSet.Config as cms`
`11`		`-import os, sys, imp, re, random`
	`11`	`+import os, sys, importlib, re, random`
`12`	`12`	`import FWCore.ParameterSet.VarParsing as VarParsing`
`13`	`13`
`14`	`14`	`####################################################################`