feat: add bot class
This commit is contained in:
		| @@ -9,6 +9,7 @@ from PyQt6.QtCore import QObject, QThread, QTimer, pyqtSignal, pyqtSlot | ||||
| from PyQt6.QtGui import QKeyEvent | ||||
| from PyQt6.QtWidgets import QMainWindow | ||||
|  | ||||
| from src.bot import Bot | ||||
| from src.command import ApplySnapshotCommand, CarControl, Command, ControlCommand, RecordingCommand, ResetCommand | ||||
| from src.record_file import RecordFile | ||||
| from src.recorder_ui import Ui_Recorder | ||||
| @@ -162,9 +163,11 @@ class RecorderWindow(Ui_Recorder, QMainWindow): | ||||
|         self.recordDataButton.clicked.connect(self.toggle_record) | ||||
|         self.resetButton.clicked.connect(self.rollback) | ||||
|  | ||||
|         self.bot: Optional[Bot] = None | ||||
|         self.autopiloting = False | ||||
|  | ||||
|         self.autopilotButton.clicked.connect(self.toggle_autopilot) | ||||
|         self.autopilotButton.setDisabled(True) | ||||
|  | ||||
|         self.saveRecordButton.clicked.connect(self.save_record) | ||||
|  | ||||
| @@ -263,8 +266,15 @@ class RecorderWindow(Ui_Recorder, QMainWindow): | ||||
|         self.snapshots.append(snapshot) | ||||
|         self.nbrSnapshotSaved.setText(str(len(self.snapshots))) | ||||
|  | ||||
|         if self.autopiloting and self.bot is not None: | ||||
|             self.bot.on_snapshot_received(snapshot) | ||||
|  | ||||
|     def shutdown(self): | ||||
|         self.close_signal.emit() | ||||
|  | ||||
|     def send_command(self, command: Command): | ||||
|         self.send_signal.emit(command) | ||||
|  | ||||
|     def register_bot(self, bot: Bot): | ||||
|         self.bot = bot | ||||
|         self.autopilotButton.setDisabled(False) | ||||
|   | ||||
		Reference in New Issue
	
	Block a user