From 7ec08a76031e8107ce1848d3336cd114cd825d67 Mon Sep 17 00:00:00 2001
From: Justin Lin <caterpillar@openhome.cc>
Date: Tue, 30 Apr 2019 08:20:41 +0800
Subject: [PATCH] support number

---
 src/m_scaling.scad | 14 ++++++++------
 1 file changed, 8 insertions(+), 6 deletions(-)

diff --git a/src/m_scaling.scad b/src/m_scaling.scad
index 64c6a501..8bfe1f40 100644
--- a/src/m_scaling.scad
+++ b/src/m_scaling.scad
@@ -1,6 +1,8 @@
-function m_scaling(v) = [
-    [v[0], 0, 0, 0],
-    [0, v[1], 0, 0],
-    [0, 0, v[2], 0],
-    [0, 0, 0, 1]
-];
\ No newline at end of file
+function m_scaling(s) = 
+    let(v = len(s) == 3 ? s : [s, s, s])
+    [
+        [v[0], 0, 0, 0],
+        [0, v[1], 0, 0],
+        [0, 0, v[2], 0],
+        [0, 0, 0, 1]
+    ];
\ No newline at end of file