Ejemplo n.º 1
0
from optparse import OptionParser
parser=OptionParser()
parser.add_option("-i","--inputFile")
parser.add_option("-p","--profileFitFile")
parser.add_option("-L","--libDir")
parser.add_option("-n","--numberOfEvents",default=-1)
parser.add_option("-d","--dir") # DQM_HOME directory
parser.add_option("-o","--outputDir")
#parser.add_option("-p","--plotsDir")

(options,args)=parser.parse_args()

import ROOT as r
r.gROOT.SetBatch(1)
r.gSystem.Load(options.libDir+"/libBTFTBSW.so")

file = r.TFile.Open(options.inputFile)
if (not file.IsOpen()):
    print "Cannot open "+ options.inputFile
    exit(-1)
tree = file.Get("eventRawData")

os.system('mkdir -p %s'%options.outputDir)

a=r.iMCP_BTF_simpleRecoTree(tree)
a.maxEntries=int(options.numberOfEvents)
a.mcpFitWaveFile=options.profileFitFile
a.outFile=options.outputDir+"/"+os.path.splitext(path_leaf(options.inputFile))[0]+"_RECO.root"
a.Loop()