Merge pull request #2727 from codingS3b/short_visualizer_import

Shorter python imports for postprocessing tools
ComputationalRadiationPhysics · Oct 15, 2018 · 36eabde · 36eabde
2 parents 8ed9645 + a7662fe
commit 36eabde
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/lib/python/picongpu/plugins/__init__.py b/lib/python/picongpu/plugins/__init__.py
@@ -0,0 +1,8 @@
+from .energy_histogram import EnergyHistogram
+from .phase_space import PhaseSpace
+from .png import PNG
+
+
+__all__ = ["EnergyHistogram",
+           "PhaseSpace",
+           "PNG"]
diff --git a/lib/python/picongpu/plugins/plot_mpl/__init__.py b/lib/python/picongpu/plugins/plot_mpl/__init__.py
@@ -0,0 +1,9 @@
+from .energy_histogram_visualizer import Visualizer as \
+    EnergyHistogramMPL
+from .phase_space_visualizer import Visualizer as PhaseSpaceMPL
+from .png_visualizer import Visualizer as PNGMPL
+
+
+__all__ = ["EnergyHistogramMPL",
+           "PhaseSpaceMPL",
+           "PNGMPL"]