ARQ-Simulation/Coder.py at main · kolbryt/ARQ-Simulation · GitHub

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
from Transmitter import Transmitter


class Coder:

    code_choice = [0, 0, 0]
    number_of_bits_in_packet = 3
    segments_x = []
    segments = []
    packets = []

    @staticmethod
    def set_code_choice(code_choice):
        if code_choice == 1:
            Coder.code_choice = [0, 0, 0]
        if code_choice == 2:
            Coder.code_choice = [0, 0, 1]
        if code_choice == 3:
            Coder.code_choice = [0, 1, 0]
        if code_choice == 4:
            Coder.code_choice = [0, 1, 1]

    @staticmethod
    def set_number_of_bits_in_packet(number_of_bits_in_packet):
        Coder.number_of_bits_in_packet = number_of_bits_in_packet

    @staticmethod
    def code_parity_bit(divided_message):

        for i in range(0, len(divided_message)):
            sum_of_bits = 0
            segment = divided_message[i]
            for j in range(0, len(segment)):
                sum_of_bits += segment[j] % 2

            if sum_of_bits % 2 == 0:
                divided_message[i].append(0)
            else:
                divided_message[i].append(1)

        packets = []
        for i in range(0, len(divided_message)):
            segment = divided_message[i]
            packet = []
            packet.append(0)
            packet.append(0)
            packet.append(0)
            for j in range(0, len(segment)):
                packet.append(segment[j])
            packets.append(packet)

        Coder.read_coded_message(packets)
        Transmitter.send_message(packets)
        Coder.packets = packets

    @staticmethod
    def code_crc_32(divided_message, polynomial_bitstring, initial_filler):
        print("code_crc_32 in progress")
        polynomial_bitstring = polynomial_bitstring.lstrip('0')
        len_input = len(divided_message)
        initial_padding = (len(polynomial_bitstring) - 1) * initial_filler
        input_padded_array = list(divided_message + initial_padding)
        while '1' in input_padded_array[:len_input]:
            cur_shift = input_padded_array.index('1')
            for i in range(len(polynomial_bitstring)):
                input_padded_array[cur_shift + i] \
                    = str(int(polynomial_bitstring[i] != input_padded_array[cur_shift + i]))
        packets = divided_message + ''.join(input_padded_array)[len_input:]

        Coder.read_coded_message(packets)
        Transmitter.send_message(packets)
        Coder.packets = packets


    @staticmethod
    def code_repetition_code(divided_message):
        packets = []
        for i in range(0, len(divided_message)):
            segment = divided_message[i]
            new_segment = [0, 1, 0]
            for j in range(0, len(segment)):
                new_segment.append(segment[j])
                new_segment.append(segment[j])
                new_segment.append(segment[j])
            packets.append(new_segment)

        Coder.read_coded_message(packets)
        Transmitter.send_message(packets)
        Coder.packets = packets

    @staticmethod
    def code_2_from_k(divided_message):

        """
                010111011

                010 - segment
                111
                011

                010 1 - packet
                111 1
                011 0

                000 010 1 - ram

                0 -> 11000
                1 -> 10100
        """

        packets = []
        for i in range(0, len(divided_message)):
            segment = divided_message[i]
            packet = []
            packet.append(0)
            packet.append(1)
            packet.append(1)
            for j in range(0, len(segment)):
                if segment[j] == 0:
                    packet.append(1)
                    packet.append(1)
                    packet.append(0)
                    packet.append(0)
                    packet.append(0)
                if segment[j] == 1:
                    packet.append(1)
                    packet.append(0)
                    packet.append(1)
                    packet.append(0)
                    packet.append(0)
            packets.append(packet)

        Coder.read_coded_message(packets)
        Transmitter.send_message(packets)
        Coder.packets = packets


    @staticmethod
    def choose_code_type(segments):
        if Coder.code_choice[0] == 0 and Coder.code_choice[1] == 0 and Coder.code_choice[2] == 0:
            Coder.code_parity_bit(segments)
        if Coder.code_choice[0] == 0 and Coder.code_choice[1] == 0 and Coder.code_choice[2] == 1:
            Coder.code_crc_32(segments, '101101010101010101010101010101010', '0')
        if Coder.code_choice[0] == 0 and Coder.code_choice[1] == 1 and Coder.code_choice[2] == 0:
            Coder.code_repetition_code(segments)
        if Coder.code_choice[0] == 0 and Coder.code_choice[1] == 1 and Coder.code_choice[2] == 1:
            Coder.code_2_from_k(segments)

    @staticmethod
    def message_divide(message):

        number_of_segments = len(message)//Coder.number_of_bits_in_packet
        segments = []
        x = 0
        for i in range(0, number_of_segments):
            segment = []
            for j in range(0, Coder.number_of_bits_in_packet):
                segment.append(message[x])
                x += 1
            segments.append(segment)

        Coder.read_divided_message(segments)
        Coder.segments = segments[:]
        Coder.segments_x = segments[:]
        Coder.choose_code_type(segments)

    @staticmethod
    def read_coded_message(coded_message):
        print("Coded message:")
        for i in range(0, len(coded_message)):
            for j in range(0, len(coded_message[i])):
                print(coded_message[i][j], end=" ")
            print()

    @staticmethod
    def read_divided_message(segments):
        print("Divided message:")
        for i in range(0, len(segments)):
            for j in range(0, len(segments[i])):
                print(segments[i][j], end=" ")
            print()