import glob, sys, os, string, io, ROOT, math, commands
from ROOT import std

pandapath=os.environ.get('PANDAPATH')
sys.path.append(pandapath+'/macro/tpc/FOPI/python/argparse-1.2.1')
import argparse

parser=argparse.ArgumentParser(description='submit multiple digi files and merge the afterwards')
parser.add_argument('mcfile',help='the mc file',type=str)
parser.add_argument('field',help='the drift field',type=str)
parser.add_argument('gas',help='the gas',type=str)
parser.add_argument('gain',help='the gain',type=str)
parser.add_argument('--devmap',help='the deviation map to use ( "" if none)',type=str,default="")
parser.add_argument('--junksize',help='the size of the single files',type=int,default=5000)

args=parser.parse_args()
ROOT.gROOT.ProcessLine(".x rootlogon.C") 


mcfile=ROOT.TFile(args.mcfile,'read')
tree=mcfile.Get('cbmsim')
totEv=tree.GetEntries()

start=0
stop=args.junksize

digicounter=-1
jobids=''
while start<totEv:
    digicounter+=1
    runfname='submitDigi_'+args.mcfile.split('/')[-1]+'_'+str(start)+'_'+str(stop)
    runfile=open('./'+runfname+'.run','write')
    line=''
    line+='root -l -q -b macro/tpc/FOPI/runDigiFOPICosmics.C\('
    line+='\\\"'+args.mcfile+'\\\"'
    line+=','+args.field
    line+=',\\\"'+args.gas+'\\\"' #gas mixture
    line+=','+str(args.gain) #gem gain
    #if args.distort=='None':
    #    line+=',\\\"\\\"'
    #else:
    line+=',\\\"'+args.devmap+'\\\"' #distortion map
    line+=','+str(stop)
    line+=','+str(start)
    line+=','+str(digicounter)
    line+='\)\n'

    runfile.write('pwd\n')
    runfile.write('export PANDAPATH='+pandapath+'\n')
    runfile.write('cd $PANDAPATH\n')
    runfile.write('pwd\n')
    runfile.write('. ./setenv\n')
    runfile.write(line)
    runfile.close()

    start=stop
    stop+=args.junksize

    os.system('chmod a+x ./'+runfname+'.run')
    qsub='qsub -N '+runfname+' -o ~/sgeout/'+runfname+' -j oe ./'+runfname+'.run'
    print qsub
    jobid=10 #dont forget to comment this out
    status, jobid = commands.getstatusoutput(qsub)
    jobids+=str(jobid)+':'

jobids=jobids[0:len(jobids)-1]

path=os.path.dirname(args.mcfile)
diginame=args.mcfile.split('/')[-1]
diginame=diginame.replace('mc.root','raw.root')
merge=''
merge+='python ./FOPI/BatchTools/mergerecofiles.py '
merge+=' --path '+path
merge+=' --out '+path+'/'+diginame
#merge+=' --delIfEntry '+str(totEv)
merge+=' --pattern splitraw'
merge+=' \n'

mergefname='mergeDigi_'+args.mcfile.split('/')[-1]+'.merge'
mergefile=open('./'+mergefname,'write')

mergefile.write('pwd\n')
mergefile.write('export PANDAPATH='+pandapath+'\n')
mergefile.write('cd $PANDAPATH\n')
mergefile.write('pwd\n')
mergefile.write('. ./setenv\n')
mergefile.write(merge)
mergefile.close()
os.system('chmod a+x ./'+mergefname)
qsub='qsub -N '+mergefname+' -o ~/sgeout/'+mergefname+' -j oe ./'+mergefname
qsub+=' -W depend=afterok:'+jobids
print qsub
status, jobid = commands.getstatusoutput(qsub)