Add unit test for mass matrix access

Author: alxbilger

bindings/Sofa/tests/CMakeLists.txt

@@ -16,6 +16,7 @@ set(PYTHON_FILES
     ${CMAKE_CURRENT_SOURCE_DIR}/Core/BaseObject.py
     ${CMAKE_CURRENT_SOURCE_DIR}/Core/Controller.py
     ${CMAKE_CURRENT_SOURCE_DIR}/Core/ForceField.py
+    ${CMAKE_CURRENT_SOURCE_DIR}/Core/Mass.py
     ${CMAKE_CURRENT_SOURCE_DIR}/Core/DataEngine.py
     ${CMAKE_CURRENT_SOURCE_DIR}/Core/MyRestShapeForceField.py
     ${CMAKE_CURRENT_SOURCE_DIR}/Core/BaseLink.py

bindings/Sofa/tests/Core/Mass.py

@@ -0,0 +1,68 @@
+# coding: utf8
+
+import unittest
+import Sofa
+import Sofa.Core
+import Sofa.Helper
+import Sofa.Simulation
+import SofaRuntime
+
+class Test(unittest.TestCase):
+
+    @staticmethod
+    def simulate_beam(linear_solver_template):
+        root = Sofa.Core.Node("rootNode")
+
+        loop = root.addObject('DefaultAnimationLoop')
+
+        root.addObject('RequiredPlugin', name='Sofa.Component.ODESolver.Backward')
+        root.addObject('RequiredPlugin', name='Sofa.Component.LinearSolver.Direct')
+        root.addObject('RequiredPlugin', name='Sofa.Component.Engine.Select')
+        root.addObject('RequiredPlugin', name='Sofa.Component.Constraint.Projective')
+        root.addObject('RequiredPlugin', name='Sofa.Component.SolidMechanics.FEM.Elastic')
+        root.addObject('RequiredPlugin', name='Sofa.Component.Mass')
+
+        root.addObject('EulerImplicitSolver', rayleighStiffness="0.1", rayleighMass="0.1")
+        root.addObject('SparseLDLSolver', applyPermutation="false", template=linear_solver_template)
+
+        root.addObject('MechanicalObject', name="DoFs")
+        mass = root.addObject('MeshMatrixMass', name="mass", totalMass="320")
+        root.addObject('RegularGridTopology', name="grid", nx="4", ny="4", nz="20", xmin="-9", xmax="-6", ymin="0", ymax="3", zmin="0", zmax="19")
+        root.addObject('BoxROI', name="box", box="-10 -1 -0.0001  -5 4 0.0001")
+        root.addObject('FixedConstraint', indices="@box.indices")
+        root.addObject('HexahedronFEMForceField', name="FEM", youngModulus="4000", poissonRatio="0.3", method="large")
+
+        matrix_accessor = root.addObject(MatrixAccessController('MatrixAccessor', name='matrixAccessor', mass=mass))
+
+        Sofa.Simulation.init(root)
+        Sofa.Simulation.animate(root, 0.0001)
+
+        return matrix_accessor.mass_matrix
+
+    def test_stiffness_matrix_access_scalar(self):
+
+        M = self.simulate_beam("CompressedRowSparseMatrixd")
+
+        self.assertEqual(M.ndim, 2)
+        self.assertEqual(M.shape, (960, 960))
+        self.assertEqual(M.nnz, 9480)
+
+    def test_stiffness_matrix_access_blocks3x3(self):
+
+        M = self.simulate_beam("CompressedRowSparseMatrixMat3x3d")
+
+        self.assertEqual(M.ndim, 2)
+        self.assertEqual(M.shape, (960, 960))
+        self.assertEqual(M.nnz, 9480)
+
+
+
+class MatrixAccessController(Sofa.Core.Controller):
+
+
+    def __init__(self, *args, **kwargs):
+        Sofa.Core.Controller.__init__(self, *args, **kwargs)
+        self.mass = kwargs.get("mass")
+
+    def onAnimateEndEvent(self, event):
+        self.mass_matrix = self.mass.assembleMMatrix()