Merge branch 'master' into awhitening
[aubio.git] / python / demos / demo_sink.py
1 #! /usr/bin/env python
2
3 import sys
4 from aubio import source, sink
5
6 if __name__ == '__main__':
7     if len(sys.argv) < 3:
8         print('usage: %s <inputfile> <outputfile> [samplerate] [hop_size]' % sys.argv[0])
9         sys.exit(1)
10
11     if len(sys.argv) > 3: samplerate = int(sys.argv[3])
12     else: samplerate = 0
13     if len(sys.argv) > 4: hop_size = int(sys.argv[4])
14     else: hop_size = 256
15
16     f = source(sys.argv[1], samplerate, hop_size)
17     if samplerate == 0: samplerate = f.samplerate
18     g = sink(sys.argv[2], samplerate)
19
20     total_frames, read = 0, hop_size
21     while read:
22         vec, read = f()
23         g(vec, read)
24         total_frames += read
25     outstr = "wrote %.2fs" % (total_frames / float(samplerate))
26     outstr += " (%d frames in" % total_frames
27     outstr += " %d blocks" % (total_frames // f.hop_size)
28     outstr += " at %dHz)" % f.samplerate
29     outstr += " from " + f.uri
30     outstr += " to " + g.uri
31     print(outstr)