immich-frame/src/display_manager.cpp

#include "display_manager.h"
#include "power_manager.h"
#include <M5Unified.h>

// Map 6-color palette indices to RGB565 for M5GFX
static const uint16_t PALETTE_565[6] = {
    0x0000,  // Black
    0xFFFF,  // White
    0xF800,  // Red (approximate)
    0x07E0,  // Green (approximate)
    0x001F,  // Blue (approximate)
    0xFFE0   // Yellow (approximate)
};

void DisplayManager::begin(PowerManager& power) {
    _power = &power;
    _display = &M5.Display;

    // Ensure no automatic e-ink refreshes happen
    _display->setAutoDisplay(false);

    // Set rotation for landscape (device physically rotated)
    _display->setRotation(3);

    Serial.printf("[display] Initialized: %dx%d, rotation=%d\n",
                  _display->width(), _display->height(), _display->getRotation());
}

void DisplayManager::showSetupScreen() {
    _power->enableEPDPower();

    // Batch all drawing — nothing hits the panel until display() at the end
    _display->startWrite();
    _display->fillScreen(TFT_WHITE);

    // Generate QR code for WiFi auto-connect
    const char* qrData = "WIFI:T:nopass;S:PaperColor-Setup;;";
    QRCode qrcode;
    uint8_t qrcodeData[qrcode_getBufferSize(6)];
    qrcode_initText(&qrcode, qrcodeData, 6, ECC_LOW, qrData);

    // Draw QR code centered vertically, left-of-center horizontally
    int qrSize = qrcode.size;
    int scale = 4;
    int qrPixels = qrSize * scale;
    int qrX = (DISPLAY_WIDTH / 3) - (qrPixels / 2);
    int qrY = (DISPLAY_HEIGHT - qrPixels) / 2;

    for (int y = 0; y < qrSize; y++) {
        for (int x = 0; x < qrSize; x++) {
            uint16_t color = qrcode_getModule(&qrcode, x, y) ? TFT_BLACK : TFT_WHITE;
            _display->fillRect(qrX + x * scale, qrY + y * scale, scale, scale, color);
        }
    }

    // Text instructions to the right of QR code
    int textX = DISPLAY_WIDTH / 2 + 40;
    _display->setTextColor(TFT_BLACK);
    _display->setTextDatum(middle_left);
    _display->setTextSize(2);
    _display->drawString("PaperColor", textX, DISPLAY_HEIGHT / 2 - 60);
    _display->setTextSize(1);
    _display->drawString("Scan QR to connect", textX, DISPLAY_HEIGHT / 2 - 20);
    _display->drawString("or join WiFi:", textX, DISPLAY_HEIGHT / 2 + 10);
    _display->setTextSize(1.5);
    _display->drawString("PaperColor-Setup", textX, DISPLAY_HEIGHT / 2 + 40);
    _display->setTextSize(1);
    _display->drawString("Then open:", textX, DISPLAY_HEIGHT / 2 + 70);
    _display->drawString("192.168.4.1", textX, DISPLAY_HEIGHT / 2 + 95);

    _display->endWrite();

    // Single e-ink refresh — the only panel update during boot
    triggerRefresh();
    _power->disableEPDPower();
}

void DisplayManager::showImage(const ProcessedImage& img) {
    if (!img.valid || img.framebuffer == nullptr) {
        Serial.println("[display] Invalid image — skipping");
        return;
    }

    _power->enableEPDPower();

    Serial.println("[display] Writing framebuffer to e-ink...");
    unsigned long start = millis();

    // Write pixel by pixel using palette-mapped colors
    _display->startWrite();
    for (uint16_t y = 0; y < img.height; y++) {
        for (uint16_t x = 0; x < img.width; x++) {
            uint8_t colorIdx = img.framebuffer[y * img.width + x];
            _display->writePixel(x, y, PALETTE_565[colorIdx]);
        }
    }
    _display->endWrite();

    triggerRefresh();

    unsigned long elapsed = millis() - start;
    Serial.printf("[display] Refresh complete in %lu ms\n", elapsed);

    _power->disableEPDPower();
}

void DisplayManager::showMessage(const char* title, const char* body) {
    _power->enableEPDPower();

    _display->startWrite();
    _display->fillScreen(TFT_WHITE);
    _display->setTextColor(TFT_BLACK);
    _display->setTextDatum(middle_center);
    _display->setTextSize(2);
    _display->drawString(title, DISPLAY_WIDTH / 2, DISPLAY_HEIGHT / 2 - 30);
    _display->setTextSize(1);
    _display->drawString(body, DISPLAY_WIDTH / 2, DISPLAY_HEIGHT / 2 + 20);
    _display->endWrite();

    triggerRefresh();
    _power->disableEPDPower();
}

void DisplayManager::showMetadata(const AssetInfo& info, uint8_t metaFlags, MetaPosition pos) {
    if (metaFlags == 0) return;  // No metadata to show

    // Build metadata string
    String metaText = "";

    if ((metaFlags & META_DATE) && info.dateTime.length() > 0) {
        // Extract just the date portion (YYYY-MM-DD)
        metaText += info.dateTime.substring(0, 10);
    }
    if ((metaFlags & META_LOCATION) && info.city.length() > 0) {
        if (metaText.length() > 0) metaText += "  |  ";
        metaText += info.city;
    }
    if ((metaFlags & META_PEOPLE) && !info.people.empty()) {
        if (metaText.length() > 0) metaText += "  |  ";
        for (size_t i = 0; i < info.people.size(); i++) {
            if (i > 0) metaText += ", ";
            metaText += info.people[i];
        }
    }
    if ((metaFlags & META_ALBUM)) {
        // Album name would need to be passed separately — skip for now
    }
    if ((metaFlags & META_CAMERA) && info.camera.length() > 0) {
        if (metaText.length() > 0) metaText += "  |  ";
        metaText += info.camera;
    }

    if (metaText.length() == 0) return;

    // Draw semi-transparent bar with text
    uint16_t barY = (pos == MetaPosition::Top) ? 0 : (DISPLAY_HEIGHT - 30);
    _display->fillRect(0, barY, DISPLAY_WIDTH, 30, TFT_BLACK);
    _display->setTextColor(TFT_WHITE);
    _display->setTextDatum(middle_center);
    _display->setTextSize(1);
    _display->drawString(metaText.c_str(), DISPLAY_WIDTH / 2, barY + 15);
}

void DisplayManager::triggerRefresh() {
    // M5GFX handles e-ink refresh internally when using the EPD panel driver
    // The display() call triggers the actual e-ink refresh cycle (10-20s)
    _display->display();
}

uint16_t DisplayManager::paletteToColor565(uint8_t index) {
    if (index >= DISPLAY_COLORS) return 0;
    return PALETTE_565[index];
}