mirror of
https://github.com/portapack-mayhem/mayhem-firmware.git
synced 2024-12-15 04:28:10 +00:00
ef03f020ce
* Add IQ Trim app WIP * Add 'trim' checkbox to capture * Implement trimming * Finish auto-trim * Stray //
118 lines
3.3 KiB
C++
118 lines
3.3 KiB
C++
/*
|
|
* 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
|