arduino

[Lws0530]

你好 如果我想使用arduino控制 我该如何操作 在哪个模块中修改呢 如果可以的话我想跟您交换联系方式
Hello, if I want to use Arduino for control, how should I proceed and where should I make modifications in the code? If possible, I would like to exchange contact information with you.
我的微信号码是Voteme33
My WeChat number is Voteme33.

[DEVILENMO]

你只需要创建一个新的类，继承 BaseModules.py 中的 AutoAimModule 类并实现这几个方法：

class AutoAimModule:
    def __init__(self):
        raise NotImplementedError('Subclass must implement this method.')

    def auto_shoot(self, target_list: list) -> None:
        raise NotImplementedError('Subclass must implement this method.')

    def auto_aim(self, target_list: list) -> bool:
        raise NotImplementedError('Subclass must implement this method.')

然后把你自己实现的 AutoAimModule 类的子类实例化，传入位置是在 main.py 中：

if __name__ == '__main__':
    view_range = (640, 640)
    print('Initing detect module...')
    detect_module = YoloDetector('./weights/apex_v8s.pt')  # set your model file here, .pt or .trt
    print('Initing auto aiming module...')
    auto_aiming_module = DeadEyeAutoAimingModule(view_range)
    dead_eye = DeadEyeCore(detect_module, auto_aiming_module, view_range)

目前我是直接传入的 auto_aiming_module 是 DeadEyeAutoAimingModule 类的实例，这个类是我自己基于 pydirectinput 和 pynput 实现的瞄准控制类。它的定义始于 DefaultModules.py 文件中第70行：

class DeadEyeAutoAimingModule(AutoAimModule):
    ...

而等你自己的控制类实现以后，你只需要把这个位置：

auto_aiming_module = DeadEyeAutoAimingModule(view_range)

改成你自己的类就可以。
此外，你可以将这个 DeadEyeAutoAimingModule 类作为参考，这里有对函数传入参数的解析示例：

for target in target_list:
    tag, left_top, right_bottom = target.label, target.left_top, target.right_bottom

顺便提醒一句，我目前 DetectModule 只测试了 YoloDetector ，还没有时间去测试 TRTPredictor ，因此目前我只能保证使用 .pt 模型是不会出错的。

[Lws0530]

所以我需要创建一个新的Python文件来 创建一个新的类吗

[DEVILENMO]

都可以，看你个人，但是新创建一个文件再写一个新的类来继承 AutoAimModule 基类会更加整洁。对你来说，你只需要处理函数传入的 target_list 参数，然后告诉你的 arduino 该做什么就可以，其他工作都由我的框架完成了。你可以理解为我项目的其他部分和 arduino 或者其他的什么操作键鼠的库是完全隔离的，对于你提出的要求，只需要专注于处理目标然后选择是开枪或是移动准星。

[DEVILENMO]

更新后只需要修改实现 DefaultModules.py 中的 SimpleMouseController 即可。

class SimpleMouseController(MouseControlModule):
    def __init__(self):
        # mouse controller
        self.mouse_controller = mouse.Controller()

    def click_left_button(self):
        # ToDo: You are required to implement a mouse click function yourself
        pass

    def move_mouse(self, x: int, y: int):
        # ToDo: You are required to implement a mouse move function yourself
        pass

[Lws0530]

你好 可以帮我看一下添加树莓派的代码正不正确吗

import serial

def main():
arduino = serial.Serial('COM4', 115200, timeout=0)

def del(self):
self.close_connection()

def close_connection(self):
    if self.com_type == 'serial':
        if self.board is not None:
            self.board.close()
def move(self, x, y):
    # Add the remainder from the previous calculation
    x += self.remainder_x
    y += self.remainder_y

    # Round x and y, and calculate the new remainder
    self.remainder_x = x
    self.remainder_y = y
    x = int(x)
    y = int(y)
    self.remainder_x -= x
    self.remainder_y -= y

    if x != 0 or y != 0:
        match self.com_type:
            case 'socket' | 'serial':
                self.send_command(f'M{x},{y}\r')
                print(f'M({x}, {y})')

def click(self, delay_before_click=0):
    if (
            not self.click_thread.is_alive() and
            time.time() - self.last_click_time >= 1 / self.target_cps
    ):
        self.click_thread = threading.Thread(target=self.send_click, args=(delay_before_click,))
        self.click_thread.start()

def send_click(self, delay_before_click=0):
    time.sleep(delay_before_click)
    self.last_click_time = time.time()
    match self.com_type:
        case 'socket' | 'serial':
            self.send_command('C\r')
            print(f'C({random_delay * 1000:g})')
    time.sleep((np.random.randint(10) + 25) / 1000)  # Sleep to avoid sending another click instantly after mouseup

def send_command(self, command):
    with self.lock:
        match self.com_type:
            case 'socket':
                self.client.sendall(command.encode())
            case 'serial':
                self.board.write(command.encode())
        print(f'Sent: {command}')
        print(f'Response from {self.com_type}: {self.get_response()}')

def get_response(self):  # Waits for a response before sending a new instruction
    match self.com_type:
        case 'socket':
            return self.client.recv(4).decode()
        case 'serial':
            while True:
                receive = self.board.readline().decode('utf-8').strip()
                if len(receive) > 0:
                    return receive

[Lws0530]

实在抱歉 我有点小白 不是很清楚要怎么填写到文件里 可以帮我精简一下吗

[DEVILENMO]

实在抱歉 我有点小白 不是很清楚要怎么填写到文件里 可以帮我精简一下吗

如果你只是需要实现瞄准模块，参考你的代码，我尝试实现了一下

import serial
import threading
import time
import numpy as np

class ArduinoMouseController(MouseControlModule):
    def __init__(self, port='COM4', baudrate=115200):
        self.board = serial.Serial(port, baudrate, timeout=0)
        self.lock = threading.Lock()
        
        # 用于处理小数部分的移动
        self.remainder_x = 0
        self.remainder_y = 0
        
        # 点击相关变量
        self.last_click_time = 0
        self.target_cps = 1  # 控制点击频率
        self.click_thread = threading.Thread()

    def move_mouse(self, x: int, y: int):
        # 处理小数部分
        x += self.remainder_x
        y += self.remainder_y
        
        # 保存小数部分到下次使用
        self.remainder_x = x - int(x)
        self.remainder_y = y - int(y)
        
        x = int(x)
        y = int(y)
        
        if x != 0 or y != 0:
            self.send_command(f'M{x},{y}\r')

    def click_left_button(self):
        if (not self.click_thread.is_alive() and 
            time.time() - self.last_click_time >= 1 / self.target_cps):
            self.click_thread = threading.Thread(target=self._send_click)
            self.click_thread.start()

    def _send_click(self):
        self.last_click_time = time.time()
        self.send_command('C\r')
        # 点击后的随机延迟，避免固定间隔
        time.sleep((np.random.randint(10) + 25) / 1000)

    def send_command(self, command):
        with self.lock:
            self.board.write(command.encode())
            print(f'Sent: {command}')

    def __del__(self):
        if hasattr(self, 'board') and self.board is not None:
            self.board.close()

之后，你需要在 DefaultModules.py 的120行处把这个新模块替换我写的鼠标瞄准模块：

self.mouse_controller = ArduinoMouseController()

如果有错误，你自己稍微修复一下应该就可以

[Lws0530]

好的感谢您

[DEVILENMO]

好的感谢您

不客气