[DRAFT] Add Rigol instrument

rtestbench/core.py

@@ -470,6 +470,7 @@ def unlock(self):
 ###########
 
 from rtestbench.tools.keysight._factory import get_keysight_tool
+from rtestbench.tools.rigol._factory import get_rigol_tool
 
 
 class ToolFactory(object):
@@ -560,6 +561,11 @@ def _build_specific_tool(self, tool_info) -> Tool:
                 return get_keysight_tool(tool_info)
             except ValueError:
                 raise
+        elif tool_info.manufacturer == "Rigol Technologies":
+            try:
+                return get_rigol_tool(tool_info)
+            except ValueError:
+                raise
         else:
             raise NotImplementedError("The manufacturer {} has not been implemented yet.".format(tool_info.manufacturer))
 

rtestbench/tools/rigol/__init__.py

@@ -0,0 +1,2 @@
+
+

rtestbench/tools/rigol/_factory.py

@@ -0,0 +1,25 @@
+"""Factory functions for all rigol instruments."""
+
+
+from rtestbench.tools.rigol.oscilloscope import ds1102
+
+
+def get_rigol_tool(tool_info):
+    """Returns the dedicated rigol Tool based on the model.
+
+    Raises:
+        ValueError: The model is not known from the system.
+    """
+
+    try:
+        return get_rigol_oscilloscope(tool_info)
+    except ValueError: # When other families are added, update this part of the code.
+        raise ValueError("Unknown rigol family of instruments/tools.")
+
+
+def get_rigol_oscilloscope(tool_info):
+    # DS1102x series
+    if tool_info.model == "DS1102E":
+        return ds1102.DS1102E(tool_info)
+    else:
+        raise ValueError("Unknown rigol electrometer model")

rtestbench/tools/rigol/oscilloscope/__init__.py

@@ -0,0 +1,7 @@
+
+# __all__ = [
+#     'ds1102e',
+# ]
+
+
+# from . import _interface_v1