ManchesterEncoding.py

from pathlib import Path

INPUT_FILE = "input.txt"
OUTPUT_FILE = "received_file.txt"

# Manchester rule:
# 0 -> 10
# 1 -> 01

def file_to_bits(filename):
    text = Path(filename).read_text(encoding="utf-8").rstrip("\r\n")
    data = text.encode("utf-8")

    bits = ""
    for byte in data:
        bits += format(byte, "08b")

    return bits, len(data)


def bits_to_manchester(bits):
    encoded = ""

    for bit in bits:
        if bit == "0":
            encoded += "10"
        elif bit == "1":
            encoded += "01"

    return encoded


def manchester_to_bits(encoded):
    bits = ""
    errors = 0

    if len(encoded) % 2 != 0:
        errors += 1
        encoded = encoded[:-1]

    for i in range(0, len(encoded), 2):
        pair = encoded[i:i+2]

        if pair == "10":
            bits += "0"
        elif pair == "01":
            bits += "1"
        else:
            bits += "0"
            errors += 1

    return bits, errors


def bits_to_file(bits, output_filename, original_size):
    data = bytearray()

    for i in range(0, len(bits), 8):
        byte = bits[i:i+8]

        if len(byte) == 8:
            data.append(int(byte, 2))

    data = data[:original_size]
    Path(output_filename).write_bytes(data)


def compare_files(file1, file2):
    original = Path(file1).read_text(encoding="utf-8").rstrip("\r\n").encode("utf-8")
    received = Path(file2).read_bytes()

    return original == received


def main():
    bits, original_size = file_to_bits(INPUT_FILE)

    manchester = bits_to_manchester(bits)

    decoded_bits, decode_errors = manchester_to_bits(manchester)

    bits_to_file(decoded_bits, OUTPUT_FILE, original_size)

    success = compare_files(INPUT_FILE, OUTPUT_FILE)

    Path("binary.txt").write_text(bits, encoding="utf-8")
    Path("manchester.txt").write_text(manchester, encoding="utf-8")
    Path("decoded_bits.txt").write_text(decoded_bits, encoding="utf-8")

    print("Manchester Encoding Project")
    print("---------------------------")
    print("Input file:", INPUT_FILE)
    print("Output file:", OUTPUT_FILE)
    print("Original bits:", len(bits))
    print("Manchester bits:", len(manchester))
    print("Decode errors:", decode_errors)
    print("Transmission success:", success)


if __name__ == "__main__":
    main()