| 1 |
mzanetti |
1.1 |
#!/usr/bin/env python
|
| 2 |
|
|
import ROOT
|
| 3 |
|
|
import sys, os, math, argparse
|
| 4 |
mzanetti |
1.2 |
from auxiliar import getDataset, lep3Datasets
|
| 5 |
|
|
from fillLep3Tree import fillLep3Tree
|
| 6 |
mzanetti |
1.1 |
|
| 7 |
|
|
parser = argparse.ArgumentParser(prog=os.path.basename(sys.argv[0]),description="analysis")
|
| 8 |
|
|
parser.add_argument('-d',dest='dataset',action='store',required=True,help='dataset')
|
| 9 |
mzanetti |
1.2 |
parser.add_argument('-o',dest='outputDir',action='store',required=False,help='output directory')
|
| 10 |
|
|
parser.add_argument('-f',dest='firstFile',action='store',required=False,help='first file to process')
|
| 11 |
mzanetti |
1.1 |
parser.add_argument('-n',dest='maxNumberOfFiles',action='store',required=False,help='max number of files')
|
| 12 |
|
|
args=parser.parse_args()
|
| 13 |
mzanetti |
1.2 |
if not args.outputDir: args.outputDir='./'
|
| 14 |
|
|
if not args.firstFile: args.firstFile='0'
|
| 15 |
mzanetti |
1.1 |
if not args.maxNumberOfFiles: args.maxNumberOfFiles='99999'
|
| 16 |
|
|
|
| 17 |
mzanetti |
1.2 |
if args.dataset in lep3Datasets:
|
| 18 |
|
|
counter = 0
|
| 19 |
|
|
for inputFile in getDataset(lep3Datasets[args.dataset][0]):
|
| 20 |
|
|
counter+=1
|
| 21 |
|
|
if counter < int(args.firstFile): continue
|
| 22 |
|
|
print inputFile
|
| 23 |
|
|
outputFile = args.outputDir+args.dataset+'_'+inputFile[inputFile.rfind('_')+1:-5]+'.root'
|
| 24 |
|
|
#fillLep3Tree(inputFile, outputFile, lep3Datasets[args.dataset][1])
|
| 25 |
|
|
os.system('./fillTree.py -i '+ inputFile + ' -o '+outputFile )
|
| 26 |
mzanetti |
1.3 |
if counter==int(args.maxNumberOfFiles)+int(args.firstFile)-1: break
|
| 27 |
mzanetti |
1.1 |
|
