IQ Trim app and Capture auto-trim (#1456)

* Add IQ Trim app WIP * Add 'trim' checkbox to capture * Implement trimming * Finish auto-trim * Stray //
2025-08-12 00:20:03 -04:00 · 2023-09-21 08:43:10 -07:00 · 2023-09-21 08:43:10 -07:00 · ef03f020ce
commit ef03f020ce
parent 070eb4003e
11 changed files with 537 additions and 51 deletions
--- a/firmware/application/apps/ui_iq_trim.cpp
+++ b/firmware/application/apps/ui_iq_trim.cpp
@ -0,0 +1,118 @@
+/*
+ * Copyright (C) 2023 Kyle Reed
+ *
+ * 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_iq_trim.hpp"
+
+#include "complex.hpp"
+#include "portapack.hpp"
+#include "ui_fileman.hpp"
+
+using namespace portapack;
+namespace fs = std::filesystem;
+
+namespace ui {
+
+IQTrimView::IQTrimView(NavigationView& nav) {
+    add_children({
+        &labels,
+        &field_path,
+        &text_range,
+        &button_trim,
+    });
+
+    field_path.on_select = [this, &nav](TextField&) {
+        auto open_view = nav.push<FileLoadView>(".C*");
+        open_view->push_dir(u"CAPTURES");
+        open_view->on_changed = [this](fs::path path) {
+            read_capture(path);
+            path_ = std::move(path);
+            refresh_ui();
+        };
+    };
+
+    button_trim.on_select = [this, &nav](Button&) {
+        if (!path_.empty() && trim_range_.file_size > 0) {
+            progress_ui.show_trimming();
+            TrimFile(path_, trim_range_);
+            read_capture(path_);
+            refresh_ui();
+        } else {
+            nav.display_modal("Error", "Select a file first.");
+        }
+    };
+}
+
+void IQTrimView::paint(Painter& painter) {
+    if (!path_.empty()) {
+        // Draw power buckets.
+        for (size_t i = 0; i < power_buckets_.size(); ++i) {
+            auto amp = power_buckets_[i].power;
+            painter.draw_vline(
+                pos_lines + Point{(int)i, 0},
+                height_lines,
+                Color(amp, amp, amp));
+        }
+
+        // Draw trim range edges.
+        int start_x = screen_width * trim_range_.start / trim_range_.file_size;
+        int end_x = screen_width * trim_range_.end / trim_range_.file_size;
+
+        painter.draw_vline(
+            pos_lines + Point{start_x, 0},
+            height_lines,
+            Color::dark_green());
+        painter.draw_vline(
+            pos_lines + Point{end_x, 0},
+            height_lines,
+            Color::dark_red());
+    }
+}
+
+void IQTrimView::focus() {
+    field_path.focus();
+}
+
+void IQTrimView::refresh_ui() {
+    field_path.set_text(path_.filename().string());
+    text_range.set(to_string_dec_uint(trim_range_.start) + "-" + to_string_dec_uint(trim_range_.end));
+    set_dirty();
+}
+
+bool IQTrimView::read_capture(const fs::path& path) {
+    power_buckets_ = {};
+    PowerBuckets buckets{
+        .p = power_buckets_.data(),
+        .size = power_buckets_.size()};
+
+    progress_ui.show_reading();
+    auto range = ComputeTrimRange(path, amp_threshold, &buckets, progress_ui.get_callback());
+    progress_ui.clear();
+
+    if (range) {
+        trim_range_ = *range;
+        return true;
+    } else {
+        trim_range_ = {};
+        return false;
+    }
+}
+
+}  // namespace ui