added Time Base Mode query and write

1 files changed, 20 insertions, 0 deletions

diff --git a/qolab/hardware/scope/rigolds1054z.py b/qolab/hardware/scope/rigolds1054z.py
index ced67d2..b41086c 100644
--- a/qolab/hardware/scope/rigolds1054z.py
+++ b/qolab/hardware/scope/rigolds1054z.py
@@ -92,6 +92,26 @@ class RigolDS1054z(ScopeSCPI):
         return res
 
     @BasicInstrument.tsdb_append
+    def getTimeBaseMode(self):
+        """Get he mode of the horizontal timebase.
+
+        We expect {MAIN|XY|ROLL}.
+        MAIN stands for normal voltage vs time mode.
+        XY stands X vs Y mode.
+        ROLL stands for rolling mode.
+        """
+        res = self.query(":TIMebase:MODE?")
+        return res
+
+    @BasicInstrument.tsdb_append
+    def setTimeBaseMode(self, val):
+        """Set trigger mode.
+
+        Takes {MAIN|XY|ROLL}.
+        """
+        self.write(f":TIMebase:MODE? {val}")
+
+    @BasicInstrument.tsdb_append
     def getTriggerMode(self):
         """Get trigger mode.