initial namespace chages

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