profiler.py

Go to the documentation of this file.

"""
Reads data from a serial port and saves it to a local file, then analyzes the data.
Make sure to set COM_PORT to the correct value before running.
"""
 
from collections import defaultdict
 
import numpy as np
import serial
 
COM_PORT = "COM11"
MESSAGES_TO_SAMPLE = 1000
LOCAL_FILENAME = "debug_log.txt"
 
ser = serial.Serial(
    port=COM_PORT,
    baudrate=115200,
    parity=serial.PARITY_NONE,
    stopbits=serial.STOPBITS_ONE,
    bytesize=serial.EIGHTBITS,
    timeout=0,
)
 
print("Connected to: " + ser.portstr)
 
# Handle closing the connection
def disconnect():
    print("Closing connection")
    ser.close()
 
# Read serial and write to file
f = open(LOCAL_FILENAME, "w")
print("Starting to write to file...")
 
messages_counted = 0
while messages_counted < MESSAGES_TO_SAMPLE:
    try:
        line = ser.readline().decode("utf-8").strip()
        if line == "":
            continue
        f.write(f"{line}\n")
        messages_counted += 1
 
    except Exception:
        disconnect()
        break
f.close()
 
section_current = defaultdict(int)
section_times = defaultdict(list)
with open(LOCAL_FILENAME, "r") as f:
    lines = f.readlines()[1:]
    for line in lines:
        segments = line.split()
        if len(segments) != 3:
            continue
        name, action, time = segments
        if action == "START":
            section_current[name] = int(time)
        elif action == "END":
            if section_current[name] == 0:  # Ignore if no start time
                continue
            section_times[name].append(int(time) - section_current[name])
 
# Print the average time for each section
print("=" * 55)
print("Average time per section:")
 
for name in sorted(section_times.keys()):
    # print(section_times[name])
    print(f"{name}")
    print(f" * AVG : {np.mean(section_times[name]):>6.3f} microseconds")
    print(f" * STD : {np.std(section_times[name]):>6.3f} microseconds")
    print(f" * MIN : {np.min(section_times[name]):>6.0f} microseconds")
    print(f" * MAX : {np.max(section_times[name]):>6.0f} microseconds")
    print(f" * %   : {np.percentile(section_times[name], [10, 25, 50, 75, 90])}")