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

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()