From ba43b0930f82b15cf828bc3c6915443fab900700 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?T=C3=96R=C3=96K=20Attila?= <torokati44@gmail.com>
Date: Sun, 19 Mar 2023 21:08:40 +0100
Subject: [PATCH] render: Add Bitmap::to_rgb()

---
 render/src/bitmap.rs | 26 ++++++++++++++++++++++++++
 1 file changed, 26 insertions(+)

diff --git a/render/src/bitmap.rs b/render/src/bitmap.rs
index a046b5977..ed6140626 100644
--- a/render/src/bitmap.rs
+++ b/render/src/bitmap.rs
@@ -88,6 +88,32 @@ impl Bitmap {
         }
     }
 
+    pub fn to_rgb(mut self) -> Self {
+        // Converts this bitmap to RGB, if it is not already.
+        match self.format {
+            BitmapFormat::Rgb => {} // no-op
+            BitmapFormat::Rgba => unreachable!("Can't convert RGBA Bitmap to RGB"),
+            BitmapFormat::Yuv420p => {
+                let luma_len = (self.width * self.height) as usize;
+                let chroma_len = (self.chroma_width() * self.chroma_height()) as usize;
+
+                let y = &self.data[0..luma_len];
+                let u = &self.data[luma_len..luma_len + chroma_len];
+                let v = &self.data[luma_len + chroma_len..luma_len + 2 * chroma_len];
+
+                self.data = yuv420_to_rgba(y, u, v, self.width as usize)
+                    .chunks_exact(4)
+                    .flat_map(|rgba| [rgba[0], rgba[1], rgba[2]])
+                    .collect();
+            }
+            BitmapFormat::Yuva420p => unreachable!("Can't convert YUVA Bitmap to RGB"),
+        }
+
+        self.format = BitmapFormat::Rgb;
+
+        self
+    }
+
     pub fn to_rgba(mut self) -> Self {
         // Converts this bitmap to RGBA, if it is not already.
         match self.format {