From 8b885b09025450a2ac04838009ea1908c5864479 Mon Sep 17 00:00:00 2001
From: "Ernest M. van der Linden" <ernst.vanderlinden@ernestoz.com>
Date: Mon, 6 Jan 2025 19:01:07 +0100
Subject: [PATCH] Implemented macro for format_groups

Replaced manual const arrays with a macro-based approach for format groups.

All existing functionality remains unchanged.
---
 nokhwa-core/src/frame_format.rs | 128 +++++++++++++-------------------
 1 file changed, 52 insertions(+), 76 deletions(-)

diff --git a/nokhwa-core/src/frame_format.rs b/nokhwa-core/src/frame_format.rs
index 1776069..f40c4a6 100644
--- a/nokhwa-core/src/frame_format.rs
+++ b/nokhwa-core/src/frame_format.rs
@@ -79,82 +79,58 @@ pub enum FrameFormat {
     Custom([u8; 8]),
 }
 
-// FIXME: Fix these frame format lists! Maybe move to using a macro..?
-impl FrameFormat {
-    pub const ALL: &'static [FrameFormat] = &[
-        FrameFormat::H263,
-        FrameFormat::H264,
-        FrameFormat::H265,
-        FrameFormat::Av1,
-        FrameFormat::Avc1,
-        FrameFormat::Mpeg1,
-        FrameFormat::Mpeg2,
-        FrameFormat::Mpeg4,
-        FrameFormat::MJpeg,
-        FrameFormat::XVid,
-        FrameFormat::VP8,
-        FrameFormat::VP9,
-        FrameFormat::Yuyv422,
-        FrameFormat::Uyvy422,
-        FrameFormat::Nv12,
-        FrameFormat::Nv21,
-        FrameFormat::Yv12,
-        FrameFormat::Luma8,
-        FrameFormat::Luma16,
-        FrameFormat::Rgb332,
-        FrameFormat::RgbA8888,
-    ];
-
-    pub const COMPRESSED: &'static [FrameFormat] = &[
-        FrameFormat::H263,
-        FrameFormat::H264,
-        FrameFormat::H265,
-        FrameFormat::Av1,
-        FrameFormat::Avc1,
-        FrameFormat::Mpeg1,
-        FrameFormat::Mpeg2,
-        FrameFormat::Mpeg4,
-        FrameFormat::MJpeg,
-        FrameFormat::XVid,
-        FrameFormat::VP8,
-        FrameFormat::VP9,
-    ];
-
-    pub const CHROMA: &'static [FrameFormat] = &[
-        FrameFormat::Yuyv422,
-        FrameFormat::Uyvy422,
-        FrameFormat::Nv12,
-        FrameFormat::Nv21,
-        FrameFormat::Yv12,
-    ];
-
-    pub const LUMA: &'static [FrameFormat] = &[FrameFormat::Luma8, FrameFormat::Luma16];
-
-    pub const RGB: &'static [FrameFormat] = &[FrameFormat::Rgb332, FrameFormat::RgbA8888];
-
-    pub const COLOR_FORMATS: &'static [FrameFormat] = &[
-        FrameFormat::H265,
-        FrameFormat::H264,
-        FrameFormat::H263,
-        FrameFormat::Av1,
-        FrameFormat::Avc1,
-        FrameFormat::Mpeg1,
-        FrameFormat::Mpeg2,
-        FrameFormat::Mpeg4,
-        FrameFormat::MJpeg,
-        FrameFormat::XVid,
-        FrameFormat::VP8,
-        FrameFormat::VP9,
-        FrameFormat::Yuyv422,
-        FrameFormat::Uyvy422,
-        FrameFormat::Nv12,
-        FrameFormat::Nv21,
-        FrameFormat::Yv12,
-        FrameFormat::Rgb332,
-        FrameFormat::RgbA8888,
-    ];
-
-    pub const GRAYSCALE: &'static [FrameFormat] = &[FrameFormat::Luma8, FrameFormat::Luma16];
+macro_rules! define_frame_format_groups {
+    (
+        $(
+            $group_name:ident => [
+                $($format:ident),* $(,)?
+            ]
+        ),* $(,)?
+    ) => {
+        impl FrameFormat {
+            $(
+                pub const $group_name: &'static [FrameFormat] = &[
+                    $(FrameFormat::$format),*
+                ];
+            )*
+        }
+    };
+}
+
+define_frame_format_groups! {
+    ALL => [
+        H263, H264, H265, Av1, Avc1, Mpeg1, Mpeg2, Mpeg4, MJpeg, XVid,
+        VP8, VP9, Yuyv422, Uyvy422, Nv12, Nv21, Yv12, Ayuv444, Yvyu422,
+        I420, Yvu9, Luma8, Luma16, Depth16, Rgb332, Rgb555, Rgb565,
+        Rgb888, RgbA8888, ARgb8888, Bayer8, Bayer16
+    ],
+    COMPRESSED => [
+        H263, H264, H265, Av1, Avc1, Mpeg1, Mpeg2, Mpeg4, MJpeg, XVid,
+        VP8, VP9
+    ],
+    CHROMA => [
+        Yuyv422, Uyvy422, Nv12, Nv21, Yv12, Ayuv444, Yvyu422, I420, Yvu9
+    ],
+    LUMA => [
+        Luma8, Luma16
+    ],
+    RGB => [
+        Rgb332, Rgb555, Rgb565, Rgb888, RgbA8888, ARgb8888
+    ],
+    COLOR_FORMATS => [
+        H265, H264, H263, Av1, Avc1, Mpeg1, Mpeg2, Mpeg4, MJpeg, XVid,
+        VP8, VP9, Yuyv422, Uyvy422, Nv12, Nv21, Yv12, Ayuv444, Yvyu422,
+        I420, Rgb332, Rgb555, Rgb565, Rgb888, RgbA8888, ARgb8888
+    ],
+    GRAYSCALE => [
+        Luma8, Luma16
+    ],
+    DEPTH => [
+        Depth16
+    ],
+    BAYER => [
+        Bayer8, Bayer16
+    ]
 }
 
 impl Display for FrameFormat {