source: python/demos/demo_tss.py @ 4120fbc

feature/autosinkfeature/constantqfeature/pitchshiftfeature/pydocstringsfeature/timestretchpitchshiftsamplertimestretchyinfft+
Last change on this file since 4120fbc was 4120fbc, checked in by Paul Brossier <piem@piem.org>, 4 years ago

python/demos: python3 and double precision compatibility

  • Property mode set to 100755
File size: 1.5 KB
RevLine 
[7175ed4]1#! /usr/bin/env python
[d3b1eb1]2
3import sys
4from aubio import source, sink, pvoc, tss
5
6if __name__ == '__main__':
[d0faeca]7    if len(sys.argv) < 2:
[4120fbc]8        print('usage: %s <inputfile> <outputfile_transient> <outputfile_steady>' % sys.argv[0])
[d0faeca]9        sys.exit(1)
10
11    samplerate = 44100
[4120fbc]12    win_s = 1024       # fft size
13    hop_s = win_s // 4 # block size
[d0faeca]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])
[8fb567c]47    show()
Note: See TracBrowser for help on using the repository browser.