ERF_Orbit.cpp File Reference

#include <ERF_Orbit.H>

Include dependency graph for ERF_Orbit.cpp:

Functions
void	zenith (int &calday, amrex::MultiFab *clat, amrex::MultiFab *clon, const amrex::Vector< int > &rank_offsets, real1d &coszrs, int &ncol, const Real &eccen, const Real &mvelpp, const Real &lambm0, const Real &obliqr, amrex::Real uniform_angle)

Function Documentation

zenith()

void zenith	(	int &	calday,
		amrex::MultiFab *	clat,
		amrex::MultiFab *	clon,
		const amrex::Vector< int > &	rank_offsets,
		real1d &	coszrs,
		int &	ncol,
		const Real &	eccen,
		const Real &	mvelpp,
		const Real &	lambm0,
		const Real &	obliqr,
		amrex::Real	uniform_angle
	)

{
    Real delta;    // Solar declination angle  in radians
    Real eccf;     // Earth orbit eccentricity factor
 
    // Populate delta & eccf
    shr_orb_decl(calday, eccen, mvelpp, lambm0, obliqr, delta, eccf);
 
    // If we have a valid pointer, go through the whole machinery
    if (clat) {
        for (MFIter mfi(*clat); mfi.isValid(); ++mfi) {
            const auto& tbx = mfi.tilebox();
            auto nx = tbx.length(0);
 
            auto lat_array = clat->array(mfi);
            auto lon_array = clon->array(mfi);
 
            const int offset = rank_offsets[mfi.LocalIndex()];
 
            // NOTE: lat/lon are 2D multifabs!
            ParallelFor(tbx, [=] AMREX_GPU_DEVICE (int i, int j, int /*k*/)
            {
                auto icol = (j-tbx.smallEnd(1))*nx + (i-tbx.smallEnd(0)) + 1 + offset;
                coszrs(icol) = shr_orb_cosz(calday, lat_array(i,j,0), lon_array(i,j,0), delta, uniform_angle);
            });
       }
    }
    // Use constant value near center of USA or the uniform angle
    else {
        Real cons_lat = 40.0;
        Real cons_lon = 100.0;
        Real val = shr_orb_cosz(calday, cons_lat, cons_lon, delta, uniform_angle);
        yakl::memset(coszrs, val);
    }
}

Here is the call graph for this function: