main.py

#!/usr/bin/env python3.7

# Copyright 2022, Gurobi Optimization, LLC

# Solve a traveling salesman problem on a randomly generated set of
# points using lazy constraints.   The base MIP model only includes
# 'degree-2' constraints, requiring each node to have exactly
# two incident edges.  Solutions to this model may contain subtours -
# tours that don't visit every city.  The lazy constraint callback
# adds new constraints to cut them off.

import sys
import math
import random
from itertools import combinations
import import_function
from import_function import *
from run_scenario import *
from os import listdir
from os.path import isfile, join
from datetime import datetime
import pathlib
#exec(open("import_function.py").read())
#exec(open("run_scenario.py").read())

if len(sys.argv) == 1:
    explicit_input_folder_location  = str(pathlib.Path(__file__).parent.resolve()) + "\\inputs\\"
    explicit_output_folder_location = str(pathlib.Path(__file__).parent.resolve()) + "\\outputs\\"
    rum_lim_minutes = 30
else:
    explicit_input_folder_location  = str(pathlib.Path(sys.argv[0]).parent) + "\\inputs\\"
    explicit_output_folder_location = str(pathlib.Path(sys.argv[0]).parent) + "\\outputs\\"
    rum_lim_minutes = float(sys.argv[1])
input_file_names = [f for f in listdir(explicit_input_folder_location) if isfile(join(explicit_input_folder_location, f))]
run_outputs = []

#this will loop all the inputs we have put in the folder
count = 0
for scenario_input_file in input_file_names:
    input_objects, input_links, input_global = import_inputs(explicit_input_folder_location + scenario_input_file)
    count += 1
    print(scenario_input_file + " start at: " + str(datetime.now()) + " | " + str(count) + " / " + str(len(input_file_names)))
    run_outputs_single = run_scenario(input_objects, input_links, input_global,
                 scenario_name = scenario_input_file, 
                 rum_lim_minutes = rum_lim_minutes, 
                 disable_costly_constraints = False, 
                 force_1_to_catch_a_bus = False,
                 show_fig = False,
                 return_resolution_stats = True)
    run_outputs += [run_outputs_single]

with open(explicit_output_folder_location + 'resolution_stats_output.txt', 'w') as f:
        """Resolution Stats"""
        f.write('Resolution Stats')
        f.write('\n')
        f.write('run_id score upper_bound run_time(s) sol_count')
        f.write('\n')
        for id, scenario in zip(range(len(run_outputs)), run_outputs):
            f.write(str(id) + " ")
            f.write(scenario[0])
            f.write('\n')