From 730b21c90c7d2925516a5ec85e5520802c9ab868 Mon Sep 17 00:00:00 2001
From: Dan Porter <dan.porter@diamond.ac.uk>
Date: Wed, 14 Aug 2024 10:25:25 +0100
Subject: [PATCH] Updated python_script.yaml with babelscan script

---
 python_script.yaml | 14 +++++++++-----
 1 file changed, 9 insertions(+), 5 deletions(-)

diff --git a/python_script.yaml b/python_script.yaml
index 5a60727..e6fc4fa 100644
--- a/python_script.yaml
+++ b/python_script.yaml
@@ -33,13 +33,17 @@ spec:
       command: ["python"]
       source: |
         import numpy as np
-        import time
-        import h5py
+        import matplotlib.pyplot as plt
+        import babelscan
+        
+        from babelscan.plotting_matplotlib import set_plot_defaults
+        set_plot_defaults()  # set custom matplotlib rcParams
+        i16 = babelscan.instrument_from_config('i16')
   
         filename = '{{workflow.parameters.visitdir}}/{{inputs.parameters.scanno}}.nxs'
         print(filename)
+        scan = i16.scan(filename)
+        print(scan)
+        cmd = scan('cmd')
   
-        with h5py.File(filename, 'r') as hdf:
-          dataset = hdf['entry1/scan_command']
-          cmd = dataset.asstr()[()]
         {{inputs.parameters.python-code}}