Add matrix primitives

src/ansys/geometry/core/primitives/__init__.py

@@ -1,5 +1,16 @@
 """PyGeometry primitives subpackage."""
+from ansys.geometry.core.primitives.matrix import Matrix33, Matrix44, RotationMatrix
 from ansys.geometry.core.primitives.point import Point2D, Point3D
 from ansys.geometry.core.primitives.vector import UnitVector2D, UnitVector3D, Vector2D, Vector3D
 
-__all__ = ["Point2D", "Point3D", "UnitVector2D", "UnitVector3D", "Vector2D", "Vector3D"]
+__all__ = [
+    "Point2D",
+    "Point3D",
+    "UnitVector2D",
+    "UnitVector3D",
+    "Vector2D",
+    "Vector3D",
+    "Matrix33",
+    "Matrix44",
+    "RotationMatrix",
+]

src/ansys/geometry/core/primitives/matrix.py

@@ -0,0 +1,78 @@
+from math import cos, sin
+
+import numpy as np
+
+from ansys.geometry.core.primitives.vector import Vector2D, Vector3D
+
+
+class Matrix33(np.ndarray):
+    def __new__(cls, input):
+        """Constructor for ``Point3D``."""
+
+        matrix = np.asarray(input).view(cls)
+
+        if matrix is None or matrix.ndim != 2 or matrix.shape != (3, 3):
+            raise ValueError("Matrix33 should only be a 2D array of shape (3,3).")
+
+        if not np.issubdtype(matrix.dtype, np.number) or not isinstance(matrix, (np.ndarray)):
+            raise ValueError("The input parameters should be integer or float.")
+
+        return matrix
+
+    def inverse(self):
+        det = np.linalg.det(self)
+        if det == 0:
+            raise ValueError("The determinent of matrix is zero, cannot be inversed")
+        return np.linalg.inv(self)
+
+
+class Matrix44(np.ndarray):
+    def __new__(cls, input):
+        """Constructor for ``Point3D``."""
+
+        matrix = np.asarray(input).view(cls)
+
+        if matrix is None or matrix.ndim != 2 or matrix.shape != (4, 4):
+            raise ValueError("Matrix44 shouldonly be a 2D array of shape (4,4).")
+
+        if not np.issubdtype(matrix.dtype, np.number) or not isinstance(matrix, (np.ndarray)):
+            raise ValueError("The input parameters should be integer or float.")
+
+        return matrix
+
+    def inverse(self):
+        det = np.linalg.det(self)
+        if det == 0:
+            raise ValueError("The determinent of matrix is zero, cannot be inversed")
+        return np.linalg.inv(self)
+
+
+# from ansys.geometry.core.primitives import Matrix33
+# a = Matrix33([[1,2,3],[2,5,6],[2,5,10]])
+# b = Matrix33([[1,0,0],[0,1,0],[0,0,1]])
+class RotationMatrix(Matrix33):
+    def __new__(cls, input, theta):
+        obj = Matrix33(input)
+        rot = np.array([[cos(theta), -sin(theta), 0], [sin(theta), cos(theta), 0], [0, 0, 1]])
+        return np.multiply(rot, obj)
+
+
+class TranslationMatrix(Matrix33):
+    def __new__(cls, input, v: Vector2D):
+        obj = Matrix33(input)
+        translate = np.array([[0, 0, v.x], [0, 0, v.y], [0, 0, 1]])
+        return np.multiply(obj, translate)
+
+
+class TranslateRotateMatrix(Matrix44):
+    def __new__(cls, input, angle, v: Vector3D):
+        obj = Matrix44(input)
+        transrot = np.array(
+            [
+                [0, 0, 0, v.x],
+                [0, cos(angle), -sin(angle), v.y],
+                [0, sin(angle), cos(angle), v.z],
+                [0, 0, 0, 1],
+            ]
+        )
+        return np.multiply(obj, transrot)