diff --git a/src/main/java/com/minelittlepony/client/render/entity/feature/AbstractPonyFeature.java b/src/main/java/com/minelittlepony/client/render/entity/feature/AbstractPonyFeature.java index 8643ef28..ec5c9c35 100644 --- a/src/main/java/com/minelittlepony/client/render/entity/feature/AbstractPonyFeature.java +++ b/src/main/java/com/minelittlepony/client/render/entity/feature/AbstractPonyFeature.java @@ -43,6 +43,11 @@ public abstract class AbstractPonyFeature getModelWrapper() { return context.getInternalRenderer().getModelWrapper(); } diff --git a/src/main/java/com/minelittlepony/client/render/entity/feature/HeldItemFeature.java b/src/main/java/com/minelittlepony/client/render/entity/feature/HeldItemFeature.java index d4335033..35b6c063 100644 --- a/src/main/java/com/minelittlepony/client/render/entity/feature/HeldItemFeature.java +++ b/src/main/java/com/minelittlepony/client/render/entity/feature/HeldItemFeature.java @@ -2,6 +2,7 @@ package com.minelittlepony.client.render.entity.feature; import com.minelittlepony.api.model.BodyPart; import com.minelittlepony.client.model.IPonyModel; +import com.minelittlepony.client.render.IPonyRenderContext; import net.minecraft.client.render.VertexConsumerProvider; import net.minecraft.client.render.entity.feature.FeatureRendererContext; @@ -17,8 +18,12 @@ import net.minecraft.util.Arm; public class HeldItemFeature & IPonyModel & ModelWithArms> extends HeldItemFeatureRenderer { - public HeldItemFeature(FeatureRendererContext context, HeldItemRenderer renderer) { - super(context, renderer); + private final IPonyRenderContext context; + + @SuppressWarnings("unchecked") + public HeldItemFeature(IPonyRenderContext context, HeldItemRenderer renderer) { + super((FeatureRendererContext)context, renderer); + this.context = context; } protected ItemStack getLeftItem(T entity) { @@ -39,7 +44,7 @@ public class HeldItemFeature & ItemStack right = getRightItem(entity); if (!left.isEmpty() || !right.isEmpty()) { - M model = getContextModel(); + M model = context.getInternalRenderer().getModel(); stack.push(); diff --git a/src/main/java/com/minelittlepony/client/render/entity/feature/IllagerHeldItemFeature.java b/src/main/java/com/minelittlepony/client/render/entity/feature/IllagerHeldItemFeature.java index 1ef001fd..ad39e485 100644 --- a/src/main/java/com/minelittlepony/client/render/entity/feature/IllagerHeldItemFeature.java +++ b/src/main/java/com/minelittlepony/client/render/entity/feature/IllagerHeldItemFeature.java @@ -1,16 +1,16 @@ package com.minelittlepony.client.render.entity.feature; import net.minecraft.client.render.VertexConsumerProvider; -import net.minecraft.client.render.entity.feature.FeatureRendererContext; import net.minecraft.client.render.item.HeldItemRenderer; import net.minecraft.client.util.math.MatrixStack; import net.minecraft.entity.mob.IllagerEntity; import com.minelittlepony.client.model.entity.race.AlicornModel; +import com.minelittlepony.client.render.IPonyRenderContext; public class IllagerHeldItemFeature> extends HeldItemFeature { - public IllagerHeldItemFeature(FeatureRendererContext livingPony, HeldItemRenderer renderer) { + public IllagerHeldItemFeature(IPonyRenderContext livingPony, HeldItemRenderer renderer) { super(livingPony, renderer); }