ERF_UpdateRhoQtSources_SDMCongestus3DCold.H

Go to the documentation of this file.

    ParmParse pp_prob("prob");
 
    Real advection_moisture_rate = 1.0e-8; pp_prob.query("advection_moisture_rate", advection_moisture_rate);
    Real source_cutoff           = 500.0;  pp_prob.query("source_cutoff", source_cutoff);
 
    auto prob_lo = geom.ProbLoArray();
    auto prob_hi = geom.ProbHiArray();
    auto dx = geom.CellSizeArray();
    // On a mesh other than ConstantDz, take the heights from the levels, not from dx[2]
    const bool use_z_phys = use_physical_z();
 
    // Define a point (xc,yc) at the center of the domain
    Real xc = 0.5 * (prob_lo[0] + prob_hi[0]);
    Real yc = 0.5 * (prob_lo[1] + prob_hi[1]);
 
    for (MFIter mfi(*qsrc, TilingIfNotGPU()); mfi.isValid(); ++mfi)
    {
        const auto &box = mfi.tilebox();
        const Array4<Real>& qsrc_arr = qsrc->array(mfi);
        if (box.length(0) != 1)
        {
            // Spatially varying source over x,y
            ParallelFor(box, [=] AMREX_GPU_DEVICE(int i, int j, int k) noexcept
            {
                auto x = prob_lo[0] + (i + 0.5) * dx[0];
                auto y = prob_lo[1] + (j + 0.5) * dx[1];
                auto z = use_z_phys ? d_zlevels_arr[k] : (k + 0.5) * dx[2];
 
                const auto c = 1700.0*1700.0;
                const auto r = std::sqrt((x-xc)*(x-xc) + (y-yc)*(y-yc));
                const auto rsqr = r*r;
                qsrc_arr(i, j, k) = (advection_moisture_rate*exp(-rsqr / c))*exp(-z/source_cutoff);
            });
        } else {
            // src is a function over Z only
            ParallelFor(box, [=] AMREX_GPU_DEVICE(int i, int j, int k) {
                qsrc_arr(i, j, k) = 0.0;
            });
        }
    }