|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
|
import FWCore.ParameterSet.Config as cms
import sys
import argparse
from IOPool.Input.fixReading_12_4_X_Files import fixReading_12_4_X_Files
parser = argparse.ArgumentParser(prog=sys.argv[0], description='Test Run 3 Scouting data formats')
parser.add_argument("--electronVersion", type=int, help="electron data format version (default: 7)", default=7)
parser.add_argument("--photonVersion", type=int, help="photon data format version (default: 6)", default=6)
parser.add_argument("--vertexVersion", type=int, help="photon data format version (default: 4)", default=4)
parser.add_argument("--inputFile", type=str, help="Input file name (default: testRun3Scouting.root)", default="testRun3Scouting.root")
parser.add_argument("--outputFileName", type=str, help="Output file name (default: testRun3Scouting2.root)", default="testRun3Scouting2.root")
parser.add_argument("-f", "--fixStreamerInfo", action="store_true")
args = parser.parse_args()
process = cms.Process("READ")
if args.fixStreamerInfo:
process = fixReading_12_4_X_Files(process)
print("FixingStreamerInfos")
process.source = cms.Source("PoolSource", fileNames = cms.untracked.vstring("file:"+args.inputFile))
process.testReadRun3Scouting = cms.EDAnalyzer("TestReadRun3Scouting",
# I stick to values exactly convertable to float
# to avoid potential rounding issues in the test.
expectedCaloJetsValues = cms.vdouble(
2.0, 4.0 , 6.0, 8.0, 10.0,
12.0, 14.0, 16.0, 18.0, 20.0,
22.0, 24.0, 26.0, 28.0, 30.0,
32.0),
caloJetsTag = cms.InputTag("run3ScoutingProducer", "", "PROD"),
electronClassVersion = cms.int32(args.electronVersion),
expectedElectronFloatingPointValues = cms.vdouble(
10.0, 20.0, 30.0, 40.0, 50.0,
60.0, 70.0, 80.0, 90.0, 100.0,
110.0, 120.0, 130.0, 140.0, 150.0,
160.0, 170.0, 180.0, 190.0, 200.0,
210.0, 220.0, 230.0, 240.0, 250.0,
260.0, 270.0, 280.0, 290.0, 300.0,
310.0, 320.0, 330.0),
expectedElectronIntegralValues = cms.vint32(10, 20, 30, 40, 50, 60, 70, 80),
electronsTag = cms.InputTag("run3ScoutingProducer", "", "PROD"),
expectedMuonFloatingPointValues = cms.vdouble(
10.0, 20.0, 30.0, 40.0, 50.0,
60.0, 70.0, 80.0, 90.0, 100.0,
110.0, 120.0, 130.0, 140.0, 150.0,
160.0, 170.0, 180.0, 190.0, 200.0,
210.0, 220.0, 230.0, 240.0, 250.0,
260.0, 270.0, 280.0, 290.0, 300.0,
310.0, 320.0, 330.0, 340.0, 350.0,
360.0, 370.0
),
expectedMuonIntegralValues = cms.vint32(
10, 20, 30, 40, 50,
60, 70, 80, 90, 100,
110, 120, 130, 140, 150,
160, 170, 180, 190, 200,
210, 220, 230, 240, 250,
260
),
muonsTag = cms.InputTag("run3ScoutingProducer", "", "PROD"),
expectedParticleFloatingPointValues = cms.vdouble(
11.0, 21.0, 31.0, 41.0, 51.0,
61.0, 71.0, 81.0, 91.0, 101.0,
111.0
),
expectedParticleIntegralValues = cms.vint32(
11, 21, 31, 41, 51
),
particlesTag = cms.InputTag("run3ScoutingProducer", "", "PROD"),
expectedPFJetFloatingPointValues = cms.vdouble(
12.0, 22.0, 32.0, 42.0, 52.0,
62.0, 72.0, 82.0, 92.0, 102.0,
112.0, 122.0, 132.0, 142.0, 152.0
),
expectedPFJetIntegralValues = cms.vint32(
12, 22, 32, 42, 52,
62, 72, 82
),
pfJetsTag = cms.InputTag("run3ScoutingProducer", "", "PROD"),
photonClassVersion = cms.int32(args.photonVersion),
expectedPhotonFloatingPointValues = cms.vdouble(
14.0, 23.0, 33.0, 43.0, 53.0,
63.0, 73.0, 83.0, 93.0, 103.0,
113.0, 123.0, 133.0, 143.0, 153.0,
163.0, 173.0
),
expectedPhotonIntegralValues = cms.vint32(
14, 23, 33, 43, 53
),
photonsTag = cms.InputTag("run3ScoutingProducer", "", "PROD"),
expectedTrackFloatingPointValues = cms.vdouble(
14.0, 24.0, 34.0, 44.0, 54.0,
64.0, 74.0, 84.0, 94.0, 104.0,
114.0, 124.0, 134.0, 144.0, 154.0,
164.0, 174.0, 184.0, 194.0, 204.0,
214.0, 224.0, 234.0, 244.0, 254.0,
264.0, 274.0, 284.0, 294.0
),
expectedTrackIntegralValues = cms.vint32(
14, 24, 34, 44, 54
),
tracksTag = cms.InputTag("run3ScoutingProducer", "", "PROD"),
vertexClassVersion = cms.int32(args.vertexVersion),
expectedVertexFloatingPointValues = cms.vdouble(
15.0, 25.0, 35.0, 45.0, 55.0,
65.0, 75.0, 85.0, 95.0, 105.0
),
expectedVertexIntegralValues = cms.vint32(
15, 25, 35
),
vertexesTag = cms.InputTag("run3ScoutingProducer", "", "PROD")
)
process.out = cms.OutputModule("PoolOutputModule",
fileName = cms.untracked.string(args.outputFileName),
fastCloning = cms.untracked.bool(False)
)
process.path = cms.Path(process.testReadRun3Scouting)
process.endPath = cms.EndPath(process.out)
|