From 19aec13a1060a521913abc6bd669080171d43594 Mon Sep 17 00:00:00 2001
From: gingerBill <bill@gingerbill.org>
Date: Sun, 6 Feb 2022 11:42:59 +0000
Subject: Support rank-2 arrays (matrix-like) for `transpose`

---
 src/types.cpp | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

(limited to 'src/types.cpp')

diff --git a/src/types.cpp b/src/types.cpp
index e0d35a12c..9ee6ba359 100644
--- a/src/types.cpp
+++ b/src/types.cpp
@@ -363,6 +363,7 @@ enum TypeInfoFlag : u32 {
 enum : int {
 	MATRIX_ELEMENT_COUNT_MIN = 1,
 	MATRIX_ELEMENT_COUNT_MAX = 16,
+	MATRIX_ELEMENT_MAX_SIZE = MATRIX_ELEMENT_COUNT_MAX * (2 * 8), // complex128
 };
 
 
@@ -1583,6 +1584,24 @@ Type *core_array_type(Type *t) {
 	}
 }
 
+i32 type_math_rank(Type *t) {
+	i32 rank = 0;
+	for (;;) {
+		t = base_type(t);
+		switch (t->kind) {
+		case Type_Array:
+			rank += 1;
+			t = t->Array.elem;
+			break;
+		case Type_Matrix:
+			rank += 2;
+			t = t->Matrix.elem;
+			break;
+		default:
+			return rank;
+		}
+	}
+}
 
 
 Type *base_complex_elem_type(Type *t) {
-- 
cgit v1.2.3