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

import com.mojang.blaze3d.vertex.PoseStack;
import net.fabricmc.api.EnvType;
import net.fabricmc.api.Environment;
import net.minecraft.client.model.DolphinModel;
import net.minecraft.client.renderer.MultiBufferSource;
import net.minecraft.client.renderer.entity.ItemRenderer;
import net.minecraft.client.renderer.entity.RenderLayerParent;
import net.minecraft.client.renderer.entity.state.DolphinRenderState;
import net.minecraft.client.renderer.texture.OverlayTexture;
import net.minecraft.client.resources.model.BakedModel;
import net.minecraft.util.Mth;
import net.minecraft.world.item.ItemDisplayContext;
import net.minecraft.world.item.ItemStack;

@Environment(EnvType.CLIENT)
public class DolphinCarryingItemLayer extends RenderLayer<DolphinRenderState, DolphinModel> {
	private final ItemRenderer itemRenderer;

	public DolphinCarryingItemLayer(RenderLayerParent<DolphinRenderState, DolphinModel> renderer, ItemRenderer itemRenderer) {
		super(renderer);
		this.itemRenderer = itemRenderer;
	}

	public void render(PoseStack poseStack, MultiBufferSource bufferSource, int packedLight, DolphinRenderState renderState, float yRot, float xRot) {
		ItemStack itemStack = renderState.getMainHandItem();
		BakedModel bakedModel = renderState.getMainHandItemModel();
		if (bakedModel != null) {
			poseStack.pushPose();
			float f = 1.0F;
			float g = -1.0F;
			float h = Mth.abs(renderState.xRot) / 60.0F;
			if (renderState.xRot < 0.0F) {
				poseStack.translate(0.0F, 1.0F - h * 0.5F, -1.0F + h * 0.5F);
			} else {
				poseStack.translate(0.0F, 1.0F + h * 0.8F, -1.0F + h * 0.2F);
			}

			this.itemRenderer.render(itemStack, ItemDisplayContext.GROUND, false, poseStack, bufferSource, packedLight, OverlayTexture.NO_OVERLAY, bakedModel);
			poseStack.popPose();
		}
	}
}