1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
|
/* File produced by Kranc */
#define KRANC_C
#include <assert.h>
#include <math.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include "cctk.h"
#include "cctk_Arguments.h"
#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)
#define QAD(x) (SQR(SQR(x)))
#define INV(x) ((1.0) / (x))
#define SQR(x) ((x) * (x))
#define CUB(x) ((x) * (x) * (x))
extern "C" void hydro_RHS_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 */, "ML_hydro::ene_grouprhs","flat");
if (ierr < 0)
CCTK_WARN(1, "Failed to register flat BC for ML_hydro::ene_grouprhs.");
ierr = Boundary_SelectGroupForBC(cctkGH, CCTK_ALL_FACES, GenericFD_GetBoundaryWidth(cctkGH), -1 /* no table */, "ML_hydro::mass_grouprhs","flat");
if (ierr < 0)
CCTK_WARN(1, "Failed to register flat BC for ML_hydro::mass_grouprhs.");
ierr = Boundary_SelectGroupForBC(cctkGH, CCTK_ALL_FACES, GenericFD_GetBoundaryWidth(cctkGH), -1 /* no table */, "ML_hydro::mom_grouprhs","flat");
if (ierr < 0)
CCTK_WARN(1, "Failed to register flat BC for ML_hydro::mom_grouprhs.");
return;
}
static void hydro_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 imin[3], int const imax[3], int const n_subblock_gfs, CCTK_REAL * restrict const subblock_gfs[])
{
DECLARE_CCTK_ARGUMENTS;
DECLARE_CCTK_PARAMETERS;
/* 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);
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 p1o2dx = 0.5*INV(dx);
CCTK_REAL const p1o2dy = 0.5*INV(dy);
CCTK_REAL const p1o2dz = 0.5*INV(dz);
CCTK_REAL const p1o4dxdy = 0.25*INV(dx*dy);
CCTK_REAL const p1o4dxdz = 0.25*INV(dx*dz);
CCTK_REAL const p1o4dydz = 0.25*INV(dy*dz);
CCTK_REAL const p1odx2 = INV(SQR(dx));
CCTK_REAL const p1ody2 = INV(SQR(dy));
CCTK_REAL const p1odz2 = 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 */
#pragma omp parallel
CCTK_LOOP3(hydro_RHS,
i,j,k, imin[0],imin[1],imin[2], imax[0],imax[1],imax[2],
cctk_lsh[0],cctk_lsh[1],cctk_lsh[2])
{
ptrdiff_t const index = di*i + dj*j + dk*k;
/* Assign local copies of grid functions */
CCTK_REAL eneflux1L = eneflux1[index];
CCTK_REAL eneflux2L = eneflux2[index];
CCTK_REAL eneflux3L = eneflux3[index];
CCTK_REAL massflux1L = massflux1[index];
CCTK_REAL massflux2L = massflux2[index];
CCTK_REAL massflux3L = massflux3[index];
CCTK_REAL momflux11L = momflux11[index];
CCTK_REAL momflux12L = momflux12[index];
CCTK_REAL momflux13L = momflux13[index];
CCTK_REAL momflux21L = momflux21[index];
CCTK_REAL momflux22L = momflux22[index];
CCTK_REAL momflux23L = momflux23[index];
CCTK_REAL momflux31L = momflux31[index];
CCTK_REAL momflux32L = momflux32[index];
CCTK_REAL momflux33L = momflux33[index];
/* Include user supplied include files */
/* Precompute derivatives */
CCTK_REAL const PDstandardNth1eneflux1 = PDstandardNth1(&eneflux1[index]);
CCTK_REAL const PDstandardNth2eneflux2 = PDstandardNth2(&eneflux2[index]);
CCTK_REAL const PDstandardNth3eneflux3 = PDstandardNth3(&eneflux3[index]);
CCTK_REAL const PDstandardNth1massflux1 = PDstandardNth1(&massflux1[index]);
CCTK_REAL const PDstandardNth2massflux2 = PDstandardNth2(&massflux2[index]);
CCTK_REAL const PDstandardNth3massflux3 = PDstandardNth3(&massflux3[index]);
CCTK_REAL const PDstandardNth1momflux11 = PDstandardNth1(&momflux11[index]);
CCTK_REAL const PDstandardNth2momflux12 = PDstandardNth2(&momflux12[index]);
CCTK_REAL const PDstandardNth3momflux13 = PDstandardNth3(&momflux13[index]);
CCTK_REAL const PDstandardNth1momflux21 = PDstandardNth1(&momflux21[index]);
CCTK_REAL const PDstandardNth2momflux22 = PDstandardNth2(&momflux22[index]);
CCTK_REAL const PDstandardNth3momflux23 = PDstandardNth3(&momflux23[index]);
CCTK_REAL const PDstandardNth1momflux31 = PDstandardNth1(&momflux31[index]);
CCTK_REAL const PDstandardNth2momflux32 = PDstandardNth2(&momflux32[index]);
CCTK_REAL const PDstandardNth3momflux33 = PDstandardNth3(&momflux33[index]);
/* Calculate temporaries and grid functions */
CCTK_REAL massrhsL = -PDstandardNth1massflux1 -
PDstandardNth2massflux2 - PDstandardNth3massflux3;
CCTK_REAL mom1rhsL = -PDstandardNth1momflux11 -
PDstandardNth2momflux12 - PDstandardNth3momflux13;
CCTK_REAL mom2rhsL = -PDstandardNth1momflux21 -
PDstandardNth2momflux22 - PDstandardNth3momflux23;
CCTK_REAL mom3rhsL = -PDstandardNth1momflux31 -
PDstandardNth2momflux32 - PDstandardNth3momflux33;
CCTK_REAL enerhsL = -PDstandardNth1eneflux1 - PDstandardNth2eneflux2 -
PDstandardNth3eneflux3;
/* Copy local copies back to grid functions */
enerhs[index] = enerhsL;
massrhs[index] = massrhsL;
mom1rhs[index] = mom1rhsL;
mom2rhs[index] = mom2rhsL;
mom3rhs[index] = mom3rhsL;
}
CCTK_ENDLOOP3(hydro_RHS);
}
extern "C" void hydro_RHS(CCTK_ARGUMENTS)
{
DECLARE_CCTK_ARGUMENTS;
DECLARE_CCTK_PARAMETERS;
if (verbose > 1)
{
CCTK_VInfo(CCTK_THORNSTRING,"Entering hydro_RHS_Body");
}
if (cctk_iteration % hydro_RHS_calc_every != hydro_RHS_calc_offset)
{
return;
}
const char *const groups[] = {
"ML_hydro::eneflux_group",
"ML_hydro::ene_grouprhs",
"ML_hydro::massflux_group",
"ML_hydro::mass_grouprhs",
"ML_hydro::momflux_group",
"ML_hydro::mom_grouprhs"};
GenericFD_AssertGroupStorage(cctkGH, "hydro_RHS", 6, groups);
GenericFD_EnsureStencilFits(cctkGH, "hydro_RHS", 1, 1, 1);
GenericFD_LoopOverInterior(cctkGH, hydro_RHS_Body);
if (verbose > 1)
{
CCTK_VInfo(CCTK_THORNSTRING,"Leaving hydro_RHS_Body");
}
}
|
