/*
 * 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 "proc_ook.hpp"
#include "portapack_shared_memory.hpp"
#include "sine_table_int8.hpp"
#include "event_m4.hpp"

#include <cstdint>

inline void OOKProcessor::write_sample(const buffer_c8_t& buffer, uint8_t bit_value, size_t i) {
    int8_t re, im;

    if (bit_value) {
        phase = (phase + 200);  // What ?
        sphase = phase + (64 << 18);

        re = (sine_table_i8[(sphase & 0x03FC0000) >> 18]);
        im = (sine_table_i8[(phase & 0x03FC0000) >> 18]);
    } else {
        re = 0;
        im = 0;
    }

    buffer.p[i] = {re, im};
}

bool OOKProcessor::scan_init(unsigned int order) {
    if (order > MAX_DE_BRUIJN_ORDER)
        return false;

    scan_done = false;
    scan_progress = 0;

    k = 0;
    idx = 1;

    duval_symbols = 2;  // 2 for binary, 3 for ternary encoders
    duval_length = 0;
    duval_bit = 0;
    duval_sample_bit = 0;
    duval_symbol = 0;

    memset(v, 0, sizeof(v));
    return true;
}

bool OOKProcessor::scan_encode(const buffer_c8_t& buffer, size_t& buf_ptr) {
    // encode data: 0 = 1000, 1 = 1110
    // @TODO: make this user-configurable
    const uint8_t sym[] = {0b0001, 0b0111};
    constexpr auto symbol_length = 4;

    // iterate over every symbol in the sequence and convert it to bits with required bitrate
    for (; duval_bit < duval_length; duval_bit++) {
        auto val = v_tmp[duval_bit];
        for (; duval_symbol < symbol_length; duval_symbol++) {
            auto s = sym[val] & (1 << duval_symbol);
            for (; duval_sample_bit < samples_per_bit; duval_sample_bit++) {
                if (buf_ptr >= buffer.count) {
                    // buffer is full - continue next time
                    txprogress_message.done = false;
                    txprogress_message.progress = scan_progress++;
                    shared_memory.application_queue.push(txprogress_message);
                    return false;
                }
                write_sample(buffer, s, buf_ptr++);
            }
            duval_sample_bit = 0;
        }
        duval_symbol = 0;
    }
    duval_bit = 0;
    return true;
}

inline size_t OOKProcessor::duval_algo_step() {
    size_t buf_ptr = 0;
    const unsigned int w = de_bruijn_length;

    // Duval's algorithm for generating de Bruijn sequence
    while (idx) {
        if (w % idx == 0) {
            for (unsigned int k = 0; k < idx; k++)
                v_tmp[buf_ptr++] = v[k];
            k = 0;
        }

        for (unsigned int j = 0; j < w - idx; j++)
            v[idx + j] = v[j];

        for (idx = w; (idx > 0) && (v[idx - 1] >= duval_symbols - 1); idx--)
            ;

        if (idx)
            v[idx - 1]++;

        if (buf_ptr) {
            // we fill at most de_bruijn_length number of elements
            return buf_ptr;
        }
    }

    return 0;
}

void OOKProcessor::scan_process(const buffer_c8_t& buffer) {
    size_t buf_ptr = 0;

    // transmit any leftover bits from previous step
    if (!scan_encode(buffer, buf_ptr))
        return;

    while (1) {
        // calculate next chunk of deBruijn sequence
        duval_length = duval_algo_step();

        if (duval_length == 0) {
            // last chunk - done
            if (!scan_done) {
                txprogress_message.done = true;
                shared_memory.application_queue.push(txprogress_message);
            }
            scan_done = 1;

            // clear the remaining buffer in case we have any bytes left
            for (size_t i = buf_ptr; i < buffer.count; i++)
                buffer.p[i] = {0, 0};

            break;
        }

        duval_bit = 0;
        duval_sample_bit = 0;
        duval_symbol = 0;

        // encode the sequence into required format
        if (!scan_encode(buffer, buf_ptr))
            break;
    }
}

void OOKProcessor::execute(const buffer_c8_t& buffer) {
    // This is called at 2.28M/2048 = 1113Hz

    if (!configured) return;

    if (de_bruijn_length) {
        scan_process(buffer);
        return;
    }

    for (size_t i = 0; i < buffer.count; i++) {
        // Synthesis at 2.28M/10 = 228kHz
        if (!s) {
            s = 10 - 1;
            if (sample_count >= samples_per_bit) {
                if (configured) {
                    if (bit_pos >= length) {
                        // End of data
                        if (pause_counter == 0) {
                            pause_counter = pause;
                            cur_bit = 0;
                        } else if (pause_counter == 1) {
                            if (repeat_counter < repeat) {
                                // Repeat
                                cur_bit = shared_memory.bb_data.data[0] & 0x80;
                                txprogress_message.progress = repeat_counter + 1;
                                txprogress_message.done = false;
                                shared_memory.application_queue.push(txprogress_message);
                                bit_pos = 1;
                                repeat_counter++;
                            } else {
                                // Stop
                                cur_bit = 0;
                                txprogress_message.done = true;
                                shared_memory.application_queue.push(txprogress_message);
                                configured = false;
                            }
                            pause_counter = 0;
                        } else {
                            pause_counter--;
                        }
                    } else {
                        cur_bit = (shared_memory.bb_data.data[bit_pos >> 3] << (bit_pos & 7)) & 0x80;
                        bit_pos++;
                    }
                }

                sample_count = 0;
            } else {
                sample_count++;
            }
        } else {
            s--;
        }

        write_sample(buffer, cur_bit, i);
    }
}

void OOKProcessor::on_message(const Message* const p) {
    const auto message = *reinterpret_cast<const OOKConfigureMessage*>(p);

    if (message.id == Message::ID::OOKConfigure) {
        configured = false;

        repeat = message.repeat - 1;
        length = message.stream_length;
        pause = message.pause_symbols + 1;
        de_bruijn_length = message.de_bruijn_length;
        samples_per_bit = message.samples_per_bit;

        if (!length && !samples_per_bit) {
            // shutdown
            return;
        }

        if (de_bruijn_length) {
            if (!scan_init(de_bruijn_length))
                return;
        } else {
            samples_per_bit /= 10;
        }

        pause_counter = 0;
        s = 0;
        sample_count = samples_per_bit;
        repeat_counter = 0;
        bit_pos = 0;
        cur_bit = 0;
        txprogress_message.progress = 0;
        txprogress_message.done = false;
        configured = true;
    }
}

int main() {
    EventDispatcher event_dispatcher{std::make_unique<OOKProcessor>()};
    event_dispatcher.run();
    return 0;
}