diff options
Diffstat (limited to 'ML_WaveToy/src/WT_RHS.cc')
-rw-r--r-- | ML_WaveToy/src/WT_RHS.cc | 118 |
1 files changed, 118 insertions, 0 deletions
diff --git a/ML_WaveToy/src/WT_RHS.cc b/ML_WaveToy/src/WT_RHS.cc new file mode 100644 index 0000000..ed0c933 --- /dev/null +++ b/ML_WaveToy/src/WT_RHS.cc @@ -0,0 +1,118 @@ +/* File produced by Kranc */ + +#define KRANC_C + +#include <assert.h> +#include <math.h> +#include <stdio.h> +#include <stdlib.h> +#include "cctk.h" +#include "cctk_Arguments.h" +#include "cctk_Parameters.h" +#include "GenericFD.h" +#include "Differencing.h" +#include "Vectors.hh" +#include "loopcontrol.h" + +/* Define macros used in calculations */ +#define INITVALUE (42) +#define INV(x) ((1.0) / (x)) +#define SQR(x) ((x) * (x)) +#define CUB(x) ((x) * (x) * (x)) +#define QAD(x) ((x) * (x) * (x) * (x)) + +static void WT_RHS_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[]) +{ + DECLARE_CCTK_ARGUMENTS; + DECLARE_CCTK_PARAMETERS; + + + /* Declare finite differencing variables */ + + if (verbose > 1) + { + CCTK_VInfo(CCTK_THORNSTRING,"Entering WT_RHS_Body"); + } + + if (cctk_iteration % WT_RHS_calc_every != WT_RHS_calc_offset) + { + return; + } + + /* Include user-supplied include files */ + + /* Initialise finite differencing variables */ + CCTK_REAL const dx = CCTK_DELTA_SPACE(0); + CCTK_REAL const dy = CCTK_DELTA_SPACE(1); + CCTK_REAL const dz = CCTK_DELTA_SPACE(2); + int const di = 1; + int const dj = CCTK_GFINDEX3D(cctkGH,0,1,0) - CCTK_GFINDEX3D(cctkGH,0,0,0); + int const dk = CCTK_GFINDEX3D(cctkGH,0,0,1) - CCTK_GFINDEX3D(cctkGH,0,0,0); + CCTK_REAL const dxi = 1.0 / dx; + CCTK_REAL const dyi = 1.0 / dy; + CCTK_REAL const dzi = 1.0 / dz; + CCTK_REAL const khalf = 0.5; + CCTK_REAL const kthird = 1/3.0; + CCTK_REAL const ktwothird = 2.0/3.0; + CCTK_REAL const kfourthird = 4.0/3.0; + CCTK_REAL const keightthird = 8.0/3.0; + CCTK_REAL const hdxi = 0.5 * dxi; + CCTK_REAL const hdyi = 0.5 * dyi; + CCTK_REAL const hdzi = 0.5 * dzi; + + /* Initialize predefined quantities */ + CCTK_REAL const p1o12dx = INV(dx)/12.; + CCTK_REAL const p1o12dy = INV(dy)/12.; + CCTK_REAL const p1o12dz = INV(dz)/12.; + CCTK_REAL const p1o144dxdy = (INV(dx)*INV(dy))/144.; + CCTK_REAL const p1o144dxdz = (INV(dx)*INV(dz))/144.; + CCTK_REAL const p1o144dydz = (INV(dy)*INV(dz))/144.; + CCTK_REAL const pm1o12dx2 = -pow(dx,-2)/12.; + CCTK_REAL const pm1o12dy2 = -pow(dy,-2)/12.; + CCTK_REAL const pm1o12dz2 = -pow(dz,-2)/12.; + + /* Loop over the grid points */ + #pragma omp parallel + LC_LOOP3 (WT_RHS, + i,j,k, min[0],min[1],min[2], max[0],max[1],max[2], + cctk_lsh[0],cctk_lsh[1],cctk_lsh[2]) + { + // int index = INITVALUE; + int const index = CCTK_GFINDEX3D(cctkGH,i,j,k); + /* Declare derivatives */ + // CCTK_REAL PDstandardNth11u = INITVALUE; + // CCTK_REAL PDstandardNth22u = INITVALUE; + // CCTK_REAL PDstandardNth33u = INITVALUE; + + /* Assign local copies of grid functions */ + CCTK_REAL_VEC rhoL = vec_load(rho[index]); + CCTK_REAL_VEC uL = vec_load(u[index]); + + /* Include user supplied include files */ + + /* Precompute derivatives */ + CCTK_REAL_VEC const PDstandardNth11u = PDstandardNth11(u, i, j, k); + CCTK_REAL_VEC const PDstandardNth22u = PDstandardNth22(u, i, j, k); + CCTK_REAL_VEC const PDstandardNth33u = PDstandardNth33(u, i, j, k); + + /* Calculate temporaries and grid functions */ + CCTK_REAL_VEC urhsL = rhoL; + + CCTK_REAL_VEC rhorhsL = PDstandardNth11u + PDstandardNth22u + + PDstandardNth33u; + + + /* Copy local copies back to grid functions */ + vec_store_nta(rhorhs[index],rhorhsL); + vec_store_nta(urhs[index],urhsL); + } + LC_ENDLOOP3 (WT_RHS); +} + +extern "C" void WT_RHS(CCTK_ARGUMENTS) +{ + DECLARE_CCTK_ARGUMENTS; + DECLARE_CCTK_PARAMETERS; + + GenericFD_LoopOverInterior(cctkGH, &WT_RHS_Body); +} |