def test_chain(self):
s = Matrix44.scale(10, 20, 30)
t = Matrix44.translate(10, 20, 30)
c = Matrix44.chain(s, t)
x = diag((10., 20., 30., 1.))
x[3, 0] = 10.
x[3, 1] = 20.
x[3, 2] = 30.
self.assertTrue(equal_matrix(c, x))
def test_chain2(self):
s = Matrix44.scale(10, 20, 30)
t = Matrix44.translate(10, 20, 30)
r = Matrix44.axis_rotate(angle=pi / 2, axis=(0., 0., 1.))
points = ((23., 97., .5), (2., 7., 13.))
p1 = s.transform_vectors(points)
p1 = t.transform_vectors(p1)
p1 = r.transform_vectors(p1)
c = Matrix44.chain(s, t, r)
p2 = c.transform_vectors(points)
self.assertTrue(equal_vectors(p1, p2))
def test_invalid_init(self):
with self.assertRaises(ValueError):
Matrix44((0, 1, 2, 3), (4, 5, 6, 7), (8, 9, 10, 11),
(12, 13, 14, 15, 16))
with self.assertRaises(ValueError):
Matrix44(
(0, 1, 2, 3),
(4, 5, 6, 7),
(8, 9, 10, 11),
)
with self.assertRaises(ValueError):
Matrix44(
[0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16])
with self.assertRaises(ValueError):
Matrix44([0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14])
def test_init_4(self):
matrix = Matrix44((0, 1, 2, 3), (4, 5, 6, 7), (8, 9, 10, 11),
(12, 13, 14, 15))
self.assertEqual(matrix.get_row(0), (0.0, 1.0, 2.0, 3.0))
self.assertEqual(matrix.get_row(1), (4.0, 5.0, 6.0, 7.0))
self.assertEqual(matrix.get_row(2), (8.0, 9.0, 10.0, 11.0))
self.assertEqual(matrix.get_row(3), (12.0, 13.0, 14.0, 15.0))
def test_init_1(self):
matrix = Matrix44(
[0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15])
assert matrix.get_row(0) == (0.0, 1.0, 2.0, 3.0)
assert matrix.get_row(1) == (4.0, 5.0, 6.0, 7.0)
assert matrix.get_row(2) == (8.0, 9.0, 10.0, 11.0)
assert matrix.get_row(3) == (12.0, 13.0, 14.0, 15.0)
def test_translate(self):
t = Matrix44.translate(10, 20, 30)
x = diag((1., 1., 1., 1.))
x[3, 0] = 10.
x[3, 1] = 20.
x[3, 2] = 30.
self.assertTrue(equal_matrix(t, x))
def test_set(self):
matrix = Matrix44()
matrix.set((2., 3., 4., 5.), (6., 7., 8., 9.), (10., 11., 12., 13.),
(14., 15., 16., 17.))
self.assertEqual(matrix.get_row(0), (2.0, 3.0, 4.0, 5.0))
self.assertEqual(matrix.get_row(1), (6.0, 7.0, 8.0, 9.0))
self.assertEqual(matrix.get_row(2), (10.0, 11.0, 12.0, 13.0))
self.assertEqual(matrix.get_row(3), (14.0, 15.0, 16.0, 17.0))
def test_set(self):
matrix = Matrix44()
matrix.set((2., 3., 4., 5.), (6., 7., 8., 9.), (10., 11., 12., 13.),
(14., 15., 16., 17.))
assert matrix.get_row(0) == (2.0, 3.0, 4.0, 5.0)
assert matrix.get_row(1) == (6.0, 7.0, 8.0, 9.0)
assert matrix.get_row(2) == (10.0, 11.0, 12.0, 13.0)
assert matrix.get_row(3) == (14.0, 15.0, 16.0, 17.0)
def test_transpose(self):
matrix = Matrix44((0, 1, 2, 3), (4, 5, 6, 7), (8, 9, 10, 11),
(12, 13, 14, 15))
matrix.transpose()
self.assertEqual(matrix.get_row(0), (0.0, 4.0, 8.0, 12.0))
self.assertEqual(matrix.get_row(1), (1.0, 5.0, 9.0, 13.0))
self.assertEqual(matrix.get_row(2), (2.0, 6.0, 10.0, 14.0))
self.assertEqual(matrix.get_row(3), (3.0, 7.0, 11.0, 15.0))
def test_matrix44_to_ocs():
ocs = OCS(EXTRUSION)
matrix = Matrix44.ucs(ocs.ux, ocs.uy, ocs.uz)
assert is_close_points(
matrix.transform((-9.56460754, 8.44764172, 9.97894327)),
(9.41378764657076, 13.15481838975576, 0.8689258932616031),
places=6,
)
def test_transpose(self):
matrix = Matrix44((0, 1, 2, 3), (4, 5, 6, 7), (8, 9, 10, 11),
(12, 13, 14, 15))
matrix.transpose()
assert matrix.get_row(0) == (0.0, 4.0, 8.0, 12.0)
assert matrix.get_row(1) == (1.0, 5.0, 9.0, 13.0)
assert matrix.get_row(2) == (2.0, 6.0, 10.0, 14.0)
assert matrix.get_row(3) == (3.0, 7.0, 11.0, 15.0)
def test_z_rotate(self):
alpha = radians(25)
t = Matrix44.z_rotate(alpha)
x = diag((1., 1., 1., 1.))
x[0, 0] = cos(alpha)
x[1, 0] = -sin(alpha)
x[0, 1] = sin(alpha)
x[1, 1] = cos(alpha)
self.assertTrue(equal_matrix(t, x))
def test_y_rotate(self):
alpha = radians(25)
t = Matrix44.y_rotate(alpha)
x = diag((1., 1., 1., 1.))
x[0, 0] = cos(alpha)
x[2, 0] = sin(alpha)
x[0, 2] = -sin(alpha)
x[2, 2] = cos(alpha)
assert equal_matrix(t, x) is True
def test_set_col(self):
matrix = Matrix44()
matrix.set_col(0, (2., 3., 4., 5.))
assert matrix.get_col(0) == (2.0, 3.0, 4.0, 5.0)
matrix.set_col(1, (6., 7., 8., 9.))
assert matrix.get_col(1) == (6.0, 7.0, 8.0, 9.0)
matrix.set_col(2, (10., 11., 12., 13.))
assert matrix.get_col(2) == (10.0, 11.0, 12.0, 13.0)
matrix.set_col(3, (14., 15., 16., 17.))
assert matrix.get_col(3) == (14.0, 15.0, 16.0, 17.0)
def fetch_matrix(tags: 'Tags', code: int) -> Tuple[Tags, Optional[Matrix44]]:
values = []
remaining = Tags()
for tag in tags:
if tag.code == code:
values.append(tag.value)
if len(values) == 16:
# enough values collected, code 43 is maybe used for two matrices
code = -1
else:
remaining.append(tag)
if len(values) == 16:
# only if valid matrix
return remaining, Matrix44(values)
else:
return tags, None
def test_iter(self):
values = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15]
matrix = Matrix44(values)
for v1, m1 in zip(values, matrix):
assert isclose(v1, m1)
def test_scale(self):
t = Matrix44.scale(10, 20, 30)
x = diag((10., 20., 30., 1.))
self.assertTrue(equal_matrix(t, x))
def diag(values):
m = Matrix44()
for i, value in enumerate(values):
m[i, i] = value
return m
def test_get_row(self):
matrix = Matrix44()
self.assertEqual(matrix.get_row(0), (1.0, 0.0, 0.0, 0.0))
self.assertEqual(matrix.get_row(1), (0.0, 1.0, 0.0, 0.0))
self.assertEqual(matrix.get_row(2), (0.0, 0.0, 1.0, 0.0))
self.assertEqual(matrix.get_row(3), (0.0, 0.0, 0.0, 1.0))
def test_copy(self):
values = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15]
m1 = Matrix44(values)
matrix = m1.copy()
for v1, m1 in zip(values, matrix):
assert isclose(v1, m1)
def test_copy(self):
values = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15]
m1 = Matrix44(values)
matrix = m1.copy()
for v1, m1 in zip(values, matrix):
self.assertAlmostEqual(v1, m1)
def test_init_0(self):
matrix = Matrix44()
self.assertAlmostEqual(matrix[0, 0], 1.)
self.assertAlmostEqual(matrix[1, 1], 1.)
self.assertAlmostEqual(matrix[2, 2], 1.)
self.assertAlmostEqual(matrix[3, 3], 1.)
def test_get_row(self):
matrix = Matrix44()
assert matrix.get_row(0) == (1.0, 0.0, 0.0, 0.0)
assert matrix.get_row(1) == (0.0, 1.0, 0.0, 0.0)
assert matrix.get_row(2) == (0.0, 0.0, 1.0, 0.0)
assert matrix.get_row(3) == (0.0, 0.0, 0.0, 1.0)
def __init__(self):
super().__init__()
self.transformation_matrix_revolved_entity = Matrix44()
def test_inverse_error(self):
m = Matrix44([1] * 16)
self.assertRaises(ZeroDivisionError, m.inverse)
def test_transform(self):
t = Matrix44.scale(2., .5, 1.)
r = t.transform((10., 20., 30.))
self.assertEqual(r, (20., 10., 30.))
def __init__(self):
super().__init__()
self.transformation_matrix_sweep_entity = Matrix44()
self.transformation_matrix_path_entity = Matrix44()
self.sweep_entity_transformation_matrix = Matrix44()
self.path_entity_transformation_matrix = Matrix44()
def __init__(self, doc: 'Drawing' = None):
super().__init__(doc)
self.transformation_matrix_revolved_entity = Matrix44()
def __init__(self, doc: 'Drawing' = None):
super().__init__(doc)
self.transformation_matrix_sweep_entity = Matrix44()
self.transformation_matrix_path_entity = Matrix44()
self.sweep_entity_transformation_matrix = Matrix44()
self.path_entity_transformation_matrix = Matrix44()
def get_matrix(subclass: 'Tags', code: int) -> Matrix44:
values = [tag.value for tag in subclass.find_all(code)]
if len(values) != 16:
raise DXFStructureError('Invalid transformation matrix.')
return Matrix44(values)
