package net.minecraft.client.renderer.debug;

import it.unimi.dsi.fastutil.longs.LongOpenHashSet;
import it.unimi.dsi.fastutil.longs.LongSet;
import net.minecraft.client.Minecraft;
import net.minecraft.client.renderer.culling.Frustum;
import net.minecraft.core.BlockPos;
import net.minecraft.core.SectionPos;
import net.minecraft.gizmos.Gizmos;
import net.minecraft.gizmos.TextGizmo.Style;
import net.minecraft.util.ARGB;
import net.minecraft.util.debug.DebugValueAccess;
import net.minecraft.world.level.Level;
import net.minecraft.world.level.LightLayer;
import net.minecraft.world.phys.Vec3;

public class LightDebugRenderer implements DebugRenderer.SimpleDebugRenderer {
	private final Minecraft minecraft;
	private final boolean showBlockLight;
	private final boolean showSkyLight;
	private static final int MAX_RENDER_DIST = 10;

	public LightDebugRenderer(final Minecraft minecraft, final boolean showBlockLight, final boolean showSkyLight) {
		this.minecraft = minecraft;
		this.showBlockLight = showBlockLight;
		this.showSkyLight = showSkyLight;
	}

	@Override
	public void emitGizmos(
		final double camX, final double camY, final double camZ, final DebugValueAccess debugValues, final Frustum frustum, final float partialTicks
	) {
		Level level = this.minecraft.level;
		BlockPos playerPos = BlockPos.containing(camX, camY, camZ);
		LongSet set = new LongOpenHashSet();

		for (BlockPos blockPos : BlockPos.betweenClosed(playerPos.offset(-10, -10, -10), playerPos.offset(10, 10, 10))) {
			int skyBrightness = level.getBrightness(LightLayer.SKY, blockPos);
			long sectionNode = SectionPos.blockToSection(blockPos.asLong());
			if (set.add(sectionNode)) {
				Gizmos.billboardText(
					level.getChunkSource().getLightEngine().getDebugData(LightLayer.SKY, SectionPos.of(sectionNode)),
					new Vec3(
						SectionPos.sectionToBlockCoord(SectionPos.x(sectionNode), 8),
						SectionPos.sectionToBlockCoord(SectionPos.y(sectionNode), 8),
						SectionPos.sectionToBlockCoord(SectionPos.z(sectionNode), 8)
					),
					Style.forColorAndCentered(-65536).withScale(4.8F)
				);
			}

			if (skyBrightness != 15 && this.showSkyLight) {
				int color = ARGB.srgbLerp(skyBrightness / 15.0F, -16776961, -16711681);
				Gizmos.billboardText(String.valueOf(skyBrightness), Vec3.atLowerCornerWithOffset(blockPos, 0.5, 0.25, 0.5), Style.forColorAndCentered(color));
			}

			if (this.showBlockLight) {
				int blockBrightness = level.getBrightness(LightLayer.BLOCK, blockPos);
				if (blockBrightness != 0) {
					int color = ARGB.srgbLerp(blockBrightness / 15.0F, -5636096, -256);
					Gizmos.billboardText(String.valueOf(level.getBrightness(LightLayer.BLOCK, blockPos)), Vec3.atCenterOf(blockPos), Style.forColorAndCentered(color));
				}
			}
		}
	}
}