pdo: add common PDO trait

OpenDevicePartnership · Feb 10, 2025 · 286160b · 286160b
1 parent d7ea0bd
commit 286160b
Show file tree

Hide file tree

Showing 3 changed files with 76 additions and 0 deletions.
diff --git a/src/pdo/mod.rs b/src/pdo/mod.rs
@@ -110,3 +110,35 @@ impl TryFrom<u32> for ApdoKind {
         }
     }
 }
+
+/// Common PDO trait
+pub trait Common {
+    /// Get the PDO kind
+    fn kind(&self) -> PdoKind;
+    /// Get the APDO kind
+    fn apdo_kind(&self) -> Option<ApdoKind>;
+}
+
+/// Top-level PDO type
+#[derive(Copy, Clone, Debug, PartialEq, Eq)]
+#[cfg_attr(feature = "defmt", derive(defmt::Format))]
+pub enum Pdo {
+    Source(source::Pdo),
+    Sink(sink::Pdo),
+}
+
+impl Common for Pdo {
+    fn kind(&self) -> PdoKind {
+        match self {
+            Pdo::Source(pdo) => pdo.kind(),
+            Pdo::Sink(pdo) => pdo.kind(),
+        }
+    }
+
+    fn apdo_kind(&self) -> Option<ApdoKind> {
+        match self {
+            Pdo::Source(pdo) => pdo.apdo_kind(),
+            Pdo::Sink(pdo) => pdo.apdo_kind(),
+        }
+    }
+}
diff --git a/src/pdo/sink.rs b/src/pdo/sink.rs
@@ -18,6 +18,28 @@ pub enum Pdo {
     Augmented(Apdo),
 }
 
+impl Common for Pdo {
+    fn kind(&self) -> PdoKind {
+        match self {
+            Pdo::Fixed(_) => PdoKind::Fixed,
+            Pdo::Battery(_) => PdoKind::Battery,
+            Pdo::Variable(_) => PdoKind::Variable,
+            Pdo::Augmented(_) => PdoKind::Augmented,
+        }
+    }
+
+    fn apdo_kind(&self) -> Option<ApdoKind> {
+        match self {
+            Pdo::Augmented(apdo) => Some(match apdo {
+                Apdo::SprPps(_) => ApdoKind::SprPps,
+                Apdo::EprAvs(_) => ApdoKind::EprAvs,
+                Apdo::SprAvs(_) => ApdoKind::SprAvs,
+            }),
+            _ => None,
+        }
+    }
+}
+
 impl TryFrom<u32> for Pdo {
     type Error = PdError;
 

diff --git a/src/pdo/source.rs b/src/pdo/source.rs
@@ -18,6 +18,28 @@ pub enum Pdo {
     Augmented(Apdo),
 }
 
+impl Common for Pdo {
+    fn kind(&self) -> PdoKind {
+        match self {
+            Pdo::Fixed(_) => PdoKind::Fixed,
+            Pdo::Battery(_) => PdoKind::Battery,
+            Pdo::Variable(_) => PdoKind::Variable,
+            Pdo::Augmented(_) => PdoKind::Augmented,
+        }
+    }
+
+    fn apdo_kind(&self) -> Option<ApdoKind> {
+        match self {
+            Pdo::Augmented(apdo) => Some(match apdo {
+                Apdo::SprPps(_) => ApdoKind::SprPps,
+                Apdo::EprAvs(_) => ApdoKind::EprAvs,
+                Apdo::SprAvs(_) => ApdoKind::SprAvs,
+            }),
+            _ => None,
+        }
+    }
+}
+
 impl TryFrom<u32> for Pdo {
     type Error = PdError;