#!/usr/bin/python3
import os
import sys
import subprocess
from pathlib import Path

FILENAME = f"/tmp/rx_100kB_{str(os.getpid())}"


def program_device():
    # build new firmware with SGPIO_DEBUG mode enabled
    print("Programming device...")
    fw_dir  = os.getcwd() + "/firmware/hackrf_usb/build"
    del_dir = subprocess.run(["rm", "-rf", "firmware/hackrf_usb/build"])
    mk_dir  = subprocess.run(["mkdir", "firmware/hackrf_usb/build"])
    cmake   = subprocess.run(["cmake", "-D", "SGPIO_DEBUG=1", ".."],
                             cwd=fw_dir, stdout=subprocess.DEVNULL)
    make    = subprocess.run(["make"],
                             cwd=fw_dir, stdout=subprocess.DEVNULL)
    program = subprocess.run(["./ci-scripts/test-firmware-program.sh"],
                             stdout=subprocess.DEVNULL)


def capture():
    shortfall_count = -1
    capture_tries = 0

    while shortfall_count != 0:
        print("Capturing data...")
        rx_100kB = subprocess.run(["host/build/hackrf-tools/src/hackrf_transfer",
                                    "-r", FILENAME, "-d", "RunningFromRAM",
                                    "-n", "50000", "-s", "20000000"],
                                    capture_output=True, encoding="UTF-8")
        print(rx_100kB.stdout)
        print(rx_100kB.stderr)
        print(f"Wrote capture data to file: {FILENAME}")

        debug_state_proc = subprocess.run(["host/build/hackrf-tools/src/hackrf_debug", "--state"],
                                            capture_output=True, encoding="UTF-8")
        print(debug_state_proc.stdout)
        print(debug_state_proc.stderr)
        capture_tries += 1
        debug_state = debug_state_proc.stdout.split("\n")
        shortfalls_line = [s for s in debug_state if s.startswith("Number of shortfalls")]
        shortfall_count = [int(c) for c in shortfalls_line[0].split() if c.isdigit()][0]

        if capture_tries == 10:
            print("Unable to transmit data with 0 shortfalls. " \
            "This is not indicative of a device failure. " \
            "Likely an issue with the testing infrastructure.")
            sys.exit(1)


def check_bytes():
    print(f"Checking length of {FILENAME}")
    rx_data = Path(FILENAME).read_bytes()
    if len(rx_data) != 100000: # file should be 100k bytes when using 50k samples
        print(f"ERROR: Only {str(len(rx_data))} bytes found in file, expected 100k.")
        sys.exit(1)
    else:
        print("Correct file size found.")

    # check that each byte = prev_byte + 1 except at wraparound bounds
    print("Checking bytes...")
    for i in range(1, len(rx_data)):
        if rx_data[i-1] != rx_data[i] - 1:
            if not (rx_data[i] == 0 and rx_data[i-1] == 255):
                print(f"ERROR: Incorrect data value found at location {str(i)} in {FILENAME}:")
                # print up to 5 values starting from at most 1 value before error occurence
                j = -1
                while j < 4:
                    if i + j < len(rx_data) and i + j > -1:
                        print(f"{str(i+j)} : {str(rx_data[i+j])}")
                    j = j + 1
                sys.exit(1)
    print("Successfully validated all bytes in file.\nSGPIO debug test passed.")


def main():
    program_device()
    capture()
    check_bytes()


if __name__ == "__main__":
    main()