TrackProducer/test/TrackRefit.py

0001 import FWCore.ParameterSet.Config as cms
0002
0003 process = cms.Process("Refitting")
0004
0005 ### Standard Configurations
0006 #process.load("RecoVertex.BeamSpotProducer.BeamSpot_cfi")
0007 #process.load("Configuration.StandardSequences.MagneticField_cff")
0008 #process.load('Configuration.Geometry.GeometryRecoDB_cff')
0009 #process.load("Configuration.StandardSequences.FrontierConditions_GlobalTag_cff")
0010
0011 process.load('Configuration.StandardSequences.Services_cff')
0012 process.load('FWCore.MessageService.MessageLogger_cfi')
0013 process.load('Configuration.EventContent.EventContent_cff')
0014 process.load('Configuration.StandardSequences.GeometryRecoDB_cff')
0015 process.load('Configuration.StandardSequences.MagneticField_cff')
0016 process.load('Configuration.StandardSequences.RawToDigi_cff')
0017 process.load('Configuration.StandardSequences.L1Reco_cff')
0018 process.load('Configuration.StandardSequences.Reconstruction_cff')
0019
0020 process.load('Configuration.StandardSequences.FrontierConditions_GlobalTag_cff')
0021 from Configuration.AlCa.GlobalTag import GlobalTag
0022 # choose!
0023 process.GlobalTag = GlobalTag(process.GlobalTag, 'auto:run2_data_GRun', '')
0024 # process.GlobalTag = GlobalTag(process.GlobalTag, 'auto:run2_mc_GRun', '')
0025
0026
0027
0028
0029 ### Track refitter specific stuff
0030 import RecoTracker.TrackProducer.TrackRefitter_cfi
0031 import CommonTools.RecoAlgos.recoTrackRefSelector_cfi
0032 process.mytkselector = CommonTools.RecoAlgos.recoTrackRefSelector_cfi.recoTrackRefSelector.clone()
0033 process.mytkselector.quality = ['highPurity']
0034 process.mytkselector.min3DLayer = 2
0035 process.mytkselector.ptMin = 0.5
0036 process.mytkselector.tip = 1.0
0037 process.myRefittedTracks = RecoTracker.TrackProducer.TrackRefitter_cfi.TrackRefitter.clone()
0038 process.myRefittedTracks.src= 'mytkselector'
0039 process.myRefittedTracks.NavigationSchool = ''
0040 process.myRefittedTracks.Fitter = 'FlexibleKFFittingSmoother'
0041
0042
0043 ### and an analyzer
0044 process.trajCout = cms.EDAnalyzer('TrajectoryAnalyzer',
0045    trajectoryInput=cms.InputTag('myRefittedTracks')
0046 )
0047
0048 process.source = cms.Source ("PoolSource",
0049                              fileNames=cms.untracked.vstring('file:pickevents_1.root',
0050                             ),
0051                              skipEvents=cms.untracked.uint32(0)
0052                              )
0053
0054 process.maxEvents = cms.untracked.PSet(input = cms.untracked.int32(1))
0055
0056 process.Path = cms.Path(process.mytkselector+process.myRefittedTracks+process.trajCout)
0057