aboutsummaryrefslogtreecommitdiff
path: root/src/texturePacks/java/moe/nea/firmament/mixins/custommodels/ReplaceHeadModel.java
blob: 1b9bc81fdf6302e2fe6713f76f35ac7d12af2589 (plain)
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
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
package moe.nea.firmament.mixins.custommodels;

import moe.nea.firmament.features.texturepack.HeadModelChooser;
import net.minecraft.client.renderer.item.ItemModelResolver;
import net.minecraft.client.renderer.entity.LivingEntityRenderer;
import net.minecraft.client.model.EntityModel;
import net.minecraft.client.renderer.entity.state.LivingEntityRenderState;
import net.minecraft.client.renderer.item.ItemStackRenderState;
import net.minecraft.world.entity.EquipmentSlot;
import net.minecraft.world.entity.LivingEntity;
import net.minecraft.world.item.ItemDisplayContext;
import org.spongepowered.asm.mixin.Final;
import org.spongepowered.asm.mixin.Mixin;
import org.spongepowered.asm.mixin.Shadow;
import org.spongepowered.asm.mixin.Unique;
import org.spongepowered.asm.mixin.injection.At;
import org.spongepowered.asm.mixin.injection.Inject;
import org.spongepowered.asm.mixin.injection.callback.CallbackInfo;

@Mixin(LivingEntityRenderer.class)
public class ReplaceHeadModel<T extends LivingEntity, S extends LivingEntityRenderState, M extends EntityModel<? super S>> {
	@Shadow
	@Final
	protected ItemModelResolver itemModelResolver;

	@Unique
	private ItemStackRenderState tempRenderState = new ItemStackRenderState();

	@Inject(
		method = "extractRenderState(Lnet/minecraft/world/entity/LivingEntity;Lnet/minecraft/client/renderer/entity/state/LivingEntityRenderState;F)V",
		at = @At("TAIL")
	)
	private void replaceHeadModel(
		T livingEntity, S livingEntityRenderState, float f, CallbackInfo ci
	) {
		var headItemStack = livingEntity.getItemBySlot(EquipmentSlot.HEAD);

		HeadModelChooser.INSTANCE.getIS_CHOOSING_HEAD_MODEL().set(true);
		tempRenderState.clear();
		this.itemModelResolver.updateForLiving(tempRenderState, headItemStack, ItemDisplayContext.HEAD, livingEntity);
		HeadModelChooser.INSTANCE.getIS_CHOOSING_HEAD_MODEL().set(false);

		if (HeadModelChooser.HasExplicitHeadModelMarker.cast(tempRenderState)
			.isExplicitHeadModel_Firmament()) {
			livingEntityRenderState.wornHeadType = null;
			var temp = livingEntityRenderState.headItem;
			livingEntityRenderState.headItem = tempRenderState;
			tempRenderState = temp;
		}
	}
}