summaryrefslogtreecommitdiff
path: root/gr-atsc/src/lib/gen_encoder.py
blob: 6c8fabf7ada77e5a3579a8c2e6936570b96cbc26 (plain)
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
#!/usr/bin/python
#
# Copyright 2002 Free Software Foundation, Inc.
#
# This file is part of GNU Radio
#
# GNU Radio 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, or (at your option)
# any later version.
#
# GNU Radio 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 GNU Radio; see the file COPYING.  If not, write to
# the Free Software Foundation, Inc., 51 Franklin Street,
# Boston, MA 02110-1301, USA.
#

def output(input, state):
    x2 = (input >> 1) & 0x1
    x1 = (input >> 0) & 0x1
    s = (state >> 2) & 0x1
    t = (state >> 1) & 0x1
    u = (state >> 0) & 0x1

    z0 = u
    z1 = x1
    z2 = x2 ^ s
    return (z2 << 2) | (z1 << 1) | z0


def next_state(input, state):
    x2 = (input >> 1) & 0x1
    x1 = (input >> 0) & 0x1
    s0 = (state >> 2) & 0x1
    t0 = (state >> 1) & 0x1
    u0 = (state >> 0) & 0x1

    s1 = x2 ^ s0
    t1 = u0
    u1 = t0 ^ x1

    return (s1 << 2) | (t1 << 1) | u1

print "@@@ NEXT @@@"

for i in range (32):
    state = (i >> 2) & 0x7
    input = i & 0x3
    print next_state (input, state)


print "@@@ OUTPUT @@@"

for i in range (32):
    state = (i >> 2) & 0x7
    input = i & 0x3
    print output (input, state)