import re
import math
import serial
# import pyserial
import serial.tools.list_ports as sertools
import time
import struct

##########################################################################################
# Function to parse raw measurement values into a structure format
def parseMesaurementData(serial_data):
    # Define the format for one DataPacket (36 bytes)
    format_string = '<IfIfffIff'
    packet_size = struct.calcsize(format_string)

    # Replace this with your actual binary data from serial
    binary_data = serial_data  # e.g., ser.read(expected_length)

    # Parse all packets
    packets = []
    for i in range(0, len(binary_data), packet_size):
        chunk = binary_data[i:i+packet_size]
        if len(chunk) == packet_size:
            unpacked = struct.unpack(format_string, chunk)
            packet = {
                'sensorNum': unpacked[0],
                'impFreq': unpacked[1],
                'impDataPoints': unpacked[2],
                'impRzMag': unpacked[3],
                'impRzPhase': unpacked[4],
                'rtdFreq': unpacked[5],
                'rtdDataPoints': unpacked[6],
                'rtdRzMag': unpacked[7],
                'rtdRzPhase': unpacked[8]
            }
            packets.append(packet)

    # Print parsed packets
    for idx, packet in enumerate(packets):
        print(f"Packet {idx+1}: {packet}")

    print('\n')

##########################################################################################
def countMeasurementPackets(serial_data):
    # Define the format for one DataPacket (36 bytes)
    format_string = '<IfIfffIff'
    packet_size = struct.calcsize(format_string)

    # Calculate how many complete packets are in the data
    num_packets = len(serial_data) // packet_size

    print(f"Number of packets received: {num_packets}")

##########################################################################################
# Function to parse raw measurement settings into a structure format
def parseMeasurementSettings(binary_data):
    # Define the format string: 3 floats + 4 unsigned 32-bit integers
    format_string = '<fffIIII'  # Little-endian: <

    # Check if the binary data is the correct size
    expected_size = struct.calcsize(format_string)
    if len(binary_data) != expected_size:
        print(f"Invalid data length: expected {expected_size}, got {len(binary_data)}")
        return

    # Unpack the binary data
    unpacked = struct.unpack(format_string, binary_data)

    # Map the unpacked values to field names
    settings = {
        'SinFreq': unpacked[0],
        'DacVoltPP': unpacked[1],
        'BiasVolt': unpacked[2],
        'HstiaRtiaSel': unpacked[3],
        'AdcPgaGain': unpacked[4],
        'DftNum': unpacked[5],
        'ADCAvgNum': unpacked[6]
    }

    print(f"\nMeasurement Settings:")
    # Print the parsed settings
    for key, value in settings.items():
        print(f"{key}: {value}")

    print('\n')

##########################################################################################
# Function to parse EEPROM data
def parseEEPROMdata(binary_data):
    max_doubles = 8
    max_floats = 16
    double_size = 8
    float_size = 4

    # Calculate how many values we can actually unpack
    available_doubles = min(len(binary_data) // double_size, max_doubles)
    remaining_bytes = len(binary_data) - (available_doubles * double_size)
    available_floats = min(remaining_bytes // float_size, max_floats)

    # Unpack doubles
    double_format = f'<{available_doubles}d'
    double_values = struct.unpack(double_format, binary_data[:available_doubles * double_size])

    # Unpack floats
    float_format = f'<{available_floats}f'
    float_start = available_doubles * double_size
    float_values = struct.unpack(float_format, binary_data[float_start:float_start + available_floats * float_size])

    # Print results
    print("EEPROM Double Values:")
    for i, val in enumerate(double_values):
        print(f"  doubleValue[{i}]: {val}")

    print("EEPROM Float Values:")
    for i, val in enumerate(float_values):
        print(f"  floatValue[{i}]: {val}")

    print('\n')

##########################################################################################