|
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
|
#!/usr/bin/env python3
from builtins import range
import re
import argparse
import math
import fileinput
# Set up argrument parser
helpEpilog = ''''''
parser = argparse.ArgumentParser(description='Take card file, blank all INFI directives and insert the INFI directives from the modifier file instead.',
epilog=helpEpilog,
formatter_class=argparse.RawDescriptionHelpFormatter)
parser.add_argument('inCfg', action='store',
help='name of the config-template')
parser.add_argument('modCfg', action='store',
help='name of the modifier file from mps_split')
parser.add_argument('outCfg', action='store',
help='name of modified output file')
parser.add_argument('isn', action='store',
help='number of the job (three digit number with leading zeros)')
parser.add_argument("--max-events", dest = "max_events", type = int,
help = "maximum number of events to process")
parser.add_argument("--skip-events", dest = "skip_events", type = int,
help = "number of events to skip before processing")
# Parse arguments
args = parser.parse_args()
inCfg = args.inCfg
modCfg = args.modCfg
outCfg = args.outCfg
isn = args.isn
# open input file
with open(inCfg, 'r') as INFILE:
body = INFILE.read()
# read modifier file
with open(modCfg, 'r') as MODFILE:
mods = MODFILE.read()
mods = mods.strip()
# prepare the new fileNames directive. Delete first line if necessary.
fileNames = mods.split('\n')
if 'CastorPool=' in fileNames[0]:
del fileNames[0]
# replace ISN number (input is a string of three digit number with leading zeros though)
body = re.sub(re.compile('ISN',re.M), isn, body)
# print to outCfg
with open(outCfg, 'w') as OUTFILE:
OUTFILE.write(body)
# Number of total files and number of extends for cms.vstring are needed
numberOfFiles = len(fileNames)
numberOfExtends = int(math.ceil(numberOfFiles/255.))
# Create and insert the readFile.extend lines
for j in range(numberOfExtends):
insertBlock = "readFiles.extend([\n "
i=0
currentStart = j*255
while (i<255) and ((currentStart+i)<numberOfFiles):
entry = fileNames[currentStart+i].strip()
if (i==254) or ((currentStart+i+1)==numberOfFiles):
insertBlock += "\'"+entry+"\'])\n"
else:
insertBlock += "\'"+entry+"\',\n "
i+=1
for line in fileinput.input(outCfg, inplace=1):
print(line,end='')
if re.match('readFiles\s*=\s*cms.untracked.vstring()',line):
print(insertBlock,end='')
if args.skip_events is not None:
with open(outCfg, "a") as f:
f.write("process.source.skipEvents = cms.untracked.uint32({0:d})\n"
.format(args.skip_events))
if args.max_events is not None:
with open(outCfg, "a") as f:
f.write("process.maxEvents = cms.untracked.PSet(input = "
"cms.untracked.int32({0:d}))\n".format(args.max_events))
|