from src.io_tsp import ProblemInstance
import os
# OOT = out-of-time

# Run #0 unknown

# Run #1
# alpha = 0.5, beta = 7, evap = 0.5, weight = instance.best_sol
# if instance.nPoints > 1000: ants = 550 loops = 15
# elif instance.nPoints > 195: ants = 1000 loops = 55
# else: ants = 2500 loops = 150
#   pr439=112263 pcb442=53699 d198=16199 fl1577=<OOT> ch130=6332 u1060=247703
#   kroA100=21737 eil76=554 rat783=<OOT> lin318=45580

# Run #2
# alpha, beta, evap, weight = (0.9, 8, 0.4, 100_000)
# ants, loops = (900, 6) if instance.nPoints > 1100 \
#     else (600, 15) if instance.nPoints > 1000 \
#     else (750, 30) if instance.nPoints > 700 \
#     else (975, 40) if instance.nPoints > 500 \
#     else (1000, 50) if instance.nPoints > 300 \
#     else (1100, 70) if instance.nPoints > 195 else (2700, 140)
#   pr439=111322 pcb442=52176 d198=16177 ch130=6269 u1060=246085
#   kroA100=21665 eil76=550 lin318=43675
# Separate run: fl1577=24238 (132s) rat783=9389 (174s)

dir = "/Users/maggicl/Git/AI2020BsC/c_prob/"
def ant_colony_opt(instance):
    fname = dir + instance.name + ".txt"

    # write .mat file for C++
    if not os.path.exists(fname):
        with open(fname, "w") as f:
            for i in range(instance.nPoints):
                print(" ".join(map(str, instance.dist_matrix[i])), file=f)

    alpha, beta, evap, weight = (0.9, 8, 0.4, 100_000)
    ants, loops = (1000,7) if instance.nPoints > 1100 \
        else (600,15) if instance.nPoints > 1000 \
        else (750,30) if instance.nPoints > 700 \
        else (975,40) if instance.nPoints > 500 \
        else (1000,50) if instance.nPoints > 300 \
        else (1100,70) if instance.nPoints > 195 else (2700, 140)

    # Call C++ program
    cmd = dir + "aco " + str(instance.nPoints) + " " + str(loops) + " " + str(ants) + \
        " " + str(alpha) + " " + str(beta) +  " " + str(evap) +  " " + str(weight) +" < " + fname + " &2>/dev/null"
    print(cmd)
    solution = eval(os.popen(cmd).read())
    solution.append(solution[0])
    return solution

if __name__ == "__main__":
    ant_colony_opt(ProblemInstance("../problems/eil76.tsp"))