ESPGeiger

#define VERSION "V1.5.0"

#include <WiFiManager.h>
#include <WiFi.h>
#include <SSD1306.h>
#include <esp_task_wdt.h>

#define GEIGER_INPUT_PIN 13
#define BUTTON_PIN 14
#define CONVERSION_FACTOR 154.0
#define PERIOD_LOG 15
#define CONFIG_PORTAL_TIMEOUT 120
#define WDT_TIMEOUT_MS 10000

int counts = 0;
int cpm = 0;
float usv = 0.0;
unsigned long lastCountTime, startCountTime;

SSD1306 display(0x3c, 5, 4);

void IRAM_ATTR ISR_impulse() {
  counts++;
}

// ========== OLED ==========
void displayInit() {
  display.init();
  display.flipScreenVertically();
  display.setTextAlignment(TEXT_ALIGN_LEFT);
  display.setFont(ArialMT_Plain_10);
  display.drawString(0, 0, "Radiation Monitor");
  display.display();
}

void showStatus(const String &line1, const String &line2 = "") {
  display.clear();
  display.setFont(ArialMT_Plain_10);
  display.drawString(0, 20, line1);
  display.drawString(0, 34, line2);
  display.display();
}

void showReadings(int cpm, float usv) {
  display.clear();
  display.setFont(ArialMT_Plain_10);
  display.drawString(0, 0, "Radiation Monitor");
  display.setFont(ArialMT_Plain_16);
  display.drawString(0, 18, "CPM: " + String(cpm));
  display.drawString(0, 40, "uSv/h: " + String(usv, 2));
  display.display();
}

// ========== WIFI SETUP ==========
void connectWiFi() {
  WiFi.mode(WIFI_STA);
  WiFi.begin();

  showStatus("Connecting to WiFi...");
  Serial.println("Connecting to saved WiFi...");

  unsigned long start = millis();
  while (WiFi.status() != WL_CONNECTED && millis() - start < 15000) {
    delay(250);
    Serial.print(".");
  }

  if (WiFi.status() == WL_CONNECTED) {
    Serial.print("\nConnected! IP: ");
    Serial.println(WiFi.localIP());
    showStatus("WiFi Connected", WiFi.localIP().toString());
    delay(1000);
  } else {
    Serial.println("\nWiFi not connected.");
    showStatus("WiFi not set.", "Press button to config");
  }
}

void launchConfigPortal() {
  Serial.println("Launching WiFi config portal...");

  WiFiManager wm;
  wm.setConfigPortalTimeout(CONFIG_PORTAL_TIMEOUT);

  wm.resetSettings();  // ? Clear previous credentials!

  showStatus("Starting WiFi", "Config Portal...");

  esp_task_wdt_deinit();  // Disable watchdog during portal

  if (!wm.startConfigPortal("ESPGeiger")) {
    Serial.println("WiFiManager timeout. Restarting...");
    showStatus("Config Timeout", "Rebooting...");
    delay(2000);
    ESP.restart();
  }

  Serial.println("WiFi credentials saved.");
  showStatus("WiFi Config", "Rebooting...");
  delay(2000);
  ESP.restart();
}


// ========== SETUP ==========
void setup() {
  Serial.begin(115200);
  Serial.println("Starting Geiger Counter " VERSION);

  pinMode(GEIGER_INPUT_PIN, INPUT);
  pinMode(BUTTON_PIN, INPUT_PULLUP);

  displayInit();
  connectWiFi();

  // Only attach interrupts if WiFi is working
  if (WiFi.status() == WL_CONNECTED) {
    attachInterrupt(digitalPinToInterrupt(GEIGER_INPUT_PIN), ISR_impulse, FALLING);
    startCountTime = lastCountTime = millis();
    esp_task_wdt_config_t wdt_config = {
  .timeout_ms = WDT_TIMEOUT_MS,
  .idle_core_mask = (1 << portNUM_PROCESSORS) - 1,
  .trigger_panic = true
};
esp_task_wdt_init(&wdt_config);

    esp_task_wdt_add(NULL);
  }
}

// ========== LOOP ==========
void loop() {
  // On-demand config: press button any time
  if (digitalRead(BUTTON_PIN) == LOW) {
    delay(50); // Debounce
    if (digitalRead(BUTTON_PIN) == LOW) {
      Serial.println("WiFi config button pressed.");
      launchConfigPortal();
    }
  }

  // Only run main logic if WiFi is working
  if (WiFi.status() == WL_CONNECTED) {
    esp_task_wdt_reset();

    if (millis() - lastCountTime > PERIOD_LOG * 1000) {
      unsigned long now = millis();
      cpm = (60000 * counts) / (now - startCountTime);
      usv = (float)cpm / CONVERSION_FACTOR;

      counts = 0;
      startCountTime = now;
      lastCountTime += PERIOD_LOG * 1000;

      showReadings(cpm, usv);

      Serial.print("CPM: ");
      Serial.print(cpm);
      Serial.print(" | uSv/h: ");
      Serial.println(usv, 2);
    }
  }

  delay(50);
}