Source code for sfs.plot3d

"""3D plots of sound fields etc."""
import matplotlib.pyplot as _plt


def secondary_sources(x0, n0, a0=None, *, w=0.08, h=0.08):
    """Plot positions and normals of a 3D secondary source distribution."""
    fig = _plt.figure(figsize=(15, 15))
    ax = fig.add_subplot(111, projection='3d')
    q = ax.quiver(x0[:, 0], x0[:, 1], x0[:, 2], n0[:, 0],
                  n0[:, 1], n0[:, 2], length=0.1)
    _plt.xlabel('x (m)')
    _plt.ylabel('y (m)')
    _plt.title('Secondary Sources')
    return q