package net.minecraft.client.renderer.block;

import com.mojang.blaze3d.vertex.PoseStack;
import com.mojang.math.MatrixUtil;
import it.unimi.dsi.fastutil.ints.IntArrayList;
import it.unimi.dsi.fastutil.ints.IntList;
import it.unimi.dsi.fastutil.objects.ObjectArrayList;
import java.util.List;
import net.fabricmc.api.EnvType;
import net.fabricmc.api.Environment;
import net.fabricmc.fabric.api.client.rendering.v1.FabricRenderState;
import net.minecraft.client.renderer.Sheets;
import net.minecraft.client.renderer.SubmitNodeCollector;
import net.minecraft.client.renderer.block.dispatch.BlockStateModelPart;
import net.minecraft.client.renderer.rendertype.RenderType;
import net.minecraft.client.renderer.rendertype.RenderTypes;
import net.minecraft.client.renderer.special.SpecialModelRenderer;
import net.minecraft.client.renderer.texture.TextureAtlas;
import net.minecraft.util.LightCoordsUtil;
import net.minecraft.util.RandomSource;
import org.joml.Matrix4fc;
import org.jspecify.annotations.Nullable;

@Environment(EnvType.CLIENT)
public class BlockModelRenderState implements FabricRenderState {
	public static final int[] EMPTY_TINTS = new int[0];
	@Nullable
	private List<BlockStateModelPart> modelParts;
	@Nullable
	private Matrix4fc transformation;
	@Nullable
	private RenderType renderType;
	@Nullable
	private SpecialModelRenderer<?> specialRenderer;
	@Nullable
	private Matrix4fc specialRendererTransformation;
	@Nullable
	private IntList tintLayers;
	public int blockLightCoords;
	@Nullable
	private RandomSource randomSource;

	public void clear() {
		this.modelParts = null;
		this.transformation = null;
		this.renderType = null;
		this.specialRenderer = null;
		this.specialRendererTransformation = null;
		this.blockLightCoords = 0;
		if (this.tintLayers != null) {
			this.tintLayers.clear();
		}
	}

	public IntList tintLayers() {
		if (this.tintLayers == null) {
			this.tintLayers = new IntArrayList();
		}

		return this.tintLayers;
	}

	public <T> void setupSpecialModel(final SpecialModelRenderer<T> renderer, final Matrix4fc transformation) {
		this.specialRenderer = renderer;
		this.specialRendererTransformation = identityToNull(transformation);
	}

	public List<BlockStateModelPart> setupModel(final Matrix4fc transformation, final boolean hasTranslucency) {
		this.transformation = identityToNull(transformation);
		this.renderType = hasTranslucency ? Sheets.translucentBlockItemSheet() : Sheets.cutoutBlockItemSheet();
		if (this.modelParts == null) {
			this.modelParts = new ObjectArrayList<>();
		} else {
			this.modelParts.clear();
		}

		return this.modelParts;
	}

	public void submit(
		final PoseStack poseStack, final SubmitNodeCollector submitNodeCollector, final int externalLightCoords, final int overlayCoords, final int outlineColor
	) {
		this.submitModel(this.renderType, poseStack, submitNodeCollector, externalLightCoords, overlayCoords, outlineColor);
		if (this.specialRenderer != null) {
			int lightCoords = LightCoordsUtil.max(externalLightCoords, this.blockLightCoords);
			if (this.specialRendererTransformation != null) {
				poseStack.pushPose();
				poseStack.mulPose(this.specialRendererTransformation);
				submitSpecialRenderer(this.specialRenderer, poseStack, submitNodeCollector, lightCoords, overlayCoords, outlineColor);
				poseStack.popPose();
			} else {
				submitSpecialRenderer(this.specialRenderer, poseStack, submitNodeCollector, lightCoords, overlayCoords, outlineColor);
			}
		}
	}

	@Nullable
	private static Matrix4fc identityToNull(final Matrix4fc transformation) {
		return MatrixUtil.checkPropertyRaw(transformation, 4) ? null : transformation;
	}

	private void submitModel(
		final RenderType renderType,
		final PoseStack poseStack,
		final SubmitNodeCollector submitNodeCollector,
		final int externalLightCoords,
		final int overlayCoords,
		final int outlineColor
	) {
		if (this.modelParts != null && !this.modelParts.isEmpty()) {
			List<BlockStateModelPart> modelPartsCopy = new ObjectArrayList<>(this.modelParts);
			int[] tints = this.tintLayers != null ? this.tintLayers.toArray(EMPTY_TINTS) : EMPTY_TINTS;
			int lightCoords = LightCoordsUtil.max(externalLightCoords, this.blockLightCoords);
			if (this.transformation != null) {
				poseStack.pushPose();
				poseStack.mulPose(this.transformation);
				submitNodeCollector.submitBlockModel(poseStack, renderType, modelPartsCopy, tints, lightCoords, overlayCoords, outlineColor);
				poseStack.popPose();
			} else {
				submitNodeCollector.submitBlockModel(poseStack, renderType, modelPartsCopy, tints, lightCoords, overlayCoords, outlineColor);
			}
		}
	}

	private static void submitSpecialRenderer(
		final SpecialModelRenderer<?> renderer,
		final PoseStack poseStack,
		final SubmitNodeCollector submitNodeCollector,
		final int lightCoords,
		final int overlayCoords,
		final int outlineColor
	) {
		renderer.submit(null, poseStack, submitNodeCollector, lightCoords, overlayCoords, false, outlineColor);
	}

	public void submitOnlyOutline(
		final PoseStack poseStack, final SubmitNodeCollector submitNodeCollector, final int lightCoords, final int overlayCoords, final int outlineColor
	) {
		this.submitModel(RenderTypes.outline(TextureAtlas.LOCATION_BLOCKS), poseStack, submitNodeCollector, lightCoords, overlayCoords, outlineColor);
	}

	public void submitWithZOffset(
		final PoseStack poseStack, final SubmitNodeCollector submitNodeCollector, final int lightCoords, final int overlayCoords, final int outlineColor
	) {
		this.submitModel(
			RenderTypes.entitySolidZOffsetForward(TextureAtlas.LOCATION_BLOCKS), poseStack, submitNodeCollector, lightCoords, overlayCoords, outlineColor
		);
	}

	public boolean isEmpty() {
		return this.modelParts == null && this.specialRenderer == null;
	}

	public RandomSource scratchRandomSource(final long seed) {
		if (this.randomSource == null) {
			this.randomSource = RandomSource.create(seed);
		} else {
			this.randomSource.setSeed(seed);
		}

		return this.randomSource;
	}
}