wip

Author: jokasimr

src/ess/estia/load.py

@@ -1,18 +1,23 @@
+import h5py
 import scipp as sc
 
 from ..reflectometry.types import (
     Filename,
     RawDetectorData,
     RunType,
 )
-from .mcstas import parse_events_ascii
+from .mcstas import parse_events_ascii, parse_events_h5
 
 
 def load_mcstas_events(
     filename: Filename[RunType],
 ) -> RawDetectorData[RunType]:
-    with open(filename) as f:
-        da = parse_events_ascii(f)
+    if h5py.is_hdf5(filename):
+        with h5py.File(filename) as f:
+            da = parse_events_h5(f)
+    else:
+        with open(filename) as f:
+            da = parse_events_ascii(f)
 
     da.coords['sample_rotation'] = sc.scalar(
         float(da.coords['omegaa'].value), unit='deg'

src/ess/estia/mcstas.py

@@ -47,3 +47,15 @@ def parse_events_ascii(lines):
                 da.coords[k] = sc.scalar(v)
             return da
     raise ValueError('Could not parse the file as a list of events.')
+
+
+def parse_events_h5(f):
+    pass
+
+
+'''
+def parse_events_h5(f):
+    f['entry1/data']
+    events = load_nx(f, 'NXentry/NXdetector/NXdata')
+    parameters = load_nx(f, 'NXentry/simulation/Param')
+'''