feat(ufmt): add feature

Cargo.toml

@@ -3,6 +3,14 @@ name = "soft-serial"
 version = "0.1.0"
 edition = "2021"
 
+[dependencies.ufmt]
+version = "*"
+optional = true
+
+[dependencies.ring-buffer]
+git = "https://gitea.doryan04.ru/doryan/ring-buffer"
+optional = true
+
 [dependencies.arduino-hal]
 git = "https://github.com/rahix/avr-hal"
 rev = "3e362624547462928a219c40f9ea8e3a64f21e5f"
@@ -12,6 +20,8 @@ git = "https://gitea.doryan04.ru/TheEmbeddedRust/static-pins"
 
 [features]
 sparkfun-promicro = ["arduino-hal/sparkfun-promicro", "static-pins/sparkfun-promicro"]
+ring-buf = ["dep:ring-buffer"]
+ufmt = ["dep:ufmt"]
 
 [dependencies.avr-device]
 version = "0.5.4"

src/lib.rs

@@ -4,6 +4,7 @@ use core::marker::PhantomData;
 
 use arduino_hal::{
     delay_us,
+    hal::port::{PB4, PB5, PE6},
     port::{
         mode::{Input, PullUp},
         Pin, PinOps,
@@ -13,6 +14,9 @@ use arduino_hal::{
 use avr_device::asm::delay_cycles;
 
 use static_pins::StaticPinOps;
+use ufmt::derive::uDebug;
+
+mod structures;
 
 pub type PollResult = Result<(), PollError>;
 pub type ReadByteResult = Result<u8, CorruptedData>;
@@ -22,11 +26,18 @@ const SERIAL_DELAY: u32 = 4;
 const FIRST_HALF_SERIAL_DELAY: u32 = SERIAL_DELAY / 2;
 const SECOND_HALF_SERIAL_DELAY: u32 = SERIAL_DELAY - FIRST_HALF_SERIAL_DELAY;
 
-const READING_ADJUST: u32 = 16;
-const FIRST_ENTRY_READING: u32 = 30;
-const LSB: u8 = 0x01;
-const MSB: u8 = 0x80;
+const READING_ADJUST: u32 = 10;
+const TX_DELAY_CYCLES: u32 = 22;
+const MSB: u16 = 0x0800;
+const PACKET_MASK: u16 = 0x0FFF;
 
+#[inline(always)]
+fn crc_calculate(value: u8) -> u8 {
+    ((value >> 6) | (value >> 4) | (value >> 2) | value) & 0x0F
+}
+
+#[cfg(feature = "ufmt")]
+#[derive(uDebug)]
 pub enum PollError {
     NotFound,
     NotReady,
@@ -36,40 +47,81 @@ pub struct HalfDuplexSerial<P> {
     _pin: PhantomData<Pin<Input<PullUp>, P>>,
 }
 
+#[cfg(debug_assertions)]
+#[inline(always)]
+fn debug_pb4_low() {
+    PB4::set_low();
+}
+
+#[cfg(debug_assertions)]
+#[inline(always)]
+fn debug_pb4_high() {
+    PB4::set_high();
+}
+
+#[allow(dead_code)]
+#[cfg(debug_assertions)]
+#[inline(always)]
+fn debug_pb4_pulse_positive() {
+    debug_pb4_low();
+    debug_pb4_high();
+}
+
+#[allow(dead_code)]
+#[cfg(debug_assertions)]
+#[inline(always)]
+fn debug_pb4_pulse_negative() {
+    debug_pb4_high();
+    debug_pb4_low();
+}
+
 impl<P> HalfDuplexSerial<P>
 where
     P: PinOps + StaticPinOps,
 {
-    #[inline]
     pub fn new(_pin: Pin<Input<PullUp>, P>) -> Self {
+        if cfg!(debug_assertions) {
+            PB4::into_output_high();
+            PB5::into_output_high();
+            PE6::into_output_high();
+        }
         Self {
             _pin: PhantomData {},
         }
     }
 
     pub fn poll(&self) -> PollResult {
+        PE6::set_low();
         P::into_output();
-        delay_cycles(1);
+
+        delay_cycles(2);
+
         P::into_pull_up_input();
 
-        delay_us(SERIAL_DELAY);
+        delay_us(SERIAL_DELAY + FIRST_HALF_SERIAL_DELAY);
 
         if P::is_low() {
+            PE6::set_high();
             return PollResult::Err(PollError::NotFound);
         }
 
         delay_us(SERIAL_DELAY);
 
         if P::is_high() {
+            PE6::set_high();
             return PollResult::Err(PollError::NotReady);
         }
 
         while P::is_low() {}
 
+        PE6::set_high();
+
         PollResult::Ok(())
     }
 
     pub fn response(&self) {
+        PE6::set_low();
+
         P::into_output_high();
 
         delay_us(FIRST_HALF_SERIAL_DELAY);
@@ -78,23 +130,42 @@ where
 
         delay_us(SERIAL_DELAY);
 
+        P::set_high();
         P::into_pull_up_input();
+
+        PE6::set_high();
     }
 
-    #[inline(never)]
+    #[inline]
     pub fn sync_transmitter(&self) {
+        if cfg!(debug_assertions) {
+            PB5::set_low();
+        }
+
         P::into_output();
 
         delay_us(SERIAL_DELAY);
 
         P::set_high();
+
+        if cfg!(debug_assertions) {
+            PB5::set_high();
+        }
     }
 
-    #[inline(never)]
+    #[inline]
     pub fn sync_reciever(&self) {
+        if cfg!(debug_assertions) {
+            PB5::set_low();
+        }
+
         while P::is_high() {}
 
         while P::is_low() {}
+
+        if cfg!(debug_assertions) {
+            PB5::set_high();
+        }
     }
 
     #[inline]
@@ -103,42 +174,88 @@ where
     }
 }
 
-pub trait SoftSerialWriter<P, T>
+pub trait SoftSerialWriter<P, T>: SoftSerialByteWriter<P>
+where
+    P: PinOps + StaticPinOps,
+{
+    fn write_bytes(&self, transmit_data: T);
+}
+
+pub trait SoftSerialReader<P, T>: SoftSerialByteReader<P>
+where
+    P: PinOps + StaticPinOps,
+{
+    fn read_bytes(&self, recieve_data: T) -> Result<(), CorruptedData>;
+}
+
+pub trait SoftSerialByteWriter<P>
 where
     P: PinOps + StaticPinOps,
 {
     #[inline(never)]
-    fn write_byte(&self, data: u8) {
-        let (mut data, mut parity_bit) = (data, 0);
+    fn write_byte(&self, transmit_data: u8) {
+        let mut data = ((transmit_data as u16) << 4) | (crc_calculate(transmit_data) as u16);
+
+        for _ in 0..(u8::BITS + 4) {
+            debug_pb4_low();
 
-        for _ in 0..8 {
             if data & MSB == 0 {
                 P::set_high();
-                parity_bit ^= 0;
             } else {
                 P::set_low();
-                parity_bit ^= 1;
             }
 
+            data = (data << 1) & PACKET_MASK;
             delay_us(SERIAL_DELAY);
 
-            data <<= 1;
+            debug_pb4_high();
         }
 
-        // Hamming code and CRC are very weightful and slow, so I use simple parity check
-
-        if parity_bit == 0 {
-            P::set_high();
-        } else {
-            P::set_low();
-        }
-
-        delay_us(SERIAL_DELAY);
+        P::set_high();
+        delay_cycles(TX_DELAY_CYCLES);
     }
-
-    fn write_bytes(&self, transmit_data: T);
 }
 
+pub trait SoftSerialByteReader<P>
+where
+    P: PinOps + StaticPinOps,
+{
+    #[inline(never)]
+    fn read_byte(&self) -> ReadByteResult {
+        let mut packet = 0u16;
+
+        for _ in 0..(u8::BITS + 4) {
+            debug_pb4_low();
+
+            delay_us(FIRST_HALF_SERIAL_DELAY);
+            delay_cycles(READING_ADJUST);
+
+            if P::is_low() {
+                packet = (packet << 1) | 1;
+            } else {
+                packet <<= 1;
+            }
+
+            delay_us(SECOND_HALF_SERIAL_DELAY);
+
+            debug_pb4_high();
+        }
+
+        let data = (packet >> 4) as u8;
+        let received_crc = (packet & 0x000F) as u8;
+        let calculated_crc = crc_calculate(data);
+
+        if received_crc != calculated_crc {
+            return Err((received_crc, calculated_crc));
+        }
+
+        Ok(data)
+    }
+}
+
+impl<P: PinOps + StaticPinOps> SoftSerialByteWriter<P> for HalfDuplexSerial<P> {}
+impl<P: PinOps + StaticPinOps> SoftSerialByteReader<P> for HalfDuplexSerial<P> {}
+
 impl<P> SoftSerialWriter<P, &[u8]> for HalfDuplexSerial<P>
 where
     P: PinOps + StaticPinOps,
@@ -151,59 +268,15 @@ where
     }
 }
 
-pub trait SoftSerialReader<P, T>
-where
-    P: PinOps + StaticPinOps,
-{
-    #[inline(never)]
-    fn read_byte(&self) -> ReadByteResult {
-        let (mut data, mut reciever_parity_bit) = (0, 0);
-
-        delay_cycles(FIRST_ENTRY_READING);
-        for _ in 0..8 {
-            delay_us(FIRST_HALF_SERIAL_DELAY);
-
-            data <<= 1;
-
-            if P::is_low() {
-                data |= 1;
-                reciever_parity_bit ^= 1;
-            } else {
-                data |= 0;
-                reciever_parity_bit ^= 0;
-            }
-
-            delay_cycles(READING_ADJUST);
-            delay_us(SECOND_HALF_SERIAL_DELAY);
-        }
-
-        delay_us(FIRST_HALF_SERIAL_DELAY);
-
-        let transmitter_parity_bit = (P::read() >> P::PIN_NUM) & LSB;
-
-        delay_cycles(READING_ADJUST);
-        delay_us(SECOND_HALF_SERIAL_DELAY);
-
-        if reciever_parity_bit == transmitter_parity_bit {
-            return Err((data, reciever_parity_bit));
-        }
-
-        Ok(data)
-    }
-
-    fn read_bytes(&self, recieve_data: T);
-}
-
 impl<P> SoftSerialReader<P, &mut [u8]> for HalfDuplexSerial<P>
 where
     P: PinOps + StaticPinOps,
 {
-    fn read_bytes(&self, recieve_data: &mut [u8]) {
+    fn read_bytes(&self, recieve_data: &mut [u8]) -> Result<(), CorruptedData> {
         for byte in recieve_data {
-            if let Ok(data) = self.read_byte() {
-                *byte = data;
-            }
+            *byte = self.read_byte()?;
             self.sync_reciever();
         }
+        Ok(())
     }
 }

src/structures/mod.rs

@@ -0,0 +1,52 @@
+use arduino_hal::port::PinOps;
+
+use static_pins::StaticPinOps;
+
+use crate::{
+    HalfDuplexSerial, SoftSerialByteReader, SoftSerialByteWriter, SoftSerialReader,
+    SoftSerialWriter,
+};
+
+#[cfg(feature = "ring-buf")]
+pub mod ring_buf {
+    use crate::CorruptedData;
+
+    use super::*;
+    use ring_buffer::RingBuffer;
+
+    impl<const N: usize, P> SoftSerialWriter<P, &mut RingBuffer<N>> for HalfDuplexSerial<P>
+    where
+        P: PinOps + StaticPinOps,
+    {
+        #[inline(always)]
+        fn write_bytes(&self, transmit_data: &mut RingBuffer<N>) {
+            self.write_byte(transmit_data.len() as u8);
+
+            for byte in transmit_data {
+                self.sync_transmitter();
+                self.write_byte(byte);
+            }
+
+            self.sync_transmitter();
+        }
+    }
+
+    impl<const N: usize, P> SoftSerialReader<P, &mut RingBuffer<N>> for HalfDuplexSerial<P>
+    where
+        P: PinOps + StaticPinOps,
+    {
+        #[inline(always)]
+        fn read_bytes(&self, recieve_data: &mut RingBuffer<N>) -> Result<(), CorruptedData> {
+            let len = self.read_byte()?;
+
+            for _ in 0..len {
+                self.sync_reciever();
+                recieve_data.push(self.read_byte()?);
+            }
+
+            self.sync_reciever();
+
+            Ok(())
+        }
+    }
+}