ERF_UpdateWSubsidence_SineMassFlux.H

Go to the documentation of this file.

    // Compute density from state on device, copy to host
    Gpu::DeviceVector<Real> rho_d(khi+1, 0.0);
    auto rho_arr = rho_d.data();
    for (MFIter mfi(state, TilingIfNotGPU()); mfi.isValid(); ++mfi) {
        const auto &box = mfi.tilebox();
        auto state_arr = state.const_array(mfi);
        ParallelFor(box, [=] AMREX_GPU_DEVICE (int i, int j, int k) noexcept {
            rho_arr[k] = state_arr(i, j, k, Rho_comp);
        });
    }
 
    Vector<Real> rho_h(khi+1, 0.0);
    Gpu::copy(Gpu::deviceToHost, rho_d.begin(), rho_d.end(), rho_h.begin());
 
    // Staggered rho via linear interpolation
    Vector<Real> rho_stag(khi+2, 0.0);
    rho_stag[0]     = 1.5*rho_h[0]   - 0.5*rho_h[1];
    rho_stag[khi+1] = 1.5*rho_h[khi] - 0.5*rho_h[khi-1];
    for (int k = 1; k <= khi; k++) {
        rho_stag[k] = 0.5*(rho_h[k-1]+rho_h[k]);
    }
 
    // Sinusoidal mass flux: w = 2*sin(pi*t/600)/rho for t < 600
    for (int k = 0; k <= khi+1; k++) {
        if (time < 600) {
            wbar[k] = 2*sin(PI*time/600)/rho_stag[k];
        } else {
            wbar[k] = 0.0;
        }
    }
 
    // Copy from host version to device version
    amrex::Gpu::copy(amrex::Gpu::hostToDevice, wbar.begin(), wbar.end(), d_wbar.begin());