Merge pull request #122 from svalinn/parallelize_nwl

Parallelize nwl
svalinn · Jul 16, 2024 · fe943e2 · fe943e2
2 parents 8e8e537 + 37c4ce9
commit fe943e2
Show file tree

Hide file tree

Showing 2 changed files with 156 additions and 106 deletions.
diff --git a/Examples/nwl_plot_example.py b/Examples/nwl_plot_example.py
@@ -2,19 +2,21 @@
 
 
 # Define simulation parameters
-dagmc_geom = 'nwl_geom.h5m'
-source_mesh = 'source_mesh.h5m'
+dagmc_geom = "nwl_geom.h5m"
+source_mesh = "source_mesh.h5m"
 tor_ext = 90.0
-ss_file = 'source_strengths.txt'
+ss_file = "source_strengths.txt"
 num_parts = 100000
 
 nwl.nwl_transport(dagmc_geom, source_mesh, tor_ext, ss_file, num_parts)
 
 # Define first wall geometry and plotting parameters
-source_file = 'surface_source.h5'
-plas_eq = 'wout_vmec.nc'
+source_file = "surface_source.h5"
+plas_eq = "wout_vmec.nc"
 tor_ext = 90.0
 pol_ext = 360.0
 wall_s = 1.08
 
-nwl.nwl_plot(source_file, ss_file, plas_eq, tor_ext, pol_ext, wall_s)
+nwl.nwl_plot(
+    source_file, ss_file, plas_eq, tor_ext, pol_ext, wall_s, num_threads=2
+)