initial commit

ROFLS_Arena_Controller/ROFLS_Arena_Controller.ino

@@ -0,0 +1,217 @@
+// ROFLS+ Arena Controller
+
+#include <WiFi.h>
+#include <esp_now.h> // automatically installed for ESP32 boards, I think?
+#include <Preferences.h> // automatically installed for ESP32 boards
+#include <FastLED.h> // https://fastled.io/
+#include <avdweb_Switch.h> // https://github.com/avdwebLibraries/avdweb_Switch
+#include <CountDown.h> // https://github.com/RobTillaart/CountDown
+#include <StopWatch.h> // https://github.com/RobTillaart/StopWatch_RT
+
+// Hardware connections
+#define START_BTN_PIN 1
+#define PAUSE_BTN_PIN 2
+#define PIT_BTN_PIN 3
+#define RESET_BTN_PIN 4
+#define RUMBLE_SWITCH_PIN 5
+#define PIT_ENABLE_SWITCH_PIN 6
+#define RED_TEAM_BTN_PIN 7
+#define RED_TEAM_LED_PIN 39
+#define BLUE_TEAM_BTN_PIN 6
+#define BLUE_TEAM_LED_PIN 40
+#define PIT_RELEASE_PIN 35
+
+// LED Strip setup
+#define NUM_LEDS_PER_DIGIT 56
+#define NUM_OF_DIGITS 4
+#define NUM_LEDS_COLON 8
+#define NUM_LEDS_TIMER (NUM_LEDS_PER_DIGIT * NUM_OF_DIGITS + NUM_LEDS_COLON) // + 1 because of the makeshift Levelshifter
+#define LED_DATA_PIN_TIMER 16
+// This is an array of leds.  One item for each led in your strip.
+CRGB leds_TIMER[NUM_LEDS_TIMER];
+
+// define buttons and switches
+Switch buttonSTART = Switch(START_BTN_PIN);
+Switch buttonPAUSE = Switch(PAUSE_BTN_PIN);
+Switch buttonPIT = Switch(PIT_BTN_PIN);
+Switch buttonRESET = Switch(RESET_BTN_PIN);
+Switch switchRUMBLE = Switch(RUMBLE_SWITCH_PIN);
+Switch switchPIT = Switch(PIT_ENABLE_SWITCH_PIN);
+Switch buttonREDTEAM = Switch(RED_TEAM_BTN_PIN);
+Switch buttonBLUETEAM = Switch(BLUE_TEAM_BTN_PIN);
+
+bool buttonSTARTvar = false;
+bool buttonPAUSEvar = false;
+bool buttonPITvar = false;
+bool buttonRESETvar = false;
+bool buttonREDTEAMvar = false;
+bool buttonBLUETEAMvar = false;
+unsigned long PITopenTimestamp = 0;
+const long PITopenTime = 1000; // activate solenoid for 1000ms
+bool PITreleased = false;
+
+const int countdownTIME = 3; // countdown timer length in minutes
+const int countdownToFight = 3; // countdown timer length in seconds
+const int PITreleaseTime = 90; // automatic pit release time in seconds until end of countdown
+bool countdownPAUSED = false;
+CountDown FightCountDown[1];
+
+// Rumble stopwatch
+StopWatch rumbleTIME;
+
+int prevMINUTES = 0;
+int prevSECONDS = 0;
+int prevCLOCKRED = 0;
+int prevCLOCKGREEN = 0;
+int prevCLOCKBLUE = 0;
+
+int CLOCK_LED_BRIGHTNESS = 16; // 64 is okay
+
+int BLINK_COUNTER_REDTEAM = 0;
+int BLINK_COUNTER_BLUETEAM = 0;
+
+bool ARENA_READY = true;
+bool REDTEAM_READY = false;
+bool BLUETEAM_READY = false;
+
+// ESP-NOW config
+
+// Structure example to send data
+// Must match the receiver structure
+typedef struct struct_message {
+  bool buttonSTARTremote;
+  bool buttonPAUSEremote;
+  bool buttonPITremote;
+  bool buttonRESETremote;
+  bool buttonREDTEAMremote;
+  bool buttonBLUETEAMremote;
+} struct_message;
+
+// Create a struct_message called remoteDATA
+struct_message remoteDATA;
+
+// callback function that will be executed when data is received
+void OnDataRecv(const uint8_t * mac, const uint8_t *incomingData, int len) {
+  memcpy(&remoteDATA, incomingData, sizeof(remoteDATA));
+  if (remoteDATA.buttonSTARTremote) {
+    buttonSTARTvar = true;
+  }
+  if (remoteDATA.buttonPAUSEremote) {
+    buttonPAUSEvar = true;
+  }
+  if (remoteDATA.buttonPITremote) {
+    buttonPITvar = true;
+  }
+  if (remoteDATA.buttonRESETremote) {
+    buttonRESETvar = true;
+  }
+}
+
+void setup() {
+  // set outputs
+  pinMode(RED_TEAM_LED_PIN, OUTPUT);
+  pinMode(BLUE_TEAM_LED_PIN, OUTPUT);
+  pinMode(PIT_RELEASE_PIN, OUTPUT);
+
+  digitalWrite(BLUE_TEAM_LED_PIN, LOW);
+  digitalWrite(RED_TEAM_LED_PIN, LOW);
+  digitalWrite(PIT_RELEASE_PIN, LOW);
+
+  Serial.begin(115200);
+
+  // Set device as a Wi-Fi Station
+  WiFi.mode(WIFI_STA);
+
+  // Init ESP-NOW
+  if (esp_now_init() != ESP_OK) {
+    Serial.println("Error initializing ESP-NOW");
+    return;
+  }
+  // Once ESPNow is successfully Init, we will register for recv CB to
+  // get recv packer info
+  esp_now_register_recv_cb(esp_now_recv_cb_t(OnDataRecv));
+
+
+  // set rumble stopwatch resolution to seconds
+
+	// sanity check delay - allows reprogramming if accidently blowing power w/leds
+  delay(2000);
+  // This function sets up the leds and tells the controller about them
+  FastLED.addLeds<WS2811Controller800Khz, LED_DATA_PIN_TIMER, GRB>(leds_TIMER, NUM_LEDS_TIMER);  // GRB ordering is typical
+  //FastLED.setMaxRefreshRate(10, true);
+  FastLED.setMaxPowerInVoltsAndMilliamps(5,2000); // Limit to 10W of output power
+}
+
+int XDAS = 255;
+bool toggle = false;
+void loop() {
+  // poll all the switch/button inputs
+  pollInput();
+  // deactivate solenoids if needed
+  checkPIT();
+
+  // Normal fight routine
+  while (!switchRUMBLE.on()) {
+    // poll all the switch/button inputs
+    pollInput();
+    // deactivate solenoids if needed
+    checkPIT();
+    // start button logic
+    if (buttonSTARTvar) {
+      buttonSTARTvar = false;
+      if (!FightCountDown[0].isRunning() && countdownPAUSED == false) {
+        if (buttonREDTEAMvar && buttonBLUETEAMvar) {
+          buttonREDTEAMvar = false;
+          buttonBLUETEAMvar = false;
+          FightCountDown[0].start(0,0,countdownTIME,0);
+        }
+      }
+      else {
+        if (buttonREDTEAMvar && buttonBLUETEAMvar) {
+          buttonREDTEAMvar = false;
+          buttonBLUETEAMvar = false;
+          countdownPAUSED = false;
+          FightCountDown[0].cont();
+        }
+      }
+    }
+    // pause button logic
+    if (buttonPAUSEvar) {
+      buttonPAUSEvar = false;
+      if (FightCountDown[0].isRunning()) {
+        countdownPAUSED = true;
+        FightCountDown[0].stop();
+      }
+    }
+    // pit button logic
+    if (buttonPITvar) {
+      buttonPITvar = false;
+      openPIT();
+    }
+    // automatic pit release
+    if (FightCountDown[0].remaining() <= PITreleaseTime && FightCountDown[0].remaining() > 0 && switchPIT.on()) {
+      openPIT();
+    }
+    // reset button logic
+    if (buttonRESETvar) {
+      buttonRESETvar = false;
+      PITreleased = false;
+      ESP.restart();
+    }
+    blink_LED_RedTeam(200);
+    blink_LED_BlueTeam(200);
+    // update the LED Display
+    showTimeDisplay((FightCountDown[0].remaining()/60%10), (FightCountDown[0].remaining()%60), 0, XDAS, 0);
+  }
+
+
+  // Rumble fightroutine
+  while (switchRUMBLE.on()) {
+    // poll all the switch/button inputs
+    pollInput();
+    // deactivate solenoids if needed
+    checkPIT();
+    // update the LED Display
+    showTimeDisplay(12, 34, XDAS, 0, XDAS);
+  }
+}