From f681cc20357e92e1511a2aee7122bdd97b70a1b7 Mon Sep 17 00:00:00 2001
From: "Zheng, Lei" <realthunder.dev@gmail.com>
Date: Wed, 18 Oct 2017 17:51:01 +0800
Subject: [PATCH] utils: add getElementRotation()

---
 utils.py | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/utils.py b/utils.py
index fa27664..accd7e0 100644
--- a/utils.py
+++ b/utils.py
@@ -212,7 +212,7 @@ def getElementPos(obj):
     return pos
 
 
-def getElementNormal(obj,reverse=False):
+def getElementRotation(obj,reverse=False):
     axis = None
     face = getElement(obj,Part.Face)
     if face:
@@ -256,7 +256,12 @@ def getElementNormal(obj,reverse=False):
                     if np.std( D, axis=0 ).max() < 10**-9: #then linear curve
                         return D[0]
     if axis:
-        q = FreeCAD.Rotation(FreeCAD.Vector(0,0,-1 if reverse else 1),axis).Q
+        return FreeCAD.Rotation(FreeCAD.Vector(0,0,-1 if reverse else 1),axis)
+
+def getElementNormal(obj,reverse=False):
+    rot = getElementRotation(obj,reverse)
+    if rot:
+        q = rot.Q
         return q[3],q[0],q[1],q[2]
 
 def getElementCircular(obj):