From 8daff5e677e4975852120a4f9c6b826a73d2df69 Mon Sep 17 00:00:00 2001
From: Connor McLaughlin <stenzek@gmail.com>
Date: Wed, 17 Jun 2020 22:46:13 +1000
Subject: [PATCH] Common: Add dimensional_array.h

---
 src/common/CMakeLists.txt         |  1 +
 src/common/common.vcxproj         |  1 +
 src/common/common.vcxproj.filters |  1 +
 src/common/dimensional_array.h    | 80 +++++++++++++++++++++++++++++++
 4 files changed, 83 insertions(+)
 create mode 100644 src/common/dimensional_array.h

diff --git a/src/common/CMakeLists.txt b/src/common/CMakeLists.txt
index 336dff1b4..768174854 100644
--- a/src/common/CMakeLists.txt
+++ b/src/common/CMakeLists.txt
@@ -22,6 +22,7 @@ add_library(common
   cpu_detect.h
   cubeb_audio_stream.cpp
   cubeb_audio_stream.h
+  dimensional_array.h
   event.cpp
   event.h
   fifo_queue.h
diff --git a/src/common/common.vcxproj b/src/common/common.vcxproj
index 9848168ab..57f492b4c 100644
--- a/src/common/common.vcxproj
+++ b/src/common/common.vcxproj
@@ -50,6 +50,7 @@
     <ClInclude Include="d3d11\staging_texture.h" />
     <ClInclude Include="d3d11\stream_buffer.h" />
     <ClInclude Include="d3d11\texture.h" />
+    <ClInclude Include="dimensional_array.h" />
     <ClInclude Include="event.h" />
     <ClInclude Include="fifo_queue.h" />
     <ClInclude Include="file_system.h" />
diff --git a/src/common/common.vcxproj.filters b/src/common/common.vcxproj.filters
index b7d3b0b47..ea31c228d 100644
--- a/src/common/common.vcxproj.filters
+++ b/src/common/common.vcxproj.filters
@@ -66,6 +66,7 @@
     </ClInclude>
     <ClInclude Include="window_info.h" />
     <ClInclude Include="cd_image_hasher.h" />
+    <ClInclude Include="dimensional_array.h" />
   </ItemGroup>
   <ItemGroup>
     <ClCompile Include="jit_code_buffer.cpp" />
diff --git a/src/common/dimensional_array.h b/src/common/dimensional_array.h
new file mode 100644
index 000000000..459cd705b
--- /dev/null
+++ b/src/common/dimensional_array.h
@@ -0,0 +1,80 @@
+// Sourced from https://github.com/BreadFish64/ScaleFish/blob/master/common/dimensional_array.hpp
+// Copyright (c) 2020 BreadFish64
+//
+// Permission is hereby granted, free of charge, to any person obtaining a copy
+// of this software and associated documentation files (the "Software"), to deal
+// in the Software without restriction, including without limitation the rights
+// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+// copies of the Software, and to permit persons to whom the Software is
+// furnished to do so, subject to the following conditions:
+//
+// The above copyright notice and this permission notice shall be included in all
+// copies or substantial portions of the Software.
+//
+// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+// SOFTWARE.
+
+#pragma once
+
+#include <array>
+#include <cstdint>
+#include <type_traits>
+
+namespace detail {
+template<typename T, std::size_t rank, std::size_t... sizes>
+struct DimensionalArrayExplicitRank;
+
+template<std::size_t rank, std::size_t... sizes>
+constexpr std::size_t GetRankSize()
+{
+  constexpr std::size_t size_array[] = {sizes...};
+  return size_array[rank - 1];
+}
+
+template<typename T, std::size_t rank, std::size_t... sizes>
+using GetArrayImplType =
+  std::array<std::conditional_t<rank == 1, T, DimensionalArrayExplicitRank<T, rank - 1, sizes...>>,
+             GetRankSize<rank, sizes...>()>;
+
+template<typename T, std::size_t rank_param, std::size_t... sizes>
+struct DimensionalArrayExplicitRank : public GetArrayImplType<T, rank_param, sizes...>
+{
+  static constexpr std::size_t rank = rank_param;
+  static_assert(rank > 0, "Attempted to create dimensional array with rank less than 1");
+  using ArrayImplType = GetArrayImplType<T, rank, sizes...>;
+  using ArrayImplType::ArrayImplType;
+
+  template<typename Callable>
+  void enumerate(const Callable& f)
+  {
+    for (auto& it : *this)
+    {
+      if constexpr (rank == 1)
+        f(it);
+      else
+        it.enumerate(f);
+    }
+  }
+
+  template<typename Callable>
+  void enumerate(const Callable& f) const
+  {
+    for (const auto& it : *this)
+    {
+      if constexpr (rank == 1)
+        f(it);
+      else
+        it.enumerate(f);
+    }
+  }
+};
+} // namespace detail
+
+template<typename T, std::size_t... sizes>
+using DimensionalArray = detail::DimensionalArrayExplicitRank<T, sizeof...(sizes), sizes...>;
+