from math import fabs as fabs
from math import sqrt as sqrt
from scipy.special import erfc as erfc


class StartUPTest:

    @staticmethod
    def monobit_test(binary_data: str):

        length_of_bit_string = len(binary_data)

        # Variable for S(n)
        count = 0
        # Iterate each bit in the string and compute for S(n)
        for bit in binary_data:
            if bit == 48:
                # If bit is 0, then -1 from the S(n)
                count -= 1
            elif bit == 49:
                # If bit is 1, then +1 to the S(n)
                count += 1

        # Compute the test statistic
        sObs = count / sqrt(length_of_bit_string)

        # Compute p-Value
        p_value = erfc(fabs(sObs) / sqrt(2))

        # return a p_value and randomness result
        return (p_value, (p_value >= 0.01))

    @staticmethod
    def test5(binary_data: str):

        ShiftFeld = [0] * 5000
        MaxKorrFeld = [0] * 5000

        # Fill BitFeldB with data

        for tau in range(1, 5001):
            Z_tau = 0
            for i in range(5000):
                Z_tau += binary_data[i] ^ binary_data[i + tau]
            ShiftFeld[tau - 1] = Z_tau

        #Debugging
        #for i in range(5000):
        #    print(ShiftFeld[i], end=' ')

        # Find the index of the maximum deviation from 2500
        max_deviation = 0
        for tau in range(5000):
            deviation = abs(ShiftFeld[tau] - 2500)
            if deviation > max_deviation:
                max_deviation = deviation

        # Find all indices with the maximum deviation
        j = 0
        for tau in range(5000):
            deviation = abs(ShiftFeld[tau] - 2500)
            if deviation == max_deviation:
                MaxKorrFeld[j] = tau
                j += 1

        print("Maximale Z_tau-Abweichung von 2500:", max_deviation)
        print("Aufgetreten für Shifts:")
        for k in range(j):
            print("Shift:", MaxKorrFeld[k] + 1)

        tau = MaxKorrFeld[0]
        Z_tau = 0
        for i in range(10000, 15000):
            Z_tau += StartUPTest.charToInt(i, binary_data) ^ StartUPTest.charToInt(i + tau + 1, binary_data)
        tau += 1

        ok = Z_tau > 2326 and Z_tau < 2674
        return (Z_tau, ok)

    @staticmethod
    def charToInt(index, binary_data: str):
        value = 0
        if binary_data[index] == 49:
            value = 1
        else:
            value = 0
        return value