From 8e96639d5a2bd3dbdaee886557c200e3cc6923f7 Mon Sep 17 00:00:00 2001
From: Justin Lin <caterpillar@openhome.cc>
Date: Tue, 30 Apr 2019 08:29:56 +0800
Subject: [PATCH] support number

---
 src/m_rotation.scad | 16 +++++++++-------
 1 file changed, 9 insertions(+), 7 deletions(-)

diff --git a/src/m_rotation.scad b/src/m_rotation.scad
index 48882d45..4f120227 100644
--- a/src/m_rotation.scad
+++ b/src/m_rotation.scad
@@ -58,11 +58,13 @@ function _m_zRotation(a) =
         [0, 0, 0, 1]
     ];    
 
+function _xyz_rotation(a) =
+    let(ang = len(a) == 3 ? a : [0, 0, a])
+    __m_multiply(
+        _m_zRotation(ang[2]), __m_multiply(
+            _m_yRotation(ang[1]), _m_xRotation(ang[0])
+        )
+    );
+
 function m_rotation(a, v) = 
-    v == undef ? 
-        __m_multiply(
-            _m_zRotation(a[2]), __m_multiply(
-                _m_yRotation(a[1]), _m_xRotation(a[0])
-            )
-        ) :
-        _q_rotation(a, v);
\ No newline at end of file
+    v == undef ? _xyz_rotation(a) : _q_rotation(a, v);
\ No newline at end of file