First commit

files/improved-shapers/calibrate_shaper_config.py

@@ -0,0 +1,39 @@
+class CalibrateShaperConfig:
+    def __init__(self, config):
+        self.printer = config.get_printer();
+        
+        shaper_type = config.get('shaper_type', 'mzv')
+        self.shaper_type_x = config.get('shaper_type_x' , shaper_type)
+        self.shaper_freq_x = config.getfloat('shaper_freq_x', 0., minval=0.)
+
+        self.shaper_type_y = config.get('shaper_type_y' , shaper_type)
+        self.shaper_freq_y = config.getfloat('shaper_freq_y', 0., minval=0.)
+
+        # Register commands
+        gcode = config.get_printer().lookup_object('gcode')
+        gcode.register_command("SAVE_INPUT_SHAPER", self.cmd_save_input_shaper)
+
+    def get_status(self, eventtime):
+        return {}
+
+    def cmd_save_input_shaper(self, gcmd):
+        self.shaper_freq_x = gcmd.get_float('SHAPER_FREQ_X',
+                                            self.shaper_freq_x, minval=0.)
+        self.shaper_type_x = gcmd.get('SHAPER_TYPE_X', self.shaper_type_x)
+
+        self.shaper_freq_y = gcmd.get_float('SHAPER_FREQ_Y',
+                                            self.shaper_freq_y, minval=0.)
+        self.shaper_type_y = gcmd.get('SHAPER_TYPE_Y', self.shaper_type_y)
+
+        configfile = self.printer.lookup_object('configfile')
+
+        configfile.set('input_shaper', 'shaper_type_x', self.shaper_type_x)
+        configfile.set('input_shaper', 'shaper_freq_x',
+                       '%.1f' % (self.shaper_freq_x,))
+
+        configfile.set('input_shaper', 'shaper_type_y', self.shaper_type_y)
+        configfile.set('input_shaper', 'shaper_freq_y',
+                       '%.1f' % (self.shaper_freq_y,))
+
+def load_config(config):
+    return CalibrateShaperConfig(config)