package net.minecraft.client.renderer.entity;

import com.mojang.blaze3d.vertex.PoseStack;
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.Model;
import net.minecraft.client.model.geom.ModelLayerLocation;
import net.minecraft.client.model.geom.ModelLayers;
import net.minecraft.client.model.object.boat.BoatModel;
import net.minecraft.client.renderer.SubmitNodeCollector;
import net.minecraft.client.renderer.entity.state.BoatRenderState;
import net.minecraft.client.renderer.rendertype.RenderTypes;
import net.minecraft.client.renderer.texture.OverlayTexture;
import net.minecraft.util.Unit;

@Environment(EnvType.CLIENT)
public class BoatRenderer extends AbstractBoatRenderer {
	private final Model.Simple waterPatchModel;
	private final EntityModel<BoatRenderState> model;

	public BoatRenderer(final EntityRendererProvider.Context context, final ModelLayerLocation modelId) {
		super(context, modelId.model().withPath((UnaryOperator<String>)(p -> "textures/entity/" + p + ".png")));
		this.waterPatchModel = new Model.Simple(context.bakeLayer(ModelLayers.BOAT_WATER_PATCH), t -> RenderTypes.waterMask());
		this.model = new BoatModel(context.bakeLayer(modelId));
	}

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

	@Override
	protected void submitTypeAdditions(
		final BoatRenderState state, final PoseStack poseStack, final SubmitNodeCollector submitNodeCollector, final int lightCoords
	) {
		if (!state.isUnderWater) {
			submitNodeCollector.submitModel(
				this.waterPatchModel, Unit.INSTANCE, poseStack, this.texture, lightCoords, OverlayTexture.NO_OVERLAY, state.outlineColor, null
			);
		}
	}
}