Add basic class and example

Author: henrikjacobsenfys

examples/example_data/diffusion_data_example.h5

@@ -0,0 +1,50 @@
+{
+ "cells": [
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "51b7b8be",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from easydynamics.experiment import Experiment\n",
+    "\n",
+    "import scipp as sc\n"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "eb91e49a",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "vanadium_experiment=Experiment(\"Vanadium\")\n",
+    "vanadium_experiment.load_hdf5(filename=\"example_data/vanadium_data_example.h5\")\n",
+    "\n",
+    "vanadium_experiment.plot_data()\n"
+   ]
+  }
+ ],
+ "metadata": {
+  "kernelspec": {
+   "display_name": "newdynamics",
+   "language": "python",
+   "name": "python3"
+  },
+  "language_info": {
+   "codemirror_mode": {
+    "name": "ipython",
+    "version": 3
+   },
+   "file_extension": ".py",
+   "mimetype": "text/x-python",
+   "name": "python",
+   "nbconvert_exporter": "python",
+   "pygments_lexer": "ipython3",
+   "version": "3.11.13"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 5
+}

examples/example_data/vanadium_data_example.h5

@@ -0,0 +1,7 @@
+from .experiment import (
+    Experiment,
+)
+
+__all__ = [
+    "Experiment",
+]

examples/experiment_example.ipynb

@@ -0,0 +1,110 @@
+from typing import Optional, Union
+
+import plopp as pp
+import scipp as sc
+from easyscience.job.experiment import ExperimentBase
+from scipp.io import load_hdf5 as sc_load_hdf5
+from scipp.io import save_hdf5 as sc_save_hdf5
+
+
+class Experiment(ExperimentBase):
+    """
+    Holds data from an experiment as a sc.DataArray along with metadata.
+    """
+
+    def __init__(
+        self,
+        name: str,
+        data: Optional[Union[sc.DataArray, str]] = None,
+        *args,
+        **kwargs,
+    ):
+        super(Experiment, self).__init__(name, *args, **kwargs)
+
+        if isinstance(data, str):
+            self.load_hdf5(filename=data)
+
+        elif isinstance(data, sc.DataArray):
+            self._data = data
+
+    def load_hdf5(self, filename: str):
+        """
+        Load data from an HDF5 file.
+
+        Args:
+            file_path (str): Path to the data file.
+            name (str): Name to assign to the loaded dataset.
+        """
+        if not isinstance(filename, str):
+            raise TypeError(f"Filename must be a string, not {type(filename).__name__}")
+
+        # TODO: Add checks of dimensions etc. I'm not yet sure what dimensions I want to allow, so for now I trust myself.
+
+        self._data = sc_load_hdf5(filename)
+
+    def save_hdf5(self, filename: Optional[str] = None):
+        """Save the dataset to HDF5.
+
+        Args:
+            filename (str): Path to the output HDF5 file.
+        """
+
+        if filename is None:
+            filename = f"{self.name}.h5"
+
+        if not isinstance(filename, str):
+            raise TypeError(f"Filename must be a string, not {type(filename).__name__}")
+
+        if self._data is None:
+            raise ValueError("No data to save.")
+
+        import os
+
+        os.makedirs(os.path.dirname(filename), exist_ok=True)
+        sc_save_hdf5(self._data, filename)
+
+    def plot_data(self):
+        """Plot the dataset using plopp."""
+
+        if self._data is None:
+            raise ValueError("No data to plot. Please load data first.")
+
+        if not self._in_notebook():
+            raise RuntimeError(
+                "plot_data() can only be used in a Jupyter notebook environment."
+            )
+
+        from IPython.display import display
+
+        fig = pp.plot(self._data.transpose(), title=f"{self.name}")
+        display(fig)
+
+    @staticmethod
+    def _in_notebook():
+        try:
+            from IPython import get_ipython
+
+            shell = get_ipython().__class__.__name__
+            if shell == "ZMQInteractiveShell":
+                return True  # Jupyter notebook or JupyterLab
+            elif shell == "TerminalInteractiveShell":
+                return False  # Terminal IPython
+            else:
+                return False
+        except (NameError, ImportError):
+            return False  # Standard Python (no IPython)
+
+    @staticmethod
+    def _in_notebook():
+        try:
+            from IPython import get_ipython
+
+            shell = get_ipython().__class__.__name__
+            if shell == "ZMQInteractiveShell":
+                return True  # Jupyter notebook or JupyterLab
+            elif shell == "TerminalInteractiveShell":
+                return False  # Terminal IPython
+            else:
+                return False
+        except (NameError, ImportError):
+            return False  # Standard Python (no IPython)