diff options
Diffstat (limited to 'Examples/EM/src/EM_constraints.cc')
-rw-r--r-- | Examples/EM/src/EM_constraints.cc | 144 |
1 files changed, 85 insertions, 59 deletions
diff --git a/Examples/EM/src/EM_constraints.cc b/Examples/EM/src/EM_constraints.cc index d26cb66..dc40783 100644 --- a/Examples/EM/src/EM_constraints.cc +++ b/Examples/EM/src/EM_constraints.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,43 +22,40 @@ #define SQR(x) ((x) * (x)) #define CUB(x) ((x) * (x) * (x)) -static void EM_constraints_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[]) +extern "C" void EM_constraints_SelectBCs(CCTK_ARGUMENTS) { DECLARE_CCTK_ARGUMENTS; DECLARE_CCTK_PARAMETERS; + CCTK_INT ierr = 0; + ierr = Boundary_SelectGroupForBC(cctkGH, CCTK_ALL_FACES, GenericFD_GetBoundaryWidth(cctkGH), -1 /* no table */, "EM::constraints","flat"); + if (ierr < 0) + CCTK_WARN(1, "Failed to register flat BC for EM::constraints."); + return; +} + +static void EM_constraints_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_constraints_Body"); - } - - if (cctk_iteration % EM_constraints_calc_every != EM_constraints_calc_offset) - { - return; - } - - const char *groups[] = {"EM::B_group","EM::constraints","EM::El_group"}; - GenericFD_AssertGroupStorage(cctkGH, "EM_constraints", 3, groups); - - GenericFD_EnsureStencilFits(cctkGH, "EM_constraints", 1, 1, 1); - /* 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); @@ -93,48 +92,54 @@ static void EM_constraints_Body(cGH const * restrict const cctkGH, int const dir 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_constraints, + 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 B1L = B1[index]; - CCTK_REAL B2L = B2[index]; - CCTK_REAL B3L = B3[index]; - CCTK_REAL El1L = El1[index]; - CCTK_REAL El2L = El2[index]; - CCTK_REAL El3L = El3[index]; - - - /* Include user supplied include files */ - - /* Precompute derivatives */ - CCTK_REAL const PDstandard2nd1B1 = PDstandard2nd1(&B1[index]); - CCTK_REAL const PDstandard2nd2B2 = PDstandard2nd2(&B2[index]); - CCTK_REAL const PDstandard2nd3B3 = PDstandard2nd3(&B3[index]); - CCTK_REAL const PDstandard2nd1El1 = PDstandard2nd1(&El1[index]); - CCTK_REAL const PDstandard2nd2El2 = PDstandard2nd2(&El2[index]); - CCTK_REAL const PDstandard2nd3El3 = PDstandard2nd3(&El3[index]); - - /* Calculate temporaries and grid functions */ - CCTK_REAL CElL = PDstandard2nd1El1 + PDstandard2nd2El2 + - PDstandard2nd3El3; - - CCTK_REAL CBL = PDstandard2nd1B1 + PDstandard2nd2B2 + - PDstandard2nd3B3; - - /* Copy local copies back to grid functions */ - CB[index] = CBL; - CEl[index] = CElL; - } - } + ptrdiff_t const index = di*i + dj*j + dk*k; + + /* Assign local copies of grid functions */ + + CCTK_REAL B1L = B1[index]; + CCTK_REAL B2L = B2[index]; + CCTK_REAL B3L = B3[index]; + CCTK_REAL El1L = El1[index]; + CCTK_REAL El2L = El2[index]; + CCTK_REAL El3L = El3[index]; + + + /* Include user supplied include files */ + + /* Precompute derivatives */ + CCTK_REAL const PDstandard2nd1B1 = PDstandard2nd1(&B1[index]); + CCTK_REAL const PDstandard2nd2B2 = PDstandard2nd2(&B2[index]); + CCTK_REAL const PDstandard2nd3B3 = PDstandard2nd3(&B3[index]); + CCTK_REAL const PDstandard2nd1El1 = PDstandard2nd1(&El1[index]); + CCTK_REAL const PDstandard2nd2El2 = PDstandard2nd2(&El2[index]); + CCTK_REAL const PDstandard2nd3El3 = PDstandard2nd3(&El3[index]); + + /* Calculate temporaries and grid functions */ + CCTK_REAL CElL = PDstandard2nd1El1 + PDstandard2nd2El2 + + PDstandard2nd3El3; + + CCTK_REAL CBL = PDstandard2nd1B1 + PDstandard2nd2B2 + + PDstandard2nd3B3; + + /* Copy local copies back to grid functions */ + CB[index] = CBL; + CEl[index] = CElL; } + CCTK_ENDLOOP3 (EM_constraints); } extern "C" void EM_constraints(CCTK_ARGUMENTS) @@ -142,5 +147,26 @@ extern "C" void EM_constraints(CCTK_ARGUMENTS) DECLARE_CCTK_ARGUMENTS; DECLARE_CCTK_PARAMETERS; - GenericFD_LoopOverEverything(cctkGH, &EM_constraints_Body); + + if (verbose > 1) + { + CCTK_VInfo(CCTK_THORNSTRING,"Entering EM_constraints_Body"); + } + + if (cctk_iteration % EM_constraints_calc_every != EM_constraints_calc_offset) + { + return; + } + + const char *groups[] = {"EM::B_group","EM::constraints","EM::El_group"}; + GenericFD_AssertGroupStorage(cctkGH, "EM_constraints", 3, groups); + + GenericFD_EnsureStencilFits(cctkGH, "EM_constraints", 1, 1, 1); + + GenericFD_LoopOverInterior(cctkGH, &EM_constraints_Body); + + if (verbose > 1) + { + CCTK_VInfo(CCTK_THORNSTRING,"Leaving EM_constraints_Body"); + } } |