Use the correct entity renderer when culling leashed entities

Author: FiniteReality

src/main/java/org/embeddedt/embeddium/impl/mixin/features/render/entity/cull/EntityRendererMixin.java

@@ -2,18 +2,25 @@
 
 import com.llamalad7.mixinextras.injector.ModifyExpressionValue;
 import com.llamalad7.mixinextras.sugar.Local;
+import net.minecraft.client.renderer.entity.EntityRenderDispatcher;
 import net.minecraft.world.entity.Leashable;
 import org.embeddedt.embeddium.impl.render.EmbeddiumWorldRenderer;
 import net.minecraft.client.renderer.culling.Frustum;
 import net.minecraft.client.renderer.entity.EntityRenderer;
 import net.minecraft.world.entity.Entity;
+import org.spongepowered.asm.mixin.Final;
 import org.spongepowered.asm.mixin.Mixin;
+import org.spongepowered.asm.mixin.Shadow;
 import org.spongepowered.asm.mixin.injection.At;
 import org.spongepowered.asm.mixin.injection.Inject;
 import org.spongepowered.asm.mixin.injection.callback.CallbackInfoReturnable;
 
 @Mixin(EntityRenderer.class)
 public abstract class EntityRendererMixin<T extends Entity> {
+    @Shadow
+    @Final
+    protected EntityRenderDispatcher entityRenderDispatcher;
+
     @ModifyExpressionValue(method = "shouldRender", at = @At(value = "INVOKE", target = "Lnet/minecraft/client/renderer/culling/Frustum;isVisible(Lnet/minecraft/world/phys/AABB;)Z", ordinal = 0))
     private boolean checkSectionForCullingMain(boolean isWithinFrustum, @Local(ordinal = 0, argsOnly = true) T entity) {
         if(!isWithinFrustum) {
@@ -36,7 +43,9 @@ private boolean checkSectionForCullingMain(boolean isWithinFrustum, @Local(ordin
         // Check if the entity is in a visible chunk section
 
         var renderer = EmbeddiumWorldRenderer.instanceNullable();
+        //noinspection DataFlowIssue - Vanilla has already checked getLeashHolder is non-null.
+        var leashHolderRenderer = this.entityRenderDispatcher.getRenderer(leashable.getLeashHolder());
 
-        return renderer == null || renderer.isEntityVisible(leashable.getLeashHolder(), (EntityRenderer)(Object)this);
+        return renderer == null || renderer.isEntityVisible(leashable.getLeashHolder(), leashHolderRenderer);
     }
 }