TrackFindingTMTT/test/make_stubs.py

0001 import FWCore.ParameterSet.Config as cms
0002 import FWCore.Utilities.FileUtils as FileUtils
0003 import FWCore.ParameterSet.VarParsing as VarParsing
0004
0005 process = cms.Process("STUBS")
0006
0007 process.load('Configuration.Geometry.GeometryExtended2023D17Reco_cff')
0008 process.load('Configuration.Geometry.GeometryExtended2023D17_cff')
0009 process.load('Configuration.StandardSequences.MagneticField_cff')
0010 process.load('Configuration.EventContent.EventContent_cff')
0011 process.load('Configuration.StandardSequences.FrontierConditions_GlobalTag_cff')
0012 from Configuration.AlCa.GlobalTag import GlobalTag
0013 process.GlobalTag = GlobalTag(process.GlobalTag, 'auto:upgradePLS3', '')
0014
0015 process.load("FWCore.MessageLogger.MessageLogger_cfi")
0016
0017 options = VarParsing.VarParsing ('analysis')
0018
0019 #--- Specify input MC
0020 options.register('inputMC', '../../../Samples91X/930pre3/TTbar/PU200.txt', VarParsing.VarParsing.multiplicity.singleton, VarParsing.VarParsing.varType.string, "Files to be processed")
0021
0022 #--- Specify number of events to process.
0023 options.register('Events',1000,VarParsing.VarParsing.multiplicity.singleton, VarParsing.VarParsing.varType.int,"Number of Events to analyze")
0024
0025 options.parseArguments()
0026
0027 #--- input and output
0028 list = FileUtils.loadListFromFile(options.inputMC)
0029 readFiles = cms.untracked.vstring(*list)
0030
0031
0032 process.options   = cms.untracked.PSet( wantSummary = cms.untracked.bool(False) )
0033 process.maxEvents = cms.untracked.PSet( input = cms.untracked.int32(options.Events) )
0034
0035 process.source = cms.Source ("PoolSource",
0036                             fileNames = readFiles,
0037                             )
0038
0039 process.Timing = cms.Service("Timing", summaryOnly = cms.untracked.bool(True))
0040
0041 process.RAWSIMoutput = cms.OutputModule("PoolOutputModule",
0042     splitLevel = cms.untracked.int32(0),
0043     eventAutoFlushCompressedSize = cms.untracked.int32(5242880),
0044     outputCommands = process.RAWSIMEventContent.outputCommands,
0045     fileName = cms.untracked.string('output_PU200.root'), ## ADAPT IT ##
0046     dataset = cms.untracked.PSet(
0047         filterName = cms.untracked.string(''),
0048         dataTier = cms.untracked.string('GEN-SIM')
0049     )
0050 )
0051 process.RAWSIMoutput.outputCommands.append('keep  *_*_ClusterAccepted_*')
0052 process.RAWSIMoutput.outputCommands.append('keep  *_*_StubAccepted_*')
0053 process.RAWSIMoutput.outputCommands.append('keep  *_*_MergedTrackTruth_*')
0054 process.RAWSIMoutput_step    = cms.EndPath(process.RAWSIMoutput)
0055
0056
0057 process.load('L1Trigger.TrackTrigger.TrackTrigger_cff')
0058 process.load('SimTracker.TrackTriggerAssociation.TrackTriggerAssociator_cff')
0059 process.TTClusterAssociatorFromPixelDigis.digiSimLinks = cms.InputTag("simSiPixelDigis","Tracker")
0060 process.p = cms.Path(process.TrackTriggerClustersStubs * process.TrackTriggerAssociatorClustersStubs)
0061