#!/usr/bin/env python
import subprocess
import argparse

parser = argparse.ArgumentParser(description="Run (fast) simulation and analysis")

parser.add_argument('-nEv', '--number_of_events', type=int, help="Number of events to be processed", metavar='N', default=100)
parser.add_argument('-sqrts', '--center_of_mass_energy', type=float, help="Center of mass energy, sqrt(s)", metavar='sqrts', default=6.5)
parser.add_argument('-dec', '--decayfile', type=str, help="Filename of decayfile to be used", metavar='dec', default="D0_Kc_PIc.dec")
parser.add_argument('--prefix', type=str, help="Output prefix", default="D0_K_Pi")

parser.add_argument('--no_analysis', action="store_true", help="Don't do an analysis after fast simulation")
parser.add_argument('--no_sim', action="store_true", help="Don't do a simulation before analysis")


args = parser.parse_args()


nOfEvents = args.number_of_events
sqrts = args.center_of_mass_energy
decayfile = args.decayfile
prefix = args.prefix

if not (args.no_sim):
	subprocess.call(["root", "-q", "-l", "-b", 'simfast.C("'+ str(prefix) +'", "' + str(decayfile) + '", '+ str(sqrts) +', '+ str(nOfEvents) +', "pbarpSystem0")'])

if not (args.no_analysis):
	subprocess.call(["root", "-q", "-l", "-b", 'ana_d0.C("'+ str(prefix) +'_fast.root", '+ str(nOfEvents) +', '+ str(sqrts) +')' ])