From 2d8945dc1c5a799476c10bf54d7eb1bca3a0fd36 Mon Sep 17 00:00:00 2001 From: Dmitry Butyugin Date: Sun, 14 Mar 2021 14:58:29 +0100 Subject: graph_accelerometer: Support axis-level comparison of adxl345 data Signed-off-by: Dmitry Butyugin --- klippy/extras/shaper_calibrate.py | 4 ++++ 1 file changed, 4 insertions(+) (limited to 'klippy/extras/shaper_calibrate.py') diff --git a/klippy/extras/shaper_calibrate.py b/klippy/extras/shaper_calibrate.py index 1df4c943..71e5e489 100644 --- a/klippy/extras/shaper_calibrate.py +++ b/klippy/extras/shaper_calibrate.py @@ -139,6 +139,8 @@ class CalibrationData: self.psd_y = psd_y self.psd_z = psd_z self._psd_list = [self.psd_sum, self.psd_x, self.psd_y, self.psd_z] + self._psd_map = {'x': self.psd_x, 'y': self.psd_y, 'z': self.psd_z, + 'all': self.psd_sum} self.data_sets = 1 def join(self, other): np = self.numpy @@ -159,6 +161,8 @@ class CalibrationData: psd /= self.freq_bins + .1 # Remove low-frequency noise psd[self.freq_bins < MIN_FREQ] = 0. + def get_psd(self, axis='all'): + return self._psd_map[axis] CalibrationResult = collections.namedtuple( -- cgit v1.2.3-70-g09d2