src/main/java/moe/nea/firmament/mixins/custommodels/ApplyHeadModelInItemRenderer.java


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35

package moe.nea.firmament.mixins.custommodels;

import com.llamalad7.mixinextras.sugar.Local;
import com.llamalad7.mixinextras.sugar.ref.LocalRef;
import moe.nea.firmament.features.texturepack.BakedModelExtra;
import net.minecraft.client.render.VertexConsumerProvider;
import net.minecraft.client.render.item.ItemRenderer;
import net.minecraft.client.render.model.BakedModel;
import net.minecraft.client.util.math.MatrixStack;
import net.minecraft.item.ItemStack;
import net.minecraft.item.ModelTransformationMode;
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(ItemRenderer.class)
public class ApplyHeadModelInItemRenderer {
	@Inject(method = "renderItem(Lnet/minecraft/item/ItemStack;Lnet/minecraft/item/ModelTransformationMode;ZLnet/minecraft/client/util/math/MatrixStack;Lnet/minecraft/client/render/VertexConsumerProvider;IILnet/minecraft/client/render/model/BakedModel;ZF)V",
		at = @At("HEAD"))
	private void applyHeadModel(ItemStack stack, ModelTransformationMode transformationMode, boolean leftHanded,
	                            MatrixStack matrices, VertexConsumerProvider vertexConsumers, int light, int overlay,
	                            BakedModel model, boolean useInventoryModel, float z, CallbackInfo ci,
	                            @Local(argsOnly = true) LocalRef<BakedModel> modelMut
	) {
		if (transformationMode == ModelTransformationMode.HEAD
			&& model instanceof BakedModelExtra extra) {
			var headModel = extra.getHeadModel_firmament();
			if (headModel != null) {
				modelMut.set(headModel);
			}
		}
	}
}