ham-ref/satdump
1
0
Fork 0
mirror of https://github.com/SatDump/SatDump synced 2026-08-13 17:47:30 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions
satdump/src-interface/viewer2/viewer.cpp

319 lines
14 KiB
C++
Raw Blame History

#include "viewer.h"
// #include "image_handler.h"
// #include "radiation_handler.h"
// #include "scatterometer_handler.h"
// #include "core/config.h"
// #include "products/dataset.h"
#include "common/event_bus.h"
#include "common/utils.h"
// #include "resources.h"
#include "core/plugin.h"
#include "core/style.h"
#include "dsp/cyclo_test.h"
#include "dsp/flowgraph/dsp_flowgraph_handler.h"
#include "handlers/dataset/dataset_handler.h"
#include "handlers/dummy_handler.h"
#include "handlers/product/image_product_handler.h" // TODOREWORK CLEAN
#include "handlers/projection/projection_handler.h"
#include "handlers/trash/trash_handler.h"
#include "handlers/vector/shapefile_handler.h"
// TODOREWORK
#include "resources.h"
// TODOREWORK
#include "dsp/newrec.h"
#include "dsp/waterfall_test.h"
#include "common/image/io.h"
#include "imgui/imgui_filedrop.h"
namespace satdump
{
namespace viewer
{
ViewerApplication::ViewerApplication() : Application("viewer")
{
master_handler = std::make_shared<handlers::DummyHandler>("MasterHandlerViewer");
// Add trashcan
trash_handler = std::make_shared<handlers::DummyHandler>("TrashHandlerViewer");
auto trash_h = std::make_shared<handlers::TrashHandler>();
trash_h->setCanBeDragged(false);
trash_handler->addSubHandler(trash_h);
trash_handler->setCanBeDraggedTo(false);
// Enable dropping files onto the viewer. TODOREWORK, check the viewer IS active!?
eventBus->register_handler<imgui_utils::FileDropEvent>(
[this](const imgui_utils::FileDropEvent &v)
{
for (auto &f : v.files)
tryOpenFileInViewer(f);
});
// TODOREWORK. Returns the last selected handler of a specific type if available
eventBus->register_handler<GetLastSelectedOfTypeEvent>(
[this](const GetLastSelectedOfTypeEvent &v)
{
if (last_selected_handler.count(v.type))
v.h = last_selected_handler[v.type];
else
v.h = nullptr;
});
// TODOREWORK remove
tryOpenFileInViewer("/home/alan/Downloads/SatDump_NEWPRODS/metop_test/dataset.json");
}
ViewerApplication::~ViewerApplication()
{
if (file_open_thread.joinable())
file_open_thread.join();
}
void ViewerApplication::drawPanel()
{
ImGui::SetNextItemWidth(ImGui::GetWindowWidth() / 2);
if (ImGui::CollapsingHeader("General", ImGuiTreeNodeFlags_DefaultOpen))
{
auto prev_curr = curr_handler;
if (ImGui::BeginTable("##masterhandlertable", 1, ImGuiTableFlags_RowBg | ImGuiTableFlags_Borders))
{
ImGui::TableSetupColumn("##masterhandlertable_col", ImGuiTableColumnFlags_None);
ImGui::TableNextColumn();
master_handler->drawTreeMenu(curr_handler);
trash_handler->drawTreeMenu(curr_handler);
ImGui::EndTable();
}
if (prev_curr != curr_handler)
last_selected_handler.insert_or_assign(curr_handler->getID(), curr_handler);
if (curr_handler)
curr_handler->drawMenu();
}
}
void ViewerApplication::drawMenuBar()
{
// Handle File Stuff TODOREWORK?
{
if (file_open_dialog.update())
{
std::string prod_path = file_open_dialog.getPath();
if (prod_path.size() > 0)
tryOpenFileInViewer(prod_path);
else
logger->trace("No file selected");
}
}
if (ImGui::BeginMenuBar())
{
// Main "Open" menu, for files, other handlers, etc
if (ImGui::BeginMenu("File"))
{
file_open_dialog.render("Open File", "Open File", "", {{"All Files", "*"}});
// TODOREWORK remove
if (ImGui::BeginMenu("Hardcoded"))
{
if (ImGui::MenuItem("Load KMSS"))
tryOpenFileInViewer("/home/alan/Downloads/SatDump_NEWPRODS/KMSS_24/dataset.json");
if (ImGui::MenuItem("Load Sterna"))
tryOpenFileInViewer("/home/alan/Downloads/SatDump_NEWPRODS/aws_pfm_cadu/dataset.json");
if (ImGui::MenuItem("Load MSUGS"))
tryOpenFileInViewer("/home/alan/Downloads/20241231_132953_ARKTIKA-M 2_dat/MSUGS_VIS1/product.cbor");
if (ImGui::MenuItem("Load MSUGS 2"))
tryOpenFileInViewer("/home/alan/Downloads/SatDump_NEWPRODS/20250104_071415_ELEKTRO-L_3_dat/dataset.json");
if (ImGui::MenuItem("Load MetOp"))
tryOpenFileInViewer("/home/alan/Downloads/SatDump_NEWPRODS/metop_test/dataset.json");
if (ImGui::MenuItem("Load L3"))
tryOpenFileInViewer("/data_ssd/ELEKTRO-L3/20250104_071415_ELEKTRO-L 3.dat_OUT/dataset.json");
if (ImGui::MenuItem("Load JPSS-1"))
tryOpenFileInViewer("/home/alan/Downloads/SatDump_NEWPRODS/202411271228_NOAA_20/dataset.json");
if (ImGui::MenuItem("Load JPSS-2"))
tryOpenFileInViewer("/home/alan/Downloads/SatDump_NEWPRODS/n21_day/dataset.json");
if (ImGui::MenuItem("Load APT"))
tryOpenFileInViewer("/home/alan/Downloads/audio_137912500Hz_11-39-56_04-03-2024_wav/dataset.json");
if (ImGui::MenuItem("Load GOES"))
tryOpenFileInViewer("/home/alan/Downloads/SatDump_NEWPRODS/goes_hrit_jvital2013_cadu/IMAGES/GOES-16/Full Disk/2024-04-17_18-00-20/product.cbor");
if (ImGui::MenuItem("Load Shapefile"))
{
auto shp_h = std::make_shared<handlers::ShapefileHandler>(resources::getResourcePath("maps/ne_10m_admin_0_countries.shp"));
master_handler->addSubHandler(shp_h);
}
if (ImGui::MenuItem("Load Shapefile FRA_1"))
{
auto shp_h = std::make_shared<handlers::ShapefileHandler>("/home/alan/Downloads/gadm41_FRA_shp/gadm41_FRA_1.shp");
master_handler->addSubHandler(shp_h);
}
if (ImGui::MenuItem("Load Shapefile FRA_2"))
{
auto shp_h = std::make_shared<handlers::ShapefileHandler>("/home/alan/Downloads/gadm41_FRA_shp/gadm41_FRA_2.shp");
master_handler->addSubHandler(shp_h);
}
ImGui::EndMenu();
}
ImGui::EndMenu();
}
if (ImGui::BeginMenu("Handler"))
{
if (ImGui::BeginMenu("Add"))
{ // TODOREWORK?
if (ImGui::MenuItem("Projection"))
master_handler->addSubHandler(std::make_shared<handlers::ProjectionHandler>());
if (ImGui::MenuItem("DSP Flowgraph"))
master_handler->addSubHandler(std::make_shared<handlers::DSPFlowGraphHandler>());
if (ImGui::MenuItem("Waterfall TEST"))
master_handler->addSubHandler(std::make_shared<handlers::WaterfallTestHandler>());
if (ImGui::MenuItem("NewRec TEST"))
master_handler->addSubHandler(std::make_shared<handlers::NewRecHandler>());
if (ImGui::MenuItem("CycloHelper TEST"))
master_handler->addSubHandler(std::make_shared<handlers::CycloHelperHandler>());
ImGui::EndMenu();
}
if (curr_handler && ImGui::BeginMenu("Config"))
{
if (ImGui::MenuItem("JSON To Clipboard"))
ImGui::SetClipboardText(curr_handler->getConfig().dump(4).c_str());
if (ImGui::MenuItem("JSON From Clipboard"))
curr_handler->setConfig(nlohmann::json::parse(ImGui::GetClipboardText()));
ImGui::EndMenu();
}
ImGui::EndMenu();
}
if (curr_handler)
{
if (ImGui::BeginMenu("Current"))
{
curr_handler->drawMenuBar();
ImGui::EndMenu();
}
}
eventBus->fire_event<RenderLoadMenuElementsEvent>({curr_handler, master_handler});
ImGui::EndMenuBar();
}
}
void ViewerApplication::drawContents() { ImGui::Text("No handler selected!"); }
void ViewerApplication::drawUI()
{
ImVec2 viewer_size = ImGui::GetContentRegionAvail();
if (ImGui::BeginTable("##wiever_table", 2, ImGuiTableFlags_Resizable | ImGuiTableFlags_SizingStretchProp | ImGuiTableFlags_BordersInnerV))
{
ImGui::TableSetupColumn("##panel_v", ImGuiTableColumnFlags_None, viewer_size.x * panel_ratio);
ImGui::TableSetupColumn("##view", ImGuiTableColumnFlags_None, viewer_size.x * (1.0f - panel_ratio));
ImGui::TableNextColumn();
float left_width = ImGui::GetColumnWidth(0);
float right_width = viewer_size.x - left_width;
if (left_width != last_width && last_width != -1)
panel_ratio = left_width / viewer_size.x;
last_width = left_width;
ImGui::BeginChild("ViewerChildPanel", {left_width, float(viewer_size.y - 10)}, false, ImGuiWindowFlags_MenuBar);
drawMenuBar();
drawPanel();
ImGui::EndChild();
ImGui::TableNextColumn();
ImGui::BeginGroup();
if (curr_handler)
curr_handler->drawContents({float(right_width - 9 * ui_scale), float(viewer_size.y)});
else
drawContents();
ImGui::EndGroup();
ImGui::EndTable();
}
}
void ViewerApplication::tryOpenFileInViewer(std::string path)
{
if (file_open_thread.joinable())
file_open_thread.join();
auto fun = [this, path]()
{
try
{
if (!std::filesystem::path(path).has_extension())
{
logger->error("Invalid file, no extension!");
return;
}
if (std::filesystem::path(path).extension().string() == ".cbor")
{
logger->trace("Viewer loading product " + path);
try
{
auto prod_h = handlers::getProductHandlerForProduct(products::loadProduct(path));
master_handler->addSubHandler(prod_h);
}
catch (std::exception &e)
{
logger->error("Error loading product! Maybe not a valid product? Details : %s", e.what());
}
}
else if (std::filesystem::path(path).extension().string() == ".json")
{
logger->trace("Viewer loading dataset " + path);
try
{
products::DataSet dataset;
dataset.load(path);
std::shared_ptr<handlers::DatasetHandler> dat_h = std::make_shared<handlers::DatasetHandler>(std::filesystem::path(path).parent_path().string(), dataset);
master_handler->addSubHandler(dat_h);
}
catch (std::exception &e)
{
logger->error("Error loading dataset! Maybe not a valid dataset? Details : %s", e.what());
}
}
else if (std::filesystem::path(path).extension().string() == ".shp")
{
logger->trace("Viewer loading shapefile " + path);
master_handler->addSubHandler(std::make_shared<handlers::ShapefileHandler>(path));
}
else
{
logger->trace("Viewer loading image " + path);
image::Image img;
image::load_img(img, path);
if (img.size() > 0)
master_handler->addSubHandler(std::make_shared<handlers::ImageHandler>(img, std::filesystem::path(path).stem().string()));
else
logger->error("Could not open this file as image!"); // TODOREWORK Probably check before this?
}
}
catch (std::exception &e)
{
logger->error("Error opening file! => %s", e.what());
}
};
file_open_thread = std::thread(fun);
}
} // namespace viewer
}; // namespace satdump
Reference in a new issue View git blame Copy permalink