1 files changed, 107 insertions, 0 deletions
diff --git a/ACO.py b/ACO.py
new file mode 100755
index 0000000..3ec1ffb
--- /dev/null
+++ b/ACO.py
@@ -0,0 +1,107 @@
+#!/usr/bin/env python3
+import math
+import time
+import random
+
+
+Nodes = [
+    [0, 10, 20, 30, 40],
+    [10, 0, 20, 30, 40],
+    [20, 20, 0, 30, 40],
+    [30, 30, 30, 0, 40],
+    [40, 40, 40, 40, 0],
+]
+
+Pheromone = [
+    [0, 0, 0, 0, 0],
+    [0, 0, 0, 0, 0],
+    [0, 0, 0, 0, 0],
+    [0, 0, 0, 0, 0],
+    [0, 0, 0, 0, 0],
+]
+
+
+class Ant:
+    def __init__(self, node):
+        self.node = node
+        self.route = int()
+        self.exclusion_list = list()
+
+    def __repr__(self):
+        return "{node: %s}" % (self.node)
+
+    def __str__(self):
+        return "{node: %s}" % (self.node)
+
+    def nextHop(self):
+        n = math.ceil(random.random() * len(Nodes)) - 1
+        if (
+            n not in self.exclusion_list
+            and n != self.exclusion_list[-1]
+            and Nodes[self.exclusion_list[-1]][n] != 0
+        ):
+            self.exclusion_list.append(n)
+            return n
+        else:
+            if len(self.exclusion_list) == len(Nodes):
+                return None
+            else:
+                self.nextHop()
+
+    def nextHop2(self):
+        pass
+
+    def walk(self):
+        self.exclusion_list.append(self.node)
+        for _ in range(0, len(Nodes)):
+            next = self.nextHop()
+            if next is not None:
+                self.route += Nodes[self.node][next]
+                self.node = next
+
+    def printRoute(self):
+        print(self.exclusion_list)
+
+
+class ACO:
+    def __init__(self, ant_count):
+        self.ant_count = ant_count
+        self.Ants = self.factory()
+
+    def factory(self):
+        result = list()
+        for _ in range(0, self.ant_count):
+            node = math.ceil(random.random() * 5) - 1
+            result.append(Ant(node))
+        return result
+
+    def fitness(self):
+        pass
+
+    def run(self):
+        for ant in self.Ants:
+            ant.walk()
+            ant.printRoute()
+            for i in range(0, len(Nodes) - 1):
+                Pheromone[ant.exclusion_list[i]][ant.exclusion_list[i + 1]] = (
+                    1 / ant.route
+                )
+
+
+def main():
+    random.seed(time.time())
+    rho = 0
+    alpha = 1
+    beta = 1
+    aco = ACO(5)
+    # for ant in aco.Ants:
+    #     print("ant:", ant)
+    aco.run()
+    for i in range(0, len(Pheromone)):
+        for j in range(0, len(Pheromone)):
+            print(Pheromone[i][j], end=" ")
+        print()
+
+
+if __name__ == "__main__":
+    main()