Add screenshot to AgilentE8357A

DavidLutton · Apr 13, 2024 · 193a96d · 193a96d
1 parent 99cbea9
commit 193a96d
Showing 1 changed file with 16 additions and 1 deletion.
diff --git a/src/labtoolkit/NetworkAnalyser/AgilentE8357A.py b/src/labtoolkit/NetworkAnalyser/AgilentE8357A.py
@@ -2,6 +2,8 @@
 from ..SCPI import SCPI
 import numpy as np
 import pandas as pd
+import PIL.Image as Image
+import io
 
 
 class AgilentE8357A(IEEE488, SCPI):
@@ -142,7 +144,20 @@ def reference(self):
         """
         return self.query('SENSe:ROSCillator:SOURce?')
 
-
+    def screenshot(self):
+        self.write('HCOPY:FILE \'screen.png\'')
+
+        image = Image.open(io.BytesIO(
+            self.query_binary_values(
+                ':MMEMory:TRANsfer? \'screen.png\'',
+                datatype='B', 
+                is_big_endian=False, 
+                container=bytearray
+            )
+        ))
+        self.write('MMEMory:DELete \'screen.png\'')
+        return image
+
     '''
     @property
     def form(self):