Visualization of (continuous) secondary source contour by polygon

sfs/plot.py

@@ -3,7 +3,8 @@
 import matplotlib.pyplot as plt
 from matplotlib.patches import PathPatch
 from matplotlib.path import Path
-from matplotlib.collections import PatchCollection
+from matplotlib.collections import PatchCollection, LineCollection
+from matplotlib.colors import ListedColormap
 from mpl_toolkits import axes_grid1
 from mpl_toolkits.mplot3d import Axes3D
 import numpy as np
@@ -86,6 +87,43 @@ def secondarysource_2d(x0, n0, grid=None):
  ax.add_artist(ss)
 
 
+def secondarysourcecontour_2d(x0, a0=True, ax=None):
+ """Draw contour of secondary source distribution as polygon.
+
+ Parameters
+ ----------
+ x0 : (N, 3) array_like
+ Secondary source positions.
+ a0 : bool or (N,) array_like, optional
+ Active secondary sources.
+ ax : Axes object, optional
+ The loudspeakers are plotted into this `matplotlib.axes.Axes`
+ object or -- if not specified -- into the current axes.
+ """
+ x0 = util.asarray_of_rows(x0)
+ a0 = util.asarray_1d(a0)
+
+ # colormap with black and gray to indicate active parts
+ cmap = ListedColormap(np.stack((.5*np.ones(3), np.zeros(3))))
+
+ # setup line collection with secondary source contour
+ x0 = np.append(x0, x0[0:1, :], axis=0)
+ points = np.array([x0[:, 0], x0[:, 1]]).T.reshape(-1, 1, 2)
+ segments = np.concatenate([points[:-1], points[1:]], axis=1)
+ lc = LineCollection(segments, cmap=cmap)
+ lc.set_linewidth(2)
+
+ # set color of line to indicate active parts
+ a0 = np.append(a0, a0[0])
+ active = np.logical_and(a0[0:-1], a0[1:])
+ lc.set_array(active)
+
+ # add collection of lines to current axis
+ if ax is None:
+ ax = plt.gca()
+ plt.gca().add_collection(lc)
+
+
 def loudspeaker_2d(x0, n0, a0=0.5, size=0.08, show_numbers=False, grid=None,
  ax=None):
  """Draw loudspeaker symbols at given locations and angles.