package net.minecraft.client.renderer.entity;

import net.fabricmc.api.EnvType;
import net.fabricmc.api.Environment;
import net.minecraft.client.model.WardenModel;
import net.minecraft.client.model.geom.ModelLayers;
import net.minecraft.client.renderer.entity.layers.WardenEmissiveLayer;
import net.minecraft.resources.ResourceLocation;
import net.minecraft.util.Mth;
import net.minecraft.world.entity.monster.warden.Warden;

@Environment(EnvType.CLIENT)
public class WardenRenderer extends MobRenderer<Warden, WardenModel<Warden>> {
	private static final ResourceLocation TEXTURE = ResourceLocation.withDefaultNamespace("textures/entity/warden/warden.png");
	private static final ResourceLocation BIOLUMINESCENT_LAYER_TEXTURE = ResourceLocation.withDefaultNamespace(
		"textures/entity/warden/warden_bioluminescent_layer.png"
	);
	private static final ResourceLocation HEART_TEXTURE = ResourceLocation.withDefaultNamespace("textures/entity/warden/warden_heart.png");
	private static final ResourceLocation PULSATING_SPOTS_TEXTURE_1 = ResourceLocation.withDefaultNamespace("textures/entity/warden/warden_pulsating_spots_1.png");
	private static final ResourceLocation PULSATING_SPOTS_TEXTURE_2 = ResourceLocation.withDefaultNamespace("textures/entity/warden/warden_pulsating_spots_2.png");

	public WardenRenderer(EntityRendererProvider.Context context) {
		super(context, new WardenModel<>(context.bakeLayer(ModelLayers.WARDEN)), 0.9F);
		this.addLayer(new WardenEmissiveLayer<>(this, BIOLUMINESCENT_LAYER_TEXTURE, (warden, f, g) -> 1.0F, WardenModel::getBioluminescentLayerModelParts));
		this.addLayer(
			new WardenEmissiveLayer<>(
				this, PULSATING_SPOTS_TEXTURE_1, (warden, f, g) -> Math.max(0.0F, Mth.cos(g * 0.045F) * 0.25F), WardenModel::getPulsatingSpotsLayerModelParts
			)
		);
		this.addLayer(
			new WardenEmissiveLayer<>(
				this,
				PULSATING_SPOTS_TEXTURE_2,
				(warden, f, g) -> Math.max(0.0F, Mth.cos(g * 0.045F + (float) Math.PI) * 0.25F),
				WardenModel::getPulsatingSpotsLayerModelParts
			)
		);
		this.addLayer(new WardenEmissiveLayer<>(this, TEXTURE, (warden, f, g) -> warden.getTendrilAnimation(f), WardenModel::getTendrilsLayerModelParts));
		this.addLayer(new WardenEmissiveLayer<>(this, HEART_TEXTURE, (warden, f, g) -> warden.getHeartAnimation(f), WardenModel::getHeartLayerModelParts));
	}

	public ResourceLocation getTextureLocation(Warden warden) {
		return TEXTURE;
	}
}