ENH: Add general loading routine for corrections

src/darsia/__init__.py

@@ -57,6 +57,7 @@
 from darsia.corrections.color.colorcorrection import *
 from darsia.corrections.color.relativecolorcorrection import *
 from darsia.corrections.color.experimentalcolorcorrection import *
+from darsia.corrections.readcorrection import *
 from darsia.image.coordinatetransformation import *  # Requires affine correction
 from darsia.signals.models.basemodel import *
 from darsia.signals.models.combinedmodel import *

src/darsia/corrections/readcorrection.py

@@ -0,0 +1,41 @@
+"""Module containing common reading functionality for corrections."""
+
+from pathlib import Path
+
+import numpy as np
+
+from darsia import (
+    ColorCorrection,
+    CurvatureCorrection,
+    DriftCorrection,
+    IlluminationCorrection,
+    TypeCorrection,
+)
+
+AnyCorrection = (
+    TypeCorrection
+    | DriftCorrection
+    | CurvatureCorrection
+    | IlluminationCorrection
+    | ColorCorrection
+)
+
+
+def read_correction(path: Path) -> AnyCorrection:
+    """General function to read corrections from a file.
+
+    Args:
+        path (Path): path to npz file
+
+    Returns:
+        BaseCorrection: correction object
+
+    """
+
+    # Read class name from npz file
+    class_name = np.load(path, allow_pickle=True)["class_name"].item()
+
+    # Load correction from file
+    correction = eval(class_name)()
+    correction.load(path)
+    return correction