Replace unsafe code with raw trait

This commit is contained in:
Jaby 2023-05-01 23:01:35 +02:00
parent de4ba5e1f3
commit 5ccbd2ebd1
4 changed files with 24 additions and 14 deletions

View File

@ -59,23 +59,23 @@ fn encode<T: PSXImageConverter>(image: T, color_depth: ColorType, clut_align: Cl
};
let header = Header::encode(width, height, pal_width, pal_height).ok_or(Error::from_callback(|| {format!("Image size (width: {}, height: {}) needs to be even", width, height)}))?;
tool_helper::raw::write_generic(output, header)?;
tool_helper::raw::write_raw(output, &header)?;
if let Some(palette) = palette {
let mut color_count = pal_width*pal_height;
for color in palette {
tool_helper::raw::write_generic(output, color.get_raw())?;
tool_helper::raw::write_raw(output, color)?;
color_count -= 1;
}
while color_count > 0 {
tool_helper::raw::write_generic(output, PSXColor::black().get_raw())?;
tool_helper::raw::write_raw(output, &PSXColor::black())?;
color_count -= 1;
}
}
for color in image {
tool_helper::raw::write_generic(output, color.get_raw())?;
tool_helper::raw::write_raw(output, &color)?;
}
Ok(())

View File

@ -1,4 +1,4 @@
use tool_helper::{*, bits::BitRange};
use tool_helper::{*, bits::BitRange, raw::RawConversion};
#[repr(packed(1))]
#[allow(dead_code)]
@ -102,10 +102,6 @@ impl Color {
Color{value}
}
pub const fn get_raw(&self) -> u16 {
self.value.to_le()
}
const fn new(stp: u8, red: u8, green: u8, blue: u8) -> Color {
let value = set_member_value!(set_member_value!(set_member_value!(set_member_value!(0,
stp, 0, u16),
@ -122,6 +118,12 @@ impl Color {
make_member_getter_setter!(blue, Self::COLOR_SHIFT, u16);
}
impl RawConversion<2> for Color {
fn convert_to_raw(&self) -> [u8; 2] {
self.value.to_le_bytes()
}
}
#[allow(dead_code)]
impl Header {
const TEX_WIDTH_BIT_RANGE: BitRange = BitRange::from_to(0, 8);
@ -146,6 +148,12 @@ impl Header {
}
}
impl RawConversion<4> for Header {
fn convert_to_raw(&self) -> [u8; 4] {
self.value.to_le_bytes()
}
}
pub trait PSXImageConverter: std::iter::Iterator<Item = Color> {
fn width(&self) -> u16;
fn height(&self) -> u16;

View File

@ -1,6 +1,6 @@
[package]
name = "tool_helper"
version = "0.8.1"
version = "0.9.0"
edition = "2021"
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html

View File

@ -1,7 +1,9 @@
use super::{Error, Write};
pub fn write_generic<T>(output: &mut dyn Write, value: T) -> Result<usize, Error> {
let raw = unsafe {std::slice::from_raw_parts(&value as *const _ as *const u8, std::mem::size_of_val(&value))};
pub trait RawConversion<const COUNT:usize> {
fn convert_to_raw(&self) -> [u8; COUNT];
}
Error::try_or_new(output.write(raw))
pub fn write_raw<T:RawConversion<COUNT>, const COUNT:usize>(output: &mut dyn Write, value: &T) -> Result<usize, Error> {
Error::try_or_new(output.write(&value.convert_to_raw()))
}