From 086b1e48828740ce215ba3609dcb899c8687bfa0 Mon Sep 17 00:00:00 2001
From: mrb0nk500 <b0nk@b0nk.xyz>
Date: Thu, 2 Mar 2023 12:17:44 -0400
Subject: TArray: Make `{,fast_}fill_with()` type generic

---
 context.h | 41 ++++++++++++++++++++++++-----------------
 1 file changed, 24 insertions(+), 17 deletions(-)

(limited to 'context.h')

diff --git a/context.h b/context.h
index 2a3e8dc..09156df 100644
--- a/context.h
+++ b/context.h
@@ -203,6 +203,11 @@ public:
 		return reinterpret_cast<T2 *>(m_data);
 	};
 
+	template<typename T2>
+	size_t size_as() {
+		return byte_size() / sizeof(T2);
+	};
+
 	template<typename T2>
 	TArray<T2, (n * sizeof(T))/sizeof(T2)> &to() {
 		typedef TArray<T2, sizeof(m_data)/sizeof(T2)> to_type;
@@ -217,12 +222,14 @@ public:
 		memset(m_data, val, byte_size());
 	};
 
-	void fill_with(u8 val) {
-		_fill_with<false>(val);
+	template<typename T2>
+	void fill_with(const T2 val) {
+		_fill_with<T2, false>(val);
 	};
 
-	void fast_fill_with(u8 val) {
-		_fill_with<true>(val);
+	template<typename T2>
+	void fast_fill_with(const T2 val) {
+		_fill_with<T2, true>(val);
 	};
 
 	template<typename T2>
@@ -246,26 +253,26 @@ public:
 	};
 
 private:
-	template<bool do_unroll_check>
-	void _fill_with(u8 val) {
-		u8 *bytes = as_bytes();
-		size_t size = byte_size();
+	template<typename T2, bool do_unroll_check>
+	void _fill_with(const T2 val) {
+		T2 *data = as<T2>();
+		size_t size = size_as<T2>();
 		if (do_unroll_check && size <= 8) {
 			int i = 0;
 			switch (size) {
-				case 8: bytes[i++] = val;
-				case 7: bytes[i++] = val;
-				case 6: bytes[i++] = val;
-				case 5: bytes[i++] = val;
-				case 4: bytes[i++] = val;
-				case 3: bytes[i++] = val;
-				case 2: bytes[i++] = val;
-				case 1: bytes[i++] = val;
+				case 8: data[i++] = val;
+				case 7: data[i++] = val;
+				case 6: data[i++] = val;
+				case 5: data[i++] = val;
+				case 4: data[i++] = val;
+				case 3: data[i++] = val;
+				case 2: data[i++] = val;
+				case 1: data[i++] = val;
 				default: break;
 			}
 		} else {
 			for (int i = 0; i < size; ++i) {
-				bytes[i] = val;
+				data[i] = val;
 			}
 		}
 	};
-- 
cgit v1.2.3-13-gbd6f