#!/usr/bin/env python
'''
Date        : 08/04/2015  
Created by  : Fulya Kaplan
Description : This main python script runs scheduler and ember simulations successively until completion.
'''

import os, sys
from xml.dom.minidom import parse
import xml.dom.minidom

from optparse import OptionParser
import math
import numpy as np

# Function to run linux commands
def run(cmd):
    #print(cmd)
    os.system(cmd)

def clear_files(options):

    erFile = open(options.emberRunningFile, 'w')
    ecFile = open(options.emberCompletedFile, 'w')

    erFile.close()
    ecFile.close()

def delete_logs():

    cmd = "rm motif*.log"
    run(cmd)

def run_sim (options):

    # Run scheduler for the first time and create the first snapshot
    init_cmd  = "sst ./%s" %(options.schedPythonFile)
    run(init_cmd)

    # Do the following in a loop until the simulation is completed
    # Parse scheduler snapshot->run ember->Parse ember output->run scheduler->...
    while( is_not_empty(options.xmlFile) ):
        ember_cmd = "./%s --xml %s > %s" %(options.sched_parser, options.xmlFile, options.emberOutFile)
        run(ember_cmd)

        sched_cmd = "./%s --xml %s --emberOut %s --schedPy %s --ember_completed %s --ember_running %s " %(options.ember_parser, options.xmlFile, options.emberOutFile, options.schedPythonFile, options.emberCompletedFile, options.emberRunningFile)
        run(sched_cmd)

    delete_logs()

def is_not_empty(fileName):

    try:
        if os.stat(fileName).st_size > 0:
            return True
        else:
            return False 
    except OSError:
        print "No file"

'''
def set_path_emberLoad(options):

    emberpath="/home/fkaplan/SST/scratch/src/sst-simulator/sst/elements/ember/test"
    cmd = "sed -i \"s|PATH|%s|g\" emberLoad.py" %(emberpath)
    print cmd
    run(cmd)
'''
def grep_set_fileNames(options):

    # Parser script names are defined by default
    options.sched_parser = "snapshotParser_sched.py"
    options.ember_parser = "snapshotParser_ember.py"

    # Grep other file names from the scheduler input configuration file
    traceFile = open(options.schedPythonFile, 'r')

    for line in traceFile:
        if "traceName" in line:
            temp = line.split(':')[1]
            temp = temp.split('\"')[1]
            options.xmlFile = temp + ".snapshot.xml"
        elif "completedJobsTrace" in line:
            temp = line.split(':')[1]
            temp = temp.split('\"')[1]
            options.emberCompletedFile = temp
        elif "runningJobsTrace" in line:
            temp = line.split(':')[1]
            temp = temp.split('\"')[1]
            options.emberRunningFile = temp

    traceFile.close()

    return (options)


def main():

    parser = OptionParser(usage="usage: %prog [options]")
    parser.add_option("--emberOut",  action='store', dest="emberOutFile", help="Name of the ember output file.")
    parser.add_option("--schedPy",  action='store', dest="schedPythonFile", help="Name of the python file that holds the scheduler parameters.")
    (options, args) = parser.parse_args()


    options = grep_set_fileNames(options)
    clear_files(options)
    #set_path_emberLoad(options)
    run_sim(options)
    

if __name__ == '__main__':
    main()