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

import net.fabricmc.api.EnvType;
import net.fabricmc.api.Environment;
import net.minecraft.client.model.ArrowModel;
import net.minecraft.client.model.PlayerModel;
import net.minecraft.client.model.geom.ModelLayers;
import net.minecraft.client.renderer.entity.LivingEntityRenderer;
import net.minecraft.client.renderer.entity.TippableArrowRenderer;
import net.minecraft.client.renderer.entity.EntityRendererProvider.Context;
import net.minecraft.client.renderer.entity.state.PlayerRenderState;

@Environment(EnvType.CLIENT)
public class ArrowLayer<M extends PlayerModel> extends StuckInBodyLayer<M> {
	public ArrowLayer(LivingEntityRenderer<?, PlayerRenderState, M> renderer, Context context) {
		super(renderer, new ArrowModel(context.bakeLayer(ModelLayers.ARROW)), TippableArrowRenderer.NORMAL_ARROW_LOCATION, StuckInBodyLayer.PlacementStyle.IN_CUBE);
	}

	@Override
	protected int numStuck(PlayerRenderState renderState) {
		return renderState.arrowCount;
	}
}