diff options
Diffstat (limited to 'Examples/Modules/space_charge_initialization/analysis.py')
| -rwxr-xr-x | Examples/Modules/space_charge_initialization/analysis.py | 13 |
1 files changed, 9 insertions, 4 deletions
diff --git a/Examples/Modules/space_charge_initialization/analysis.py b/Examples/Modules/space_charge_initialization/analysis.py index cb382b9c3..a9a40470d 100755 --- a/Examples/Modules/space_charge_initialization/analysis.py +++ b/Examples/Modules/space_charge_initialization/analysis.py @@ -30,15 +30,20 @@ r0 = 2.e-6 # Open data file filename = sys.argv[1] ds = yt.load( filename ) +# yt 4.0+ has rounding issues with our domain data: +# RuntimeError: yt attempted to read outside the boundaries +# of a non-periodic domain along dimension 0. +if 'force_periodicity' in dir(ds): ds.force_periodicity() + # Extract data ad0 = ds.covering_grid(level=0, left_edge=ds.domain_left_edge, dims=ds.domain_dimensions) -Ex_array = ad0['Ex'].to_ndarray().squeeze() +Ex_array = ad0[("mesh", "Ex")].to_ndarray().squeeze() if ds.dimensionality == 2: # Rename the z dimension as y, so as to make this script work for 2d and 3d - Ey_array = ad0['Ez'].to_ndarray().squeeze() + Ey_array = ad0[("mesh", "Ez")].to_ndarray().squeeze() elif ds.dimensionality == 3: - Ey_array = ad0['Ey'].to_ndarray() - Ez_array = ad0['Ez'].to_ndarray() + Ey_array = ad0[("mesh", "Ey")].to_ndarray() + Ez_array = ad0[("mesh", "Ez")].to_ndarray() # Extract grid coordinates Nx, Ny, Nz = ds.domain_dimensions |