# 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
mesh = discretize.TreeMesh([32, 32, 32])
mesh.max_level
# Expected:
## 5
#
# Next we define the bottom points of the prism, its heights, and the level we
# want to refine to, then refine the mesh.
#
triangle = [[0.14, 0.31, 0.21], [0.32, 0.96, 0.34], [0.87, 0.23, 0.12]]
height = 0.35
levels = 5
mesh.refine_vertical_trianglular_prism(triangle, height, levels)
#
# Now lets look at the mesh.
#
v = mesh.cell_levels_by_index(np.arange(mesh.n_cells))
fig, axs = plt.subplots(1, 3, figsize=(12,4))
mesh.plot_slice(v, ax=axs[0], normal='x', grid=True, clim=[2, 5])
mesh.plot_slice(v, ax=axs[1], normal='y', grid=True, clim=[2, 5])
mesh.plot_slice(v, ax=axs[2], normal='z', grid=True, clim=[2, 5])
plt.show()