From 034f7d7e38e391e209ffacd063a14628b485800d Mon Sep 17 00:00:00 2001
From: Gabriele Musco <gabmus@disroot.org>
Date: Tue, 26 Dec 2023 11:29:06 +0000
Subject: [PATCH] feat: some traits and functions for wivrn config and related
 enums

---
 src/file_builders/wivrn_config.rs | 47 +++++++++++++++++++++++++++++--
 1 file changed, 45 insertions(+), 2 deletions(-)

diff --git a/src/file_builders/wivrn_config.rs b/src/file_builders/wivrn_config.rs
index 1802e97..17ad4be 100644
--- a/src/file_builders/wivrn_config.rs
+++ b/src/file_builders/wivrn_config.rs
@@ -24,11 +24,11 @@ impl Display for Encoder {
 }
 
 impl Encoder {
-    pub fn iter() -> Iter<'static, Encoder> {
+    pub fn iter() -> Iter<'static, Self> {
         [Self::X264, Self::Nvenc, Self::Vaapi].iter()
     }
 
-    pub fn as_vec() -> Vec<Encoder> {
+    pub fn as_vec() -> Vec<Self> {
         vec![Self::X264, Self::Nvenc, Self::Vaapi]
     }
 
@@ -50,6 +50,36 @@ pub enum Codec {
     Hevc,
 }
 
+impl Display for Codec {
+    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
+        f.write_str(match self {
+            Self::H264 => "h264",
+            Self::H265 => "h265",
+            Self::Avc => "AVC",
+            Self::Hevc => "HEVC",
+        })
+    }
+}
+
+impl Codec {
+    pub fn iter() -> Iter<'static, Self> {
+        [Self::H264, Self::H265, Self::Avc, Self::Hevc].iter()
+    }
+
+    pub fn as_vec() -> Vec<Self> {
+        vec![Self::H264, Self::H265, Self::Avc, Self::Hevc]
+    }
+
+    pub fn as_number(&self) -> u32 {
+        match self {
+            Self::H264 => 0,
+            Self::H265 => 1,
+            Self::Avc => 2,
+            Self::Hevc => 3,
+        }
+    }
+}
+
 #[derive(Debug, Clone, PartialEq, Serialize, Deserialize)]
 pub struct WivrnConfEncoder {
     pub encoder: Encoder,
@@ -64,6 +94,19 @@ pub struct WivrnConfEncoder {
     pub group: Option<i32>,
 }
 
+impl Default for WivrnConfEncoder {
+    fn default() -> Self {
+        Self {
+            encoder: Encoder::X264,
+            codec: Codec::H264,
+            bitrate: None,
+            width: None,
+            height: None,
+            group: None,
+        }
+    }
+}
+
 #[derive(Debug, Clone, PartialEq, Serialize, Deserialize)]
 pub struct WivrnConfig {
     #[serde(skip_serializing_if = "Option::is_none")]