ref: update handler on_complete_trajectory

main.py

@@ -4,27 +4,21 @@ from visualization import VisualizationManager
 from trajectory_drawer import TrajectoryDrawer
 from yandex_map import YandexMap
 
-# Создаем менеджер визуализации
+def main():
-# viz_manager = VisualizationManager("Drone Autopilot - Global Map & Detection")
+    global trajectoryDrawer
 
-# Создаем симулятор с AutoPilot для обработки изображений
+    yandexMap = YandexMap()
-# Передаем менеджер визуализации в автопилот
+    yandexMap.savePhoto('map.jpg')
-# simulator = Simulator(RandomPilot(), AutoPilot(viz_manager=viz_manager), viz_manager=viz_manager)
+    yandexMap.destroy()
 
-yandexMap = YandexMap()
-yandexMap.savePhoto('map.jpg')
-# yandexMap.destroy()
 
-trajectoryDrawer = TrajectoryDrawer('map.jpg')
+    trajectoryDrawer = TrajectoryDrawer('map.jpg')
-trajectoryDrawer.on_complete_trajectory = lambda x: print(x)
+    trajectoryDrawer.on_complete_trajectory = onCompleteTrajectory
-trajectoryDrawer.show()
+    trajectoryDrawer.show()
 
-# Использование
+def onCompleteTrajectory(points):
-# if __name__ == "__main__":
+    print("I've got points. Here these: ")
-#     # Укажите путь к вашему изображению
+    print(points)
-#     image_path = "map.jpg"  # Замените на путь к вашему изображению
+
-    
+if __name__ == "__main__":
-#     drawer = TrajectoryDrawer(image_path)
+    main()
-#     drawer.show()
-# Запускаем симуляцию
-# simulator.loop()

trajectory_drawer.py

@@ -60,7 +60,8 @@ class TrajectoryDrawer:
                 
             # Можно сохранить координаты в файл
             # self.save_trajectory()
-            self.on_complete_trajectory("GOOD")
+            self.on_complete_trajectory(self.points)
+            self.destroy()
             
     def clear_trajectory(self):
         self.points = []
@@ -74,6 +75,9 @@ class TrajectoryDrawer:
             for i, (x, y) in enumerate(self.points):
                 f.write(f"{i+1},{x:.4f},{y:.4f}\n")
         print("Координаты сохранены в trajectory_coordinates.txt")
+
+    def destroy(self):
+        plt.close(self.fig)
         
     def show(self):
         plt.tight_layout()