3diy
/
klipper
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

scripts: Support multiple inputs in graph_accelerometer.py raw mode 

Signed-off-by: Dmitry Butyugin <dmbutyugin@google.com>
This commit is contained in:
Dmitry Butyugin 2021-09-29 22:50:17 +02:00 committed by KevinOConnor
parent ea65670239
commit 14bf9f8ba8
1 changed files with 21 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
scripts/graph_accelerometer.py
View File

@ -34,24 +34,31 @@ def parse_log(logname, opts):
# Raw accelerometer graphing # Raw accelerometer graphing
###################################################################### ######################################################################
def plot_accel(data, logname): def plot_accel(datas, lognames):
fig, axes = matplotlib.pyplot.subplots(nrows=3, sharex=True)
axes[0].set_title("\n".join(wrap(
"Accelerometer data (%s)" % (', '.join(lognames)), MAX_TITLE_LENGTH)))
axis_names = ['x', 'y', 'z']
for data, logname in zip(datas, lognames):
if isinstance(data, shaper_calibrate.CalibrationData): if isinstance(data, shaper_calibrate.CalibrationData):
raise error("Cannot plot raw accelerometer data using the processed" raise error("Cannot plot raw accelerometer data using the processed"
" resonances, raw_data input is required") " resonances, raw_data input is required")
first_time = data[0, 0] first_time = data[0, 0]
times = data[:,0] - first_time times = data[:,0] - first_time
fig, axes = matplotlib.pyplot.subplots(nrows=3, sharex=True)
axes[0].set_title("\n".join(wrap("Accelerometer data (%s)" % (logname,),
MAX_TITLE_LENGTH)))
axis_names = ['x', 'y', 'z']
for i in range(len(axis_names)): for i in range(len(axis_names)):
avg = data[:,i+1].mean() avg = data[:,i+1].mean()
adata = data[:,i+1] - data[:,i+1].mean() adata = data[:,i+1] - data[:,i+1].mean()
ax = axes[i] ax = axes[i]
ax.plot(times, adata, alpha=0.8) label = '\n'.join(wrap(logname, 60)) + ' (%+.3f mm/s^2)' % (-avg,)
ax.plot(times, adata, alpha=0.8, label=label)
axes[-1].set_xlabel('Time (s)')
fontP = matplotlib.font_manager.FontProperties()
fontP.set_size('x-small')
for i in range(len(axis_names)):
ax = axes[i]
ax.grid(True) ax.grid(True)
ax.set_ylabel('%s accel (%+.3f)\n(mm/s^2)' % (axis_names[i], -avg)) ax.legend(loc='best', prop=fontP)
axes[-1].set_xlabel('Time (%+.3f)\n(s)' % (-first_time,)) ax.set_ylabel('%s accel' % (axis_names[i],))
fig.tight_layout() fig.tight_layout()
return fig return fig
@ -246,9 +253,7 @@ def main():
# Draw graph # Draw graph
if options.raw: if options.raw:
if len(args) > 1: fig = plot_accel(datas, args)
opts.error("Only 1 input is supported in raw mode")
fig = plot_accel(datas[0], args[0])
elif options.specgram: elif options.specgram:
if len(args) > 1: if len(args) > 1:
opts.error("Only 1 input is supported in specgram mode") opts.error("Only 1 input is supported in specgram mode")