DieMitchell/ReSkate
1
0 Fork 0
mirror of https://github.com/Dingo-Shenanigans/ReSkate.git synced 2026-10-23 17:35:57 +02:00
Code Issues Packages Projects Releases Wiki Activity
Files
805cd0d6f5806b1603b0af1109a3041e540f2dc6
ReSkate/Extension/UI/Overlay/plugin_overlay.cpp
T
AyeZeeBB 805cd0d6f5 Plugins: Animator Studio's screen 1:1, on general drawing, screen and pose APIs 
- The studio plugin draws PR 297's screen itself: skate.'s Trick List menu and card, the joint timeline with key rows, dragging joints on the skater, keyboard and pad controls with their prompts, the unsaved-changes card, the recording strip and the loading screen. It lists a trick library made with tools/make_trick_library.py (the PR's, here beside it), and checks itself with studio test.
- Drawing: d:rough_rect, d:halftone, d:scribble, d:clip, d:measure, a text wrap width, the title_large font, text up to 400, and the loading screen's kickflipping board as d:game_image("loading_board") with a frame.
- reskate.ui.screen with panel = false: input taken, nothing drawn, the plugin draws it.
- reskate.skater.joint gives each joint's parent; reskate.skater.pose takes an offset.
2026-10-20 17:22:19 -05:00

361 lines
17 KiB
C++
Raw Blame History

#include "plugin_overlay.h"
#include "Extension/Plugins/plugin_camera.h"
#include "Extension/Plugins/plugin_ui.h"
#include "Extension/UI/Overlay/game_images.h"
#include "Extension/UI/Overlay/overlay_internal.h"
#include "Extension/UI/Overlay/xray_overlay.h"
#include "Extension/UI/Overlay/overlay.h"
#include "Extension/UI/Overlay/skate_style.h"
#include <algorithm>
#include <imgui_internal.h>
#include <memory>
#include <mutex>
namespace dingosdk::overlay {
namespace {
// Render thread: the layers taken by plugins_pending, drawn by draw_plugins.
std::shared_ptr<const std::vector<plugin_ui::Layer>>& frame() {
static auto* value = new std::shared_ptr<const std::vector<plugin_ui::Layer>>;
return *value;
}
ImVec2 at(const plugin_ui::Vec3& p) { return ImVec2(p[0], p[1]); }
ImU32 shadow_of(ImU32 colour) {
const auto alpha = static_cast<ImU32>(((colour >> IM_COL32_A_SHIFT) & 0xff) * 0.75f);
return IM_COL32(0, 0, 0, alpha);
}
void text(ImDrawList* draw, ImFont* font, float k, ImVec2 where, const plugin_ui::Shape& shape) {
const float size = shape.style.size * k;
if (shape.style.wrap > 0) {
// Wrapped: from its top left, as d:measure counts it.
const float wrap = shape.style.wrap * k;
if (shape.style.shadow) {
const float offset = std::max(1.0f, size * 0.06f);
draw->AddText(font, size, ImVec2(where.x + offset, where.y + offset), shadow_of(shape.style.colour), shape.text.c_str(), nullptr, wrap);
}
draw->AddText(font, size, where, shape.style.colour, shape.text.c_str(), nullptr, wrap);
return;
}
const auto width = font->CalcTextSizeA(size, FLT_MAX, 0.0f, shape.text.c_str()).x;
if (shape.style.align == 1) where.x -= width * 0.5f;
else if (shape.style.align == 2) where.x -= width;
if (shape.style.shadow) {
const float offset = std::max(1.0f, size * 0.06f);
draw->AddText(font, size, ImVec2(where.x + offset, where.y + offset), shadow_of(shape.style.colour), shape.text.c_str());
}
draw->AddText(font, size, where, shape.style.colour, shape.text.c_str());
}
void circle(ImDrawList* draw, ImVec2 centre, float radius, const plugin_ui::Style& style, float k) {
if (radius <= 0) return;
if (style.fill) draw->AddCircleFilled(centre, radius, style.colour);
else draw->AddCircle(centre, radius, style.colour, 0, style.thickness * k);
}
}
// The mouse while a plugin's screen is open, for reskate.input.cursor (render thread writes it).
struct CursorState {
std::mutex mutex;
PluginCursor now;
bool open{};
};
static CursorState& cursor_state() {
static auto* value = new CursorState;
return *value;
}
bool read_plugin_cursor(PluginCursor& out) noexcept {
auto& c = cursor_state();
std::lock_guard lock(c.mutex);
if (!c.open) return false;
out = c.now;
c.now.wheel = 0;
return true;
}
static void note_cursor(bool open, bool over_screen) {
auto& c = cursor_state();
std::lock_guard lock(c.mutex);
c.open = open;
if (!open) {
c.now = {};
return;
}
const auto& io = ImGui::GetIO();
c.now.x = io.MousePos.x;
c.now.y = io.MousePos.y;
c.now.left = io.MouseDown[0];
c.now.right = io.MouseDown[1];
c.now.middle = io.MouseDown[2];
c.now.wheel += io.MouseWheel;
c.now.over_screen = over_screen;
}
// The screen Esc closed, until its plugin's next build no longer has it open.
static std::string& closed_screen() {
static auto* value = new std::string;
return *value;
}
static const plugin_ui::Panel* open_screen() {
const auto panels = plugin_ui::latest_panels();
static std::shared_ptr<const std::vector<plugin_ui::Panel>> held;
held = panels;
if (!held) return nullptr;
for (const auto& panel : *held)
if (panel.screen) {
if (closed_screen() == panel.plugin + '\n' + panel.title) return nullptr;
return &panel;
}
closed_screen().clear();
return nullptr;
}
// Called before the overlay's ImGui context exists or is made current, too: its size is read from
// the context itself (what its last frame had), never through ImGui's current one.
bool plugins_pending() {
detail::state().plugin_screen_visible.store(open_screen() != nullptr);
if (const auto* context = detail::state().context) {
const auto display = context->IO.DisplaySize;
if (display.x > 0 && display.y > 0) plugin_ui::set_display(display.x, display.y);
}
frame() = plugin_ui::latest_layers();
return frame() && !frame()->empty();
}
void draw_plugin_screen() {
const auto* screen = open_screen();
if (!screen) {
note_cursor(false, false);
return;
}
const auto display = ImGui::GetIO().DisplaySize;
const float k = std::max(1.0f, display.y / 1080.0f);
auto& s = detail::state();
// No panel: the input is the plugin's, which draws its screen itself; Esc still closes it.
if (screen->dock == 3) {
note_cursor(true, false);
if (ImGui::IsKeyPressed(ImGuiKey_Escape, false) && !ImGui::GetIO().WantTextInput) {
plugin_ui::act({screen->plugin, screen->title, plugin_ui::close_widget});
closed_screen() = screen->plugin + '\n' + screen->title;
s.plugin_screen_visible.store(false);
}
return;
}
// Docked: a panel down one side, the game showing beside it.
const bool docked = screen->dock != 0;
const float panel = docked ? std::round(display.x * std::clamp(screen->width, 0.2f, 0.6f)) : display.x;
ImGui::SetNextWindowPos(ImVec2(screen->dock == 2 ? display.x - panel : 0.0f, 0));
ImGui::SetNextWindowSize(ImVec2(panel, display.y));
ImGui::SetNextWindowBgAlpha(docked ? 0.88f : 0.94f);
ImGui::PushStyleVar(ImGuiStyleVar_WindowPadding, docked ? ImVec2(24 * k, 24 * k) : ImVec2(64 * k, 48 * k));
const auto flags = ImGuiWindowFlags_NoDecoration | ImGuiWindowFlags_NoMove | ImGuiWindowFlags_NoSavedSettings;
const auto& io = ImGui::GetIO();
const float left = screen->dock == 2 ? display.x - panel : 0.0f;
note_cursor(true, io.MousePos.x >= left && io.MousePos.x < left + panel);
if (ImGui::Begin("##plugin-screen", nullptr, flags)) {
const float wide = docked ? ImGui::GetContentRegionAvail().x * 0.6f : 480 * k;
const float button = docked ? ImGui::GetContentRegionAvail().x : 320 * k;
ImGui::SetWindowFontScale(1.25f * k);
if (s.menu.heading) ImGui::PushFont(s.menu.heading);
ImGui::TextUnformatted(screen->title.c_str());
if (s.menu.heading) ImGui::PopFont();
ImGui::TextDisabled("%s - Esc closes it", screen->plugin.c_str());
ImGui::Separator();
ImGui::Spacing();
int item = 0;
for (const auto& w : screen->widgets) {
ImGui::PushID(item++);
plugin_ui::Action action{screen->plugin, screen->title, w.id};
bool acted = false;
using Kind = plugin_ui::Widget::Kind;
switch (w.kind) {
case Kind::heading:
ImGui::Spacing();
if (s.menu.bold) ImGui::PushFont(s.menu.bold);
ImGui::TextUnformatted(w.text.c_str());
if (s.menu.bold) ImGui::PopFont();
break;
case Kind::label: ImGui::TextWrapped("%s", w.text.c_str()); break;
case Kind::separator: ImGui::Separator(); break;
case Kind::card:
ImGui::Spacing();
if (s.menu.heading) ImGui::PushFont(s.menu.heading);
ImGui::TextUnformatted(w.text.c_str());
if (s.menu.heading) ImGui::PopFont();
if (!w.help.empty()) ImGui::TextDisabled("%s", w.help.c_str());
break;
case Kind::note: ImGui::TextDisabled("%s", w.text.c_str()); break;
case Kind::warn: ImGui::TextColored(ImVec4(1.0f, 0.75f, 0.3f, 1.0f), "%s", w.text.c_str()); break;
case Kind::info: ImGui::Text("%s: %s", w.text.c_str(), w.help.c_str()); break;
case Kind::tabs:
for (int i = 0; i < static_cast<int>(w.choices.size()); ++i) {
if (i) ImGui::SameLine();
ImGui::PushID(i);
if (ImGui::Selectable(w.choices[static_cast<std::size_t>(i)].c_str(), i + 1 == w.chosen, 0,
ImGui::CalcTextSize(w.choices[static_cast<std::size_t>(i)].c_str()))) {
action.chosen = i + 1;
acted = true;
}
ImGui::PopID();
}
break;
case Kind::bar:
ImGui::ProgressBar((w.value - w.minimum) / (w.maximum - w.minimum), ImVec2(wide, 0), w.text.c_str());
break;
case Kind::button: acted = ImGui::Button(w.text.c_str(), ImVec2(button, 0)); break;
case Kind::toggle: {
bool on = w.on;
if (ImGui::Checkbox(w.text.c_str(), &on)) {
action.on = on;
acted = true;
}
break;
}
case Kind::slider: {
float value = w.value;
ImGui::SetNextItemWidth(wide);
if (ImGui::SliderFloat(w.text.c_str(), &value, w.minimum, w.maximum, "%.2f", ImGuiSliderFlags_AlwaysClamp)) {
action.value = value;
acted = true;
}
break;
}
case Kind::dropdown: {
int selected = std::clamp(w.chosen - 1, 0, static_cast<int>(w.choices.size()) - 1);
ImGui::SetNextItemWidth(wide);
if (ImGui::BeginCombo(w.text.c_str(), w.choices.empty() ? "" : w.choices[static_cast<std::size_t>(selected)].c_str())) {
for (int i = 0; i < static_cast<int>(w.choices.size()); ++i)
if (ImGui::Selectable(w.choices[static_cast<std::size_t>(i)].c_str(), i == selected)) {
action.chosen = i + 1;
acted = true;
}
ImGui::EndCombo();
}
break;
}
}
if (acted) plugin_ui::act(std::move(action));
ImGui::PopID();
}
// Esc closes it at once here; its plugin hears screen_closed.
if (ImGui::IsKeyPressed(ImGuiKey_Escape, false) && !ImGui::GetIO().WantTextInput) {
plugin_ui::act({screen->plugin, screen->title, plugin_ui::close_widget});
closed_screen() = screen->plugin + '\n' + screen->title;
s.plugin_screen_visible.store(false);
}
}
ImGui::End();
ImGui::PopStyleVar();
}
void draw_plugins() {
const auto layers = frame();
if (!layers || layers->empty()) return;
const auto display = ImGui::GetIO().DisplaySize;
if (display.x <= 0 || display.y <= 0) return;
const float k = display.y / 1080.0f;
auto& s = detail::state();
auto* fallback = ImGui::GetFont();
const std::array<ImFont*, 5> fonts{s.menu.bold ? s.menu.bold : fallback, s.menu.body ? s.menu.body : fallback,
s.menu.heading ? s.menu.heading : fallback, s.menu.title ? s.menu.title : fallback,
s.menu.title_large ? s.menu.title_large : s.menu.title ? s.menu.title : fallback};
auto* draw = ImGui::GetBackgroundDrawList();
const auto camera = plugin_camera::projector(display.x, display.y);
// The fonts' advances for d:measure, again whenever the atlas is new.
static const void* measured{};
if (measured != ImGui::GetIO().Fonts) {
measured = ImGui::GetIO().Fonts;
for (int i = 0; i < 5; ++i) {
auto* font = fonts[static_cast<std::size_t>(i)];
plugin_ui::detail::FontMetrics m;
m.size = font->FontSize;
for (int c = 32; c < 128; ++c) m.advance[static_cast<std::size_t>(c - 32)] = font->GetCharAdvance(static_cast<ImWchar>(c));
plugin_ui::set_font(i, m);
}
}
using Kind = plugin_ui::Shape::Kind;
for (const auto& layer : *layers) {
int clips = 0;
for (const auto& shape : layer.shapes) {
const auto& style = shape.style;
const auto& p = shape.points;
auto* font = fonts[static_cast<std::size_t>(std::clamp(style.font, 0, 4))];
switch (shape.kind) {
case Kind::text: text(draw, font, k, at(p[0]), shape); break;
case Kind::rect:
if (style.fill) draw->AddRectFilled(at(p[0]), at(p[1]), style.colour, style.rounding * k);
else draw->AddRect(at(p[0]), at(p[1]), style.colour, style.rounding * k, 0, style.thickness * k);
break;
case Kind::line: draw->AddLine(at(p[0]), at(p[1]), style.colour, style.thickness * k); break;
case Kind::circle: circle(draw, at(p[0]), shape.radius, style, k); break;
case Kind::poly: {
std::vector<ImVec2> points;
for (const auto& point : p) points.push_back(at(point));
if (style.fill) draw->AddConvexPolyFilled(points.data(), static_cast<int>(points.size()), style.colour);
else draw->AddPolyline(points.data(), static_cast<int>(points.size()), style.colour, ImDrawFlags_Closed, style.thickness * k);
break;
}
case Kind::world_text:
if (const auto on = camera ? camera->project(p[0]) : std::nullopt) text(draw, font, k, ImVec2((*on)[0], (*on)[1]), shape);
break;
case Kind::world_line: {
const auto a = camera ? camera->project(p[0]) : std::nullopt;
const auto b = camera ? camera->project(p[1]) : std::nullopt;
if (a && b) draw->AddLine(ImVec2((*a)[0], (*a)[1]), ImVec2((*b)[0], (*b)[1]), style.colour, style.thickness * k);
break;
}
case Kind::world_circle:
if (const auto on = camera ? camera->project(p[0]) : std::nullopt)
circle(draw, ImVec2((*on)[0], (*on)[1]), std::min(shape.radius * camera->focal / (*on)[2], display.y), style, k);
break;
case Kind::image: {
ImVec2 size;
if (const auto id = detail::plugin_image(shape.text, size))
draw->AddImageRounded(id, at(p[0]), at(p[1]), ImVec2(style.uv[0], style.uv[1]), ImVec2(style.uv[2], style.uv[3]), style.colour,
style.rounding * k);
break;
}
case Kind::world_image: {
ImVec2 size;
const auto id = detail::plugin_image(shape.text, size);
const auto on = camera ? camera->project(p[0]) : std::nullopt;
if (!id || !on || size.y <= 0) break;
const float span_x = std::abs(style.uv[2] - style.uv[0]), span_y = std::abs(style.uv[3] - style.uv[1]);
const float aspect = span_y > 0 ? size.x * span_x / (size.y * span_y) : 1.0f;
const float height = std::min(shape.radius * camera->focal / (*on)[2], display.y), width = height * aspect;
draw->AddImage(id, ImVec2((*on)[0] - width * .5f, (*on)[1] - height * .5f), ImVec2((*on)[0] + width * .5f, (*on)[1] + height * .5f),
ImVec2(style.uv[0], style.uv[1]), ImVec2(style.uv[2], style.uv[3]), style.colour);
break;
}
case Kind::game_image:
(void)draw_game_image(draw, shape.text, at(p[0]), at(p[1]), style.colour, static_cast<GameImageFit>(std::clamp(shape.fit, 0, 2)),
shape.edge * k, shape.frame);
break;
case Kind::gradient:
if (shape.colours.size() == 4)
draw->AddRectFilledMultiColor(at(p[0]), at(p[1]), shape.colours[0], shape.colours[1], shape.colours[2], shape.colours[3]);
break;
case Kind::skeleton: {
if (shape.colours.size() != skater_body::count) break;
std::array<ImU32, skater_body::count> paint{};
std::copy(shape.colours.begin(), shape.colours.end(), paint.begin());
draw_xray(draw, paint, static_cast<float>(style.colour >> IM_COL32_A_SHIFT & 0xff) / 255.0f);
break;
}
case Kind::rough_rect: skate_theme::rough_rect(draw, at(p[0]), at(p[1]), style.colour, shape.seed, shape.radius * k); break;
case Kind::scribble: skate_theme::scribble(draw, at(p[0]), at(p[1]), shape.seed, shape.radius * k); break;
case Kind::halftone: theme::halftone(draw, at(p[0]), shape.columns, shape.rows, style.colour); break;
case Kind::clip:
if (clips >= 16) break;
draw->PushClipRect(at(p[0]), at(p[1]), true);
++clips;
break;
case Kind::unclip:
if (clips > 0) {
draw->PopClipRect();
--clips;
}
break;
}
}
for (; clips > 0; --clips) draw->PopClipRect();
}
}
}
Reference in new issue View Git Blame Copy Permalink