Sebi/Sim-Tello-KI-fun
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Remove files on disk

Browse Source
This commit is contained in:
Andrew Johnson
2025-11-09 13:10:05 +01:00
parent 3cac34355f
commit 73454a03c8
3 changed files with 5 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
.gitignore vendored
View File

@@ -87,4 +87,3 @@ cython_debug/
.devcontainer/devcontainer.json .devcontainer/devcontainer.json
output/ output/
tello_recording/

2
tello_sim/command_server.py
View File

@@ -1,6 +1,6 @@
import os import os
import socket import socket
from ursina import * from ursina import * # type: ignore
from time import time from time import time
import cv2 import cv2
from ursina_adapter import UrsinaAdapter from ursina_adapter import UrsinaAdapter

15
tello_sim/ursina_adapter.py
View File

@@ -57,11 +57,6 @@ class UrsinaAdapter():
self.stream_active = False self.stream_active = False
self.is_connected = False self.is_connected = False
self.recording_folder = "tello_recording" self.recording_folder = "tello_recording"
# Create recording folder if it doesn't exist
if not os.path.exists(self.recording_folder):
os.makedirs(self.recording_folder)
self.frame_count = 0 self.frame_count = 0
self.saved_frames = [] self.saved_frames = []
self.screenshot_interval = 3 self.screenshot_interval = 3
@@ -833,7 +828,7 @@ class UrsinaAdapter():
def capture_frame(self): def capture_frame(self):
"""Capture and save the latest FPV frame from update()""" """Capture the latest FPV frame. Optionally save to disk if save_frames_to_disk is True."""
if not self.stream_active: if not self.stream_active:
print("[Capture] Stream not active. Cannot capture frame.") print("[Capture] Stream not active. Cannot capture frame.")
return return
@@ -842,11 +837,9 @@ class UrsinaAdapter():
print("[Capture] No latest frame available.") print("[Capture] No latest frame available.")
return return
frame_path = os.path.join(self.recording_folder, f"frame_{self.frame_count}.png") # Always increment frame count for tracking
cv2.imwrite(frame_path, self.latest_frame)
self.saved_frames.append(frame_path)
self.frame_count += 1 self.frame_count += 1
print(f"[Capture] Screenshot {self.frame_count} saved: {frame_path}") print(f"[Capture] Frame {self.frame_count} captured (memory only)")
def set_speed(self, x: int): def set_speed(self, x: int):
"""Set drone speed by adjusting acceleration force. """Set drone speed by adjusting acceleration force.
@@ -904,7 +897,7 @@ class UrsinaAdapter():
if self.bezier_mode: if self.bezier_mode:
t_now = time() t_now = time()
elapsed = t_now - self.bezier_start_time elapsed = t_now - self.bezier_start_time # type: ignore
t = min(1.0, elapsed / self.bezier_duration) t = min(1.0, elapsed / self.bezier_duration)
# Bézier point # Bézier point