From e0bbb1b7ea5c7bc24763d1863a94ef74f0966ccb Mon Sep 17 00:00:00 2001
From: Brad Lewis <22850972+BradLewis@users.noreply.github.com>
Date: Wed, 15 Oct 2025 17:08:48 -0400
Subject: Format matrix comp lits that use ranges

---
 odinfmt.sh                                          |  2 +-
 src/odin/printer/visit.odin                         | 21 +++++++++++++++------
 tools/odinfmt/tests/.snapshots/matrix_comp_lit.odin |  5 +++++
 tools/odinfmt/tests/matrix_comp_lit.odin            |  7 +++++++
 4 files changed, 28 insertions(+), 7 deletions(-)
 create mode 100644 tools/odinfmt/tests/.snapshots/matrix_comp_lit.odin
 create mode 100644 tools/odinfmt/tests/matrix_comp_lit.odin

diff --git a/odinfmt.sh b/odinfmt.sh
index fa37947..40ba5d3 100755
--- a/odinfmt.sh
+++ b/odinfmt.sh
@@ -4,7 +4,7 @@ if [[ $1 == "debug" ]]
 then
     shift
 
-	odin build tools/odinfmt/main.odin -file -show-timings -collection:src=src -out:odinfmt -o:none
+	odin build tools/odinfmt/main.odin -file -show-timings -collection:src=src -out:odinfmt -o:none -debug
     exit 0
 fi
 
diff --git a/src/odin/printer/visit.odin b/src/odin/printer/visit.odin
index 734a3a3..8c820d4 100644
--- a/src/odin/printer/visit.odin
+++ b/src/odin/printer/visit.odin
@@ -1808,13 +1808,13 @@ visit_expr :: proc(
 			document = cons_with_nopl(document, group(visit_expr(p, v.type)))
 
 			if matrix_type, ok := v.type.derived.(^ast.Matrix_Type);
-			   ok && len(v.elems) > 0 && is_matrix_type_constant(matrix_type) {
+			   ok && is_matrix_type_constant(matrix_type) && is_matrix_filled_comp_lit(matrix_type, v) {
 				document = cons(document, visit_begin_brace(p, v.pos, .Comp_Lit))
 
 				set_source_position(p, v.open)
 				document = cons(
 					document,
-					nest(cons(newline_position(p, 1, v.elems[0].pos), visit_matrix_comp_lit(p, v, matrix_type))),
+					nest(cons(newline_position(p, 1, v.elems[0].pos), visit_matrix_comp_lit(p, matrix_type, v))),
 				)
 				set_source_position(p, v.end)
 
@@ -1911,7 +1911,6 @@ visit_expr :: proc(
 	case ^ast.Tag_Expr:
 		document = cons(
 			text(v.op.text),
-			break_with_no_newline(),
 			text(v.name),
 			break_with_no_newline(),
 			visit_expr(p, v.expr),
@@ -1945,13 +1944,23 @@ is_matrix_type_constant :: proc(matrix_type: ^ast.Matrix_Type) -> bool {
 }
 
 @(private)
-visit_matrix_comp_lit :: proc(p: ^Printer, comp_lit: ^ast.Comp_Lit, matrix_type: ^ast.Matrix_Type) -> ^Document {
-	document := empty()
-
+is_matrix_filled_comp_lit :: proc(matrix_type: ^ast.Matrix_Type, comp_lit: ^ast.Comp_Lit) -> bool {
 	//these values have already been validated
 	row_count, _ := strconv.parse_int(matrix_type.row_count.derived.(^ast.Basic_Lit).tok.text)
 	column_count, _ := strconv.parse_int(matrix_type.column_count.derived.(^ast.Basic_Lit).tok.text)
 
+	if row_count * column_count > len(comp_lit.elems) {
+		return false
+	}
+	return true
+}
+
+@(private)
+visit_matrix_comp_lit :: proc(p: ^Printer, matrix_type: ^ast.Matrix_Type, comp_lit: ^ast.Comp_Lit) -> ^Document {
+	document := empty()
+	row_count, _ := strconv.parse_int(matrix_type.row_count.derived.(^ast.Basic_Lit).tok.text)
+	column_count, _ := strconv.parse_int(matrix_type.column_count.derived.(^ast.Basic_Lit).tok.text)
+
 	if row_count * column_count > len(comp_lit.elems) {
 		p.errored_out = true
 		return document
diff --git a/tools/odinfmt/tests/.snapshots/matrix_comp_lit.odin b/tools/odinfmt/tests/.snapshots/matrix_comp_lit.odin
new file mode 100644
index 0000000..a90e451
--- /dev/null
+++ b/tools/odinfmt/tests/.snapshots/matrix_comp_lit.odin
@@ -0,0 +1,5 @@
+package odinfmt_test
+
+main :: proc() {
+	foo := #row_major matrix[4, 4]int{4 ..< 8 = 1, 1 ..< 2 = 2}
+}
diff --git a/tools/odinfmt/tests/matrix_comp_lit.odin b/tools/odinfmt/tests/matrix_comp_lit.odin
new file mode 100644
index 0000000..74e71d2
--- /dev/null
+++ b/tools/odinfmt/tests/matrix_comp_lit.odin
@@ -0,0 +1,7 @@
+package odinfmt_test
+
+main :: proc() {
+	foo := #row_major  matrix[4,4]int {
+		4..<8 = 1, 1..<2 = 2,
+	} 
+}
-- 
cgit v1.2.3