mirror of
https://github.com/SatDump/SatDump
synced 2026-08-13 17:47:30 -04:00
428 lines
19 KiB
C++
428 lines
19 KiB
C++
#include "recorder.h"
|
|
#include "core/config.h"
|
|
#include "common/utils.h"
|
|
#include "core/style.h"
|
|
#include "logger.h"
|
|
#include "imgui/imgui_stdlib.h"
|
|
#include "core/pipeline.h"
|
|
|
|
#include "main_ui.h"
|
|
|
|
#include "imgui/imgui_internal.h"
|
|
|
|
#include "processing.h"
|
|
|
|
namespace satdump
|
|
{
|
|
RecorderApplication::RecorderApplication()
|
|
: Application("recorder"), pipeline_selector(true)
|
|
{
|
|
dsp::registerAllSources();
|
|
|
|
if (config::main_cfg["user"].contains("recorder_state"))
|
|
deserialize_config(config::main_cfg["user"]["recorder_state"]);
|
|
|
|
automated_live_output_dir = config::main_cfg["satdump_output_directories"]["live_processing_autogen"]["value"].get<bool>();
|
|
|
|
sources = dsp::getAllAvailableSources();
|
|
|
|
for (dsp::SourceDescriptor src : sources)
|
|
{
|
|
logger->debug("Device " + src.name);
|
|
sdr_select_string += src.name + '\0';
|
|
}
|
|
|
|
source_ptr = dsp::getSourceFromDescriptor(sources[0]);
|
|
source_ptr->open();
|
|
source_ptr->set_frequency(100e6);
|
|
|
|
splitter = std::make_shared<dsp::SplitterBlock>(source_ptr->output_stream);
|
|
splitter->set_output_2nd(false);
|
|
splitter->set_output_3rd(false);
|
|
|
|
fft = std::make_shared<dsp::FFTBlock>(splitter->output_stream);
|
|
fft->set_fft_settings(fft_size);
|
|
fft->start();
|
|
|
|
file_sink = std::make_shared<dsp::FileSinkBlock>(splitter->output_stream_2);
|
|
file_sink->set_output_sample_type(dsp::CF_32);
|
|
file_sink->start();
|
|
|
|
fft_plot = std::make_shared<widgets::FFTPlot>(fft->output_stream->writeBuf, fft_size, -10, 20, 10);
|
|
waterfall_plot = std::make_shared<widgets::WaterfallPlot>(fft->output_stream->writeBuf, fft_size, 2000);
|
|
}
|
|
|
|
RecorderApplication::~RecorderApplication()
|
|
{
|
|
source_ptr->stop();
|
|
source_ptr->close();
|
|
splitter->stop();
|
|
fft->stop();
|
|
file_sink->stop();
|
|
}
|
|
|
|
void RecorderApplication::drawUI()
|
|
{
|
|
ImVec2 recorder_size = ImGui::GetContentRegionAvail();
|
|
// recorder_size.y -= ImGui::GetCursorPosY();
|
|
|
|
ImGui::BeginGroup();
|
|
|
|
float wf_size = recorder_size.y - (is_processing ? 240 * ui_scale : 0);
|
|
ImGui::BeginChild("RecorderChildPanel", {float(recorder_size.x * panel_ratio), wf_size}, false);
|
|
{
|
|
if (ImGui::CollapsingHeader("Device"))
|
|
{
|
|
if (is_started)
|
|
style::beginDisabled();
|
|
if (ImGui::Combo("Source", &sdr_select_id, sdr_select_string.c_str()))
|
|
{
|
|
source_ptr = getSourceFromDescriptor(sources[sdr_select_id]);
|
|
source_ptr->open();
|
|
source_ptr->set_frequency(100e6);
|
|
}
|
|
if (ImGui::Button("Refresh"))
|
|
{
|
|
sources = dsp::getAllAvailableSources();
|
|
|
|
sdr_select_string.clear();
|
|
for (dsp::SourceDescriptor src : sources)
|
|
{
|
|
logger->debug("Device " + src.name);
|
|
sdr_select_string += src.name + '\0';
|
|
}
|
|
|
|
while (sdr_select_id >= (int)sources.size())
|
|
sdr_select_id--;
|
|
|
|
source_ptr = getSourceFromDescriptor(sources[sdr_select_id]);
|
|
source_ptr->open();
|
|
source_ptr->set_frequency(100e6);
|
|
}
|
|
if (is_started)
|
|
style::endDisabled();
|
|
|
|
if (ImGui::InputDouble("MHz", &frequency_mhz))
|
|
source_ptr->set_frequency(frequency_mhz * 1e6);
|
|
|
|
source_ptr->drawControlUI();
|
|
|
|
if (!is_started)
|
|
{
|
|
if (ImGui::Button("Start"))
|
|
{
|
|
source_ptr->start();
|
|
splitter->input_stream = source_ptr->output_stream;
|
|
splitter->start();
|
|
is_started = true;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
if (ImGui::Button("Stop"))
|
|
{
|
|
splitter->stop_tmp();
|
|
source_ptr->stop();
|
|
is_started = false;
|
|
}
|
|
}
|
|
}
|
|
|
|
if (ImGui::CollapsingHeader("FFT"))
|
|
{
|
|
ImGui::SliderFloat("FFT Max", &fft_plot->scale_max, -80, 80);
|
|
ImGui::SliderFloat("FFT Min", &fft_plot->scale_min, -80, 80);
|
|
ImGui::SliderFloat("Avg Rate", &fft->avg_rate, 0.01, 0.99);
|
|
ImGui::Checkbox("Show Waterfall", &show_waterfall);
|
|
}
|
|
|
|
if (fft_plot->scale_max < fft_plot->scale_min)
|
|
{
|
|
fft_plot->scale_min = waterfall_plot->scale_min;
|
|
fft_plot->scale_max = waterfall_plot->scale_max;
|
|
}
|
|
else if (fft_plot->scale_min > fft_plot->scale_max)
|
|
{
|
|
fft_plot->scale_min = waterfall_plot->scale_min;
|
|
fft_plot->scale_max = waterfall_plot->scale_max;
|
|
}
|
|
else
|
|
{
|
|
waterfall_plot->scale_min = fft_plot->scale_min;
|
|
waterfall_plot->scale_max = fft_plot->scale_max;
|
|
}
|
|
|
|
if (ImGui::CollapsingHeader("Processing"))
|
|
{
|
|
if (!is_started)
|
|
style::beginDisabled();
|
|
if (is_processing)
|
|
style::beginDisabled();
|
|
pipeline_selector.renderSelectionBox(ImGui::GetContentRegionAvail().x);
|
|
if (!automated_live_output_dir)
|
|
pipeline_selector.drawMainparamsLive();
|
|
pipeline_selector.renderParamTable();
|
|
if (is_processing)
|
|
style::endDisabled();
|
|
|
|
if (!is_processing)
|
|
{
|
|
if (ImGui::Button("Start###startprocessing"))
|
|
{
|
|
if (pipeline_selector.outputdirselect.file_valid || automated_live_output_dir)
|
|
{
|
|
pipeline_params = pipeline_selector.getParameters();
|
|
pipeline_params["samplerate"] = source_ptr->get_samplerate();
|
|
pipeline_params["baseband_format"] = "f32";
|
|
pipeline_params["buffer_size"] = STREAM_BUFFER_SIZE; // This is required, as we WILL go over the (usually) default 8192 size
|
|
|
|
if (automated_live_output_dir)
|
|
{
|
|
const time_t timevalue = time(0);
|
|
std::tm *timeReadable = gmtime(&timevalue);
|
|
std::string timestamp = std::to_string(timeReadable->tm_year + 1900) + "-" +
|
|
(timeReadable->tm_mon + 1 > 9 ? std::to_string(timeReadable->tm_mon + 1) : "0" + std::to_string(timeReadable->tm_mon + 1)) + "-" +
|
|
(timeReadable->tm_mday > 9 ? std::to_string(timeReadable->tm_mday) : "0" + std::to_string(timeReadable->tm_mday)) + "_" +
|
|
(timeReadable->tm_hour > 9 ? std::to_string(timeReadable->tm_hour) : "0" + std::to_string(timeReadable->tm_hour)) + "-" +
|
|
(timeReadable->tm_min > 9 ? std::to_string(timeReadable->tm_min) : "0" + std::to_string(timeReadable->tm_min));
|
|
pipeline_output_dir = config::main_cfg["satdump_output_directories"]["live_processing_path"]["value"].get<std::string>() + "/" +
|
|
timestamp + "_" +
|
|
pipelines[pipeline_selector.pipeline_id].name + "_" +
|
|
std::to_string(long(source_ptr->d_frequency / 1e6)) + "Mhz";
|
|
std::filesystem::create_directories(pipeline_output_dir);
|
|
logger->info("Generated folder name : " + pipeline_output_dir);
|
|
}
|
|
else
|
|
{
|
|
pipeline_output_dir = pipeline_selector.outputdirselect.getPath();
|
|
}
|
|
|
|
live_pipeline = std::make_unique<LivePipeline>(pipelines[pipeline_selector.pipeline_id], pipeline_params, pipeline_output_dir);
|
|
splitter->output_stream_3 = std::make_shared<dsp::stream<complex_t>>();
|
|
live_pipeline->start(splitter->output_stream_3, ui_thread_pool);
|
|
splitter->set_output_3rd(true);
|
|
|
|
is_processing = true;
|
|
}
|
|
else
|
|
{
|
|
error = "Please select a valid output directory!";
|
|
}
|
|
}
|
|
}
|
|
else
|
|
{
|
|
if (ImGui::Button("Stop##stopprocessing"))
|
|
{
|
|
is_processing = false;
|
|
splitter->set_output_3rd(false);
|
|
live_pipeline->stop();
|
|
|
|
if (config::main_cfg["user_interface"]["finish_processing_after_live"]["value"].get<bool>() && live_pipeline->getOutputFiles().size() > 0)
|
|
{
|
|
Pipeline pipeline = pipelines[pipeline_selector.pipeline_id];
|
|
std::string input_file = live_pipeline->getOutputFiles()[0];
|
|
int start_level = pipeline.live_cfg[pipeline.live_cfg.size() - 1].first;
|
|
std::string input_level = pipeline.steps[start_level].level_name;
|
|
ui_thread_pool.push([=](int)
|
|
{ processing::process(pipeline.name, input_level, input_file, pipeline_output_dir, pipeline_params); });
|
|
}
|
|
|
|
live_pipeline.reset();
|
|
}
|
|
}
|
|
|
|
ImGui::SameLine();
|
|
ImGui::TextColored(ImColor(255, 0, 0), "%s", error.c_str());
|
|
|
|
if (!is_started)
|
|
style::endDisabled();
|
|
}
|
|
|
|
if (ImGui::CollapsingHeader("Recording"))
|
|
{
|
|
if (is_recording)
|
|
style::beginDisabled();
|
|
if (ImGui::Combo("Format", &select_sample_format, "f32\0"
|
|
"s16\0"
|
|
"s8\0"
|
|
#ifdef BUILD_ZIQ
|
|
"ziq s8\0"
|
|
"ziq s16\0"
|
|
"ziq f32\0"
|
|
#endif
|
|
))
|
|
{
|
|
if (select_sample_format == 0)
|
|
file_sink->set_output_sample_type(dsp::CF_32);
|
|
else if (select_sample_format == 1)
|
|
file_sink->set_output_sample_type(dsp::IS_16);
|
|
else if (select_sample_format == 2)
|
|
file_sink->set_output_sample_type(dsp::IS_8);
|
|
#ifdef BUILD_ZIQ
|
|
else if (select_sample_format == 3)
|
|
{
|
|
file_sink->set_output_sample_type(dsp::ZIQ);
|
|
ziq_bit_depth = 8;
|
|
}
|
|
else if (select_sample_format == 4)
|
|
{
|
|
file_sink->set_output_sample_type(dsp::ZIQ);
|
|
ziq_bit_depth = 16;
|
|
}
|
|
else if (select_sample_format == 5)
|
|
{
|
|
file_sink->set_output_sample_type(dsp::ZIQ);
|
|
ziq_bit_depth = 32;
|
|
}
|
|
#endif
|
|
}
|
|
if (is_recording)
|
|
style::endDisabled();
|
|
|
|
if (file_sink->get_written() < 1e9)
|
|
ImGui::Text("Size : %.2f MB", file_sink->get_written() / 1e6);
|
|
else
|
|
ImGui::Text("Size : %.2f GB", file_sink->get_written() / 1e9);
|
|
|
|
#ifdef BUILD_ZIQ
|
|
if (select_sample_format == 3 || select_sample_format == 4 || select_sample_format == 5)
|
|
{
|
|
if (file_sink->get_written_raw() < 1e9)
|
|
ImGui::Text("Size (raw) : %.2f MB", file_sink->get_written_raw() / 1e6);
|
|
else
|
|
ImGui::Text("Size (raw) : %.2f GB", file_sink->get_written_raw() / 1e9);
|
|
}
|
|
#endif
|
|
|
|
ImGui::Text("File : %s", recorder_filename.c_str());
|
|
|
|
ImGui::Spacing();
|
|
|
|
if (!is_recording)
|
|
ImGui::TextColored(ImColor(255, 0, 0), "IDLE");
|
|
else
|
|
ImGui::TextColored(ImColor(0, 255, 0), "RECORDING");
|
|
|
|
ImGui::Spacing();
|
|
|
|
if (!is_recording)
|
|
{
|
|
if (ImGui::Button("Start###startrecording"))
|
|
{
|
|
splitter->set_output_2nd(true);
|
|
|
|
const time_t timevalue = time(0);
|
|
std::tm *timeReadable = gmtime(&timevalue);
|
|
std::string timestamp =
|
|
(timeReadable->tm_hour > 9 ? std::to_string(timeReadable->tm_hour) : "0" + std::to_string(timeReadable->tm_hour)) + "-" +
|
|
(timeReadable->tm_min > 9 ? std::to_string(timeReadable->tm_min) : "0" + std::to_string(timeReadable->tm_min)) + "-" +
|
|
(timeReadable->tm_sec > 9 ? std::to_string(timeReadable->tm_sec) : "0" + std::to_string(timeReadable->tm_sec));
|
|
|
|
std::string filename = config::main_cfg["satdump_output_directories"]["recording_path"]["value"].get<std::string>() +
|
|
"/" + timestamp + "_" + std::to_string(source_ptr->get_samplerate()) + "SPS_" +
|
|
std::to_string(long(frequency_mhz * 1e6)) + "Hz";
|
|
|
|
recorder_filename = file_sink->start_recording(filename, source_ptr->get_samplerate(), ziq_bit_depth);
|
|
|
|
logger->info("Recording to " + recorder_filename);
|
|
|
|
is_recording = true;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
if (ImGui::Button("Stop##stoprecording"))
|
|
{
|
|
file_sink->stop_recording();
|
|
splitter->set_output_2nd(false);
|
|
recorder_filename = "";
|
|
is_recording = false;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
ImGui::EndChild();
|
|
ImGui::EndGroup();
|
|
|
|
ImGui::SameLine();
|
|
|
|
ImGui::BeginGroup();
|
|
ImGui::BeginChild("RecorderFFT", {float(recorder_size.x * (1.0 - panel_ratio)), wf_size}, false, ImGuiWindowFlags_NoScrollbar | ImGuiWindowFlags_NoScrollWithMouse);
|
|
{
|
|
float fft_height = wf_size * (show_waterfall ? (1.0 - waterfall_ratio) : 1.0);
|
|
float wf_height = wf_size * waterfall_ratio;
|
|
fft_plot->draw({float(recorder_size.x * (1.0 - panel_ratio) - 4), fft_height});
|
|
if (show_waterfall)
|
|
waterfall_plot->draw({float(recorder_size.x * (1.0 - panel_ratio) - 4), wf_height * 4}, is_started);
|
|
|
|
float offset = 35 * ui_scale;
|
|
|
|
ImVec2 mouse_pos = ImGui::GetMousePos();
|
|
if ((mouse_pos.y > offset + fft_height - 10 * ui_scale &&
|
|
mouse_pos.y < offset + fft_height + 10 * ui_scale &&
|
|
mouse_pos.x > recorder_size.x * 0.20) ||
|
|
dragging_waterfall)
|
|
{
|
|
ImGui::SetMouseCursor(ImGuiMouseCursor_ResizeNS);
|
|
if (ImGui::IsMouseDown(ImGuiMouseButton_Left))
|
|
{
|
|
float new_y = mouse_pos.y - offset;
|
|
float new_ratio = new_y / (fft_height + wf_height);
|
|
if (new_ratio > 0.1 && new_ratio < 0.9)
|
|
waterfall_ratio = 1.0 - new_ratio;
|
|
}
|
|
else
|
|
dragging_waterfall = false;
|
|
if (ImGui::IsMouseClicked(ImGuiMouseButton_Left))
|
|
dragging_waterfall = true;
|
|
}
|
|
else if (ImGui::GetMouseCursor() != ImGuiMouseCursor_ResizeEW)
|
|
ImGui::SetMouseCursor(ImGuiMouseCursor_Arrow);
|
|
}
|
|
ImGui::EndChild();
|
|
ImGui::EndGroup();
|
|
|
|
ImVec2 mouse_pos = ImGui::GetMousePos();
|
|
if ((mouse_pos.x > recorder_size.x * panel_ratio + 15 * ui_scale - 10 * ui_scale &&
|
|
mouse_pos.x < recorder_size.x * panel_ratio + 15 * ui_scale + 10 * ui_scale &&
|
|
mouse_pos.y > 35 * ui_scale) ||
|
|
dragging_panel)
|
|
{
|
|
ImGui::SetMouseCursor(ImGuiMouseCursor_ResizeEW);
|
|
if (ImGui::IsMouseDown(ImGuiMouseButton_Left))
|
|
{
|
|
float new_x = mouse_pos.x;
|
|
float new_ratio = new_x / recorder_size.x;
|
|
if (new_ratio > 0.1 && new_ratio < 0.9)
|
|
panel_ratio = new_ratio;
|
|
}
|
|
else
|
|
dragging_panel = false;
|
|
if (ImGui::IsMouseClicked(ImGuiMouseButton_Left))
|
|
dragging_panel = true;
|
|
}
|
|
else if (ImGui::GetMouseCursor() != ImGuiMouseCursor_ResizeNS)
|
|
ImGui::SetMouseCursor(ImGuiMouseCursor_Arrow);
|
|
|
|
if (is_processing)
|
|
{
|
|
float y_pos = ImGui::GetCursorPosY() + 35 * ui_scale;
|
|
float live_width = recorder_size.x + 16 * ui_scale;
|
|
float live_height = 250;
|
|
float winwidth = live_pipeline->modules.size() > 0 ? live_width / live_pipeline->modules.size() : live_width;
|
|
float currentPos = 0;
|
|
for (std::shared_ptr<ProcessingModule> module : live_pipeline->modules)
|
|
{
|
|
ImGui::SetNextWindowPos({currentPos, y_pos});
|
|
ImGui::SetNextWindowSize({(float)winwidth, (float)live_height});
|
|
module->drawUI(false);
|
|
currentPos += winwidth;
|
|
|
|
// if (ImGui::GetCurrentContext()->last_window != NULL)
|
|
// currentPos += ImGui::GetCurrentContext()->last_window->Size.x;
|
|
// logger->info(ImGui::GetCurrentContext()->last_window->Name);
|
|
}
|
|
}
|
|
}
|
|
};
|