TrackProducer/test/refitFromAOD.py

0001 import FWCore.ParameterSet.Config as cms
0002 import FWCore.ParameterSet.VarParsing as VarParsing
0003
0004 from Configuration.Eras.Era_Run2_2016_cff import Run2_2016
0005 process = cms.Process('RECO2',Run2_2016)
0006
0007 options = VarParsing.VarParsing('analysis')
0008 options.register('globalTag',
0009                  "auto:run2_mc", # default value
0010                  VarParsing.VarParsing.multiplicity.singleton, # singleton or list
0011                  VarParsing.VarParsing.varType.string, # string, int, or float
0012                  "input file name")
0013 options.parseArguments()
0014
0015 # import of standard configurations
0016 process.load('Configuration.StandardSequences.Services_cff')
0017 process.load('SimGeneral.HepPDTESSource.pythiapdt_cfi')
0018 process.load('FWCore.MessageService.MessageLogger_cfi')
0019 process.load('Configuration.EventContent.EventContent_cff')
0020 process.load('SimGeneral.MixingModule.mixNoPU_cfi')
0021 process.load('Configuration.StandardSequences.GeometryRecoDB_cff')
0022 process.load('Configuration.StandardSequences.MagneticField_cff')
0023 process.load('Configuration.StandardSequences.Reconstruction_cff')
0024 process.load('Configuration.StandardSequences.EndOfProcess_cff')
0025 process.load('Configuration.StandardSequences.FrontierConditions_GlobalTag_cff')
0026
0027 process.source = cms.Source("PoolSource",
0028                             fileNames = cms.untracked.vstring(options.inputFiles),
0029                             secondaryFileNames = cms.untracked.vstring()
0030                             )
0031
0032 process.maxEvents = cms.untracked.PSet(
0033     input = cms.untracked.int32(options.maxEvents)
0034 )
0035
0036 process.options = cms.untracked.PSet()
0037
0038 # Production Info
0039 process.configurationMetadata = cms.untracked.PSet(
0040     annotation = cms.untracked.string('step1 nevts:1'),
0041     name = cms.untracked.string('Applications'),
0042     version = cms.untracked.string('$Revision: 1.19 $')
0043 )
0044
0045 # Output definition
0046
0047 process.RECOSIMoutput = cms.OutputModule("PoolOutputModule",
0048     dataset = cms.untracked.PSet(
0049         dataTier = cms.untracked.string(''),
0050         filterName = cms.untracked.string('')
0051     ),
0052     fileName = cms.untracked.string('step1_RECO.root'),
0053     outputCommands = process.RECOSIMEventContent.outputCommands,
0054     splitLevel = cms.untracked.int32(0)
0055 )
0056
0057 # Additional output definition
0058 process.RECOSIMoutput.outputCommands = cms.untracked.vstring("keep *_myRefittedTracks_*_*")
0059
0060
0061 # Other statements
0062 from Configuration.AlCa.GlobalTag import GlobalTag
0063 process.GlobalTag = GlobalTag(process.GlobalTag, options.globalTag, '')
0064
0065 import RecoTracker.TrackProducer.trackExtraRekeyer_cfi
0066 process.trackExtraRekeyer = RecoTracker.TrackProducer.trackExtraRekeyer_cfi.trackExtraRekeyer.clone(
0067     src = "generalTracks",
0068     association = "muonReducedTrackExtras"
0069 )
0070
0071 import RecoTracker.TrackProducer.TrackRefitter_cfi
0072 process.myRefittedTracks = RecoTracker.TrackProducer.TrackRefitter_cfi.TrackRefitter.clone(
0073     src= 'trackExtraRekeyer',
0074     NavigationSchool = '',
0075     Fitter = 'FlexibleKFFittingSmoother'
0076 )
0077
0078 # Path and EndPath definitions
0079 process.reconstruction_step = cms.Path(process.trackExtraRekeyer*process.myRefittedTracks)
0080 process.endjob_step = cms.EndPath(process.endOfProcess)
0081 process.RECOSIMoutput_step = cms.EndPath(process.RECOSIMoutput)
0082
0083 # Schedule definition
0084 process.schedule = cms.Schedule(process.reconstruction_step,process.endjob_step,process.RECOSIMoutput_step)