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.CatModel;
import net.minecraft.client.model.geom.EntityModelSet;
import net.minecraft.client.model.geom.ModelLayers;
import net.minecraft.client.renderer.MultiBufferSource;
import net.minecraft.client.renderer.entity.RenderLayerParent;
import net.minecraft.client.renderer.entity.state.CatRenderState;
import net.minecraft.resources.ResourceLocation;
import net.minecraft.world.item.DyeColor;

@Environment(EnvType.CLIENT)
public class CatCollarLayer extends RenderLayer<CatRenderState, CatModel> {
	private static final ResourceLocation CAT_COLLAR_LOCATION = ResourceLocation.withDefaultNamespace("textures/entity/cat/cat_collar.png");
	private final CatModel adultModel;
	private final CatModel babyModel;

	public CatCollarLayer(RenderLayerParent<CatRenderState, CatModel> renderer, EntityModelSet modelSet) {
		super(renderer);
		this.adultModel = new CatModel(modelSet.bakeLayer(ModelLayers.CAT_COLLAR));
		this.babyModel = new CatModel(modelSet.bakeLayer(ModelLayers.CAT_BABY_COLLAR));
	}

	public void render(PoseStack poseStack, MultiBufferSource bufferSource, int packedLight, CatRenderState renderState, float yRot, float xRot) {
		DyeColor dyeColor = renderState.collarColor;
		if (dyeColor != null) {
			int i = dyeColor.getTextureDiffuseColor();
			CatModel catModel = renderState.isBaby ? this.babyModel : this.adultModel;
			coloredCutoutModelCopyLayerRender(catModel, CAT_COLLAR_LOCATION, poseStack, bufferSource, packedLight, renderState, i);
		}
	}
}