Add multidimensional drift correction and strip-based alignment - EDS, 4DSTEM prior to arXiv

pyproject.toml

@@ -44,6 +44,7 @@ dependencies = [
   "colorspacious",
   "dill",
   "numpy>2",
+  "pandas>=2.2",
   "matplotlib",
   "scipy",
   "tqdm",

src/quantem/imaging/__init__.py

@@ -1,3 +1,15 @@
-from quantem.imaging.drift import DriftCorrection as DriftCorrection
+"""Imaging tools for scientific image analysis."""
+
+from quantem.imaging.drift import (
+    CorrectionResult as CorrectionResult,
+    DriftCorrection as DriftCorrection,
+    StripPass as StripPass,
+    pair_spectrum_image_references as pair_spectrum_image_references,
+)
+from quantem.imaging.drift.io import (
+    read_emd as read_emd,
+    read_emd_eds as read_emd_eds,
+    read_emd_metadata as read_emd_metadata,
+)
 from quantem.imaging.lattice import Lattice as Lattice
 from quantem.imaging.lattice_visualization import PLOT_REGISTRY as PLOT_REGISTRY