Add KeyData.run_value()

Author: philsmt

extra_data/keydata.py

@@ -105,7 +105,7 @@ class KeyData:
     """
     def __init__(
             self, source, key, *, train_ids, files, section, dtype, eshape,
-            inc_suspect_trains=True,
+            is_single_run, inc_suspect_trains=True,
     ):
         self.source = source
         self.key = key
@@ -115,6 +115,7 @@ def __init__(
         self.dtype = dtype
         self.entry_shape = eshape
         self.ndim = len(eshape) + 1
+        self.is_single_run = is_single_run
         self.inc_suspect_trains = inc_suspect_trains
 
     def _find_chunks(self):
@@ -343,6 +344,7 @@ def _only_tids(self, tids, files=None):
             section=self.section,
             dtype=self.dtype,
             eshape=self.entry_shape,
+            is_single_run=self.is_single_run,
             inc_suspect_trains=self.inc_suspect_trains,
         )
 
@@ -472,6 +474,29 @@ def as_single_value(self, rtol=1e-5, atol=0.0, reduce_by=None):
 
         return value
 
+    def run_value(self, allow_multi_run=False):
+        """Get the RUN value for this key if it exists.
+
+        This method is intended for use with data from a single run. If you
+        combine data from multiple runs, it will raise MultiRunError.
+
+        Returns the RUN parameter value corresponding to this key.
+        """
+
+        from .sourcedata import SourceData  # Prevent cyclic import.
+
+        # Construct minimal SourceData object to obtain RUN value.
+        return SourceData(
+            self.source,
+            sel_keys=None,
+            train_ids=self.train_ids,
+            files=self.files,
+            section=self.section,
+            canonical_name=self.source,
+            is_single_run=self.is_single_run,
+            inc_suspect_trains=self.inc_suspect_trains
+        ).run_value(self.key, allow_multi_run=allow_multi_run)
+
     # Getting data as different kinds of array: -------------------------------
 
     def ndarray(self, roi=(), out=None):

extra_data/sourcedata.py

@@ -118,6 +118,7 @@ def __getitem__(self, key):
             section=self.section,
             dtype=ds0.dtype,
             eshape=ds0.shape[1:],
+            is_single_run=self.is_single_run,
             inc_suspect_trains=self.inc_suspect_trains,
         )
 

extra_data/tests/test_keydata.py

@@ -363,6 +363,17 @@ def test_single_value(mock_sa3_control_data, monkeypatch):
     np.testing.assert_equal(intensity.as_single_value(rtol=1), np.median(data))
 
 
+def test_run_value(mock_sa3_control_data):
+    f = H5File(mock_sa3_control_data)
+
+    flux = f['SA3_XTD10_XGM/XGM/DOOCS', 'pulseEnergy.photonFlux']
+    assert flux.run_value() == 0.0
+
+    imager = f['SA3_XTD10_IMGFEL/CAM/BEAMVIEW:daqOutput', 'data.image.pixels']
+    with pytest.raises(ValueError):
+        assert imager.run_value()
+
+
 def test_ndarray_out(mock_spb_raw_run):
     f = RunDirectory(mock_spb_raw_run)
     cam = f['SPB_IRU_CAM/CAM/SIDEMIC:daqOutput', 'data.image.dims']