/*
 * Copyright (C) 2014 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 "irq_controls.hpp"
#include "max283x.hpp"
#include "portapack.hpp"
#include "string_format.hpp"
#include "ui_receiver.hpp"
#include "ui_freqman.hpp"

#ifndef PRALINE
#include "audio.hpp"
#endif

using namespace portapack;

namespace ui {

/* FrequencyField ********************************************************/

FrequencyField::FrequencyField(
    const Point parent_pos)
    : Widget{{parent_pos, {8 * 10, 16}}},
      length_{11},
      range_{rf::tuning_range} {
    initial_switch_config_ = get_switches_long_press_config();
    set_focusable(true);
}

FrequencyField::FrequencyField(
    const Point parent_pos,
    const rf::FrequencyRange range)
    : Widget{{parent_pos, {8 * 10, 16}}},
      length_{11},
      range_{range} {
    initial_switch_config_ = get_switches_long_press_config();
    set_focusable(true);
}

FrequencyField::~FrequencyField() {
    reset_switch_config();
}

rf::Frequency FrequencyField::value() const {
    return value_;
}

void FrequencyField::set_value(rf::Frequency new_value) {
    // While in digit mode, don't update if the value is
    // out of range. That way going out of range doesn't
    // cause all the other digits to reset.
    if (digit_mode_ && !range_.contains_inc(new_value))
        return;

    new_value = clamp_value(new_value);

    if (new_value != value_) {
        value_ = new_value;
        if (on_change) {
            on_change(value_);
        }
        set_dirty();
    }
}

void FrequencyField::set_step(rf::Frequency new_value) {
    // TODO: Quantize current frequency to a step of the new size?
    step_ = new_value;
}

void FrequencyField::set_allow_digit_mode(bool allowed) {
    allow_digit_mode_ = allowed;

    if (!allowed && digit_mode_) {
        digit_mode_ = false;
        reset_switch_config();
        set_dirty();
    }
}

void FrequencyField::getAccessibilityText(std::string& result) {
    result = to_string_dec_int(value_);
}
void FrequencyField::getWidgetName(std::string& result) {
    result = "FrequencyField";
}

void FrequencyField::paint(Painter& painter) {
    const auto str_value = to_string_short_freq(value_);
    const auto paint_style = has_focus() ? style().invert() : style();

    painter.draw_string(
        screen_pos(),
        paint_style,
        str_value);

    // Highlight current digit in digit_mode.
    if (digit_mode_) {
        auto p = screen_pos();
        p += {digit_ * char_width, 0};
        painter.draw_char(p, *Theme::getInstance()->option_active, str_value[digit_]);
    }
}

bool FrequencyField::on_key(KeyEvent event) {
    if (event == KeyEvent::Select) {
        // Toggle 'digit' mode with long-press.
        if (digit_mode_ || key_is_long_pressed(event)) {
            digit_mode_ = !digit_mode_;
            set_dirty();
            return true;
        }

        if (on_edit) {
            on_edit();
            return true;
        }
    }

    if (digit_mode_) {
        constexpr uint8_t decimal_pos = 4;
        int8_t delta = 0;

        switch (event) {
            case KeyEvent::Up:
                set_value(value_ + digit_step());
                break;
            case KeyEvent::Down:
                set_value(value_ - digit_step());
                break;
            case KeyEvent::Left:
                delta = -1;
                break;
            case KeyEvent::Right:
                delta = 1;
                break;
            default:
                return false;
        }

        if (delta != 0) {
            digit_ += delta;

            // If on decimal, skip it.
            if (digit_ == decimal_pos)
                digit_ += delta;
            // Otherwise ensure in bounds.
            else
                digit_ = clip<int8_t>(digit_, 0, 8);

            set_dirty();
        }

        return true;
    }

    return false;
}

bool FrequencyField::on_keyboard(KeyboardEvent key) {
    if (key == '+' || key == ' ') return on_encoder(1);
    if (key == '-') return on_encoder(-1);

    return false;
}

bool FrequencyField::on_encoder(const EncoderEvent delta) {
    if (digit_mode_)
        set_value(value_ + (delta * digit_step()));
    else
        set_value(value_ + (delta * step_));
    return true;
}

bool FrequencyField::on_touch(const TouchEvent event) {
    if (event.type == TouchEvent::Type::Start) {
        focus();
    }
    return true;
}

void FrequencyField::on_focus() {
    if (on_show_options)
        on_show_options();

    enable_switch_config();
}

void FrequencyField::on_blur() {
    reset_switch_config();
}

rf::Frequency FrequencyField::digit_step() const {
    constexpr rf::Frequency steps[] = {
        1'000'000'000,
        100'000'000,
        10'000'000,
        1'000'000,
        0,  // Decimal point position.
        100'000,
        10'000,
        1'000,
        100,
    };

    return digit_ < std::size(steps) ? steps[digit_] : 0;
}

rf::Frequency FrequencyField::clamp_value(rf::Frequency value) {
    return range_.clip(value);
}

void FrequencyField::enable_switch_config() {
    // Don't enable long press testing when not allowed.
    if (!allow_digit_mode_)
        return;

    // Enable long press on "Select".
    SwitchesState config;
    config[toUType(Switch::Sel)] = true;
    set_switches_long_press_config(config);
}

void FrequencyField::reset_switch_config() {
    set_switches_long_press_config(initial_switch_config_);
}

/* FrequencyKeypadView ***************************************************/

bool FrequencyKeypadView::on_encoder(const EncoderEvent delta) {
    focused_button += delta;
    if (focused_button < 0) {
        focused_button = buttons.size() - 1;
    } else if (focused_button >= (int16_t)buttons.size()) {
        focused_button = 0;
    }
    buttons[focused_button].focus();
    return true;
}

FrequencyKeypadView::FrequencyKeypadView(
    NavigationView& nav,
    const rf::Frequency value) {
    button_w = screen_width / 3;
    add_child(&text_value);

    const auto button_fn = [this](Button& button) {
        this->on_button(button);
    };

    const char* const key_caps = "123456789<0.";

    int n = 0;
    for (auto& button : buttons) {
        add_child(&button);
        const std::string label{
            key_caps[n]};
        button.id = n;
        button.on_highlight = [this](Button& button) {
            focused_button = button.id;
        };
        button.on_select = button_fn;
        button.set_parent_rect({(n % 3) * button_w,
                                (n / 3) * button_h + 25,
                                button_w, button_h});
        button.set_text(label);
        n++;
    }

    add_children({&button_save_ghz,
                  &button_save_khz,
                  &button_save_mhz,
                  &button_load,
                  &button_clear,
                  &button_done_ghz,
                  &button_done_mhz,
                  &button_done_khz});

    button_save_mhz.on_select = [this, &nav](Button&) {
        nav.push<FrequencySaveView>(this->value(FrequencyUnit::MHZ));  // pass as mhz and handle unit convert logic there, cuz idk if pass others would loss something
    };
    button_save_ghz.on_select = [this, &nav](Button&) {
        nav.push<FrequencySaveView>(this->value(FrequencyUnit::GHZ));
    };
    button_save_khz.on_select = [this, &nav](Button&) {
        nav.push<FrequencySaveView>(this->value(FrequencyUnit::KHZ));
    };

    button_load.on_select = [this, &nav](Button&) {
        auto load_view = nav.push<FrequencyLoadView>();
        load_view->on_frequency_loaded = [this](rf::Frequency value) {
            set_value(value);
        };
    };
    button_clear.on_select = [this](Button&) {
        mhz.clear();
        submhz.clear();
        clear_field_if_digits_entered = true;
        if (state == State::DigitSubMHz) {
            state = State::DigitMHz;
        }
        draw_input_hint();
        update_text();
    };

    button_done_ghz.on_select = [this, &nav](Button&) {
        if (on_changed)
            on_changed(this->value(FrequencyUnit::GHZ));
        nav.pop();
    };

    button_done_mhz.on_select = [this, &nav](Button&) {
        if (on_changed)
            on_changed(this->value(FrequencyUnit::MHZ));
        nav.pop();
    };

    button_done_khz.on_select = [this, &nav](Button&) {
        if (on_changed)
            on_changed(this->value(FrequencyUnit::KHZ));
        nav.pop();
    };

    set_value(value);
}

void FrequencyKeypadView::focus() {
    button_done_mhz.focus();
}

rf::Frequency FrequencyKeypadView::value(FrequencyUnit frequency_uni) const {
    switch (frequency_uni) {
        case FrequencyUnit::GHZ:
            return mhz.as_int() * 1000000000ULL + submhz.as_int() * submhz_base * 1000;
        case FrequencyUnit::MHZ:
            return mhz.as_int() * 1000000ULL + submhz.as_int() * submhz_base;
        case FrequencyUnit::KHZ:
            return mhz.as_int() * 1000ULL + submhz.as_int() * submhz_base / 1000;
        default:
            return mhz.as_int() * 1000000ULL + submhz.as_int() * submhz_base;
    }
}

void FrequencyKeypadView::set_value(const rf::Frequency new_value) {
    mhz.set(new_value / 1000000);
    mhz.remove_leading_zeros();

    submhz.set((new_value % 1000000) / submhz_base);
    submhz.remove_trailing_zeros();

    update_text();
}

void FrequencyKeypadView::on_button(Button& button) {
    const auto s = button.text();
    if (s == ".") {
        field_toggle();
    } else if (s == "<") {
        digit_delete();
    } else {
        digit_add(s[0]);
    }
    update_text();
}

bool FrequencyKeypadView::on_keyboard(const KeyboardEvent key) {
    if (key == 8) {
        digit_delete();
        update_text();
        return true;
    }
    if (key >= '0' && key <= '9') {
        digit_add(key);
        update_text();
        return true;
    }
    if (key == '.') {
        field_toggle();
        update_text();
        return true;
    }
    return false;
}

void FrequencyKeypadView::digit_add(const char c) {
    if (state == State::DigitMHz) {
        if (clear_field_if_digits_entered) {
            mhz.clear();
        }
        mhz.add_digit(c);
    } else {
        submhz.add_digit(c);
    }
    clear_field_if_digits_entered = false;
}

void FrequencyKeypadView::digit_delete() {
    if (state == State::DigitMHz) {
        mhz.delete_digit();
    } else {
        submhz.delete_digit();
    }
}

void FrequencyKeypadView::field_toggle() {
    if (state == State::DigitMHz) {
        state = State::DigitSubMHz;
        submhz.clear();
    } else {
        state = State::DigitMHz;
        clear_field_if_digits_entered = true;
    }
    draw_input_hint();
}

void FrequencyKeypadView::draw_input_hint() {
    set_dirty();
}

void FrequencyKeypadView::paint(Painter& painter) {
    View::paint(painter);

    const bool s = state == State::DigitMHz;
    painter.draw_hline(
        {0, 36},
        8 * 4,
        s ? Color::white() : Color::black());
    painter.draw_hline(
        {5 * 8, 36},
        8 * 4,
        s ? Color::black() : Color::white());
}

void FrequencyKeypadView::update_text() {
    const auto s = mhz.as_string() + "." + submhz.as_string();
    text_value.set(s);
}

/* FrequencyOptionsView **************************************************/

FrequencyOptionsView::FrequencyOptionsView(
    const Rect parent_rect,
    const Style* const style)
    : View{parent_rect} {
    set_style(style);

    field_step.on_change = [this](size_t n, OptionsField::value_t v) {
        (void)n;
        this->on_step_changed(v);
    };

    field_ppm.on_change = [this](int32_t v) {
        this->on_reference_ppm_correction_changed(v);
    };

    add_children({
        &text_step,
        &field_step,
    });

    if (portapack::clock_manager.get_reference().source == ClockManager::ReferenceSource::Xtal) {
        add_child(&field_ppm);
        add_child(&text_ppm);
    }
}

void FrequencyOptionsView::set_step(rf::Frequency f) {
    field_step.set_by_nearest_value(f);
}

void FrequencyOptionsView::set_reference_ppm_correction(int32_t v) {
    field_ppm.set_value(v);
}

void FrequencyOptionsView::on_step_changed(rf::Frequency v) {
    if (on_change_step) {
        on_change_step(v);
    }
}

void FrequencyOptionsView::on_reference_ppm_correction_changed(int32_t v) {
    if (on_change_reference_ppm_correction) {
        on_change_reference_ppm_correction(v);
    }
}

/* RFAmpField ************************************************************/

RFAmpField::RFAmpField(
    Point parent_pos)
    : NumberField{
          parent_pos,
          1,
          {0, 1},
          1,
          ' ',
      } {
    set_value(receiver_model.rf_amp());

    on_change = [](int32_t v) {
        receiver_model.set_rf_amp(v);
    };
}

/* RadioGainOptionsView **************************************************/

RadioGainOptionsView::RadioGainOptionsView(
    const Rect parent_rect,
    const Style* const style)
    : View{parent_rect} {
    set_style(style);

    add_children({
        &label_rf_amp,
        &field_rf_amp,
    });
}

/* LNAGainField **********************************************************/

LNAGainField::LNAGainField(
    Point parent_pos)
    : NumberField{
          parent_pos,
          2,
          {max283x::lna::gain_db_range.minimum, max283x::lna::gain_db_range.maximum},
          max283x::lna::gain_db_step,
          ' ',
      } {
    set_value(receiver_model.lna());

    on_change = [](int32_t v) {
        receiver_model.set_lna(v);
    };
}

void LNAGainField::on_focus() {
    if (on_show_options) {
        on_show_options();
    }
}

/* VGAGainField **********************************************************/

VGAGainField::VGAGainField(
    Point parent_pos)
    : NumberField{
          parent_pos,
          2,
          {max283x::vga::gain_db_range.minimum, max283x::vga::gain_db_range.maximum},
          max283x::vga::gain_db_step,
          ' ',
      } {
    set_value(receiver_model.vga());

    on_change = [](int32_t v) {
        receiver_model.set_vga(v);
    };
}

void VGAGainField::on_focus() {
    if (on_show_options) {
        on_show_options();
    }
}

/* AudioVolumeField *******************************************************/

AudioVolumeField::AudioVolumeField(
    Point parent_pos)
    : NumberField{
          parent_pos,
          /* length */ 2,
          /* range */ {0, 99},
          /* step */ 1,
          /* fill char */ ' '} {
    set_value(receiver_model.normalized_headphone_volume());

#ifdef PRALINE
    on_change = [](int32_t v) {
        receiver_model.set_normalized_headphone_volume(v);
#else
    on_change = [](int32_t vol) {
        // don't call receiver model, because this widget shuld be able to handle volume settinsg from any app, regardless of the receiver model's enables state. like the tx apps should be able to set volume too.
        // this is identical to the receiver model's method
        uint8_t v = clip<uint8_t>(vol, 0, 99);
        auto new_volume = volume_t::decibel(v - 99) + audio::headphone::volume_range().max;
        persistent_memory::set_headphone_volume(new_volume);
        audio::headphone::set_volume(new_volume);
#endif
    };
}

} /* namespace ui */