def test_create_cell(self):
# met.create_nodes_from_mg(self.mg)
grid = met.Grid(self.mg)
l1 = len(grid.cells)
self.assertEqual(len(grid.ndb.levels[0]), 81)
self.assertEqual(grid.idx_edge_pairs, [(0, 1), (0, 2), (1, 3), (2, 3)])
childs1 = grid.cells[0].make_childs()
self.assertEqual(len(childs1), 4)
expected_vertices = np.array([[-4., -4.], [-4., -3.5], [-3.5, -4.],
[-3.5, -3.5]])
self.assertTrue(
np.all(childs1[0].get_vertex_coords() == expected_vertices))
# five nodes had to be inserted
self.assertEqual(len(grid.ndb.levels[1]), 5)
childs2 = childs1[0].make_childs()
self.assertEqual(len(grid.ndb.levels[2]), 5)
self.assertEqual(childs1[0].child_cells, childs2)
self.assertEqual(childs2[0].parent_cell, childs1[0])
self.assertEqual(len(grid.levels[0]), l1)
self.assertEqual(len(grid.levels[1]), len(childs1))
self.assertEqual(len(grid.levels[2]), len(childs2))
if 0:
plt.plot(*grid.all_mg_points, '.')
plot_cells2d([gc] + childs1 + childs2, show=True)
def test_plot(self):
# create images where each new cell is shown
grid = met.Grid(self.mg)
l1 = len(grid.cells)
childs1 = grid.cells[0].make_childs()
self.assertEqual(len(grid.ndb.levels[1]), 19)
l2 = len(grid.cells)
self.assertEqual(len(childs1), 8)
self.assertEqual(l2, l1 + len(childs1))
expected_vertices = np.array([[-4.0, -4.0, -4.0], [-4.0, -4.0, -3.5],
[-4.0, -3.5, -4.0], [-4.0, -3.5, -3.5],
[-3.5, -4.0, -4.], [-3.5, -4.0, -3.5],
[-3.5, -3.5, -4.0], [-3.5, -3.5, -3.5]])
self.assertTrue(
np.all(childs1[0].get_vertex_coords() == expected_vertices))
childs2 = childs1[0].make_childs()
self.assertEqual(len(grid.ndb.levels[2]), 19)
self.assertEqual(childs1[0].child_cells, childs2)
self.assertEqual(childs2[0].parent_cell, childs1[0])
self.assertEqual(len(grid.cells), l1 + len(childs1) + len(childs2))
if 0:
plot_cells = grid.cells[:1] + [grid.cells[-16]] + grid.cells[-8:]
plot_cells3d(plot_cells,
imax=None,
show=True,
all_points=grid.all_mg_points)
def test_create_cells(self):
grid = met.Grid(self.mg)
self.assertEqual(list(grid.cells[0].vertex_nodes[0].coords),
[-4.0, -4.0, -4.0])
self.assertEqual(list(grid.cells[0].vertex_nodes[3].coords),
[-4.0, -3.0, -3.0])
def test_refinement(self):
grid = met.Grid(self.mg)
# perform refinement and get a point_collection (pc)
pc0 = pc = grid.refinement_step(met.func_sphere_nd)
# check inner and outer points
self.assertEqual(pc.ip.shape, (3, 7))
self.assertEqual(pc.op.shape, (3, 722))
# check inner and outer boundary points
self.assertEqual(pc.ibp.shape, (3, 7))
self.assertEqual(pc.obp.shape, (3, 74))
# plot inner and outer points (level 0)
ax = plt.figure().add_subplot(1, 1, 1, projection='3d')
ax.plot(*grid.all_mg_points, '.', ms=1, color="k", alpha=0.5)
ax.plot(*pc.ip, 'o', ms=3, color="r", alpha=0.3)
ax.plot(*pc.op, 'o', ms=3, color="b", alpha=0.01)
ax.plot(*pc.ibp, 'o', ms=3, color="r", alpha=1)
ax.plot(*pc.obp, 'o', ms=3, color="b", alpha=1)
# plot level 1
pc1 = pc = grid.refinement_step(met.func_sphere_nd)
ax.plot(*pc.ibp, 'o', ms=3, color="r", alpha=1)
ax.plot(*pc.obp, 'o', ms=3, color="b", alpha=1)
# advance to level 3
pc2 = grid.refinement_step(met.func_sphere_nd)
pc3 = grid.refinement_step(met.func_sphere_nd)
# new figure
ax = plt.figure().add_subplot(1, 1, 1, projection='3d')
# ax.plot(*grid.all_mg_points, '.', ms=1, color="k", alpha=0.5)
ax.plot(*pc0.ibp, 'o', ms=3, color="m", alpha=1)
ax.plot(*pc1.ibp, 'o', ms=3, color="r", alpha=1)
ax.plot(*pc2.ibp, 'o', ms=3, color="g", alpha=1)
ax.plot(*pc3.ibp, 'o', ms=3, color="b", alpha=1)
# -> plot the involved cells for debugging
plt.savefig("n3_level3.png")
if exec_show:
plt.show()
def _test_plot(self):
# create images where each new cell is shown
grid = met.Grid(self.mg)
plt.plot(*grid.all_mg_points, '.')
plt.savefig("tmp_0.png")
for i, cell in enumerate(grid.cells):
edges = np.array(cell.get_edge_coords())
plt.plot(*edges.T)
plt.savefig("tmp_{:03d}.png".format(i))
ipd.IPS()
def test_make_childs_bug(self):
grid = met.Grid(self.mg)
childs1 = grid.cells[0].make_childs()
diff_arr = np.array(grid.vertex_local_idcs) * 0.5
# ensure that the lower left is the first vertex (this is needed as reference for the next child generation)
for c in childs1:
node_coords = np.array([vn.coords for vn in c.vertex_nodes])
reference_node = node_coords[0, :]
# the other nodes must be reachable from the reference node via the vectors in diff_arr
node_coords2 = reference_node + diff_arr
self.assertTrue(np.allclose(node_coords, node_coords2))
def test_refinement(self):
plot_points = False
exec_show = False
grid = met.Grid(self.mg)
pc0 = pc = grid.refinement_step(met.func_sphere_nd)
ic0 = grid.inhomogeneous_cells[0]
ac0 = grid.levels[0]
plot_cells2d(ac0, color="0.7", alpha=0.3)
plot_cells2d(ic0, color="0.7", alpha=0.3)
plt.gca().add_patch(plt.Circle([0, 0], 1.3**.5, alpha=0.5))
# there are 12 inhomogeneous cells (manually verified)
self.assertEqual(len(ic0), 12)
self.assertEqual(pc.ibp.shape, (2, 5))
self.assertEqual(pc.obp.shape, (2, 16))
if plot_points:
# plot inner and outer points (level 0)
plt.plot(*pc.op, "bo", alpha=0.2, ms=5)
plt.plot(*pc.ip, "ro", alpha=0.2, ms=5)
plt.plot(*pc.obp, "bo", ms=3)
plt.plot(*pc.ibp, "ro", ms=3)
plt.title("levels 0")
plt.axis("square")
plt.savefig("n2_level0.png")
# -----
self.assertEqual(grid.max_level, 0)
pc1 = pc = grid.refinement_step(met.func_sphere_nd)
self.assertEqual(grid.max_level, 1)
self.assertEqual(pc.ibp.shape, (2, 16))
self.assertEqual(pc.obp.shape, (2, 24))
plt.cla()
if plot_points:
# plot inner and outer points (level 1)
plt.plot(*pc.op, "go", alpha=0.2, ms=5)
plt.plot(*pc.ip, "mo", alpha=0.2, ms=5)
plt.plot(*pc.obp, "go", ms=3)
plt.plot(*pc.ibp, "mo", ms=3)
pc2 = pc = grid.refinement_step(met.func_sphere_nd)
ac0 = grid.levels[0]
ic0 = grid.inhomogeneous_cells[0]
ic1 = grid.inhomogeneous_cells[1]
ac1 = grid.levels[1]
ac2 = grid.levels[2]
ic2 = grid.inhomogeneous_cells[2]
plot_cells2d(ac0, color="0.7", alpha=0.3)
plot_cells2d(ac1, color="0.7", alpha=0.5)
plot_cells2d(ac2, color="0.7", alpha=0.8)
plot_cells2d(ic0, color=(.5, 0, 0))
plot_cells2d(ic1, color=(.7, 0, 0))
plot_cells2d(ic2, color=(.9, 0, 0))
plt.gca().add_patch(plt.Circle([0, 0], 1.3**.5))
plt.title("levels 0, 1, 2")
plt.savefig("n2_level2.png")
if exec_show:
plt.show()
# -----
pc2 = pc = grid.refinement_step(met.func_sphere_nd)
# plot inner and outer points (level 2)
plt.cla()
plt.axis("square")
plt.axis([-1.8, 1.8, -1.8, 1.8])
plt.gca().add_patch(plt.Circle([0, 0], 1.3**.5, alpha=0.5))
#
# plt.plot(*pc.op, "go", alpha=0.2, ms=5)
# plt.plot(*pc.ip, "mo", alpha=0.2, ms=5)
plt.plot(*pc.obp, "go", ms=3)
plt.plot(*pc.ibp, "mo", ms=3)
plt.title("levels 0, 1, 2")
plt.savefig("n2_level2_points.png")
if exec_show:
plt.show()