100 lines
3.8 KiB
C++
100 lines
3.8 KiB
C++
#include <hex/helpers/utils.hpp>
|
|
|
|
#include <content/visualizer_helpers.hpp>
|
|
|
|
#include <implot.h>
|
|
#include <imgui.h>
|
|
|
|
#include <hex/ui/imgui_imhex_extensions.h>
|
|
#include <pl/patterns/pattern_bitfield.hpp>
|
|
|
|
namespace hex::plugin::visualizers {
|
|
|
|
void drawDigitalSignalVisualizer(pl::ptrn::Pattern &, bool shouldReset, std::span<const pl::core::Token::Literal> arguments) {
|
|
auto *bitfield = dynamic_cast<pl::ptrn::PatternBitfield*>(arguments[0].toPattern().get());
|
|
if (bitfield == nullptr)
|
|
throw std::logic_error("Digital signal visualizer only works with bitfields.");
|
|
|
|
|
|
struct DataPoint {
|
|
std::array<ImVec2, 2> points;
|
|
std::string label;
|
|
std::string value;
|
|
ImColor color;
|
|
};
|
|
static std::vector<DataPoint> dataPoints;
|
|
static ImVec2 lastPoint;
|
|
|
|
if (shouldReset) {
|
|
dataPoints.clear();
|
|
lastPoint = { 0, 0 };
|
|
|
|
bitfield->forEachEntry(0, bitfield->getEntryCount(), [&](u64, pl::ptrn::Pattern *entry) {
|
|
size_t bitSize;
|
|
if (auto bitfieldField = dynamic_cast<pl::ptrn::PatternBitfieldField*>(entry); bitfieldField != nullptr)
|
|
bitSize = bitfieldField->getBitSize();
|
|
else
|
|
bitSize = entry->getSize() * 8;
|
|
|
|
auto value = entry->getValue();
|
|
bool high = value.toUnsigned() > 0;
|
|
dataPoints.emplace_back(
|
|
std::array<ImVec2, 2> { lastPoint, { lastPoint.x, high ? 1.0F : 0.0F } },
|
|
entry->getDisplayName(),
|
|
entry->getFormattedValue(),
|
|
entry->getColor()
|
|
);
|
|
|
|
lastPoint = dataPoints.back().points[1];
|
|
lastPoint.x += float(bitSize);
|
|
});
|
|
|
|
dataPoints.push_back({
|
|
.points = { lastPoint, { lastPoint.x, 0 } },
|
|
.label = "",
|
|
.value = "",
|
|
.color = ImColor(0x00)
|
|
});
|
|
}
|
|
|
|
if (ImPlot::BeginPlot("##Signal", ImVec2(600_scaled, 200_scaled), ImPlotFlags_NoLegend | ImPlotFlags_NoFrame | ImPlotFlags_NoMenus | ImPlotFlags_NoMouseText)) {
|
|
ImPlot::SetupAxisLimitsConstraints(ImAxis_X1, 0, lastPoint.x);
|
|
|
|
ImPlot::SetupAxis(ImAxis_Y1, "", ImPlotAxisFlags_LockMin | ImPlotAxisFlags_LockMax);
|
|
ImPlot::SetupAxisFormat(ImAxis_Y1, "");
|
|
ImPlot::SetupAxisLimits(ImAxis_Y1, -0.1F, 1.1F);
|
|
|
|
for (size_t i = 0; i < dataPoints.size() - 1; i += 1) {
|
|
const auto &left = dataPoints[i];
|
|
const auto &right = dataPoints[i + 1];
|
|
|
|
{
|
|
auto x = left.points[1].x + ((right.points[0].x - left.points[1].x) / 2);
|
|
ImPlot::Annotation(x, 0.55F, left.color, {}, false, "%s", left.label.c_str());
|
|
ImPlot::Annotation(x, 0.40F, left.color, {}, false, "%s", left.value.c_str());
|
|
}
|
|
|
|
{
|
|
ImVec2 min = ImPlot::PlotToPixels(ImPlotPoint(left.points[0].x, 0));
|
|
ImVec2 max = ImPlot::PlotToPixels(ImPlotPoint(right.points[1].x, 1));
|
|
|
|
ImPlot::PushPlotClipRect();
|
|
auto transparentColor = left.color;
|
|
transparentColor.Value.w = 0.2F;
|
|
ImPlot::GetPlotDrawList()->AddRectFilled(min, max, transparentColor);
|
|
ImPlot::PopPlotClipRect();
|
|
}
|
|
}
|
|
|
|
ImPlot::PushStyleVar(ImPlotStyleVar_LineWeight, 2_scaled);
|
|
ImPlot::PlotLineG("Signal", [](int idx, void*) -> ImPlotPoint {
|
|
return dataPoints[idx / 2].points[idx % 2];
|
|
}, nullptr, dataPoints.size() * 2);
|
|
ImPlot::PopStyleVar();
|
|
|
|
ImPlot::EndPlot();
|
|
}
|
|
}
|
|
|
|
}
|