Add Piezo.zero().

[pypiezo.git] / pypiezo / base.py

diff --git a/pypiezo/base.py b/pypiezo/base.py
index 6f9eefc9e172e94aa703166745e61f157cf181da..25461c717675b0df8b9348dc0dd4eba06e52d9ab 100644 (file)
--- a/pypiezo/base.py
+++ b/pypiezo/base.py
@@ -912,3 +912,16 @@ class Piezo (object):
         for i,name in enumerate(input_names):
             ret[name] = input_data[:,i]
         return ret
+
+    def zero(self, axis_names=None, **kwargs):
+        zeros = []
+        if axis_names is None:
+            axis_names = [axis.name for axis in self.axes]
+        for axis_name in axis_names:
+            axis = self.axis_by_name(axis_name)
+            config = self.config.select_config(
+                'axes', axis_name, get_attribute=get_axis_name)['channel']
+            zero = convert_volts_to_bits(config, 0)
+            zeros.append(zero)
+            self.jump(axis_name, zero)
+        return zeros