#!/usr/bin/env python

import argparse

import logging

import sys

import time

import traceback

from datetime import datetime

import numpy as np

import pandas as pd

import methods

import utils

from method import Method

from neighbor import Neighbor

def main(method, solution_file, args):

logging.shutdown()

def get_all_methods(gifts, log):

return [sub(gifts, log) for sub in Method.__subclasses__()]

if __name__ == "__main__":

start_time = time.time()

log = utils.get_logger("santa-sleigh")

Neighbor.log = log

gifts = pd.read_csv("data/gifts.csv")

methods = {method.name: method for method in get_all_methods(gifts, log)}

parser = argparse.ArgumentParser()

# general arguments

parser.add_argument("method", choices=methods.keys(), help="method to try")

parser.add_argument("--random-seed", required=False, type=int, nargs="?", const=42, help="Random seed")

# method-specific arguments

parser.add_argument("--from-file", required=False, help=

"Pattern to match files under 'data/' against which contains a solution to load as basis for the new evaluation")

parser.add_argument("--alpha", required=False, type=int,

help="Factor for temperature decrease in SA")

parser.add_argument("--temperature", required=False, type=int,

help="Initial temperature to use for SA")

args = parser.parse_args()

evaluation_id = "{}-{}".format(args.method, datetime.utcnow().strftime("%Y-%m-%d-%H:%M:%S"))

args.evaluation_id = evaluation_id

solution_file = "solutions/{}_{}.csv".format(evaluation_id, args.random_seed)

log.debug("method: '{}', args: '{}'".format(args.method, args))

np.random.seed(args.random_seed)

main(methods[args.method], solution_file, args)