import os
import time
import spidev
from StartUpTest import StartUPTest
from TotOnline import TotOnline

def read_data(num_numbers, bits_per_number):
    spi = spidev.SpiDev()
    spi.open(0, 0)  # Öffnet die SPI-Kommunikation (Bus 0, Device 0)
    spi.max_speed_hz = 16000000  # Legt die maximale Taktrate fest

    # Calculate the total number of bytes required to accommodate the desired number of bits
    total_bytes = (num_numbers * bits_per_number + 7) // 8

    # Define the filename
    filename = f'{num_numbers}numbers_{bits_per_number}bits.bin'

    with open(filename, 'wb') as file:
        bytes_received = 0
        while bytes_received < total_bytes:
            # Read a byte from SPI
            data = spi.readbytes(1)
            file.write(bytes(data))
            bytes_received += 1
            time.sleep(0.0000001)

    return filename

def convert_to_binary(filename, num_numbers, bits_per_number):
    with open(filename, 'rb') as f:
        # Read the contents of the file as bytes
        content = f.read()

    # Convert the bytes to a string of binary digits
    binary_str = ''.join(format(byte, '08b') for byte in content)

    # Truncate the binary string to the desired length
    truncated_binary_str = binary_str[:num_numbers * bits_per_number]

    # Write the truncated binary string back to the file with separators
    binary_filename = f'{num_numbers}numbers_{bits_per_number}bits_binary.txt'
    with open(binary_filename, 'w') as f:
        for i in range(0, len(truncated_binary_str), bits_per_number):
            binary_number = truncated_binary_str[i:i+bits_per_number]
            f.write(binary_number)

    return binary_filename


def process_data_in_chunks(data, chunk_size=1000000):
    chunks = [data[i:i+chunk_size] for i in range(0, len(data), chunk_size)]
    return chunks

def perform_startup_tests(filename):
    with open(filename, 'rb') as f:
        # Read the contents of the file as bytes
        content = f.read()

        print("Run all StartUPTests")
        binary_chunks = process_data_in_chunks(content)
        first_chunk = binary_chunks[0]
        result = StartUPTest.run_all_tests(first_chunk)
        print(result)
        return result
        
def perform_tot_online_tests(filename):
    with open(filename, 'rb') as f:
        # Read the contents of the file as bytes
        content = f.read()

        print("Run all TotOnline tests")
        binary_chunks = process_data_in_chunks(content)
        first_chunk = binary_chunks[0]
        result = TotOnline.run_all_tests(first_chunk)
        print(result)
        return result

def analyze_data(num_numbers, bits_per_number, startup=False):

    if startup:
        filename = read_data(10, 80000)
        binary_filename = convert_to_binary(filename, 10, 80000)
        perform_startup_tests(binary_filename)
        perform_tot_online_tests(binary_filename)
        os.unlink(filename)
        os.unlink(binary_filename)

    else:
        filename = read_data(num_numbers, bits_per_number)
        binary_filename = convert_to_binary(filename, num_numbers, bits_per_number)
        hex_filename = convert_to_hex(binary_filename, num_numbers, bits_per_number)
        result = perform_tot_online_tests(binary_filename)
        with open(hex_filename, 'rb') as f:
            # Read the contents of the file as bytes
            hex_return = f.read()
            print(hex_return)
            os.unlink(filename)
            os.unlink(hex_filename)
            os.unlink(binary_filename)
            if result:
                return hex_return
            else:
                filename = read_data(10, 80000)
                binary_filename = convert_to_binary(filename, 10, 80000)
                SUTres = perform_startup_tests(binary_filename)
                TOTres = perform_tot_online_tests(binary_filename)
                if SUTres and TOTres:
                    os.unlink(filename)
                    os.unlink(binary_filename)
                    return hex_return
                else:
                    return False

# Teste den Code
result = analyze_data(50, 2560, startup=False)
print(result)