Added structure directory and improved sort by fitness

src/structure/population.py

@@ -0,0 +1,79 @@
+class Population:
+
+    def __init__(self, chromosome_list = None):
+        """Intiialize the population with fitness of value None, and a set of chromosomes dependant on user-passed parameter"""
+
+        if chromosome_list is None:
+          self.chromosome_list = []
+        else:
+          self.chromosome_list = chromosome_list
+
+        self.fitness = None
+        self.mating_pool = []
+
+
+    def size(self):
+        """Returns the size of the population"""
+        return len(self.chromosome_list)
+
+
+    def get_closet_fitness(self,value):
+        """Get the chomosome that has the closets fitness to the value defined"""
+        pass
+
+
+    def add_chromosome(self, chromosome, index = -1):
+        """Adds a chromosome to the population at the input index, defaulted to the end of the chromosome set"""
+        if index == -1:
+            index = len(self.chromosome_list)
+        self.chromosome_list.insert(index, chromosome)
+
+
+    def remove_chromosome(self, index):
+        """removes a chromosome from the indicated index"""
+        del self.chromosome_list[index]
+
+
+    def get_all_chromosomes(self):
+        """returns all chromosomes in the population"""
+        return self.chromosome_list
+
+
+    def get_fitness(self):
+        """returns the population's fitness"""
+        return self.fitness
+
+
+    def set_all_chromosomes(self, chromosomes):
+        self.chromosome_list = chromosomes
+
+
+    def set_chromosome(self, chromosome, index = -1):
+        if index == -1:
+            index = len(self.chromosomes)-1
+        self.chromosome_list[index] = chromosome
+
+
+    def set_fitness(self, fitness):
+        """Sets the fitness value of the population"""
+        self.fitness = fitness
+
+
+    def __repr__(self):
+        """Returns a string representation of the entire population"""
+        pass
+
+
+    def print_all(self):
+        """Prints information about the population in the following format:
+        Current population
+        Chromosome 1 - [gene][gene][gene][.etc] / Chromosome fitness -
+        Chromosome 2 - [gene][gene][gene][.etc] / Chromosome fitness -
+        etc.
+        """
+
+        print("Current population:")
+
+        for index in range(self.size()):
+            print(f'Chromosome - {index} {self.chromosome_list[index]}', end = "")
+            print(f' / Fitness = {self.chromosome_list[index].get_fitness()}')