Merge pull request #430 from European-XFEL/kd-size

Implement .nbytes, .size_mb, and .size_gb for KeyData

docs/reading_files.rst

@@ -177,6 +177,12 @@ below, e.g.::
 
    .. automethod:: data_counts
 
+   .. autoattribute:: nbytes
+
+   .. autoattribute:: size_mb
+
+   .. autoattribute:: size_gb
+
    .. automethod:: ndarray
 
    .. automethod:: series

extra_data/keydata.py

@@ -91,6 +91,21 @@ def shape(self):
         """
         return (sum(c.total_count for c in self._data_chunks),) + self.entry_shape
 
+    @property
+    def nbytes(self):
+        """The number of bytes this data would take up in memory."""
+        return self.dtype.itemsize * np.prod(self.shape)
+
+    @property
+    def size_mb(self):
+        """The size of the data in memory in megabytes."""
+        return self.nbytes / 1e6
+
+    @property
+    def size_gb(self):
+        """The size of the data in memory in gigabytes."""
+        return self.nbytes / 1e9
+
     @property
     def source_file_paths(self):
         paths = []

extra_data/tests/test_keydata.py

@@ -34,6 +34,9 @@ def test_get_keydata(mock_spb_raw_run):
         'RAW-R0238-DA01-S00000.h5', 'RAW-R0238-DA01-S00001.h5'
     }
 
+    data = xgm_beam_x.ndarray()
+    assert xgm_beam_x.nbytes == data.nbytes
+
 def test_select_trains(mock_spb_raw_run):
     run = RunDirectory(mock_spb_raw_run)
     xgm_beam_x = run['SPB_XTD9_XGM/DOOCS/MAIN', 'beamPosition.ixPos.value']