about summary refs log tree commit diff
path: root/src/block/screendescriptor.rs
blob: dc0257da9d8f13391244b3bc71e316ad48d7a901 (plain)
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
use std::convert::TryInto;

pub struct ScreenDescriptor {
	pub width: u16,
	pub height: u16,
	pub packed: u8,
	pub background_color_index: u8,
	pub pixel_aspect_ratio: u8,
}

impl ScreenDescriptor {
	pub fn new(width: u16, height: u16) -> Self {
		Self {
			width,
			height,
			packed: 0,
			background_color_index: 0,
			pixel_aspect_ratio: 0,
		}
	}

	pub fn set_color_table_present(&mut self, is_present: bool) {
		if is_present {
			self.packed |= 0b1000_0000;
		} else {
			self.packed &= 0b0111_1111;
		}
	}

	pub fn set_color_table_size(&mut self, size: u8) {
		println!("scts: {}", size);
		// GCT size is calulated by raising two to this number plus one,
		// so we have to work backwards.
		let size = (size as f32).log2().ceil() - 1f32;
		self.packed |= size as u8;
	}

	//TODO: Setter for sort flag in packed field
	//TODO: Setter for color resolution in packed field

	pub fn color_table_present(&self) -> bool {
		self.packed & 0b1000_0000 != 0
	}

	pub fn color_table_len(&self) -> usize {
		crate::packed_to_color_table_length(self.packed & 0b0000_0111)
	}
}

impl From<&ScreenDescriptor> for Box<[u8]> {
	fn from(lsd: &ScreenDescriptor) -> Self {
		let mut vec = vec![];
		vec.extend_from_slice(&lsd.width.to_le_bytes());
		vec.extend_from_slice(&lsd.height.to_le_bytes());
		vec.push(lsd.packed);
		vec.push(lsd.background_color_index);
		vec.push(lsd.pixel_aspect_ratio);

		vec.into_boxed_slice()
	}
}

impl From<[u8; 7]> for ScreenDescriptor {
	fn from(arr: [u8; 7]) -> Self {
		let width = u16::from_le_bytes(arr[0..2].try_into().unwrap());
		let height = u16::from_le_bytes(arr[2..4].try_into().unwrap());
		let packed = arr[4];
		let background_color_index = arr[5];
		let pixel_aspect_ratio = arr[6];

		Self {
			width,
			height,
			packed,
			background_color_index,
			pixel_aspect_ratio,
		}
	}
}