macro_rules! block_decoder{
($name: expr, $block_width: expr, $block_height: expr, $raw_block_size: expr, $block_decode_func: expr) => {
paste::item! {
#[doc = "Decodes a " $name " encoded texture into an image"]
pub fn [<decode_ $name>](data: &[u8], width: usize, height: usize, image: &mut [u32]) -> Result<(), &'static str> {
const BLOCK_WIDTH: usize = $block_width;
const BLOCK_HEIGHT: usize = $block_height;
const BLOCK_SIZE: usize = BLOCK_WIDTH * BLOCK_HEIGHT;
let num_blocks_x: usize = width.div_ceil(BLOCK_WIDTH);
let num_blocks_y: usize = height.div_ceil(BLOCK_HEIGHT);
let mut buffer: [u32; BLOCK_SIZE] = [crate::color::color(0,0,0,255); BLOCK_SIZE];
if data.len() < num_blocks_x * num_blocks_y * $raw_block_size {
return Err("Not enough data to decode image!");
}
if image.len() < width * height {
return Err("Image buffer is too small!");
}
let mut data_offset = 0;
(0..num_blocks_y).for_each(|by| {
(0..num_blocks_x).for_each(|bx| {
$block_decode_func(&data[data_offset..], &mut buffer);
crate::color::copy_block_buffer(
bx,
by,
width,
height,
BLOCK_WIDTH,
BLOCK_HEIGHT,
&buffer,
image,
);
data_offset += $raw_block_size;
});
});
Ok(())
}
}
};
}
macro_rules! CRND_HUFF_DECODE {
($codec: expr, $model: expr, $symbol: expr) => {
$symbol = match $codec.decode($model) {
Ok(s) => s,
Err(_) => return false,
}
};
}
macro_rules! WRITE_TO_INT_BUFFER {
($buf: expr, $index: expr, $val: expr) => {
let t_index = ($index * 4) as usize;
#[cfg(target_endian = "little")]
let tiles = $val.to_le_bytes();
#[cfg(target_endian = "big")]
let tiles = $val.to_be_bytes();
$buf[t_index] = tiles[0];
$buf[t_index + 1] = tiles[1];
$buf[t_index + 2] = tiles[2];
$buf[t_index + 3] = tiles[3]
};
}
macro_rules! WRITE_OR_U8_INTO_U16_BUFFER {
($buf: expr, $index: expr, $val: expr) => {
let t_index = ($index >> 1) as usize;
let t_val = $buf[t_index].to_be_bytes();
if $index & 1 != 1 {
$buf[t_index] = ((t_val[0] as u16) << 8) | ((t_val[1] as u16) | $val as u16);
} else {
$buf[t_index] = (((t_val[0] as u16) | ($val as u16)) << 8) | (t_val[1] as u16);
}
};
}
pub(crate) use {
block_decoder, CRND_HUFF_DECODE, WRITE_OR_U8_INTO_U16_BUFFER, WRITE_TO_INT_BUFFER,
};