1use crate::{
15 byte_converter::*, converting_callback_receiver::ConvertingCallbackReceiver, converting_receiver::ConvertingReceiver, device::*,
16 ip_connection::GetRequestSender,
17};
18pub enum HatBrickFunction {
19 SetSleepMode,
20 GetSleepMode,
21 SetBrickletPower,
22 GetBrickletPower,
23 GetVoltages,
24 SetVoltagesCallbackConfiguration,
25 GetVoltagesCallbackConfiguration,
26 SetRtcDriver,
27 GetRtcDriver,
28 GetSpitfpErrorCount,
29 SetBootloaderMode,
30 GetBootloaderMode,
31 SetWriteFirmwarePointer,
32 WriteFirmware,
33 SetStatusLedConfig,
34 GetStatusLedConfig,
35 GetChipTemperature,
36 Reset,
37 WriteUid,
38 ReadUid,
39 GetIdentity,
40 CallbackVoltages,
41}
42impl From<HatBrickFunction> for u8 {
43 fn from(fun: HatBrickFunction) -> Self {
44 match fun {
45 HatBrickFunction::SetSleepMode => 1,
46 HatBrickFunction::GetSleepMode => 2,
47 HatBrickFunction::SetBrickletPower => 3,
48 HatBrickFunction::GetBrickletPower => 4,
49 HatBrickFunction::GetVoltages => 5,
50 HatBrickFunction::SetVoltagesCallbackConfiguration => 6,
51 HatBrickFunction::GetVoltagesCallbackConfiguration => 7,
52 HatBrickFunction::SetRtcDriver => 9,
53 HatBrickFunction::GetRtcDriver => 10,
54 HatBrickFunction::GetSpitfpErrorCount => 234,
55 HatBrickFunction::SetBootloaderMode => 235,
56 HatBrickFunction::GetBootloaderMode => 236,
57 HatBrickFunction::SetWriteFirmwarePointer => 237,
58 HatBrickFunction::WriteFirmware => 238,
59 HatBrickFunction::SetStatusLedConfig => 239,
60 HatBrickFunction::GetStatusLedConfig => 240,
61 HatBrickFunction::GetChipTemperature => 242,
62 HatBrickFunction::Reset => 243,
63 HatBrickFunction::WriteUid => 248,
64 HatBrickFunction::ReadUid => 249,
65 HatBrickFunction::GetIdentity => 255,
66 HatBrickFunction::CallbackVoltages => 8,
67 }
68 }
69}
70pub const HAT_BRICK_RTC_DRIVER_PCF8523: u8 = 0;
71pub const HAT_BRICK_RTC_DRIVER_DS1338: u8 = 1;
72pub const HAT_BRICK_BOOTLOADER_MODE_BOOTLOADER: u8 = 0;
73pub const HAT_BRICK_BOOTLOADER_MODE_FIRMWARE: u8 = 1;
74pub const HAT_BRICK_BOOTLOADER_MODE_BOOTLOADER_WAIT_FOR_REBOOT: u8 = 2;
75pub const HAT_BRICK_BOOTLOADER_MODE_FIRMWARE_WAIT_FOR_REBOOT: u8 = 3;
76pub const HAT_BRICK_BOOTLOADER_MODE_FIRMWARE_WAIT_FOR_ERASE_AND_REBOOT: u8 = 4;
77pub const HAT_BRICK_BOOTLOADER_STATUS_OK: u8 = 0;
78pub const HAT_BRICK_BOOTLOADER_STATUS_INVALID_MODE: u8 = 1;
79pub const HAT_BRICK_BOOTLOADER_STATUS_NO_CHANGE: u8 = 2;
80pub const HAT_BRICK_BOOTLOADER_STATUS_ENTRY_FUNCTION_NOT_PRESENT: u8 = 3;
81pub const HAT_BRICK_BOOTLOADER_STATUS_DEVICE_IDENTIFIER_INCORRECT: u8 = 4;
82pub const HAT_BRICK_BOOTLOADER_STATUS_CRC_MISMATCH: u8 = 5;
83pub const HAT_BRICK_STATUS_LED_CONFIG_OFF: u8 = 0;
84pub const HAT_BRICK_STATUS_LED_CONFIG_ON: u8 = 1;
85pub const HAT_BRICK_STATUS_LED_CONFIG_SHOW_HEARTBEAT: u8 = 2;
86pub const HAT_BRICK_STATUS_LED_CONFIG_SHOW_STATUS: u8 = 3;
87
88#[derive(Clone, Copy, Debug, Default, PartialEq, Eq, Hash)]
89pub struct SleepMode {
90 pub power_off_delay: u32,
91 pub power_off_duration: u32,
92 pub raspberry_pi_off: bool,
93 pub bricklets_off: bool,
94 pub enable_sleep_indicator: bool,
95}
96impl FromByteSlice for SleepMode {
97 fn bytes_expected() -> usize { 11 }
98 fn from_le_byte_slice(bytes: &[u8]) -> SleepMode {
99 SleepMode {
100 power_off_delay: <u32>::from_le_byte_slice(&bytes[0..4]),
101 power_off_duration: <u32>::from_le_byte_slice(&bytes[4..8]),
102 raspberry_pi_off: <bool>::from_le_byte_slice(&bytes[8..9]),
103 bricklets_off: <bool>::from_le_byte_slice(&bytes[9..10]),
104 enable_sleep_indicator: <bool>::from_le_byte_slice(&bytes[10..11]),
105 }
106 }
107}
108
109#[derive(Clone, Copy, Debug, Default, PartialEq, Eq, Hash)]
110pub struct Voltages {
111 pub voltage_usb: u16,
112 pub voltage_dc: u16,
113}
114impl FromByteSlice for Voltages {
115 fn bytes_expected() -> usize { 4 }
116 fn from_le_byte_slice(bytes: &[u8]) -> Voltages {
117 Voltages { voltage_usb: <u16>::from_le_byte_slice(&bytes[0..2]), voltage_dc: <u16>::from_le_byte_slice(&bytes[2..4]) }
118 }
119}
120
121#[derive(Clone, Copy, Debug, Default, PartialEq, Eq, Hash)]
122pub struct VoltagesCallbackConfiguration {
123 pub period: u32,
124 pub value_has_to_change: bool,
125}
126impl FromByteSlice for VoltagesCallbackConfiguration {
127 fn bytes_expected() -> usize { 5 }
128 fn from_le_byte_slice(bytes: &[u8]) -> VoltagesCallbackConfiguration {
129 VoltagesCallbackConfiguration {
130 period: <u32>::from_le_byte_slice(&bytes[0..4]),
131 value_has_to_change: <bool>::from_le_byte_slice(&bytes[4..5]),
132 }
133 }
134}
135
136#[derive(Clone, Copy, Debug, Default, PartialEq, Eq, Hash)]
137pub struct VoltagesEvent {
138 pub voltage_usb: u16,
139 pub voltage_dc: u16,
140}
141impl FromByteSlice for VoltagesEvent {
142 fn bytes_expected() -> usize { 4 }
143 fn from_le_byte_slice(bytes: &[u8]) -> VoltagesEvent {
144 VoltagesEvent { voltage_usb: <u16>::from_le_byte_slice(&bytes[0..2]), voltage_dc: <u16>::from_le_byte_slice(&bytes[2..4]) }
145 }
146}
147
148#[derive(Clone, Copy, Debug, Default, PartialEq, Eq, Hash)]
149pub struct SpitfpErrorCount {
150 pub error_count_ack_checksum: u32,
151 pub error_count_message_checksum: u32,
152 pub error_count_frame: u32,
153 pub error_count_overflow: u32,
154}
155impl FromByteSlice for SpitfpErrorCount {
156 fn bytes_expected() -> usize { 16 }
157 fn from_le_byte_slice(bytes: &[u8]) -> SpitfpErrorCount {
158 SpitfpErrorCount {
159 error_count_ack_checksum: <u32>::from_le_byte_slice(&bytes[0..4]),
160 error_count_message_checksum: <u32>::from_le_byte_slice(&bytes[4..8]),
161 error_count_frame: <u32>::from_le_byte_slice(&bytes[8..12]),
162 error_count_overflow: <u32>::from_le_byte_slice(&bytes[12..16]),
163 }
164 }
165}
166
167#[derive(Clone, Debug, Default, PartialEq, Eq, Hash)]
168pub struct Identity {
169 pub uid: String,
170 pub connected_uid: String,
171 pub position: char,
172 pub hardware_version: [u8; 3],
173 pub firmware_version: [u8; 3],
174 pub device_identifier: u16,
175}
176impl FromByteSlice for Identity {
177 fn bytes_expected() -> usize { 25 }
178 fn from_le_byte_slice(bytes: &[u8]) -> Identity {
179 Identity {
180 uid: <String>::from_le_byte_slice(&bytes[0..8]),
181 connected_uid: <String>::from_le_byte_slice(&bytes[8..16]),
182 position: <char>::from_le_byte_slice(&bytes[16..17]),
183 hardware_version: <[u8; 3]>::from_le_byte_slice(&bytes[17..20]),
184 firmware_version: <[u8; 3]>::from_le_byte_slice(&bytes[20..23]),
185 device_identifier: <u16>::from_le_byte_slice(&bytes[23..25]),
186 }
187 }
188}
189
190#[derive(Clone)]
192pub struct HatBrick {
193 device: Device,
194}
195impl HatBrick {
196 pub const DEVICE_IDENTIFIER: u16 = 111;
197 pub const DEVICE_DISPLAY_NAME: &'static str = "HAT Brick";
198 pub fn new<T: GetRequestSender>(uid: &str, req_sender: T) -> HatBrick {
200 let mut result = HatBrick { device: Device::new([2, 0, 2], uid, req_sender, 0) };
201 result.device.response_expected[u8::from(HatBrickFunction::SetSleepMode) as usize] = ResponseExpectedFlag::False;
202 result.device.response_expected[u8::from(HatBrickFunction::GetSleepMode) as usize] = ResponseExpectedFlag::AlwaysTrue;
203 result.device.response_expected[u8::from(HatBrickFunction::SetBrickletPower) as usize] = ResponseExpectedFlag::False;
204 result.device.response_expected[u8::from(HatBrickFunction::GetBrickletPower) as usize] = ResponseExpectedFlag::AlwaysTrue;
205 result.device.response_expected[u8::from(HatBrickFunction::GetVoltages) as usize] = ResponseExpectedFlag::AlwaysTrue;
206 result.device.response_expected[u8::from(HatBrickFunction::SetVoltagesCallbackConfiguration) as usize] = ResponseExpectedFlag::True;
207 result.device.response_expected[u8::from(HatBrickFunction::GetVoltagesCallbackConfiguration) as usize] =
208 ResponseExpectedFlag::AlwaysTrue;
209 result.device.response_expected[u8::from(HatBrickFunction::SetRtcDriver) as usize] = ResponseExpectedFlag::False;
210 result.device.response_expected[u8::from(HatBrickFunction::GetRtcDriver) as usize] = ResponseExpectedFlag::AlwaysTrue;
211 result.device.response_expected[u8::from(HatBrickFunction::GetSpitfpErrorCount) as usize] = ResponseExpectedFlag::AlwaysTrue;
212 result.device.response_expected[u8::from(HatBrickFunction::SetBootloaderMode) as usize] = ResponseExpectedFlag::AlwaysTrue;
213 result.device.response_expected[u8::from(HatBrickFunction::GetBootloaderMode) as usize] = ResponseExpectedFlag::AlwaysTrue;
214 result.device.response_expected[u8::from(HatBrickFunction::SetWriteFirmwarePointer) as usize] = ResponseExpectedFlag::False;
215 result.device.response_expected[u8::from(HatBrickFunction::WriteFirmware) as usize] = ResponseExpectedFlag::AlwaysTrue;
216 result.device.response_expected[u8::from(HatBrickFunction::SetStatusLedConfig) as usize] = ResponseExpectedFlag::False;
217 result.device.response_expected[u8::from(HatBrickFunction::GetStatusLedConfig) as usize] = ResponseExpectedFlag::AlwaysTrue;
218 result.device.response_expected[u8::from(HatBrickFunction::GetChipTemperature) as usize] = ResponseExpectedFlag::AlwaysTrue;
219 result.device.response_expected[u8::from(HatBrickFunction::Reset) as usize] = ResponseExpectedFlag::False;
220 result.device.response_expected[u8::from(HatBrickFunction::WriteUid) as usize] = ResponseExpectedFlag::False;
221 result.device.response_expected[u8::from(HatBrickFunction::ReadUid) as usize] = ResponseExpectedFlag::AlwaysTrue;
222 result.device.response_expected[u8::from(HatBrickFunction::GetIdentity) as usize] = ResponseExpectedFlag::AlwaysTrue;
223 result
224 }
225
226 pub fn get_response_expected(&mut self, fun: HatBrickFunction) -> Result<bool, GetResponseExpectedError> {
241 self.device.get_response_expected(u8::from(fun))
242 }
243
244 pub fn set_response_expected(&mut self, fun: HatBrickFunction, response_expected: bool) -> Result<(), SetResponseExpectedError> {
253 self.device.set_response_expected(u8::from(fun), response_expected)
254 }
255
256 pub fn set_response_expected_all(&mut self, response_expected: bool) { self.device.set_response_expected_all(response_expected) }
258
259 pub fn get_api_version(&self) -> [u8; 3] { self.device.api_version }
262
263 pub fn get_voltages_callback_receiver(&self) -> ConvertingCallbackReceiver<VoltagesEvent> {
272 self.device.get_callback_receiver(u8::from(HatBrickFunction::CallbackVoltages))
273 }
274
275 pub fn set_sleep_mode(
298 &self,
299 power_off_delay: u32,
300 power_off_duration: u32,
301 raspberry_pi_off: bool,
302 bricklets_off: bool,
303 enable_sleep_indicator: bool,
304 ) -> ConvertingReceiver<()> {
305 let mut payload = vec![0; 11];
306 payload[0..4].copy_from_slice(&<u32>::to_le_byte_vec(power_off_delay));
307 payload[4..8].copy_from_slice(&<u32>::to_le_byte_vec(power_off_duration));
308 payload[8..9].copy_from_slice(&<bool>::to_le_byte_vec(raspberry_pi_off));
309 payload[9..10].copy_from_slice(&<bool>::to_le_byte_vec(bricklets_off));
310 payload[10..11].copy_from_slice(&<bool>::to_le_byte_vec(enable_sleep_indicator));
311
312 self.device.set(u8::from(HatBrickFunction::SetSleepMode), payload)
313 }
314
315 pub fn get_sleep_mode(&self) -> ConvertingReceiver<SleepMode> {
317 let payload = vec![0; 0];
318
319 self.device.get(u8::from(HatBrickFunction::GetSleepMode), payload)
320 }
321
322 pub fn set_bricklet_power(&self, bricklet_power: bool) -> ConvertingReceiver<()> {
324 let mut payload = vec![0; 1];
325 payload[0..1].copy_from_slice(&<bool>::to_le_byte_vec(bricklet_power));
326
327 self.device.set(u8::from(HatBrickFunction::SetBrickletPower), payload)
328 }
329
330 pub fn get_bricklet_power(&self) -> ConvertingReceiver<bool> {
332 let payload = vec![0; 0];
333
334 self.device.get(u8::from(HatBrickFunction::GetBrickletPower), payload)
335 }
336
337 pub fn get_voltages(&self) -> ConvertingReceiver<Voltages> {
352 let payload = vec![0; 0];
353
354 self.device.get(u8::from(HatBrickFunction::GetVoltages), payload)
355 }
356
357 pub fn set_voltages_callback_configuration(&self, period: u32, value_has_to_change: bool) -> ConvertingReceiver<()> {
370 let mut payload = vec![0; 5];
371 payload[0..4].copy_from_slice(&<u32>::to_le_byte_vec(period));
372 payload[4..5].copy_from_slice(&<bool>::to_le_byte_vec(value_has_to_change));
373
374 self.device.set(u8::from(HatBrickFunction::SetVoltagesCallbackConfiguration), payload)
375 }
376
377 pub fn get_voltages_callback_configuration(&self) -> ConvertingReceiver<VoltagesCallbackConfiguration> {
383 let payload = vec![0; 0];
384
385 self.device.get(u8::from(HatBrickFunction::GetVoltagesCallbackConfiguration), payload)
386 }
387
388 pub fn set_rtc_driver(&self, rtc_driver: u8) -> ConvertingReceiver<()> {
403 let mut payload = vec![0; 1];
404 payload[0..1].copy_from_slice(&<u8>::to_le_byte_vec(rtc_driver));
405
406 self.device.set(u8::from(HatBrickFunction::SetRtcDriver), payload)
407 }
408
409 pub fn get_rtc_driver(&self) -> ConvertingReceiver<u8> {
418 let payload = vec![0; 0];
419
420 self.device.get(u8::from(HatBrickFunction::GetRtcDriver), payload)
421 }
422
423 pub fn get_spitfp_error_count(&self) -> ConvertingReceiver<SpitfpErrorCount> {
435 let payload = vec![0; 0];
436
437 self.device.get(u8::from(HatBrickFunction::GetSpitfpErrorCount), payload)
438 }
439
440 pub fn set_bootloader_mode(&self, mode: u8) -> ConvertingReceiver<u8> {
463 let mut payload = vec![0; 1];
464 payload[0..1].copy_from_slice(&<u8>::to_le_byte_vec(mode));
465
466 self.device.get(u8::from(HatBrickFunction::SetBootloaderMode), payload)
467 }
468
469 pub fn get_bootloader_mode(&self) -> ConvertingReceiver<u8> {
478 let payload = vec![0; 0];
479
480 self.device.get(u8::from(HatBrickFunction::GetBootloaderMode), payload)
481 }
482
483 pub fn set_write_firmware_pointer(&self, pointer: u32) -> ConvertingReceiver<()> {
490 let mut payload = vec![0; 4];
491 payload[0..4].copy_from_slice(&<u32>::to_le_byte_vec(pointer));
492
493 self.device.set(u8::from(HatBrickFunction::SetWriteFirmwarePointer), payload)
494 }
495
496 pub fn write_firmware(&self, data: [u8; 64]) -> ConvertingReceiver<u8> {
505 let mut payload = vec![0; 64];
506 payload[0..64].copy_from_slice(&<[u8; 64]>::to_le_byte_vec(data));
507
508 self.device.get(u8::from(HatBrickFunction::WriteFirmware), payload)
509 }
510
511 pub fn set_status_led_config(&self, config: u8) -> ConvertingReceiver<()> {
525 let mut payload = vec![0; 1];
526 payload[0..1].copy_from_slice(&<u8>::to_le_byte_vec(config));
527
528 self.device.set(u8::from(HatBrickFunction::SetStatusLedConfig), payload)
529 }
530
531 pub fn get_status_led_config(&self) -> ConvertingReceiver<u8> {
539 let payload = vec![0; 0];
540
541 self.device.get(u8::from(HatBrickFunction::GetStatusLedConfig), payload)
542 }
543
544 pub fn get_chip_temperature(&self) -> ConvertingReceiver<i16> {
551 let payload = vec![0; 0];
552
553 self.device.get(u8::from(HatBrickFunction::GetChipTemperature), payload)
554 }
555
556 pub fn reset(&self) -> ConvertingReceiver<()> {
563 let payload = vec![0; 0];
564
565 self.device.set(u8::from(HatBrickFunction::Reset), payload)
566 }
567
568 pub fn write_uid(&self, uid: u32) -> ConvertingReceiver<()> {
574 let mut payload = vec![0; 4];
575 payload[0..4].copy_from_slice(&<u32>::to_le_byte_vec(uid));
576
577 self.device.set(u8::from(HatBrickFunction::WriteUid), payload)
578 }
579
580 pub fn read_uid(&self) -> ConvertingReceiver<u32> {
583 let payload = vec![0; 0];
584
585 self.device.get(u8::from(HatBrickFunction::ReadUid), payload)
586 }
587
588 pub fn get_identity(&self) -> ConvertingReceiver<Identity> {
598 let payload = vec![0; 0];
599
600 self.device.get(u8::from(HatBrickFunction::GetIdentity), payload)
601 }
602}