package net.minecraft.client.renderer.entity;

import net.fabricmc.api.EnvType;
import net.fabricmc.api.Environment;
import net.minecraft.client.renderer.entity.EntityRendererProvider.Context;
import net.minecraft.client.renderer.entity.state.TippableArrowRenderState;
import net.minecraft.resources.ResourceLocation;
import net.minecraft.world.entity.projectile.Arrow;

@Environment(EnvType.CLIENT)
public class TippableArrowRenderer extends ArrowRenderer<Arrow, TippableArrowRenderState> {
	public static final ResourceLocation NORMAL_ARROW_LOCATION = ResourceLocation.withDefaultNamespace("textures/entity/projectiles/arrow.png");
	public static final ResourceLocation TIPPED_ARROW_LOCATION = ResourceLocation.withDefaultNamespace("textures/entity/projectiles/tipped_arrow.png");

	public TippableArrowRenderer(Context context) {
		super(context);
	}

	protected ResourceLocation getTextureLocation(TippableArrowRenderState tippableArrowRenderState) {
		return tippableArrowRenderState.isTipped ? TIPPED_ARROW_LOCATION : NORMAL_ARROW_LOCATION;
	}

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

	public void extractRenderState(Arrow arrow, TippableArrowRenderState tippableArrowRenderState, float f) {
		super.extractRenderState(arrow, tippableArrowRenderState, f);
		tippableArrowRenderState.isTipped = arrow.getColor() > 0;
	}
}