import json

from tabulate import tabulate

from PackingAlgorithm import PackingAlgorithm

from Visualization import Visualization

NUM_OF_INDIVIDUALS = 40

NUM_OF_GENERATIONS = 600

PC = 0.8

PM1 = 0.5

PM2 = 0.2

K = 4

ROTATIONS = 6

if __name__ == "__main__":

with open('input.json', 'r') as outfile:

data = json.load(outfile)

problem_indices = list(data.keys())

for p_ind in problem_indices:

visualization = Visualization(p_ind)

print("Running Problem Set {}".format(p_ind))

print(tabulate([['Generations', NUM_OF_GENERATIONS], ['Individuals', NUM_OF_INDIVIDUALS],

['Rotations', ROTATIONS], ['Crossover Prob.', PC], ['Mutation Prob1', PM1],

['Mutation Prob2', PM2], ['Tournament Size', K]], headers=['Parameter', 'Value'],

tablefmt="github"))

print()

truck_dimension = data[p_ind]['truck dimension']

boxes = data[p_ind]['boxes']

total_value = data[p_ind]['total value']

boxParams = {}

for i, params in enumerate(boxes):

boxParams[i] = params

pa = PackingAlgorithm(box_params=boxParams, truck_dimension=truck_dimension, total_value=total_value, population_size=NUM_OF_INDIVIDUALS, generations=NUM_OF_GENERATIONS,

k=K, pc=PC, pm1=PM1, pm2=PM2, rotation=ROTATIONS)

pa.optimize()

visualization.plot_average_fitness(pa.avg_fitness)

true_solution = data[p_ind]['solution']

visualization.draw_final_rank1_solutions(pa.population)

visualization.draw_plotly_final_rank1_solutions(pa.population)

visualization.draw_plotly_true_solution(true_solution)

visualization.draw_pareto(pa.population)