package net.minecraft.client.renderer.entity.layers;

import com.mojang.blaze3d.vertex.PoseStack;
import com.mojang.math.Axis;
import net.fabricmc.api.EnvType;
import net.fabricmc.api.Environment;
import net.minecraft.client.model.WitchModel;
import net.minecraft.client.renderer.ItemInHandRenderer;
import net.minecraft.client.renderer.MultiBufferSource;
import net.minecraft.client.renderer.entity.RenderLayerParent;
import net.minecraft.world.entity.LivingEntity;
import net.minecraft.world.item.ItemStack;
import net.minecraft.world.item.Items;

@Environment(EnvType.CLIENT)
public class WitchItemLayer<T extends LivingEntity> extends CrossedArmsItemLayer<T, WitchModel<T>> {
	public WitchItemLayer(RenderLayerParent<T, WitchModel<T>> renderer, ItemInHandRenderer itemInHandRenderer) {
		super(renderer, itemInHandRenderer);
	}

	@Override
	public void render(
		PoseStack poseStack,
		MultiBufferSource buffer,
		int packedLight,
		T livingEntity,
		float limbSwing,
		float limbSwingAmount,
		float partialTicks,
		float ageInTicks,
		float netHeadYaw,
		float headPitch
	) {
		ItemStack itemStack = livingEntity.getMainHandItem();
		poseStack.pushPose();
		if (itemStack.is(Items.POTION)) {
			this.getParentModel().getHead().translateAndRotate(poseStack);
			this.getParentModel().getNose().translateAndRotate(poseStack);
			poseStack.translate(0.0625F, 0.25F, 0.0F);
			poseStack.mulPose(Axis.ZP.rotationDegrees(180.0F));
			poseStack.mulPose(Axis.XP.rotationDegrees(140.0F));
			poseStack.mulPose(Axis.ZP.rotationDegrees(10.0F));
			poseStack.translate(0.0F, -0.4F, 0.4F);
		}

		super.render(poseStack, buffer, packedLight, livingEntity, limbSwing, limbSwingAmount, partialTicks, ageInTicks, netHeadYaw, headPitch);
		poseStack.popPose();
	}
}