From d7d324ac76aa13b50ae722519c12fe46e02fd2dc Mon Sep 17 00:00:00 2001
From: psucien <bad_cast@protonmail.com>
Date: Sun, 2 Jun 2024 22:46:40 +0200
Subject: [PATCH] amdgpu: surface bits per element LUT

---
 src/video_core/amdgpu/pixel_format.cpp | 21 +++++++++++++++++----
 src/video_core/amdgpu/pixel_format.h   |  3 ++-
 2 files changed, 19 insertions(+), 5 deletions(-)

diff --git a/src/video_core/amdgpu/pixel_format.cpp b/src/video_core/amdgpu/pixel_format.cpp
index f963370d..5f6eb903 100644
--- a/src/video_core/amdgpu/pixel_format.cpp
+++ b/src/video_core/amdgpu/pixel_format.cpp
@@ -40,17 +40,30 @@ std::string_view NameOf(NumberFormat fmt) {
     }
 }
 
-u32 NumComponents(DataFormat format) {
-    constexpr std::array numComponentsPerElement = {
+int NumComponents(DataFormat format) {
+    constexpr std::array num_components_per_element = {
         0,  1,  1,  2,  1,  2,  3,  3,  4,  4,  4,  2, 4, 3, 4, -1, 3, 4, 4, 4, 2,
         2,  2,  -1, -1, -1, -1, -1, -1, -1, -1, -1, 3, 3, 3, 4, 4,  4, 1, 2, 3, 4,
         -1, -1, 2,  2,  2,  2,  2,  2,  2,  2,  2,  2, 2, 2, 2, 2,  3, 1, 1};
 
     const u32 index = static_cast<u32>(format);
-    if (index >= numComponentsPerElement.size()) {
+    if (index >= num_components_per_element.size()) {
         return 0;
     }
-    return numComponentsPerElement[index];
+    return num_components_per_element[index];
+}
+
+int NumBits(DataFormat format) {
+    const std::array num_bits_per_element = {
+        0,  8,  16, 16, 32, 32, 32, 32, 32, 32, 32, 64, 64, 96, 128, -1, 16, 16, 16, 16, 32,
+        32, 64, -1, -1, -1, -1, -1, -1, -1, -1, -1, 16, 16, 32, 4,   8,  8,  4,  8,  8,  8,
+        -1, -1, 8,  8,  8,  8,  8,  8,  16, 16, 32, 32, 32, 64, 64,  8,  16, 1,  1};
+
+    const u32 index = static_cast<u32>(format);
+    if (index >= num_bits_per_element.size()) {
+        return 0;
+    }
+    return num_bits_per_element[index];
 }
 
 } // namespace AmdGpu
diff --git a/src/video_core/amdgpu/pixel_format.h b/src/video_core/amdgpu/pixel_format.h
index 7555cdb3..22d102af 100644
--- a/src/video_core/amdgpu/pixel_format.h
+++ b/src/video_core/amdgpu/pixel_format.h
@@ -63,7 +63,8 @@ enum class NumberFormat : u32 {
 
 [[nodiscard]] std::string_view NameOf(NumberFormat fmt);
 
-u32 NumComponents(DataFormat format);
+int NumComponents(DataFormat format);
+int NumBits(DataFormat format);
 
 } // namespace AmdGpu