use std::num::NonZeroU16;

use anyhow::{Context, Result};
use indexmap::IndexSet;
use serde::{Deserialize, Serialize};

use super::chunk::{Chunk, SectionIterItem};
use crate::{
	resource::{Biome, BlockFlag, BlockType},
	types::*,
};

#[derive(Debug, Clone, Copy, PartialEq, Eq, Serialize, Deserialize)]
pub struct BlockHeight(pub i32);

impl BlockHeight {
	/// Constructs a new [BlockHeight] from section and block Y indices
	///
	/// Returns an error if the resulting coordindate does not fit into
	/// an [i32].
	pub fn new(section: SectionY, block: BlockY) -> Result<Self> {
		let height = section
			.0
			.checked_mul(BLOCKS_PER_CHUNK as i32)
			.and_then(|y| y.checked_add_unsigned(block.0.into()))
			.context("Block height out of bounds")?;
		Ok(BlockHeight(height))
	}
}

pub type BlockArray = LayerBlockArray<Option<BlockType>>;
pub type BiomeArray = LayerBlockArray<Option<NonZeroU16>>;
pub type BlockLightArray = LayerBlockArray<u8>;
pub type DepthArray = LayerBlockArray<Option<BlockHeight>>;

struct LayerEntry<'a> {
	block: &'a mut Option<BlockType>,
	biome: &'a mut Option<NonZeroU16>,
	block_light: &'a mut u8,
	depth: &'a mut Option<BlockHeight>,
}

impl<'a> LayerEntry<'a> {
	fn is_empty(&self) -> bool {
		self.block.is_none()
	}

	fn done(&self) -> bool {
		self.depth.is_some()
	}

	fn fill(
		&mut self,
		biome_list: &mut IndexSet<Biome>,
		section: SectionIterItem,
		coords: SectionBlockCoords,
	) -> Result<bool> {
		let Some(block_type) = section
			.section
			.block_at(coords)?
			.filter(|block_type| block_type.is(BlockFlag::Opaque))
		else {
			if self.is_empty() {
				*self.block_light = section.block_light.block_light_at(coords);
			}

			return Ok(false);
		};

		if self.is_empty() {
			*self.block = Some(block_type);
			if let Some(biome) = section.biomes.biome_at(section.y, coords)? {
				let (biome_index, _) = biome_list.insert_full(*biome);
				*self.biome = NonZeroU16::new(
					(biome_index + 1)
						.try_into()
						.expect("biome index not in range"),
				);
			}
		}

		if block_type.is(BlockFlag::Water) {
			return Ok(false);
		}

		let height = BlockHeight::new(section.y, coords.y)?;
		*self.depth = Some(height);

		Ok(true)
	}
}

#[derive(Debug, Default)]
pub struct LayerData {
	pub blocks: Box<BlockArray>,
	pub biomes: Box<BiomeArray>,
	pub block_light: Box<BlockLightArray>,
	pub depths: Box<DepthArray>,
}

impl LayerData {
	fn entry(&mut self, coords: LayerBlockCoords) -> LayerEntry {
		LayerEntry {
			block: &mut self.blocks[coords],
			biome: &mut self.biomes[coords],
			block_light: &mut self.block_light[coords],
			depth: &mut self.depths[coords],
		}
	}
}

/// Fills in a [BlockInfoArray] with the information of the chunk's top
/// block layer
///
/// For each (X, Z) coordinate pair, the topmost opaque block is
/// determined as the block that should be visible on the rendered
/// map. For water blocks, the height of the first non-water block
/// is additionally filled in as the water depth.
pub fn top_layer(biome_list: &mut IndexSet<Biome>, chunk: &Chunk) -> Result<Option<LayerData>> {
	use BLOCKS_PER_CHUNK as N;

	if chunk.is_empty() {
		return Ok(None);
	}

	let mut done = 0;
	let mut ret = LayerData::default();

	for section in chunk.sections().rev() {
		for y in BlockY::iter().rev() {
			for z in BlockZ::iter() {
				for x in BlockX::iter() {
					let xz = LayerBlockCoords { x, z };

					let mut entry = ret.entry(xz);
					if entry.done() {
						continue;
					}

					let coords = SectionBlockCoords { xz, y };
					if !entry.fill(biome_list, section, coords)? {
						continue;
					}

					assert!(entry.done());
					done += 1;
					if done == N * N {
						break;
					}
				}
			}
		}
	}

	Ok(Some(ret))
}