mayhem-firmware/firmware/application/external/signal_hunter/ui_signal_hunter.cpp

/*
 * 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.
 */

#include "ui_signal_hunter.hpp"
#include "receiver_model.hpp"
#include "baseband_api.hpp"
#include "ui_fileman.hpp"
#include "freqman_db.hpp"
#include "string_format.hpp"
#include "io_file.hpp"
#include "rtc_time.hpp"

using namespace portapack;

namespace ui::external_app::signal_hunter {

// --- TAB 1: Main View ---
HunterMainView::HunterMainView(Rect parent_rect, SignalHunterAppView& parent)
    : View(parent_rect), parent_app(parent) {
    add_children({&text_current_freq, &text_status, &button_start_stop, &text_hits});
    update_frequency(433920000);

    button_start_stop.on_select = [this](Button&) {
        parent_app.is_hunting = !parent_app.is_hunting;
        if (parent_app.is_hunting) {
            button_start_stop.set_text("STOP");
            // Set initial frequency based on set mode (single freq / freq hopping)
            if (parent_app.freq_hop_mode && !parent_app.frequency_list.empty()) {
                parent_app.current_freq_index = 0;
                auto freq = parent_app.frequency_list[0];
                receiver_model.set_target_frequency(freq);
                update_frequency(freq);
            } else {
                receiver_model.set_target_frequency(parent_app.single_frequency);
                update_frequency(parent_app.single_frequency);
            }
            update_status("HUNTING...", Theme::getInstance()->fg_green);
            parent_app.send_hunter_config(true);
        } else {
            button_start_stop.set_text("START");
            update_status("IDLE", Theme::getInstance()->fg_light);
            set_recording_state(false);
            parent_app.send_hunter_config(false);
        }
    };
}

void HunterMainView::on_show() {
    if (!parent_app.freq_hop_mode) {
        current_freq_ = receiver_model.target_frequency();
    }

    update_frequency(current_freq_);  // force repaint
}

void HunterMainView::focus() {
    if (!parent_app.freq_hop_mode) {
        current_freq_ = receiver_model.target_frequency();
    }

    update_frequency(current_freq_);
    button_start_stop.focus();
}

void HunterMainView::set_recording_state(bool recording) {
    if (recording)
        text_current_freq.set_style(Theme::getInstance()->fg_red);
    else
        text_current_freq.set_style(Theme::getInstance()->fg_light);

    text_current_freq.set_dirty();    // force ui update
    update_frequency(current_freq_);  // repaint with the new color
}

void HunterMainView::update_frequency(rf::Frequency freq) {
    current_freq_ = freq;
    text_current_freq.set(to_string_dec_uint(freq) + " Hz");
}

void HunterMainView::update_status(const std::string& status, const Style* style) {
    text_status.set(status);
    text_status.set_style(style);
}

void HunterMainView::update_hits(uint32_t hits) {
    text_hits.set("Hits: " + to_string_dec_uint(hits));
}

// --- TAB 2: Freqs View ---
HunterFreqsView::HunterFreqsView(Rect parent_rect, SignalHunterAppView& parent)
    : View(parent_rect), parent_app(parent) {
    add_children({&button_load_file,
                  &button_clear,
                  &labels,
                  &field_dwell,
                  &text_loaded_info});

    field_dwell.set_value(parent_app.hop_dwell_ms);
    field_dwell.on_change = [this](int32_t v) { parent_app.hop_dwell_ms = v; };

    button_load_file.on_select = [this](Button&) {
        auto* view = parent_app.get_nav().push<FileLoadView>(".TXT");
        if (view) {
            view->on_changed = [this](std::filesystem::path path) {
                parent_app.frequency_list.clear();
                parent_app.current_freq_index = 0;
                FreqmanDB db{};
                db.open(path);
                for (const auto& entry : db) {
                    if (entry.frequency_a > 0)
                        parent_app.frequency_list.push_back(entry.frequency_a);
                }
                parent_app.freqman_file = path.stem().string();

                // Switch to HOP mode if a file was loaded
                if (!parent_app.frequency_list.empty())
                    parent_app.freq_hop_mode = true;

                update_list_count();

                // Update the mode display in the Config tab
                if (parent_app.get_config_view()) {
                    parent_app.get_config_view()->update_mode_display();
                }
            };
        }
    };

    button_clear.on_select = [this](Button&) {
        parent_app.frequency_list.clear();
        parent_app.current_freq_index = 0;
        parent_app.freq_hop_mode = false;

        update_list_count();

        // Update the mode display in the Config tab
        if (parent_app.get_config_view()) {
            parent_app.get_config_view()->update_mode_display();
        }
    };
}

void HunterFreqsView::update_list_count() {
    text_loaded_info.set("Loaded: " +
                         to_string_dec_uint(parent_app.frequency_list.size()) + " freqs");
}

void HunterFreqsView::focus() {
    button_load_file.focus();
}

// --- TAB 3: Config View ---
HunterConfigView::HunterConfigView(Rect parent_rect, SignalHunterAppView& parent)
    : View(parent_rect), parent_app(parent), field_single_freq{{UI_POS_X(2), UI_POS_Y(4)}, parent.get_nav()} {
    add_children({&button_mode,
                  &field_single_freq,
                  &labels,
                  &field_threshold,
                  &field_hang_time,
                  &text_info_config});

    button_mode.on_select = [this](Button&) {
        parent_app.freq_hop_mode = !parent_app.freq_hop_mode;
        update_mode_display();
    };

    field_single_freq.set_value(parent_app.single_frequency);

    field_threshold.set_value(parent_app.energy_threshold);
    field_threshold.on_change = [this](int32_t v) { parent_app.energy_threshold = v; };

    field_hang_time.set_value(parent_app.hangtime_ms);
    field_hang_time.on_change = [this](int32_t v) { parent_app.hangtime_ms = v; };

    update_mode_display();
}

void HunterConfigView::on_show() {
    auto current = receiver_model.target_frequency();
    parent_app.single_frequency = current;
    field_single_freq.set_value(current);
}

void HunterConfigView::update_mode_display() {
    if (parent_app.freq_hop_mode)
        button_mode.set_text("MODE: HOP (" +
                             to_string_dec_uint(parent_app.frequency_list.size()) + " freqs)");
    else
        button_mode.set_text("MODE: SINGLE");
}

void HunterConfigView::focus() {
    field_threshold.focus();
}

// --- Main App View ---
SignalHunterAppView::SignalHunterAppView(ui::NavigationView& nav)
    : frequency_list{},
      nav_(nav) {
    baseband::run_prepared_image(portapack::memory::map::m4_code.base());

    receiver_model.set_target_frequency(433920000);
    receiver_model.set_baseband_bandwidth(1750000);
    receiver_model.set_sampling_rate(2000000);
    receiver_model.enable();

    view_main = std::make_unique<HunterMainView>(content_rect, *this);
    view_freqs = std::make_unique<HunterFreqsView>(content_rect, *this);
    view_config = std::make_unique<HunterConfigView>(content_rect, *this);

    tab_view = std::make_unique<TabView>(std::initializer_list<TabView::TabDef>{
        {"Target", Color::cyan(), view_main.get()},
        {"Freqs", Color::green(), view_freqs.get()},
        {"Config", Color::yellow(), view_config.get()}});
    tab_view->set_parent_rect(view_rect);

    add_children({&field_lna,
                  &field_vga,
                  &field_rf_amp,
                  &rssi,
                  tab_view.get(),
                  view_main.get(),
                  view_freqs.get(),
                  view_config.get()});

    view_freqs->hidden(true);
    view_config->hidden(true);

    using namespace app_settings;
    settings_ = std::make_unique<SettingsManager>(
        "ext_signal_hunter", Mode::RX,
        SettingBindings{
            {"threshold"sv, &energy_threshold},
            {"hangtime_ms"sv, &hangtime_ms},
            {"hop_dwell_ms"sv, &hop_dwell_ms},
            {"single_freq"sv, &single_frequency},
            {"file"sv, &freqman_file}});
}

SignalHunterAppView::~SignalHunterAppView() {
    if (!freq_hop_mode) {
        single_frequency = receiver_model.target_frequency();
    }

    // Safely shutdown recording
    stop_recording();
    receiver_model.disable();
    baseband::shutdown();
}

void SignalHunterAppView::focus() {
    if (tab_view)
        tab_view->focus();
}

// TIMER LOGIC: Based on display refresh rate
void SignalHunterAppView::on_frame_sync() {
    // Timer running only if, HUNTING, in HOPPING mode, not recording + we have the freqs list
    if (is_hunting && freq_hop_mode && !capture_thread && !frequency_list.empty()) {
        hop_timer_ms += 17;  // 1 frame in a 60 Hz display ~ 16.6 ms

        // if Dwell Time has passed, change frequency
        if (hop_timer_ms >= hop_dwell_ms) {
            hop_timer_ms = 0;

            current_freq_index = (current_freq_index + 1) % frequency_list.size();
            auto next_freq = frequency_list[current_freq_index];
            receiver_model.set_target_frequency(next_freq);

            if (view_main) {
                view_main->update_frequency(next_freq);
            }
        }
    } else {
        hop_timer_ms = 0;  // if not hunting / recording -> timer = 0
    }
}

// Recording control logic via M0 <-> M4 IPC
void SignalHunterAppView::send_hunter_config(bool start) {
    // Send config to baseband processor via IPC mechanism
    baseband::set_hunter_config(energy_threshold, hangtime_ms, start);

    if (!start) {
        stop_recording();
    }
}

void SignalHunterAppView::on_hunter_trigger(const HunterTriggerMessage* /*message*/) {
    // M4 has detected signal above threshold - start I/Q capture
    if (!capture_thread && is_hunting) {
        trigger_hits++;
        view_main->update_hits(trigger_hits);
        view_main->update_status("RECORDING!", Theme::getInstance()->fg_red);
        view_main->set_recording_state(true);

        start_recording();
    }
}

void SignalHunterAppView::on_hunter_stop(const HunterStopMessage*) {
    if (capture_thread) {
        stop_recording();
        view_main->set_recording_state(false);

        if (is_hunting) {
            if (freq_hop_mode && !frequency_list.empty()) {
                // Next frequency
                current_freq_index = (current_freq_index + 1) % frequency_list.size();
                auto next_freq = frequency_list[current_freq_index];
                receiver_model.set_target_frequency(next_freq);
                view_main->update_frequency(next_freq);

                // Reset the timer
                hop_timer_ms = 0;
            }
            view_main->update_status("HUNTING...", Theme::getInstance()->fg_green);
        }
    }
}

void SignalHunterAppView::start_recording() {
    rtc::RTC datetime{};
    rtc_time::now(datetime);

    // Generate timestamped filename for capture session
    const std::string capture_filename = "HNT_" +
                                         to_string_dec_uint(datetime.year(), 4, '0') +
                                         to_string_dec_uint(datetime.month(), 2, '0') +
                                         to_string_dec_uint(datetime.day(), 2, '0') + "T" +
                                         to_string_dec_uint(datetime.hour(), 2, '0') +
                                         to_string_dec_uint(datetime.minute(), 2, '0') +
                                         to_string_dec_uint(datetime.second(), 2, '0');

    // Use FileWriter for direct raw I/Q dump to avoid M0 CPU bottlenecks.
    // FileConvertWriter is heavier
    auto writer = std::make_unique<FileWriter>();
    auto create_error = writer->create(std::filesystem::path(u"/CAPTURES") / (capture_filename + ".C16"));

    if (create_error.is_valid()) {
        if (view_main) view_main->update_status("SD ERROR", Theme::getInstance()->fg_red);

        send_hunter_config(false);
        is_hunting = false;
        if (view_main) view_main->set_start_button_text("START");

        return;
    }

    current_capture_filename = capture_filename;

    capture_thread = std::make_unique<CaptureThread>(
        std::move(writer),
        4096,
        4,
        []() {},
        [](File::Error) {});

    if (view_main) view_main->update_status("RECORDING!", Theme::getInstance()->fg_red);
}

void SignalHunterAppView::stop_recording() {
    // Stop I/Q capture and release SD card for metadata write
    capture_thread.reset();

    // Generate metadata file after capture completes
    if (!current_capture_filename.empty()) {
        File txt_file;
        auto txt_error = txt_file.create(std::filesystem::path(u"/CAPTURES") / (current_capture_filename + ".TXT"));

        // is_valid() returns true only if error is present (counter-intuitive)
        if (!txt_error.is_valid()) {
            std::string metadata =
                "center_frequency=" + to_string_dec_uint(receiver_model.target_frequency()) + "\n" +
                // Divide main sample rate by decimation factor (8) to match C16 sample rate
                "sample_rate=" + to_string_dec_uint(receiver_model.sampling_rate() / 8) + "\n";

            txt_file.write(metadata.c_str(), metadata.length());
        }
        // Clear filename for next capture cycle
        current_capture_filename.clear();
    }
}

}  // namespace ui::external_app::signal_hunter