export packet size as constant

Usefull for SPI, where we need provide read buffer simultaneously with write buffer
ublox-rs · Mar 13, 2020 · 0f45945 · 0f45945
1 parent 681e328
commit 0f45945
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 5 deletions.
diff --git a/ublox/src/ubx_packets.rs b/ublox/src/ubx_packets.rs
@@ -94,6 +94,8 @@ pub struct UbxPacketRequest {
 }
 
 impl UbxPacketRequest {
+    pub const PACKET_LEN: usize = 8;
+
     #[inline]
     pub fn request_for<T: UbxPacketMeta>() -> Self {
         Self {
@@ -107,7 +109,7 @@ impl UbxPacketRequest {
     }
 
     #[inline]
-    pub fn to_packet_bytes(self) -> [u8; 8] {
+    pub fn to_packet_bytes(self) -> [u8; Self::PACKET_LEN] {
         let mut ret = [
             SYNC_CHAR_1,
             SYNC_CHAR_2,

diff --git a/ublox_derive/src/output.rs b/ublox_derive/src/output.rs
@@ -245,9 +245,11 @@ pub fn generate_send_code_for_packet(pack_descr: &PackDesc) -> Vec<TokenStream>
         let packet_payload_size_u16 = u16::try_from(packet_payload_size).unwrap();
         ret.push(quote! {
             impl #payload_struct {
+                pub const PACKET_LEN: usize = #packet_size;
+
                 #[inline]
-                pub fn to_packet_bytes(self) -> [u8; #packet_size] {
-                    let mut ret = [0u8; #packet_size];
+                pub fn to_packet_bytes(self) -> [u8; Self::PACKET_LEN] {
+                    let mut ret = [0u8; Self::PACKET_LEN];
                     ret[0] = SYNC_CHAR_1;
                     ret[1] = SYNC_CHAR_2;
                     ret[2] = #main_name::CLASS;

diff --git a/ublox_derive/src/tests.rs b/ublox_derive/src/tests.rs
@@ -222,9 +222,11 @@ fn test_ubx_packet_send() {
                 pub a: u8,
             }
             impl TestBuilder {
+                pub const PACKET_LEN: usize = 17usize;
+
                 #[inline]
-                pub fn to_packet_bytes(self) -> [u8; 17usize] {
-                    let mut ret = [0u8; 17usize];
+                pub fn to_packet_bytes(self) -> [u8; Self::PACKET_LEN] {
+                    let mut ret = [0u8; Self::PACKET_LEN];
                     ret[0] = SYNC_CHAR_1;
                     ret[1] = SYNC_CHAR_2;
                     ret[2] = Test::CLASS;