options.parseArguments()
runOnMC = options.runOnMC
mLSPskim = float(options.mLSP)
runSMS = False
if options.datasetName:
runSMS = 'SMS' in options.datasetName
runOnMC = 'START' in options.datasetName
if not options.inputFiles:
if options.inputDirectory is not None:
listDirectory(options.inputDirectory, options.inputFiles, options.maxFiles)
if True:
names = [f for f in options.datasetName.split('/') if f]
if runOnMC:
options.model = names[0].split('-')[1].split('_')[0]
if options.index < 0:
name = '%s-%s-%s.root' % (names[0],names[1],names[-1])
else:
name = '%s-%s-%s_%d.root' % (names[0],names[1],names[-1],options.index)
options.outputFile = os.path.join(options.outputDirectory,name)
files = getFiles(
[options.datasetName],
'lucieg',
'susy_tree_CMG_[0-9]+_[0-9]+.root'
# 'susy_tree_CMG_[0-9]+.root'
)
if options.maxFiles > 0:
options.inputFiles = files[0:options.maxFiles]
options.register ('index',
-1, # default value
VarParsing.multiplicity.singleton, # singleton or list
VarParsing.varType.int, # string, int, or float
"The file index to run on")
options.parseArguments()
if True:
names = [f for f in options.datasetName.split('/') if f]
if options.index == -1:
name = '%s-%s-%s-BPOG.root' % (names[0],names[1],names[-1])
else:
name = '%s-%s-%s-BPOG_%d.root' % (names[0],names[1],names[-1],options.index)
options.outputFile = os.path.join(options.outputDirectory,name)
options.model = names[0].split('-')[1].split('_')[0]
pickleFile = options.outputFile.replace('.root','.pkl')
files = getFiles(
[options.datasetName],
'wreece',
'susy_tree_CMG_[0-9]+.root'
)
print 'The number of files to run on is: %d' % len(files)
if options.index > -1:
chunks = []
chunk = []
for f in files:
if len(chunk) <= options.maxFiles:
chunk.append(f)
