output_handler.cc

#include "output_handler.h"
#include <Arduino.h>
#include <M5Stack.h>

void DrawWing() {
  int x = 60;
  int y = 20;
  int w = 200;
  int h = 200;
  int t = 20;
  int k;
  k = (w-t)/4;
  M5.Lcd.fillTriangle(x, y, x+t, y, x + k, y+h, GREEN);
  M5.Lcd.fillTriangle(x+t, y, x+k, y+h, x+k+t, y+h, GREEN);
  M5.Lcd.fillTriangle(x+2*k, y, x+2*k+t, y, x+k, y+h, GREEN);
  M5.Lcd.fillTriangle(x+2*k+t, y, x+k, y+h, x+k+t, y+h, GREEN);
  x += 2*k;
  M5.Lcd.fillTriangle(x, y, x+t, y, x + k, y+h, GREEN);
  M5.Lcd.fillTriangle(x+t, y, x+k, y+h, x+k+t, y+h, GREEN);
  M5.Lcd.fillTriangle(x+2*k, y, x+2*k+t, y, x+k, y+h, GREEN);
  M5.Lcd.fillTriangle(x+2*k+t, y, x+k, y+h, x+k+t, y+h, GREEN);
}

void DrawRing() {
  int x = 60;
  int y = 20;
  int w = 200;
  int h = 200;
  int t = 20;
  M5.Lcd.fillEllipse(x+w/2, y+h/2, w/2, h/2, RED);
  M5.Lcd.fillEllipse(x+w/2, y+h/2, w/2-t, h/2-t, BLACK);
}

void DrawSlope() {
  int x = 60;
  int y = 20;
  int w = 200;
  int h = 200;
  int t = 20;
  M5.Lcd.fillTriangle(x+w-t*1.5, y, x+w, y, x, y+w, BLUE);
  M5.Lcd.fillTriangle(x+w, y, x, y+w, x+t*1.5, y+w, BLUE);
  M5.Lcd.fillRect(x+t, y+h-t, w-t, t, BLUE);
}

void HandleOutput(tflite::ErrorReporter* error_reporter, int kind) {
  // light (red: wing, blue: ring, green: slope)
  if (kind == 0) {
    error_reporter->Report(
        "WING:\n\r*         *         *\n\r *       * *       "
        "*\n\r  *     *   *     *\n\r   *   *     *   *\n\r    * *       "
        "* *\n\r     *         *\n\r");
    M5.Lcd.fillScreen(BLACK);
    DrawWing();
  } else if (kind == 1) {
    error_reporter->Report(
        "RING:\n\r          *\n\r       *     *\n\r     *         *\n\r "
        "   *           *\n\r     *         *\n\r       *     *\n\r      "
        "    *\n\r");
    M5.Lcd.fillScreen(BLACK);
    DrawRing();
  } else if (kind == 2) {
    error_reporter->Report(
        "SLOPE:\n\r        *\n\r       *\n\r      *\n\r     *\n\r    "
        "*\n\r   *\n\r  *\n\r * * * * * * * *\n\r");
    M5.Lcd.fillScreen(BLACK);
    DrawSlope();
  }
}