Simple code cleanup.
[debian/gnuradio] / gr-noaa / apps / hrpt_decode.py
1 #!/usr/bin/env python
2 ##################################################
3 # Gnuradio Python Flow Graph
4 # Title: Hrpt Decode
5 # Generated: Sun Nov  8 10:49:01 2009
6 ##################################################
7
8 from gnuradio import eng_notation
9 from gnuradio import gr
10 from gnuradio import noaa
11 from gnuradio.eng_option import eng_option
12 from gnuradio.gr import firdes
13 from optparse import OptionParser
14 import math, os
15
16 class hrpt_decode(gr.top_block):
17
18         def __init__(self, decim=32, pll_alpha=0.01, clock_alpha=0.01, input_filename="frames.hrpt"):
19                 gr.top_block.__init__(self, "Hrpt Decode")
20
21                 ##################################################
22                 # Parameters
23                 ##################################################
24                 self.decim = decim
25                 self.pll_alpha = pll_alpha
26                 self.clock_alpha = clock_alpha
27                 self.input_filename = input_filename
28
29                 ##################################################
30                 # Variables
31                 ##################################################
32                 self.sym_rate = sym_rate = 600*1109
33                 self.sample_rate = sample_rate = 64e6/decim
34                 self.sps = sps = sample_rate/sym_rate
35                 self.max_clock_offset = max_clock_offset = 100e-6
36                 self.max_carrier_offset = max_carrier_offset = 2*math.pi*100e3/sample_rate
37                 self.hs = hs = int(sps/2.0)
38
39                 ##################################################
40                 # Blocks
41                 ##################################################
42                 self.decoder = noaa.hrpt_decoder(True,True)
43                 self.file_source = gr.file_source(gr.sizeof_short*1, input_filename, False)
44
45                 ##################################################
46                 # Connections
47                 ##################################################
48                 self.connect((self.file_source, 0), (self.decoder, 0))
49
50         def set_decim(self, decim):
51                 self.decim = decim
52                 self.set_sample_rate(64e6/self.decim)
53
54         def set_pll_alpha(self, pll_alpha):
55                 self.pll_alpha = pll_alpha
56
57         def set_clock_alpha(self, clock_alpha):
58                 self.clock_alpha = clock_alpha
59
60         def set_input_filename(self, input_filename):
61                 self.input_filename = input_filename
62
63         def set_sym_rate(self, sym_rate):
64                 self.sym_rate = sym_rate
65                 self.set_sps(self.sample_rate/self.sym_rate)
66
67         def set_sample_rate(self, sample_rate):
68                 self.sample_rate = sample_rate
69                 self.set_max_carrier_offset(2*math.pi*100e3/self.sample_rate)
70                 self.set_sps(self.sample_rate/self.sym_rate)
71
72         def set_sps(self, sps):
73                 self.sps = sps
74                 self.set_hs(int(self.sps/2.0))
75
76         def set_max_clock_offset(self, max_clock_offset):
77                 self.max_clock_offset = max_clock_offset
78
79         def set_max_carrier_offset(self, max_carrier_offset):
80                 self.max_carrier_offset = max_carrier_offset
81
82         def set_hs(self, hs):
83                 self.hs = hs
84
85 if __name__ == '__main__':
86         parser = OptionParser(option_class=eng_option, usage="%prog: [options]")
87         parser.add_option("-d", "--decim", dest="decim", type="intx", default=32,
88                 help="Set decim [default=%default]")
89         parser.add_option("-p", "--pll-alpha", dest="pll_alpha", type="eng_float", default=eng_notation.num_to_str(0.01),
90                 help="Set pll_alpha [default=%default]")
91         parser.add_option("-s", "--clock-alpha", dest="clock_alpha", type="eng_float", default=eng_notation.num_to_str(0.01),
92                 help="Set clock_alpha [default=%default]")
93         parser.add_option("-F", "--input-filename", dest="input_filename", type="string", default="frames.hrpt",
94                 help="Set frames.hrpt [default=%default]")
95         (options, args) = parser.parse_args()
96         tb = hrpt_decode(decim=options.decim, pll_alpha=options.pll_alpha, clock_alpha=options.clock_alpha, input_filename=options.input_filename)
97         tb.run()
98