diff options
Diffstat (limited to 'plot.py')
| -rw-r--r-- | plot.py | 52 |
1 files changed, 52 insertions, 0 deletions
@@ -0,0 +1,52 @@ +# Copyright (c) 2009 John Glover, National University of Ireland, Maynooth +# +# This program is free software; you can redistribute it and/or modify +# it under the terms of the GNU General Public License as published by +# the Free Software Foundation; either version 2 of the License, or +# (at your option) any later version. +# +# This program is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU General Public License for more details. +# +# You should have received a copy of the GNU General Public License +# along with this program; if not, write to the Free Software +# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA + +from pylab import plot, show + +def _plot_frame_peaks(frame, frame_number): + "Plot one frame, which is a list of Peak objects" + x_values = [frame_number for x in range(len(frame))] + y_values = [peak.frequency for peak in frame] + plot(x_values, y_values, "ro") + +def plot_peaks(peaks): + "Plot peaks found by a peak detection algorithm" + for frame_number, frame in enumerate(peaks): + _plot_frame_peaks(frame, frame_number) + +def plot_frame_peaks(peaks): + "Plot peaks in one frame" + x_values = [] + y_values = [] + for peak in peaks: + x_values.append(peak.frequency) + y_values.append(peak.amplitude) + plot(x_values, y_values, 'ro') + +def plot_partials(partials, show_peaks=True): + "Plot partials created by a partial tracking algorithm" + num_frames = max([partial.get_last_frame() for partial in partials]) + peaks = [[] for f in range(num_frames)] + for partial in partials: + x_values = [] + y_values = [] + for peak_number, peak in enumerate(partial.peaks): + x_values.append(partial.starting_frame + peak_number) + y_values.append(peak.frequency) + peaks[partial.starting_frame + peak_number].append(peak) + plot(x_values, y_values, "b") + if show_peaks: + plot_peaks(peaks)
\ No newline at end of file |