discretize.TreeMesh.refine_triangle

TreeMesh.refine_triangle(self, triangle, levels, finalize=True, diagonal_balance=None)

Refine the TreeMesh along the triangle to the desired level.

Refines the TreeMesh by determining if a cell intersects the given triangle(s) to the prescribed level(s).

Parameters:

triangle(N, 3, dim) array_like

The nodes of the triangle(s).

levelsint or (N) array_like of int

The level to refine intersecting cells to.

finalizebool, optional

Whether to finalize after refining

diagonal_balancebool or None, optional

Whether to balance cells diagonally in the refinement, None implies using the same setting used to instantiate the TreeMesh`.

Examples

We create a simple mesh and refine the TreeMesh such that all cells that intersect the line segment path are at the given levels.

>>> import discretize
>>> import matplotlib.pyplot as plt
>>> import matplotlib.patches as patches
>>> tree_mesh = discretize.TreeMesh([32, 32])
>>> tree_mesh.max_level
5

Next we define the points along the line and the level we want to refine to, and refine the mesh.

>>> triangle = [[0.14, 0.31], [0.32, 0.96], [0.23, 0.87]]
>>> levels = 5
>>> tree_mesh.refine_triangle(triangle, levels)

Now lets look at the mesh, and overlay the line on it to ensure it refined where we wanted it to.

>>> ax = tree_mesh.plot_grid()
>>> tri = patches.Polygon(triangle, fill=False)
>>> ax.add_patch(tri)
>>> plt.show()

(Source code, png, pdf)