package net.minecraft.client.renderer.entity;

import java.util.function.UnaryOperator;
import net.fabricmc.api.EnvType;
import net.fabricmc.api.Environment;
import net.minecraft.client.model.EntityModel;
import net.minecraft.client.model.RaftModel;
import net.minecraft.client.model.geom.ModelLayerLocation;
import net.minecraft.client.renderer.RenderType;
import net.minecraft.client.renderer.entity.EntityRendererProvider.Context;
import net.minecraft.client.renderer.entity.state.BoatRenderState;
import net.minecraft.resources.ResourceLocation;

@Environment(EnvType.CLIENT)
public class RaftRenderer extends AbstractBoatRenderer {
	private final EntityModel<BoatRenderState> model;
	private final ResourceLocation texture;

	public RaftRenderer(Context context, ModelLayerLocation modelLayer) {
		super(context);
		this.texture = modelLayer.model().withPath((UnaryOperator<String>)(string -> "textures/entity/" + string + ".png"));
		this.model = new RaftModel(context.bakeLayer(modelLayer));
	}

	@Override
	protected EntityModel<BoatRenderState> model() {
		return this.model;
	}

	@Override
	protected RenderType renderType() {
		return this.model.renderType(this.texture);
	}
}