Implement layout

This commit is contained in:
Paul Brinkmeier 2025-04-24 19:50:26 +02:00
parent f0c94d8d7d
commit cf7bb96517

View File

@ -1,37 +1,68 @@
#include <ArduinoJson.h> #include <ArduinoJson.h>
#include <ESP8266WiFi.h>
#include <SPI.h>
#include <Wire.h>
#include <Adafruit_GFX.h> #include <Adafruit_GFX.h>
#include <Adafruit_SSD1306.h> #include <Adafruit_ST7735.h>
#include <ESP8266WiFi.h>
#include <ESP8266HTTPClient.h> #include <ESP8266HTTPClient.h>
#include <SPI.h>
#include <WiFiClientSecureBearSSL.h> #include <WiFiClientSecureBearSSL.h>
#include "bitmaps.h" #include "bitmaps.h"
#include "wifi_credentials.h" #include "wifi_credentials.h"
// Screen stuff #define TFT_CS D8
#define SCREEN_WIDTH 128 // OLED display width, in pixels #define TFT_DC D4
#define SCREEN_HEIGHT 64 // OLED display height, in pixels #define TFT_RESET D3
#define OLED_RESET -1 // Reset pin # (or -1 if sharing Arduino reset pin)
#define SCREEN_ADDRESS 0x3C
// Button stuff // Button stuff
#define BUTTON_PIN D6 #define BUTTON_PIN D6
// Display stuff
#define FRAME_DELAY 80 #define FRAME_DELAY 80
#define COLOR_BG (rgb(0, 0, 0))
#define COLOR_DIVIDER (rgb(0xff, 0x60, 0x0d))
#define COLOR_TEXT (rgb(0xff, 0x60, 0x0d))
// Layout
#define WIDTH 160
#define HEIGHT 128
#define CX 6
#define CY 8
#define MARGIN_LEFT 5
#define MARGIN_RIGHT 5
#define MAIN_X MARGIN_LEFT
#define MAIN_Y 9
#define MAIN_WIDTH (WIDTH - 2 * MARGIN_LEFT)
#define MAIN_HEIGHT (8 * CY)
#define CLOCK_X MARGIN_LEFT
#define CLOCK_Y 113
#define STATUS_X 95
#define STATUS_Y CLOCK_Y
#define STATUS_WIDTH (10 * CX)
#define STATUS_HEIGHT CY
// Peripherals // Peripherals
Adafruit_SSD1306 display(SCREEN_WIDTH, SCREEN_HEIGHT, &Wire, OLED_RESET); Adafruit_ST7735 display(TFT_CS, TFT_DC, TFT_RESET);
// App state // App state
enum { typedef enum {
CONNECTING, CONNECTING,
SHOWING_DEPARTURES SHOWING_DEPARTURES
} state = CONNECTING; } State;
State state = CONNECTING;
State previous_state = state;
int frame = 0; int frame = 0;
// Use this variable to trigger state enter effects
int state_changed = 1;
int buttonPushed = false; int buttonPushed = false;
JsonDocument departures; JsonDocument departures;
@ -39,27 +70,38 @@ void ICACHE_RAM_ATTR onButtonFalling() {
buttonPushed = true; buttonPushed = true;
} }
uint16_t rgb(uint16_t r, uint16_t g, uint16_t b) {
// Color layout: RRRRRGGGGGGBBBBB (5, 6, 5)
return ((b >> 3) & 0b11111) << 11 | ((g >> 2) & 0b111111) << 5 | ((r >> 3) & 0b11111);
}
void drawLayout() {
display.fillScreen(COLOR_BG);
display.drawFastHLine(5, 4, 150, COLOR_DIVIDER);
display.drawFastHLine(5, 109, 150, COLOR_DIVIDER);
display.drawFastHLine(5, 123, 150, COLOR_DIVIDER);
display.setCursor(CLOCK_X, CLOCK_Y);
display.print("13:12");
}
void setup() { void setup() {
Serial.begin(9600); Serial.begin(9600);
// SSD1306_SWITCHCAPVCC = generate display voltage from 3.3V internally Serial.println("serial init");
if(!display.begin(SSD1306_SWITCHCAPVCC, SCREEN_ADDRESS)) { display.initR();
Serial.println("SSD1306 allocation failed"); display.setSPISpeed(40000000);
for(;;); // Don't proceed, loop forever display.setRotation(3);
} Serial.println("display initialized");
Serial.println("SSD1306 initialized"); Serial.printf("display dimensions: %" PRId16 "x%" PRId16 "\n", display.width(), display.height());
display.clearDisplay();
display.setTextSize(1);
display.setTextColor(COLOR_TEXT);
drawLayout();
WiFi.begin(WIFI_SSID, WIFI_PASSWORD); WiFi.begin(WIFI_SSID, WIFI_PASSWORD);
pinMode(BUTTON_PIN, INPUT_PULLUP); pinMode(BUTTON_PIN, INPUT_PULLUP);
attachInterrupt(BUTTON_PIN, onButtonFalling, FALLING); attachInterrupt(BUTTON_PIN, onButtonFalling, FALLING);
display.setTextColor(SSD1306_INVERSE);
display.setTextSize(1);
display.setCursor(0, 0);
display.printf("Connecting to\n%s\n", WIFI_SSID);
display.display();
/* /*
display.print(" 5 Mannheim sofort\n" display.print(" 5 Mannheim sofort\n"
"26 Kirchheim 9 min\n" "26 Kirchheim 9 min\n"
@ -67,11 +109,6 @@ void setup() {
*/ */
} }
void drawSpinner(int frame) {
display.fillRect(120, 56, spinny_dims[0], spinny_dims[1], SSD1306_BLACK);
display.drawBitmap(120, 56, spinny[frame % spinny_dims[2]], spinny_dims[0], spinny_dims[1], SSD1306_WHITE);
}
// TODO: Error handling // TODO: Error handling
String fetchDepartures() { String fetchDepartures() {
std::unique_ptr<BearSSL::WiFiClientSecure> https(new BearSSL::WiFiClientSecure); std::unique_ptr<BearSSL::WiFiClientSecure> https(new BearSSL::WiFiClientSecure);
@ -79,10 +116,9 @@ String fetchDepartures() {
HTTPClient client; HTTPClient client;
if (!client.begin(*https, "vrnp.beany.club", 443, "/departures?stop_id=de:08221:1225&platform=A")) { if (!client.begin(*https, "vrnp.beany.club", 443, "/departures?stop_id=de:08221:1225&platform=A")) {
display.clearDisplay(); display.fillScreen(COLOR_BG);
display.setCursor(0, 0); display.setCursor(0, 0);
display.print("begin failed"); display.print("begin failed");
display.display();
for (;;); for (;;);
} }
@ -90,10 +126,9 @@ String fetchDepartures() {
client.addHeader("Accept", "application/json"); client.addHeader("Accept", "application/json");
int statusCode = client.GET(); int statusCode = client.GET();
if (statusCode != 200) { if (statusCode != 200) {
display.clearDisplay(); display.fillScreen(COLOR_BG);
display.setCursor(0, 0); display.setCursor(0, 0);
display.printf("http non-ok: %d\n", statusCode); display.printf("http non-ok: %d\n", statusCode);
display.display();
for (;;); for (;;);
} }
@ -101,20 +136,29 @@ String fetchDepartures() {
} }
void loop() { void loop() {
if (state != previous_state) {
state_changed = 1;
previous_state = state;
}
logic_loop();
frame++;
state_changed = 0;
}
void logic_loop() {
switch (state) { switch (state) {
case CONNECTING: { case CONNECTING: {
if (WiFi.status() != WL_CONNECTED) { if (WiFi.status() != WL_CONNECTED) {
drawSpinner(frame++); display.fillRect(STATUS_X, STATUS_Y, STATUS_WIDTH, STATUS_HEIGHT, COLOR_BG);
display.display(); display.setCursor(STATUS_X, STATUS_Y);
display.printf(" connect %c", "|/-\\"[frame % 4]);
delay(FRAME_DELAY); delay(FRAME_DELAY);
return; return;
} }
display.clearDisplay(); display.fillRect(STATUS_X, STATUS_Y, STATUS_WIDTH, STATUS_HEIGHT, COLOR_BG);
display.fillRect(0, 56, 128, 8, SSD1306_INVERSE); display.setCursor(STATUS_X, STATUS_Y);
display.setCursor(0, 56); display.print(" fetching");
display.print("requesting departures");
display.display();
String departuresRaw = fetchDepartures(); String departuresRaw = fetchDepartures();
deserializeJson(departures, departuresRaw); deserializeJson(departures, departuresRaw);
@ -130,48 +174,24 @@ void loop() {
return; return;
} }
/* display.fillRect(STATUS_X, STATUS_Y, STATUS_WIDTH, STATUS_HEIGHT, COLOR_BG);
if (buttonPushed) { display.fillRect(MAIN_X, MAIN_Y, MAIN_WIDTH, MAIN_HEIGHT, COLOR_BG);
buttonPushed = false;
delay(FRAME_DELAY);
return;
}
*/
// Marquee effect for direction strings longer than 11 characters int line = 0;
// Neither particularly efficient nor legible
// (This could use some love)
int availableWidth = 11;
display.clearDisplay();
display.setCursor(0, 0);
for (JsonVariant departure : departures["departures"].as<JsonArray>()) { for (JsonVariant departure : departures["departures"].as<JsonArray>()) {
const char *directionStr = departure["direction"].as<const char *>(); const char *directionStr = departure["direction"].as<const char *>();
char buf[128] = {0}; display.setCursor(MAIN_X, MAIN_Y + (CY + 3) * line);
const char *dirStart; display.printf("%2s %-15.15s %6s",
if (strlen(directionStr) <= availableWidth || (strlen(directionStr) + 3 + availableWidth) > 127) {
dirStart = directionStr;
} else {
memcpy(buf, directionStr, strlen(directionStr));
memcpy(buf + strlen(directionStr), " ", 3);
memcpy(buf + strlen(directionStr) + 3, directionStr, availableWidth);
buf[strlen(directionStr) + 3 + availableWidth] = '\0';
dirStart = buf + frame % (strlen(directionStr) + 3);
}
display.printf("%2s %-11.11s %6s\n",
departure["symbol"].as<const char*>(), departure["symbol"].as<const char*>(),
dirStart, directionStr,
departure["leaving"].as<const char*>() departure["leaving"].as<const char*>()
); );
line++;
} }
display.display();
frame++; delay(25000);
delay(500); state = CONNECTING;
return; return;
} }
} }