From a93739474b5d209d10158c77d60fbf34454fb41d Mon Sep 17 00:00:00 2001
From: Justin Lin <justin.sdk@gmail.com>
Date: Wed, 16 Mar 2022 07:59:41 +0800
Subject: [PATCH] refactor

---
 src/maze/mz_square_cells.scad | 15 ++++++++++++---
 1 file changed, 12 insertions(+), 3 deletions(-)

diff --git a/src/maze/mz_square_cells.scad b/src/maze/mz_square_cells.scad
index ec984412..12b4741b 100644
--- a/src/maze/mz_square_cells.scad
+++ b/src/maze/mz_square_cells.scad
@@ -12,8 +12,17 @@ use <_impl/_mz_square_cells_impl.scad>;
 use <mz_square_initialize.scad>;
 
 function mz_square_cells(rows, columns, start = [0, 0], init_cells, x_wrapping = false, y_wrapping = false, seed) = 
+    let(
+        init_undef = is_undef(init_cells),
+        mz = init_undef ? mz_square_initialize(rows, columns) : init_cells,
+        rc = init_undef ? [rows, columns] : [len([for(cell = mz) if(cell.y == 0) undef]), len([for(cell = mz) if(cell.x == 0) undef])]
+    )
     go_maze( 
-        start.x, start.y,   
-        is_undef(init_cells) ? mz_square_initialize(rows, columns) : init_cells,  
-        rows, columns, x_wrapping, y_wrapping, seed
+        start.x, 
+        start.y,   
+        mz,  
+        rc[0], rc[1], 
+        x_wrapping, 
+        y_wrapping, 
+        seed
     );