github方案移植 步骤二 方案的代码部分 单片机主程序

[wukong]

#include "Arduino.h"
#include "Transformation.h"
#include "AbsMouse.h"
#include <Keyboard.h>
#include "Wire.h"

#define triggerA 2
#define triggerB 3
#define mag 9
#define lid 8
#define fireLED 4
#define warningLED 15

//采集模块实例化
#include <wukongM700.h>
//设置第一个参数为cs引脚，第二个参数为单片机采集时光枪的编号
PointsTracking irtracker(4, 1);
int irPoints[8];
bool PointFound[4];

int positionX[] = {1023, 1023, 1023, 1023};
int positionY[] = {1023, 1023, 1023, 1023};
int cornersX[] = {1023, 1023, 1023, 1023};
int cornersY[] = {1023, 1023, 1023, 1023};
boolean dataComplete = false;
int button_triggerA = 0;
int button_triggerB = 0;
int button_mag = 0;
int button_lid = 0;
int warningLEDState = 0;

//声明屏幕分辨率为1920*1080
int screenW = 1920;
int screenH = 1080;
//中心点偏移参考点
int gunCenterX = 595;
int gunCenterY = 325;

void setup() {
  pinMode(fireLED, OUTPUT);
  pinMode(warningLED, OUTPUT);
  pinMode(triggerA, INPUT_PULLUP);
  pinMode(triggerB, INPUT_PULLUP);
  pinMode(mag, INPUT_PULLUP);
  pinMode(lid, INPUT_PULLUP);
  Serial.begin(19200);
  AbsMouse.init(screenW, screenH);
  camera.begin();
  Keyboard.begin();
  digitalWrite(fireLED, LOW);
  digitalWrite(warningLED, LOW);
  irtracker.DeviceInit();
  if (irtracker.deviceIsOk) {
    Serial.println("point tracting device is ok");
    camx = irtracker.get_xpixel();
    camy = irtracker.get_ypixel();
    Serial.print("x:"); Serial.print(camx); Serial.print(" y:"); Serial.println(camy);
  }
  else {
    Serial.println("pleas check point tracting device' spi setting");
  }
}

void loop() {
  handleButtons();
  getCameraData();
}

//按键的监听与驱动
void handleButtons() {
  int triggerA_now = digitalRead(triggerA);
  int triggerB_now = digitalRead(triggerB);
  int mag_now = digitalRead(mag);
  int lid_now = digitalRead(lid);
  if (triggerA_now != button_triggerA)
  {
    button_triggerA = triggerA_now;
    if (button_triggerA == 0)
    {
      AbsMouse.press(MOUSE_LEFT);
      digitalWrite(fireLED, HIGH);
    }
    else {
      AbsMouse.release(MOUSE_LEFT);
      digitalWrite(fireLED, LOW);
    }
  }
  if (triggerB_now != button_triggerB)
  {
    button_triggerB = triggerB_now;
    if (button_triggerB == 0) {
      AbsMouse.press(MOUSE_RIGHT);
    }
    else {
      AbsMouse.release(MOUSE_RIGHT);
    }
  }
  if (lid_now != button_lid) {
    button_lid = lid_now;
    if (button_lid == 0) {
      AbsMouse.press(MOUSE_MIDDLE);
    }
    else {
      AbsMouse.release(MOUSE_MIDDLE);
    }
  }
  if (mag_now != button_mag) {
    button_mag = mag_now;
    if (button_mag == 0) {
      Keyboard.press(KEY_RETURN);
    }
  }
  else {
    Keyboard.release(KEY_RETURN);
  }
}

//获取模组光点采集
void getCameraData() {
  if (irtracker.deviceIsOk) {
    irtracker.get_mot_data(irPoints);
    for (int i = 0; i < 4; i++)
    {
      if (irPoints[2 * i] < 0 || irPoints[2 * i + 1] < 0) {
        PointFound[i] = false;
      }
      else {
        PointFound[i] = true;
      }
    }   
  }
  if (PointFound[0] && PointFound[1] && PointFound[2] && PointFound[3]) {
    for (uint t = 0; t < 4; t++) {
      positionX[t] = irPoints[2 * t];
      positionY[t] = irPoints[2 * t + 1];
    }
    int orderedX[] = {0, 1, 2, 3};
    for (int i = 0; i < 3 ; i++) {
      for (int j = i + 1; j < 4; j++) {
        if (positionX[orderedX[i]] < positionX[orderedX[j]]) {
          int temp = orderedX[i];
          orderedX[i] = orderedX[j];
          orderedX[j] = temp;
        }
      }
    }
    if (positionY[orderedX[0]] < positionY[orderedX[1]]) {
      cornersX[0] = positionX[orderedX[0]];
      cornersY[0] = positionY[orderedX[0]];
      cornersX[2] = positionX[orderedX[1]];
      cornersY[2] = positionY[orderedX[1]];
    } else {
      cornersX[0] = positionX[orderedX[1]];
      cornersY[0] = positionY[orderedX[1]];
      cornersX[2] = positionX[orderedX[0]];
      cornersY[2] = positionY[orderedX[0]];
    }

    if (positionY[orderedX[2]] < positionY[orderedX[3]]) {
      cornersX[1] = positionX[orderedX[2]];
      cornersY[1] = positionY[orderedX[2]];
      cornersX[3] = positionX[orderedX[3]];
      cornersY[3] = positionY[orderedX[3]];
    } else {
      cornersX[1] = positionX[orderedX[3]];
      cornersY[1] = positionY[orderedX[3]];
      cornersX[3] = positionX[orderedX[2]];
      cornersY[3] = positionY[orderedX[2]];
    }

    Transformation trans(cornersX, cornersY, screenW, screenH, gunCenterX, gunCenterY);
    AbsMouse.move(trans.u(), trans.v());
    setWarningLED(0);
  } else {
    setWarningLED(1);
  }
}

void setWarningLED(int x) {
  if (x == 1) {
    digitalWrite(warningLED, HIGH);
  }
  else {
    digitalWrite(warningLED, LOW);
  }
}