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

import com.mojang.blaze3d.vertex.PoseStack;
import com.mojang.blaze3d.vertex.VertexConsumer;
import net.fabricmc.api.EnvType;
import net.fabricmc.api.Environment;
import net.minecraft.client.model.EntityModel;
import net.minecraft.client.renderer.MultiBufferSource;
import net.minecraft.client.renderer.RenderType;
import net.minecraft.client.renderer.entity.RenderLayerParent;
import net.minecraft.client.renderer.entity.state.LivingEntityRenderState;
import net.minecraft.client.renderer.entity.state.SaddleableRenderState;
import net.minecraft.client.renderer.texture.OverlayTexture;
import net.minecraft.resources.ResourceLocation;

@Environment(EnvType.CLIENT)
public class SaddleLayer<S extends LivingEntityRenderState & SaddleableRenderState, M extends EntityModel<? super S>> extends RenderLayer<S, M> {
	private final ResourceLocation textureLocation;
	private final M adultModel;
	private final M babyModel;

	public SaddleLayer(RenderLayerParent<S, M> renderer, M adultModel, M babyModel, ResourceLocation textureLocation) {
		super(renderer);
		this.adultModel = adultModel;
		this.babyModel = babyModel;
		this.textureLocation = textureLocation;
	}

	public SaddleLayer(RenderLayerParent<S, M> renderer, M model, ResourceLocation textureLocation) {
		this(renderer, model, model, textureLocation);
	}

	public void render(PoseStack poseStack, MultiBufferSource bufferSource, int packedLight, S renderState, float yRot, float xRot) {
		if (renderState.isSaddled()) {
			M entityModel = renderState.isBaby ? this.babyModel : this.adultModel;
			entityModel.setupAnim(renderState);
			VertexConsumer vertexConsumer = bufferSource.getBuffer(RenderType.entityCutoutNoCull(this.textureLocation));
			entityModel.renderToBuffer(poseStack, vertexConsumer, packedLight, OverlayTexture.NO_OVERLAY);
		}
	}
}