gr-noaa: Switched to 'double rate BPSK' HRPT synchronization
[debian/gnuradio] / gr-noaa / apps / demod_hrpt_file.py
1 #!/usr/bin/env python
2 ##################################################
3 # Gnuradio Python Flow Graph
4 # Title: Demod Hrpt File
5 # Generated: Sun Nov  1 18:39:07 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 demod_hrpt_file(gr.top_block):
17
18         def __init__(self, decim=16, pll_alpha=0.05, clock_alpha=0.05, sync_alpha=0.05, input_filename="usrp.dat", output_filename="frames.dat"):
19                 gr.top_block.__init__(self, "Demod Hrpt File")
20
21                 ##################################################
22                 # Parameters
23                 ##################################################
24                 self.decim = decim
25                 self.pll_alpha = pll_alpha
26                 self.clock_alpha = clock_alpha
27                 self.sync_alpha = sync_alpha
28                 self.input_filename = input_filename
29                 self.output_filename = output_filename
30
31                 ##################################################
32                 # Variables
33                 ##################################################
34                 self.sym_rate = sym_rate = 600*1109
35                 self.sample_rate = sample_rate = 64e6/decim
36                 self.sps = sps = sample_rate/sym_rate
37                 self.max_clock_offset = max_clock_offset = 0.1
38                 self.max_carrier_offset = max_carrier_offset = 2*math.pi*100e3/sample_rate
39                 self.hs = hs = int(sps/2.0)
40
41                 ##################################################
42                 # Blocks
43                 ##################################################
44                 self.agc = gr.agc_cc(1e-5, 1.0, 1.0/32768.0, 1.0)
45                 self.cs2cf = gr.interleaved_short_to_complex()
46                 self.file_source = gr.file_source(gr.sizeof_short*1, input_filename, False)
47                 self.gr_binary_slicer_fb_0 = gr.binary_slicer_fb()
48                 self.gr_clock_recovery_mm_xx_0 = gr.clock_recovery_mm_ff(sps/2.0, clock_alpha**2/4.0, 0.5, clock_alpha, max_clock_offset)
49                 self.gr_file_sink_0 = gr.file_sink(gr.sizeof_short*1, output_filename)
50                 self.gr_moving_average_xx_0 = gr.moving_average_ff(hs, 1.0/hs, 4000)
51                 self.noaa_hrpt_bit_sync_0 = noaa.hrpt_bit_sync()
52                 self.noaa_hrpt_decoder_0 = noaa.hrpt_decoder()
53                 self.noaa_hrpt_deframer_0 = noaa.hrpt_deframer()
54                 self.pll = noaa.hrpt_pll_cf(pll_alpha, pll_alpha**2/4.0, max_carrier_offset)
55
56                 ##################################################
57                 # Connections
58                 ##################################################
59                 self.connect((self.file_source, 0), (self.cs2cf, 0))
60                 self.connect((self.gr_clock_recovery_mm_xx_0, 0), (self.gr_binary_slicer_fb_0, 0))
61                 self.connect((self.gr_moving_average_xx_0, 0), (self.gr_clock_recovery_mm_xx_0, 0))
62                 self.connect((self.pll, 0), (self.gr_moving_average_xx_0, 0))
63                 self.connect((self.gr_binary_slicer_fb_0, 0), (self.noaa_hrpt_bit_sync_0, 0))
64                 self.connect((self.cs2cf, 0), (self.agc, 0))
65                 self.connect((self.agc, 0), (self.pll, 0))
66                 self.connect((self.noaa_hrpt_deframer_0, 0), (self.gr_file_sink_0, 0))
67                 self.connect((self.noaa_hrpt_deframer_0, 0), (self.noaa_hrpt_decoder_0, 0))
68                 self.connect((self.noaa_hrpt_bit_sync_0, 0), (self.noaa_hrpt_deframer_0, 0))
69
70         def set_decim(self, decim):
71                 self.decim = decim
72                 self.set_sample_rate(64e6/self.decim)
73
74         def set_pll_alpha(self, pll_alpha):
75                 self.pll_alpha = pll_alpha
76                 self.pll.set_alpha(self.pll_alpha)
77                 self.pll.set_beta(self.pll_alpha**2/4.0)
78
79         def set_clock_alpha(self, clock_alpha):
80                 self.clock_alpha = clock_alpha
81                 self.gr_clock_recovery_mm_xx_0.set_gain_omega(self.clock_alpha**2/4.0)
82                 self.gr_clock_recovery_mm_xx_0.set_gain_mu(self.clock_alpha)
83
84         def set_sync_alpha(self, sync_alpha):
85                 self.sync_alpha = sync_alpha
86
87         def set_input_filename(self, input_filename):
88                 self.input_filename = input_filename
89
90         def set_output_filename(self, output_filename):
91                 self.output_filename = output_filename
92
93         def set_sym_rate(self, sym_rate):
94                 self.sym_rate = sym_rate
95                 self.set_sps(self.sample_rate/self.sym_rate)
96
97         def set_sample_rate(self, sample_rate):
98                 self.sample_rate = sample_rate
99                 self.set_max_carrier_offset(2*math.pi*100e3/self.sample_rate)
100                 self.set_sps(self.sample_rate/self.sym_rate)
101
102         def set_sps(self, sps):
103                 self.sps = sps
104                 self.set_hs(int(self.sps/2.0))
105                 self.gr_clock_recovery_mm_xx_0.set_omega(self.sps/2.0)
106
107         def set_max_clock_offset(self, max_clock_offset):
108                 self.max_clock_offset = max_clock_offset
109
110         def set_max_carrier_offset(self, max_carrier_offset):
111                 self.max_carrier_offset = max_carrier_offset
112                 self.pll.set_max_offset(self.max_carrier_offset)
113
114         def set_hs(self, hs):
115                 self.hs = hs
116                 self.gr_moving_average_xx_0.set_length_and_scale(self.hs, 1.0/self.hs)
117
118 if __name__ == '__main__':
119         parser = OptionParser(option_class=eng_option, usage="%prog: [options]")
120         parser.add_option("-d", "--decim", dest="decim", type="intx", default=16,
121                 help="Set decim [default=%default]")
122         parser.add_option("-p", "--pll-alpha", dest="pll_alpha", type="eng_float", default=eng_notation.num_to_str(0.05),
123                 help="Set pll_alpha [default=%default]")
124         parser.add_option("-a", "--clock-alpha", dest="clock_alpha", type="eng_float", default=eng_notation.num_to_str(0.05),
125                 help="Set clock_alpha [default=%default]")
126         parser.add_option("-s", "--sync-alpha", dest="sync_alpha", type="eng_float", default=eng_notation.num_to_str(0.05),
127                 help="Set sync_alpha [default=%default]")
128         parser.add_option("-F", "--input-filename", dest="input_filename", type="string", default="usrp.dat",
129                 help="Set usrp.dat [default=%default]")
130         parser.add_option("-o", "--output-filename", dest="output_filename", type="string", default="frames.dat",
131                 help="Set frames.dat [default=%default]")
132         (options, args) = parser.parse_args()
133         tb = demod_hrpt_file(decim=options.decim, pll_alpha=options.pll_alpha, clock_alpha=options.clock_alpha, sync_alpha=options.sync_alpha, input_filename=options.input_filename, output_filename=options.output_filename)
134         tb.run()
135