97 lines
3.5 KiB
C
97 lines
3.5 KiB
C
#ifndef AUBINATOR_VIEWER_URB_H
|
|
#define AUBINATOR_VIEWER_URB_H
|
|
|
|
#include "aubinator_viewer.h"
|
|
|
|
#include "imgui/imgui.h"
|
|
|
|
struct AubinatorViewerUrb {
|
|
|
|
float RowHeight;
|
|
|
|
AubinatorViewerUrb() {
|
|
RowHeight = 10.0f;
|
|
}
|
|
|
|
bool _Hovered(const ImVec2& mouse, bool window_hovered,
|
|
const ImVec2& tl, const ImVec2& br) {
|
|
return window_hovered &&
|
|
tl.x <= mouse.x && tl.y <= mouse.y &&
|
|
br.x > mouse.x && br.y > mouse.y;
|
|
}
|
|
|
|
void DrawAllocation(const char *label,
|
|
int n_stages,
|
|
int end_urb_offset,
|
|
const char *stage_names[],
|
|
const struct aub_decode_urb_stage_state *stages) {
|
|
const ImVec2 label_size = ImGui::CalcTextSize("VS entry: ", NULL, true);
|
|
ImVec2 graph_size(ImGui::CalcItemWidth(), 2 * n_stages * label_size.y);
|
|
|
|
ImGui::BeginChild(label, ImVec2(0, graph_size.y), false);
|
|
|
|
ImDrawList* draw_list = ImGui::GetWindowDrawList();
|
|
|
|
const float row_height = MAX2(RowHeight, label_size.y);
|
|
const float width = ImGui::GetContentRegionAvailWidth() - label_size.x;
|
|
const float alloc_delta = width / end_urb_offset;
|
|
const ImVec2 window_pos = ImGui::GetWindowPos();
|
|
const ImVec2 mouse_pos = ImGui::GetMousePos();
|
|
const bool window_hovered = ImGui::IsWindowHovered();
|
|
|
|
int const_idx = 0;
|
|
for (int s = 0; s < n_stages; s++) {
|
|
const float x = window_pos.x + label_size.x;
|
|
const float y = window_pos.y + s * row_height;
|
|
|
|
ImVec2 alloc_pos(window_pos.x, y);
|
|
ImVec2 alloc_tl(x + stages[s].start * alloc_delta, y);
|
|
ImVec2 alloc_br(x + (stages[s].start +
|
|
stages[s].n_entries * stages[s].size) * alloc_delta,
|
|
y + row_height);
|
|
ImVec2 const_tl(x + const_idx * alloc_delta, y);
|
|
ImVec2 const_br(x + (const_idx + stages[s].const_rd_length) * alloc_delta,
|
|
y + row_height);
|
|
|
|
char label[40];
|
|
snprintf(label, sizeof(label), "%s: ", stage_names[s]);
|
|
draw_list->AddText(alloc_pos, ImGui::GetColorU32(ImGuiCol_Text), label);
|
|
|
|
float r, g, b;
|
|
bool hovered;
|
|
|
|
/* URB allocation */
|
|
hovered = _Hovered(mouse_pos, window_hovered, alloc_tl, alloc_br);
|
|
ImGui::ColorConvertHSVtoRGB((2 * s) * 1.0f / (2 * n_stages),
|
|
1.0f, hovered ? 1.0f : 0.8f,
|
|
r, g, b);
|
|
draw_list->AddRectFilled(alloc_tl, alloc_br, ImColor(r, g, b));
|
|
if (hovered) {
|
|
ImGui::SetTooltip("%s: start=%u end=%u",
|
|
stage_names[s],
|
|
stages[s].start,
|
|
stages[s].start + stages[s].n_entries * stages[s].size);
|
|
}
|
|
|
|
/* Constant URB input */
|
|
hovered = _Hovered(mouse_pos, window_hovered, const_tl, const_br);
|
|
ImGui::ColorConvertHSVtoRGB((2 * s + 1) * 1.0f / (2 * n_stages),
|
|
1.0f, hovered ? 1.0f : 0.8f,
|
|
r, g, b);
|
|
draw_list->AddRectFilled(const_tl, const_br, ImColor(r, g, b));
|
|
if (hovered) {
|
|
ImGui::SetTooltip("%s constant: start=%u end=%u",
|
|
stage_names[s],
|
|
stages[s].rd_offset,
|
|
stages[s].rd_offset + stages[s].rd_length);
|
|
}
|
|
|
|
const_idx += stages[s].const_rd_length;
|
|
}
|
|
|
|
ImGui::EndChild();
|
|
}
|
|
};
|
|
|
|
#endif /* AUBINATOR_VIEWER_URB_H */
|