SiStripCommon/test/testCalibTree_nano_G2.py

0001 from __future__ import print_function
0002
0003 ## adapted from produceCalibrationTree_template_cfg.py
0004
0005 import FWCore.ParameterSet.Config as cms
0006 ##from CalibTracker.SiStripCommon.shallowTree_test_template import * ## TODO get rid of this one
0007
0008 process = cms.Process('CALIB')
0009 process.load('Configuration/StandardSequences/MagneticField_cff')
0010 process.load('Configuration.Geometry.GeometryRecoDB_cff')
0011 process.load('Configuration.StandardSequences.FrontierConditions_GlobalTag_cff')
0012
0013 from Configuration.AlCa.GlobalTag import GlobalTag
0014 process.GlobalTag = GlobalTag(process.GlobalTag, "auto:run3_data_PromptAnalysis")
0015
0016 process.load('FWCore.MessageService.MessageLogger_cfi')
0017 process.load('Configuration.StandardSequences.Services_cff')
0018
0019 process.maxEvents = cms.untracked.PSet(input=cms.untracked.int32(-1))
0020
0021 process.source = cms.Source("PoolSource",
0022                             fileNames = cms.untracked.vstring("/store/express/Run2023F/StreamExpress/ALCARECO/SiStripCalMinBias-Express-v1/000/373/710/00000/e2df2f78-b95a-4f33-ae22-add59aa2903f.root"))
0023
0024 process.options   = cms.untracked.PSet( wantSummary = cms.untracked.bool(True) )
0025 process.MessageLogger.cerr.FwkReport.reportEvery = 10000
0026
0027 inTracks = cms.InputTag("ALCARECOSiStripCalMinBias")
0028
0029 process.load('CalibTracker.SiStripCommon.prescaleEvent_cfi')
0030 process.load('CalibTracker.Configuration.Filter_Refit_cff')
0031 ## use CalibrationTracks (for clusters) and CalibrationTracksRefit (for tracks)
0032 process.CalibrationTracks.src = inTracks
0033
0034 tracksForCalib = cms.InputTag("CalibrationTracksRefit")
0035
0036 process.prescaleEvent.prescale = 1
0037
0038 process.TkCalSeq = cms.Sequence(process.prescaleEvent*process.MeasurementTrackerEvent*process.trackFilterRefit)
0039
0040 process.load("PhysicsTools.NanoAOD.nano_cff")
0041 process.load("PhysicsTools.NanoAOD.NanoAODEDMEventContent_cff")
0042
0043 ## as a test: it should be possible to add tracks fully at configuration level (+ declaring the plugin)
0044 from PhysicsTools.NanoAOD.common_cff import *
0045 ## this is equivalent to ShallowTrackProducer as configured for the gain calibration
0046 process.tracksTable = cms.EDProducer("SimpleTrackFlatTableProducer",
0047         src=tracksForCalib,
0048         cut=cms.string(""),
0049         name=cms.string("track"),
0050         doc=cms.string("SiStripCalMinBias ALCARECO tracks"),
0051         singleton=cms.bool(False),
0052         extension=cms.bool(False),
0053         variables=cms.PSet(
0054             chi2ndof=Var("chi2()/ndof", float),
0055             pt=Var("pt()", float),
0056             hitsvalid=Var("numberOfValidHits()", int), ## unsigned?
0057             phi=Var("phi()", float),
0058             eta=Var("eta()", float),
0059             )
0060         )
0061 process.load("CalibTracker.SiStripCommon.tkInstLumiTable_cfi")
0062 process.tkInstLumiTable.extension = True
0063 process.load("CalibTracker.SiStripCommon.siStripGainCalibTable_cfi")
0064 process.siStripGainCalibTable.Tracks = tracksForCalib
0065
0066 process.nanoCTPath = cms.Path(process.TkCalSeq*
0067         process.nanoMetadata*process.tkInstLumiTable
0068         *process.tracksTable
0069         *process.siStripGainCalibTable
0070         )
0071
0072 process.out = cms.OutputModule("NanoAODOutputModule",
0073         fileName=cms.untracked.string("CalibTreeMC_nano_G2.root"),
0074         outputCommands=process.NANOAODEventContent.outputCommands
0075         )
0076
0077 process.end = cms.EndPath(process.out)