/*
 * Copyright (C) 2015 Jared Boone, ShareBrained Technology, Inc.
 * Copyright (C) 2016 Furrtek
 *
 * This file is part of PortaPack.
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2, or (at your option)
 * any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; see the file COPYING.  If not, write to
 * the Free Software Foundation, Inc., 51 Franklin Street,
 * Boston, MA 02110-1301, USA.
 */

#include "ui_navigation.hpp"

// #include "modules.h"

#include "portapack.hpp"
#include "event_m0.hpp"
#include "bmp_splash.hpp"
#include "bmp_modal_warning.hpp"
#include "portapack_persistent_memory.hpp"
#include "portapack_shared_memory.hpp"

#include "ui_about_simple.hpp"
#include "ui_adsb_rx.hpp"
#include "ui_adsb_tx.hpp"
#include "ui_afsk_rx.hpp"
#include "ui_aprs_rx.hpp"
#include "ui_btle_rx.hpp"
#include "ui_nrf_rx.hpp"
#include "ui_aprs_tx.hpp"
#include "ui_bht_tx.hpp"
#include "ui_coasterp.hpp"
#include "ui_debug.hpp"
#include "ui_encoders.hpp"
#include "ui_fileman.hpp"
#include "ui_freqman.hpp"
#include "ui_jammer.hpp"
// #include "ui_keyfob.hpp"
#include "ui_lcr.hpp"
#include "ui_mictx.hpp"
#include "ui_morse.hpp"
// #include "ui_numbers.hpp"
// #include "ui_nuoptix.hpp"
// #include "ui_playdead.hpp"
#include "ui_pocsag_tx.hpp"
#include "ui_rds.hpp"
#include "ui_remote.hpp"
#include "ui_scanner.hpp"
#include "ui_search.hpp"
#include "ui_recon.hpp"
#include "ui_level.hpp"
#include "ui_sd_wipe.hpp"
#include "ui_settings.hpp"
#include "ui_siggen.hpp"
#include "ui_sonde.hpp"
#include "ui_sstvtx.hpp"
// #include "ui_test.hpp"
#include "ui_text_editor.hpp"
#include "ui_tone_search.hpp"
#include "ui_touchtunes.hpp"
#include "ui_playlist.hpp"
#include "ui_view_wav.hpp"
#include "ui_whipcalc.hpp"
#include "ui_flash_utility.hpp"
#include "ui_sd_over_usb.hpp"
#include "ui_spectrum_painter.hpp"

// #include "acars_app.hpp"
#include "ais_app.hpp"
#include "analog_audio_app.hpp"
#include "analog_tv_app.hpp"
#include "capture_app.hpp"
#include "ert_app.hpp"
#include "lge_app.hpp"
#include "pocsag_app.hpp"
#include "replay_app.hpp"
#include "gps_sim_app.hpp"
#include "soundboard_app.hpp"
#include "tpms_app.hpp"

#include "core_control.hpp"
#include "ui_looking_glass_app.hpp"
#include "file.hpp"
#include "png_writer.hpp"

using portapack::receiver_model;
using portapack::transmitter_model;

namespace ui {

/* SystemStatusView ******************************************************/

SystemStatusView::SystemStatusView(
    NavigationView& nav)
    : nav_(nav) {
    static constexpr Style style_systemstatus{
        .font = font::fixed_8x16,
        .background = Color::dark_grey(),
        .foreground = Color::white(),
    };

    add_children({
        &backdrop,
        &button_back,
        &title,
        &button_title,
        &button_converter,
        &button_speaker,
        &button_stealth,
        //&button_textentry,
        &button_camera,
        &button_sleep,
        &button_bias_tee,
        &button_clock_status,
        &sd_card_status_view,
    });

    if (portapack::persistent_memory::config_speaker())
        button_speaker.hidden(false);
    else
        button_speaker.hidden(true);

    if (portapack::persistent_memory::config_hide_converter()) {
        button_converter.hidden(true);
    } else {
        button_converter.hidden(false);
        if (portapack::persistent_memory::config_converter()) {
            button_converter.set_foreground(Color::red());
        } else {
            button_converter.set_foreground(Color::light_grey());
        }
    }

    button_back.id = -1;  // Special ID used by FocusManager
    title.set_style(&style_systemstatus);

    if (portapack::persistent_memory::stealth_mode())
        button_stealth.set_foreground(ui::Color::green());

    /*if (!portapack::persistent_memory::ui_config_textentry())
                button_textentry.set_bitmap(&bitmap_icon_keyboard);
        else
                button_textentry.set_bitmap(&bitmap_icon_unistroke);*/

    refresh();

    button_back.on_select = [this](ImageButton&) {
        if (this->on_back)
            this->on_back();
    };

    button_title.on_select = [this](ImageButton&) {
        this->on_title();
    };

    button_converter.on_select = [this](ImageButton&) {
        this->on_converter();
    };

    button_speaker.on_select = [this](ImageButton&) {
        this->on_speaker();
    };

    button_stealth.on_select = [this](ImageButton&) {
        this->on_stealth();
    };

    button_bias_tee.on_select = [this](ImageButton&) {
        this->on_bias_tee();
    };

    /*button_textentry.on_select = [this](ImageButton&) {
                this->on_textentry();
        };*/

    button_camera.on_select = [this](ImageButton&) {
        this->on_camera();
    };

    button_sleep.on_select = [this](ImageButton&) {
        DisplaySleepMessage message;
        EventDispatcher::send_message(message);
    };

    button_clock_status.on_select = [this](ImageButton&) {
        this->on_clk();
    };
}

void SystemStatusView::refresh() {
    if (portapack::persistent_memory::config_hide_converter()) {
        button_converter.hidden(true);
    } else {
        if (portapack::persistent_memory::config_updown_converter()) {
            button_converter.set_bitmap(&bitmap_icon_downconvert);
        } else {
            button_converter.set_bitmap(&bitmap_icon_upconvert);
        }
        button_converter.hidden(false);
        if (portapack::persistent_memory::config_converter()) {
            button_converter.set_foreground(Color::red());
        } else {
            button_converter.set_foreground(Color::light_grey());
        }
    }
    // Retune to take converter change in account
    receiver_model.set_tuning_frequency(portapack::persistent_memory::tuned_frequency());
    transmitter_model.set_tuning_frequency(portapack::persistent_memory::tuned_frequency());

    if (!portapack::persistent_memory::config_speaker()) {
        button_speaker.set_foreground(Color::light_grey());
        button_speaker.set_bitmap(&bitmap_icon_speaker_mute);
        button_speaker.hidden(false);
    } else {
        button_speaker.hidden(true);
    }

    if (portapack::get_antenna_bias()) {
        button_bias_tee.set_bitmap(&bitmap_icon_biast_on);
        button_bias_tee.set_foreground(ui::Color::yellow());
    } else {
        button_bias_tee.set_bitmap(&bitmap_icon_biast_off);
        button_bias_tee.set_foreground(ui::Color::light_grey());
    }

    if (portapack::clock_manager.get_reference().source == ClockManager::ReferenceSource::External) {
        button_clock_status.set_bitmap(&bitmap_icon_clk_ext);
    } else {
        button_clock_status.set_bitmap(&bitmap_icon_clk_int);
    }

    if (portapack::persistent_memory::clkout_enabled()) {
        button_clock_status.set_foreground(ui::Color::green());
    } else {
        button_clock_status.set_foreground(ui::Color::light_grey());
    }

    set_dirty();
}

void SystemStatusView::set_back_enabled(bool new_value) {
    if (new_value) {
        add_child(&button_back);
    } else {
        remove_child(&button_back);
    }
}

void SystemStatusView::set_title_image_enabled(bool new_value) {
    if (new_value) {
        add_child(&button_title);
    } else {
        remove_child(&button_title);
    }
}

void SystemStatusView::set_title(const std::string new_value) {
    if (new_value.empty()) {
        title.set(default_title);
    } else {
        title.set(new_value);
    }
}

void SystemStatusView::on_converter() {
    if (!portapack::persistent_memory::config_converter()) {
        portapack::persistent_memory::set_config_converter(true);
        button_converter.set_foreground(Color::red());
    } else {
        portapack::persistent_memory::set_config_converter(false);
        button_converter.set_foreground(Color::light_grey());
    }
    receiver_model.set_tuning_frequency(portapack::persistent_memory::tuned_frequency());  // Retune
}

void SystemStatusView::on_speaker() {
    if (!portapack::speaker_mode) {
        portapack::set_speaker_mode(true);
        button_speaker.set_foreground(Color::green());
        button_speaker.set_bitmap(&bitmap_icon_speaker);
    } else {
        portapack::set_speaker_mode(false);
        button_speaker.set_foreground(Color::light_grey());
        button_speaker.set_bitmap(&bitmap_icon_speaker_mute);
    }
}

void SystemStatusView::on_stealth() {
    bool mode = not portapack::persistent_memory::stealth_mode();

    portapack::persistent_memory::set_stealth_mode(mode);

    button_stealth.set_foreground(mode ? Color::green() : Color::light_grey());
}

void SystemStatusView::on_bias_tee() {
    if (!portapack::get_antenna_bias()) {
        nav_.display_modal("Bias voltage", "Enable DC voltage on\nantenna connector?", YESNO, [this](bool v) {
            if (v) {
                portapack::set_antenna_bias(true);
                receiver_model.set_antenna_bias();
                transmitter_model.set_antenna_bias();
                refresh();
            }
        });
    } else {
        portapack::set_antenna_bias(false);
        receiver_model.set_antenna_bias();
        transmitter_model.set_antenna_bias();

        // Ensure this is disabled. The models don't actually
        // update the radio unless they are 'enabled_'.
        radio::set_antenna_bias(false);
        refresh();
    }
}

/*void SystemStatusView::on_textentry() {
        uint8_t cfg;

        cfg = portapack::persistent_memory::ui_config_textentry();
        portapack::persistent_memory::set_config_textentry(cfg ^ 1);

        if (!cfg)
                button_textentry.set_bitmap(&bitmap_icon_unistroke);
        else
                button_textentry.set_bitmap(&bitmap_icon_keyboard);
}*/

void SystemStatusView::on_camera() {
    ensure_directory("SCREENSHOTS");
    auto path = next_filename_matching_pattern(u"SCREENSHOTS/SCR_????.PNG");

    if (path.empty()) {
        return;
    }

    PNGWriter png;
    auto create_error = png.create(path);
    if (create_error.is_valid()) {
        return;
    }

    for (int i = 0; i < 320; i++) {
        std::array<ColorRGB888, 240> row;
        portapack::display.read_pixels({0, i, 240, 1}, row);
        png.write_scanline(row);
    }
}

void SystemStatusView::on_clk() {
    bool v = portapack::persistent_memory::clkout_enabled();
    if (v) {
        v = false;
    } else {
        v = true;
    }
    portapack::clock_manager.enable_clock_output(v);
    portapack::persistent_memory::set_clkout_enabled(v);
    refresh();
}

void SystemStatusView::on_title() {
    if (nav_.is_top())
        nav_.push<AboutView>();
    else
        nav_.pop();
}

/* Information View *****************************************************/

InformationView::InformationView(
    NavigationView& nav)
    : nav_(nav) {
    static constexpr Style style_infobar{
        .font = font::fixed_8x16,
        .background = {33, 33, 33},
        .foreground = Color::white(),
    };

    add_children({&backdrop,
                  &version,
                  &ltime});

    version.set_style(&style_infobar);

    ltime.set_hide_clock(portapack::persistent_memory::hide_clock());
    ltime.set_style(&style_infobar);
    ltime.set_seconds_enabled(true);
    ltime.set_date_enabled(portapack::persistent_memory::clock_with_date());
    set_dirty();
}

void InformationView::refresh() {
    ltime.set_hide_clock(portapack::persistent_memory::hide_clock());
    ltime.set_seconds_enabled(true);
    ltime.set_date_enabled(portapack::persistent_memory::clock_with_date());
}

/* Navigation ************************************************************/

bool NavigationView::is_top() const {
    return view_stack.size() == 1;
}

View* NavigationView::push_view(std::unique_ptr<View> new_view) {
    free_view();

    const auto p = new_view.get();
    view_stack.emplace_back(ViewState{std::move(new_view), {}});

    update_view();

    return p;
}

void NavigationView::pop() {
    pop(true);
}

void NavigationView::pop_modal() {
    // Pop modal view + underlying app view.
    // TODO: this shouldn't be necessary.
    pop(false);
    pop(true);
}

void NavigationView::display_modal(
    const std::string& title,
    const std::string& message) {
    display_modal(title, message, INFO, nullptr);
}

void NavigationView::display_modal(
    const std::string& title,
    const std::string& message,
    const modal_t type,
    const std::function<void(bool)> on_choice) {
    /* If a modal view is already visible, don't display another */
    if (!modal_view) {
        modal_view = push<ModalMessageView>(title, message, type, on_choice);
    }
}

void NavigationView::pop(bool update) {
    if (view() == modal_view) {
        modal_view = nullptr;
    }

    // Can't pop last item from stack.
    if (view_stack.size() > 1) {
        auto on_pop = view_stack.back().on_pop;

        free_view();
        view_stack.pop_back();

        if (update)
            update_view();

        if (on_pop) on_pop();
    }
}

void NavigationView::free_view() {
    remove_child(view());
}

void NavigationView::update_view() {
    const auto new_view = view_stack.back().view.get();

    add_child(new_view);
    new_view->set_parent_rect({{0, 0}, size()});

    focus();
    set_dirty();

    if (on_view_changed) {
        on_view_changed(*new_view);
    }
}

Widget* NavigationView::view() const {
    return children_.empty() ? nullptr : children_[0];
}

void NavigationView::focus() {
    if (view()) {
        view()->focus();
    }
}

bool NavigationView::set_on_pop(std::function<void()> on_pop) {
    if (view_stack.size() <= 1)
        return false;

    auto& top = view_stack.back();
    if (top.on_pop)
        return false;

    top.on_pop = on_pop;
    return true;
}

/* ReceiversMenuView *****************************************************/

ReceiversMenuView::ReceiversMenuView(NavigationView& nav) {
    if (portapack::persistent_memory::show_gui_return_icon()) {
        add_items({{"..", ui::Color::light_grey(), &bitmap_icon_previous, [&nav]() { nav.pop(); }}});
    }
    add_items({
        {
            "ADS-B",
            ui::Color::green(),
            &bitmap_icon_adsb,
            [&nav]() { nav.push<ADSBRxView>(); },
        },
        //{ "ACARS",	ui::Color::yellow(),	&bitmap_icon_adsb,			[&nav](){ nav.push<ACARSAppView>(); }, },
        {"AIS Boats", ui::Color::green(), &bitmap_icon_ais, [&nav]() { nav.push<AISAppView>(); }},
        {"AFSK", ui::Color::yellow(), &bitmap_icon_modem, [&nav]() { nav.push<AFSKRxView>(); }},
        {"BTLE", ui::Color::yellow(), &bitmap_icon_btle, [&nav]() { nav.push<BTLERxView>(); }},
        {"NRF", ui::Color::yellow(), &bitmap_icon_nrf, [&nav]() { nav.push<NRFRxView>(); }},
        {"Audio", ui::Color::green(), &bitmap_icon_speaker, [&nav]() { nav.push<AnalogAudioView>(); }},
        {"Analog TV", ui::Color::yellow(), &bitmap_icon_sstv, [&nav]() { nav.push<AnalogTvView>(); }},
        {"ERT Meter", ui::Color::green(), &bitmap_icon_ert, [&nav]() { nav.push<ERTAppView>(); }},
        {"POCSAG", ui::Color::green(), &bitmap_icon_pocsag, [&nav]() { nav.push<POCSAGAppView>(); }},
        {"Radiosnde", ui::Color::green(), &bitmap_icon_sonde, [&nav]() { nav.push<SondeView>(); }},
        {"TPMS Cars", ui::Color::green(), &bitmap_icon_tpms, [&nav]() { nav.push<TPMSAppView>(); }},
        {"Recon", ui::Color::green(), &bitmap_icon_scanner, [&nav]() { nav.push<ReconView>(); }},
        {"Level", ui::Color::green(), &bitmap_icon_options_radio, [&nav]() { nav.push<LevelView>(); }},
        {"APRS", ui::Color::green(), &bitmap_icon_aprs, [&nav]() { nav.push<APRSRXView>(); }}
        /*
                { "DMR", 		ui::Color::dark_grey(),	&bitmap_icon_dmr,		[&nav](){ nav.push<NotImplementedView>(); } },
                { "SIGFOX", 	ui::Color::dark_grey(),	&bitmap_icon_fox,		[&nav](){ nav.push<NotImplementedView>(); } }, // SIGFRXView
                { "LoRa", 		ui::Color::dark_grey(),	&bitmap_icon_lora,		[&nav](){ nav.push<NotImplementedView>(); } },
                { "SSTV", 		ui::Color::dark_grey(), &bitmap_icon_sstv,		[&nav](){ nav.push<NotImplementedView>(); } },
                { "TETRA", 		ui::Color::dark_grey(),	&bitmap_icon_tetra,		[&nav](){ nav.push<NotImplementedView>(); } },*/
    });

    // set_highlighted(0);		// Default selection is "Audio"
}

/* TransmittersMenuView **************************************************/

TransmittersMenuView::TransmittersMenuView(NavigationView& nav) {
    if (portapack::persistent_memory::show_gui_return_icon()) {
        add_items({{"..", ui::Color::light_grey(), &bitmap_icon_previous, [&nav]() { nav.pop(); }}});
    }
    add_items({
        {"ADS-B [S]", ui::Color::yellow(), &bitmap_icon_adsb, [&nav]() { nav.push<ADSBTxView>(); }},
        {"APRS", ui::Color::green(), &bitmap_icon_aprs, [&nav]() { nav.push<APRSTXView>(); }},
        {"BHT Xy/EP", ui::Color::green(), &bitmap_icon_bht, [&nav]() { nav.push<BHTView>(); }},
        {"GPS Sim", ui::Color::yellow(), &bitmap_icon_gps_sim, [&nav]() { nav.push<GpsSimAppView>(); }},
        {"Jammer", ui::Color::green(), &bitmap_icon_jammer, [&nav]() { nav.push<JammerView>(); }},
        //{ "Key fob",		ui::Color::orange(),	&bitmap_icon_keyfob,	[&nav](){ nav.push<KeyfobView>(); } },
        {"LGE tool", ui::Color::yellow(), &bitmap_icon_lge, [&nav]() { nav.push<LGEView>(); }},
        {"Morse", ui::Color::green(), &bitmap_icon_morse, [&nav]() { nav.push<MorseView>(); }},
        {"BurgerPgr", ui::Color::yellow(), &bitmap_icon_burger, [&nav]() { nav.push<CoasterPagerView>(); }},
        //{ "Nuoptix DTMF", 	ui::Color::green(),		&bitmap_icon_nuoptix,	[&nav](){ nav.push<NuoptixView>(); } },
        {"OOK", ui::Color::yellow(), &bitmap_icon_remote, [&nav]() { nav.push<EncodersView>(); }},
        {"POCSAG", ui::Color::green(), &bitmap_icon_pocsag, [&nav]() { nav.push<POCSAGTXView>(); }},
        {"RDS", ui::Color::green(), &bitmap_icon_rds, [&nav]() { nav.push<RDSView>(); }},
        {"Soundbrd", ui::Color::green(), &bitmap_icon_soundboard, [&nav]() { nav.push<SoundBoardView>(); }},
        {"SSTV", ui::Color::green(), &bitmap_icon_sstv, [&nav]() { nav.push<SSTVTXView>(); }},
        {"TEDI/LCR", ui::Color::yellow(), &bitmap_icon_lcr, [&nav]() { nav.push<LCRView>(); }},
        {"TouchTune", ui::Color::yellow(), &bitmap_icon_remote, [&nav]() { nav.push<TouchTunesView>(); }},
        {"Playlist", ui::Color::green(), &bitmap_icon_scanner, [&nav]() { nav.push<PlaylistView>(); }},
        {"S.Painter", ui::Color::orange(), &bitmap_icon_morse, [&nav]() { nav.push<SpectrumPainterView>(); }},
        //{ "Remote",			ui::Color::dark_grey(),	&bitmap_icon_remote,	[&nav](){ nav.push<RemoteView>(); } },
    });
}

/* UtilitiesMenuView *****************************************************/

UtilitiesMenuView::UtilitiesMenuView(NavigationView& nav) {
    if (portapack::persistent_memory::show_gui_return_icon()) {
        add_items({{"..", ui::Color::light_grey(), &bitmap_icon_previous, [&nav]() { nav.pop(); }}});
    }
    add_items({
        //{ "Test app", 		ui::Color::dark_grey(),	nullptr,				[&nav](){ nav.push<TestView>(); } },
        {"Freq. manager", ui::Color::green(), &bitmap_icon_freqman, [&nav]() { nav.push<FrequencyManagerView>(); }},
        {"File manager", ui::Color::yellow(), &bitmap_icon_dir, [&nav]() { nav.push<FileManagerView>(); }},
        {"Notepad", ui::Color::dark_cyan(), &bitmap_icon_notepad, [&nav]() { nav.push<TextEditorView>(); }},
        {"Signal gen", ui::Color::green(), &bitmap_icon_cwgen, [&nav]() { nav.push<SigGenView>(); }},
        //{ "Tone search",	ui::Color::dark_grey(), nullptr,					[&nav](){ nav.push<ToneSearchView>(); } },
        {"Wav viewer", ui::Color::yellow(), &bitmap_icon_soundboard, [&nav]() { nav.push<ViewWavView>(); }},
        {"Antenna length", ui::Color::green(), &bitmap_icon_tools_antenna, [&nav]() { nav.push<WhipCalcView>(); }},

        {"Wipe SD card", ui::Color::red(), &bitmap_icon_tools_wipesd, [&nav]() { nav.push<WipeSDView>(); }},
        {"Flash Utility", ui::Color::red(), &bitmap_icon_temperature, [&nav]() { nav.push<FlashUtilityView>(); }},
        {"SD over USB", ui::Color::yellow(), &bitmap_icon_hackrf, [&nav]() { nav.push<SdOverUsbView>(); }},
    });
    set_max_rows(2);  // allow wider buttons
}

/* SystemMenuView ********************************************************/

void SystemMenuView::hackrf_mode(NavigationView& nav) {
    nav.push<ModalMessageView>("HackRF mode", " This mode enables HackRF\n functionality. To return,\n  press the reset button.\n\n  Switch to HackRF mode?", YESNO,
                               [this](bool choice) {
                                   if (choice) {
                                       EventDispatcher::request_stop();
                                   }
                               });
}

SystemMenuView::SystemMenuView(NavigationView& nav) {
    add_items({
        //{ "Play dead",				ui::Color::red(),		&bitmap_icon_playdead,	[&nav](){ nav.push<PlayDeadView>(); } },
        {"Receive", ui::Color::cyan(), &bitmap_icon_receivers, [&nav]() { nav.push<ReceiversMenuView>(); }},
        {"Transmit", ui::Color::cyan(), &bitmap_icon_transmit, [&nav]() { nav.push<TransmittersMenuView>(); }},
        {"Capture", ui::Color::red(), &bitmap_icon_capture, [&nav]() { nav.push<CaptureAppView>(); }},
        {"Replay", ui::Color::green(), &bitmap_icon_replay, [&nav]() { nav.push<ReplayAppView>(); }},
        {"Search", ui::Color::yellow(), &bitmap_icon_search, [&nav]() { nav.push<SearchView>(); }},
        {"Scanner", ui::Color::yellow(), &bitmap_icon_scanner, [&nav]() { nav.push<ScannerView>(); }},
        {"Microphone", ui::Color::yellow(), &bitmap_icon_microphone, [&nav]() { nav.push<MicTXView>(); }},
        {"Looking Glass", ui::Color::yellow(), &bitmap_icon_looking, [&nav]() { nav.push<GlassView>(); }},
        {"Utilities", ui::Color::cyan(), &bitmap_icon_utilities, [&nav]() { nav.push<UtilitiesMenuView>(); }},
        {"Settings", ui::Color::cyan(), &bitmap_icon_setup, [&nav]() { nav.push<SettingsMenuView>(); }},
        {"Debug", ui::Color::light_grey(), &bitmap_icon_debug, [&nav]() { nav.push<DebugMenuView>(); }},
        {"HackRF", ui::Color::cyan(), &bitmap_icon_hackrf, [this, &nav]() { hackrf_mode(nav); }},
        //{ "About", 		ui::Color::cyan(),			nullptr,				[&nav](){ nav.push<AboutView>(); } }
    });
    set_max_rows(2);  // allow wider buttons
    set_arrow_enabled(false);
    // set_highlighted(1);		// Startup selection
}

/* SystemView ************************************************************/

static constexpr ui::Style style_default{
    .font = ui::font::fixed_8x16,
    .background = ui::Color::black(),
    .foreground = ui::Color::white()};

SystemView::SystemView(
    Context& context,
    const Rect parent_rect)
    : View{parent_rect},
      context_(context) {
    set_style(&style_default);

    constexpr ui::Dim status_view_height = 16;
    constexpr ui::Dim info_view_height = 16;

    add_child(&status_view);
    status_view.set_parent_rect({{0, 0},
                                 {parent_rect.width(), status_view_height}});
    status_view.on_back = [this]() {
        this->navigation_view.pop();
    };

    add_child(&navigation_view);
    navigation_view.set_parent_rect({{0, status_view_height},
                                     {parent_rect.width(), static_cast<ui::Dim>(parent_rect.height() - status_view_height)}});

    add_child(&info_view);
    info_view.set_parent_rect({{0, 19 * 16},
                               {parent_rect.width(), info_view_height}});

    navigation_view.on_view_changed = [this](const View& new_view) {
        if (!this->navigation_view.is_top()) {
            remove_child(&info_view);
        } else {
            add_child(&info_view);
            info_view.refresh();
        }

        this->status_view.set_back_enabled(!this->navigation_view.is_top());
        this->status_view.set_title_image_enabled(this->navigation_view.is_top());
        this->status_view.set_title(new_view.title());
        this->status_view.set_dirty();
    };

    // portapack::persistent_memory::set_playdead_sequence(0x8D1);

    // Initial view
    /*if ((portapack::persistent_memory::playing_dead() == 0x5920C1DF) ||		// Enable code
                (portapack::persistent_memory::ui_config() & 16)) {					// Login option
                navigation_view.push<PlayDeadView>();
        } else {*/

    navigation_view.push<SystemMenuView>();

    File pmem_flag_file_handle;
    std::string pmem_flag_file = "/SETTINGS/PMEM_FILEFLAG";
    auto result = pmem_flag_file_handle.open(pmem_flag_file);
    if (!result.is_valid()) {
        portapack::persistent_memory::load_persistent_settings_from_file("SETTINGS/pmem_settings");
    }

    if (portapack::persistent_memory::config_splash()) {
        navigation_view.push<BMPView>();
    }
    status_view.set_back_enabled(false);
    status_view.set_title_image_enabled(true);
    status_view.set_dirty();
    // else
    //	navigation_view.push<SystemMenuView>();

    //}
}

Context& SystemView::context() const {
    return context_;
}

void SystemView::toggle_overlay() {
    if (overlay_active) {
        this->remove_child(&this->overlay);
        this->set_dirty();
        shared_memory.request_m4_performance_counter = 0;
    } else {
        this->add_child(&this->overlay);
        this->set_dirty();
        shared_memory.request_m4_performance_counter = 1;
        shared_memory.m4_cpu_usage = 0;
        shared_memory.m4_heap_usage = 0;
        shared_memory.m4_stack_usage = 0;
    }

    overlay_active = !overlay_active;
}

void SystemView::paint_overlay() {
    static bool last_paint_state = false;
    if (overlay_active) {
        // paint background only every other second
        if ((((chTimeNow() >> 10) & 0x01) == 0x01) == last_paint_state)
            return;

        last_paint_state = !last_paint_state;
        this->overlay.set_dirty();
    }
}

/* ***********************************************************************/

void BMPView::focus() {
    button_done.focus();
}

BMPView::BMPView(NavigationView& nav) {
    add_children({&button_done});

    button_done.on_select = [this, &nav](Button&) {
        nav.pop();
    };
}

void BMPView::paint(Painter&) {
    if (!portapack::display.drawBMP2({0, 0}, "splash.bmp"))
        portapack::display.drawBMP({(240 - 230) / 2, (320 - 50) / 2 - 10}, splash_bmp, false);
}

/* NotImplementedView ****************************************************/

/*NotImplementedView::NotImplementedView(NavigationView& nav) {
        button_done.on_select = [&nav](Button&){
                nav.pop();
        };

        add_children({
                &text_title,
                &button_done,
        });
}

void NotImplementedView::focus() {
        button_done.focus();
}*/

/* ModalMessageView ******************************************************/

ModalMessageView::ModalMessageView(
    NavigationView& nav,
    const std::string& title,
    const std::string& message,
    const modal_t type,
    const std::function<void(bool)> on_choice)
    : title_{title},
      message_{message},
      type_{type},
      on_choice_{on_choice} {
    if (type == INFO) {
        add_child(&button_ok);

        button_ok.on_select = [this, &nav](Button&) {
            if (on_choice_)
                on_choice_(true);  // Assumes handler will pop.
            else
                nav.pop();
        };
    } else if (type == YESNO) {
        add_children({&button_yes,
                      &button_no});

        button_yes.on_select = [this, &nav](Button&) {
            if (on_choice_) on_choice_(true);
            nav.pop();
        };
        button_no.on_select = [this, &nav](Button&) {
            if (on_choice_) on_choice_(false);
            nav.pop();
        };
    } else if (type == YESCANCEL) {
        add_children({&button_yes,
                      &button_no});

        button_yes.on_select = [this, &nav](Button&) {
            if (on_choice_) on_choice_(true);
            nav.pop();
        };
        button_no.on_select = [this, &nav](Button&) {
            // if (on_choice_) on_choice_(false);
            nav.pop_modal();
        };
    } else {  // ABORT
        add_child(&button_ok);

        button_ok.on_select = [this, &nav](Button&) {
            if (on_choice_) on_choice_(true);
            nav.pop_modal();
        };
    }
}

void ModalMessageView::paint(Painter& painter) {
    size_t pos, i = 0, start = 0;

    portapack::display.drawBMP({100, 48}, modal_warning_bmp, false);

    // Terrible...
    while ((pos = message_.find("\n", start)) != std::string::npos) {
        painter.draw_string(
            {1 * 8, (Coord)(120 + (i * 16))},
            style(),
            message_.substr(start, pos - start));
        i++;
        start = pos + 1;
    }
    painter.draw_string(
        {1 * 8, (Coord)(120 + (i * 16))},
        style(),
        message_.substr(start, pos));
}

void ModalMessageView::focus() {
    if ((type_ == YESNO) || (type_ == YESCANCEL)) {
        button_yes.focus();
    } else {
        button_ok.focus();
    }
}

} /* namespace ui */