ref: e47ade3a8ec171fb24a7b38126070851c81ca599
dir: /python/aubio/gnuplot.py/
"""Copyright (C) 2004 Paul Brossier <piem@altern.org>
print aubio.__LICENSE__ for the terms of use
"""
__LICENSE__ = """\
Copyright (C) 2004 Paul Brossier <piem@altern.org>
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., 675 Mass Ave, Cambridge, MA 02139, USA.
"""
__notesheight = 0.25
def audio_to_array(filename):
import aubio.aubioclass
import numarray
hopsize = 2048
filei = aubio.aubioclass.sndfile(filename)
framestep = 1/(filei.samplerate()+0.)
channels = filei.channels()
myvec = aubio.aubioclass.fvec(hopsize,channels)
data = []
readsize = hopsize
while (readsize==hopsize):
readsize = filei.read(hopsize,myvec)
#for i in range(channels):
i = 0
curpos = 0
while (curpos < readsize):
data.append(myvec.get(curpos,i))
curpos+=1
time = numarray.arange(len(data))*framestep
return time,data
def plot_audio(filenames, fileout=None, start=0, end=None, noaxis=None):
g = gnuplot_init(fileout)
d = []
todraw = len(filenames)
xorig = 0.
xsize = 1./todraw
g.gnuplot('set multiplot;')
while (len(filenames)):
time,data = audio_to_array(filenames.pop(0))
d.append(make_audio_plot(time,data))
if not noaxis and todraw==1:
g.xlabel('Time (s)')
g.ylabel('Amplitude')
g.gnuplot('set size %f,1.;' % (xsize) )
g.gnuplot('set origin %f,0.;' % (xorig) )
g.gnuplot('set style data lines; \
set yrange [-1.:1.]; \
set xrange [0:%f]' % time[-1])
g.plot(d.pop(0))
xorig += 1./todraw
g.gnuplot('unset multiplot;')
def audio_to_spec(filename):
from aubioclass import fvec,cvec,pvoc,sndfile
from math import log
bufsize = 256*16
hopsize = bufsize/4 # could depend on filelength
filei = sndfile(filename)
srate = float(filei.samplerate())
framestep = hopsize/srate
freqstep = srate/bufsize
channels = filei.channels()
myvec = fvec(hopsize,channels)
myfft = cvec(bufsize,channels)
pv = pvoc(bufsize,hopsize,channels)
data,time,freq = [],[],[]
for f in range(bufsize/2):
freq.append(f*freqstep)
readsize = hopsize
frameread = 0
while (readsize==hopsize):
readsize = filei.read(hopsize,myvec)
pv.do(myvec,myfft)
frame = []
i = 0 #for i in range(channels):
curpos = 0
while (curpos < bufsize/2):
frame.append(log(myfft.get(curpos,i)**2+0.000001))
curpos+=1
time.append(frameread*framestep)
data.append(frame)
frameread += 1
# crop data if unfinished frames
if len(data[-1]) != len(data[0]):
data = data[0:-2]
time = time[0:-2]
# verify size consistency
assert len(data) == len(time)
assert len(data[0]) == len(freq)
return data,time,freq
def plot_spec(filename, outplot='',extension='', fileout=None, start=0, end=None, noaxis=None,log=1):
import Gnuplot
g = gnuplot_create(outplot,extension)
data,time,freq = audio_to_spec(filename)
xorig = 0.
xsize = 1.#/todraw
if not noaxis:
g.xlabel('Time (s)')
g.ylabel('Frequency (Hz)')
g.gnuplot('set pm3d map')
#g.gnuplot('set palette rgbformulae 30,31,32')
#g.gnuplot('set palette')
g.gnuplot('set xrange [0.:%f]' % time[-1])
g.gnuplot('set yrange [1.:%f]' % (freq[-1]/1.))
if log:
g.gnuplot('set yrange [10.1:%f]' % (freq[-1]/1.))
g.gnuplot('set log y')
g.splot(Gnuplot.GridData(data,time,freq, binary=1))
#xorig += 1./todraw
def downsample_audio(time,data,maxpoints=10000):
""" resample audio data to last only maxpoints """
import numarray
length = len(time)
downsample = length/maxpoints
if downsample == 0: downsample = 1
x = numarray.array(time).resize(length)[0:-1:downsample]
y = numarray.array(data).resize(length)[0:-1:downsample]
return x,y
def make_audio_plot(time,data,maxpoints=10000):
""" create gnuplot plot from an audio file """
import numarray
import Gnuplot, Gnuplot.funcutils
length = len(time)
downsample = length/maxpoints
if downsample == 0: downsample = 1
x = numarray.array(time).resize(length)[0:-1:downsample]
y = numarray.array(data).resize(length)[0:-1:downsample]
return Gnuplot.Data(x,y,with='lines')
def gnuplot_init(outplot,debug=0,persist=1):
# prepare the plot
import Gnuplot
g = Gnuplot.Gnuplot(debug=debug, persist=persist)
if outplot == 'stdout':
g("set terminal png fontfile 'p052023l.pfb'")
#g('set output \'%s\'' % outplot)
elif outplot:
extension = outplot.split('.')[-1]
if extension == 'ps': extension = 'postscript'
g('set terminal %s' % extension)
g('set output \'%s\'' % outplot)
return g
def gnuplot_create(outplot='',extension='',debug=0,persist=1):
import Gnuplot
g = Gnuplot.Gnuplot(debug=debug, persist=persist)
if not extension or not outplot: return g
if extension == 'ps': ext, extension = '.ps' , 'postscript'
elif extension == 'png': ext, extension = '.png', 'png'
elif extension == 'svg': ext, extension = '.svg', 'svg'
else: exit("ERR: unknown plot extension")
g('set terminal %s' % extension)
if outplot != "stdout":
g('set output \'%s%s\'' % (outplot,ext))
return g