import subprocess,argparse,os,sys

parser=argparse.ArgumentParser(description='run high statistic on plot roofit shit')
parser.add_argument('recofile',help='the reco file to process',type=str)
parser.add_argument('outfile',help='the rootfile to write everything into',type=str)
parser.add_argument('events',help='number of events',type=int)
parser.add_argument('--start',help='the startevent',type=int,default=0)
parser.add_argument('--keep',help='skip if outfile is already existing',action='store_true')
parser.add_argument('--merge',help='also merge at the end',action='store_true')

args=parser.parse_args()

junksize=100
ev=-junksize

junk_num=-1
while ev<args.events:
    ev+=junksize
    junk_num+=1
    tmpout=args.outfile.replace('.root','_tmp{0}.root'.format(junk_num))

    print tmpout
    if os.path.isfile(tmpout) and args.keep:
        print 'skipping',tmpout
        continue

    if ev<args.start:
        continue

    command=''
    command+='python FOPI/mberger/plot_digisOnPlaneRooFit.py'
    command+=' '+args.recofile
    command+=' --outfile '+tmpout
    command+=' --events '+str(ev+junksize)
    command+=' --goto '+str(ev)
    command+=' --batch'
    print command
    os.system(command)

    '''
    moved_out=args.outfile.replace('.root','_tmp2.root')
    command=''
    command+='mv {0} {1}'.format(args.outfile,moved_out)
    os.system(command)

    command=''
    command+='hadd {0} {1} {2}'.format(args.outfile,tmpout,moved_out)
    #p=subprocess.call(command)
    os.system(command)

    #p=subprocess.call('rm {0}'.format(tmpout))
    os.system('rm {0} {1}'.format(tmpout,moved_out))
    '''

if args.merge:
    os.system('hadd -f {0} {1}'.format(args.outfile,args.outfile.replace('.root','_tmp*')))