Trackball/firmware/util/visualize_data.py

#!/usr/bin/env python

import matplotlib.pyplot as plt
import numpy as np
import sys
import math

if len(sys.argv) < 2:
    print("Usage:")
    print("    " + sys.argv[0] + " filename")
    sys.exit(0)

fig, axes = plt.subplots(6, 1, figsize=(5, 15))

data = []
lines_heading = []
with open(sys.argv[1]) as f:
    lines = f.readlines()
    lines_heading = lines[0].split(',')
    lines_data = lines[1:]
    for line in lines_data:
        nums = line.split(',')
        line_data = []
        for r in nums:
            line_data.append(int(r))
        data.append(line_data)

start_time = data[0][0]
for i in range(0, len(data)):
    data[i][0] = (data[i][0] - start_time) / 1000.0 / 1000.0

print("samples: " + str(len(data)))

x = [ r[0] for r in data ]

for n in range(3, 9):
    y = [ r[n] for r in data ]
    im = axes[n - 3].plot(x, y)
    axes[n - 3].set_title(lines_heading[n])

plt.suptitle(sys.argv[1], fontsize=18)
plt.tight_layout()
plt.show()