From d5a7a7f00fc9b171cdd485fb26987f4aa6eb0f6b Mon Sep 17 00:00:00 2001 From: Dmitry Butyugin Date: Fri, 22 Oct 2021 20:46:20 +0200 Subject: input_shaper: Define input shapers in a single place in Python code Signed-off-by: Dmitry Butyugin --- scripts/calibrate_shaper.py | 12 ++++++------ scripts/graph_accelerometer.py | 10 +++++----- 2 files changed, 11 insertions(+), 11 deletions(-) (limited to 'scripts') diff --git a/scripts/calibrate_shaper.py b/scripts/calibrate_shaper.py index 5bbc9eea..3d4bceff 100755 --- a/scripts/calibrate_shaper.py +++ b/scripts/calibrate_shaper.py @@ -6,12 +6,12 @@ # # This file may be distributed under the terms of the GNU GPLv3 license. from __future__ import print_function -import optparse, os, sys +import importlib, optparse, os, sys from textwrap import wrap import numpy as np, matplotlib sys.path.append(os.path.join(os.path.dirname(os.path.realpath(__file__)), - '..', 'klippy', 'extras')) -from shaper_calibrate import CalibrationData, ShaperCalibrate + '..', 'klippy')) +shaper_calibrate = importlib.import_module('.shaper_calibrate', 'extras') MAX_TITLE_LENGTH=65 @@ -25,7 +25,7 @@ def parse_log(logname): return np.loadtxt(logname, comments='#', delimiter=',') # Parse power spectral density data data = np.loadtxt(logname, skiprows=1, comments='#', delimiter=',') - calibration_data = CalibrationData( + calibration_data = shaper_calibrate.CalibrationData( freq_bins=data[:,0], psd_sum=data[:,4], psd_x=data[:,1], psd_y=data[:,2], psd_z=data[:,3]) calibration_data.set_numpy(np) @@ -41,8 +41,8 @@ def parse_log(logname): # Find the best shaper parameters def calibrate_shaper(datas, csv_output, max_smoothing): - helper = ShaperCalibrate(printer=None) - if isinstance(datas[0], CalibrationData): + helper = shaper_calibrate.ShaperCalibrate(printer=None) + if isinstance(datas[0], shaper_calibrate.CalibrationData): calibration_data = datas[0] for data in datas[1:]: calibration_data.add_data(data) diff --git a/scripts/graph_accelerometer.py b/scripts/graph_accelerometer.py index 472530d3..990bd23b 100755 --- a/scripts/graph_accelerometer.py +++ b/scripts/graph_accelerometer.py @@ -5,12 +5,12 @@ # Copyright (C) 2020 Dmitry Butyugin # # This file may be distributed under the terms of the GNU GPLv3 license. -import optparse, os, sys +import importlib, optparse, os, sys from textwrap import wrap import numpy as np, matplotlib sys.path.append(os.path.join(os.path.dirname(os.path.realpath(__file__)), - '..', 'klippy', 'extras')) -from shaper_calibrate import ShaperCalibrate + '..', 'klippy')) +shaper_calibrate = importlib.import_module('.shaper_calibrate', 'extras') MAX_TITLE_LENGTH=65 @@ -56,7 +56,7 @@ def plot_accel(data, logname): # Calculate estimated "power spectral density" def calc_freq_response(data, max_freq): - helper = ShaperCalibrate(printer=None) + helper = shaper_calibrate.ShaperCalibrate(printer=None) return helper.process_accelerometer_data(data) def calc_specgram(data, axis): @@ -155,7 +155,7 @@ def plot_specgram(data, logname, max_freq, axis): ###################################################################### def write_frequency_response(datas, output): - helper = ShaperCalibrate(printer=None) + helper = shaper_calibrate.ShaperCalibrate(printer=None) calibration_data = helper.process_accelerometer_data(datas[0]) for data in datas[1:]: calibration_data.add_data(helper.process_accelerometer_data(data)) -- cgit v1.2.3-70-g09d2