You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
287 lines
11 KiB
287 lines
11 KiB
|
|
#pragma once
|
|
#include "imgui.h"
|
|
#include "imgui_internal.h"
|
|
|
|
|
|
|
|
namespace ImGui {
|
|
|
|
static ImU32 InvertColorU32(ImU32 in)
|
|
{
|
|
ImVec4 in4 = ColorConvertU32ToFloat4(in);
|
|
in4.x = 1.f - in4.x;
|
|
in4.y = 1.f - in4.y;
|
|
in4.z = 1.f - in4.z;
|
|
return GetColorU32(in4);
|
|
}
|
|
|
|
static ImVec2 add(const ImVec2& a, const ImVec2& b){
|
|
return ImVec2(a.x+b.x, a.y+b.y);
|
|
}
|
|
static ImVec2 subtract(const ImVec2& a, const ImVec2& b){
|
|
return ImVec2(a.x-b.x, a.y-b.y);
|
|
}
|
|
|
|
static void PlotMultiLines(
|
|
const std::vector<std::vector<int>> &data,
|
|
std::string title,
|
|
const std::vector<std::string> &labels,
|
|
const std::vector<ImColor> &colors,
|
|
float maximum_scale,
|
|
float minimum_scale,
|
|
sf::Vector2f graph_size
|
|
){
|
|
|
|
int values_count = data.at(0).size();
|
|
|
|
ImGuiWindow* window = GetCurrentWindow();
|
|
if (window->SkipItems)
|
|
return;
|
|
|
|
ImGuiContext& g = *GImGui;
|
|
const ImGuiStyle& style = g.Style;
|
|
|
|
const ImVec2 label_size = ImGui::CalcTextSize(title.c_str(), NULL, true);
|
|
|
|
if (graph_size.x == 0.0f)
|
|
graph_size.x = CalcItemWidth();
|
|
if (graph_size.y == 0.0f)
|
|
graph_size.y = label_size.y + (style.FramePadding.y * 2);
|
|
|
|
const ImRect frame_bb(window->DC.CursorPos,
|
|
add(window->DC.CursorPos, graph_size));
|
|
const ImRect inner_bb(add(frame_bb.Min,style.FramePadding),
|
|
subtract(frame_bb.Max , style.FramePadding));
|
|
ImVec2 thing = ImVec2(label_size.x > 0.0f ?
|
|
style.ItemInnerSpacing.x + label_size.x : 0.0f,
|
|
0);
|
|
const ImRect total_bb(frame_bb.Min, add(frame_bb.Max , thing));
|
|
|
|
ItemSize(total_bb, style.FramePadding.y);
|
|
if (!ItemAdd(total_bb, 0))
|
|
return;
|
|
|
|
RenderFrame(frame_bb.Min, frame_bb.Max, GetColorU32(ImGuiCol_FrameBg),
|
|
true, style.FrameRounding);
|
|
|
|
int res_w = ImMin((int) graph_size.x, values_count );
|
|
int item_count = values_count;
|
|
int text_margin = 10;
|
|
|
|
for (int data_idx = 0; data_idx < data.size(); ++data_idx)
|
|
{
|
|
const float t_step = 1.0f / (float) res_w;
|
|
|
|
float v0 = data.at(data_idx).at(0);
|
|
float t0 = 0.0f;
|
|
ImVec2 tp0 = ImVec2(t0,
|
|
1.0f - ImSaturate((v0 - minimum_scale) / (maximum_scale - minimum_scale))); // Point in the normalized space of our target rectangle
|
|
|
|
const ImU32 col_base = colors.at(data_idx);
|
|
const ImU32 col_hovered = InvertColorU32(colors.at(data_idx));
|
|
|
|
for (int n = 0; n < res_w; n++)
|
|
{
|
|
const float t1 = t0 + t_step;
|
|
const int v1_idx = (int) (t0 * item_count + 0.5f);
|
|
IM_ASSERT(v1_idx >= 0 && v1_idx < values_count);
|
|
const float v1 = data.at(data_idx).at(v1_idx);
|
|
const ImVec2 tp1 = ImVec2(t1,
|
|
1.0f - ImSaturate((v1 - minimum_scale) / (maximum_scale - minimum_scale)));
|
|
|
|
// NB: Draw calls are merged together by the DrawList system. Still, we should render our batch are lower level to save a bit of CPU.
|
|
ImVec2 pos0 = ImLerp(inner_bb.Min, inner_bb.Max, tp0);
|
|
ImVec2 pos1 = ImLerp(inner_bb.Min, inner_bb.Max, tp1);//ImVec2(tp1.x, 1.0f));
|
|
|
|
window->DrawList->AddLine(pos0, pos1, col_base);
|
|
|
|
|
|
t0 = t1;
|
|
tp0 = tp1;
|
|
}
|
|
|
|
// Render the text label
|
|
ImGui::PushStyleColor(ImGuiCol_Text, (ImVec4)colors.at(data_idx));
|
|
RenderText(ImVec2(frame_bb.Max.x + style.ItemInnerSpacing.x, inner_bb.Min.y + text_margin * data_idx), labels.at(data_idx).c_str());
|
|
ImGui::PopStyleColor(1);
|
|
}
|
|
|
|
RenderText(ImVec2(frame_bb.Max.x + style.ItemInnerSpacing.x, inner_bb.Max.y), title.c_str());
|
|
}
|
|
|
|
|
|
static void PlotMultiEx(
|
|
ImGuiPlotType plot_type,
|
|
const char* label,
|
|
int num_datas,
|
|
const char** names,
|
|
const ImColor* colors,
|
|
float(*getter)(const void* data, int idx),
|
|
const void * const * datas,
|
|
int values_count,
|
|
float scale_min,
|
|
float scale_max,
|
|
ImVec2 graph_size){
|
|
|
|
const int values_offset = 0;
|
|
|
|
ImGuiWindow* window = GetCurrentWindow();
|
|
if (window->SkipItems)
|
|
return;
|
|
|
|
ImGuiContext& g = *GImGui;
|
|
const ImGuiStyle& style = g.Style;
|
|
|
|
const ImVec2 label_size = ImGui::CalcTextSize(label, NULL, true);
|
|
if (graph_size.x == 0.0f)
|
|
graph_size.x = CalcItemWidth();
|
|
if (graph_size.y == 0.0f)
|
|
graph_size.y = label_size.y + (style.FramePadding.y * 2);
|
|
|
|
|
|
|
|
const ImRect frame_bb(window->DC.CursorPos, add(window->DC.CursorPos, graph_size));
|
|
const ImRect inner_bb(add(frame_bb.Min,style.FramePadding), subtract(frame_bb.Max , style.FramePadding));
|
|
ImVec2 thing = ImVec2(label_size.x > 0.0f ? style.ItemInnerSpacing.x + label_size.x : 0.0f, 0);
|
|
const ImRect total_bb(frame_bb.Min, add(frame_bb.Max , thing));
|
|
|
|
|
|
ItemSize(total_bb, style.FramePadding.y);
|
|
if (!ItemAdd(total_bb, 0))
|
|
return;
|
|
|
|
// Determine scale from values if not specified
|
|
if (scale_min == FLT_MAX || scale_max == FLT_MAX)
|
|
{
|
|
float v_min = FLT_MAX;
|
|
float v_max = -FLT_MAX;
|
|
for (int data_idx = 0; data_idx < num_datas; ++data_idx)
|
|
{
|
|
for (int i = 0; i < values_count; i++)
|
|
{
|
|
const float v = getter(datas[data_idx], i);
|
|
v_min = ImMin(v_min, v);
|
|
v_max = ImMax(v_max, v);
|
|
}
|
|
}
|
|
if (scale_min == FLT_MAX)
|
|
scale_min = v_min;
|
|
if (scale_max == FLT_MAX)
|
|
scale_max = v_max;
|
|
}
|
|
|
|
RenderFrame(frame_bb.Min, frame_bb.Max, GetColorU32(ImGuiCol_FrameBg), true, style.FrameRounding);
|
|
|
|
int res_w = ImMin((int) graph_size.x, values_count) + ((plot_type == ImGuiPlotType_Lines) ? -1 : 0);
|
|
int item_count = values_count + ((plot_type == ImGuiPlotType_Lines) ? -1 : 0);
|
|
|
|
// Tooltip on hover
|
|
// int v_hovered = -1;
|
|
// if (IsHovered(inner_bb, 0))
|
|
// {
|
|
// const float t = ImClamp((g.IO.MousePos.x - inner_bb.Min.x) / (inner_bb.Max.x - inner_bb.Min.x), 0.0f, 0.9999f);
|
|
// const int v_idx = (int) (t * item_count);
|
|
// IM_ASSERT(v_idx >= 0 && v_idx < values_count);
|
|
//
|
|
// // std::string toolTip;
|
|
// ImGui::BeginTooltip();
|
|
// const int idx0 = (v_idx + values_offset) % values_count;
|
|
// if (plot_type == ImGuiPlotType_Lines)
|
|
// {
|
|
// const int idx1 = (v_idx + 1 + values_offset) % values_count;
|
|
// Text("%8d %8d | Name", v_idx, v_idx+1);
|
|
// for (int dataIdx = 0; dataIdx < num_datas; ++dataIdx)
|
|
// {
|
|
// const float v0 = getter(datas[dataIdx], idx0);
|
|
// const float v1 = getter(datas[dataIdx], idx1);
|
|
// TextColored(colors[dataIdx], "%8.4g %8.4g | %s", v0, v1, names[dataIdx]);
|
|
// }
|
|
// }
|
|
// else if (plot_type == ImGuiPlotType_Histogram)
|
|
// {
|
|
// for (int dataIdx = 0; dataIdx < num_datas; ++dataIdx)
|
|
// {
|
|
// const float v0 = getter(datas[dataIdx], idx0);
|
|
// TextColored(colors[dataIdx], "%d: %8.4g | %s", v_idx, v0, names[dataIdx]);
|
|
// }
|
|
// }
|
|
// ImGui::EndTooltip();
|
|
// v_hovered = v_idx;
|
|
// }
|
|
|
|
for (int data_idx = 0; data_idx < num_datas; ++data_idx)
|
|
{
|
|
const float t_step = 1.0f / (float) res_w;
|
|
|
|
float v0 = getter(datas[data_idx], (0 + values_offset) % values_count);
|
|
float t0 = 0.0f;
|
|
ImVec2 tp0 = ImVec2(t0, 1.0f - ImSaturate((v0 - scale_min) / (scale_max - scale_min))); // Point in the normalized space of our target rectangle
|
|
|
|
const ImU32 col_base = colors[data_idx];
|
|
const ImU32 col_hovered = InvertColorU32(colors[data_idx]);
|
|
|
|
//const ImU32 col_base = GetColorU32((plot_type == ImGuiPlotType_Lines) ? ImGuiCol_PlotLines : ImGuiCol_PlotHistogram);
|
|
//const ImU32 col_hovered = GetColorU32((plot_type == ImGuiPlotType_Lines) ? ImGuiCol_PlotLinesHovered : ImGuiCol_PlotHistogramHovered);
|
|
|
|
for (int n = 0; n < res_w; n++)
|
|
{
|
|
const float t1 = t0 + t_step;
|
|
const int v1_idx = (int) (t0 * item_count + 0.5f);
|
|
IM_ASSERT(v1_idx >= 0 && v1_idx < values_count);
|
|
const float v1 = getter(datas[data_idx], (v1_idx + values_offset + 1) % values_count);
|
|
const ImVec2 tp1 = ImVec2(t1, 1.0f - ImSaturate((v1 - scale_min) / (scale_max - scale_min)));
|
|
|
|
// NB: Draw calls are merged together by the DrawList system. Still, we should render our batch are lower level to save a bit of CPU.
|
|
ImVec2 pos0 = ImLerp(inner_bb.Min, inner_bb.Max, tp0);
|
|
ImVec2 pos1 = ImLerp(inner_bb.Min, inner_bb.Max, (plot_type == ImGuiPlotType_Lines) ? tp1 : ImVec2(tp1.x, 1.0f));
|
|
if (plot_type == ImGuiPlotType_Lines)
|
|
{
|
|
window->DrawList->AddLine(pos0, pos1, col_base);
|
|
}
|
|
else if (plot_type == ImGuiPlotType_Histogram)
|
|
{
|
|
if (pos1.x >= pos0.x + 2.0f)
|
|
pos1.x -= 1.0f;
|
|
window->DrawList->AddRectFilled(pos0, pos1, col_base);
|
|
}
|
|
|
|
t0 = t1;
|
|
tp0 = tp1;
|
|
}
|
|
}
|
|
|
|
RenderText(ImVec2(frame_bb.Max.x + style.ItemInnerSpacing.x, inner_bb.Min.y), label);
|
|
}
|
|
|
|
inline void PlotMultiLines(
|
|
const char* label,
|
|
int num_datas,
|
|
const char** names,
|
|
const ImColor* colors,
|
|
float(*getter)(const void* data, int idx),
|
|
const void * const * datas,
|
|
int values_count,
|
|
float scale_min,
|
|
float scale_max,
|
|
ImVec2 graph_size)
|
|
{
|
|
PlotMultiEx(ImGuiPlotType_Lines, label, num_datas, names, colors, getter, datas, values_count, scale_min, scale_max, graph_size);
|
|
}
|
|
|
|
inline void PlotMultiHistograms(
|
|
const char* label,
|
|
int num_hists,
|
|
const char** names,
|
|
const ImColor* colors,
|
|
float(*getter)(const void* data, int idx),
|
|
const void * const * datas,
|
|
int values_count,
|
|
float scale_min,
|
|
float scale_max,
|
|
ImVec2 graph_size)
|
|
{
|
|
PlotMultiEx(ImGuiPlotType_Histogram, label, num_hists, names, colors, getter, datas, values_count, scale_min, scale_max, graph_size);
|
|
}
|
|
|
|
} // namespace ImGui
|