Python与Arduino高效实时数据交互：基于串口通信的坐标传输教程

pip install pyserial

import serial
import time
import random # 模拟OpenCV获取的坐标数据

# 配置串口参数
# 根据你的系统和Arduino连接的端口进行修改
# Windows系统通常是'COMx'，Linux/macOS系统通常是'/dev/ttyUSBx' 或 '/dev/tty.usbmodemxxx'
SERIAL_PORT = 'COM3'  
BAUD_RATE = 9600

try:
    # 初始化串口
    ser = serial.Serial(SERIAL_PORT, BAUD_RATE, timeout=1)
    print(f"串口 {SERIAL_PORT} 以 {BAUD_RATE} 波特率打开成功。")

    while True:
        # 模拟从OpenCV获取X、Y坐标
        # 实际应用中，这里会是你的OpenCV人脸追踪逻辑
        x_coord = random.randint(0, 640)  # 假设屏幕宽度640
        y_coord = random.randint(0, 480)  # 假设屏幕高度480

        # 将坐标格式化为CSV字符串，并添加换行符作为数据包结束标志
        data_to_send = f"{x_coord},{y_coord}\n"

        # 将字符串编码为字节流发送
        ser.write(data_to_send.encode('utf-8'))
        print(f"发送数据: {data_to_send.strip()}") # .strip()去除末尾换行符，便于显示

        time.sleep(0.1) # 每100毫秒发送一次数据

except serial.SerialException as e:
    print(f"串口错误: {e}")
except KeyboardInterrupt:
    print("程序被用户中断。")
finally:
    if 'ser' in locals() and ser.is_open:
        ser.close()
        print("串口已关闭。")

// 定义缓冲区大小，足以容纳预期的数据字符串
const int BUFFER_SIZE = 64; 
char receivedChars[BUFFER_SIZE]; // 存储接收到的字符
boolean newData = false;         // 标志是否有新数据

int x_coord = 0; // 解析出的X坐标
int y_coord = 0; // 解析出的Y坐标

void setup() {
  // 初始化串口通信，波特率必须与Python端一致
  Serial.begin(9600); 
  Serial.println("Arduino已启动，等待Python数据...");
}

void loop() {
  recvWithEndMarker(); // 接收数据
  if (newData == true) {
    parseData();       // 解析数据
    // 在这里使用解析出的x_coord和y_coord来控制你的设备，例如舵机
    Serial.print("接收到X: ");
    Serial.print(x_coord);
    Serial.print(", Y: ");
    Serial.println(y_coord);

    newData = false; // 重置标志位
  }
  // 其他非阻塞任务可以在这里执行
}

// 接收以换行符（'\n'）结尾的数据包
void recvWithEndMarker() {
  static byte ndx = 0; // 索引
  char rc;             // 接收到的字符

  while (Serial.available() > 0 && newData == false) {
    rc = Serial.read(); // 读取一个字符

    if (rc != '\n') { // 如果不是换行符
      receivedChars[ndx] = rc; // 存入缓冲区
      ndx++;
      if (ndx >= BUFFER_SIZE) { // 防止缓冲区溢出
        ndx = BUFFER_SIZE - 1;
      }
    } else { // 遇到换行符，表示一个数据包结束
      receivedChars[ndx] = '\0'; // 添加字符串结束符
      newData = true;            // 设置新数据标志
      ndx = 0;                   // 重置索引
    }
  }
}

// 解析CSV格式的数据
void parseData() {
  char *strtokIndex; // 用于strtok函数的指针

  // 复制一份数据，因为strtok会修改原始字符串
  char tempChars[BUFFER_SIZE];
  strcpy(tempChars, receivedChars);

  // 获取第一个逗号前的字符串（X坐标）
  strtokIndex = strtok(tempChars, ",");
  if (strtokIndex != NULL) {
    x_coord = atoi(strtokIndex); // 转换为整数
  } else {
    // 错误处理，例如设置默认值或打印错误
    x_coord = 0; 
  }

  // 获取第二个逗号后的字符串（Y坐标）
  strtokIndex = strtok(NULL, ","); // 继续从上次strtok停止的位置查找
  if (strtokIndex != NULL) {
    y_coord = atoi(strtokIndex); // 转换为整数
  } else {
    // 错误处理
    y_coord = 0;
  }
}