ref: 81fe32734168e80d53cd575535a63ce99b1eb6a0
dir: /python/demos/demo_tss.py/
#! /usr/bin/env python
import sys
from aubio import source, sink, pvoc, tss
if __name__ == '__main__':
if len(sys.argv) < 2:
print 'usage: %s <inputfile> <outputfile_transient> <outputfile_steady>' % sys.argv[0]
sys.exit(1)
samplerate = 44100
win_s = 512 # fft size
hop_s = win_s / 2 # block size
threshold = 0.26
f = source(sys.argv[1], samplerate, hop_s)
g = sink(sys.argv[2], samplerate)
h = sink(sys.argv[3], samplerate)
pv = pvoc(win_s, hop_s) # phase vocoder
pw = pvoc(win_s, hop_s) # another phase vocoder
t = tss(win_s, hop_s) # transient steady state separation
t.set_threshold(threshold)
read = hop_s
while read:
samples, read = f() # read file
spec = pv(samples) # compute spectrum
trans_spec, stead_spec = t(spec) # transient steady-state separation
transients = pv.rdo(trans_spec) # overlap-add synthesis of transients
steadstate = pw.rdo(stead_spec) # overlap-add synthesis of steady states
g(transients, read) # write transients to output
h(steadstate, read) # write steady states to output
del f, g, h # finish writing the files now
from demo_spectrogram import get_spectrogram
from pylab import subplot, show
subplot(311)
get_spectrogram(sys.argv[1])
subplot(312)
get_spectrogram(sys.argv[2])
subplot(313)
get_spectrogram(sys.argv[3])
show()