package net.minecraft.world.level.lighting;

import net.minecraft.core.BlockPos;
import net.minecraft.core.SectionPos;
import net.minecraft.world.level.ChunkPos;
import net.minecraft.world.level.LevelHeightAccessor;
import net.minecraft.world.level.LightLayer;
import net.minecraft.world.level.chunk.DataLayer;
import net.minecraft.world.level.chunk.LightChunkGetter;
import org.jspecify.annotations.Nullable;

public class LevelLightEngine implements LightEventListener {
	public static final int LIGHT_SECTION_PADDING = 1;
	public static final LevelLightEngine EMPTY = new LevelLightEngine();
	protected final LevelHeightAccessor levelHeightAccessor;
	@Nullable
	private final LightEngine<?, ?> blockEngine;
	@Nullable
	private final LightEngine<?, ?> skyEngine;

	public LevelLightEngine(final LightChunkGetter chunkSource, final boolean hasBlockLight, final boolean hasSkyLight) {
		this.levelHeightAccessor = chunkSource.getLevel();
		this.blockEngine = hasBlockLight ? new BlockLightEngine(chunkSource) : null;
		this.skyEngine = hasSkyLight ? new SkyLightEngine(chunkSource) : null;
	}

	private LevelLightEngine() {
		this.levelHeightAccessor = LevelHeightAccessor.create(0, 0);
		this.blockEngine = null;
		this.skyEngine = null;
	}

	@Override
	public void checkBlock(final BlockPos pos) {
		if (this.blockEngine != null) {
			this.blockEngine.checkBlock(pos);
		}

		if (this.skyEngine != null) {
			this.skyEngine.checkBlock(pos);
		}
	}

	@Override
	public boolean hasLightWork() {
		return this.skyEngine != null && this.skyEngine.hasLightWork() ? true : this.blockEngine != null && this.blockEngine.hasLightWork();
	}

	@Override
	public int runLightUpdates() {
		int count = 0;
		if (this.blockEngine != null) {
			count += this.blockEngine.runLightUpdates();
		}

		if (this.skyEngine != null) {
			count += this.skyEngine.runLightUpdates();
		}

		return count;
	}

	@Override
	public void updateSectionStatus(final SectionPos pos, final boolean sectionEmpty) {
		if (this.blockEngine != null) {
			this.blockEngine.updateSectionStatus(pos, sectionEmpty);
		}

		if (this.skyEngine != null) {
			this.skyEngine.updateSectionStatus(pos, sectionEmpty);
		}
	}

	@Override
	public void setLightEnabled(final ChunkPos pos, final boolean enable) {
		if (this.blockEngine != null) {
			this.blockEngine.setLightEnabled(pos, enable);
		}

		if (this.skyEngine != null) {
			this.skyEngine.setLightEnabled(pos, enable);
		}
	}

	@Override
	public void propagateLightSources(final ChunkPos pos) {
		if (this.blockEngine != null) {
			this.blockEngine.propagateLightSources(pos);
		}

		if (this.skyEngine != null) {
			this.skyEngine.propagateLightSources(pos);
		}
	}

	public LayerLightEventListener getLayerListener(final LightLayer layer) {
		if (layer == LightLayer.BLOCK) {
			return (LayerLightEventListener)(this.blockEngine == null ? LayerLightEventListener.DummyLightLayerEventListener.INSTANCE : this.blockEngine);
		} else {
			return (LayerLightEventListener)(this.skyEngine == null ? LayerLightEventListener.DummyLightLayerEventListener.INSTANCE : this.skyEngine);
		}
	}

	public String getDebugData(final LightLayer layer, final SectionPos pos) {
		if (layer == LightLayer.BLOCK) {
			if (this.blockEngine != null) {
				return this.blockEngine.getDebugData(pos.asLong());
			}
		} else if (this.skyEngine != null) {
			return this.skyEngine.getDebugData(pos.asLong());
		}

		return "n/a";
	}

	public LayerLightSectionStorage.SectionType getDebugSectionType(final LightLayer layer, final SectionPos pos) {
		if (layer == LightLayer.BLOCK) {
			if (this.blockEngine != null) {
				return this.blockEngine.getDebugSectionType(pos.asLong());
			}
		} else if (this.skyEngine != null) {
			return this.skyEngine.getDebugSectionType(pos.asLong());
		}

		return LayerLightSectionStorage.SectionType.EMPTY;
	}

	public void queueSectionData(final LightLayer layer, final SectionPos pos, @Nullable final DataLayer data) {
		if (layer == LightLayer.BLOCK) {
			if (this.blockEngine != null) {
				this.blockEngine.queueSectionData(pos.asLong(), data);
			}
		} else if (this.skyEngine != null) {
			this.skyEngine.queueSectionData(pos.asLong(), data);
		}
	}

	public void retainData(final ChunkPos pos, final boolean retain) {
		if (this.blockEngine != null) {
			this.blockEngine.retainData(pos, retain);
		}

		if (this.skyEngine != null) {
			this.skyEngine.retainData(pos, retain);
		}
	}

	public int getRawBrightness(final BlockPos pos, final int skyDampen) {
		int skyLight = this.skyEngine == null ? 0 : this.skyEngine.getLightValue(pos) - skyDampen;
		int blockLight = this.blockEngine == null ? 0 : this.blockEngine.getLightValue(pos);
		return Math.max(blockLight, skyLight);
	}

	public boolean lightOnInColumn(final long sectionZeroNode) {
		return this.blockEngine == null
			|| this.blockEngine.storage.lightOnInColumn(sectionZeroNode) && (this.skyEngine == null || this.skyEngine.storage.lightOnInColumn(sectionZeroNode));
	}

	public int getLightSectionCount() {
		return this.levelHeightAccessor.getSectionsCount() + 2;
	}

	public int getMinLightSection() {
		return this.levelHeightAccessor.getMinSectionY() - 1;
	}

	public int getMaxLightSection() {
		return this.getMinLightSection() + this.getLightSectionCount();
	}
}