From f21e9ee71281ec8665a3009cd1a350ffde3b7046 Mon Sep 17 00:00:00 2001
From: gingerBill <bill@gingerbill.org>
Date: Wed, 25 May 2022 18:59:47 +0100
Subject: [PATCH] Allow basic casting of simd vectors

---
 src/check_expr.cpp        | 12 ++++++++++++
 src/llvm_backend_expr.cpp | 32 ++++++++++++++++++++++++++++++++
 2 files changed, 44 insertions(+)

diff --git a/src/check_expr.cpp b/src/check_expr.cpp
index a4dfade98..9a7cd5b8b 100644
--- a/src/check_expr.cpp
+++ b/src/check_expr.cpp
@@ -2688,6 +2688,18 @@ bool check_is_castable_to(CheckerContext *c, Operand *operand, Type *y) {
 		return true;
 	}
 
+	if (is_type_simd_vector(src) && is_type_simd_vector(dst)) {
+		if (src->SimdVector.count != dst->SimdVector.count) {
+			return false;
+		}
+		Type *elem_src = base_array_type(src);
+		Type *elem_dst = base_array_type(dst);
+		Operand x = {};
+		x.type = elem_src;
+		x.mode = Addressing_Value;
+		return check_is_castable_to(c, &x, elem_dst);
+	}
+
 	return false;
 }
 
diff --git a/src/llvm_backend_expr.cpp b/src/llvm_backend_expr.cpp
index 133df4d41..f4b5702bb 100644
--- a/src/llvm_backend_expr.cpp
+++ b/src/llvm_backend_expr.cpp
@@ -1820,6 +1820,38 @@ lbValue lb_emit_conv(lbProcedure *p, lbValue value, Type *t) {
 		return res;
 	}
 
+	if (is_type_simd_vector(src) && is_type_simd_vector(dst)) {
+		Type *src_elem = core_array_type(src);
+		Type *dst_elem = core_array_type(dst);
+
+		GB_ASSERT(src->SimdVector.count == dst->SimdVector.count);
+
+		lbValue res = {};
+		res.type = t;
+		if (are_types_identical(src_elem, dst_elem)) {
+			res.value = value.value;
+		} else if (is_type_float(src_elem) && is_type_integer(dst_elem)) {
+			if (is_type_unsigned(dst_elem)) {
+				res.value = LLVMBuildFPToUI(p->builder, value.value, lb_type(m, t), "");
+			} else {
+				res.value = LLVMBuildFPToSI(p->builder, value.value, lb_type(m, t), "");
+			}
+		} else if (is_type_integer(src_elem) && is_type_float(dst_elem)) {
+			if (is_type_unsigned(src_elem)) {
+				res.value = LLVMBuildUIToFP(p->builder, value.value, lb_type(m, t), "");
+			} else {
+				res.value = LLVMBuildSIToFP(p->builder, value.value, lb_type(m, t), "");
+			}
+		} else if (is_type_integer(src_elem) && is_type_integer(dst_elem)) {
+			res.value = LLVMBuildIntCast2(p->builder, value.value, lb_type(m, t), !is_type_unsigned(src_elem), "");
+		} else if (is_type_float(src_elem) && is_type_float(dst_elem)) {
+			res.value = LLVMBuildFPCast(p->builder, value.value, lb_type(m, t), "");
+		} else {
+			GB_PANIC("Unhandled simd vector conversion: %s -> %s", type_to_string(src), type_to_string(dst));
+		}
+		return res;
+	}
+
 	// Pointer <-> uintptr
 	if (is_type_pointer(src) && is_type_uintptr(dst)) {
 		lbValue res = {};