diff options
Diffstat (limited to 'Examples/EM/src/EM_initial.cc')
-rw-r--r-- | Examples/EM/src/EM_initial.cc | 134 |
1 files changed, 74 insertions, 60 deletions
diff --git a/Examples/EM/src/EM_initial.cc b/Examples/EM/src/EM_initial.cc index 9b2f83e..8d0e0b0 100644 --- a/Examples/EM/src/EM_initial.cc +++ b/Examples/EM/src/EM_initial.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 EM_initial_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 EM_initial_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 EM_initial_Body"); - } - - if (cctk_iteration % EM_initial_calc_every != EM_initial_calc_offset) - { - return; - } - - const char *groups[] = {"EM::B_group","EM::El_group","grid::coordinates"}; - GenericFD_AssertGroupStorage(cctkGH, "EM_initial", 3, 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); @@ -92,50 +80,56 @@ static void EM_initial_Body(cGH const * restrict const cctkGH, int const dir, in CCTK_REAL const pm1o12dy2 = -0.0833333333333333333333333333333*INV(SQR(dy)); CCTK_REAL const pm1o12dz2 = -0.0833333333333333333333333333333*INV(SQR(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 (EM_initial, + 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 xL = x[index]; - CCTK_REAL yL = y[index]; - - - /* Include user supplied include files */ - - /* Precompute derivatives */ - - /* Calculate temporaries and grid functions */ - CCTK_REAL El1L = Cos(2*Pi*(xL + yL))*ToReal(sigma); - - CCTK_REAL El2L = Cos(2*Pi*xL)*(-1 + ToReal(sigma)) - Cos(2*Pi*(xL + - yL))*ToReal(sigma); - - CCTK_REAL El3L = 0; - - CCTK_REAL B1L = 0; - - CCTK_REAL B2L = 0; - - CCTK_REAL B3L = -(Cos(2*Pi*xL)*(-1 + ToReal(sigma))) + Cos(2*Pi*(xL + - yL))*ToReal(sigma); - - /* Copy local copies back to grid functions */ - B1[index] = B1L; - B2[index] = B2L; - B3[index] = B3L; - El1[index] = El1L; - El2[index] = El2L; - El3[index] = El3L; - } - } + ptrdiff_t const index = di*i + dj*j + dk*k; + + /* Assign local copies of grid functions */ + + CCTK_REAL xL = x[index]; + CCTK_REAL yL = y[index]; + + + /* Include user supplied include files */ + + /* Precompute derivatives */ + + /* Calculate temporaries and grid functions */ + CCTK_REAL El1L = Cos(2*(xL + yL)*Pi)*ToReal(sigma); + + CCTK_REAL El2L = Cos(2*xL*Pi)*(-1 + ToReal(sigma)) - Cos(2*(xL + + yL)*Pi)*ToReal(sigma); + + CCTK_REAL El3L = 0; + + CCTK_REAL B1L = 0; + + CCTK_REAL B2L = 0; + + CCTK_REAL B3L = -(Cos(2*xL*Pi)*(-1 + ToReal(sigma))) + Cos(2*(xL + + yL)*Pi)*ToReal(sigma); + + /* Copy local copies back to grid functions */ + B1[index] = B1L; + B2[index] = B2L; + B3[index] = B3L; + El1[index] = El1L; + El2[index] = El2L; + El3[index] = El3L; } + CCTK_ENDLOOP3 (EM_initial); } extern "C" void EM_initial(CCTK_ARGUMENTS) @@ -143,5 +137,25 @@ extern "C" void EM_initial(CCTK_ARGUMENTS) DECLARE_CCTK_ARGUMENTS; DECLARE_CCTK_PARAMETERS; + + if (verbose > 1) + { + CCTK_VInfo(CCTK_THORNSTRING,"Entering EM_initial_Body"); + } + + if (cctk_iteration % EM_initial_calc_every != EM_initial_calc_offset) + { + return; + } + + const char *groups[] = {"EM::B_group","EM::El_group","grid::coordinates"}; + GenericFD_AssertGroupStorage(cctkGH, "EM_initial", 3, groups); + + GenericFD_LoopOverEverything(cctkGH, &EM_initial_Body); + + if (verbose > 1) + { + CCTK_VInfo(CCTK_THORNSTRING,"Leaving EM_initial_Body"); + } } |