feat: restart tx

Author: AstrickHarren

esp-hal/src/dma/buffers.rs

@@ -1540,7 +1540,7 @@ impl DmaTxStreamBufView {
         let dma_start = self.descriptor_idx * chunk_size + self.descriptor_offset;
         let dma_end = truncate_by(dma_start + buf.len(), dma_size);
 
-        if dma_start < dma_end {
+        if dma_start <= dma_end {
             self.buf.buffer[dma_start..dma_end].copy_from_slice(buf);
         } else {
             self.buf.buffer[dma_start..].copy_from_slice(&buf[..dma_size - dma_start]);
@@ -1576,11 +1576,6 @@ impl DmaTxStreamBufView {
             }
         }
 
-        info!(
-            "self.descriptor_idx: {}, self.descriptor_offset: {}, dma_start: {}, dma_end: {}",
-            self.descriptor_idx, self.descriptor_offset, dma_start, dma_end
-        );
-
         bytes_to_fill
     }
 }

esp-hal/src/dma/mod.rs

@@ -2989,6 +2989,13 @@ pub(crate) mod asynch {
             {
                 self.tx.clear_out(DmaTxInterrupt::Done);
                 Poll::Ready(Ok(()))
+            } else if self
+                .tx
+                .pending_out_interrupts()
+                .contains(DmaTxInterrupt::TotalEof)
+            {
+                self.tx.clear_interrupts();
+                return Poll::Ready(Err(DmaError::Late));
             } else if self
                 .tx
                 .pending_out_interrupts()
@@ -2998,8 +3005,11 @@ pub(crate) mod asynch {
                 Poll::Ready(Err(DmaError::DescriptorError))
             } else {
                 self.tx.waker().register(cx.waker());
-                self.tx
-                    .listen_out(DmaTxInterrupt::Done | DmaTxInterrupt::DescriptorError);
+                self.tx.listen_out(
+                    DmaTxInterrupt::Done
+                        | DmaTxInterrupt::DescriptorError
+                        | DmaTxInterrupt::TotalEof,
+                );
                 Poll::Pending
             }
         }

esp-hal/src/i2s/master.rs

@@ -1704,13 +1704,10 @@ pub mod asynch {
         dma::{
             DmaEligible,
             DmaTxBuffer,
-            ReadBuffer,
             RxCircularState,
-            TxCircularState,
             WriteBuffer,
-            asynch::{DmaRxDoneChFuture, DmaRxFuture, DmaTxDoneChFuture, DmaTxFuture},
+            asynch::{DmaRxDoneChFuture, DmaRxFuture, DmaTxDoneChFuture},
         },
-        i2s::master::private::RegBlock,
     };
 
     impl<'d> I2sTx<'d, Async> {
@@ -1788,6 +1785,17 @@ pub mod asynch {
             self.i2s_tx.tx_channel.is_done()
         }
 
+        /// Stops and restarts the DMA transfer.
+        pub fn restart(self) -> Result<Self, Error> {
+            let (i2s, buf) = self.stop();
+            i2s.write(buf)
+        }
+
+        /// Checks if the DMA transfer has an error.
+        pub fn has_error(&self) -> bool {
+            self.i2s_tx.tx_channel.has_error()
+        }
+
         /// Waits for any DMA process to be made.
         pub async fn process(&mut self) -> Result<(), Error> {
             DmaTxDoneChFuture::new(&mut self.i2s_tx.tx_channel).await?;