import sys
from ml import ml
from ml2 import ml2
from ml3 import ml3
import numpy as np
from cPickle import load
from itertools import product
from math import exp
import time


if __name__ == "__main__":
    time.sleep(3)
    if len(sys.argv) < 2:
        sys.exit("usage: {0} <file>".format(sys.argv[0]))
    root_victims, victims, non_victims, age = load(open(sys.argv[1]))
    
    # alphas = np.arange(1, 200, 10)  # parameter of the time component
    alphas = np.logspace(-6,1,num=5)
    # deltas = np.arange(.00003, .0003, .00002)  # parameter of the structural component
    deltas = np.logspace(-6,-1,num=5)
    lmbdas = np.logspace(-8,-3,num=5)
    # lmbda = 1
    with open("out.log", "w") as fh:
        for alpha, delta, lmbda in product(alphas, deltas, lmbdas):
        # for alpha, delta in product(alphas, deltas):
            beta, roots, ll = ml(root_victims, victims, non_victims, alpha, delta, lmbda)
            print "\t".join(map(str, [alpha, delta, beta, roots, ll]))
            fh.write("\t".join(map(str, [alpha, delta, beta, roots, ll])) + "\n")
            fh.flush()

    # alphas = np.arange(.15, .25, .01)  # parameter of the time component
    # # alphas = np.logspace(-4,-.5,num=15)
    # deltas = np.arange(.08, .1, .01)  # parameter of the structural component
    # # deltas = np.logspace(-2,-.1,num=15)
    # lmbda = 0.10#np.logspace(-12,-3,num=10)
    # with open("out.log", "w") as fh:
    #     for alpha, delta in product(alphas, deltas):
    #         lmbda, roots, ll = ml2(root_victims, victims, non_victims, alpha, delta, lmbda)
    #         print "\t".join(map(str, [alpha, delta, lmbda, roots, ll]))
    #         fh.write("\t".join(map(str, [alpha, delta, lmbda, roots, ll])) + "\n")
    #         fh.flush()

    # alpha = 0.0373
    # delta = 0.00518
    # beta, roots, ll = ml(root_victims, victims, non_victims, alpha, delta, 1.)
    # print "\t".join(map(str, [alpha, delta, beta, roots, ll]))

# from ml import ml
# alphas = np.arange(.01, .05, .01)  # parameter of the time component
# # alphas = np.logspace(-4,1,num=5)
# deltas = np.arange(.001, .02, .003)  # parameter of the structural component
# # deltas = np.logspace(-10,-1,num=8)
# lmbda = 1.
# with open("out.log", "w") as fh:
#     for alpha, delta in product(alphas, deltas):
#         beta, roots, ll = ml(root_victims, victims, non_victims, alpha, delta, lmbda)
#         print "\t".join(map(str, [alpha, delta, beta, roots, ll]))
#         fh.write("\t".join(map(str, [alpha, delta, beta, roots, ll])) + "\n")
#         fh.flush()