use crate::{Image, Pixel, Transform};

impl Image<u16> {
	pub fn transform_iter(&self) -> (usize, usize, impl Iterator<Item = Pixel> + use<'_>) {
		let (final_width, final_height) = if self.transform.will_swap_coordinates() {
			(self.height, self.width)
		} else {
			(self.width, self.height)
		};

		let index_0_0 = inverse_transform_index(self.transform, 0, 0, self.width, self.height);
		let index_0_1 = inverse_transform_index(self.transform, 0, 1, self.width, self.height);
		let index_1_0 = inverse_transform_index(self.transform, 1, 0, self.width, self.height);

		let column_step = (index_0_1.0 - index_0_0.0, index_0_1.1 - index_0_0.1);
		let row_step = (index_1_0.0 - index_0_0.0, index_1_0.1 - index_0_0.1);
		let mut index = index_0_0;

		let channels = self.channels as usize;

		(
			final_width,
			final_height,
			(0..final_height).flat_map(move |row| {
				let temp = (0..final_width).map(move |column| {
					let initial_index = (self.width as i64 * index.0 + index.1) as usize;
					let pixel = &self.data[channels * initial_index..channels * (initial_index + 1)];
					index = (index.0 + column_step.0, index.1 + column_step.1);

					Pixel {
						values: pixel.try_into().unwrap(),
						row,
						column,
					}
				});

				index = (index.0 + row_step.0, index.1 + row_step.1);

				temp
			}),
		)
	}
}

pub fn inverse_transform_index(transform: Transform, mut row: usize, mut column: usize, width: usize, height: usize) -> (i64, i64) {
	let value = match transform {
		Transform::Horizontal => 0,
		Transform::MirrorHorizontal => 1,
		Transform::Rotate180 => 3,
		Transform::MirrorVertical => 2,
		Transform::MirrorHorizontalRotate270 => 4,
		Transform::Rotate90 => 6,
		Transform::MirrorHorizontalRotate90 => 7,
		Transform::Rotate270 => 5,
	};

	if value & 4 != 0 {
		std::mem::swap(&mut row, &mut column)
	}

	if value & 2 != 0 {
		row = height - 1 - row;
	}

	if value & 1 != 0 {
		column = width - 1 - column;
	}

	(row as i64, column as i64)
}