mirror of
https://github.com/portapack-mayhem/mayhem-firmware.git
synced 2026-09-03 21:39:01 +00:00
Matej Sochan
112 lines
3.7 KiB
C++
112 lines
3.7 KiB
C++
/*
|
|
* Copyright (C) 2026 Matej Sochan
|
|
*
|
|
* 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.
|
|
*/
|
|
|
|
#ifndef __PROC_SIGNAL_HUNTER_H__
|
|
#define __PROC_SIGNAL_HUNTER_H__
|
|
|
|
#include <atomic>
|
|
|
|
#include "baseband_processor.hpp"
|
|
#include "baseband_thread.hpp"
|
|
#include "rssi_thread.hpp"
|
|
#include "message.hpp"
|
|
#include "dsp_decimate.hpp"
|
|
#include "stream_input.hpp"
|
|
|
|
#include <cstdint>
|
|
#include <cstddef>
|
|
#include <array>
|
|
#include <memory>
|
|
|
|
class SignalHunterProcessor : public BasebandProcessor {
|
|
public:
|
|
void execute(const buffer_c8_t& buffer) override;
|
|
void on_message(const Message* const message) override;
|
|
|
|
private:
|
|
static constexpr size_t baseband_fs = 2000000;
|
|
|
|
// Hardcoded Decimate-by-8 FIR filter.
|
|
// 2,000,000 Hz / 8 = 250,000 Hz target sample rate.
|
|
// Used to avoid SD Card bottlenecks
|
|
dsp::decimate::FIRC8xR16x24FS4Decim8 decim_0{};
|
|
|
|
std::array<complex16_t, 512> dst_buffer_data{};
|
|
const buffer_c16_t dst_buffer{dst_buffer_data.data(), dst_buffer_data.size()};
|
|
|
|
// --- Energy sliding window for threshold detection ---
|
|
static constexpr size_t WINDOW_SIZE = 64;
|
|
uint32_t window_buf[WINDOW_SIZE]{};
|
|
size_t window_idx{0};
|
|
uint32_t window_sum{0};
|
|
|
|
std::atomic<bool> flush_pending{false};
|
|
size_t flush_start_idx{0};
|
|
|
|
// Pre-trigger IQ ring buffer
|
|
// IQ_RING_SAMPLES = 2048 provides ~8ms pre-roll buffer (2048 / 250kHz).
|
|
// This captures OOK preambles efficiently while maintaining strict 8KB flush size to prevent
|
|
// Out-Of-Memory (OOM) / HardFault crashes on M0 CaptureThread, which has only 16KB buffer constraint.
|
|
static constexpr size_t IQ_RING_SAMPLES = 2048;
|
|
std::array<complex16_t, IQ_RING_SAMPLES> iq_ring{};
|
|
size_t iq_ring_idx{0};
|
|
|
|
uint32_t energy_threshold{5000};
|
|
|
|
std::atomic<bool> hunting{false};
|
|
std::atomic<bool> stream_active{false};
|
|
std::atomic<bool> stream_close_requested{false};
|
|
|
|
bool configured{false};
|
|
|
|
// State machine tracking signal detection lifecycle
|
|
enum class HuntState {
|
|
// Awaiting energy threshold crossing
|
|
IDLE,
|
|
// Trigger sent; waiting for CaptureConfigMessage from M0 to create stream
|
|
AWAITING_STREAM,
|
|
// Stream active; writing live decimated samples
|
|
RECORDING,
|
|
// Energy dropped below threshold; checking if signal returns
|
|
HANGTIME,
|
|
// Stop sent; waiting for CaptureConfigMessage(nullptr) from M0 to destroy stream
|
|
AWAITING_CLOSE
|
|
};
|
|
|
|
std::atomic<HuntState> hunt_state{HuntState::IDLE};
|
|
|
|
uint32_t hangtime_counter{0};
|
|
// Dynamic hangtime configuration: holds sample count in post-decimation domain (250 kHz rate).
|
|
// Converted from milliseconds: hangtime_ms * 250 samples/ms.
|
|
// Set dynamically via HunterConfigMessage to allow configurable silence tolerance.
|
|
uint32_t hangtime_samples_limit{0};
|
|
|
|
std::unique_ptr<StreamInput> stream{};
|
|
|
|
void configure();
|
|
void reset_hunt_state();
|
|
|
|
BasebandThread baseband_thread{baseband_fs, this, baseband::Direction::Receive};
|
|
RSSIThread rssi_thread{};
|
|
};
|
|
|
|
#endif /*__PROC_SIGNAL_HUNTER_H__*/
|