6c3efbdb3b
Support for creation of Rotation from matrices which is a combination of non uniform scale and a rotation Fixes according to review Scale -1 is Uniform, Not NoScaling Fix hasScale() when negative scale
631 lines
25 KiB
Python
631 lines
25 KiB
Python
#***************************************************************************
|
|
#* Copyright (c) 2004 Juergen Riegel <juergen.riegel@web.de> *
|
|
#* *
|
|
#* This file is part of the FreeCAD CAx development system. *
|
|
#* *
|
|
#* This program is free software; you can redistribute it and/or modify *
|
|
#* it under the terms of the GNU Lesser General Public License (LGPL) *
|
|
#* as published by the Free Software Foundation; either version 2 of *
|
|
#* the License, or (at your option) any later version. *
|
|
#* for detail see the LICENCE text file. *
|
|
#* *
|
|
#* FreeCAD is distributed in the hope that it will be useful, *
|
|
#* but WITHOUT ANY WARRANTY; without even the implied warranty of *
|
|
#* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the *
|
|
#* GNU Library General Public License for more details. *
|
|
#* *
|
|
#* You should have received a copy of the GNU Library General Public *
|
|
#* License along with FreeCAD; if not, write to the Free Software *
|
|
#* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 *
|
|
#* USA *
|
|
#* *
|
|
#***************************************************************************/
|
|
|
|
import FreeCAD, os, unittest, tempfile, math
|
|
from FreeCAD import Base
|
|
|
|
class ConsoleTestCase(unittest.TestCase):
|
|
def setUp(self):
|
|
self.count = 0
|
|
|
|
def testPrint(self):
|
|
FreeCAD.Console.PrintMessage(" Printing message\n")
|
|
FreeCAD.Console.PrintError(" Printing error\n")
|
|
FreeCAD.Console.PrintWarning(" Printing warning\n")
|
|
FreeCAD.Console.PrintLog(" Printing Log\n")
|
|
|
|
def testSynchronPrintFromThread(self):
|
|
# http://python-kurs.eu/threads.php
|
|
try:
|
|
import _thread as thread, time
|
|
except Exception:
|
|
import thread, time
|
|
def adder():
|
|
lock.acquire()
|
|
self.count=self.count+1
|
|
# call of Console method is thread-safe
|
|
FreeCAD.Console.PrintMessage("Call from Python thread: count="+str(self.count)+"\n")
|
|
lock.release()
|
|
|
|
lock=thread.allocate_lock()
|
|
for i in range(10):
|
|
thread.start_new(adder,())
|
|
|
|
time.sleep(3)
|
|
self.assertEqual(self.count, 10, "Synchronization of threads failed")
|
|
FreeCAD.Console.PrintMessage(str(self.count)+"\n")
|
|
|
|
def testAsynchronPrintFromThread(self):
|
|
# http://python-kurs.eu/threads.php
|
|
try:
|
|
import _thread as thread, time
|
|
except Exception:
|
|
import thread, time
|
|
def adder():
|
|
self.count=self.count+1
|
|
# call of Console method is thread-safe
|
|
FreeCAD.Console.PrintMessage("Call from Python thread (not synchronized): count="+str(self.count)+"\n")
|
|
|
|
lock=thread.allocate_lock()
|
|
for i in range(10):
|
|
thread.start_new(adder,())
|
|
|
|
time.sleep(3)
|
|
FreeCAD.Console.PrintMessage(str(self.count)+"\n")
|
|
|
|
# def testStatus(self):
|
|
# SLog = FreeCAD.GetStatus("Console","Log")
|
|
# SErr = FreeCAD.GetStatus("Console","Err")
|
|
# SWrn = FreeCAD.GetStatus("Console","Wrn")
|
|
# SMsg = FreeCAD.GetStatus("Console","Msg")
|
|
# FreeCAD.SetStatus("Console","Log",1)
|
|
# FreeCAD.SetStatus("Console","Err",1)
|
|
# FreeCAD.SetStatus("Console","Wrn",1)
|
|
# FreeCAD.SetStatus("Console","Msg",1)
|
|
# self.assertEqual(FreeCAD.GetStatus("Console","Msg"),1,"Set and read status failed (Console,Msg)")
|
|
# self.assertEqual(FreeCAD.GetStatus("Console","Err"),1,"Set and read status failed (Console,Err)")
|
|
# self.assertEqual(FreeCAD.GetStatus("Console","Wrn"),1,"Set and read status failed (Console,Wrn)")
|
|
# self.assertEqual(FreeCAD.GetStatus("Console","Log"),1,"Set and read status failed (Console,Log)")
|
|
# FreeCAD.SetStatus("Console","Log",0)
|
|
# FreeCAD.SetStatus("Console","Err",0)
|
|
# FreeCAD.SetStatus("Console","Wrn",0)
|
|
# FreeCAD.SetStatus("Console","Msg",0)
|
|
# self.assertEqual(FreeCAD.GetStatus("Console","Msg"),0,"Set and read status failed (Console,Msg)")
|
|
# self.assertEqual(FreeCAD.GetStatus("Console","Err"),0,"Set and read status failed (Console,Err)")
|
|
# self.assertEqual(FreeCAD.GetStatus("Console","Wrn"),0,"Set and read status failed (Console,Wrn)")
|
|
# self.assertEqual(FreeCAD.GetStatus("Console","Log"),0,"Set and read status failed (Console,Log)")
|
|
# FreeCAD.SetStatus("Console","Log",SLog)
|
|
# FreeCAD.SetStatus("Console","Err",SErr)
|
|
# FreeCAD.SetStatus("Console","Wrn",SWrn)
|
|
# FreeCAD.SetStatus("Console","Msg",SMsg)
|
|
|
|
def tearDown(self):
|
|
pass
|
|
|
|
def testILoggerBlocker(self):
|
|
if FreeCAD.GuiUp:
|
|
import QtUnitGui
|
|
QtUnitGui.testILoggerBlocker()
|
|
|
|
class ParameterTestCase(unittest.TestCase):
|
|
def setUp(self):
|
|
self.TestPar = FreeCAD.ParamGet("System parameter:Test")
|
|
|
|
def testGroup(self):
|
|
#FreeCAD.Console.PrintLog("Base::ParameterTestCase::testGroup\n")
|
|
# check on Group creation
|
|
Temp = self.TestPar.GetGroup("44")
|
|
self.assertTrue(self.TestPar.HasGroup("44"),"Test on created group failed")
|
|
# check on Deletion
|
|
self.TestPar.RemGroup("44")
|
|
self.assertTrue(self.TestPar.HasGroup("44"),"A referenced group must not be deleted")
|
|
Temp = 0
|
|
|
|
def testGroupNames(self):
|
|
with self.assertRaises(ValueError):
|
|
# no empty groups allowed
|
|
self.TestPar.GetGroup("")
|
|
grp1 = self.TestPar.GetGroup("////Sub1/////Sub2/////")
|
|
grp2 = self.TestPar.GetGroup("Sub1/Sub2")
|
|
self.assertEqual(grp1.GetGroupName(), "Sub2")
|
|
self.assertEqual(grp2.GetGroupName(), "Sub2")
|
|
|
|
# check on special conditions
|
|
def testInt(self):
|
|
#FreeCAD.Console.PrintLog("Base::ParameterTestCase::testInt\n")
|
|
#Temp = FreeCAD.ParamGet("System parameter:Test/44")
|
|
# check on Int
|
|
self.TestPar.SetInt("44",4711)
|
|
self.assertEqual(self.TestPar.GetInt("44"), 4711,"In and out error at Int")
|
|
# check on Deletion
|
|
self.TestPar.RemInt("44")
|
|
self.assertEqual(self.TestPar.GetInt("44",1), 1,"Deletion error at Int")
|
|
|
|
|
|
def testBool(self):
|
|
#FreeCAD.Console.PrintLog("Base::ParameterTestCase::testBool\n")
|
|
# check on Int
|
|
self.TestPar.SetBool("44",1)
|
|
self.assertEqual(self.TestPar.GetBool("44"), 1,"In and out error at Bool")
|
|
# check on Deletion
|
|
self.TestPar.RemBool("44")
|
|
self.assertEqual(self.TestPar.GetBool("44",0), 0,"Deletion error at Bool")
|
|
|
|
def testFloat(self):
|
|
#FreeCAD.Console.PrintLog("Base::ParameterTestCase::testFloat\n")
|
|
#Temp = FreeCAD.ParamGet("System parameter:Test/44")
|
|
# check on Int
|
|
self.TestPar.SetFloat("44",4711.4711)
|
|
self.assertEqual(self.TestPar.GetFloat("44"), 4711.4711,"In and out error at Float")
|
|
# check on Deletion
|
|
self.TestPar.RemFloat("44")
|
|
self.assertEqual(self.TestPar.GetFloat("44",1.1), 1.1,"Deletion error at Float")
|
|
|
|
def testString(self):
|
|
#FreeCAD.Console.PrintLog("Base::ParameterTestCase::testFloat\n")
|
|
#Temp = FreeCAD.ParamGet("System parameter:Test/44")
|
|
# check on Int
|
|
self.TestPar.SetString("44","abcdefgh")
|
|
self.assertEqual(self.TestPar.GetString("44"), "abcdefgh","In and out error at String")
|
|
# check on Deletion
|
|
self.TestPar.RemString("44")
|
|
self.assertEqual(self.TestPar.GetString("44","hallo"), "hallo","Deletion error at String")
|
|
|
|
def testNesting(self):
|
|
# Parameter testing
|
|
#FreeCAD.Console.PrintLog("Base::ParameterTestCase::testNesting\n")
|
|
for i in range(50):
|
|
self.TestPar.SetFloat(str(i),4711.4711)
|
|
self.TestPar.SetInt(str(i),4711)
|
|
self.TestPar.SetBool(str(i),1)
|
|
Temp = self.TestPar.GetGroup(str(i))
|
|
for l in range(50):
|
|
Temp.SetFloat(str(l),4711.4711)
|
|
Temp.SetInt(str(l),4711)
|
|
Temp.SetBool(str(l),1)
|
|
Temp = 0
|
|
|
|
def testExportImport(self):
|
|
# Parameter testing
|
|
#FreeCAD.Console.PrintLog("Base::ParameterTestCase::testNesting\n")
|
|
self.TestPar.SetFloat("ExTest",4711.4711)
|
|
self.TestPar.SetInt("ExTest",4711)
|
|
self.TestPar.SetString("ExTest","4711")
|
|
self.TestPar.SetBool("ExTest",1)
|
|
Temp = self.TestPar.GetGroup("ExTest")
|
|
Temp.SetFloat("ExTest",4711.4711)
|
|
Temp.SetInt("ExTest",4711)
|
|
Temp.SetString("ExTest","4711")
|
|
Temp.SetBool("ExTest",1)
|
|
TempPath = tempfile.gettempdir() + os.sep + "ExportTest.FCExport"
|
|
|
|
self.TestPar.Export(TempPath)
|
|
Temp = self.TestPar.GetGroup("ImportTest")
|
|
Temp.Import(TempPath)
|
|
self.assertEqual(Temp.GetFloat("ExTest"), 4711.4711,"ExportImport error")
|
|
Temp = 0
|
|
|
|
def tearDown(self):
|
|
#remove all
|
|
TestPar = FreeCAD.ParamGet("System parameter:Test")
|
|
TestPar.Clear()
|
|
|
|
class AlgebraTestCase(unittest.TestCase):
|
|
def setUp(self):
|
|
pass
|
|
|
|
def testAngle(self):
|
|
v1 = FreeCAD.Vector(0,0,0.000001)
|
|
v2 = FreeCAD.Vector(0,0.000001,0)
|
|
self.assertAlmostEqual(v1.getAngle(v2), math.pi/2)
|
|
self.assertAlmostEqual(v2.getAngle(v1), math.pi/2)
|
|
|
|
def testVector2d(self):
|
|
v = FreeCAD.Base.Vector2d(1.0, 1.0)
|
|
v.rotate(math.pi/2)
|
|
self.assertAlmostEqual(v.x, -1.0)
|
|
self.assertAlmostEqual(v.y, 1.0)
|
|
|
|
def testAngleWithNullVector(self):
|
|
v1 = FreeCAD.Vector(0,0,0)
|
|
v2 = FreeCAD.Vector(0,1,0)
|
|
self.assertTrue(math.isnan(v1.getAngle(v2)))
|
|
self.assertTrue(math.isnan(v2.getAngle(v1)))
|
|
|
|
def testMatrix(self):
|
|
m=FreeCAD.Matrix(4,2,1,0,1,1,1,0,0,0,1,0,0,0,0,1)
|
|
u=m.multiply(m.inverse())
|
|
self.assertEqual(u, FreeCAD.Matrix(),"Invalid inverse of matrix")
|
|
|
|
def testRotAndMoveMatrix(self):
|
|
m1=FreeCAD.Matrix()
|
|
m1.move(10,5,-3)
|
|
m1.rotateY(.2)
|
|
m2=FreeCAD.Matrix()
|
|
m2.rotateY(.2)
|
|
m2.move(10,5,-3)
|
|
m3=FreeCAD.Matrix()
|
|
m3.move(10,5,-3)
|
|
m4=FreeCAD.Matrix()
|
|
m4.rotateY(.2)
|
|
self.assertNotEqual(m1, m3*m4, "Wrong multiplication order")
|
|
self.assertEqual(m1, m4*m3 , "Wrong multiplication order")
|
|
self.assertEqual(m2, m3*m4 , "Wrong multiplication order")
|
|
self.assertNotEqual(m2, m4*m3, "Wrong multiplication order")
|
|
|
|
|
|
def testRotationFromMatrix(self):
|
|
m1 = FreeCAD.Matrix()
|
|
m1.rotateZ(.2)
|
|
m1.scale(0.5)
|
|
m1.rotateY(.2)
|
|
m1.scale(3)
|
|
m1.move(10,5,-3)
|
|
r1 = FreeCAD.Rotation(m1)
|
|
m2 = FreeCAD.Matrix()
|
|
m2.rotateZ(.2)
|
|
m2.rotateY(.2)
|
|
m2.move(10,5,-3)
|
|
r2 = FreeCAD.Rotation(m2)
|
|
self.assertTrue(r1.isSame(r2, 1e-7), 'Scale on matrix influenced rotation')
|
|
m3 = FreeCAD.Matrix()
|
|
m3.scale(-1)
|
|
r3 = FreeCAD.Rotation(m3)
|
|
r0 = FreeCAD.Rotation()
|
|
self.assertTrue(r3.isSame(r0, 1e-7), 'Scale on matrix influenced rotation')
|
|
m4 = FreeCAD.Matrix()
|
|
m4.scale(1.25,1.0,0.25)
|
|
m4.move(4,5,6)
|
|
r24 = FreeCAD.Rotation(m2*m4)
|
|
r42 = FreeCAD.Rotation(m4*m2)
|
|
self.assertTrue(r2.isSame(r24, 1e-7), 'Scale on matrix influenced rotation')
|
|
self.assertTrue(r2.isSame(r42, 1e-7), 'Scale on matrix influenced rotation')
|
|
|
|
|
|
def testRotation(self):
|
|
r=FreeCAD.Rotation(1,0,0,0) # 180 deg around (1,0,0)
|
|
self.assertEqual(r.Axis, FreeCAD.Vector(1,0,0))
|
|
self.assertAlmostEqual(math.fabs(r.Angle), math.fabs(math.pi))
|
|
|
|
r=r.multiply(r) # identity
|
|
self.assertEqual(r.Axis, FreeCAD.Vector(0,0,1))
|
|
self.assertAlmostEqual(r.Angle, 0)
|
|
|
|
r=FreeCAD.Rotation(1,0,0,0)
|
|
r.Q=(0,0,0,1) # update axis and angle
|
|
s=FreeCAD.Rotation(0,0,0,1)
|
|
self.assertEqual(r.Axis, s.Axis)
|
|
self.assertAlmostEqual(r.Angle, s.Angle)
|
|
self.assertTrue(r.isSame(s))
|
|
|
|
r=FreeCAD.Rotation(1,0,0,0)
|
|
r.Matrix=FreeCAD.Matrix() # update axis and angle
|
|
s=FreeCAD.Rotation(0,0,0,1)
|
|
self.assertEqual(r.Axis, s.Axis)
|
|
self.assertAlmostEqual(r.Angle, s.Angle)
|
|
self.assertTrue(r.isSame(s))
|
|
|
|
r=FreeCAD.Rotation(1,0,0,0)
|
|
r.Axes=(FreeCAD.Vector(0,0,1),FreeCAD.Vector(0,0,1)) # update axis and angle
|
|
s=FreeCAD.Rotation(0,0,0,1)
|
|
self.assertEqual(r.Axis, s.Axis)
|
|
self.assertAlmostEqual(r.Angle, s.Angle)
|
|
self.assertTrue(r.isSame(s))
|
|
|
|
#add 360 deg to angle
|
|
r=FreeCAD.Rotation(FreeCAD.Vector(1,0,0),270)
|
|
s=FreeCAD.Rotation(FreeCAD.Vector(1,0,0),270+360)
|
|
self.assertEqual(r.Axis, s.Axis)
|
|
#self.assertAlmostEqual(r.Angle, s.Angle + 2*math.pi)
|
|
self.assertTrue(r.isSame(s))
|
|
|
|
#subtract 360 deg from angle using Euler angles
|
|
r=FreeCAD.Rotation(0,0,180)
|
|
r.invert()
|
|
s=FreeCAD.Rotation(0,0,-180)
|
|
self.assertTrue(r.isSame(s))
|
|
|
|
#subtract 360 deg from angle using quaternion
|
|
r=FreeCAD.Rotation(1,0,0,0)
|
|
s=FreeCAD.Rotation(-1,0,0,0)
|
|
#angles have the same sign
|
|
if r.Angle * s.Angle > 0:
|
|
self.assertEqual(r.Axis, s.Axis*(-1))
|
|
else:
|
|
self.assertAlmostEqual(r.Angle, -s.Angle)
|
|
self.assertTrue(r.isSame(s))
|
|
r.invert()
|
|
self.assertTrue(r.isSame(s))
|
|
|
|
# gimbal lock (north pole)
|
|
r=FreeCAD.Rotation()
|
|
r.setYawPitchRoll(20, 90, 10)
|
|
a=r.getYawPitchRoll()
|
|
s=FreeCAD.Rotation()
|
|
s.setYawPitchRoll(*a)
|
|
self.assertAlmostEqual(a[0], 0.0)
|
|
self.assertAlmostEqual(a[1], 90.0)
|
|
self.assertAlmostEqual(a[2], -10.0)
|
|
self.assertTrue(r.isSame(s, 1e-12))
|
|
|
|
# gimbal lock (south pole)
|
|
r=FreeCAD.Rotation()
|
|
r.setYawPitchRoll(20, -90, 10)
|
|
a=r.getYawPitchRoll()
|
|
s=FreeCAD.Rotation()
|
|
s.setYawPitchRoll(*a)
|
|
self.assertAlmostEqual(a[0], 0.0)
|
|
self.assertAlmostEqual(a[1], -90.0)
|
|
self.assertAlmostEqual(a[2], 30.0)
|
|
self.assertTrue(r.isSame(s, 1e-12))
|
|
|
|
def testInverted(self):
|
|
p = FreeCAD.Placement()
|
|
p.Rotation.Angle = math.pi / 2
|
|
self.assertEqual(abs(p.inverse().Rotation.Angle), p.Rotation.Angle)
|
|
|
|
def testYawPitchRoll(self):
|
|
def getYPR1(yaw, pitch, roll):
|
|
r = FreeCAD.Rotation()
|
|
r.setYawPitchRoll(yaw, pitch, roll)
|
|
return r
|
|
def getYPR2(yaw, pitch, roll):
|
|
rx = FreeCAD.Rotation()
|
|
ry = FreeCAD.Rotation()
|
|
rz = FreeCAD.Rotation()
|
|
|
|
rx.Axis = FreeCAD.Vector(1,0,0)
|
|
ry.Axis = FreeCAD.Vector(0,1,0)
|
|
rz.Axis = FreeCAD.Vector(0,0,1)
|
|
|
|
rx.Angle = math.radians(roll)
|
|
ry.Angle = math.radians(pitch)
|
|
rz.Angle = math.radians(yaw)
|
|
|
|
return rz.multiply(ry).multiply(rx)
|
|
|
|
angles = []
|
|
angles.append((10,10,10))
|
|
angles.append((13,45,-24))
|
|
angles.append((10,-90,20))
|
|
|
|
for i in angles:
|
|
r = getYPR1(*i)
|
|
s = getYPR2(*i)
|
|
self.assertTrue(r.isSame(s, 1e-12))
|
|
|
|
def testBounding(self):
|
|
b=FreeCAD.BoundBox()
|
|
b.setVoid()
|
|
self.assertFalse(b.isValid(),"Bbox is not invalid")
|
|
b.add(0,0,0)
|
|
self.assertTrue(b.isValid(), "Bbox is invalid")
|
|
self.assertEqual(b.XLength, 0, "X length > 0")
|
|
self.assertEqual(b.YLength, 0, "Y length > 0")
|
|
self.assertEqual(b.ZLength, 0, "Z length > 0")
|
|
self.assertEqual(b.Center, FreeCAD.Vector(0,0,0), "Center is not at (0,0,0)")
|
|
self.assertTrue(b.isInside(b.Center), "Center is not inside Bbox")
|
|
b.add(2,2,2)
|
|
self.assertTrue(b.isInside(b.getIntersectionPoint(b.Center,FreeCAD.Vector(0,1,0))),"Intersection point is not inside Bbox")
|
|
self.assertTrue(b.intersect(b),"Bbox doesn't intersect with itself")
|
|
self.assertFalse(b.intersected(FreeCAD.BoundBox(4,4,4,6,6,6)).isValid(),"Bbox should not intersect with Bbox outside")
|
|
self.assertEqual(b.intersected(FreeCAD.BoundBox(-2,-2,-2,2,2,2)).Center, b.Center,"Bbox is not a full subset")
|
|
|
|
def testMultLeftOrRight(self):
|
|
doc = FreeCAD.newDocument()
|
|
obj = doc.addObject("App::FeatureTestPlacement")
|
|
|
|
p1 = Base.Placement()
|
|
p1.Base = Base.Vector(10, 10, 10)
|
|
|
|
p2 = Base.Placement()
|
|
p2.Rotation.Angle = math.radians(90)
|
|
|
|
obj.Input1 = p1
|
|
obj.Input2 = p2
|
|
doc.recompute()
|
|
|
|
self.assertTrue(obj.MultRight.isSame(p1 * p2))
|
|
self.assertFalse(obj.MultRight.isSame(p2 * p1))
|
|
self.assertTrue(obj.MultLeft.isSame(p2 * p1))
|
|
self.assertFalse(obj.MultLeft.isSame(p1 * p2))
|
|
|
|
FreeCAD.closeDocument(doc.Name)
|
|
|
|
class MatrixTestCase(unittest.TestCase):
|
|
def setUp(self):
|
|
self.mat = FreeCAD.Matrix()
|
|
|
|
def testOrder(self):
|
|
self.mat = FreeCAD.Matrix(1.0,2.0,3.0,4.0)
|
|
self.assertEqual(self.mat.A11, 1.0)
|
|
self.assertEqual(self.mat.A12, 2.0)
|
|
self.assertEqual(self.mat.A13, 3.0)
|
|
self.assertEqual(self.mat.A14, 4.0)
|
|
|
|
def testScalar(self):
|
|
res = self.mat * 0.0
|
|
for i in range(16):
|
|
self.assertEqual(res.A[i], 0.0)
|
|
|
|
def testAddition(self):
|
|
res1 = self.mat * 2.0
|
|
res2 = self.mat + self.mat
|
|
for i in range(16):
|
|
self.assertEqual(res1.A[i], res2.A[i])
|
|
|
|
def testMinus(self):
|
|
res = self.mat - self.mat
|
|
for i in range(16):
|
|
self.assertEqual(res.A[i], 0.0)
|
|
|
|
def testVector(self):
|
|
vec = FreeCAD.Vector(1, 1, 1)
|
|
vec = self.mat * vec
|
|
self.assertEqual(vec.x, 1.0)
|
|
self.assertEqual(vec.y, 1.0)
|
|
self.assertEqual(vec.z, 1.0)
|
|
|
|
def testVectorMult(self):
|
|
vec = FreeCAD.Vector(1, 1, 1)
|
|
with self.assertRaises(TypeError):
|
|
vec * "string"
|
|
|
|
def testRotation(self):
|
|
rot = FreeCAD.Rotation()
|
|
res = self.mat * rot
|
|
self.assertEqual(type(res), FreeCAD.Matrix)
|
|
|
|
def testPlacement(self):
|
|
plm = FreeCAD.Placement()
|
|
res = self.mat * plm
|
|
self.assertEqual(type(res), FreeCAD.Matrix)
|
|
|
|
def testMatrix(self):
|
|
mat = FreeCAD.Matrix()
|
|
res = self.mat * mat
|
|
self.assertEqual(type(res), FreeCAD.Matrix)
|
|
|
|
def testMatrixPlacementMatrix(self):
|
|
# Example taken from https://forum.freecadweb.org/viewtopic.php?f=3&t=61000
|
|
mat = FreeCAD.Matrix(-0.470847778020266,
|
|
0.8150598976807029,
|
|
0.3376088628746235,
|
|
-11.25290913640202,
|
|
-0.8822144756796808,
|
|
-0.4350066260577338,
|
|
-0.180185641360483,
|
|
-2876.45492562325,
|
|
1.955470978815492e-9,
|
|
-0.3826834326750831,
|
|
0.923879538425552,
|
|
941.3822018176414)
|
|
plm = FreeCAD.Placement(mat)
|
|
mat = plm.toMatrix()
|
|
self.assertEqual(mat.hasScale(), FreeCAD.ScaleType.NoScaling)
|
|
|
|
def testAnything(self):
|
|
with self.assertRaises(NotImplementedError):
|
|
self.mat * "string"
|
|
|
|
def testUnity(self):
|
|
mat = FreeCAD.Matrix(2,0,0,0, 0,1,0,0, 0,0,2,0, 0,0,0,-1)
|
|
self.assertFalse(mat.isUnity())
|
|
mat.unity()
|
|
self.assertTrue(mat.isUnity())
|
|
|
|
def testPower(self):
|
|
mat = FreeCAD.Matrix(2,0,0,0, 0,1,0,0, 0,0,2,0, 0,0,0,-1)
|
|
with self.assertRaises(NotImplementedError):
|
|
mat ** "string"
|
|
|
|
mat2 = mat ** 0
|
|
self.assertTrue(mat2.isUnity())
|
|
self.assertEqual(mat ** -1, mat.inverse())
|
|
self.assertEqual(mat ** 1, mat)
|
|
self.assertEqual(mat ** 2, mat * mat)
|
|
self.assertEqual(mat ** 3, mat * mat * mat)
|
|
mat.nullify()
|
|
with self.assertRaises(RuntimeError):
|
|
mat ** -1
|
|
|
|
def testScale(self):
|
|
self.mat.scale(1., 2., 3.)
|
|
self.assertEqual(self.mat.determinant(), 6.0)
|
|
self.assertEqual(self.mat.hasScale(), FreeCAD.ScaleType.NonUniformLeft)
|
|
self.mat.unity()
|
|
self.assertEqual(self.mat.hasScale(), FreeCAD.ScaleType.NoScaling)
|
|
self.mat.scale(2., 2., 2.)
|
|
self.assertEqual(self.mat.hasScale(), FreeCAD.ScaleType.Uniform)
|
|
self.mat.rotateX(1.0)
|
|
self.assertEqual(self.mat.hasScale(), FreeCAD.ScaleType.Uniform)
|
|
self.mat.scale(1., 2., 3.)
|
|
self.assertEqual(self.mat.hasScale(), FreeCAD.ScaleType.NonUniformLeft)
|
|
self.mat.unity()
|
|
self.mat.scale(1., 2., 3.)
|
|
self.mat.rotateX(1.0)
|
|
self.assertEqual(self.mat.hasScale(), FreeCAD.ScaleType.NonUniformRight)
|
|
self.mat.unity()
|
|
self.mat.setCol(0, FreeCAD.Vector(1,2,3))
|
|
self.mat.setCol(1, FreeCAD.Vector(1,2,3))
|
|
self.mat.setCol(2, FreeCAD.Vector(1,2,3))
|
|
self.assertEqual(self.mat.hasScale(), FreeCAD.ScaleType.Other)
|
|
self.mat.unity()
|
|
self.mat.setRow(0, FreeCAD.Vector(1,2,3))
|
|
self.mat.setRow(1, FreeCAD.Vector(1,2,3))
|
|
self.mat.setRow(2, FreeCAD.Vector(1,2,3))
|
|
self.assertEqual(self.mat.hasScale(), FreeCAD.ScaleType.Other)
|
|
self.mat.unity()
|
|
self.mat.scale(-1.)
|
|
self.assertEqual(self.mat.hasScale(), FreeCAD.ScaleType.Uniform)
|
|
self.mat.scale(-2.)
|
|
self.assertEqual(self.mat.hasScale(), FreeCAD.ScaleType.Uniform)
|
|
|
|
def testShearing(self):
|
|
self.mat.setRow(1, FreeCAD.Vector(0,1,1))
|
|
self.assertEqual(self.mat.hasScale(), FreeCAD.ScaleType.Other)
|
|
|
|
def testMultLeftOrRight(self):
|
|
mat1 = FreeCAD.Matrix()
|
|
mat1.rotateX(1.0)
|
|
|
|
mat2 = FreeCAD.Matrix()
|
|
mat2.scale(1., 2., 3.)
|
|
self.assertEqual((mat1 * mat2).hasScale(), FreeCAD.ScaleType.NonUniformRight)
|
|
self.assertEqual((mat2 * mat1).hasScale(), FreeCAD.ScaleType.NonUniformLeft)
|
|
|
|
def testNull(self):
|
|
self.assertFalse(self.mat.isNull())
|
|
self.mat.nullify()
|
|
self.assertTrue(self.mat.isNull())
|
|
|
|
def testUnity(self):
|
|
self.assertTrue(self.mat.isUnity())
|
|
self.mat.nullify()
|
|
self.assertFalse(self.mat.isUnity())
|
|
self.mat.unity()
|
|
self.assertTrue(self.mat.isUnity())
|
|
|
|
def testColRow(self):
|
|
with self.assertRaises(TypeError):
|
|
self.mat.col("string")
|
|
with self.assertRaises(TypeError):
|
|
self.mat.row("string")
|
|
self.assertEqual(type(self.mat.col(0)), FreeCAD.Vector)
|
|
self.assertEqual(type(self.mat.row(0)), FreeCAD.Vector)
|
|
self.mat.setCol(0, FreeCAD.Vector(1,0,0))
|
|
self.mat.setRow(0, FreeCAD.Vector(1,0,0))
|
|
|
|
def testTrace(self):
|
|
self.mat.scale(2., 2., 2.)
|
|
self.assertEqual(self.mat.trace(), FreeCAD.Vector(2., 2., 2.))
|
|
|
|
def testNumberProtocol(self):
|
|
with self.assertRaises(NotImplementedError):
|
|
self.mat / 2.0
|
|
with self.assertRaises(NotImplementedError):
|
|
self.mat % 2.0
|
|
with self.assertRaises(NotImplementedError):
|
|
divmod(self.mat, 2.0)
|
|
with self.assertRaises(NotImplementedError):
|
|
float(self.mat)
|
|
with self.assertRaises(NotImplementedError):
|
|
int(self.mat)
|
|
with self.assertRaises(NotImplementedError):
|
|
self.mat | self.mat
|
|
with self.assertRaises(NotImplementedError):
|
|
self.mat & self.mat
|
|
with self.assertRaises(NotImplementedError):
|
|
self.mat ^ self.mat
|
|
with self.assertRaises(NotImplementedError):
|
|
self.mat << 2
|
|
with self.assertRaises(NotImplementedError):
|
|
self.mat >> 2
|
|
with self.assertRaises(NotImplementedError):
|
|
~self.mat
|
|
with self.assertRaises(NotImplementedError):
|
|
abs(self.mat)
|
|
self.assertEqual(+self.mat, self.mat)
|
|
self.assertEqual(-self.mat, self.mat * -1)
|
|
self.assertTrue(bool(self.mat))
|