Merge pull request #21 from orukusaki/dma-write-buffer

Add impl for embedded_dma::WriteBuffer
bernii · Jul 14, 2023 · 641eb5c · 641eb5c
2 parents 1aa1ce2 + 48df5b2
commit 641eb5c
Showing 1 changed file with 15 additions and 1 deletion.
diff --git a/src/lib.rs b/src/lib.rs
@@ -40,7 +40,7 @@
 //! ```
 
 #![no_std]
-use embedded_dma::ReadBuffer;
+use embedded_dma::{ReadBuffer, WriteBuffer};
 use embedded_graphics::{
     draw_target::DrawTarget,
     geometry::OriginDimensions,
@@ -281,6 +281,20 @@ unsafe impl<C: PixelColor, B: DMACapableFrameBufferBackend<Color = C>> ReadBuffe
     }
 }
 
+unsafe impl<C: PixelColor, B: DMACapableFrameBufferBackend<Color = C>> WriteBuffer
+    for FrameBuf<C, B>
+{
+    type Word = u8;
+    unsafe fn write_buffer(&mut self) -> (*mut Self::Word, usize) {
+        (
+            (self.data.data_ptr() as *mut Self::Word),
+            self.height
+                * self.width
+                * (core::mem::size_of::<C>() / core::mem::size_of::<Self::Word>()),
+        )
+    }
+}
+
 #[cfg(test)]
 mod tests {
     extern crate std;