Imported Upstream version 3.2.2
[debian/gnuradio] / gnuradio-examples / python / audio / dial_tone_wav.py
1 #!/usr/bin/env python
2 #
3 # Copyright 2004,2005,2007,2008 Free Software Foundation, Inc.
4
5 # This file is part of GNU Radio
6
7 # GNU Radio is free software; you can redistribute it and/or modify
8 # it under the terms of the GNU General Public License as published by
9 # the Free Software Foundation; either version 3, or (at your option)
10 # any later version.
11
12 # GNU Radio is distributed in the hope that it will be useful,
13 # but WITHOUT ANY WARRANTY; without even the implied warranty of
14 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15 # GNU General Public License for more details.
16
17 # You should have received a copy of the GNU General Public License
18 # along with GNU Radio; see the file COPYING.  If not, write to
19 # the Free Software Foundation, Inc., 51 Franklin Street,
20 # Boston, MA 02110-1301, USA.
21
22
23 # GNU Radio example program to record a dial tone to a WAV file
24
25 from gnuradio import gr
26 from gnuradio.eng_option import eng_option
27 from optparse import OptionParser
28
29 class my_top_block(gr.top_block):
30
31     def __init__(self):
32         gr.top_block.__init__(self)
33
34         usage = "%prog: [options] filename"
35         parser = OptionParser(option_class=eng_option, usage=usage)
36         parser.add_option("-r", "--sample-rate", type="eng_float", default=48000,
37                           help="set sample rate to RATE (48000)")
38         parser.add_option("-N", "--samples", type="eng_float", default=None,
39                           help="number of samples to record")
40         (options, args) = parser.parse_args ()
41         if len(args) != 1 or options.samples is None:
42             parser.print_help()
43             raise SystemExit, 1
44
45         sample_rate = int(options.sample_rate)
46         ampl = 0.1
47
48         src0 = gr.sig_source_f (sample_rate, gr.GR_SIN_WAVE, 350, ampl)
49         src1 = gr.sig_source_f (sample_rate, gr.GR_SIN_WAVE, 440, ampl)
50         head0 = gr.head(gr.sizeof_float, int(options.samples))
51         head1 = gr.head(gr.sizeof_float, int(options.samples))
52         dst = gr.wavfile_sink(args[0], 2, int(options.sample_rate), 16)
53
54         self.connect(src0, head0, (dst, 0))
55         self.connect(src1, head1, (dst, 1))
56
57 if __name__ == '__main__':
58     try:
59         my_top_block().run()
60     except KeyboardInterrupt:
61         pass