kinect/codes/Azure-Kinect-Sensor-SDK/tools/k4aviewer/k4aimguiextensions.h

266 lines
6.7 KiB
C
Raw Normal View History

2024-03-06 18:05:53 +00:00
// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License.
#ifndef K4AIMGUIEXTENSIONS_H
#define K4AIMGUIEXTENSIONS_H
// System headers
//
#include <algorithm>
#include <functional>
#include <string>
#include <utility>
#include <vector>
// Library headers
//
#include "k4aimgui_all.h"
// Project headers
//
namespace k4aviewer
{
namespace ImGuiExtensions
{
enum class ButtonColor
{
Red,
Yellow,
Green
};
// Sets all future buttons to color until cleared or destroyed.
// Optionally set enabled=false if you don't want to change color under some circumstances
//
class ButtonColorChanger
{
public:
ButtonColorChanger(ButtonColor color, bool enabled = true) : m_active(enabled)
{
if (!m_active)
{
return;
}
float hue = 0.0f;
float sat = 0.6f;
float val = 0.6f;
switch (color)
{
case ButtonColor::Green:
hue = 0.4f;
break;
case ButtonColor::Yellow:
hue = 0.15f;
break;
case ButtonColor::Red:
hue = 0.0f;
break;
}
constexpr float hoveredSvColorOffset = 0.1f;
constexpr float activeSvColorOffset = 0.2f;
ImGui::PushStyleColor(ImGuiCol_Button, ImVec4(ImColor::HSV(hue, sat, val)));
ImGui::PushStyleColor(ImGuiCol_ButtonHovered,
ImVec4(ImColor::HSV(hue, sat + hoveredSvColorOffset, val + hoveredSvColorOffset)));
ImGui::PushStyleColor(ImGuiCol_ButtonActive,
ImVec4(ImColor::HSV(hue, sat + activeSvColorOffset, val + activeSvColorOffset)));
}
void Clear()
{
if (m_active)
{
m_active = false;
ImGui::PopStyleColor(3);
}
}
~ButtonColorChanger()
{
if (m_active)
{
Clear();
}
}
private:
bool m_active;
};
enum class TextColor
{
Normal,
Warning
};
class TextColorChanger
{
public:
TextColorChanger(TextColor color, bool enabled = true) : m_active(enabled)
{
if (!m_active)
{
return;
}
ImVec4 colorVec(1.0f, 1.0f, 1.0f, 1.0f);
switch (color)
{
case TextColor::Normal:
colorVec = ImVec4(1.0f, 1.0f, 1.0f, 1.0f);
break;
case TextColor::Warning:
colorVec = ImVec4(1.0f, 1.0f, 0.0f, 1.0f);
break;
}
ImGui::PushStyleColor(ImGuiCol_Text, colorVec);
}
void Clear()
{
if (m_active)
{
m_active = false;
ImGui::PopStyleColor();
}
}
~TextColorChanger()
{
if (m_active)
{
Clear();
}
}
private:
bool m_active;
};
template<typename T> T ShowDisableableControl(std::function<T()> showControlFunction, const bool enabled)
{
if (!enabled)
{
ImGui::PushStyleVar(ImGuiStyleVar_Alpha, ImGui::GetStyle().Alpha * 0.5f);
ImGui::PushItemFlag(ImGuiItemFlags_Disabled, true);
}
const T result = showControlFunction();
if (!enabled)
{
ImGui::PopItemFlag();
ImGui::PopStyleVar();
}
return result;
}
// Shows a dear imgui combo box based on a list of value-presentation string pairs,
// updating currentValue if the value was modified
//
template<typename T>
bool K4AComboBox(const char *label,
const char *noItemsText,
const ImGuiComboFlags flags,
const std::vector<std::pair<T, std::string>> &items,
T *currentValue,
bool enabled = true)
{
return ShowDisableableControl<bool>(
[label, noItemsText, flags, &items, currentValue]() {
bool wasUpdated = false;
auto selector = [currentValue](const std::pair<T, std::string> &a) { return a.first == *currentValue; };
auto currentlySelected = std::find_if(items.begin(), items.end(), selector);
const char *message = noItemsText;
if (currentlySelected != items.end())
{
message = currentlySelected->second.c_str();
}
if (ImGui::BeginCombo(label, message, flags))
{
for (const auto &item : items)
{
bool selected = item.first == *currentValue;
if (ImGui::Selectable(item.second.c_str(), selected))
{
*currentValue = item.first;
wasUpdated = true;
}
if (selected)
{
ImGui::SetItemDefaultFocus();
}
}
ImGui::EndCombo();
}
return wasUpdated;
},
enabled);
}
// Behaves like a normal ImGui::Button, but with support for being disabled
//
bool K4AButton(const char *label, bool enabled = true);
bool K4AButton(const char *label, const ImVec2 &size, bool enabled = true);
// Behaves like a normal ImGui::Checkbox, but with support for being disabled
//
bool K4ACheckbox(const char *label, bool *checked, bool enabled = true);
// Behaves like a normal ImGui::RadioButton, but with support for being disabled
//
bool K4ARadioButton(const char *label, bool active, bool enabled = true);
bool K4ARadioButton(const char *label, int *v, int vButton, bool enabled = true);
// Behaves like a normal ImGui::InputScalar, but with support for being disabled
//
bool K4AInputScalar(const char *label,
ImGuiDataType dataType,
void *dataPtr,
const void *step,
const void *stepFast,
const char *format,
bool enabled = true);
// Behaves like a normal ImGui::SliderInt, but with support for being disabled
//
bool K4ASliderInt(const char *label, int *value, int valueMin, int valueMax, const char *format, bool enabled = true);
// Behaves like a normal ImGui::SliderFloat, but with support for being disabled
//
bool K4ASliderFloat(const char *label,
float *value,
float valueMin,
float valueMax,
const char *format,
float power,
bool enabled = true);
// Shows a vertical slider
//
bool K4AVSliderFloat(const char *name, ImVec2 size, float *value, float minValue, float maxValue, const char *label);
// Shows vertical text
//
void K4AVText(const char *str);
// Shows a tooltip if the most recently-drawn control was hovered and show == true
//
void K4AShowTooltip(const char *msg, bool show = true);
} // namespace ImGuiExtensions
} // namespace k4aviewer
#endif