diff options
Diffstat (limited to 'Examples/EulerSR/src/eulersr_cons_calc_intercell_conserved_2.cc')
-rw-r--r-- | Examples/EulerSR/src/eulersr_cons_calc_intercell_conserved_2.cc | 219 |
1 files changed, 117 insertions, 102 deletions
diff --git a/Examples/EulerSR/src/eulersr_cons_calc_intercell_conserved_2.cc b/Examples/EulerSR/src/eulersr_cons_calc_intercell_conserved_2.cc index 2263c6a..83616ac 100644 --- a/Examples/EulerSR/src/eulersr_cons_calc_intercell_conserved_2.cc +++ b/Examples/EulerSR/src/eulersr_cons_calc_intercell_conserved_2.cc @@ -12,6 +12,8 @@ #include "cctk_Parameters.h" #include "GenericFD.h" #include "Differencing.h" +#include "cctk_Loop.h" +#include "loopcontrol.h" /* Define macros used in calculations */ #define INITVALUE (42) @@ -20,42 +22,28 @@ #define SQR(x) ((x) * (x)) #define CUB(x) ((x) * (x) * (x)) -static void eulersr_cons_calc_intercell_conserved_2_Body(cGH const * restrict const cctkGH, int const dir, int const face, CCTK_REAL const normal[3], CCTK_REAL const tangentA[3], CCTK_REAL const tangentB[3], int const min[3], int const max[3], int const n_subblock_gfs, CCTK_REAL * restrict const subblock_gfs[]) +static void eulersr_cons_calc_intercell_conserved_2_Body(cGH const * restrict const cctkGH, int const dir, int const face, CCTK_REAL const normal[3], CCTK_REAL const tangentA[3], CCTK_REAL const tangentB[3], int const imin[3], int const imax[3], int const n_subblock_gfs, CCTK_REAL * restrict const subblock_gfs[]) { DECLARE_CCTK_ARGUMENTS; DECLARE_CCTK_PARAMETERS; - /* Declare the variables used for looping over grid points */ - CCTK_INT i, j, k; - // CCTK_INT index = INITVALUE; - /* Declare finite differencing variables */ - if (verbose > 1) - { - CCTK_VInfo(CCTK_THORNSTRING,"Entering eulersr_cons_calc_intercell_conserved_2_Body"); - } - - if (cctk_iteration % eulersr_cons_calc_intercell_conserved_2_calc_every != eulersr_cons_calc_intercell_conserved_2_calc_offset) - { - return; - } - - const char *groups[] = {"EulerSR::Den_lr_group","EulerSR::epsi_lr_group","EulerSR::h_group","EulerSR::p_group","EulerSR::rho_lr_group","EulerSR::S1_lr_group","EulerSR::S2_lr_group","EulerSR::S3_lr_group","EulerSR::tau_lr_group","EulerSR::v1_lr_group","EulerSR::v2_lr_group","EulerSR::v3_lr_group","EulerSR::W_group"}; - GenericFD_AssertGroupStorage(cctkGH, "eulersr_cons_calc_intercell_conserved_2", 13, groups); - - /* Include user-supplied include files */ /* Initialise finite differencing variables */ ptrdiff_t const di = 1; ptrdiff_t const dj = CCTK_GFINDEX3D(cctkGH,0,1,0) - CCTK_GFINDEX3D(cctkGH,0,0,0); ptrdiff_t const dk = CCTK_GFINDEX3D(cctkGH,0,0,1) - CCTK_GFINDEX3D(cctkGH,0,0,0); + ptrdiff_t const cdi = sizeof(CCTK_REAL) * di; + ptrdiff_t const cdj = sizeof(CCTK_REAL) * dj; + ptrdiff_t const cdk = sizeof(CCTK_REAL) * dk; CCTK_REAL const dx = ToReal(CCTK_DELTA_SPACE(0)); CCTK_REAL const dy = ToReal(CCTK_DELTA_SPACE(1)); CCTK_REAL const dz = ToReal(CCTK_DELTA_SPACE(2)); CCTK_REAL const dt = ToReal(CCTK_DELTA_TIME); + CCTK_REAL const t = ToReal(cctk_time); CCTK_REAL const dxi = INV(dx); CCTK_REAL const dyi = INV(dy); CCTK_REAL const dzi = INV(dz); @@ -74,92 +62,99 @@ static void eulersr_cons_calc_intercell_conserved_2_Body(cGH const * restrict co CCTK_REAL const p1ody = INV(dy); CCTK_REAL const p1odz = INV(dz); + /* Assign local copies of arrays functions */ + + + + /* Calculate temporaries and arrays functions */ + + /* Copy local copies back to grid functions */ + /* Loop over the grid points */ - for (k = min[2]; k < max[2]; k++) + #pragma omp parallel + CCTK_LOOP3 (eulersr_cons_calc_intercell_conserved_2, + i,j,k, imin[0],imin[1],imin[2], imax[0],imax[1],imax[2], + cctk_lsh[0],cctk_lsh[1],cctk_lsh[2]) { - for (j = min[1]; j < max[1]; j++) - { - for (i = min[0]; i < max[0]; i++) - { - int const index = CCTK_GFINDEX3D(cctkGH,i,j,k) ; - - /* Assign local copies of grid functions */ - - CCTK_REAL DenLeftL = DenLeft[index]; - CCTK_REAL DenRightL = DenRight[index]; - CCTK_REAL epsiLeftL = epsiLeft[index]; - CCTK_REAL epsiRightL = epsiRight[index]; - CCTK_REAL hL = h[index]; - CCTK_REAL pL = p[index]; - CCTK_REAL rhoLeftL = rhoLeft[index]; - CCTK_REAL rhoRightL = rhoRight[index]; - CCTK_REAL v1LeftL = v1Left[index]; - CCTK_REAL v1RightL = v1Right[index]; - CCTK_REAL v2LeftL = v2Left[index]; - CCTK_REAL v2RightL = v2Right[index]; - CCTK_REAL v3LeftL = v3Left[index]; - CCTK_REAL v3RightL = v3Right[index]; - CCTK_REAL WL = W[index]; - - - /* Include user supplied include files */ - - /* Precompute derivatives */ - - /* Calculate temporaries and grid functions */ - CCTK_REAL Wx = 1 - SQR(v1LeftL) - SQR(v2LeftL) - SQR(v3LeftL); - - WL = INV(sqrt(Wx)); - - pL = epsiLeftL*rhoLeftL*(-1 + ToReal(gamma)); - - hL = 1 + epsiLeftL + pL*INV(rhoLeftL); - - DenLeftL = rhoLeftL*WL; - - CCTK_REAL S1LeftL = hL*rhoLeftL*v1LeftL*SQR(WL); - - CCTK_REAL S2LeftL = hL*rhoLeftL*v2LeftL*SQR(WL); - - CCTK_REAL S3LeftL = hL*rhoLeftL*v3LeftL*SQR(WL); - - CCTK_REAL tauLeftL = -DenLeftL - pL + hL*rhoLeftL*SQR(WL); - - Wx = 1 - SQR(v1RightL) - SQR(v2RightL) - SQR(v3RightL); - - WL = INV(sqrt(Wx)); - - pL = epsiRightL*rhoRightL*(-1 + ToReal(gamma)); - - hL = 1 + epsiRightL + pL*INV(rhoRightL); - - DenRightL = rhoRightL*WL; - - CCTK_REAL S1RightL = hL*rhoRightL*v1RightL*SQR(WL); - - CCTK_REAL S2RightL = hL*rhoRightL*v2RightL*SQR(WL); - - CCTK_REAL S3RightL = hL*rhoRightL*v3RightL*SQR(WL); - - CCTK_REAL tauRightL = -DenRightL - pL + hL*rhoRightL*SQR(WL); - - /* Copy local copies back to grid functions */ - DenLeft[index] = DenLeftL; - DenRight[index] = DenRightL; - h[index] = hL; - p[index] = pL; - S1Left[index] = S1LeftL; - S1Right[index] = S1RightL; - S2Left[index] = S2LeftL; - S2Right[index] = S2RightL; - S3Left[index] = S3LeftL; - S3Right[index] = S3RightL; - tauLeft[index] = tauLeftL; - tauRight[index] = tauRightL; - W[index] = WL; - } - } + ptrdiff_t const index = di*i + dj*j + dk*k; + + /* Assign local copies of grid functions */ + + CCTK_REAL DenLeftL = DenLeft[index]; + CCTK_REAL DenRightL = DenRight[index]; + CCTK_REAL epsiLeftL = epsiLeft[index]; + CCTK_REAL epsiRightL = epsiRight[index]; + CCTK_REAL hL = h[index]; + CCTK_REAL pL = p[index]; + CCTK_REAL rhoLeftL = rhoLeft[index]; + CCTK_REAL rhoRightL = rhoRight[index]; + CCTK_REAL v1LeftL = v1Left[index]; + CCTK_REAL v1RightL = v1Right[index]; + CCTK_REAL v2LeftL = v2Left[index]; + CCTK_REAL v2RightL = v2Right[index]; + CCTK_REAL v3LeftL = v3Left[index]; + CCTK_REAL v3RightL = v3Right[index]; + CCTK_REAL WL = W[index]; + + + /* Include user supplied include files */ + + /* Precompute derivatives */ + + /* Calculate temporaries and grid functions */ + CCTK_REAL Wx = 1 - SQR(v1LeftL) - SQR(v2LeftL) - SQR(v3LeftL); + + WL = INV(sqrt(Wx)); + + pL = epsiLeftL*rhoLeftL*(-1 + ToReal(gamma)); + + hL = 1 + epsiLeftL + pL*INV(rhoLeftL); + + DenLeftL = rhoLeftL*WL; + + CCTK_REAL S1LeftL = hL*rhoLeftL*v1LeftL*SQR(WL); + + CCTK_REAL S2LeftL = hL*rhoLeftL*v2LeftL*SQR(WL); + + CCTK_REAL S3LeftL = hL*rhoLeftL*v3LeftL*SQR(WL); + + CCTK_REAL tauLeftL = -DenLeftL - pL + hL*rhoLeftL*SQR(WL); + + Wx = 1 - SQR(v1RightL) - SQR(v2RightL) - SQR(v3RightL); + + WL = INV(sqrt(Wx)); + + pL = epsiRightL*rhoRightL*(-1 + ToReal(gamma)); + + hL = 1 + epsiRightL + pL*INV(rhoRightL); + + DenRightL = rhoRightL*WL; + + CCTK_REAL S1RightL = hL*rhoRightL*v1RightL*SQR(WL); + + CCTK_REAL S2RightL = hL*rhoRightL*v2RightL*SQR(WL); + + CCTK_REAL S3RightL = hL*rhoRightL*v3RightL*SQR(WL); + + CCTK_REAL tauRightL = -DenRightL - pL + + hL*rhoRightL*SQR(WL); + + /* Copy local copies back to grid functions */ + DenLeft[index] = DenLeftL; + DenRight[index] = DenRightL; + h[index] = hL; + p[index] = pL; + S1Left[index] = S1LeftL; + S1Right[index] = S1RightL; + S2Left[index] = S2LeftL; + S2Right[index] = S2RightL; + S3Left[index] = S3LeftL; + S3Right[index] = S3RightL; + tauLeft[index] = tauLeftL; + tauRight[index] = tauRightL; + W[index] = WL; } + CCTK_ENDLOOP3 (eulersr_cons_calc_intercell_conserved_2); } extern "C" void eulersr_cons_calc_intercell_conserved_2(CCTK_ARGUMENTS) @@ -167,5 +162,25 @@ extern "C" void eulersr_cons_calc_intercell_conserved_2(CCTK_ARGUMENTS) DECLARE_CCTK_ARGUMENTS; DECLARE_CCTK_PARAMETERS; + + if (verbose > 1) + { + CCTK_VInfo(CCTK_THORNSTRING,"Entering eulersr_cons_calc_intercell_conserved_2_Body"); + } + + if (cctk_iteration % eulersr_cons_calc_intercell_conserved_2_calc_every != eulersr_cons_calc_intercell_conserved_2_calc_offset) + { + return; + } + + const char *groups[] = {"EulerSR::Den_lr_group","EulerSR::epsi_lr_group","EulerSR::h_group","EulerSR::p_group","EulerSR::rho_lr_group","EulerSR::S1_lr_group","EulerSR::S2_lr_group","EulerSR::S3_lr_group","EulerSR::tau_lr_group","EulerSR::v1_lr_group","EulerSR::v2_lr_group","EulerSR::v3_lr_group","EulerSR::W_group"}; + GenericFD_AssertGroupStorage(cctkGH, "eulersr_cons_calc_intercell_conserved_2", 13, groups); + + GenericFD_LoopOverEverything(cctkGH, &eulersr_cons_calc_intercell_conserved_2_Body); + + if (verbose > 1) + { + CCTK_VInfo(CCTK_THORNSTRING,"Leaving eulersr_cons_calc_intercell_conserved_2_Body"); + } } |