/*
 * Copyright (C) 2015 Jared Boone, ShareBrained Technology, Inc.
 * Copyright (C) 2017 Furrtek
 * Copyright (C) 2022 NotPike
 *
 * 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_touchtunes.hpp"
#include "encoders.hpp"

#include "baseband_api.hpp"
#include "string_format.hpp"
#include "cpld_update.hpp"

using namespace portapack;
using namespace encoders;

namespace ui {

void TouchTunesView::focus() {
    field_pin.focus();
}

TouchTunesView::~TouchTunesView() {
    transmitter_model.disable();
    hackrf::cpld::load_sram_no_verify();  // to leave all RX ok, without ghost signal problem at the exit.
    baseband::shutdown();                 // better this function at the end, not load_sram() that sometimes produces hang up.
}

void TouchTunesView::stop_tx() {
    transmitter_model.disable();
    tx_mode = IDLE;
    progressbar.set_value(0);

    // EW Mode Check
    if (check_ew.value()) {
        start_ew();
    } else {
        text_status.set("Ready");
    }
}

void TouchTunesView::on_tx_progress(const uint32_t progress, const bool done) {
    if (!done) {
        // Progress
        if (tx_mode == SINGLE)
            progressbar.set_value(progress);
        else if (tx_mode == SCAN)
            progressbar.set_value((pin * TOUCHTUNES_REPEATS) + progress);
    } else {
        // Done transmitting
        if (tx_mode == SINGLE) {
            stop_tx();
        } else if (tx_mode == SCAN) {
            if (pin == TOUCHTUNES_MAX_PIN) {
                stop_tx();
            } else {
                transmitter_model.disable();
                pin++;
                field_pin.set_value(pin);
                start_tx(scan_button_index);
            }
        }
    }
}

// EW (Electronic Warfare) Mode will jam the receiving jukebox
// while still alowing you (the hacker) to send commands
// to the target jukebox.
// EW Mode works by transmitting a CW on 433.92MHz inbetween
// transmission events.
void TouchTunesView::start_ew() {
    // Radio
    transmitter_model.set_target_frequency(433920000);
    transmitter_model.set_rf_amp(true);
    transmitter_model.set_tx_gain(47);
    transmitter_model.enable();

    // UI
    text_status.set("Jamming...");
    progressbar.set_max(1);
    progressbar.set_value(1);
}

void TouchTunesView::stop_ew() {
    // Radio
    transmitter_model.disable();

    // UI
    text_status.set("Ready");
    progressbar.set_value(0);
}

void TouchTunesView::start_tx(const uint32_t button_index) {
    // Check EW Mode
    if (check_ew.value()) {
        stop_ew();
    }

    std::string fragments = {""};
    size_t bit;
    uint64_t frame_data;

    if (check_scan.value()) {
        scan_button_index = button_index;
        tx_mode = SCAN;
        progressbar.set_max(TOUCHTUNES_MAX_PIN * TOUCHTUNES_REPEATS);
        text_status.set("Scanning...");
    } else {
        tx_mode = SINGLE;
        progressbar.set_max(TOUCHTUNES_REPEATS);
        text_status.set("Transmitting...");
    }

    frame_data = TOUCHTUNES_SYNC_WORD;  // Sync word

    // Insert pin value (LSB first)
    for (bit = 0; bit < 8; bit++) {
        frame_data <<= 1;
        if (pin & (1 << bit))
            frame_data |= 1;
    }

    // Insert button code (and its complement)
    frame_data <<= 16;
    frame_data |= (button_codes[button_index] << 8);
    frame_data |= (button_codes[button_index] ^ 0xFF);

    // Convert to OOK symbols
    for (bit = 0; bit < (8 + 8 + 16); bit++) {
        fragments += (frame_data & 0x80000000UL) ? "1000" : "10";
        frame_data <<= 1;
    }

    // Sync and end pulse
    fragments = "111111111111111100000000" + fragments + "1000";

    size_t bitstream_length = make_bitstream(fragments);

    transmitter_model.set_target_frequency(433920000);
    transmitter_model.set_sampling_rate(OOK_SAMPLERATE);
    transmitter_model.set_rf_amp(true);
    transmitter_model.set_baseband_bandwidth(1750000);
    transmitter_model.enable();

    baseband::set_ook_data(
        bitstream_length,
        OOK_SAMPLERATE / 1766,  // 560us
        TOUCHTUNES_REPEATS,
        100  // Pause
    );
}

TouchTunesView::TouchTunesView(
    NavigationView&) {
    baseband::run_image(portapack::spi_flash::image_tag_ook);

    add_children({&labels,
                  &field_pin,
                  &check_scan,
                  &check_ew,
                  &text_status,
                  &progressbar});

    field_pin.set_value(pin);

    field_pin.on_change = [this](int32_t v) {
        pin = v;
    };

    // EW Mode
    check_ew.on_select = [this](Checkbox&, bool v) {
        if (v) {
            start_ew();
        } else {
            stop_ew();
        }
    };

    const auto button_fn = [this](Button& button) {
        start_tx(button.id);
    };

    size_t n = 0;
    for (auto& entry : remote_layout) {
        buttons[n].on_select = button_fn;
        buttons[n].id = n;
        buttons[n].set_text(entry.text);
        buttons[n].set_parent_rect({entry.position + Point(8, 0),
                                    {(Dim)(entry.text.length() + 2) * 8, 4 * 8}});
        add_child(&buttons[n]);
        n++;
    }
}

} /* namespace ui */