/*
 * 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_nuoptix.hpp"

#include "ch.h"
#include "portapack.hpp"
#include "lfsr_random.hpp"
#include "string_format.hpp"

#include "portapack_shared_memory.hpp"

#include <cstring>
#include <vector>

using namespace portapack;

namespace ui {

void NuoptixView::focus() {
    number_timecode.focus();
}

NuoptixView::~NuoptixView() {
    transmitter_model.disable();
    baseband::shutdown();
}

void NuoptixView::on_tx_progress(const uint32_t progress, const bool done) {
    if (done)
        transmit(false);
    else
        progressbar.set_value(progress);
}

void NuoptixView::transmit(bool setup) {
    uint8_t mod, tone_code;
    uint8_t c;
    uint8_t dtmf_message[6];
    rtc::RTC datetime;

    if (!tx_mode) {
        transmitter_model.disable();
        return;
    }

    if (tx_mode == IMPROVISE)
        timecode = lfsr_iterate(timecode) % 1999;  // Could be 9999 but that would be one long audio track !

    if (setup) {
        progressbar.set_max(6 * 2);

        if (tx_mode == IMPROVISE) {
            // Seed from RTC
            rtcGetTime(&RTCD1, &datetime);
            timecode = datetime.day() + datetime.second();
        } else {
            timecode = number_timecode.value();
        }

        transmitter_model.set_rf_amp(true);
        transmitter_model.enable();

        dtmf_message[0] = '*';  // "Pre-tone for restart" method #1
        dtmf_message[1] = 'A';  // "Restart" method #1
    } else {
        dtmf_message[0] = '#';
        dtmf_message[1] = (timecode / 1000) % 10;
        chThdSleepMilliseconds(92);  // 141-49ms
        number_timecode.set_value(timecode);
    }

    progressbar.set_value(0);

    dtmf_message[2] = (timecode / 100) % 10;
    dtmf_message[3] = (timecode / 10) % 10;
    dtmf_message[4] = timecode % 10;

    mod = 0;
    for (c = 1; c < 5; c++)
        if (dtmf_message[c] <= 9)
            mod += dtmf_message[c];

    mod = 10 - (mod % 10);
    if (mod == 10) mod = 0;  // Is this right ?

    text_mod.set("Mod: " + to_string_dec_uint(mod));

    dtmf_message[5] = mod;

    for (c = 0; c < 6; c++) {
        tone_code = dtmf_message[c];

        if (tone_code == 'A')
            tone_code = 10;
        else if (tone_code == 'B')
            tone_code = 11;
        else if (tone_code == 'C')
            tone_code = 12;
        else if (tone_code == 'D')
            tone_code = 13;
        else if (tone_code == '#')
            tone_code = 14;
        else if (tone_code == '*')
            tone_code = 15;

        shared_memory.bb_data.tones_data.message[c * 2] = tone_code;
        shared_memory.bb_data.tones_data.message[c * 2 + 1] = 0xFF;  // Silence
    }

    for (c = 0; c < 16; c++) {
        baseband::set_tone(c * 2, dtmf_deltas[c][0], NUOPTIX_TONE_LENGTH);
        baseband::set_tone(c * 2 + 1, dtmf_deltas[c][1], NUOPTIX_TONE_LENGTH);
    }
    shared_memory.bb_data.tones_data.silence = NUOPTIX_TONE_LENGTH;  // 49ms tone, 49ms space

    audio::set_rate(audio::Rate::Hz_24000);

    baseband::set_tones_config(transmitter_model.channel_bandwidth(), 0, 6 * 2, true, true);

    timecode++;
}

NuoptixView::NuoptixView(
    NavigationView& nav) {
    baseband::run_image(portapack::spi_flash::image_tag_tones);

    add_children({&number_timecode,
                  &text_timecode,
                  &text_mod,
                  &progressbar,
                  &tx_view});

    number_timecode.set_value(1);

    tx_view.on_edit_frequency = [this, &nav]() {
        auto new_view = nav.push<FrequencyKeypadView>(transmitter_model.target_frequency());
        new_view->on_changed = [this](rf::Frequency f) {
            transmitter_model.set_target_frequency(f);
        };
    };

    tx_view.on_start = [this]() {
        tx_view.set_transmitting(true);
        tx_mode = NORMAL;
        transmit(true);
    };

    tx_view.on_stop = [this]() {
        tx_view.set_transmitting(false);
        tx_mode = IDLE;
    };

    /*button_impro.on_select = [this](Button&){
                if (tx_mode == IMPROVISE) {
                        tx_mode = IDLE;
                        button_impro.set_text("IMPROVISE");
                } else if (tx_mode == IDLE) {
                        tx_mode = IMPROVISE;
                        button_impro.set_text("STOP");
                        transmit(true);
                }
        };*/
}

}  // namespace ui