arrow layer fix

Author: ImToggle

src/main/java/org/polyfrost/polyhitbox/mixin/LayerArrowMixin.java

@@ -0,0 +1,24 @@
+package org.polyfrost.polyhitbox.mixin;
+
+import net.minecraft.client.renderer.entity.layers.LayerArrow;
+import net.minecraft.entity.EntityLivingBase;
+import org.polyfrost.polyhitbox.render.HitboxRenderer;
+import org.spongepowered.asm.mixin.Mixin;
+import org.spongepowered.asm.mixin.injection.At;
+import org.spongepowered.asm.mixin.injection.Inject;
+import org.spongepowered.asm.mixin.injection.callback.CallbackInfo;
+
+@Mixin(LayerArrow.class)
+public class LayerArrowMixin {
+
+    @Inject(method = "doRenderLayer", at = @At(value = "INVOKE", target = "Lnet/minecraft/client/renderer/RenderHelper;disableStandardItemLighting()V"))
+    private void start(EntityLivingBase entitylivingbaseIn, float f, float g, float partialTicks, float h, float i, float j, float scale, CallbackInfo ci) {
+        HitboxRenderer.INSTANCE.setDrawingLayer(true);
+    }
+
+    @Inject(method = "doRenderLayer", at = @At(value = "TAIL"))
+    private void end(EntityLivingBase entitylivingbaseIn, float f, float g, float partialTicks, float h, float i, float j, float scale, CallbackInfo ci) {
+        HitboxRenderer.INSTANCE.setDrawingLayer(false);
+    }
+
+}

src/main/kotlin/org/polyfrost/polyhitbox/render/HitboxRenderer.kt

@@ -7,9 +7,6 @@ import net.minecraft.client.renderer.vertex.DefaultVertexFormats
 import net.minecraft.entity.Entity
 import net.minecraft.util.AxisAlignedBB
 import net.minecraft.util.MovingObjectPosition
-import net.minecraftforge.client.event.RenderWorldLastEvent
-import net.minecraftforge.common.MinecraftForge
-import net.minecraftforge.fml.common.eventhandler.SubscribeEvent
 import org.lwjgl.opengl.GL11
 import org.polyfrost.polyhitbox.config.HitboxConfig
 import org.polyfrost.polyhitbox.config.ModConfig
@@ -26,6 +23,8 @@ object HitboxRenderer {
 
     var drawingWorld = false
 
+    var drawingLayer = false
+
     fun onRender() {
         if (!ModConfig.enabled) return
         if (renderQueue.isEmpty()) return
@@ -46,7 +45,7 @@ object HitboxRenderer {
     }
 
     fun tryAddToQueue(config: HitboxConfig, entity: Entity, x: Double, y: Double, z: Double, partialTicks: Float) {
-        if (drawingWorld) {
+        if (drawingWorld && !drawingLayer) {
             renderQueue.add(RenderInfo(config, entity, x, y, z, partialTicks))
         } else {
             renderHitbox(config, entity, x, y, z, partialTicks)

src/main/resources/mixins.polyhitbox.json

@@ -8,6 +8,7 @@
     "EntityMixin",
     "EntityRendererMixin",
     "GuiInventoryMixin",
+    "LayerArrowMixin",
     "RenderManagerMixin"
   ]
 }