TrackerTFP/test/test_cfg.py

0001 ################################################################################################
0002 # Run bit-accurate TMTT L1 tracking emulation.
0003 #
0004 # To run execute do
0005 # cmsRun L1Trigger/L1TTrackerTFP/test/test_cfg.py
0006 # where the arguments take default values if you don't specify them. You can change defaults below.
0007 #################################################################################################
0008
0009 import FWCore.ParameterSet.Config as cms
0010
0011 process = cms.Process( "Demo" )
0012 process.load( 'FWCore.MessageService.MessageLogger_cfi' )
0013 process.load( 'Configuration.Geometry.GeometryExtended2026D76Reco_cff' )
0014 process.load( 'Configuration.Geometry.GeometryExtended2026D76_cff' )
0015 process.load( 'Configuration.StandardSequences.MagneticField_cff' )
0016 process.load( 'Configuration.StandardSequences.FrontierConditions_GlobalTag_cff' )
0017 process.load( 'L1Trigger.TrackTrigger.TrackTrigger_cff' )
0018
0019 from Configuration.AlCa.GlobalTag import GlobalTag
0020 process.GlobalTag = GlobalTag(process.GlobalTag, 'auto:phase2_realistic', '')
0021
0022 # load code that associates stubs with mctruth
0023 process.load( 'SimTracker.TrackTriggerAssociation.StubAssociator_cff' )
0024 # load code that produces DTCStubs
0025 process.load( 'L1Trigger.TrackerDTC.ProducerED_cff' )
0026 # load code that analyzes DTCStubs
0027 process.load( 'L1Trigger.TrackerDTC.Analyzer_cff' )
0028 # cosutmize TT algorithm
0029 from L1Trigger.TrackerDTC.Customize_cff import *
0030 producerUseTMTT( process )
0031 analyzerUseTMTT( process )
0032 #--- Load code that produces tfp Stubs
0033 process.load( 'L1Trigger.TrackerTFP.Producer_cff' )
0034 from L1Trigger.TrackerTFP.Customize_cff import *
0035 setupUseTMTT( process )
0036 #--- Load code that analyzes tfp Stubs
0037 process.load( 'L1Trigger.TrackerTFP.Analyzer_cff' )
0038
0039 # build schedule
0040 process.mc = cms.Sequence( process.StubAssociator )
0041 process.dtc = cms.Sequence( process.TrackerDTCProducer + process.TrackerDTCAnalyzer )
0042 process.gp = cms.Sequence( process.TrackerTFPProducerGP + process.TrackerTFPAnalyzerGP )
0043 process.ht = cms.Sequence( process.TrackerTFPProducerHT + process.TrackerTFPAnalyzerHT )
0044 process.mht = cms.Sequence( process.TrackerTFPProducerMHT + process.TrackerTFPAnalyzerMHT )
0045 process.zht = cms.Sequence( process.TrackerTFPProducerZHT + process.TrackerTFPAnalyzerZHT )
0046 process.interIn = cms.Sequence( process.TrackerTFPProducerZHTout + process.TrackerTFPProducerKFin + process.TrackerTFPAnalyzerKFin )
0047 process.kf = cms.Sequence( process.TrackerTFPProducerKF + process.TrackerTFPAnalyzerKF )
0048 process.interOut = cms.Sequence( process.TrackerTFPProducerTT + process.TrackerTFPProducerAS )#+ process.TrackerTFPAnalyzerTT )
0049 process.tt = cms.Path( process.mc + process.dtc + process.gp + process.ht + process.mht + process.zht + process.interIn + process.kf )#+ process.interOut )
0050 process.schedule = cms.Schedule( process.tt )
0051
0052 # create options
0053 import FWCore.ParameterSet.VarParsing as VarParsing
0054 options = VarParsing.VarParsing( 'analysis' )
0055 # specify input MC
0056 Samples = [
0057   #'/store/relval/CMSSW_11_3_0_pre6/RelValSingleMuFlatPt2To100/GEN-SIM-DIGI-RAW/113X_mcRun4_realistic_v6_2026D76noPU-v1/10000/05f802b7-b0b3-4cca-8b70-754682c3bb4c.root'
0058   #'/store/relval/CMSSW_11_3_0_pre6/RelValDisplacedMuPt2To100Dxy100/GEN-SIM-DIGI-RAW/113X_mcRun4_realistic_v6_2026D76noPU-v1/00000/011da61a-9524-4a96-b91f-03e8690af3bd.root'
0059   '/store/relval/CMSSW_11_3_0_pre6/RelValTTbar_14TeV/GEN-SIM-DIGI-RAW/PU_113X_mcRun4_realistic_v6_2026D76PU200-v1/00000/00026541-6200-4eed-b6f8-d3a1fd720e9c.root',
0060   '/store/relval/CMSSW_11_3_0_pre6/RelValTTbar_14TeV/GEN-SIM-DIGI-RAW/PU_113X_mcRun4_realistic_v6_2026D76PU200-v1/00000/013d0125-8f6e-496b-8335-614398c9210d.root',
0061   '/store/relval/CMSSW_11_3_0_pre6/RelValTTbar_14TeV/GEN-SIM-DIGI-RAW/PU_113X_mcRun4_realistic_v6_2026D76PU200-v1/00000/058bd134-86de-47e1-bcde-379ed9b79e1b.root',
0062   '/store/relval/CMSSW_11_3_0_pre6/RelValTTbar_14TeV/GEN-SIM-DIGI-RAW/PU_113X_mcRun4_realistic_v6_2026D76PU200-v1/00000/0915d66c-cbd4-4ef6-9971-7dd59e198b56.root',
0063   '/store/relval/CMSSW_11_3_0_pre6/RelValTTbar_14TeV/GEN-SIM-DIGI-RAW/PU_113X_mcRun4_realistic_v6_2026D76PU200-v1/00000/09823c8d-e443-4066-8347-8c704929cb2b.root',
0064   '/store/relval/CMSSW_11_3_0_pre6/RelValTTbar_14TeV/GEN-SIM-DIGI-RAW/PU_113X_mcRun4_realistic_v6_2026D76PU200-v1/00000/0c39a1aa-93ee-41c1-8543-6d90c09114a7.root',
0065   '/store/relval/CMSSW_11_3_0_pre6/RelValTTbar_14TeV/GEN-SIM-DIGI-RAW/PU_113X_mcRun4_realistic_v6_2026D76PU200-v1/00000/0fcdcc53-fb9f-4f0b-8529-a4d60d914c14.root',
0066   '/store/relval/CMSSW_11_3_0_pre6/RelValTTbar_14TeV/GEN-SIM-DIGI-RAW/PU_113X_mcRun4_realistic_v6_2026D76PU200-v1/00000/16760a5c-9cd2-41c3-82e5-399bb962d537.root',
0067   '/store/relval/CMSSW_11_3_0_pre6/RelValTTbar_14TeV/GEN-SIM-DIGI-RAW/PU_113X_mcRun4_realistic_v6_2026D76PU200-v1/00000/1752640f-2001-4d14-9276-063ec07cea92.root',
0068   '/store/relval/CMSSW_11_3_0_pre6/RelValTTbar_14TeV/GEN-SIM-DIGI-RAW/PU_113X_mcRun4_realistic_v6_2026D76PU200-v1/00000/180712c9-31a5-4f2a-bf92-a7fbee4dabad.root'
0069 ]
0070 options.register( 'inputMC', Samples, VarParsing.VarParsing.multiplicity.singleton, VarParsing.VarParsing.varType.string, "Files to be processed" )
0071 # specify number of events to process.
0072 options.register( 'Events',100,VarParsing.VarParsing.multiplicity.singleton, VarParsing.VarParsing.varType.int, "Number of Events to analyze" )
0073 options.parseArguments()
0074
0075 process.options = cms.untracked.PSet( wantSummary = cms.untracked.bool(False) )
0076 process.maxEvents = cms.untracked.PSet( input = cms.untracked.int32(options.Events) )
0077 process.source = cms.Source(
0078   "PoolSource",
0079   fileNames = cms.untracked.vstring( options.inputMC ),
0080   #skipEvents = cms.untracked.uint32( 3 + 8 ),
0081   noEventSort = cms.untracked.bool( True ),
0082   secondaryFileNames = cms.untracked.vstring(),
0083   duplicateCheckMode = cms.untracked.string( 'noDuplicateCheck' )
0084 )
0085 #process.Timing = cms.Service( "Timing", summaryOnly = cms.untracked.bool( True ) )
0086 process.MessageLogger.cerr.enableStatistics = False
0087 process.TFileService = cms.Service( "TFileService", fileName = cms.string( "Hist.root" ) )