1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
use std::io::Cursor;
use byteorder::{LittleEndian, ReadBytesExt};
use failure::{ensure, Error};
use crate::model::{owned::TableTypeSpecBuf, TypeSpec};
#[derive(Clone, Debug)]
pub struct TypeSpecWrapper<'a> {
raw_data: &'a [u8],
}
impl<'a> TypeSpecWrapper<'a> {
pub fn new(raw_data: &'a [u8]) -> Self {
Self { raw_data }
}
pub fn to_buffer(&self) -> Result<TableTypeSpecBuf, Error> {
let mut owned = TableTypeSpecBuf::new(self.get_id()?);
let amount = self.get_amount()?;
for i in 0..amount {
owned.push_flag(self.get_flag(i)?);
}
Ok(owned)
}
}
impl<'a> TypeSpec for TypeSpecWrapper<'a> {
fn get_id(&self) -> Result<u16, Error> {
let mut cursor = Cursor::new(self.raw_data);
cursor.set_position(8);
let out_value = cursor.read_u32::<LittleEndian>()? & 0xFF;
Ok(out_value as u16)
}
fn get_amount(&self) -> Result<u32, Error> {
let mut cursor = Cursor::new(self.raw_data);
cursor.set_position(12);
Ok(cursor.read_u32::<LittleEndian>()?)
}
fn get_flag(&self, index: u32) -> Result<u32, Error> {
let amount = self.get_amount()?;
ensure!(
index < amount,
"invalid flag on index {} out of {}",
index,
amount
);
let mut cursor = Cursor::new(self.raw_data);
let flag_offset = 16 + u64::from(index) * 4;
cursor.set_position(flag_offset);
Ok(cursor.read_u32::<LittleEndian>()?)
}
}
#[cfg(test)]
mod tests {
use super::{TypeSpec, TypeSpecWrapper};
use crate::raw_chunks;
#[test]
fn it_can_decode_a_type_spec() {
let wrapper = TypeSpecWrapper::new(raw_chunks::EXAMPLE_TYPE_SPEC);
assert_eq!(4, wrapper.get_id().unwrap());
assert_eq!(1541, wrapper.get_amount().unwrap());
assert_eq!(0x40000004, wrapper.get_flag(0).unwrap());
assert_eq!(0, wrapper.get_flag(25).unwrap());
assert_eq!(6, wrapper.get_flag(1540).unwrap());
let errored_flag = wrapper.get_flag(1541);
assert!(errored_flag.is_err());
assert_eq!(
"invalid flag on index 1541 out of 1541",
errored_flag.err().unwrap().to_string()
);
}
}