diff --git a/examples/visulization/godiva-mockup/input.py b/examples/visulization/godiva-mockup/input.py index acbd361b..729ca58a 100644 --- a/examples/visulization/godiva-mockup/input.py +++ b/examples/visulization/godiva-mockup/input.py @@ -1,5 +1,6 @@ import numpy as np import mcdc, h5py +from mcdc import visualizer # ============================================================================= # Materials @@ -73,7 +74,7 @@ mcdc.source(x=[-22.0, 22.0], time=[0.0, 5.0], isotropic=True) -mcdc.visualize( +visualizer.visualize( start_time=0, end_time=1, tick_interval=0.1, material_colors={"water": [1, 0, 0]} ) # ============================================================================= diff --git a/mcdc/__init__.py b/mcdc/__init__.py index 853af00a..c902b276 100644 --- a/mcdc/__init__.py +++ b/mcdc/__init__.py @@ -26,4 +26,3 @@ save_particle_bank, ) from mcdc.main import run, prepare -from mcdc.visualizer import visualize diff --git a/mcdc/visualizer/__init__.py b/mcdc/visualizer/__init__.py new file mode 100644 index 00000000..3f800a36 --- /dev/null +++ b/mcdc/visualizer/__init__.py @@ -0,0 +1 @@ +from mcdc.visualizer.visualizer import visualize diff --git a/mcdc/visualizer.py b/mcdc/visualizer/visualizer.py similarity index 100% rename from mcdc/visualizer.py rename to mcdc/visualizer/visualizer.py