embassy-rp: Move Spinlock implementation out of critical_section_impl

3 files changed, 77 insertions, 42 deletions

diff --git a/embassy-rp/src/critical_section_impl.rs b/embassy-rp/src/critical_section_impl.rs
index d233e6fab..2e4e8f716 100644
--- a/embassy-rp/src/critical_section_impl.rs
+++ b/embassy-rp/src/critical_section_impl.rs
@@ -1,6 +1,7 @@
 use core::sync::atomic::{AtomicU8, Ordering};
 
 use crate::pac;
+use crate::spinlock::Spinlock;
 
 struct RpSpinlockCs;
 critical_section::set_impl!(RpSpinlockCs);
@@ -92,46 +93,4 @@ impl RpSpinlockCs {
     }
 }
 
-pub struct Spinlock<const N: usize>(core::marker::PhantomData<()>)
-where
-    Spinlock<N>: SpinlockValid;
-
-impl<const N: usize> Spinlock<N>
-where
-    Spinlock<N>: SpinlockValid,
-{
-    /// Try to claim the spinlock. Will return `Some(Self)` if the lock is obtained, and `None` if the lock is
-    /// already in use somewhere else.
-    pub fn try_claim() -> Option<Self> {
-        let lock = pac::SIO.spinlock(N).read();
-        if lock > 0 {
-            Some(Self(core::marker::PhantomData))
-        } else {
-            None
-        }
-    }
-
-    /// Clear a locked spin-lock.
-    ///
-    /// # Safety
-    ///
-    /// Only call this function if you hold the spin-lock.
-    pub unsafe fn release() {
-        // Write (any value): release the lock
-        pac::SIO.spinlock(N).write_value(1);
-    }
-}
-
-impl<const N: usize> Drop for Spinlock<N>
-where
-    Spinlock<N>: SpinlockValid,
-{
-    fn drop(&mut self) {
-        // This is safe because we own the object, and hence hold the lock.
-        unsafe { Self::release() }
-    }
-}
-
 pub(crate) type Spinlock31 = Spinlock<31>;
-pub trait SpinlockValid {}
-impl SpinlockValid for Spinlock<31> {}
diff --git a/embassy-rp/src/lib.rs b/embassy-rp/src/lib.rs
index 35099d07b..01d26b8e9 100644
--- a/embassy-rp/src/lib.rs
+++ b/embassy-rp/src/lib.rs
@@ -41,6 +41,7 @@ pub mod rom_data;
 #[cfg(feature = "rp2040")]
 pub mod rtc;
 pub mod spi;
+mod spinlock;
 #[cfg(feature = "time-driver")]
 pub mod time_driver;
 #[cfg(feature = "_rp235x")]
diff --git a/embassy-rp/src/spinlock.rs b/embassy-rp/src/spinlock.rs
new file mode 100644
index 000000000..7effd2ae0
--- /dev/null
+++ b/embassy-rp/src/spinlock.rs
@@ -0,0 +1,75 @@
+use crate::pac;
+
+pub struct Spinlock<const N: usize>(core::marker::PhantomData<()>)
+where
+    Spinlock<N>: SpinlockValid;
+
+impl<const N: usize> Spinlock<N>
+where
+    Spinlock<N>: SpinlockValid,
+{
+    /// Try to claim the spinlock. Will return `Some(Self)` if the lock is obtained, and `None` if the lock is
+    /// already in use somewhere else.
+    pub fn try_claim() -> Option<Self> {
+        let lock = pac::SIO.spinlock(N).read();
+        if lock > 0 {
+            Some(Self(core::marker::PhantomData))
+        } else {
+            None
+        }
+    }
+
+    /// Clear a locked spin-lock.
+    ///
+    /// # Safety
+    ///
+    /// Only call this function if you hold the spin-lock.
+    pub unsafe fn release() {
+        // Write (any value): release the lock
+        pac::SIO.spinlock(N).write_value(1);
+    }
+}
+
+impl<const N: usize> Drop for Spinlock<N>
+where
+    Spinlock<N>: SpinlockValid,
+{
+    fn drop(&mut self) {
+        // This is safe because we own the object, and hence hold the lock.
+        unsafe { Self::release() }
+    }
+}
+
+pub trait SpinlockValid {}
+impl SpinlockValid for Spinlock<0> {}
+impl SpinlockValid for Spinlock<1> {}
+impl SpinlockValid for Spinlock<2> {}
+impl SpinlockValid for Spinlock<3> {}
+impl SpinlockValid for Spinlock<4> {}
+impl SpinlockValid for Spinlock<5> {}
+impl SpinlockValid for Spinlock<6> {}
+impl SpinlockValid for Spinlock<7> {}
+impl SpinlockValid for Spinlock<8> {}
+impl SpinlockValid for Spinlock<9> {}
+impl SpinlockValid for Spinlock<10> {}
+impl SpinlockValid for Spinlock<11> {}
+impl SpinlockValid for Spinlock<12> {}
+impl SpinlockValid for Spinlock<13> {}
+impl SpinlockValid for Spinlock<14> {}
+impl SpinlockValid for Spinlock<15> {}
+impl SpinlockValid for Spinlock<16> {}
+impl SpinlockValid for Spinlock<17> {}
+impl SpinlockValid for Spinlock<18> {}
+impl SpinlockValid for Spinlock<19> {}
+impl SpinlockValid for Spinlock<20> {}
+impl SpinlockValid for Spinlock<21> {}
+impl SpinlockValid for Spinlock<22> {}
+impl SpinlockValid for Spinlock<23> {}
+impl SpinlockValid for Spinlock<24> {}
+impl SpinlockValid for Spinlock<25> {}
+impl SpinlockValid for Spinlock<26> {}
+impl SpinlockValid for Spinlock<27> {}
+impl SpinlockValid for Spinlock<28> {}
+impl SpinlockValid for Spinlock<29> {}
+impl SpinlockValid for Spinlock<30> {}
+impl SpinlockValid for Spinlock<31> {}