import serial
import time
import csv

def get_serial_port(serial_port, baud_rate, timeout=1):
    ser = serial.Serial(serial_port, baud_rate, timeout=timeout)
    time.sleep(2)
    return ser

def write_to_csv(serial_conn, csv_file_path):
    with open(csv_file_path, mode='w', newline='') as file:
        writer = csv.writer(file)
        writer.writerow(["timestamp", "raw", "voltage"])

        while True:
            line = serial_conn.readline().decode(errors='ignore').strip()
            if line.count(',') == 2:
                timestamp, raw, voltage = line.split(",")
                writer.writerow([timestamp.strip(), raw.strip(), voltage.strip()])
                print(timestamp, raw, voltage)

def main():
    serial_port = "COM5"
    baud_rate = 115200
    csv_file = "eeg_data.csv"
    
    ser = get_serial_port(serial_port, baud_rate)
    write_to_csv(ser, csv_file)

if __name__ == "__main__":
    main()