-
Notifications
You must be signed in to change notification settings - Fork 11
/
Copy pathdvbt-blade.py
executable file
·222 lines (185 loc) · 8.05 KB
/
dvbt-blade.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
#!/usr/bin/env /usr/bin/python
# Copyright 2014,2018 Ron Economos ([email protected])
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
from gnuradio import blocks
from gnuradio import digital
from gnuradio import fft
from gnuradio import gr
from gnuradio.fft import window
from gnuradio import dtv
import osmosdr
import sys
import argparse
def main(args):
d = 'Transmit a DVB-T signal using the bladeRF and gr-dtv'
parser = argparse.ArgumentParser()
mode_help = '# of carriers. Options: t2k, t8k (default).'
parser.add_argument('-m', '--mode', dest='mode', default='t8k',
type=str, help=mode_help)
ch_help = 'channel width in MHz. Options: 5, 6, 7, 8 (default).'
parser.add_argument('-c', '--channel', dest='channel', default=8,
metavar='CH', type=int, help=ch_help)
const_help = 'constellation. qpsk, qam16, qam64 (default).'
parser.add_argument('-C', '--cons', dest='cons', default='qam64',
metavar='TYPE', type=str, help=const_help)
rate_help = 'Viterbi rate. 1/2, 2/3, 3/4, 5/6, 7/8 (default).'
parser.add_argument('-r', '--rate', dest='rate', default='7/8',
type=str, help=rate_help)
guard_help = 'guard interval. 1/32 (default), 1/16, 1/8, 1/4.'
parser.add_argument('-g', '--guard', dest='interval', default='1/32',
metavar='D', type=str, help=guard_help)
freq_help = 'center frequency (Hz). Default is 429e6.'
parser.add_argument('-f', '--freq', dest='freq', default=429e6,
type=float, help=freq_help)
vga1_help = 'bladeRF TXVGA1 gain. Default is -6.'
parser.add_argument('--txvga1', dest='txvga1', default=-6,
metavar='gain', type=int, help=vga1_help)
vga2_help = 'bladeRF TXVGA2 gain. Default is 9.'
parser.add_argument('--txvga2', dest='txvga2', default=9,
metavar='gain', type=int, help=vga2_help)
outfile_help = 'write to specified file.'
parser.add_argument('-o', '--output', dest='outfile', default=None,
metavar='OUT', type=str, help=outfile_help)
parser.add_argument('infile', metavar='input-file', type=str,
help='Input file')
args = parser.parse_args()
print args
if args.mode.lower() == 't2k':
mode = dtv.T2k
elif args.mode.lower() == 't8k':
mode = dtv.T8k
else:
sys.stderr.write('Invalid mode provided: ' + args.mode + '\n')
sys.exit(1)
if args.channel < 5 or args.channel > 8:
sys.stderr.write('Invalid channel: ' + str(args.channel) + '\n')
sys.exit(1)
else:
channel_mhz = args.channel
if args.cons.lower() == 'qpsk':
constellation = dtv.MOD_QPSK
elif args.cons.lower() == 'qam16':
constellation = dtv.MOD_16QAM
elif args.cons.lower() == 'qam64':
constellation = dtv.MOD_64QAM
else:
sys.stderr.write('Invalid constellation type: ' + args.cons + '\n')
sys.exit(1)
if args.rate == '1/2':
code_rate = dtv.C1_2
elif args.rate == '2/3':
code_rate = dtv.C2_3
elif args.rate == '3/4':
code_rate = dtv.C3_4
elif args.rate == '5/6':
code_rate = dtv.C5_6
elif args.rate == '7/8':
code_rate = dtv.C7_8
else:
sys.stderr.write('Invalid Viterbi rate: ' + args.rate + '\n')
sys.exit(1)
if args.interval == '1/32':
guard_interval = dtv.GI_1_32
elif args.interval == '1/16':
guard_interval = dtv.GI_1_16
elif args.interval == '1/8':
guard_interval = dtv.GI_1_8
elif args.interval == '1/4':
guard_interval = dtv.GI_1_4
else:
sys.stderr.write('Invalid guard interval: ' + args.interval + '\n')
sys.exit(1)
if args.freq < 300e6 or args.freq > 3.8e9:
sys.stderr.write('Invalid center frequency: ' + str(args.freq) + '\n')
sys.exit(1)
else:
center_freq = int(args.freq)
if args.txvga1 < -35 or args.txvga1 > -4:
sys.stderr.write('Invalid bladeRF TXVGA1 gain: ' +
str(args.txvga1) + '\n')
sys.exit(1)
else:
txvga1_gain = args.txvga1
if args.txvga2 < 0 or args.txvga2 > 25:
sys.stderr.write('Invalid bladeRF TXVGA2 gain: ' +
str(args.txvga2) + '\n')
sys.exit(1)
else:
txvga2_gain = args.txvga2
infile = args.infile
outfile = args.outfile
symbol_rate = channel_mhz * 8000000.0 / 7
if mode == dtv.T2k:
factor = 1
carriers = 2048
elif mode == dtv.T8k:
factor = 4
carriers = 8192
if guard_interval == dtv.GI_1_32:
gi = carriers / 32
elif guard_interval == dtv.GI_1_16:
gi = carriers / 16
elif guard_interval == dtv.GI_1_8:
gi = carriers / 8
elif guard_interval == dtv.GI_1_4:
gi = carriers / 4
if channel_mhz == 8:
bandwidth = 8750000
elif channel_mhz == 7:
bandwidth = 7000000
elif channel_mhz == 6:
bandwidth = 6000000
elif channel_mhz == 5:
bandwidth = 5000000
else:
bandwidth = 8750000
tb = gr.top_block()
src = blocks.file_source(gr.sizeof_char, infile, True)
dtv_dvbt_energy_dispersal = dtv.dvbt_energy_dispersal(1 * factor)
dtv_dvbt_reed_solomon_enc = dtv.dvbt_reed_solomon_enc(2, 8, 0x11d, 255, 239, 8, 51, (8 * factor))
dtv_dvbt_convolutional_interleaver = dtv.dvbt_convolutional_interleaver((136 * factor), 12, 17)
dtv_dvbt_inner_coder = dtv.dvbt_inner_coder(1, (1512 * factor), constellation, dtv.NH, code_rate)
dtv_dvbt_bit_inner_interleaver = dtv.dvbt_bit_inner_interleaver((1512 * factor), constellation, dtv.NH, mode)
dtv_dvbt_symbol_inner_interleaver = dtv.dvbt_symbol_inner_interleaver((1512 * factor), mode, 1)
dtv_dvbt_map = dtv.dvbt_map((1512 * factor), constellation, dtv.NH, mode, 1)
dtv_dvbt_reference_signals = dtv.dvbt_reference_signals(gr.sizeof_gr_complex, (1512 * factor), carriers, constellation, dtv.NH, code_rate, code_rate, guard_interval, mode, 0, 0)
fft_vxx = fft.fft_vcc(carriers, False, (window.rectangular(carriers)), True, 10)
digital_ofdm_cyclic_prefixer = digital.ofdm_cyclic_prefixer(carriers, carriers+(gi), 0, "")
blocks_multiply_const_vxx = blocks.multiply_const_vcc((0.0022097087, ))
out = osmosdr.sink(args="bladerf=0,buffers=128,buflen=32768")
out.set_sample_rate(symbol_rate)
out.set_center_freq(center_freq, 0)
out.set_freq_corr(0, 0)
out.set_gain(txvga2_gain, 0)
out.set_bb_gain(txvga1_gain, 0)
out.set_bandwidth(bandwidth, 0)
tb.connect(src, dtv_dvbt_energy_dispersal)
tb.connect(dtv_dvbt_energy_dispersal, dtv_dvbt_reed_solomon_enc)
tb.connect(dtv_dvbt_reed_solomon_enc, dtv_dvbt_convolutional_interleaver)
tb.connect(dtv_dvbt_convolutional_interleaver, dtv_dvbt_inner_coder)
tb.connect(dtv_dvbt_inner_coder, dtv_dvbt_bit_inner_interleaver)
tb.connect(dtv_dvbt_bit_inner_interleaver, dtv_dvbt_symbol_inner_interleaver)
tb.connect(dtv_dvbt_symbol_inner_interleaver, dtv_dvbt_map)
tb.connect(dtv_dvbt_map, dtv_dvbt_reference_signals)
tb.connect(dtv_dvbt_reference_signals, fft_vxx)
tb.connect(fft_vxx, digital_ofdm_cyclic_prefixer)
tb.connect(digital_ofdm_cyclic_prefixer, blocks_multiply_const_vxx)
tb.connect(blocks_multiply_const_vxx, out)
if outfile:
dst = blocks.file_sink(gr.sizeof_gr_complex, outfile)
tb.connect(blocks_multiply_const_vxx, dst)
tb.run()
if __name__ == '__main__':
main(sys.argv[1:])