mirror of
https://github.com/SatDump/SatDump
synced 2026-08-13 17:47:30 -04:00
383 lines
No EOL
15 KiB
C++
383 lines
No EOL
15 KiB
C++
#include "pipeline_selector.h"
|
|
|
|
#include "common/detect_header.h"
|
|
#include "common/widgets/json_editor.h"
|
|
#include "core/config.h"
|
|
#include "imgui/imgui.h"
|
|
#include "imgui/imgui_stdlib.h"
|
|
#include "logger.h"
|
|
#include "utils/string.h"
|
|
|
|
#include <algorithm>
|
|
|
|
namespace satdump
|
|
{
|
|
PipelineUISelector::PipelineUISelector(bool live_mode) : live_mode(live_mode)
|
|
{
|
|
nlohmann::ordered_json params = satdump_cfg.main_cfg["user_interface"]["default_offline_parameters"];
|
|
advanced_mode = satdump::satdump_cfg.main_cfg["user_interface"]["advanced_mode"]["value"].get_ptr<nlohmann::json::boolean_t *>();
|
|
|
|
for (nlohmann::detail::iteration_proxy_value<nlohmann::detail::iter_impl<nlohmann::ordered_json>> cfg : params.items())
|
|
if (!cfg.value().contains("no_live") || !live_mode)
|
|
parameters_ui.push_back({cfg.key(), satdump::params::EditableParameter(nlohmann::json(cfg.value()))});
|
|
|
|
if (satdump_cfg.main_cfg.contains("user"))
|
|
{
|
|
if (satdump_cfg.main_cfg["user"].contains("favourite_pipelines"))
|
|
{
|
|
try
|
|
{
|
|
for (std::string pipeline_s : satdump_cfg.main_cfg["user"]["favourite_pipelines"].get<std::vector<std::string>>())
|
|
{
|
|
for (int i = 0; i < (int)pipeline::pipelines.size(); i++)
|
|
if (pipeline::pipelines[i].id == pipeline_s)
|
|
favourite.push_back(i);
|
|
}
|
|
}
|
|
catch (std::exception &)
|
|
{
|
|
}
|
|
}
|
|
}
|
|
|
|
selected_pipeline = pipeline::pipelines[pipelines_levels_select_id];
|
|
updateSelectedPipeline();
|
|
}
|
|
|
|
bool PipelineUISelector::contains(std::vector<int> tm, int n)
|
|
{
|
|
for (unsigned int i = 0; i < tm.size(); i++)
|
|
{
|
|
if (tm[i] == n)
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
void PipelineUISelector::getParamsFromInput()
|
|
{
|
|
std::string file_path = inputfileselect.getPath();
|
|
|
|
if (std::filesystem::exists(file_path) && !std::filesystem::is_directory(file_path))
|
|
{
|
|
nlohmann::json hdr;
|
|
try_get_params_from_input_file(hdr, file_path);
|
|
for (auto &v : hdr.items())
|
|
try_set_param(v.key(), v.value());
|
|
}
|
|
}
|
|
|
|
void PipelineUISelector::try_set_param(std::string name, nlohmann::json v)
|
|
{
|
|
for (std::pair<std::string, satdump::params::EditableParameter> &p : parameters_ui)
|
|
if (p.first == name)
|
|
p.second.setValue(v);
|
|
|
|
for (std::pair<std::string, satdump::params::EditableParameter> &p : parameters_ui_pipeline)
|
|
if (p.first == name)
|
|
p.second.setValue(v);
|
|
}
|
|
|
|
void PipelineUISelector::updateSelectedPipeline()
|
|
{
|
|
parameters_ui_pipeline.clear();
|
|
for (nlohmann::detail::iteration_proxy_value<nlohmann::detail::iter_impl<nlohmann::json>> cfg : selected_pipeline.editable_parameters.items())
|
|
{
|
|
auto it = std::find_if(parameters_ui.begin(), parameters_ui.end(), [&cfg](const std::pair<std::string, satdump::params::EditableParameter> &e) { return e.first == cfg.key(); });
|
|
|
|
if (live_mode)
|
|
{
|
|
if (it == parameters_ui.end() && cfg.value().contains("type"))
|
|
parameters_ui_pipeline.push_back({cfg.key(), satdump::params::EditableParameter(cfg.value())});
|
|
else if (!cfg.value().contains("type") && it != parameters_ui.end())
|
|
it->second.setValue(cfg.value()["value"]);
|
|
}
|
|
else
|
|
{
|
|
if (it == parameters_ui.end())
|
|
parameters_ui_pipeline.push_back({cfg.key(), satdump::params::EditableParameter(cfg.value())});
|
|
else
|
|
it->second.setValue(cfg.value()["value"]);
|
|
}
|
|
}
|
|
|
|
if (!live_mode)
|
|
{
|
|
pipeline_levels_str = "";
|
|
if (selected_pipeline.id != "")
|
|
for (int i = 0; i < (int)selected_pipeline.steps.size() - 1; i++)
|
|
pipeline_levels_str += selected_pipeline.steps[i].level + '\0';
|
|
|
|
if (selected_pipeline.editable_parameters.size() != 0)
|
|
getParamsFromInput();
|
|
}
|
|
}
|
|
|
|
void PipelineUISelector::renderSelectionBox(double width)
|
|
{
|
|
pipeline_mtx.lock();
|
|
if (width != -1)
|
|
ImGui::SetNextItemWidth(width);
|
|
ImGui::InputTextWithHint("##pipelinesearchbox", u8"\uf422 Search pipelines", &pipeline_search_in);
|
|
if (width != -1)
|
|
ImGui::SetNextItemWidth(width);
|
|
|
|
if (satdump_cfg.main_cfg.contains("user"))
|
|
{
|
|
if (satdump_cfg.main_cfg["user"].contains("favourite_pipelines"))
|
|
{
|
|
if (satdump_cfg.main_cfg["user"]["favourite_pipelines"].size() != favourite.size())
|
|
{
|
|
favourite.clear();
|
|
for (std::string pipeline_s : satdump_cfg.main_cfg["user"]["favourite_pipelines"].get<std::vector<std::string>>())
|
|
{
|
|
for (int i = 0; i < (int)pipeline::pipelines.size(); i++)
|
|
if (pipeline::pipelines[i].id == pipeline_s)
|
|
favourite.push_back(i);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
if (ImGui::BeginListBox("##pipelineslistbox"))
|
|
{
|
|
ImVec4 color = {0.73, 0.6, 0.15, 1.0};
|
|
|
|
bool show = !live_mode;
|
|
if (live_mode)
|
|
{
|
|
for (int p : favourite)
|
|
if (pipeline::pipelines[p].live)
|
|
{
|
|
show = true;
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (!favourite.empty() && show)
|
|
{
|
|
if (ImGui::CollapsingHeader("Favourites", ImGuiTreeNodeFlags_DefaultOpen))
|
|
{
|
|
ImGui::Spacing();
|
|
for (int k = 0; k < (int)favourite.size(); k++)
|
|
{
|
|
int n = favourite[k];
|
|
bool show = true;
|
|
if (pipeline_search_in.size() != 0)
|
|
show = satdump::isStringPresent(pipeline::pipelines[n].name, pipeline_search_in);
|
|
|
|
if (show && (!live_mode || pipeline::pipelines[n].live))
|
|
{
|
|
bool is_selected = (selected_pipeline.id == pipeline::pipelines[n].id);
|
|
ImGui::Selectable((pipeline::pipelines[n].name + "##fav").c_str(), &is_selected);
|
|
if (ImGui::IsItemHovered())
|
|
{
|
|
int pos = ImGui::GetItemRectSize().x - 25;
|
|
ImGui::SameLine(pos);
|
|
ImGui::TextColored({0, 0, 0, 0}, "%s", text.c_str());
|
|
|
|
if (is_selected != (selected_pipeline.id == pipeline::pipelines[n].id) && ImGui::IsItemHovered(ImGuiHoveredFlags_AllowWhenOverlapped))
|
|
{
|
|
favourite.erase(favourite.begin() + k);
|
|
satdump_cfg.main_cfg["user"]["favourite_pipelines"].erase(k);
|
|
continue;
|
|
}
|
|
|
|
ImGui::SameLine(pos);
|
|
text = u8"\uf005";
|
|
ImGui::TextColored(color, "%s", text.c_str());
|
|
text = u8"\uf006";
|
|
if (is_selected != (selected_pipeline.id == pipeline::pipelines[n].id))
|
|
{
|
|
selected_pipeline = pipeline::pipelines[n];
|
|
updateSelectedPipeline();
|
|
}
|
|
}
|
|
if (is_selected)
|
|
{
|
|
// selected_pipeline = pipelines[n];
|
|
// updateSelectedPipeline();
|
|
ImGui::SetItemDefaultFocus();
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
ImGui::Spacing();
|
|
ImGui::Separator();
|
|
ImGui::Spacing();
|
|
ImGui::Spacing();
|
|
}
|
|
|
|
for (int n = 0; n < (int)pipeline::pipelines.size(); n++)
|
|
{
|
|
bool show = true;
|
|
if (pipeline_search_in.size() != 0)
|
|
show = satdump::isStringPresent(pipeline::pipelines[n].name, pipeline_search_in);
|
|
|
|
if (show && (!live_mode || pipeline::pipelines[n].live))
|
|
{
|
|
bool is_selected = (selected_pipeline.id == pipeline::pipelines[n].id);
|
|
bool isfav = contains(favourite, n);
|
|
ImGui::Selectable(pipeline::pipelines[n].name.c_str(), &is_selected);
|
|
if (ImGui::IsItemHovered() || isfav)
|
|
{
|
|
int pos = ImGui::GetItemRectSize().x - 25;
|
|
ImGui::SameLine(pos);
|
|
ImGui::TextColored({0, 0, 0, 0}, "%s", text.c_str());
|
|
|
|
if (is_selected != (selected_pipeline.id == pipeline::pipelines[n].id) && ImGui::IsItemHovered(ImGuiHoveredFlags_AllowWhenOverlapped))
|
|
{
|
|
if (isfav)
|
|
{
|
|
for (int i = 0; i < (int)favourite.size(); i++)
|
|
{
|
|
if (favourite[i] == n)
|
|
{
|
|
favourite.erase(favourite.begin() + i);
|
|
satdump_cfg.main_cfg["user"]["favourite_pipelines"].erase(i);
|
|
isfav = false;
|
|
break;
|
|
}
|
|
}
|
|
continue;
|
|
}
|
|
else
|
|
{
|
|
favourite.push_back(n);
|
|
satdump_cfg.main_cfg["user"]["favourite_pipelines"].push_back(pipeline::pipelines[n].id);
|
|
isfav = true;
|
|
is_selected = !is_selected;
|
|
}
|
|
}
|
|
|
|
ImGui::SameLine(pos);
|
|
if (isfav)
|
|
text = u8"\uf005";
|
|
ImGui::TextColored(color, "%s", text.c_str());
|
|
text = u8"\uf006";
|
|
if (is_selected != (selected_pipeline.id == pipeline::pipelines[n].id))
|
|
{
|
|
selected_pipeline = pipeline::pipelines[n];
|
|
updateSelectedPipeline();
|
|
}
|
|
}
|
|
if (is_selected)
|
|
{
|
|
ImGui::SetItemDefaultFocus();
|
|
}
|
|
}
|
|
}
|
|
ImGui::EndListBox();
|
|
}
|
|
pipeline_mtx.unlock();
|
|
}
|
|
|
|
void PipelineUISelector::drawMainparams()
|
|
{
|
|
pipeline_mtx.lock();
|
|
if (ImGui::BeginTable("##pipelinesmainoptions", 2))
|
|
{
|
|
int label_width = ImGui::CalcTextSize("Output Directory").x;
|
|
ImGui::TableSetupColumn("##pipelinesmaincolumn1", ImGuiTableColumnFlags_WidthFixed, label_width);
|
|
ImGui::TableSetupColumn("##pipelinesmaincolumn2", ImGuiTableColumnFlags_WidthStretch, label_width);
|
|
|
|
ImGui::TableNextRow();
|
|
ImGui::TableSetColumnIndex(0);
|
|
ImGui::Text("Input File");
|
|
ImGui::TableSetColumnIndex(1);
|
|
if (inputfileselect.draw())
|
|
getParamsFromInput();
|
|
|
|
ImGui::TableNextRow();
|
|
ImGui::TableSetColumnIndex(0);
|
|
ImGui::Text("Output Directory");
|
|
ImGui::TableSetColumnIndex(1);
|
|
outputdirselect.draw();
|
|
|
|
ImGui::TableNextRow();
|
|
ImGui::TableSetColumnIndex(0);
|
|
ImGui::Text("Input Level");
|
|
ImGui::TableSetColumnIndex(1);
|
|
ImGui::Combo("##pipelinelevel", &pipelines_levels_select_id, pipeline_levels_str.c_str());
|
|
ImGui::EndTable();
|
|
}
|
|
|
|
pipeline_mtx.unlock();
|
|
}
|
|
|
|
void PipelineUISelector::drawMainparamsLive()
|
|
{
|
|
ImGui::Text("Output Directory :");
|
|
outputdirselect.draw();
|
|
ImGui::Spacing();
|
|
}
|
|
|
|
void PipelineUISelector::renderParamTable()
|
|
{
|
|
if (ImGui::BeginTable("##pipelineoptions", 2, ImGuiTableFlags_Borders | ImGuiTableFlags_RowBg))
|
|
{
|
|
// TODOREWORKUI?
|
|
int label_width = ImGui::CalcTextSize("Output Directory").x;
|
|
ImGui::TableSetupColumn("##pipelinesmaincolumn1", ImGuiTableColumnFlags_WidthStretch, label_width);
|
|
ImGui::TableSetupColumn("##pipelinesmaincolumn2", ImGuiTableColumnFlags_WidthStretch, label_width);
|
|
|
|
for (std::pair<std::string, satdump::params::EditableParameter> &p : parameters_ui)
|
|
p.second.draw();
|
|
for (std::pair<std::string, satdump::params::EditableParameter> &p : parameters_ui_pipeline)
|
|
p.second.draw();
|
|
ImGui::EndTable();
|
|
}
|
|
|
|
if (*advanced_mode)
|
|
{
|
|
ImGui::SetCursorPosY(ImGui::GetCursorPosY() + 5 * ui_scale);
|
|
ImGui::SeparatorText("Advanced Parameters");
|
|
ImGui::SetCursorPosY(ImGui::GetCursorPosY() - 5 * ui_scale);
|
|
pipeline_mtx.lock();
|
|
for (auto &step : selected_pipeline.steps)
|
|
if (widgets::JSONTableEditor(step.parameters, step.module.c_str()))
|
|
step.parameters = pipeline::pipelines_json[selected_pipeline.id]["work"][step.level][step.module];
|
|
|
|
pipeline_mtx.unlock();
|
|
ImGui::SetCursorPosY(ImGui::GetCursorPosY() + 5 * ui_scale);
|
|
}
|
|
}
|
|
|
|
nlohmann::json PipelineUISelector::getParameters()
|
|
{
|
|
nlohmann::json params2;
|
|
for (std::pair<std::string, satdump::params::EditableParameter> &p : parameters_ui)
|
|
params2[p.first] = p.second.getValue();
|
|
for (std::pair<std::string, satdump::params::EditableParameter> &p : parameters_ui_pipeline)
|
|
params2[p.first] = p.second.getValue();
|
|
return params2;
|
|
}
|
|
|
|
void PipelineUISelector::setParameters(nlohmann::json params)
|
|
{
|
|
pipeline_mtx.lock();
|
|
for (auto &el : params.items())
|
|
try_set_param(el.key(), el.value());
|
|
pipeline_mtx.unlock();
|
|
}
|
|
|
|
void PipelineUISelector::select_pipeline(std::string id)
|
|
{
|
|
pipeline_mtx.lock();
|
|
bool found = false;
|
|
for (int n = 0; n < (int)pipeline::pipelines.size(); n++)
|
|
{
|
|
if (id == pipeline::pipelines[n].id)
|
|
{
|
|
selected_pipeline = pipeline::pipelines[n];
|
|
found = true;
|
|
}
|
|
}
|
|
if (found)
|
|
updateSelectedPipeline();
|
|
else
|
|
logger->error("Could not find pipeline %s!", id.c_str());
|
|
pipeline_mtx.unlock();
|
|
}
|
|
} // namespace satdump
|