package net.minecraft.client.renderer.entity;

import com.mojang.blaze3d.vertex.PoseStack;
import java.util.Objects;
import net.minecraft.client.model.geom.ModelLayers;
import net.minecraft.client.model.monster.illager.IllagerModel;
import net.minecraft.client.renderer.SubmitNodeCollector;
import net.minecraft.client.renderer.entity.layers.ItemInHandLayer;
import net.minecraft.client.renderer.entity.state.EvokerRenderState;
import net.minecraft.resources.Identifier;
import net.minecraft.world.entity.monster.illager.SpellcasterIllager;

public class EvokerRenderer<T extends SpellcasterIllager> extends IllagerRenderer<T, EvokerRenderState> {
	private static final Identifier EVOKER_ILLAGER = Identifier.withDefaultNamespace("textures/entity/illager/evoker.png");

	public EvokerRenderer(final EntityRendererProvider.Context context) {
		super(context, new IllagerModel<>(context.bakeLayer(ModelLayers.EVOKER)), 0.5F);
		this.addLayer(
			new ItemInHandLayer<EvokerRenderState, IllagerModel<EvokerRenderState>>(this) {
				{
					Objects.requireNonNull(EvokerRenderer.this);
				}

				public void submit(
					final PoseStack poseStack,
					final SubmitNodeCollector submitNodeCollector,
					final int lightCoords,
					final EvokerRenderState state,
					final float yRot,
					final float xRot
				) {
					if (state.isCastingSpell) {
						super.submit(poseStack, submitNodeCollector, lightCoords, state, yRot, xRot);
					}
				}
			}
		);
	}

	public Identifier getTextureLocation(final EvokerRenderState state) {
		return EVOKER_ILLAGER;
	}

	public EvokerRenderState createRenderState() {
		return new EvokerRenderState();
	}

	public void extractRenderState(final T entity, final EvokerRenderState state, final float partialTicks) {
		super.extractRenderState(entity, state, partialTicks);
		state.isCastingSpell = entity.isCastingSpell();
	}
}