|
17 | 17 | #include "AMReX_MultiFab.H" |
18 | 18 | #include "AMReX_MultiFabUtil.H" |
19 | 19 | #include "AMReX_Orientation.H" |
| 20 | +#include "AMReX_SPACE.H" |
20 | 21 | #include "AMReX_VisMF.H" |
21 | 22 | #include <AMReX.H> |
22 | 23 |
|
@@ -110,11 +111,17 @@ auto ComputePlaneProjection(amrex::Vector<amrex::MultiFab> const &state_new, con |
110 | 111 | auto const proj_arr = projections[lev].arrays(); |
111 | 112 | amrex::ParallelFor(projections[lev], [=] AMREX_GPU_DEVICE(int bx, int i, int j, int k) { |
112 | 113 | if (dir == amrex::Direction::x) { |
113 | | - proj_arr[bx](i, j, k) = plane_arr(0, i, j); |
| 114 | + proj_arr[bx](i, j, k) = plane_arr(AMREX_D_DECL(0, i, j)); |
| 115 | +#if AMREX_SPACEDIM >= 2 |
114 | 116 | } else if (dir == amrex::Direction::y) { |
115 | | - proj_arr[bx](i, j, k) = plane_arr(i, 0, j); |
| 117 | + proj_arr[bx](i, j, k) = plane_arr(AMREX_D_DECL(i, 0, j)); |
| 118 | +#endif |
| 119 | +#if AMREX_SPACEDIM == 3 |
| 120 | + } else if (dir == amrex::Direction::z) { |
| 121 | + proj_arr[bx](i, j, k) = plane_arr(AMREX_D_DECL(i, j, 0)); |
| 122 | +#endif |
116 | 123 | } else { |
117 | | - proj_arr[bx](i, j, k) = plane_arr(i, j, k); |
| 124 | + proj_arr[bx](i, j, k) = 0.0; |
118 | 125 | } |
119 | 126 | }); |
120 | 127 | amrex::Gpu::streamSynchronize(); |
|
0 commit comments