lib: add direction method to QueueType

Add direction method to QueueType. This method return an enum, which value determines the direction (output, capture) of the queue, given its type. Signed-off-by: Albert Esteve <aesteve@redhat.com>
lu-zero · Sep 25, 2023 · 2286f4c · 2286f4c
1 parent a5e3b2f
commit 2286f4c
Showing 1 changed file with 28 additions and 0 deletions.
diff --git a/lib/src/lib.rs b/lib/src/lib.rs
@@ -36,6 +36,13 @@ use thiserror::Error;
 // ioctl: direct, safe counterparts of the V4L2 ioctls.
 // device/queue/buffer: higher abstraction, still mapping to core V4L2 mechanics.
 
+/// Possible directions for the queue
+#[derive(Clone, Copy, Debug, PartialEq, Eq, PartialOrd, Ord)]
+pub enum QueueDirection {
+    Output,
+    Capture,
+}
+
 /// Types of queues currently supported by this library.
 #[allow(unused)]
 #[derive(Clone, Copy, Debug, PartialEq, Eq, PartialOrd, Ord, N)]
@@ -64,6 +71,27 @@ impl QueueType {
             QueueType::VideoCaptureMplane | QueueType::VideoOutputMplane
         )
     }
+
+    /// Returns the direction of the queue type (Output or Capture).
+    pub fn direction(&self) -> QueueDirection {
+        match self {
+            QueueType::VideoOutput
+            | QueueType::VideoOutputMplane
+            | QueueType::VideoOverlay
+            | QueueType::VideoOutputOverlay
+            | QueueType::VbiOutput
+            | QueueType::SlicedVbiOutput
+            | QueueType::SdrOutput
+            | QueueType::MetaOutput => QueueDirection::Output,
+
+            QueueType::VideoCapture
+            | QueueType::VbiCapture
+            | QueueType::SlicedVbiCapture
+            | QueueType::VideoCaptureMplane
+            | QueueType::SdrCapture
+            | QueueType::MetaCapture => QueueDirection::Capture,
+        }
+    }
 }
 
 impl Display for QueueType {