python/demos/demo_tss.py

   1 #! /usr/bin/env python
   2
   3 import sys
   4 from aubio import source, sink, pvoc, tss
   5
   6 if __name__ == '__main__':
   7   if len(sys.argv) < 2:
   8     print 'usage: %s <inputfile> <outputfile_transient> <outputfile_steady>' % sys.argv[0]
   9     sys.exit(1)
  10
  11   samplerate = 44100
  12   win_s = 1024      # fft size
  13   hop_s = win_s / 4 # block size
  14   threshold = 0.5
  15
  16   f = source(sys.argv[1], samplerate, hop_s)
  17   g = sink(sys.argv[2], samplerate)
  18   h = sink(sys.argv[3], samplerate)
  19
  20   pva = pvoc(win_s, hop_s)    # a phase vocoder
  21   pvb = pvoc(win_s, hop_s)    # another phase vocoder
  22   t = tss(win_s, hop_s)       # transient steady state separation
  23
  24   t.set_threshold(threshold)
  25
  26   read = hop_s
  27
  28   while read:
  29     samples, read = f()               # read file
  30     spec = pva(samples)                # compute spectrum
  31     trans_spec, stead_spec = t(spec)  # transient steady-state separation
  32     transients = pva.rdo(trans_spec)   # overlap-add synthesis of transients
  33     steadstate = pvb.rdo(stead_spec)   # overlap-add synthesis of steady states
  34     g(transients, read)               # write transients to output
  35     h(steadstate, read)               # write steady states to output
  36
  37   del f, g, h                         # finish writing the files now
  38
  39   from demo_spectrogram import get_spectrogram
  40   from pylab import subplot, show
  41   subplot(311)
  42   get_spectrogram(sys.argv[1])
  43   subplot(312)
  44   get_spectrogram(sys.argv[2])
  45   subplot(313)
  46   get_spectrogram(sys.argv[3])
  47   show()