def plot(self, num_pts, color=None, alpha=None, ax=None):
"""Plot the current curve.
Args:
num_pts (int): Number of points to plot.
color (Optional[Tuple[float, float, float]]): Color as RGB profile.
alpha (Optional[float]): The alpha channel for the color.
ax (Optional[matplotlib.artist.Artist]): matplotlib axis object
to add plot to.
Returns:
matplotlib.artist.Artist: The axis containing the plot. This
may be a newly created axis.
Raises:
NotImplementedError: If the curve's dimension is not ``2``.
"""
if self._dimension != 2:
raise NotImplementedError('2D is the only supported dimension',
'Current dimension', self._dimension)
s_vals = np.linspace(0.0, 1.0, num_pts)
points = self.evaluate_multi(s_vals)
if ax is None:
ax = _plot_helpers.new_axis()
ax.plot(points[:, 0], points[:, 1], color=color, alpha=alpha)
return ax
def plot(self, pts_per_edge, color=None, ax=None, with_nodes=False):
"""Plot the current surface.
Args:
pts_per_edge (int): Number of points to plot per edge.
color (Optional[Tuple[float, float, float]]): Color as RGB profile.
ax (Optional[matplotlib.artist.Artist]): matplotlib axis object
to add plot to.
with_nodes (Optional[bool]): Determines if the control points
should be added to the plot. Off by default.
Returns:
matplotlib.artist.Artist: The axis containing the plot. This
may be a newly created axis.
Raises:
NotImplementedError: If the surface's dimension is not ``2``.
"""
if self._dimension != 2:
raise NotImplementedError('2D is the only supported dimension',
'Current dimension', self._dimension)
if ax is None:
ax = _plot_helpers.new_axis()
_plot_helpers.add_patch(ax, color, pts_per_edge, *self._get_edges())
if with_nodes:
ax.plot(self._nodes[:, 0], self._nodes[:, 1],
color='black', marker='o', linestyle='None')
return ax
def plot(self, pts_per_edge, color=None, ax=None):
"""Plot the current curved polygon.
Args:
pts_per_edge (int): Number of points to plot per curved edge.
color (Optional[Tuple[float, float, float]]): Color as RGB profile.
ax (Optional[matplotlib.artist.Artist]): matplotlib axis object
to add plot to.
Returns:
matplotlib.artist.Artist: The axis containing the plot. This
may be a newly created axis.
"""
if ax is None:
ax = _plot_helpers.new_axis()
_plot_helpers.add_patch(ax, color, pts_per_edge, *self._edges)
return ax
def _call_function_under_test():
from bezier import _plot_helpers
return _plot_helpers.new_axis()
