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

import com.google.common.collect.Maps;
import com.mojang.blaze3d.vertex.PoseStack;
import java.util.Map;
import net.fabricmc.api.EnvType;
import net.fabricmc.api.Environment;
import net.minecraft.client.model.animal.equine.HorseModel;
import net.minecraft.client.renderer.SubmitNodeCollector;
import net.minecraft.client.renderer.entity.LivingEntityRenderer;
import net.minecraft.client.renderer.entity.RenderLayerParent;
import net.minecraft.client.renderer.entity.state.HorseRenderState;
import net.minecraft.client.renderer.rendertype.RenderTypes;
import net.minecraft.resources.Identifier;
import net.minecraft.world.entity.animal.equine.Markings;

@Environment(EnvType.CLIENT)
public class HorseMarkingLayer extends RenderLayer<HorseRenderState, HorseModel> {
	private static final Identifier INVISIBLE_TEXTURE = Identifier.withDefaultNamespace("invisible");
	private static final Map<Markings, HorseMarkingLayer.HorseMarkingTextures> LOCATION_BY_MARKINGS = Maps.newEnumMap(
		Map.of(
			Markings.NONE,
			new HorseMarkingLayer.HorseMarkingTextures(INVISIBLE_TEXTURE, INVISIBLE_TEXTURE),
			Markings.WHITE,
			new HorseMarkingLayer.HorseMarkingTextures(
				Identifier.withDefaultNamespace("textures/entity/horse/horse_markings_white.png"),
				Identifier.withDefaultNamespace("textures/entity/horse/horse_markings_white_baby.png")
			),
			Markings.WHITE_FIELD,
			new HorseMarkingLayer.HorseMarkingTextures(
				Identifier.withDefaultNamespace("textures/entity/horse/horse_markings_whitefield.png"),
				Identifier.withDefaultNamespace("textures/entity/horse/horse_markings_whitefield_baby.png")
			),
			Markings.WHITE_DOTS,
			new HorseMarkingLayer.HorseMarkingTextures(
				Identifier.withDefaultNamespace("textures/entity/horse/horse_markings_whitedots.png"),
				Identifier.withDefaultNamespace("textures/entity/horse/horse_markings_whitedots_baby.png")
			),
			Markings.BLACK_DOTS,
			new HorseMarkingLayer.HorseMarkingTextures(
				Identifier.withDefaultNamespace("textures/entity/horse/horse_markings_blackdots.png"),
				Identifier.withDefaultNamespace("textures/entity/horse/horse_markings_blackdots_baby.png")
			)
		)
	);

	public HorseMarkingLayer(final RenderLayerParent<HorseRenderState, HorseModel> renderer) {
		super(renderer);
	}

	public void submit(
		final PoseStack poseStack,
		final SubmitNodeCollector submitNodeCollector,
		final int lightCoords,
		final HorseRenderState state,
		final float yRot,
		final float xRot
	) {
		HorseMarkingLayer.HorseMarkingTextures variant = (HorseMarkingLayer.HorseMarkingTextures)LOCATION_BY_MARKINGS.get(state.markings);
		Identifier texture = state.isBaby ? variant.baby : variant.adult;
		if (texture != INVISIBLE_TEXTURE && !state.isInvisible) {
			submitNodeCollector.order(1)
				.submitModel(
					this.getParentModel(),
					state,
					poseStack,
					RenderTypes.entityTranslucent(texture),
					lightCoords,
					LivingEntityRenderer.getOverlayCoords(state, 0.0F),
					state.outlineColor,
					null
				);
		}
	}

	@Environment(EnvType.CLIENT)
	private record HorseMarkingTextures(Identifier adult, Identifier baby) {
	}
}