From 647f5e50f90baf3617d3baddd8b8f2d44507d805 Mon Sep 17 00:00:00 2001
From: alex-rakowski <alexrakowski90@gmail.com>
Date: Fri, 3 Nov 2023 18:01:02 -0700
Subject: [PATCH] initial namespace chages

---
 py4DSTEM/__init__.py | 69 ++++++++++++++++++++++++++++++++++++++------
 1 file changed, 60 insertions(+), 9 deletions(-)

diff --git a/py4DSTEM/__init__.py b/py4DSTEM/__init__.py
index dcb6a861d..2d312c437 100644
--- a/py4DSTEM/__init__.py
+++ b/py4DSTEM/__init__.py
@@ -52,18 +52,69 @@
     BraggVectorMap,
 )
 
-# strain
-from py4DSTEM.process import StrainMap
+# TODO - classification
+# from py4DSTEM.process.classification import (
+#     BraggVectorClassification,
+#     ConvergenceWarning,
+#     DataCube,
+#     FastICA,
+#     Featurization,
+#     GaussianMixture,
+#     MinMaxScaler,
+#     NMF,
+#     PCA,
+#     PointListArray,
+#     RobustScaler,
+#     permutations,
+# )
+
+# diffraction
+from py4DSTEM.process.diffraction import Crystal
+
+# py4DSTEM.process.diffraction
+# ['Axes', 'Crystal', 'Crystal_Phase', 'Figure', 'Fraction', 'ListedColormap', 'Orientation', 'PointList', 'PointListArray', 'single_atom_scatter']
+
+
+# ptycho
+from py4DSTEM.process.phase import (
+    DPCReconstruction,
+    MixedstatePtychographicReconstruction,
+    MultislicePtychographicReconstruction,
+    OptimizationParameter,
+    OverlapMagneticTomographicReconstruction,
+    OverlapTomographicReconstruction,
+    ParallaxReconstruction,
+    PtychographyOptimizer,
+    SimultaneousPtychographicReconstruction,
+    SingleslicePtychographicReconstruction,
+)
 
-# TODO - crystal
-# TODO - ptycho
-# TODO - others
+# polar
+from py4DSTEM.process.polar import PolarDatacube
 
-# TODO - where
-from py4DSTEM.process import (
-    PolarDatacube,
-)
+# from py4DSTEM.process import (
+#     PolarDatacube,
+# )
+
+# strain
+from py4DSTEM.process import StrainMap
 
+# TODO WPF
+# from py4DSTEM.process.wholepatternfit import (
+#     ComplexOverlapKernelDiskLattice,
+#     DCBackground,
+#     Flag,
+#     GaussianBackground,
+#     GaussianRing,
+#     KernelDiskLattice,
+#     Parameter,
+#     SyntheticDiskLattice,
+#     SyntheticDiskMoire,
+#     WPFModel,
+#     WPFModelType,
+#     WholePatternFit,
+#     auto,
+# )
 
 ### more submodules
 # TODO