fix logic

Author: duelist-dev

examples/mouse.py

@@ -4,7 +4,7 @@
 from src.arduino_hid_controller.constants import MOUSE_RIGHT, MOUSE_LEFT
 
 with HIDController() as hid:
-    print(f"Текущая позиция мыши: {hid.get_position()}")
+    print(f"Текущая позиция мыши: {hid.mouse.get_position()}")
 
     print('Ожидание 2 секунды...')
     sleep(2)
@@ -15,23 +15,12 @@
     print('Ожидание 2 секунды...')
     sleep(2)
 
-    print('Перемещение мыши в координаты 500,300 и клик правой кнопкой')
-    hid.mouse.move_absolute(500, 300, 1)
-    hid.mouse.click(MOUSE_RIGHT)
-
-    print('Ожидание 2 секунды...')
-    sleep(2)
-
-    print('Перемещение мыши в координаты 100,100 и клик левой кнопкой')
-    hid.mouse.move_absolute(100, 100, 1)
-    hid.mouse.click(MOUSE_LEFT)
-
-    print('Ожидание 2 секунды...')
-    sleep(2)
-
-    print('Перемещение на 100 пикселей вправо/вниз и клик левой кнопкой')
-    hid.mouse.move_relative(100, 100)
-    hid.mouse.click()
+    # Square movement pattern
+    points = [(100, 100), (100, 200), (200, 200), (200, 100), (100, 100)]
+    for x, y in points:
+        hid.mouse.move_absolute(x, y, duration=2)
+        sleep(1)
+        #hid.mouse.click(MOUSE_LEFT)
 
     print('Ожидание 2 секунды...')
     sleep(2)

sketches/adruino-hid.ino

@@ -43,7 +43,7 @@ void loop() {
 
 Device parseDevice(const String& deviceStr) {
     if (deviceStr == F("keyboard")) return DEVICE_KEYBOARD;
-    if (actionStr == F("press")) return DEVICE_MOUSE;
+    if (deviceStr == F("mouse")) return DEVICE_MOUSE;
     return DEVICE_UNKNOWN;
 }
 
@@ -195,7 +195,7 @@ bool handleKeyboard(ActionKeyboard action, const String& args) {
               Serial.println(F("Error: Missing keyboard text to print"));
               return false;
           }
-          Keyboard.print(args)
+          Keyboard.print(args);
           return true;
       }
       default:
@@ -244,7 +244,10 @@ String separateString(const String& data, char separator, int index, bool getRem
             }
             found++;
             strIndex[0] = strIndex[1] + 1;
-            strIndex[1] = (i == maxIndex) ? i + 1 : i;
+            strIndex[1] = i;
+            if (i == maxIndex && data.charAt(i) != separator) {
+                strIndex[1] = i + 1;
+            }
         }
     }
     return found > index ? data.substring(strIndex[0], strIndex[1]) : "";

src/arduino_hid_controller/arduino_controller.py

@@ -78,9 +78,9 @@ def _send_command(self, device: str, action: str, *args) -> bool:
             self.__logger.warning("Попытка отправить команду без соединения")
             return False
 
-        command = f"{device}|{action}"
+        command = f"{device}|{action}|"
         if args:
-            command += "|" + "|".join(str(arg) for arg in args)
+            command += "|".join(str(arg) for arg in args)
 
         try:
             self.__serial.write(f"{command}\n".encode())

src/arduino_hid_controller/hid_controller.py

@@ -1,3 +1,4 @@
+from .arduino_controller import ArduinoController
 from .keyboard_controller import KeyboardController
 from .mouse_controller import MouseController
 
@@ -9,5 +10,12 @@ def __init__(self):
         """
         Инициализация контроллера HID-устройств
         """
-        self.keyboard = KeyboardController()
-        self.mouse = MouseController()
+        self._arduino = ArduinoController()
+        self.keyboard = KeyboardController(self._arduino)
+        self.mouse = MouseController(self._arduino)
+
+    def __enter__(self):
+        return self
+
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        return self

src/arduino_hid_controller/keyboard_controller.py

@@ -6,22 +6,21 @@
 
 class KeyboardController(ArduinoController):
     """Класс для эмуляции клавиатуры через Arduino"""
-
-    def __init__(self):
-        super().__init__()
+    def __init__(self, arduino: ArduinoController):
+        self._arduino = arduino
         self.__logger = logging.getLogger(__name__)
         self.__is_started = False
 
     def start(self) -> bool:
         """Начать эмуляцию клавиатуры"""
-        result = self._send_command("keyboard", "start")
+        result = self._arduino._send_command("keyboard", "start")
         if result:
             self.__is_started = True
         return result
 
     def stop(self) -> bool:
         """Остановить эмуляцию клавиатуры"""
-        result = self._send_command("keyboard", "stop")
+        result = self._arduino._send_command("keyboard", "stop")
         if result:
             self.__is_started = False
         return result
@@ -44,7 +43,7 @@ def press(self, key: Union[str, int]) -> bool:
         key_str = hex(key) if isinstance(key, int) else key
         if not key_str:
             self.__logger.error("Не указана клавиша для нажатия")
-        return self._send_command("keyboard", "press", key_str)
+        return self._arduino._send_command("keyboard", "press", key_str)
 
     def release(self, key: Union[str, int]) -> bool:
         """
@@ -61,7 +60,7 @@ def release(self, key: Union[str, int]) -> bool:
         if not key_str:
             self.__logger.error("Не указана клавиша для отпускания")
             return False
-        return self._send_command("keyboard", "release", key_str)
+        return self._arduino._send_command("keyboard", "release", key_str)
 
     def press_and_release(self, key: Union[str, int], delay: float = 0.05) -> bool:
         """
@@ -121,7 +120,7 @@ def release_all(self) -> bool:
         if not self.__is_started:
             self.__logger.warning("Попытка отпустить все клавиши при неактивной эмуляции")
             return False
-        return self._send_command("keyboard", "release_all")
+        return self._arduino._send_command("keyboard", "release_all")
 
     def write(self, text: str) -> bool:
         """
@@ -137,4 +136,4 @@ def write(self, text: str) -> bool:
         if not text:
             self.__logger.warning("Попытка отправить пустой текст")
             return False
-        return self._send_command("keyboard", "print", text)
+        return self._arduino._send_command("keyboard", "print", text)

src/arduino_hid_controller/mouse_controller.py

@@ -8,8 +8,8 @@
 class MouseController(ArduinoController):
     """Класс для эмуляции мыши через Arduino"""
 
-    def __init__(self):
-        super().__init__()
+    def __init__(self, arduino: ArduinoController):
+        self._arduino = arduino
         self.__logger = logging.getLogger(__name__)
         self.__current_x = None
         self.__current_y = None
@@ -38,14 +38,14 @@ def __set_positions(self):
 
     def start(self) -> bool:
         """Начать эмуляцию мыши"""
-        result = self._send_command("mouse", "start")
+        result = self._arduino._send_command("mouse", "start")
         if result:
             self.__is_started = True
         return result
 
     def stop(self) -> bool:
         """Остановить эмуляцию мыши"""
-        result = self._send_command("mouse", "stop")
+        result = self._arduino._send_command("mouse", "stop")
         if result:
             self.__is_started = False
         return result
@@ -70,7 +70,7 @@ def press(self, button: Union[str, int]) -> bool:
             self.__logger.error("Не указана кнопка мыши")
             return False
 
-        return self._send_command("mouse", "press", button_str)
+        return self._arduino._send_command("mouse", "press", button_str)
 
     def release(self, button: Union[str, int]) -> bool:
         """
@@ -88,7 +88,7 @@ def release(self, button: Union[str, int]) -> bool:
             self.__logger.error("Не указана кнопка мыши")
             return False
 
-        return self._send_command("mouse", "release", button_str)
+        return self._arduino._send_command("mouse", "release", button_str)
 
     def click(self, button: Union[str, int]) -> bool:
         """
@@ -98,54 +98,85 @@ def click(self, button: Union[str, int]) -> bool:
             button: Может быть 'left', 'right', 'middle' или код кнопки
         """
         button_str = hex(button) if isinstance(button, int) else button
-        return self._send_command("mouse", "click", button_str)
+        return self._arduino._send_command("mouse", "click", button_str)
 
+    #TODO: Работает не корректно
     def move_absolute(self, target_x: int, target_y: int, duration: float = 1.0) -> bool:
         """
-        Плавное перемещение курсора в указанные координаты за заданное время
-
-        Аргументы:
-            target_x: Конечная координата X (0 - левый край экрана)
-            target_y: Конечная координата Y (0 - верхний край экрана)
-            duration: Время перемещения в секундах (минимум 0.01)
-
-        Возвращает:
-            bool: True если перемещение успешно, False в случае ошибки
+        Усовершенствованное перемещение с коррекцией координат и проверкой отклонений.
+        Возвращает True при успешном перемещении, False при ошибке.
         """
-        # Проверка и корректировка координат
         if not self.__is_started:
             self.__logger.warning("Попытка перемещения при неактивной эмуляции")
             return False
 
         if duration <= 0:
-            self.__logger.error("Длительность должна быть положительной")
+            self.__logger.error("Некорректная длительность перемещения")
             return False
 
+        # Получаем и корректируем координаты
         self.__set_positions()
-        target_x = max(0, min(target_x, self.__screen_width - 1))
-        target_y = max(0, min(target_y, self.__screen_height - 1))
+        target_x = max(0, min(int(target_x), self.__screen_width - 1))
+        target_y = max(0, min(int(target_y), self.__screen_height - 1))
 
-        if target_x == self.__current_x and target_y == self.__current_y:
+        # Проверяем необходимость перемещения
+        if (target_x, target_y) == (self.__current_x, self.__current_y):
             return True
 
-        steps = max(1, int(min(duration, 300.0) * 60))
+        # Рассчитываем общее перемещение
+        total_x = target_x - self.__current_x
+        total_y = target_y - self.__current_y
+
+        # Оптимальные параметры перемещения
+        steps = max(1, min(int(duration * 60), 300))  # 60 шагов/сек, макс 300 шагов
         step_delay = duration / steps
-        step_x = (target_x - self.__current_x) / steps
-        step_y = (target_y - self.__current_y) / steps
+        max_deviation = 5  # Максимально допустимое отклонение в пикселях
+
+        # Основной цикл перемещения с коррекцией
+        for step in range(1, steps + 1):
+            # Плавное движение с ускорением/замедлением
+            progress = step / steps
+            eased_progress = progress  # Можно изменить на ease-in/out функцию
 
-        for i in range(steps):
-            new_x = int(self.__current_x + step_x * (i + 1))
-            new_y = int(self.__current_y + step_y * (i + 1))
-            rel_x = new_x - self.__current_x
-            rel_y = new_y - self.__current_y
+            # Целевые координаты на текущем шаге
+            new_x = self.__current_x + total_x * eased_progress
+            new_y = self.__current_y + total_y * eased_progress
+
+            # Относительное перемещение
+            rel_x = round(new_x - self.__current_x)
+            rel_y = round(new_y - self.__current_y)
 
             if rel_x != 0 or rel_y != 0:
-                if not self._send_command("mouse", "move", rel_x, rel_y):
+                # Отправляем команду перемещения
+                if not self._arduino._send_command("mouse", "move", rel_x, rel_y):
+                    self.__logger.error(f"Ошибка перемещения на шаге {step}")
                     return False
 
-                self.__current_x = new_x
-                self.__current_y = new_y
-                time.sleep(step_delay)
+                # Обновляем текущую позицию
+                self.__current_x += rel_x
+                self.__current_y += rel_y
+
+                # Проверка отклонения (дополнительная страховка)
+                expected_x = self.__current_x + total_x * eased_progress
+                expected_y = self.__current_y + total_y * eased_progress
+                deviation = ((self.__current_x - expected_x) ** 2 +
+                             (self.__current_y - expected_y) ** 2) ** 0.5
+
+                if deviation > max_deviation:
+                    self.__logger.warning(f"Коррекция отклонения: {deviation:.1f} пикселей")
+                    return self.move_absolute(target_x, target_y, duration / 2)
+
+            time.sleep(step_delay)
+
+        # Финальная коррекция
+        final_rel_x = target_x - self.__current_x
+        final_rel_y = target_y - self.__current_y
+        if final_rel_x != 0 or final_rel_y != 0:
+            success = self._arduino._send_command("mouse", "move", final_rel_x, final_rel_y)
+            if success:
+                self.__current_x = target_x
+                self.__current_y = target_y
+            return success
 
         return True
 
@@ -161,7 +192,7 @@ def move_relative(self, x: int, y: int) -> bool:
             self.__logger.warning("Попытка перемещения при неактивной эмуляции")
             return False
 
-        return self._send_command("mouse", "move", x, y)
+        return self._arduino._send_command("mouse", "move", x, y)
 
     def get_position(self) -> Tuple[int, int]:
         """Получить текущие виртуальные координаты курсора"""