Momentum-Apps/scenes/pof_scene_main.c

#include "../portal_of_flipper_i.h"
#include "../pof_token.h"

enum SubmenuIndex {
    // SubmenuIndexFigure* need to match POF_TOKEN_LIMIT
    SubmenuIndexFigure1,
    SubmenuIndexFigure2,
    SubmenuIndexFigure3,
    SubmenuIndexFigure4,
    SubmenuIndexFigure5,
    SubmenuIndexFigure6,
    SubmenuIndexFigure7,
    SubmenuIndexLoad,
};

void pof_scene_main_submenu_callback(void* context, uint32_t index) {
    PoFApp* pof = context;
    view_dispatcher_send_custom_event(pof->view_dispatcher, index);
}

void pof_scene_main_on_update(void* context) {
    PoFApp* pof = context;
    VirtualPortal* virtual_portal = pof->virtual_portal;

    Submenu* submenu = pof->submenu;
    submenu_reset(pof->submenu);

    int count = 0;
    for(int i = 0; i < POF_TOKEN_LIMIT; i++) {
        if(virtual_portal->tokens[i]->loaded) {
            PoFToken* pof_token = virtual_portal->tokens[i];
            // Unload figure
            submenu_add_item(
                submenu,
                pof_token->dev_name,
                SubmenuIndexFigure1 + i,
                pof_scene_main_submenu_callback,
                pof);
            count++;
        }
    }

    if(count < POF_TOKEN_LIMIT) {
        submenu_add_item(
            submenu, "<Load figure>", SubmenuIndexLoad, pof_scene_main_submenu_callback, pof);
    }

    submenu_set_selected_item(
        submenu, scene_manager_get_scene_state(pof->scene_manager, PoFSceneMain));
    view_dispatcher_switch_to_view(pof->view_dispatcher, PoFViewSubmenu);
}

void pof_scene_main_on_enter(void* context) {
    pof_scene_main_on_update(context);
}

bool pof_scene_main_on_event(void* context, SceneManagerEvent event) {
    PoFApp* pof = context;
    VirtualPortal* virtual_portal = pof->virtual_portal;
    bool consumed = false;

    if(event.type == SceneManagerEventTypeCustom) {
        if(event.event == SubmenuIndexLoad) {
            // Explicitly save state so that the correct item is
            // reselected if the user cancels loading a file.
            scene_manager_set_scene_state(pof->scene_manager, PoFSceneMain, SubmenuIndexLoad);
            scene_manager_next_scene(pof->scene_manager, PoFSceneFileSelect);
            consumed = true;
        } else {
            pof_token_clear(virtual_portal->tokens[event.event], true);
            pof_scene_main_on_update(context);
        }
    } else if(event.type == SceneManagerEventTypeBack) {
        scene_manager_stop(pof->scene_manager);
        view_dispatcher_stop(pof->view_dispatcher);
        consumed = true;
    }

    return consumed;
}

void pof_scene_main_on_exit(void* context) {
    PoFApp* pof = context;
    submenu_reset(pof->submenu);
}