Graf dróg dla agenta

src/agent.py

@@ -1,6 +1,17 @@
 class Agent:
     def __init__(self):
-        self.graph = []
+        self.graph = {}
+        self.poi = {}
 
-    def inform(self, coordinate_set):
-        pass
+    def inform(self, roads_pos, entities):
+        for index, pos in enumerate(roads_pos):
+            for another_pos in roads_pos[index:]:
+                if pos == another_pos:
+                    continue
+                if abs(pos[0] - another_pos[0]) <= 1 and abs(pos[1] - another_pos[1]) <= 1:
+                    if pos not in self.graph.keys():
+                        self.graph[pos] = set()
+                    if another_pos not in self.graph.keys():
+                        self.graph[another_pos] = set()
+                    self.graph[pos].add(another_pos)
+                    self.graph[another_pos].add(pos)

src/simulation.py

@@ -49,10 +49,17 @@ class TruckEntity(Entity):
             self.position = proposed_pos
 
 
+class Bin:
+    def __init__(self, size, trash_type):
+        self.size = size
+        self.trash_type = trash_type
+
+
 class HouseEntity(Entity):
     def __init__(self, state, position):
         super().__init__(state, position)
         self.tile = HOUSE_SPRITES[0]
+        self.bins = []
 
 
 class DumpEntity(Entity):
@@ -68,6 +75,7 @@ class SimulationState:
         self.dumps_pos = []
         self.entities = []
 
+        # stworzenie mapy i jednostek na podstawie pliku txt
         map_path = Path("../res/map.txt")
         with open(map_path, "r") as map_file:
             map_data = map_file.readlines()
@@ -109,7 +117,6 @@ class SimulationState:
         for entity in self.entities:
             if entity.__class__ is TruckEntity:
                 entity.move(move_agent)
-                break
 
 
 class Layer:
@@ -233,7 +240,7 @@ class Interface:
         pg.display.update()
 
     def loop(self):
-        self.agent.inform(set())
+        self.agent.inform([(int(pos.x), int(pos.y)) for pos in self.simulation_state.roads_pos], self.simulation_state.entities)
         while self.run_simulation:
             self.processUserInput()
             if not self.debug_mode: