added 2d example

Author: teseoch

polyfempy/Problems.py

@@ -83,12 +83,12 @@ def __init__(self):
 
 	def add_dirichlet_value(self, id, value, is_dirichlet_dim=None):
 		"""add the Dirichlet value value for the sideset id. Note the value must be a vector in 2D or 3D depending on the problem. is_dirichlet_dim is a vector of boolean specifying which dimentions are fixed."""
-		assert(len(value) == 3)
+		assert(len(value) == 3 or len(value) == 2)
 		tmp = {}
 		tmp["id"] = id
 		tmp["value"] = value
 		if is_dirichlet_dim is not None:
-			assert(len(is_dirichlet_dim) == 3)
+			assert(len(is_dirichlet_dim) == 3 or len(is_dirichlet_dim) == 2)
 			tmp["dimension"] = is_dirichlet_dim
 
 		self.dirichlet_boundary.append(tmp)

tests/plane_hole.py

@@ -0,0 +1,50 @@
+import unittest
+
+import polyfempy as pf
+import utils
+import os
+
+
+class BendingTest(unittest.TestCase):
+    def test_run(self):
+        dir_path = os.path.dirname(os.path.realpath(__file__))
+        mesh_path = os.path.join(dir_path, "../3rdparty/data/plane_hole.obj")
+
+        settings = pf.Settings()
+        settings.discr_order = 1
+        settings.normalize_mesh = True
+        # settings.vismesh_rel_area = 0.1
+
+
+        settings.set_material_params("E", 210000)
+        settings.set_material_params("nu", 0.3)
+
+
+        settings.tensor_formulation = pf.TensorFormulations.LinearElasticity
+
+        problem = pf.GenericTensor()
+        problem.add_dirichlet_value(1, [0, 0], [True, False])
+        problem.add_dirichlet_value(4, [0, 0], [False, True])
+
+        problem.add_neumann_value(3, [100, 0])
+
+        settings.set_problem(problem)
+
+
+        solver = pf.Solver()
+
+        solver.settings(str(settings))
+        solver.load_mesh(mesh_path)
+
+        solver.solve()
+
+        [pts, tets, disp] = solver.get_sampled_solution()
+        vertices = pts + disp
+        mises, _ = solver.get_sampled_mises_avg()
+
+        utils.plot(vertices, tets, mises)
+
+
+
+if __name__ == '__main__':
+    unittest.main()

tests/utils.py

@@ -7,7 +7,11 @@
 def plot(vertices, connectivity, function):
     x = vertices[:,0]
     y = vertices[:,1]
-    z = vertices[:,2]
+
+    if vertices.shape[1] == 3:
+        z = vertices[:,2]
+    else:
+        z = np.zeros(x.shape, dtype=x.dtype)
 
     if connectivity.shape[1] == 3:
         f = connectivity