Momentum-Apps/nightstand_clock/clock_app.c

#include <furi.h>
#include <furi_hal.h>

#include <gui/gui.h>
#include <gui/elements.h>

#include <notification/notification.h>
#include <notification/notification_messages.h>
#include <notification/notification_app.h>
#include <momentum/settings.h>

#include "clock_app.h"

/*
    This is a modified version of the default clock app intended for use overnight
    Up / Down controls the displays brightness. Down at brightness 0 turns the notification LED on and off.
*/

int brightness = 5;
bool led = false;
NotificationApp* notif = 0;

int dspBrightnessBarFrames = 0;
const int dspBrightnessBarDisplayFrames = 8;

const NotificationMessage message_red_dim = {
    .type = NotificationMessageTypeLedRed,
    .data.led.value = 0xFF / 16,
};

const NotificationMessage message_red_off = {
    .type = NotificationMessageTypeLedRed,
    .data.led.value = 0x00,
};

static const NotificationSequence led_on = {
    &message_red_dim,
    &message_do_not_reset,
    NULL,
};

static const NotificationSequence led_off = {
    &message_red_off,
    &message_do_not_reset,
    NULL,
};

static const NotificationSequence led_reset = {
    &message_red_0,
    NULL,
};

void set_backlight_brightness(float brightness) {
    notif->settings.display_brightness = brightness;
    notification_message(notif, &sequence_display_backlight_on);
}

void handle_up() {
    dspBrightnessBarFrames = dspBrightnessBarDisplayFrames;
    if(brightness < 100) {
        led = false;
        notification_message(notif, &led_off);
        brightness += 5;
    }
    set_backlight_brightness((float)(brightness / 100.f));
}

void handle_down() {
    dspBrightnessBarFrames = dspBrightnessBarDisplayFrames;
    if(brightness > 0) {
        brightness -= 5;
        if(brightness == 0) { //trigger only on the first brightness 5 -> 0 transition
            led = true;
            notification_message(notif, &led_on);
        }
    } else if(brightness == 0) { //trigger on every down press afterwards
        led = !led;
        if(led) {
            notification_message(notif, &led_on);
        } else {
            notification_message(notif, &led_off);
        }
    }
    set_backlight_brightness((float)(brightness / 100.f));
}

static void clock_input_callback(InputEvent* input_event, void* ctx) {
    furi_assert(ctx);
    FuriMessageQueue* event_queue = ctx;

    PluginEvent event = {.type = EventTypeKey, .input = *input_event};
    furi_message_queue_put(event_queue, &event, FuriWaitForever);
}

//do you are have stupid?
// idk who left that comment here, but i don't have stupid sorry
void elements_progress_bar_vertical(
    Canvas* canvas,
    uint8_t x,
    uint8_t y,
    uint8_t height,
    float progress) {
    furi_assert(canvas);
    furi_assert(((float)progress >= 0.0f) && ((float)progress <= 1.0f));

    uint8_t width = 9;

    uint8_t progress_length = roundf((1.f - progress) * (height - 2));

    canvas_set_color(canvas, ColorBlack);
    canvas_draw_box(canvas, x + 1, y + 1, width - 2, height - 2);

    canvas_set_color(canvas, ColorWhite);
    canvas_draw_box(canvas, x + 1, y + 1, width - 2, progress_length);

    canvas_set_color(canvas, ColorBlack);
    canvas_draw_rframe(canvas, x, y, width, height, 3);
}

static void clock_render_callback(Canvas* const canvas, void* ctx) {
    //canvas_clear(canvas);
    //canvas_set_color(canvas, ColorBlack);

    //avoids a bug with the brightness being reverted after the backlight-off period
    set_backlight_brightness((float)(brightness / 100.f));

    if(dspBrightnessBarFrames > 0) {
        elements_progress_bar_vertical(canvas, 119, 1, 62, (float)(brightness / 100.f));
        dspBrightnessBarFrames--;
    }

    ClockState* state = ctx;
    if(furi_mutex_acquire(state->mutex, 200) != FuriStatusOk) {
        //FURI_LOG_D(TAG, "Can't obtain mutex, requeue render");
        PluginEvent event = {.type = EventTypeTick};
        furi_message_queue_put(state->event_queue, &event, 0);
        return;
    }

    DateTime curr_dt;
    furi_hal_rtc_get_datetime(&curr_dt);
    uint32_t curr_ts = datetime_datetime_to_timestamp(&curr_dt);

    char time_string[TIME_LEN];
    char date_string[DATE_LEN];
    char meridian_string[MERIDIAN_LEN];
    char date_pct_string[DATE_PCT_LEN];
    char timer_string[20];

    if(state->time_format == LocaleTimeFormat24h) {
        snprintf(
            time_string, TIME_LEN, CLOCK_TIME_FORMAT, curr_dt.hour, curr_dt.minute, curr_dt.second);
    } else {
        bool pm12 = curr_dt.hour >= 12;
        uint8_t hour = curr_dt.hour;

        LocaleTimeFormat time_format = locale_get_time_format();
        if(time_format == LocaleTimeFormat12h) {
            if(hour > 12) {
                hour -= 12;
            }
            if(hour == 0) {
                hour = momentum_settings.midnight_format_00 ? 0 : 12;
            }
        }

        snprintf(time_string, TIME_LEN, CLOCK_TIME_FORMAT, hour, curr_dt.minute, curr_dt.second);

        snprintf(
            meridian_string,
            MERIDIAN_LEN,
            MERIDIAN_FORMAT,
            pm12 ? MERIDIAN_STRING_PM : MERIDIAN_STRING_AM);
    }

    if(state->date_format == LocaleDateFormatYMD) {
        snprintf(
            date_string, DATE_LEN, CLOCK_ISO_DATE_FORMAT, curr_dt.year, curr_dt.month, curr_dt.day);
    } else if(state->date_format == LocaleDateFormatMDY) {
        snprintf(
            date_string, DATE_LEN, CLOCK_RFC_DATE_FORMAT, curr_dt.month, curr_dt.day, curr_dt.year);
    } else {
        snprintf(
            date_string, DATE_LEN, CLOCK_RFC_DATE_FORMAT, curr_dt.day, curr_dt.month, curr_dt.year);
    }

    bool timer_running = state->timer_running;
    uint32_t timer_start_timestamp = state->timer_start_timestamp;
    uint32_t timer_stopped_seconds = state->timer_stopped_seconds;

    furi_mutex_release(state->mutex);

    canvas_set_font(canvas, FontBigNumbers);

    if(timer_start_timestamp != 0) {
        int32_t elapsed_secs = timer_running ? (curr_ts - timer_start_timestamp) :
                                               timer_stopped_seconds;
        snprintf(timer_string, 20, "%.2ld:%.2ld", elapsed_secs / 60, elapsed_secs % 60);
        if(state->time_format == LocaleTimeFormat12h) {
            canvas_draw_str_aligned(
                canvas, 56, 8, AlignCenter, AlignCenter, time_string); // DRAW TIME
        } else {
            canvas_draw_str_aligned(
                canvas, 64, 8, AlignCenter, AlignCenter, time_string); // DRAW TIME
        }
        canvas_draw_str_aligned(canvas, 64, 32, AlignCenter, AlignTop, timer_string); // DRAW TIMER
        canvas_set_font(canvas, FontSecondary);
        if(state->time_format == LocaleTimeFormat12h) {
            canvas_draw_str_aligned(canvas, 112, 8, AlignCenter, AlignCenter, meridian_string);
        }

        snprintf(
            date_pct_string, sizeof(date_pct_string), "%s   %u%%", date_string, state->battery_pct);
        canvas_draw_str_aligned(
            canvas, 64, 20, AlignCenter, AlignTop, date_pct_string); // DRAW DATE + BATTERY
        elements_button_left(canvas, "Reset");
    } else {
        canvas_draw_str_aligned(canvas, 64, 32, AlignCenter, AlignCenter, time_string);
        canvas_set_font(canvas, FontSecondary);

        if(state->time_format == LocaleTimeFormat12h) {
            snprintf(
                date_pct_string,
                sizeof(date_pct_string),
                "%s   %u%%",
                date_string,
                state->battery_pct);
            canvas_draw_str_aligned(canvas, 64, 17, AlignCenter, AlignCenter, date_pct_string);
            canvas_draw_str_aligned(canvas, 64, 48, AlignCenter, AlignCenter, meridian_string);
        } else {
            canvas_draw_str_aligned(canvas, 64, 17, AlignCenter, AlignCenter, date_string);
            snprintf(date_pct_string, sizeof(date_pct_string), "%u%%", state->battery_pct);
            canvas_draw_str_aligned(canvas, 64, 48, AlignCenter, AlignCenter, date_pct_string);
        }
    }
    if(timer_running) {
        elements_button_center(canvas, "Stop");
    } else if(timer_start_timestamp != 0 && !timer_running) {
        elements_button_center(canvas, "Start");
    }
}

static void clock_state_init(ClockState* const state) {
    state->time_format = locale_get_time_format();
    state->date_format = locale_get_date_format();
    state->battery_pct = furi_hal_power_get_pct();

    //FURI_LOG_D(TAG, "Time format: %s", state->settings.time_format == H12 ? "12h" : "24h");
    //FURI_LOG_D(TAG, "Date format: %s", state->settings.date_format == Iso ? "ISO 8601" : "RFC 5322");
    //furi_hal_rtc_get_datetime(&state->datetime);
}

// Runs every 1000ms by default
static void clock_tick(void* ctx) {
    furi_assert(ctx);
    FuriMessageQueue* event_queue = ctx;
    PluginEvent event = {.type = EventTypeTick};
    // It's OK to loose this event if system overloaded
    furi_message_queue_put(event_queue, &event, 0);
}

void timer_start_stop(ClockState* plugin_state) {
    // START/STOP TIMER
    uint32_t curr_ts = furi_hal_rtc_get_timestamp();

    if(plugin_state->timer_running) {
        // Update stopped seconds
        plugin_state->timer_stopped_seconds = curr_ts - plugin_state->timer_start_timestamp;
    } else {
        if(plugin_state->timer_start_timestamp == 0) {
            // Set starting timestamp if this is first time
            plugin_state->timer_start_timestamp = curr_ts;
        } else {
            // Timer was already running, need to slightly readjust so we don't
            // count the intervening time
            plugin_state->timer_start_timestamp = curr_ts - plugin_state->timer_stopped_seconds;
        }
    }
    plugin_state->timer_running = !plugin_state->timer_running;
}

void timer_reset_seconds(ClockState* plugin_state) {
    if(plugin_state->timer_start_timestamp != 0) {
        // Reset seconds
        plugin_state->timer_running = false;
        plugin_state->timer_start_timestamp = 0;
        plugin_state->timer_stopped_seconds = 0;
    }
}

int32_t clock_app(void* p) {
    UNUSED(p);
    ClockState* plugin_state = malloc(sizeof(ClockState));

    plugin_state->event_queue = furi_message_queue_alloc(8, sizeof(PluginEvent));
    if(plugin_state->event_queue == NULL) {
        FURI_LOG_E(TAG, "Cannot create event queue");
        free(plugin_state);
        return 255;
    }
    //FURI_LOG_D(TAG, "Event queue created");

    plugin_state->mutex = furi_mutex_alloc(FuriMutexTypeNormal);
    if(plugin_state->mutex == NULL) {
        FURI_LOG_E(TAG, "Cannot create mutex");
        furi_message_queue_free(plugin_state->event_queue);
        free(plugin_state);
        return 255;
    }
    //FURI_LOG_D(TAG, "Mutex created");

    clock_state_init(plugin_state);

    notif = furi_record_open(RECORD_NOTIFICATION);
    float tmpBrightness = notif->settings.display_brightness;
    brightness = tmpBrightness * 100; // Keep current brightness by default

    notification_message(notif, &sequence_display_backlight_enforce_on);
    notification_message(notif, &led_off);

    // Set system callbacks
    ViewPort* view_port = view_port_alloc();
    view_port_draw_callback_set(view_port, clock_render_callback, plugin_state);
    view_port_input_callback_set(view_port, clock_input_callback, plugin_state->event_queue);

    FuriTimer* timer =
        furi_timer_alloc(clock_tick, FuriTimerTypePeriodic, plugin_state->event_queue);

    if(timer == NULL) {
        FURI_LOG_E(TAG, "Cannot create timer");
        furi_mutex_free(plugin_state->mutex);
        furi_message_queue_free(plugin_state->event_queue);
        free(plugin_state);
        return 255;
    }
    //FURI_LOG_D(TAG, "Timer created");

    // Open GUI and register view_port
    Gui* gui = furi_record_open(RECORD_GUI);
    gui_add_view_port(gui, view_port, GuiLayerFullscreen);

    furi_timer_start(timer, furi_kernel_get_tick_frequency());
    //FURI_LOG_D(TAG, "Timer started");

    // Main loop
    PluginEvent event;
    for(bool processing = true; processing;) {
        FuriStatus event_status = furi_message_queue_get(plugin_state->event_queue, &event, 100);

        if(event_status != FuriStatusOk) continue;

        if(furi_mutex_acquire(plugin_state->mutex, FuriWaitForever) != FuriStatusOk) continue;
        // press events
        if(event.type == EventTypeKey) {
            if(event.input.type == InputTypeShort) {
                switch(event.input.key) {
                case InputKeyLeft:
                    // Reset seconds
                    timer_reset_seconds(plugin_state);
                    break;
                case InputKeyOk:
                    // Toggle timer
                    timer_start_stop(plugin_state);
                    break;
                case InputKeyBack:
                    // Exit the plugin
                    processing = false;
                    break;
                case InputKeyUp:
                    handle_up();
                    break;
                case InputKeyDown:
                    handle_down();
                    break;
                default:
                    break;
                }
            }
        } else if(event.type == EventTypeTick) {
            plugin_state->battery_pct = furi_hal_power_get_pct();
        }

        furi_mutex_release(plugin_state->mutex);
        view_port_update(view_port);
    }

    furi_timer_free(timer);
    view_port_enabled_set(view_port, false);
    gui_remove_view_port(gui, view_port);
    furi_record_close(RECORD_GUI);
    furi_record_close(RECORD_NOTIFICATION);
    view_port_free(view_port);
    furi_message_queue_free(plugin_state->event_queue);
    furi_mutex_free(plugin_state->mutex);
    free(plugin_state);

    set_backlight_brightness(tmpBrightness);
    notification_message(notif, &sequence_display_backlight_enforce_auto);
    notification_message(notif, &led_reset);

    return 0;
}