package net.minecraft.client.renderer;

import com.mojang.blaze3d.platform.Lighting;
import com.mojang.blaze3d.vertex.PoseStack;
import com.mojang.blaze3d.vertex.VertexConsumer;
import com.mojang.math.Axis;
import net.minecraft.client.Minecraft;
import net.minecraft.client.player.LocalPlayer;
import net.minecraft.client.renderer.block.BlockStateModelSet;
import net.minecraft.client.renderer.item.ItemStackRenderState;
import net.minecraft.client.renderer.rendertype.RenderTypes;
import net.minecraft.client.renderer.state.WindowRenderState;
import net.minecraft.client.renderer.texture.OverlayTexture;
import net.minecraft.client.renderer.texture.TextureAtlasSprite;
import net.minecraft.client.resources.model.ModelBakery;
import net.minecraft.client.resources.model.sprite.SpriteGetter;
import net.minecraft.core.BlockPos;
import net.minecraft.core.BlockPos.MutableBlockPos;
import net.minecraft.resources.Identifier;
import net.minecraft.tags.FluidTags;
import net.minecraft.util.ARGB;
import net.minecraft.util.Mth;
import net.minecraft.util.RandomSource;
import net.minecraft.world.entity.player.Player;
import net.minecraft.world.item.ItemDisplayContext;
import net.minecraft.world.item.ItemStack;
import net.minecraft.world.level.block.RenderShape;
import net.minecraft.world.level.block.state.BlockState;
import org.joml.Matrix4f;
import org.jspecify.annotations.Nullable;

public class ScreenEffectRenderer {
	private static final Identifier UNDERWATER_LOCATION = Identifier.withDefaultNamespace("textures/misc/underwater.png");
	private final Minecraft minecraft;
	private final SpriteGetter sprites;
	public static final int ITEM_ACTIVATION_ANIMATION_LENGTH = 40;
	@Nullable
	private ItemStack itemActivationItem;
	private int itemActivationTicks;
	private float itemActivationOffX;
	private float itemActivationOffY;

	public ScreenEffectRenderer(final Minecraft minecraft, final SpriteGetter sprites) {
		this.minecraft = minecraft;
		this.sprites = sprites;
	}

	public void tick() {
		if (this.itemActivationTicks > 0) {
			this.itemActivationTicks--;
			if (this.itemActivationTicks == 0) {
				this.itemActivationItem = null;
			}
		}
	}

	public void submit(
		final boolean isFirstPerson, final boolean isSleeping, final float partialTicks, final SubmitNodeCollector submitNodeCollector, final boolean hideGui
	) {
		PoseStack poseStack = new PoseStack();
		Player player = this.minecraft.player;
		if (isFirstPerson && !isSleeping) {
			BlockState blockState = getViewBlockingState(player);
			if (blockState != null) {
				BlockStateModelSet blockStateModelSet = this.minecraft.getModelManager().getBlockStateModelSet();
				TextureAtlasSprite sprite = blockStateModelSet.getParticleMaterial(blockState).sprite();
				submitBlockSprite(sprite, poseStack, submitNodeCollector, -15132391);
			}

			if (!this.minecraft.player.isSpectator()) {
				if (this.minecraft.player.isEyeInFluid(FluidTags.WATER)) {
					submitWater(this.minecraft, poseStack, submitNodeCollector);
				}

				if (this.minecraft.player.isOnFire()) {
					TextureAtlasSprite fireSprite = this.sprites.get(ModelBakery.FIRE_1);
					submitFire(poseStack, submitNodeCollector, fireSprite);
				}
			}
		}

		if (!hideGui) {
			this.renderItemActivationAnimation(poseStack, partialTicks, submitNodeCollector);
		}
	}

	private void renderItemActivationAnimation(final PoseStack poseStack, final float partialTicks, final SubmitNodeCollector submitNodeCollector) {
		if (this.itemActivationItem != null && this.itemActivationTicks > 0) {
			int tick = 40 - this.itemActivationTicks;
			float scale = (tick + partialTicks) / 40.0F;
			float ts = scale * scale;
			float tc = scale * ts;
			float smoothScale = 10.25F * tc * ts - 24.95F * ts * ts + 25.5F * tc - 13.8F * ts + 4.0F * scale;
			float piScale = smoothScale * (float) Math.PI;
			WindowRenderState windowState = this.minecraft.gameRenderer.gameRenderState().windowRenderState;
			float aspectRatio = (float)windowState.width / windowState.height;
			float offX = this.itemActivationOffX * 0.3F * aspectRatio;
			float offY = this.itemActivationOffY * 0.3F;
			poseStack.pushPose();
			poseStack.translate(offX * Mth.abs(Mth.sin(piScale * 2.0F)), offY * Mth.abs(Mth.sin(piScale * 2.0F)), -10.0F + 9.0F * Mth.sin(piScale));
			float size = 0.8F;
			poseStack.scale(0.8F, 0.8F, 0.8F);
			poseStack.mulPose(Axis.YP.rotationDegrees(900.0F * Mth.abs(Mth.sin(piScale))));
			poseStack.mulPose(Axis.XP.rotationDegrees(6.0F * Mth.cos(scale * 8.0F)));
			poseStack.mulPose(Axis.ZP.rotationDegrees(6.0F * Mth.cos(scale * 8.0F)));
			this.minecraft.gameRenderer.lighting().setupFor(Lighting.Entry.ITEMS_3D);
			ItemStackRenderState itemState = new ItemStackRenderState();
			this.minecraft.getItemModelResolver().updateForTopItem(itemState, this.itemActivationItem, ItemDisplayContext.FIXED, this.minecraft.level, null, 0);
			itemState.submit(poseStack, submitNodeCollector, 15728880, OverlayTexture.NO_OVERLAY, 0);
			poseStack.popPose();
		}
	}

	public void resetItemActivation() {
		this.itemActivationItem = null;
	}

	public void displayItemActivation(final ItemStack itemStack, final RandomSource random) {
		this.itemActivationItem = itemStack;
		this.itemActivationTicks = 40;
		this.itemActivationOffX = random.nextFloat() * 2.0F - 1.0F;
		this.itemActivationOffY = random.nextFloat() * 2.0F - 1.0F;
	}

	@Nullable
	private static BlockState getViewBlockingState(final Player player) {
		if (player.noPhysics) {
			return null;
		} else {
			MutableBlockPos testPos = new MutableBlockPos();

			for (int i = 0; i < 8; i++) {
				testPos.set(
					player.getX() + ((i >> 0) % 2 - 0.5F) * player.getBbWidth() * 0.8F,
					player.getEyeY() + ((i >> 1) % 2 - 0.5F) * 0.1F * player.getScale(),
					player.getZ() + ((i >> 2) % 2 - 0.5F) * player.getBbWidth() * 0.8F
				);
				BlockState blockState = player.level().getBlockState(testPos);
				if (blockState.getRenderShape() != RenderShape.INVISIBLE && blockState.isViewBlocking(player.level(), testPos)) {
					return blockState;
				}
			}

			return null;
		}
	}

	private static void submitBlockSprite(
		final TextureAtlasSprite sprite, final PoseStack poseStack, final SubmitNodeCollector submitNodeCollector, final int color
	) {
		submitNodeCollector.submitCustomGeometry(
			poseStack,
			RenderTypes.blockScreenEffect(sprite.atlasLocation()),
			(pose, builder) -> buildSpriteQuad(builder, pose.pose(), sprite, -1.0F, -1.0F, 1.0F, 1.0F, -0.5F, color)
		);
	}

	private static void submitWater(final Minecraft minecraft, final PoseStack poseStack, final SubmitNodeCollector submitNodeCollector) {
		LocalPlayer player = minecraft.player;
		BlockPos pos = BlockPos.containing(player.getEyePosition());
		float brightness = Lightmap.getBrightness(player.level().dimensionType(), player.level().getMaxLocalRawBrightness(pos));
		int color = ARGB.colorFromFloat(0.1F, brightness, brightness, brightness);
		float u0 = -player.getYRot() / 64.0F;
		float v0 = player.getXRot() / 64.0F;
		submitNodeCollector.submitCustomGeometry(poseStack, RenderTypes.blockScreenEffect(UNDERWATER_LOCATION), (pose, builder) -> {
			float uvSize = 4.0F;
			buildQuad(builder, pose.pose(), -1.0F, -1.0F, 1.0F, 1.0F, -0.5F, u0 + 4.0F, v0 + 4.0F, u0, v0, color);
		});
	}

	private static void submitFire(final PoseStack poseStack, final SubmitNodeCollector submitNodeCollector, final TextureAtlasSprite sprite) {
		submitNodeCollector.submitCustomGeometry(poseStack, RenderTypes.fireScreenEffect(sprite.atlasLocation()), (basePose, builder) -> {
			Matrix4f pose = new Matrix4f();
			pose.set(basePose.pose());
			pose.translate(0.24F, -0.3F, 0.0F);
			pose.rotateY((float) (-Math.PI / 18));
			buildFireQuad(sprite, builder, pose);
			pose.set(basePose.pose());
			pose.translate(-0.24F, -0.3F, 0.0F);
			pose.rotateY((float) (Math.PI / 18));
			buildFireQuad(sprite, builder, pose);
		});
	}

	private static void buildFireQuad(final TextureAtlasSprite sprite, final VertexConsumer builder, final Matrix4f pose) {
		float size = 1.0F;
		buildSpriteQuad(builder, pose, sprite, -0.5F, -0.5F, 0.5F, 0.5F, -0.5F, -436207617);
	}

	private static void buildSpriteQuad(
		final VertexConsumer builder,
		final Matrix4f pose,
		final TextureAtlasSprite sprite,
		final float x0,
		final float y0,
		final float x1,
		final float y1,
		final float z,
		final int color
	) {
		buildQuad(builder, pose, x0, y0, x1, y1, z, sprite.getU1(), sprite.getV1(), sprite.getU0(), sprite.getV0(), color);
	}

	private static void buildQuad(
		final VertexConsumer builder,
		final Matrix4f pose,
		final float x0,
		final float y0,
		final float x1,
		final float y1,
		final float z,
		final float u0,
		final float v0,
		final float u1,
		final float v1,
		final int color
	) {
		builder.addVertex(pose, x0, y0, z).setUv(u0, v0).setColor(color);
		builder.addVertex(pose, x1, y0, z).setUv(u1, v0).setColor(color);
		builder.addVertex(pose, x1, y1, z).setUv(u1, v1).setColor(color);
		builder.addVertex(pose, x0, y1, z).setUv(u0, v1).setColor(color);
	}
}