/*
 * 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"

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::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, Styles::bg_blue, 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) {
    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 + 24,
                                button_w, button_h});
        button.set_text(label);
        n++;
    }

    add_children({&button_save,
                  &button_load,
                  &button_close});

    button_save.on_select = [this, &nav](Button&) {
        nav.push<FrequencySaveView>(this->value());
    };
    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_close.on_select = [this, &nav](Button&) {
        if (on_changed)
            on_changed(this->value());
        nav.pop();
    };

    set_value(value);
}

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

rf::Frequency FrequencyKeypadView::value() const {
    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;
    }
}

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());

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

} /* namespace ui */