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.animal.parrot.ParrotModel;
import net.minecraft.client.model.geom.EntityModelSet;
import net.minecraft.client.model.geom.ModelLayers;
import net.minecraft.client.model.player.PlayerModel;
import net.minecraft.client.renderer.SubmitNodeCollector;
import net.minecraft.client.renderer.entity.ParrotRenderer;
import net.minecraft.client.renderer.entity.RenderLayerParent;
import net.minecraft.client.renderer.entity.state.AvatarRenderState;
import net.minecraft.client.renderer.entity.state.ParrotRenderState;
import net.minecraft.client.renderer.texture.OverlayTexture;
import net.minecraft.world.entity.animal.parrot.Parrot;

@Environment(EnvType.CLIENT)
public class ParrotOnShoulderLayer extends RenderLayer<AvatarRenderState, PlayerModel> {
	private final ParrotModel model;

	public ParrotOnShoulderLayer(final RenderLayerParent<AvatarRenderState, PlayerModel> renderer, final EntityModelSet modelSet) {
		super(renderer);
		this.model = new ParrotModel(modelSet.bakeLayer(ModelLayers.PARROT));
	}

	public void submit(
		final PoseStack poseStack,
		final SubmitNodeCollector submitNodeCollector,
		final int lightCoords,
		final AvatarRenderState state,
		final float yRot,
		final float xRot
	) {
		Parrot.Variant parrotOnLeftShoulder = state.parrotOnLeftShoulder;
		if (parrotOnLeftShoulder != null) {
			this.submitOnShoulder(poseStack, submitNodeCollector, lightCoords, state, parrotOnLeftShoulder, yRot, xRot, true);
		}

		Parrot.Variant parrotOnRightShoulder = state.parrotOnRightShoulder;
		if (parrotOnRightShoulder != null) {
			this.submitOnShoulder(poseStack, submitNodeCollector, lightCoords, state, parrotOnRightShoulder, yRot, xRot, false);
		}
	}

	private void submitOnShoulder(
		final PoseStack poseStack,
		final SubmitNodeCollector submitNodeCollector,
		final int lightCoords,
		final AvatarRenderState playerState,
		final Parrot.Variant parrotVariant,
		final float yRot,
		final float xRot,
		final boolean isLeft
	) {
		poseStack.pushPose();
		poseStack.translate(isLeft ? 0.4F : -0.4F, playerState.isCrouching ? -1.3F : -1.5F, 0.0F);
		ParrotRenderState parrotState = new ParrotRenderState();
		parrotState.pose = ParrotModel.Pose.ON_SHOULDER;
		parrotState.ageInTicks = playerState.ageInTicks;
		parrotState.walkAnimationPos = playerState.walkAnimationPos;
		parrotState.walkAnimationSpeed = playerState.walkAnimationSpeed;
		parrotState.yRot = yRot;
		parrotState.xRot = xRot;
		submitNodeCollector.submitModel(
			this.model, parrotState, poseStack, ParrotRenderer.getVariantTexture(parrotVariant), lightCoords, OverlayTexture.NO_OVERLAY, playerState.outlineColor, null
		);
		poseStack.popPose();
	}
}