fix: fixed debug rendering for pathfinding nodes

AlexModGuy · Dec 22, 2023 · c7f0303 · c7f0303
1 parent d98aa96
commit c7f0303
Show file tree

Hide file tree

Showing 10 changed files with 1,998 additions and 313 deletions.
diff --git a/src/main/java/com/github/alexthe666/iceandfire/client/render/pathfinding/MRenderTypes.java b/src/main/java/com/github/alexthe666/iceandfire/client/render/pathfinding/MRenderTypes.java
diff --git a/.../com/github/alexthe666/iceandfire/client/render/pathfinding/PathfindingDebugRenderer.java b/.../com/github/alexthe666/iceandfire/client/render/pathfinding/PathfindingDebugRenderer.java
@@ -0,0 +1,121 @@
+package com.github.alexthe666.iceandfire.client.render.pathfinding;
+
+import com.github.alexthe666.iceandfire.IceAndFire;
+import com.github.alexthe666.iceandfire.pathfinding.raycoms.MNode;
+import com.github.alexthe666.iceandfire.pathfinding.raycoms.WorldEventContext;
+import com.github.alexthe666.iceandfire.pathfinding.raycoms.WorldRenderMacros;
+import com.mojang.blaze3d.vertex.VertexConsumer;
+import net.minecraft.client.Minecraft;
+import net.minecraft.client.gui.Font;
+import net.minecraft.client.renderer.MultiBufferSource;
+import net.minecraft.client.renderer.RenderBuffers;
+import net.minecraft.core.BlockPos;
+import net.minecraft.world.entity.Entity;
+import org.jetbrains.annotations.NotNull;
+import org.joml.Matrix4f;
+
+import java.util.ConcurrentModificationException;
+import java.util.HashSet;
+import java.util.Set;
+
+public class PathfindingDebugRenderer {
+    public static final RenderBuffers renderBuffers = new RenderBuffers();
+    private static final MultiBufferSource.BufferSource renderBuffer = renderBuffers.bufferSource();
+    /**
+     * Set of visited nodes.
+     */
+    public static Set<MNode> lastDebugNodesVisited = new HashSet<>();
+
+    /**
+     * Set of not visited nodes.
+     */
+    public static Set<MNode> lastDebugNodesNotVisited = new HashSet<>();
+
+    /**
+     * Set of nodes that belong to the chosen path.
+     */
+    public static Set<MNode> lastDebugNodesPath = new HashSet<>();
+
+    /**
+     * Render debugging information for the pathfinding system.
+     *
+     * @param ctx rendering context
+     */
+    public static void render(final WorldEventContext ctx) {
+        try {
+            for (final MNode n : lastDebugNodesVisited) {
+                debugDrawNode(n, 0xffff0000, ctx);
+            }
+
+            for (final MNode n : lastDebugNodesNotVisited) {
+                debugDrawNode(n, 0xff0000ff, ctx);
+            }
+
+            for (final MNode n : lastDebugNodesPath) {
+                if (n.isReachedByWorker()) {
+                    debugDrawNode(n, 0xffff6600, ctx);
+                } else {
+                    debugDrawNode(n, 0xff00ff00, ctx);
+                }
+            }
+        } catch (final ConcurrentModificationException exc) {
+            IceAndFire.LOGGER.catching(exc);
+        }
+    }
+
+    private static void debugDrawNode(final MNode n, final int argbColor, final WorldEventContext ctx) {
+        ctx.poseStack.pushPose();
+        ctx.poseStack.translate(n.pos.getX() + 0.375d, n.pos.getY() + 0.375d, n.pos.getZ() + 0.375d);
+
+        final Entity entity = Minecraft.getInstance().getCameraEntity();
+        if (n.pos.closerThan(entity.blockPosition(), 5d)) {
+            renderDebugText(n, ctx);
+        }
+
+        ctx.poseStack.scale(0.25F, 0.25F, 0.25F);
+
+        WorldRenderMacros.renderBox(ctx.bufferSource, ctx.poseStack, BlockPos.ZERO, BlockPos.ZERO, argbColor);
+
+        if (n.parent != null) {
+            final Matrix4f lineMatrix = ctx.poseStack.last().pose();
+
+            final float pdx = n.parent.pos.getX() - n.pos.getX() + 0.125f;
+            final float pdy = n.parent.pos.getY() - n.pos.getY() + 0.125f;
+            final float pdz = n.parent.pos.getZ() - n.pos.getZ() + 0.125f;
+
+            final VertexConsumer buffer = ctx.bufferSource.getBuffer(WorldRenderMacros.LINES);
+
+            buffer.vertex(lineMatrix, 0.5f, 0.5f, 0.5f).color(0.75F, 0.75F, 0.75F, 1.0F).endVertex();
+            buffer.vertex(lineMatrix, pdx / 0.25f, pdy / 0.25f, pdz / 0.25f).color(0.75F, 0.75F, 0.75F, 1.0F).endVertex();
+        }
+
+        ctx.poseStack.popPose();
+    }
+
+    private static void renderDebugText(@NotNull final MNode n, final WorldEventContext ctx) {
+        final Font fontrenderer = Minecraft.getInstance().font;
+
+        final String s1 = String.format("F: %.3f [%d]", n.getCost(), n.getCounterAdded());
+        final String s2 = String.format("G: %.3f [%d]", n.getScore(), n.getCounterVisited());
+        final int i = Math.max(fontrenderer.width(s1), fontrenderer.width(s2)) / 2;
+
+        ctx.poseStack.pushPose();
+        ctx.poseStack.translate(0.0F, 0.75F, 0.0F);
+
+        ctx.poseStack.mulPose(Minecraft.getInstance().getEntityRenderDispatcher().cameraOrientation());
+        ctx.poseStack.scale(-0.014F, -0.014F, 0.014F);
+        ctx.poseStack.translate(0.0F, 18F, 0.0F);
+        final Matrix4f mat = ctx.poseStack.last().pose();
+
+        WorldRenderMacros.renderFillRectangle(ctx.bufferSource, ctx.poseStack, -i - 1, -5, 0, 2 * i + 2, 17, 0x7f000000);
+
+        ctx.poseStack.translate(0.0F, -5F, -0.1F);
+        fontrenderer.drawInBatch(s1, -fontrenderer.width(s1) / 2.0f, 1, 0xFFFFFFFF, false, mat, ctx.bufferSource, Font.DisplayMode.NORMAL, 0, 15728880);
+        ctx.poseStack.translate(0.0F, 8F, -0.1F);
+        fontrenderer.drawInBatch(s2, -fontrenderer.width(s2) / 2.0f, 1, 0xFFFFFFFF, false, mat, ctx.bufferSource, Font.DisplayMode.NORMAL, 0, 15728880);
+
+        ctx.poseStack.popPose();
+    }
+
+
+}