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

void WeatherProcessor::execute(const buffer_c8_t& buffer) {
    if (!configured) return;

    // SR = 4Mhz ,  and we are decimating by /8 in total , decim1_out clock 4Mhz /8= 500khz samples/sec.
    // buffer has   2048 complex i8 I,Q signed samples
    // decim0 out:  2048/4 = 512 complex i16 I,Q signed samples
    // decim1 out:  512/2 =  256 complex i16 I,Q signed samples
    // Regarding Filters, we are re-using existing FIR filters, @4Mhz, FIR decim1 ilter, BW =+-220Khz (at -3dB's). BW = 440kHZ.

    const auto decim_0_out = decim_0.execute(buffer, dst_buffer);       // Input:2048 complex/4 (decim factor) = 512_output complex (1024 I/Q samples)
    const auto decim_1_out = decim_1.execute(decim_0_out, dst_buffer);  // Input:512  complex/2 (decim factor) = 256_output complex ( 512 I/Q samples)
    feed_channel_stats(decim_1_out);

    for (size_t i = 0; i < decim_1_out.count; i++) {
        threshold = (low_estimate + high_estimate) / 2;
        int32_t const hysteresis = threshold / 8;  // +-12%
        int16_t re = decim_1_out.p[i].real();
        int16_t im = decim_1_out.p[i].imag();
        uint32_t mag = ((uint32_t)re * (uint32_t)re) + ((uint32_t)im * (uint32_t)im);

        mag = (mag >> 10);
        int32_t const ook_low_delta = mag - low_estimate;
        bool meashl = currentHiLow;
        if (sig_state == STATE_IDLE) {
            if (mag > (threshold + hysteresis)) {  // just become high
                meashl = true;
                sig_state = STATE_PULSE;
                numg = 0;
            } else {
                meashl = false;  // still low
                low_estimate += ook_low_delta / OOK_EST_LOW_RATIO;
                low_estimate += ((ook_low_delta > 0) ? 1 : -1);  // Hack to compensate for lack of fixed-point scaling
                // Calculate default OOK high level estimate
                high_estimate = 1.35 * low_estimate;  // Default is a ratio of low level
                high_estimate = std::max(high_estimate, min_high_level);
                high_estimate = std::min(high_estimate, (uint32_t)OOK_MAX_HIGH_LEVEL);
            }

        } else if (sig_state == STATE_PULSE) {
            ++numg;
            if (numg > 100) numg = 100;
            if (mag < (threshold - hysteresis)) {
                // check if really a bad value
                if (numg < 3) {
                    // susp
                    sig_state = STATE_GAP;
                } else {
                    numg = 0;
                    sig_state = STATE_GAP_START;
                }
                meashl = false;  // low
            } else {
                high_estimate += mag / OOK_EST_HIGH_RATIO - high_estimate / OOK_EST_HIGH_RATIO;
                high_estimate = std::max(high_estimate, min_high_level);
                high_estimate = std::min(high_estimate, (uint32_t)OOK_MAX_HIGH_LEVEL);
                meashl = true;  // still high
            }
        } else if (sig_state == STATE_GAP_START) {
            ++numg;
            if (mag > (threshold + hysteresis)) {  // New pulse?
                sig_state = STATE_PULSE;
                meashl = true;
            } else if (numg >= 3) {
                sig_state = STATE_GAP;
                meashl = false;  // gap
            }
        } else if (sig_state == STATE_GAP) {
            ++numg;
            if (mag > (threshold + hysteresis)) {  // New pulse?
                numg = 0;
                sig_state = STATE_PULSE;
                meashl = true;
            } else {
                meashl = false;
            }
        }

        if (meashl == currentHiLow && currentDuration < 30'000'000)  // allow pass 'end' signal
        {
            currentDuration += nsPerDecSamp;
        } else {  // called on change, so send the last duration and dir.
            if (currentDuration >= 30'000'000) sig_state = STATE_IDLE;
            if (protoList) protoList->feed(currentHiLow, currentDuration / 1000);
            currentDuration = nsPerDecSamp;
            currentHiLow = meashl;
        }
    }
}

void WeatherProcessor::on_message(const Message* const message) {
    switch (message->id) {
        case Message::ID::SubGhzFPRxConfigure:
            configure(*reinterpret_cast<const SubGhzFPRxConfigureMessage*>(message));
            break;

        case Message::ID::AudioBeep:
            on_beep_message(*reinterpret_cast<const AudioBeepMessage*>(message));
            break;

        default:
            break;
    }
}

void WeatherProcessor::configure(const SubGhzFPRxConfigureMessage& message) {
    baseband_fs = message.sampling_rate;
    baseband_thread.set_sampling_rate(baseband_fs);
    nsPerDecSamp = 1'000'000'000 / baseband_fs * 8;  // Scaled it due to less array buffer sampes due to /8 decimation.  250 nseg (4Mhz) * 8

    // constexpr size_t decim_0_output_fs = baseband_fs / decim_0.decimation_factor; //unused
    // constexpr size_t decim_1_output_fs = decim_0_output_fs / decim_1.decimation_factor; //unused

    decim_0.configure(taps_200k_wfm_decim_0.taps);
    decim_1.configure(taps_200k_wfm_decim_1.taps);

    configured = true;
}

void WeatherProcessor::on_beep_message(const AudioBeepMessage& message) {
    audio::dma::beep_start(message.freq, message.sample_rate, message.duration_ms);
}

int main() {
    audio::dma::init_audio_out();
    EventDispatcher event_dispatcher{std::make_unique<WeatherProcessor>()};
    event_dispatcher.run();
    return 0;
}