人脸识别门锁

#include <Arduino.h>
#include "unihiker_k10.h"
#include "AIRecognition.h"


enum DoorState {
  READY,
  ENROLLING,
  RECOGNIZING,
  GRANTED,
  DENIED
};


// Forward declarations
void setRgb(uint8_t red, uint8_t green, uint8_t blue);
void showStatus(const String &text);
void setReady();
void startEnrollment();
void startRecognition();
void onButtonAPressed();
void onButtonBPressed();
void handleRecognitionResult();

UNIHIKER_K10 k10;
AIRecognition ai;

// Face IDs are assigned from 1 upward as faces are enrolled.
// Change this only after enrolling the intended user and checking their ID.
const int AUTHORIZED_FACE_ID = 1;
const unsigned long RESULT_SHOW_MS = 3000;
const unsigned long BUTTON_DEBOUNCE_MS = 250;



DoorState doorState = READY;
unsigned long stateStartedMs = 0;
String lastStatus = "";

void setRgb(uint8_t red, uint8_t green, uint8_t blue) {
  k10.rgb->write(-1, red, green, blue);
}

void showStatus(const String &text) {
  if (text == lastStatus) {
    return;
  }
  lastStatus = text;
  // The K10 canvas is updated only when the visible status changes.
  k10.canvas->canvasClear();
  k10.canvas->canvasText(text.c_str(), 1, 0xFFFFFF);
  k10.canvas->canvasText("A: enroll   B: verify", 8, 0xFFFFFF);
  k10.canvas->updateCanvas();
}

void setReady() {
  doorState = READY;
  setRgb(0, 0, 0);
  showStatus("Door locked - show face");
}

void startEnrollment() {
  doorState = ENROLLING;
  stateStartedMs = millis();
  setRgb(0, 0, 255);  // Blue: enrollment command sent
  showStatus("Enrolling face... please wait");
  ai.sendFaceCmd(ENROLL);
}

void startRecognition() {
  doorState = RECOGNIZING;
  stateStartedMs = millis();
  setRgb(0, 0, 255);  // Blue: recognition in progress
  showStatus("Recognizing...");
  ai.sendFaceCmd(RECOGNIZE);
}

// Camera/AI operation uses the K10's button callbacks.  Keep these
// handlers brief; the actual face commands are supported by the K10 API.
void onButtonAPressed() {
  startEnrollment();
}

void onButtonBPressed() {
  startRecognition();
}

void handleRecognitionResult() {
  if (!ai.isRecognized()) {
    return;
  }

  // Read immediately: the API's recognition ID is a one-shot result.
  const int recognizedId = ai.getRecognitionID();
  stateStartedMs = millis();

  if (recognizedId == AUTHORIZED_FACE_ID) {
    doorState = GRANTED;
    setRgb(0, 255, 0);  // Green: simulated door unlocked
    showStatus("Access granted - door open");
  } else {
    doorState = DENIED;
    setRgb(255, 0, 0);  // Red: unknown or unauthorized face
    showStatus("Access denied");
  }
}

void setup() {
  k10.begin();
  k10.initScreen(2);

  ai.initAi();
  k10.initBgCamerImage();
  k10.setBgCamerImage(false);
  k10.creatCanvas();
  ai.switchAiMode(ai.NoMode);

  // Register callbacks before enabling camera/face mode, as required by
  // the K10 face-recognition example.
  k10.buttonA->setPressedCallback(onButtonAPressed);
  k10.buttonB->setPressedCallback(onButtonBPressed);

  k10.setBgCamerImage(true);
  ai.switchAiMode(ai.Face);

  k10.rgb->brightness(40);
  setReady();
}

void loop() {
  handleRecognitionResult();

  const unsigned long now = millis();
  if ((doorState == GRANTED || doorState == DENIED || doorState == ENROLLING) &&
      now - stateStartedMs >= RESULT_SHOW_MS) {
    setReady();
  }

  delay(20);
}