Pro Micro SSD1306 ADS1115 5Button Menu FINAL

#include <Joystick.h>
#include <Wire.h>
#include <Adafruit_ADS1X15.h>
#include <Adafruit_GFX.h>
#include <Adafruit_SSD1306.h>
#include <EEPROM.h>

#define SCREEN_WIDTH 128
#define SCREEN_HEIGHT 32
#define OLED_RESET    -1
#define ADC_MAX 32767

// Button pins
#define BUTTON_CLUTCH 5
#define BUTTON_BRAKE 6
#define BUTTON_GAS 4
#define BUTTON_MENU 7
#define BUTTON_HANDBRAKE 8

// Output ranges
#define OUT_MIN 0
#define OUT_MAX 32767
#define BRAKE_PSI_MIN 0
#define BRAKE_PSI_MAX 100

// Default ADC ranges
const int16_t CLUTCH_DEFAULT_ADC_MIN = 0;
const int16_t CLUTCH_DEFAULT_ADC_MAX = 32767;
const int16_t BRAKE_DEFAULT_ADC_MIN = 4096;
const int16_t BRAKE_DEFAULT_ADC_MAX = 29491;
const int16_t GAS_DEFAULT_ADC_MIN = 0;
const int16_t GAS_DEFAULT_ADC_MAX = 32767;
const int16_t HANDBRAKE_DEFAULT_ADC_MIN = 0;
const int16_t HANDBRAKE_DEFAULT_ADC_MAX = 32767;

// Fixed PSI range
const int16_t BRAKE_PSI_ADC_MIN = 4096;
const int16_t BRAKE_PSI_ADC_MAX = 29491;

// Kg lookup table (PSI 0–100 → Kg 0–138)
const uint8_t kgTable[101] PROGMEM = {
  0, 1, 3, 4, 6, 7, 8, 10, 11, 12, 14, 15, 17, 18, 19, 21, 22, 23, 25, 26,
  28, 29, 30, 32, 33, 35, 36, 37, 39, 40, 41, 43, 44, 46, 47, 48, 50, 51, 52, 54,
  55, 57, 58, 59, 61, 62, 63, 65, 66, 68, 69, 70, 72, 73, 74, 76, 77, 79, 80, 81,
  83, 84, 85, 87, 88, 90, 91, 92, 94, 95, 96, 98, 99, 101, 102, 103, 105, 106, 107, 109,
  110, 112, 113, 114, 116, 117, 118, 120, 121, 123, 124, 125, 127, 128, 130, 131, 132, 134, 135, 136, 138
};

// EEPROM addresses
#define EEPROM_CLUTCH_MIN 0
#define EEPROM_CLUTCH_MAX 2
#define EEPROM_BRAKE_MIN 4
#define EEPROM_BRAKE_MAX 6
#define EEPROM_GAS_MIN 8
#define EEPROM_GAS_MAX 10
#define EEPROM_HANDBRAKE_MIN 12
#define EEPROM_HANDBRAKE_MAX 14
#define EEPROM_INIT_FLAG 16

// Enums for states
enum CalibrationStep { NONE = 0, MIN_CAL = 1, MAX_CAL = 2 };
enum SetMessageType { MIN_SET = 1, MAX_SET = 2, DEFAULT_SET = 3 };

// Axis data structure
struct Axis {
  int16_t adcMin;
  int16_t adcMax;
  const int16_t defaultMin;
  const int16_t defaultMax;
  bool buttonPressed;
  unsigned long pressStartTime;
};

// Global objects and variables
Adafruit_SSD1306 display(SCREEN_WIDTH, SCREEN_HEIGHT, &Wire, OLED_RESET);
Adafruit_ADS1115 ads;
Joystick_ Joystick(JOYSTICK_DEFAULT_REPORT_ID, JOYSTICK_TYPE_JOYSTICK, 0, 0, true, true, true, true, false, false, false, false);
Axis axes[4] = {
  {CLUTCH_DEFAULT_ADC_MIN, CLUTCH_DEFAULT_ADC_MAX, CLUTCH_DEFAULT_ADC_MIN, CLUTCH_DEFAULT_ADC_MAX, false, 0}, // Clutch
  {BRAKE_DEFAULT_ADC_MIN, BRAKE_DEFAULT_ADC_MAX, BRAKE_DEFAULT_ADC_MIN, BRAKE_DEFAULT_ADC_MAX, false, 0},   // Brake
  {GAS_DEFAULT_ADC_MIN, GAS_DEFAULT_ADC_MAX, GAS_DEFAULT_ADC_MIN, GAS_DEFAULT_ADC_MAX, false, 0},           // Gas
  {HANDBRAKE_DEFAULT_ADC_MIN, HANDBRAKE_DEFAULT_ADC_MAX, HANDBRAKE_DEFAULT_ADC_MIN, HANDBRAKE_DEFAULT_ADC_MAX, false, 0} // Hand brake
};
byte calibratingInput = 0;
CalibrationStep calibrationStep = NONE;
unsigned long calibrationStartTime = 0;
bool showingResetMessage = false;
unsigned long resetMessageStartTime = 0;
byte resetMessageInput = 0;
bool showingSetMessage = false;
unsigned long setMessageStartTime = 0;
SetMessageType setMessageType = MIN_SET;
byte setMessageInput = 0;
int16_t setMessageValue = 0;
int16_t lastAdc[4] = {0, 0, 0, 0};
float lastBrakePSI = -1;
uint8_t lastBrakeKg = 0;
byte currentScreen = 0;
bool menuButtonPressed = false;
unsigned long lastMenuPress = 0;
unsigned long lastUpdate = 0;
const unsigned long UPDATE_INTERVAL = 10;

void saveCalibrationToEEPROM() {
  EEPROM.put(EEPROM_CLUTCH_MIN, axes[0].adcMin);
  EEPROM.put(EEPROM_CLUTCH_MAX, axes[0].adcMax);
  EEPROM.put(EEPROM_BRAKE_MIN, axes[1].adcMin);
  EEPROM.put(EEPROM_BRAKE_MAX, axes[1].adcMax);
  EEPROM.put(EEPROM_GAS_MIN, axes[2].adcMin);
  EEPROM.put(EEPROM_GAS_MAX, axes[2].adcMax);
  EEPROM.put(EEPROM_HANDBRAKE_MIN, axes[3].adcMin);
  EEPROM.put(EEPROM_HANDBRAKE_MAX, axes[3].adcMax);
  EEPROM.update(EEPROM_INIT_FLAG, 0xAA);
}

void loadCalibrationFromEEPROM() {
  if (EEPROM.read(EEPROM_INIT_FLAG) == 0xAA) {
    EEPROM.get(EEPROM_CLUTCH_MIN, axes[0].adcMin);
    EEPROM.get(EEPROM_CLUTCH_MAX, axes[0].adcMax);
    EEPROM.get(EEPROM_BRAKE_MIN, axes[1].adcMin);
    EEPROM.get(EEPROM_BRAKE_MAX, axes[1].adcMax);
    EEPROM.get(EEPROM_GAS_MIN, axes[2].adcMin);
    EEPROM.get(EEPROM_GAS_MAX, axes[2].adcMax);
    EEPROM.get(EEPROM_HANDBRAKE_MIN, axes[3].adcMin);
    EEPROM.get(EEPROM_HANDBRAKE_MAX, axes[3].adcMax);
  } else {
    saveCalibrationToEEPROM();
  }
}

void setup() {
  Wire.begin();
  if (!display.begin(SSD1306_SWITCHCAPVCC, 0x3C) || !ads.begin()) while (1);

  ads.setGain(GAIN_TWOTHIRDS);
  display.setTextSize(1);
  display.setTextColor(SSD1306_WHITE);

  Joystick.begin();
  Joystick.setXAxisRange(OUT_MIN, OUT_MAX);
  Joystick.setYAxisRange(OUT_MIN, OUT_MAX);
  Joystick.setZAxisRange(OUT_MIN, OUT_MAX);
  Joystick.setRxAxisRange(OUT_MIN, OUT_MAX);

  // Configure used pins
  pinMode(BUTTON_CLUTCH, INPUT_PULLUP);
  pinMode(BUTTON_BRAKE, INPUT_PULLUP);
  pinMode(BUTTON_GAS, INPUT_PULLUP);
  pinMode(BUTTON_MENU, INPUT_PULLUP);
  pinMode(BUTTON_HANDBRAKE, INPUT_PULLUP);

  // Configure unused pins as INPUT_PULLUP
  pinMode(0, INPUT_PULLUP);  // RX
  pinMode(1, INPUT_PULLUP);  // TX
  pinMode(9, INPUT_PULLUP);  // Free pin
  pinMode(10, INPUT_PULLUP); // SS
  pinMode(14, INPUT_PULLUP); // MISO
  pinMode(15, INPUT_PULLUP); // SCK
  pinMode(16, INPUT_PULLUP); // MOSI
  pinMode(18, INPUT_PULLUP); // A0
  pinMode(19, INPUT_PULLUP); // A1
  pinMode(20, INPUT_PULLUP); // A2
  pinMode(21, INPUT_PULLUP); // A3

  loadCalibrationFromEEPROM();
}

void handleButton(byte index, int pin) {
  if (digitalRead(pin) == LOW) {
    if (!axes[index].buttonPressed) {
      axes[index].buttonPressed = true;
      axes[index].pressStartTime = millis();
    } else if (millis() - axes[index].pressStartTime >= 3000) {
      axes[index].adcMin = axes[index].defaultMin;
      axes[index].adcMax = axes[index].defaultMax;
      saveCalibrationToEEPROM();
      showingResetMessage = true;
      resetMessageStartTime = millis();
      resetMessageInput = index + 1;
      axes[index].buttonPressed = false;
    }
  } else if (axes[index].buttonPressed) {
    axes[index].buttonPressed = false;
    if (millis() - axes[index].pressStartTime < 3000) {
      calibratingInput = index + 1;
      calibrationStep = MIN_CAL;
      calibrationStartTime = millis();
    }
  }
}

void handleMenuButton() {
  if (digitalRead(BUTTON_MENU) == LOW) {
    if (!menuButtonPressed && millis() - lastMenuPress > 200) {
      menuButtonPressed = true;
      lastMenuPress = millis();
      currentScreen = (currentScreen + 1) % 3;
    }
  } else {
    menuButtonPressed = false;
  }
}

int16_t mapInt(int16_t x, int16_t in_min, int16_t in_max, int16_t out_min, int16_t out_max) {
  return (int16_t)(((long)(x - in_min) * (out_max - out_min)) / (in_max - in_min) + out_min);
}

void updateJoystick() {
  lastAdc[0] = ads.readADC_SingleEnded(0); // Clutch
  lastAdc[1] = ads.readADC_SingleEnded(1); // Brake
  lastAdc[2] = ads.readADC_SingleEnded(2); // Gas
  lastAdc[3] = ads.readADC_SingleEnded(3); // Hand brake

  Joystick.setXAxis(mapInt(constrain(lastAdc[0], axes[0].adcMin, axes[0].adcMax), axes[0].adcMin, axes[0].adcMax, OUT_MIN, OUT_MAX));
  Joystick.setYAxis(mapInt(constrain(lastAdc[1], axes[1].adcMin, axes[1].adcMax), axes[1].adcMin, axes[1].adcMax, OUT_MIN, OUT_MAX));
  Joystick.setZAxis(mapInt(constrain(lastAdc[2], axes[2].adcMin, axes[2].adcMax), axes[2].adcMin, axes[2].adcMax, OUT_MIN, OUT_MAX));
  Joystick.setRxAxis(mapInt(constrain(lastAdc[3], axes[3].adcMin, axes[3].adcMax), axes[3].adcMin, axes[3].adcMax, OUT_MIN, OUT_MAX));
  lastBrakePSI = mapFloat(constrain(lastAdc[1], BRAKE_PSI_ADC_MIN, BRAKE_PSI_ADC_MAX), BRAKE_PSI_ADC_MIN, BRAKE_PSI_ADC_MAX, BRAKE_PSI_MIN, BRAKE_PSI_MAX);
  lastBrakeKg = pgm_read_byte(&kgTable[(uint8_t)lastBrakePSI]);
}

void updateDisplay() {
  display.clearDisplay();
  display.setCursor(0, 0);

  if (calibratingInput != 0 && (calibrationStep == MIN_CAL || calibrationStep == MAX_CAL)) {
    unsigned long elapsed = millis() - calibrationStartTime;
    int secondsLeft = 5 - (elapsed / 1000);
    if (secondsLeft < 0) secondsLeft = 0;

    display.print(F("Hold "));
    switch (calibratingInput) {
      case 1: display.print(F("Clutch")); break;
      case 2: display.print(F("Brake")); break;
      case 3: display.print(F("Gas")); break;
      case 4: display.print(F("Handbrake")); break;
    }
    display.print(calibrationStep == MIN_CAL ? F(" MIN: ") : F(" MAX: "));
    display.print(secondsLeft);
    display.setCursor(0, 10);
    display.print(F("ADC: "));
    display.print(lastAdc[calibratingInput - 1]);

    if (elapsed >= 5000) {
      int16_t currentADC = lastAdc[calibratingInput - 1];
      if (calibrationStep == MIN_CAL) {
        axes[calibratingInput - 1].adcMin = currentADC;
        saveCalibrationToEEPROM();
        showingSetMessage = true;
        setMessageStartTime = millis();
        setMessageType = MIN_SET;
        setMessageInput = calibratingInput;
        setMessageValue = currentADC;
        calibrationStep = NONE;
      } else if (calibrationStep == MAX_CAL) {
        axes[calibratingInput - 1].adcMax = currentADC;
        saveCalibrationToEEPROM();
        showingSetMessage = true;
        setMessageStartTime = millis();
        setMessageType = MAX_SET;
        setMessageInput = calibratingInput;
        setMessageValue = currentADC;
        calibratingInput = 0;
        calibrationStep = NONE;
        showingResetMessage = false;
      }
    }
  } else if (showingSetMessage && millis() - setMessageStartTime < 2000) {
    display.print(F("DONE"));
  } else if (showingResetMessage && millis() - resetMessageStartTime < 3000) {
    switch (resetMessageInput) {
      case 1: display.print(F("Clutch")); break;
      case 2: display.print(F("Brake")); break;
      case 3: display.print(F("Gas")); break;
      case 4: display.print(F("Handbrake")); break;
    }
    display.print(F(" reset to defaults"));
  } else if (showingResetMessage && millis() - resetMessageStartTime >= 3000) {
    showingResetMessage = false;
    showingSetMessage = true;
    setMessageStartTime = millis();
    setMessageType = DEFAULT_SET;
    setMessageInput = resetMessageInput;
    setMessageValue = 0;
  } else {
    if (showingSetMessage && millis() - setMessageStartTime >= 2000) {
      showingSetMessage = false;
      setMessageInput = 0;
      setMessageValue = 0;
      if (setMessageType == MIN_SET && calibratingInput != 0) {
        calibrationStep = MAX_CAL;
        calibrationStartTime = millis();
      }
    }

    if (currentScreen == 0) { // Screen 1: Min/Max/ADC
      display.print(F("C:"));
      display.print(axes[0].adcMin);
      display.print(F(" "));
      display.print(axes[0].adcMax);
      display.print(F(" "));
      display.print(lastAdc[0]);
      display.setCursor(0, 8);
      display.print(F("B:"));
      display.print(axes[1].adcMin);
      display.print(F(" "));
      display.print(axes[1].adcMax);
      display.print(F(" "));
      display.print(lastAdc[1]);
      display.setCursor(0, 16);
      display.print(F("G:"));
      display.print(axes[2].adcMin);
      display.print(F(" "));
      display.print(axes[2].adcMax);
      display.print(F(" "));
      display.print(lastAdc[2]);
      display.setCursor(0, 24);
      display.print(F("H:"));
      display.print(axes[3].adcMin);
      display.print(F(" "));
      display.print(axes[3].adcMax);
      display.print(F(" "));
      display.print(lastAdc[3]);
    } else if (currentScreen == 1) { // Screen 2: Brake PSI and Kg
      display.print(F("Brake PSI: "));
      display.print(lastBrakePSI, 2);
      display.setCursor(0, 8);
      display.print(F("Kg: "));
      display.print(lastBrakeKg);
    } else if (currentScreen == 2) { // Screen 3: Clutch/Brake/Gas/Handbrake Percentages
      int16_t clutchPercent = mapInt(constrain(lastAdc[0], axes[0].adcMin, axes[0].adcMax), axes[0].adcMin, axes[0].adcMax, 0, 100);
      int16_t brakePercent = mapInt(constrain(lastAdc[1], axes[1].adcMin, axes[1].adcMax), axes[1].adcMin, axes[1].adcMax, 0, 100);
      int16_t gasPercent = mapInt(constrain(lastAdc[2], axes[2].adcMin, axes[2].adcMax), axes[2].adcMin, axes[2].adcMax, 0, 100);
      int16_t handbrakePercent = mapInt(constrain(lastAdc[3], axes[3].adcMin, axes[3].adcMax), axes[3].adcMin, axes[3].adcMax, 0, 100);
      display.print(F("C:"));
      display.print(clutchPercent);
      display.print(F("%"));
      display.setCursor(0, 8);
      display.print(F("B:"));
      display.print(brakePercent);
      display.print(F("%"));
      display.setCursor(0, 16);
      display.print(F("G:"));
      display.print(gasPercent);
      display.print(F("%"));
      display.setCursor(0, 24);
      display.print(F("H:"));
      display.print(handbrakePercent);
      display.print(F("%"));
    }
  }
  display.display();
}

void loop() {
  unsigned long currentMillis = millis();

  if (currentMillis - lastUpdate >= UPDATE_INTERVAL) {
    if (calibratingInput == 0 && !showingResetMessage && !showingSetMessage) {
      handleButton(0, BUTTON_CLUTCH);
      handleButton(1, BUTTON_BRAKE);
      handleButton(2, BUTTON_GAS);
      handleButton(3, BUTTON_HANDBRAKE);
    }
    handleMenuButton();
    updateJoystick();
    updateDisplay();
    lastUpdate = currentMillis;
  }
}

inline float mapFloat(float x, float in_min, float in_max, float out_min, float out_max) {
  return (x - in_min) * (out_max - out_min) / (in_max - in_min) + out_min;
}