fix: special element renderers / circle renderers

1 files changed, 68 insertions, 0 deletions

diff --git a/src/main/java/moe/nea/firmament/mixins/render/renderer/MultipleSpecialGuiRenderStates.java b/src/main/java/moe/nea/firmament/mixins/render/renderer/MultipleSpecialGuiRenderStates.java
new file mode 100644
index 0000000..cc79591
--- /dev/null
+++ b/src/main/java/moe/nea/firmament/mixins/render/renderer/MultipleSpecialGuiRenderStates.java
@@ -0,0 +1,68 @@
+/*
+ * SPDX-License-Identifier: LGPL-3.0-or-later
+ * SPDX-FileCopyrightText: 2025 azureaaron via Skyblocker
+ */
+
+package moe.nea.firmament.mixins.render.renderer;
+
+import com.mojang.blaze3d.buffers.GpuBufferSlice;
+import moe.nea.firmament.util.render.MultiSpecialGuiRenderState;
+import moe.nea.firmament.util.render.MultiSpecialGuiRenderer;
+import net.minecraft.client.gui.render.GuiRenderer;
+import net.minecraft.client.gui.render.SpecialGuiElementRenderer;
+import net.minecraft.client.gui.render.state.GuiRenderState;
+import net.minecraft.client.gui.render.state.special.SpecialGuiElementRenderState;
+import net.minecraft.client.render.VertexConsumerProvider;
+import org.spongepowered.asm.mixin.Final;
+import org.spongepowered.asm.mixin.Mixin;
+import org.spongepowered.asm.mixin.Shadow;
+import org.spongepowered.asm.mixin.Unique;
+import org.spongepowered.asm.mixin.injection.At;
+import org.spongepowered.asm.mixin.injection.Inject;
+import org.spongepowered.asm.mixin.injection.callback.CallbackInfo;
+
+import java.util.HashMap;
+import java.util.Map;
+
+/**
+ * The structure of this class was roughly taken from SkyBlocker, retrieved 29.07.2025
+ */
+@Mixin(GuiRenderer.class)
+public class MultipleSpecialGuiRenderStates {
+	@Shadow
+	@Final
+	private VertexConsumerProvider.Immediate vertexConsumers;
+	@Shadow
+	@Final
+	GuiRenderState state;
+	@Unique
+	Map<MultiSpecialGuiRenderState, MultiSpecialGuiRenderer<?>> multiRenderers = new HashMap<>();
+
+	@Inject(method = "prepareSpecialElement", at = @At("HEAD"), cancellable = true)
+	private <T extends SpecialGuiElementRenderState> void onPrepareElement(T elementState, int windowScaleFactor, CallbackInfo ci) {
+		if (elementState instanceof MultiSpecialGuiRenderState multiState) {
+			@SuppressWarnings({"resource", "unchecked"})
+			var renderer = (SpecialGuiElementRenderer<T>) multiRenderers
+				.computeIfAbsent(multiState, elementState$ -> elementState$.createRenderer(this.vertexConsumers));
+			renderer.render(elementState, state, windowScaleFactor);
+			ci.cancel();
+		}
+	}
+
+	@Inject(method = "close", at = @At("TAIL"))
+	private void onClose(CallbackInfo ci) {
+		multiRenderers.values().forEach(SpecialGuiElementRenderer::close);
+	}
+
+	@Inject(method = "render(Lcom/mojang/blaze3d/buffers/GpuBufferSlice;)V", at = @At(value = "INVOKE", target = "Lnet/minecraft/client/gui/render/GuiRenderer;clearOversizedItems()V"))
+	private void onAfterRender(GpuBufferSlice fogBuffer, CallbackInfo ci) {
+		multiRenderers.values().removeIf(it -> {
+			if (it.consumeRender()) {
+				return false;
+			} else {
+				it.close();
+				return true;
+			}
+		});
+	}
+}