satdump/src-interface/viewer/radiation_handler.cpp
2022-09-13 15:06:09 -07:00

91 lines
No EOL
2.9 KiB
C++

#include "radiation_handler.h"
#include "core/config.h"
#include "resources.h"
namespace satdump
{
void RadiationViewerHandler::init()
{
products = (RadiationProducts *)ViewerHandler::products;
for (int c = 0; c < (int)products->channel_counts.size(); c++)
select_channel_image_str += "Channel " + std::to_string(c + 1) + '\0';
update();
}
void RadiationViewerHandler::update()
{
if (selected_visualization_id == 0)
{
RadiationMapCfg cfg;
cfg.channel = select_channel_image_id + 1;
cfg.radius = 5;
cfg.min = map_min;
cfg.max = map_max;
image::Image<uint16_t> map = make_radiation_map(*products, cfg);
image_view.update(map);
}
else if (selected_visualization_id == 1)
{
graph_values.clear();
graph_values.resize(products->channel_counts.size());
for (int c = 0; c < (int)products->channel_counts.size(); c++)
{
for (int i = 0; i < (int)products->channel_counts[c].size(); i++)
graph_values[c].push_back(products->channel_counts[c][i]);
}
}
}
void RadiationViewerHandler::drawMenu()
{
if (ImGui::CollapsingHeader("Vizualisation"))
{
if (ImGui::RadioButton(u8"\uf84c Map", &selected_visualization_id, 0))
update();
if (ImGui::RadioButton(u8"\uf437 Graph", &selected_visualization_id, 1))
update();
if (selected_visualization_id == 0)
{
if (ImGui::Combo("###mapchannelcomboid", &select_channel_image_id, select_channel_image_str.c_str()))
update();
ImGui::SetNextItemWidth(ImGui::GetWindowWidth()/2);
if (ImGui::DragInt("##Min", &map_min, 1.0f, 0, 255, "Min: %d", ImGuiSliderFlags_AlwaysClamp))
update();
ImGui::SameLine();
ImGui::SetNextItemWidth(ImGui::GetWindowWidth()/2);
if(ImGui::DragInt("##Max", &map_max, 1.0f, 0, 255, "Max: %d", ImGuiSliderFlags_AlwaysClamp))
update();
}
}
}
void RadiationViewerHandler::drawContents(ImVec2 win_size)
{
if (selected_visualization_id == 0)
{
image_view.draw(win_size);
}
else if (selected_visualization_id == 1)
{
ImGui::BeginChild("RadiationPlotChild");
for (int i = 0; i < (int)products->channel_counts.size(); i++){
ImGui::PlotLines(std::string("Channel " + std::to_string(i + 1)).c_str(), graph_values[i].data(), graph_values[i].size());
ImGui::Spacing();
ImGui::Separator();
ImGui::Spacing();
}
ImGui::EndChild();
}
}
float RadiationViewerHandler::drawTreeMenu()
{
return 0;
}
}