TrackingMCTruth/test/simDoubletsPhase2_TEST.py

0001 """
0002 This script runs the SimDoubletsProducer and SimDoubletsAnalyzer for Phase2.
0003
0004 To run it, you first have to produce or get the output root file from a step 2 that runs the HLT.
0005 The input file is expected to be named `step2.root` by default, but you can also rename it below.
0006 Then you can simply run the config using:
0007
0008 cmsRun simDoubletsPhase2_TEST.py
0009
0010 It will produce one DQMIO output file named `simDoublets_DQMIO.root`.
0011 This can be further processed in the harvesting step by running the simDoubletsPhase2_HARVESTING.py script.
0012 """
0013 inputFile = "step2.root"
0014
0015 import FWCore.ParameterSet.Config as cms
0016
0017 from Configuration.Eras.Era_Phase2C17I13M9_cff import Phase2C17I13M9
0018
0019 process = cms.Process("SIMDOUBLETS",Phase2C17I13M9)
0020
0021 process.maxEvents = cms.untracked.PSet(
0022     input = cms.untracked.int32(-1),
0023     output = cms.optional.untracked.allowed(cms.int32,cms.PSet)
0024 )
0025
0026 # Input source
0027 process.source = cms.Source("PoolSource",
0028     dropDescendantsOfDroppedBranches = cms.untracked.bool(False),
0029     fileNames = cms.untracked.vstring('file:%s' % inputFile),
0030     inputCommands = cms.untracked.vstring(
0031         'keep *',
0032         'drop *_hltSiPixelRecHits_*_*'   # we will reproduce the PixelRecHits to have their local position available
0033     ),
0034     secondaryFileNames = cms.untracked.vstring()
0035 )
0036
0037 ### conditions
0038 process.load('Configuration.StandardSequences.FrontierConditions_GlobalTag_cff')
0039 from Configuration.AlCa.GlobalTag import GlobalTag
0040 process.GlobalTag = GlobalTag(process.GlobalTag, 'auto:phase2_realistic_T33', '')
0041
0042 ### standard includes
0043 process.load('Configuration/StandardSequences/Services_cff')
0044 process.load('Configuration.Geometry.GeometryExtendedRun4D110Reco_cff')
0045 process.load("Configuration.StandardSequences.RawToDigi_cff")
0046 process.load("Configuration.StandardSequences.Reconstruction_cff")
0047 process.load("Configuration.EventContent.EventContent_cff")
0048 process.load("Configuration.StandardSequences.MagneticField_cff")
0049 process.load('Configuration.StandardSequences.EndOfProcess_cff')
0050
0051 ### load hltSiPixelRecHits
0052 process.load("HLTrigger.Configuration.HLT_75e33.modules.hltSiPixelRecHits_cfi")
0053 ### load hltTPClusterProducer
0054 process.load("Validation.RecoTrack.associators_cff")
0055 ### load the new EDProducer "SimDoubletsProducerPhase2"
0056 process.load("SimTracker.TrackerHitAssociation.simDoubletsProducerPhase2_cfi")
0057 ### load the new DQM EDAnalyzer "SimDoubletsAnalyzerPhase2"
0058 process.load("Validation.TrackingMCTruth.simDoubletsAnalyzerPhase2_cfi")
0059
0060 ####  set up the paths
0061 process.simDoubletPath = cms.Path(
0062     process.hltSiPixelRecHits *         # reproduce the SiPixelRecHits
0063     process.hltTPClusterProducer *      # run the cluster to TrackingParticle association
0064     process.simDoubletsProducerPhase2 * # produce the SimDoublets
0065     process.simDoubletsAnalyzerPhase2   # analyze the SimDoublets
0066 )
0067
0068 # Output definition
0069 process.DQMoutput = cms.OutputModule("DQMRootOutputModule",
0070     dataset = cms.untracked.PSet(
0071         dataTier = cms.untracked.string('DQMIO'),
0072         filterName = cms.untracked.string('')
0073     ),
0074     fileName = cms.untracked.string('file:simDoublets_DQMIO.root'),
0075     outputCommands = process.DQMEventContent.outputCommands,
0076     splitLevel = cms.untracked.int32(0)
0077 )
0078
0079
0080 process.endjob_step = cms.EndPath(process.endOfProcess)
0081 process.DQMoutput_step = cms.EndPath(process.DQMoutput)
0082
0083
0084 process.schedule = cms.Schedule(
0085       process.simDoubletPath,process.endjob_step,process.DQMoutput_step
0086 )
0087
0088 process.options = cms.untracked.PSet(
0089     numberOfThreads = cms.untracked.uint32(1),
0090     numberOfStreams = cms.untracked.uint32(1),
0091     wantSummary = cms.untracked.bool(True)
0092 )