#! /usr/bin/env python2
# RunMARS.py
# The new RunMARS python script that should do everything
# Author: Andrew Edmonds
# Created: 19th August 2014

import os
import sys
import getopt
import subprocess
from subprocess import call

################################
### Handle Command Line Options

## Some defaults
geometry_file="geometry.root"
input_file="input.root"
output_file="output.root"

## Read the command line options
opts, args = getopt.getopt(sys.argv[1:], "hg:i:o:", ["help","geometry=", "input=", "output="])
for opt, arg in opts:
    if opt in ("-h", "--help"):
        print("Usage: python RunMARS.py [--geometry=<geometry ROOT file>] [--input=<input RooTracker file>] [--output-<output RooTracker file>]")
    elif opt in ("-g", "--geometry"):
        geometry_file=arg;
    elif opt in ("-i", "--input"):
        input_file=arg;
    elif opt in ("-o", "--output"):
        output_file=arg;

## Assign these as environment variables so they can be found in the program
os.environ["MARS_GEOM_FILE"] = geometry_file
os.environ["MARS_INPUT_FILE"] = input_file
os.environ["MARS_OUTPUT_FILE"] = output_file

### End Handle Command Line Options
####################################

####################################
### Create the material input file MATER.INP
# Get the currrent major version of ROOT by splitting the string on the /
cmd = subprocess.Popen(['root-config', '--version'], stdout=subprocess.PIPE)
ROOT_VERSION = cmd.communicate()[0].split('/')

# Check that the current ROOT version is 5.34 or later
if (float(ROOT_VERSION[0]) < 5.34):
	print "ROOT version is not 5.34 or later"
	print "Should be in ICEDUST so resetting environment variables to use the version of ROOT packaged here"
	# Need to set the environment variables so that it gets the version of ROOT within ICEDUST
	os.environ["ROOTSYS"] = os.environ["ICEDUST_PROJECT"] + "/packages/ROOT/v999/amd64_linux26/root"
	os.environ["LD_LIBRARY_PATH"] = os.environ["ROOTSYS"] + "/lib:" + os.environ["LD_LIBRARY_PATH"]
	os.environ["PATH"] = os.environ["ROOTSYS"] + "/bin:" + os.environ["PATH"]
else:
	print "ROOT version is OK"

# Now run the script that will create the MATER.INP file
cmd = subprocess.Popen(["root", "-l", "CreateMaterialInputFile.C(\""+geometry_file+"\")", "-q"]).wait()

print "\nMATER.INP file created"
### End Create the material input file MATER.INP
################################################

##############################################
### Run MARS
# Need to set MARS environment variables
os.environ["MARS15"] = "/afs/in2p3.fr/throng/lpnhe/g2/restricted/mars15"

if (os.environ["MARS15"] == ""):
    print ("You must specify the location of your MARS installation in RunMARS.py");
    sys.exit();

os.environ["MARS15DAT"] = os.environ["MARS15"] + "/dat"
os.environ["SIMMARSROOT"] = os.environ["PWD"] + "/.."

os.environ["LD_LIBRARY_PATH"] = os.environ["MARS15"] + "/linux/lib:" + os.environ["LD_LIBRARY_PATH"]

# Need to add the different GCC library locations
os.environ["GCC_PATH"] = os.environ["PWD"] + "/../../../GCC/v999/amd64_sl6/bin"
os.environ["LD_LIBRARY_PATH"] = os.environ["GCC_PATH"] + "/../lib64:" + os.environ["LD_LIBRARY_PATH"]

# Need to change the location of Tcl and Tk for MARS visualisation
os.environ["TCL_LIBRARY"] = os.environ["MARS15"] + "/linux/lib/tcl8.5";
os.environ["TK_LIBRARY"] = os.environ["MARS15"] + "/linux/lib/tk8.5";

# Where the fieldmaps are located
os.environ["FIELDMAP_DIRECTORY"] = os.environ["ICEDUST_HOME"] + "/fieldMaps/140625-ICEDUST/"

# Get the current major version of ROOT by splitting the string on the /
cmd = subprocess.Popen(['root-config', '--version'], stdout=subprocess.PIPE)
ROOT_VERSION = cmd.communicate()[0].split('/')

# Check that the current ROOT version is 5.32
#if (float(ROOT_VERSION[0]) != 5.32):
#	print "ROOT version is not 5.32"
#	print "Setting the ROOT package to be the one in the MARS15 directory"
#	# Need to set the environment variables so that it doesn't get the version of ROOT within ICEDUST
#	os.environ["ROOTSYS"] = os.environ["MARS15"] + "/root-5.32p04/root"
#	os.environ["LD_LIBRARY_PATH"] = os.environ["ROOTSYS"] + "/lib:" + os.environ["LD_LIBRARY_PATH"]
#	os.environ["PATH"] = os.environ["ROOTSYS"] + "/bin:" + os.environ["PATH"]
#else:
#	print "ROOT version is OK"

# Now run the script that will create the MATER.INP file
#print os.environ["GCC_PATH"]
#print os.environ["LD_LIBRARY_PATH"]
#call(["ldd", "../amd64_sl6/SimMARS.exe"])
cmd = subprocess.Popen(["../amd64_sl6/SimMARS.exe"]).wait()

print "\nDone"
### End Run MARS
########################