Source code for mpt4py.geometry.visualization

from .plot import PlotterProtocol, PolyhedralPlottingData
from .plot_pyvista import PyvistaPlotter
from .plot_matplotlib import MatplotlibPlotter

from typing import Optional



def get_plotter(plotter_name: Optional[str] = None) -> plot.PlotterProtocol:
	"""
	Get a plotter object based on the specified backend.
	"""
	plotter_name = 'pyvista' if plotter_name is None else plotter_name
	if plotter_name == 'pyvista':
		return PyvistaPlotter()
	elif plotter_name == 'matplotlib':
		return MatplotlibPlotter()
	raise ValueError("Invalid plotting backend.")




def new_figure(backend: Optional[str] = 'pyvista'):
	r"""
    Create a new figure using the specified backend.
    """
	return get_plotter(backend)


__all__ = [
	'PlotterProtocol',
	'PolyhedralPlottingData',
	'PyvistaPlotter',
	'MatplotlibPlotter',
	'get_plotter',
	'new_figure',
]