andersj/HcalPlotter/tests.py

import FWCore.ParameterSet.Config as cms
import sys
import os
import re

process = cms.Process("Demo")

process.load("FWCore.MessageService.MessageLogger_cfi")

process.MessageLogger.cerr.FwkReport.reportEvery = 500
process.MessageLogger.suppressWarning.extend(["ecalRecHit",'ecalEBunpacker'])

process.load("andersj.HcalPlotter.hcal_tb11_cff")

# We are getting the runnumber from the command line now. No need to edit this file.
if len(sys.argv) > 2:
    #print sys.argv
    arg1 = sys.argv[2]
    RUNNUMBER = int(arg1)
    inputieta = int(sys.argv[3])
    inputiphi = int(sys.argv[4])
    #print arg1
else:
    RUNNUMBER = 0
if (RUNNUMBER < 1):
    rn = input("Enter Runnumber: ")
    RUNNUMBER = int(rn)
    tmp = input("Enter hb table ieta: ")
    inputieta = int(tmp)
    tmp = input("Enter hb table iphi: ")
    inputiphi = int(tmp)
print "Running on: {0:08d} table coordinates ({1},{2})".format(RUNNUMBER,
                                                               inputieta,
                                                               inputiphi)

process.maxEvents = cms.untracked.PSet(
    input = cms.untracked.int32(-1)
)

datadir = 'tb_data'

files = ['file:{0}/{1}'.format(datadir,x) for x in os.listdir(datadir) \
         if re.search('_{0:08d}\.\d+'.format(RUNNUMBER), x)]

#print files

process.source = cms.Source("PoolSource",
    fileNames = cms.untracked.vstring(files
    )
)

#process.source.fileNames.extend(files)

print process.source.fileNames

process.TFileService = cms.Service("TFileService",
    fileName = cms.string('test/tb2011_{0:08d}.root'.format(RUNNUMBER))
)

process.load("EventFilter.EcalRawToDigi.EcalUnpackerData_cfi")
process.load("EventFilter.EcalRawToDigi.EcalUnpackerMapping_cfi")
process.load("Geometry.CaloEventSetup.CaloGeometry_cfi")
process.load("Geometry.CaloEventSetup.CaloTopology_cfi")
process.load("Geometry.CaloEventSetup.EcalTrigTowerConstituents_cfi")
process.load("Geometry.CMSCommonData.cmsIdealGeometryXML_cfi")
process.load("Geometry.EcalMapping.EcalMapping_cfi")
process.load("Geometry.EcalMapping.EcalMappingRecord_cfi")
process.load("CalibCalorimetry.EcalLaserCorrection.ecalLaserCorrectionService_cfi")
process.load("CalibCalorimetry.EcalTrivialCondModules.EcalTrivialCondRetriever_cfi")

process.ecalEBunpacker.FEDs = cms.vint32(610)
process.ecalEBunpacker.memUnpacking = cms.bool(False)
process.ecalEBunpacker.srpUnpacking = cms.bool(False)
process.ecalEBunpacker.syncCheck = cms.bool(False)
process.ecalEBunpacker.orderedFedList = cms.vint32(610)
process.ecalEBunpacker.silentMode = cms.untracked.bool(False)

## import RecoLocalCalo.EcalRecProducers.ecalRatioUncalibRecHit_cfi
## process.ecalUncalibHit = RecoLocalCalo.EcalRecProducers.ecalRatioUncalibRecHit_cfi.ecalRatioUncalibRecHit.clone()

import RecoLocalCalo.EcalRecProducers.ecalFixedAlphaBetaFitUncalibRecHit_cfi
process.ecalUncalibHit = RecoLocalCalo.EcalRecProducers.ecalFixedAlphaBetaFitUncalibRecHit_cfi.ecalFixedAlphaBetaFitUncalibRecHit.clone()

process.ecalUncalibHit.EBdigiCollection = 'ecalEBunpacker:ebDigis'
process.ecalUncalibHit.EEdigiCollection = 'ecalEBunpacker:eeDigis'

process.load("RecoLocalCalo.EcalRecProducers.ecalRecHit_cfi")
process.ecalRecHit.ebDetIdToBeRecovered = cms.InputTag("","")
process.ecalRecHit.eeDetIdToBeRecovered = cms.InputTag("","")
process.ecalRecHit.eeFEToBeRecovered = cms.InputTag("","")
process.ecalRecHit.ebFEToBeRecovered = cms.InputTag("","")
process.ecalRecHit.recoverEBFE = cms.bool(False)
process.ecalRecHit.recoverEEFE = cms.bool(False)
process.ecalRecHit.EBuncalibRecHitCollection = 'ecalUncalibHit:EcalUncalibRecHitsEB'
process.ecalRecHit.EEuncalibRecHitCollection = 'ecalUncalibHit:EcalUncalibRecHitsEE'

process.ecalDataSequence = cms.Sequence(process.ecalEBunpacker*process.ecalUncalibHit*process.ecalRecHit)

process.load("EventFilter.HcalRawToDigi.HcalRawToDigi_cfi")
process.load("RecoLocalCalo.HcalRecProducers.HcalSimpleReconstructor_hbhe_cfi")
process.load("RecoLocalCalo.HcalRecProducers.HcalSimpleReconstructor_ho_cfi")
#process.load("RecoLocalCalo.HcalRecProducers.HcalSimpleReconstructor_hf_cfi")
process.load("andersj.HcalPlotter.HcalTBObjectUnpacker_Normal_cfi")

## process.horeco.firstSample=2
## process.horeco.samplesToAdd=6
process.horeco.firstSample=3
process.horeco.samplesToAdd=4
process.horeco.correctForPhaseContainment = False
process.horeco.correctForTimeslew = False

#process.hfreco.firstSample=2
#process.hfreco.samplesToAdd=9

process.plotanal=cms.EDAnalyzer(
    "HcalHOTBPlotAnal",
    hbheRHtag = cms.untracked.InputTag("hbheprereco"),
    hoRHtag   = cms.untracked.InputTag("horeco"),
    hfRHtag   = cms.untracked.InputTag("hfreco"),
    hcalDigiTag = cms.untracked.InputTag("hcalDigis"),
    hcalTrigTag = cms.untracked.InputTag("tbunpack"),
    ebRHtag = cms.untracked.InputTag("ecalRecHit:EcalRecHitsEB"),
    doBeamCounters = cms.untracked.bool(True),
    calibFC2GeV = cms.untracked.double(0.01),
    HistoParameters = cms.PSet(
        pedGeVlo   = cms.double(-15),
        pedGeVhi   = cms.double(15),
        pedADClo   = cms.double(0),
        pedADChi   = cms.double(49),
        ledGeVlo   = cms.double(-5),
        ledGeVhi   = cms.double(250),
        laserGeVlo = cms.double(-5),
        laserGeVhi = cms.double(350),
        otherGeVlo = cms.double(-5),
        otherGeVhi = cms.double(250),
        beamGeVlo  = cms.double(-20),
        beamGeVhi  = cms.double(500),
        #beamGeVhi  = cms.double(80*200),
        timeNSlo   = cms.double(50),
        timeNShi   = cms.double(250),
        ieta = cms.int32(inputieta),
        iphi = cms.int32(inputiphi)
        )
    )

##process.dump = cms.EDAnalyzer("EventContentAnalyzer")

#process.p=cms.Path(process.ecalDataSequence+process.hcalDigis+process.hbheprereco+process.horeco+process.hfreco+process.tbunpack+process.plotanal)
process.p=cms.Path(process.ecalDataSequence+process.hcalDigis+process.hbheprereco+process.horeco+process.tbunpack+process.plotanal)
Revision:	1.9
Committed:	Wed Aug 8 14:48:36 2012 UTC (12 years, 8 months ago) by andersj
Content type:	text/x-python
Branch:	MAIN
CVS Tags:	HEAD
Changes since 1.8:	+1 -1 lines
Log Message:	final 2011 updates
#	User	Rev	Content
1	andersj	1.1	import FWCore.ParameterSet.Config as cms
2			import sys
3	andersj	1.6	import os
4			import re
5	andersj	1.1
6			process = cms.Process("Demo")
7
8			process.load("FWCore.MessageService.MessageLogger_cfi")
9
10			process.MessageLogger.cerr.FwkReport.reportEvery = 500
11	andersj	1.9	process.MessageLogger.suppressWarning.extend(["ecalRecHit",'ecalEBunpacker'])
12	andersj	1.1
13			process.load("andersj.HcalPlotter.hcal_tb11_cff")
14
15			# We are getting the runnumber from the command line now. No need to edit this file.
16			if len(sys.argv) > 2:
17			#print sys.argv
18			arg1 = sys.argv[2]
19			RUNNUMBER = int(arg1)
20	andersj	1.7	inputieta = int(sys.argv[3])
21			inputiphi = int(sys.argv[4])
22	andersj	1.1	#print arg1
23			else:
24			RUNNUMBER = 0
25			if (RUNNUMBER < 1):
26	andersj	1.7	rn = input("Enter Runnumber: ")
27			RUNNUMBER = int(rn)
28			tmp = input("Enter hb table ieta: ")
29			inputieta = int(tmp)
30			tmp = input("Enter hb table iphi: ")
31			inputiphi = int(tmp)
32			print "Running on: {0:08d} table coordinates ({1},{2})".format(RUNNUMBER,
33			inputieta,
34			inputiphi)
35	andersj	1.1
36			process.maxEvents = cms.untracked.PSet(
37			input = cms.untracked.int32(-1)
38			)
39
40	andersj	1.6	datadir = 'tb_data'
41
42			files = ['file:{0}/{1}'.format(datadir,x) for x in os.listdir(datadir) \
43			if re.search('_{0:08d}\.\d+'.format(RUNNUMBER), x)]
44
45			#print files
46
47	andersj	1.1	process.source = cms.Source("PoolSource",
48	andersj	1.6	fileNames = cms.untracked.vstring(files
49	andersj	1.1	)
50			)
51
52	andersj	1.6	#process.source.fileNames.extend(files)
53
54			print process.source.fileNames
55
56	andersj	1.1	process.TFileService = cms.Service("TFileService",
57			fileName = cms.string('test/tb2011_{0:08d}.root'.format(RUNNUMBER))
58			)
59
60			process.load("EventFilter.EcalRawToDigi.EcalUnpackerData_cfi")
61			process.load("EventFilter.EcalRawToDigi.EcalUnpackerMapping_cfi")
62			process.load("Geometry.CaloEventSetup.CaloGeometry_cfi")
63			process.load("Geometry.CaloEventSetup.CaloTopology_cfi")
64			process.load("Geometry.CaloEventSetup.EcalTrigTowerConstituents_cfi")
65			process.load("Geometry.CMSCommonData.cmsIdealGeometryXML_cfi")
66			process.load("Geometry.EcalMapping.EcalMapping_cfi")
67			process.load("Geometry.EcalMapping.EcalMappingRecord_cfi")
68			process.load("CalibCalorimetry.EcalLaserCorrection.ecalLaserCorrectionService_cfi")
69			process.load("CalibCalorimetry.EcalTrivialCondModules.EcalTrivialCondRetriever_cfi")
70
71			process.ecalEBunpacker.FEDs = cms.vint32(610)
72			process.ecalEBunpacker.memUnpacking = cms.bool(False)
73			process.ecalEBunpacker.srpUnpacking = cms.bool(False)
74	andersj	1.2	process.ecalEBunpacker.syncCheck = cms.bool(False)
75			process.ecalEBunpacker.orderedFedList = cms.vint32(610)
76			process.ecalEBunpacker.silentMode = cms.untracked.bool(False)
77
78	andersj	1.8	## import RecoLocalCalo.EcalRecProducers.ecalRatioUncalibRecHit_cfi
79			## process.ecalUncalibHit = RecoLocalCalo.EcalRecProducers.ecalRatioUncalibRecHit_cfi.ecalRatioUncalibRecHit.clone()
80	andersj	1.2
81	andersj	1.8	import RecoLocalCalo.EcalRecProducers.ecalFixedAlphaBetaFitUncalibRecHit_cfi
82			process.ecalUncalibHit = RecoLocalCalo.EcalRecProducers.ecalFixedAlphaBetaFitUncalibRecHit_cfi.ecalFixedAlphaBetaFitUncalibRecHit.clone()
83	andersj	1.2
84			process.ecalUncalibHit.EBdigiCollection = 'ecalEBunpacker:ebDigis'
85			process.ecalUncalibHit.EEdigiCollection = 'ecalEBunpacker:eeDigis'
86
87			process.load("RecoLocalCalo.EcalRecProducers.ecalRecHit_cfi")
88			process.ecalRecHit.ebDetIdToBeRecovered = cms.InputTag("","")
89			process.ecalRecHit.eeDetIdToBeRecovered = cms.InputTag("","")
90			process.ecalRecHit.eeFEToBeRecovered = cms.InputTag("","")
91			process.ecalRecHit.ebFEToBeRecovered = cms.InputTag("","")
92			process.ecalRecHit.recoverEBFE = cms.bool(False)
93			process.ecalRecHit.recoverEEFE = cms.bool(False)
94			process.ecalRecHit.EBuncalibRecHitCollection = 'ecalUncalibHit:EcalUncalibRecHitsEB'
95			process.ecalRecHit.EEuncalibRecHitCollection = 'ecalUncalibHit:EcalUncalibRecHitsEE'
96	andersj	1.1
97	andersj	1.2	process.ecalDataSequence = cms.Sequence(process.ecalEBunpackerprocess.ecalUncalibHitprocess.ecalRecHit)
98	andersj	1.1
99			process.load("EventFilter.HcalRawToDigi.HcalRawToDigi_cfi")
100			process.load("RecoLocalCalo.HcalRecProducers.HcalSimpleReconstructor_hbhe_cfi")
101			process.load("RecoLocalCalo.HcalRecProducers.HcalSimpleReconstructor_ho_cfi")
102	andersj	1.4	#process.load("RecoLocalCalo.HcalRecProducers.HcalSimpleReconstructor_hf_cfi")
103	andersj	1.1	process.load("andersj.HcalPlotter.HcalTBObjectUnpacker_Normal_cfi")
104
105	andersj	1.4	## process.horeco.firstSample=2
106			## process.horeco.samplesToAdd=6
107			process.horeco.firstSample=3
108			process.horeco.samplesToAdd=4
109			process.horeco.correctForPhaseContainment = False
110			process.horeco.correctForTimeslew = False
111	andersj	1.1
112	andersj	1.4	#process.hfreco.firstSample=2
113			#process.hfreco.samplesToAdd=9
114	andersj	1.1
115			process.plotanal=cms.EDAnalyzer(
116			"HcalHOTBPlotAnal",
117			hbheRHtag = cms.untracked.InputTag("hbheprereco"),
118			hoRHtag = cms.untracked.InputTag("horeco"),
119			hfRHtag = cms.untracked.InputTag("hfreco"),
120			hcalDigiTag = cms.untracked.InputTag("hcalDigis"),
121			hcalTrigTag = cms.untracked.InputTag("tbunpack"),
122			ebRHtag = cms.untracked.InputTag("ecalRecHit:EcalRecHitsEB"),
123			doBeamCounters = cms.untracked.bool(True),
124			calibFC2GeV = cms.untracked.double(0.01),
125			HistoParameters = cms.PSet(
126			pedGeVlo = cms.double(-15),
127			pedGeVhi = cms.double(15),
128			pedADClo = cms.double(0),
129			pedADChi = cms.double(49),
130			ledGeVlo = cms.double(-5),
131			ledGeVhi = cms.double(250),
132			laserGeVlo = cms.double(-5),
133			laserGeVhi = cms.double(350),
134			otherGeVlo = cms.double(-5),
135			otherGeVhi = cms.double(250),
136			beamGeVlo = cms.double(-20),
137			beamGeVhi = cms.double(500),
138			#beamGeVhi = cms.double(80*200),
139			timeNSlo = cms.double(50),
140	andersj	1.7	timeNShi = cms.double(250),
141			ieta = cms.int32(inputieta),
142			iphi = cms.int32(inputiphi)
143	andersj	1.1	)
144			)
145
146			##process.dump = cms.EDAnalyzer("EventContentAnalyzer")
147
148	andersj	1.4	#process.p=cms.Path(process.ecalDataSequence+process.hcalDigis+process.hbheprereco+process.horeco+process.hfreco+process.tbunpack+process.plotanal)
149			process.p=cms.Path(process.ecalDataSequence+process.hcalDigis+process.hbheprereco+process.horeco+process.tbunpack+process.plotanal)