forked from github/esp8266-geigercounter
154 lines
3.3 KiB
C++
154 lines
3.3 KiB
C++
#include <ESP8266WiFi.h>
|
|
#include <PubSubClient.h>
|
|
#include <SoftwareSerial.h>
|
|
#include <SimpleTimer.h>
|
|
|
|
#include "settings.h"
|
|
|
|
WiFiClient wifiClient;
|
|
PubSubClient mqttClient;
|
|
SoftwareSerial geigerCounterSerial(PIN_UART_RX, PIN_UART_TX);
|
|
SimpleTimer timer;
|
|
|
|
String serialInput = "";
|
|
char serialInputHelper[RECV_LINE_SIZE];
|
|
|
|
int lastCPM = 0, currentCPM = 0;
|
|
float lastuSv = 0, currentuSv = 0;
|
|
|
|
void setup() {
|
|
|
|
Serial.begin(115200);
|
|
geigerCounterSerial.begin(BAUD_GEIGERCOUNTER);
|
|
|
|
delay(10);
|
|
|
|
Serial.print("Connecting to ");
|
|
Serial.println(ssid);
|
|
|
|
WiFi.begin(ssid, password);
|
|
|
|
while (WiFi.status() != WL_CONNECTED) {
|
|
Serial.print(".");
|
|
delay(500);
|
|
}
|
|
|
|
Serial.println("");
|
|
Serial.println("WiFi connected");
|
|
Serial.println("IP address: ");
|
|
Serial.println(WiFi.localIP());
|
|
|
|
timer.setInterval(UPDATE_INTERVAL_SECONDS * 1000L, updateRadiationValues);
|
|
}
|
|
|
|
void updateRadiationValues() {
|
|
|
|
|
|
char tmp[8];
|
|
|
|
if(currentCPM != lastCPM) {
|
|
String(currentCPM).toCharArray(tmp, 8);
|
|
Serial.print("Sending CPM");
|
|
Serial.println(tmp);
|
|
mqttClient.publish(MQTT_TOPIC_CPM, tmp, true);
|
|
}
|
|
|
|
if(currentuSv != lastuSv) {
|
|
String(currentuSv).toCharArray(tmp, 8);
|
|
Serial.print("Sending uSv");
|
|
Serial.println(tmp);
|
|
mqttClient.publish(MQTT_TOPIC_USV, tmp, true);
|
|
}
|
|
|
|
lastCPM = currentCPM;
|
|
lastuSv = currentuSv;
|
|
}
|
|
|
|
void connectMqtt() {
|
|
|
|
bool newConnection = false;
|
|
|
|
while (!mqttClient.connected()) {
|
|
mqttClient.setClient(wifiClient);
|
|
mqttClient.setServer(mqttHost, 1883);
|
|
mqttClient.connect("geigercounter", MQTT_TOPIC_LAST_WILL, 1, true, "disconnected");
|
|
|
|
delay(1000);
|
|
newConnection = true;
|
|
}
|
|
|
|
if(newConnection) {
|
|
mqttClient.publish(MQTT_TOPIC_LAST_WILL, "connected", true);
|
|
}
|
|
|
|
}
|
|
|
|
void parseReceivedLine(char* input) {
|
|
|
|
char segment = 0;
|
|
char *token;
|
|
|
|
float uSv = 0;
|
|
float cpm = 0;
|
|
|
|
token = strtok(input, delimiter);
|
|
|
|
while (token != NULL) {
|
|
|
|
switch(segment) {
|
|
|
|
// This is just for validation
|
|
case IDX_CPS_KEY: if (strcmp(token, "CPS") != 0) return; break;
|
|
case IDX_CPM_KEY: if (strcmp(token, "CPM") != 0) return; break;
|
|
case IDX_uSv_KEY: if (strcmp(token, "uSv/hr") != 0) return; break;
|
|
|
|
case IDX_CPM:
|
|
Serial.printf("Current CPM: %s\n", token);
|
|
cpm = String(token).toInt();
|
|
break;
|
|
|
|
case IDX_uSv:
|
|
Serial.printf("Current uSv/hr: %s\n", token);
|
|
uSv = String(token).toFloat();
|
|
break;
|
|
}
|
|
|
|
if(segment > 7) {
|
|
// Invalid! There should be no more than 7 segments
|
|
return;
|
|
}
|
|
|
|
token = strtok(NULL, delimiter);
|
|
segment++;
|
|
}
|
|
|
|
currentuSv = uSv;
|
|
currentCPM = cpm;
|
|
}
|
|
|
|
void loop() {
|
|
|
|
connectMqtt();
|
|
timer.run();
|
|
mqttClient.loop();
|
|
|
|
if (geigerCounterSerial.available()) {
|
|
char in = (char) geigerCounterSerial.read();
|
|
|
|
serialInput += in;
|
|
|
|
if (in == '\n') {
|
|
serialInput.toCharArray(serialInputHelper, RECV_LINE_SIZE);
|
|
parseReceivedLine(serialInputHelper);
|
|
|
|
serialInput = "";
|
|
}
|
|
|
|
// Just in case the buffer gets to big, start from scratch
|
|
if(serialInput.length() > RECV_LINE_SIZE + 10) {
|
|
serialInput = "";
|
|
}
|
|
|
|
Serial.write(in);
|
|
}
|
|
}
|