use super::*;
use crate::utils::check_image_multiple;
mod as_complete;
#[derive(Clone, Debug, Eq, PartialEq, Hash)]
pub struct GridEdgeWang {
image: RgbaImage,
}
impl GridAtlas for GridEdgeWang {
unsafe fn new(image: RgbaImage) -> Self {
Self { image }
}
fn create(image: &RgbaImage, (x, y): (u32, u32), (w, h): (u32, u32)) -> ImageResult<Self> {
let (image_w, image_h) = image.dimensions();
if x + w * 4 > image_w || y + h * 4 > image_h {
io_error("The image size has out of range", ErrorKind::InvalidInput)?;
}
let view = image.view(x, y, w * 4, h * 4);
unsafe { Ok(Self::new(view.to_image())) }
}
fn get_cell_size(&self) -> (u32, u32) {
(self.image.width() / 4, self.image.height() / 4)
}
fn get_image(&self) -> &RgbaImage {
&self.image
}
fn get_by_corner(&self, ru: bool, rd: bool, ld: bool, lu: bool) -> RgbaImage {
panic!("can not get edge wang tile by corner ({} {} {} {})", ru, rd, ld, lu)
}
fn get_by_side(&self, u: bool, r: bool, d: bool, l: bool) -> RgbaImage {
let (i, j) = wang4x4e_sub_image(u, r, d, l);
let (w, h) = self.get_cell_size();
let view = self.image.view(i * w, j * h, w, h);
view.to_image()
}
fn get_by_mask(&self, mask: u8) -> RgbaImage {
let u = mask >> 0 & 1 == 1;
let r = mask >> 2 & 1 == 1;
let d = mask >> 4 & 1 == 1;
let l = mask >> 6 & 1 == 1;
self.get_by_side(u, r, d, l)
}
fn load<P>(path: P) -> ImageResult<Self>
where
P: AsRef<Path>,
{
let image = image::open(path)?.to_rgba8();
check_image_multiple(&image, 4, 4)?;
Ok(Self { image })
}
}
fn wang4x4e_sub_image(u: bool, r: bool, d: bool, l: bool) -> (u32, u32) {
let mask = (u as u8) << 0 | (r as u8) << 1 | (d as u8) << 2 | (l as u8) << 3;
match mask {
0b0000 => (0, 3),
0b0001 => (0, 2),
0b0010 => (1, 3),
0b0011 => (1, 2),
0b0100 => (0, 0),
0b0101 => (0, 1),
0b0110 => (1, 0),
0b0111 => (1, 1),
0b1000 => (3, 3),
0b1001 => (3, 2),
0b1010 => (2, 3),
0b1011 => (2, 2),
0b1100 => (3, 0),
0b1101 => (3, 1),
0b1110 => (2, 0),
0b1111 => (2, 1),
_ => unreachable!(),
}
}