Validation/Geometry/test/runP_Tracker.py - master - githubmirror/cmssw

Validation/Geometry/test/runP_Tracker.py
140 строк5 KB
Shahzad Malik Muzaffar
[DQM-GEOMETRY] py2/3 compatibility:drop use of __future__
22 ноя 2024, 20:18
22 ноя 2024, 20:181926e30
# In order to produce everything that you need in one go, use the command:
#
# for t in {'BeamPipe','Tracker','PixBar','PixFwdMinus','PixFwdPlus','TIB','TOB','TIDB','TIDF','TEC','TkStrct','InnerServices'}; do python3 runP_Tracker.py geom=XYZ label=$t >& /dev/null &; done

import FWCore.ParameterSet.Config as cms
from FWCore.ParameterSet.VarParsing import VarParsing
import sys, re

from FWCore.PythonFramework.CmsRun import CmsRun

process = cms.Process("PROD")

process.load("SimGeneral.HepPDTESSource.pythiapdt_cfi")

"""

 The default geometry is Extended2017Plan1. If a different geometry
 is needed, the appropriate flag has to be passed at command line,
 e.g.: cmsRun runP_Tracker_cfg.py geom="XYZ"

 The default component to be monitored is the Tracker. If other
 components need to be studied, they must be supplied, one at a time,
 at the command line, e.g.: python3 runP_Tracker.py
 label="XYZ"

"""

from Validation.Geometry.plot_utils import _LABELS2COMPS

process.load("FWCore.MessageLogger.MessageLogger_cfi")

_ALLOWED_LABELS = _LABELS2COMPS.keys()

options = VarParsing('analysis')
options.register('geom',             #name
                 'Extended2017Plan1',      #default value
                 VarParsing.multiplicity.singleton,   # kind of options
                 VarParsing.varType.string,           # type of option
                 "Select the geometry to be studied"  # help message
                )

options.register('label',         #name
                 'Tracker',              #default value
                 VarParsing.multiplicity.singleton,   # kind of options
                 VarParsing.varType.string,           # type of option
                 "Select the label to be used to create output files. Default to tracker. If multiple components are selected, it defaults to the join of all components, with '_' as separator."  # help message
                )

options.setDefault('inputFiles', ['file:single_neutrino_random.root'])

options.parseArguments()
# Option validation

process.MessageLogger = cms.Service(
    "MessageLogger",
    destinations   = cms.untracked.vstring('info'),
    categories = cms.untracked.vstring(['logMsg','MaterialBudget']),
    info = cms.untracked.PSet(
        threshold = cms.untracked.string('INFO'),
        filename = cms.untracked.string('Log_%s_%s' % (options.label,options.geom)),
        logMsg = cms.untracked.PSet(limit = cms.untracked.int32(-1))
        )
    )

if options.label not in _ALLOWED_LABELS:
    print("\n*** Error, '%s' not registered as a valid components to monitor." % options.label)
    print("Allowed components:", _ALLOWED_LABELS)
    raise RuntimeError("Unknown label")

_components = _LABELS2COMPS[options.label]

#
#Geometry
#
def _adaptToRun2(det):
  if det == 'PixelForwardZminus':
    det = det.replace('minus', 'Minus')
  elif det == 'PixelForwardZplus':
    det = det.replace('plus', 'Plus')
  return det

# Load geometry either from the Database of from files
process.load("Configuration.Geometry.Geometry%sReco_cff" % options.geom)

# Customise names for Run2
if re.match('.*2016.*', options.geom):
  if isinstance(_components, list):
      for i in range(len(_components)):
          _components[i] = _adaptToRun2(_components[i])
  else:
      _components = _adaptToRun2(_components)

#
#Magnetic Field
#
process.load("Configuration.StandardSequences.MagneticField_38T_cff")

# Output of events, etc...
#
# Explicit note : since some histos/tree might be dumped directly,
#                 better NOT use PoolOutputModule !
# Detector simulation (Geant4-based)
#
process.load("SimG4Core.Application.g4SimHits_cfi")

process.load("IOMC.RandomEngine.IOMC_cff")
process.RandomNumberGeneratorService.g4SimHits.initialSeed = 9876

process.source = cms.Source("PoolSource",
    fileNames = cms.untracked.vstring(options.inputFiles)
)

process.maxEvents = cms.untracked.PSet(
    input = cms.untracked.int32(-1)
)

process.p1 = cms.Path(process.g4SimHits)
process.g4SimHits.StackingAction.TrackNeutrino = cms.bool(True)
process.g4SimHits.UseMagneticField = False
process.g4SimHits.Physics.type = 'SimG4Core/Physics/DummyPhysics'
process.g4SimHits.Physics.DummyEMPhysics = True
process.g4SimHits.Physics.CutsPerRegion = False
process.g4SimHits.Watchers = cms.VPSet(cms.PSet(
    type = cms.string('MaterialBudgetAction'),
    MaterialBudgetAction = cms.PSet(
        HistosFile = cms.string('matbdg_%s_%s.root' % (options.label,
                                                       options.geom)),
        AllStepsToTree = cms.bool(True),
        HistogramList = cms.string('Tracker'),
        SelectedVolumes = cms.vstring(_components),
        TreeFile = cms.string('matbdg_tree_%s_%s.root' % (options.label,
                                                          options.geom)),
        StopAfterProcess = cms.string('None'),
        TextFile = cms.string('matbdg_%s_%s.txt' % (options.label,
                                                     options.geom))
    )
))

cmsRun = CmsRun(process)
cmsRun.run()