visualization.convert_to_colored_map

visualization.convert_to_colored_map(psi2tree, shape=None)

Converts to a parition map of the domain. This assumes psi2tree.geometry to be a grid.

Parameters

Name Type Description Default
psi2tree psiclone.psi2tree.Psi2tree required
shape tuple of ints Shape of the 2D domain. None