main_v6.py

import pprint as pp
import helper as h
import genetic_algorithm
import generic
import copy

# class instance
ga = genetic_algorithm.Genetic_algorithm();
gnc = generic.Generic();

if __name__ == "__main__":
    for x in range(50):
        if x == 0:
            # generate random solutions
            raw_data = gnc.read_excel();
            data = gnc.convert_dict_to_list(raw_data);
            generation = ga.generate_random_solutions(data)
        else:
            # get gene data
            gene_data_list = []
            new_chromosome_list = []
            for chromosome in next_generation:
                gene_data_list = []
                for gene in chromosome:
                    gene_data = [gene, raw_data[gene]['p'], raw_data[gene]['w'], raw_data[gene]['d'], 0 , 0 , 0, 0]
                    gene_data_list.append(gene_data)
                new_chromosome_list.append(gene_data_list)
                generation = new_chromosome_list

        # calculate fitness function
        gen_calculated = []
        for chromosome in generation:
            chro_calculated = ga.fitness_function(chromosome)
            # wrap the calculation of the generation
            gen_calculated.append(copy.deepcopy(chro_calculated))

        # create new array for chromosome and total weighted tardiness
        sort_chro = []
        for l, chromosome in enumerate(gen_calculated):
            sort_gene = []
            for m, gene in enumerate(chromosome):
                sort_gene.append(gene[0])
                if (m == 5):
                    sort_gene.append(gene[7])
            sort_chro.append(list(sort_gene))

        # sort the sort_chro
        sorted_generation = sorted(sort_chro, key=lambda x: x[6])
        h.pren(sorted_generation)

        # store the minimum Ob founded
        minimum_Ob = sorted_generation[0][6]

        # selection and crossover process for making a new generation
        next_generation_candidate = ga.selection(sorted_generation)

        # mutation process
        next_generation = ga.mutation(next_generation_candidate)