use std::{collections::HashMap, io};
use ltk_io_ext::{measure, window_at};
use super::super::{BinProperty, Error};
use byteorder::{ReadBytesExt, WriteBytesExt, LE};
#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
#[derive(Debug, PartialEq)]
pub struct BinTreeObject {
pub path_hash: u32,
pub class_hash: u32,
pub properties: HashMap<u32, BinProperty>,
}
impl BinTreeObject {
pub fn from_reader<R: io::Read + io::Seek + ?Sized>(
reader: &mut R,
class_hash: u32,
legacy: bool,
) -> Result<Self, Error> {
let size = reader.read_u32::<LE>()?;
let (real_size, value) = measure(reader, |reader| {
let path_hash = reader.read_u32::<LE>()?;
let prop_count = reader.read_u16::<LE>()? as usize;
let mut properties = HashMap::with_capacity(prop_count);
for _ in 0..prop_count {
let prop = BinProperty::from_reader(reader, legacy)?;
properties.insert(prop.name_hash, prop);
}
Ok::<_, Error>(Self {
path_hash,
class_hash,
properties,
})
})?;
if size as u64 != real_size {
return Err(Error::InvalidSize(size as _, real_size));
}
Ok(value)
}
pub fn to_writer<W: io::Write + io::Seek + ?Sized>(
&self,
writer: &mut W,
legacy: bool,
) -> io::Result<()> {
if legacy {
unimplemented!("legacy BinTreeObject write");
}
let size_pos = writer.stream_position()?;
writer.write_u32::<LE>(0)?;
let (size, _) = measure(writer, |writer| {
writer.write_u32::<LE>(self.path_hash)?;
writer.write_u16::<LE>(self.properties.len() as _)?;
for prop in self.properties.values() {
prop.to_writer(writer)?;
}
Ok::<_, io::Error>(())
})?;
window_at(writer, size_pos, |writer| writer.write_u32::<LE>(size as _))?;
Ok(())
}
}