Source code for splinelib.plotting

import numpy as np
import matplotlib.pyplot as plt
from .splinelib import *


[docs]def add_spline_to_plot(spline, include_control_polygon=True, include_knots=True):
    """
    Adds given spline to current matplotlib plot

    Args:
        spline (Spline):                 Spline to add
        include_control_polygon (bool)   Optional. Whether to include control polygon in
                                         plot or not. Default is True.
        include_knots (bool)             Optional. Whether to include knots in plot or not.
                                         Default is True

    Returns:
        None
    """
    xs, ps = spline.evaluate_all()
    plt.plot(xs, ps)

    if include_control_polygon:
        plt.plot(*spline.get_control_polygon())

    min_val = np.min(ps)
    dist = np.max(ps) - min_val

    steps = [i * (dist / 8) - dist for i in range(spline.get_degree() + 1)]
    similar_knots = 0
    previous_knot = None


    def double_equals(d1, d2, tol=1e-10):
        if d1 is None or d2 is None:
            return False
        else:
            return abs(d1 - d2) < tol


    if not spline.is_parametric() and include_knots:
        for knot in spline.get_knots():
            if double_equals(knot, previous_knot):
                similar_knots += 1
            else:
                similar_knots = 0

            plt.scatter(knot, steps[similar_knots], color="k", s=10)

            previous_knot = knot