338 lines
12 KiB
C
338 lines
12 KiB
C
#ifndef SHARE_FUNC_H
|
||
#define SHARE_FUNC_H
|
||
|
||
#include <stdlib.h>
|
||
#include <stddef.h>
|
||
#include <math.h>
|
||
#include <stdio.h>
|
||
#include <omp.h>
|
||
/* 主网格：0-based -> 1D */
|
||
static inline size_t idx_ex(int i0, int j0, int k0, const int ex[3]) {
|
||
const int ex1 = ex[0], ex2 = ex[1];
|
||
return (size_t)i0 + (size_t)j0 * (size_t)ex1 + (size_t)k0 * (size_t)ex1 * (size_t)ex2;
|
||
}
|
||
|
||
/*
|
||
* fh 对应 Fortran: fh(-1:ex1, -1:ex2, -1:ex3)
|
||
* ord=2 => shift=1
|
||
* iF/jF/kF 为 Fortran 索引（可为 -1,0,1..ex）
|
||
*/
|
||
static inline size_t idx_fh_F_ord2(int iF, int jF, int kF, const int ex[3]) {
|
||
const int shift = 1;
|
||
const int nx = ex[0] + 2; // ex1 + ord
|
||
const int ny = ex[1] + 2;
|
||
|
||
const int ii = iF + shift; // 0..ex1+1
|
||
const int jj = jF + shift; // 0..ex2+1
|
||
const int kk = kF + shift; // 0..ex3+1
|
||
|
||
return (size_t)ii + (size_t)jj * (size_t)nx + (size_t)kk * (size_t)nx * (size_t)ny;
|
||
}
|
||
|
||
/*
|
||
* fh 对应 Fortran: fh(-2:ex1, -2:ex2, -2:ex3)
|
||
* ord=3 => shift=2
|
||
* iF/jF/kF 是 Fortran 索引（可为负）
|
||
*/
|
||
static inline size_t idx_fh_F(int iF, int jF, int kF, const int ex[3]) {
|
||
const int shift = 2; // ord=3 -> -2..ex
|
||
const int nx = ex[0] + 3; // ex1 + ord
|
||
const int ny = ex[1] + 3;
|
||
|
||
const int ii = iF + shift; // 0..ex1+2
|
||
const int jj = jF + shift; // 0..ex2+2
|
||
const int kk = kF + shift; // 0..ex3+2
|
||
|
||
return (size_t)ii + (size_t)jj * (size_t)nx + (size_t)kk * (size_t)nx * (size_t)ny;
|
||
}
|
||
|
||
/*
|
||
* func: (1..extc1, 1..extc2, 1..extc3) 1-based in Fortran
|
||
* funcc: (-ord+1..extc1, -ord+1..extc2, -ord+1..extc3) in Fortran
|
||
*
|
||
* C 里我们把：
|
||
* func 视为 0-based: i0=0..extc1-1, j0=0..extc2-1, k0=0..extc3-1
|
||
* funcc 用“平移下标”存为一维数组：
|
||
* iF in [-ord+1..extc1] -> ii = iF + (ord-1) in [0..extc1+ord-1]
|
||
* 总长度 nx = extc1 + ord
|
||
* 同理 ny = extc2 + ord, nz = extc3 + ord
|
||
*/
|
||
|
||
static inline size_t idx_func0(int i0, int j0, int k0, const int extc[3]) {
|
||
const int nx = extc[0], ny = extc[1];
|
||
return (size_t)i0 + (size_t)j0 * (size_t)nx + (size_t)k0 * (size_t)nx * (size_t)ny;
|
||
}
|
||
|
||
static inline size_t idx_funcc_F(int iF, int jF, int kF, int ord, const int extc[3]) {
|
||
const int shift = ord - 1; // iF = -shift .. extc1
|
||
const int nx = extc[0] + ord; // [-shift..extc1] 共 extc1+ord 个
|
||
const int ny = extc[1] + ord;
|
||
|
||
const int ii = iF + shift; // 0..extc1+shift
|
||
const int jj = jF + shift; // 0..extc2+shift
|
||
const int kk = kF + shift; // 0..extc3+shift
|
||
|
||
return (size_t)ii + (size_t)jj * (size_t)nx + (size_t)kk * (size_t)nx * (size_t)ny;
|
||
}
|
||
|
||
/*
|
||
* 等价于 Fortran:
|
||
* funcc(1:extc1,1:extc2,1:extc3)=func
|
||
* do i=0,ord-1
|
||
* funcc(-i,1:extc2,1:extc3) = funcc(i+1,1:extc2,1:extc3)*SoA(1)
|
||
* enddo
|
||
* do i=0,ord-1
|
||
* funcc(:,-i,1:extc3) = funcc(:,i+1,1:extc3)*SoA(2)
|
||
* enddo
|
||
* do i=0,ord-1
|
||
* funcc(:,:,-i) = funcc(:,:,i+1)*SoA(3)
|
||
* enddo
|
||
*/
|
||
static inline void symmetry_bd(int ord,
|
||
const int extc[3],
|
||
const double *func,
|
||
double *funcc,
|
||
const double SoA[3])
|
||
{
|
||
const int extc1 = extc[0], extc2 = extc[1], extc3 = extc[2];
|
||
|
||
// 1) funcc(1:extc1,1:extc2,1:extc3) = func
|
||
// Fortran 的 (iF=1..extc1) 对应 C 的 func(i0=0..extc1-1)
|
||
for (int k0 = 0; k0 < extc3; ++k0) {
|
||
for (int j0 = 0; j0 < extc2; ++j0) {
|
||
for (int i0 = 0; i0 < extc1; ++i0) {
|
||
const int iF = i0 + 1, jF = j0 + 1, kF = k0 + 1;
|
||
funcc[idx_funcc_F(iF, jF, kF, ord, extc)] = func[idx_func0(i0, j0, k0, extc)];
|
||
}
|
||
}
|
||
}
|
||
|
||
// 2) do i=0..ord-1: funcc(-i, 1:extc2, 1:extc3) = funcc(i+1, ...)*SoA(1)
|
||
for (int ii = 0; ii <= ord - 1; ++ii) {
|
||
const int iF_dst = -ii; // 0, -1, -2, ...
|
||
const int iF_src = ii + 1; // 1, 2, 3, ...
|
||
for (int kF = 1; kF <= extc3; ++kF) {
|
||
for (int jF = 1; jF <= extc2; ++jF) {
|
||
funcc[idx_funcc_F(iF_dst, jF, kF, ord, extc)] =
|
||
funcc[idx_funcc_F(iF_src, jF, kF, ord, extc)] * SoA[0];
|
||
}
|
||
}
|
||
}
|
||
|
||
// 3) do i=0..ord-1: funcc(:,-i, 1:extc3) = funcc(:, i+1, 1:extc3)*SoA(2)
|
||
// 注意 Fortran 这里的 ":" 表示 iF 从 (-ord+1..extc1) 全覆盖
|
||
for (int jj = 0; jj <= ord - 1; ++jj) {
|
||
const int jF_dst = -jj;
|
||
const int jF_src = jj + 1;
|
||
for (int kF = 1; kF <= extc3; ++kF) {
|
||
for (int iF = -ord + 1; iF <= extc1; ++iF) {
|
||
funcc[idx_funcc_F(iF, jF_dst, kF, ord, extc)] =
|
||
funcc[idx_funcc_F(iF, jF_src, kF, ord, extc)] * SoA[1];
|
||
}
|
||
}
|
||
}
|
||
|
||
// 4) do i=0..ord-1: funcc(:,:,-i) = funcc(:,:, i+1)*SoA(3)
|
||
for (int kk = 0; kk <= ord - 1; ++kk) {
|
||
const int kF_dst = -kk;
|
||
const int kF_src = kk + 1;
|
||
for (int jF = -ord + 1; jF <= extc2; ++jF) {
|
||
for (int iF = -ord + 1; iF <= extc1; ++iF) {
|
||
funcc[idx_funcc_F(iF, jF, kF_dst, ord, extc)] =
|
||
funcc[idx_funcc_F(iF, jF, kF_src, ord, extc)] * SoA[2];
|
||
}
|
||
}
|
||
}
|
||
}
|
||
#endif
|
||
|
||
/* 你已有的函数：idx_ex / idx_fh_F_ord2 以及 fh 的布局 */
|
||
static inline void fdderivs_xh(
|
||
int i0, int j0, int k0,
|
||
const int ex[3],
|
||
const double *fh,
|
||
int iminF, int jminF, int kminF,
|
||
int imaxF, int jmaxF, int kmaxF,
|
||
double Fdxdx, double Fdydy, double Fdzdz,
|
||
double Fdxdy, double Fdxdz, double Fdydz,
|
||
double Sdxdx, double Sdydy, double Sdzdz,
|
||
double Sdxdy, double Sdxdz, double Sdydz,
|
||
double *fxx, double *fxy, double *fxz,
|
||
double *fyy, double *fyz, double *fzz
|
||
){
|
||
const double F8 = 8.0;
|
||
const double F16 = 16.0;
|
||
const double F30 = 30.0;
|
||
const double TWO = 2.0;
|
||
|
||
const int iF = i0 + 1;
|
||
const int jF = j0 + 1;
|
||
const int kF = k0 + 1;
|
||
|
||
const size_t p = idx_ex(i0, j0, k0, ex);
|
||
|
||
/* 高阶分支：i±2,j±2,k±2 都在范围内 */
|
||
if ((iF + 2) <= imaxF && (iF - 2) >= iminF &&
|
||
(jF + 2) <= jmaxF && (jF - 2) >= jminF &&
|
||
(kF + 2) <= kmaxF && (kF - 2) >= kminF)
|
||
{
|
||
fxx[p] = Fdxdx * (
|
||
-fh[idx_fh_F_ord2(iF - 2, jF, kF, ex)] +
|
||
F16 * fh[idx_fh_F_ord2(iF - 1, jF, kF, ex)] -
|
||
F30 * fh[idx_fh_F_ord2(iF, jF, kF, ex)] -
|
||
fh[idx_fh_F_ord2(iF + 2, jF, kF, ex)] +
|
||
F16 * fh[idx_fh_F_ord2(iF + 1, jF, kF, ex)]
|
||
);
|
||
|
||
fyy[p] = Fdydy * (
|
||
-fh[idx_fh_F_ord2(iF, jF - 2, kF, ex)] +
|
||
F16 * fh[idx_fh_F_ord2(iF, jF - 1, kF, ex)] -
|
||
F30 * fh[idx_fh_F_ord2(iF, jF, kF, ex)] -
|
||
fh[idx_fh_F_ord2(iF, jF + 2, kF, ex)] +
|
||
F16 * fh[idx_fh_F_ord2(iF, jF + 1, kF, ex)]
|
||
);
|
||
|
||
fzz[p] = Fdzdz * (
|
||
-fh[idx_fh_F_ord2(iF, jF, kF - 2, ex)] +
|
||
F16 * fh[idx_fh_F_ord2(iF, jF, kF - 1, ex)] -
|
||
F30 * fh[idx_fh_F_ord2(iF, jF, kF, ex)] -
|
||
fh[idx_fh_F_ord2(iF, jF, kF + 2, ex)] +
|
||
F16 * fh[idx_fh_F_ord2(iF, jF, kF + 1, ex)]
|
||
);
|
||
|
||
/* fxy 高阶 */
|
||
{
|
||
const double t_jm2 =
|
||
( fh[idx_fh_F_ord2(iF - 2, jF - 2, kF, ex)]
|
||
-F8*fh[idx_fh_F_ord2(iF - 1, jF - 2, kF, ex)]
|
||
+F8*fh[idx_fh_F_ord2(iF + 1, jF - 2, kF, ex)]
|
||
- fh[idx_fh_F_ord2(iF + 2, jF - 2, kF, ex)] );
|
||
|
||
const double t_jm1 =
|
||
( fh[idx_fh_F_ord2(iF - 2, jF - 1, kF, ex)]
|
||
-F8*fh[idx_fh_F_ord2(iF - 1, jF - 1, kF, ex)]
|
||
+F8*fh[idx_fh_F_ord2(iF + 1, jF - 1, kF, ex)]
|
||
- fh[idx_fh_F_ord2(iF + 2, jF - 1, kF, ex)] );
|
||
|
||
const double t_jp1 =
|
||
( fh[idx_fh_F_ord2(iF - 2, jF + 1, kF, ex)]
|
||
-F8*fh[idx_fh_F_ord2(iF - 1, jF + 1, kF, ex)]
|
||
+F8*fh[idx_fh_F_ord2(iF + 1, jF + 1, kF, ex)]
|
||
- fh[idx_fh_F_ord2(iF + 2, jF + 1, kF, ex)] );
|
||
|
||
const double t_jp2 =
|
||
( fh[idx_fh_F_ord2(iF - 2, jF + 2, kF, ex)]
|
||
-F8*fh[idx_fh_F_ord2(iF - 1, jF + 2, kF, ex)]
|
||
+F8*fh[idx_fh_F_ord2(iF + 1, jF + 2, kF, ex)]
|
||
- fh[idx_fh_F_ord2(iF + 2, jF + 2, kF, ex)] );
|
||
|
||
fxy[p] = Fdxdy * ( t_jm2 - F8 * t_jm1 + F8 * t_jp1 - t_jp2 );
|
||
}
|
||
|
||
/* fxz 高阶 */
|
||
{
|
||
const double t_km2 =
|
||
( fh[idx_fh_F_ord2(iF - 2, jF, kF - 2, ex)]
|
||
-F8*fh[idx_fh_F_ord2(iF - 1, jF, kF - 2, ex)]
|
||
+F8*fh[idx_fh_F_ord2(iF + 1, jF, kF - 2, ex)]
|
||
- fh[idx_fh_F_ord2(iF + 2, jF, kF - 2, ex)] );
|
||
|
||
const double t_km1 =
|
||
( fh[idx_fh_F_ord2(iF - 2, jF, kF - 1, ex)]
|
||
-F8*fh[idx_fh_F_ord2(iF - 1, jF, kF - 1, ex)]
|
||
+F8*fh[idx_fh_F_ord2(iF + 1, jF, kF - 1, ex)]
|
||
- fh[idx_fh_F_ord2(iF + 2, jF, kF - 1, ex)] );
|
||
|
||
const double t_kp1 =
|
||
( fh[idx_fh_F_ord2(iF - 2, jF, kF + 1, ex)]
|
||
-F8*fh[idx_fh_F_ord2(iF - 1, jF, kF + 1, ex)]
|
||
+F8*fh[idx_fh_F_ord2(iF + 1, jF, kF + 1, ex)]
|
||
- fh[idx_fh_F_ord2(iF + 2, jF, kF + 1, ex)] );
|
||
|
||
const double t_kp2 =
|
||
( fh[idx_fh_F_ord2(iF - 2, jF, kF + 2, ex)]
|
||
-F8*fh[idx_fh_F_ord2(iF - 1, jF, kF + 2, ex)]
|
||
+F8*fh[idx_fh_F_ord2(iF + 1, jF, kF + 2, ex)]
|
||
- fh[idx_fh_F_ord2(iF + 2, jF, kF + 2, ex)] );
|
||
|
||
fxz[p] = Fdxdz * ( t_km2 - F8 * t_km1 + F8 * t_kp1 - t_kp2 );
|
||
}
|
||
|
||
/* fyz 高阶 */
|
||
{
|
||
const double t_km2 =
|
||
( fh[idx_fh_F_ord2(iF, jF - 2, kF - 2, ex)]
|
||
-F8*fh[idx_fh_F_ord2(iF, jF - 1, kF - 2, ex)]
|
||
+F8*fh[idx_fh_F_ord2(iF, jF + 1, kF - 2, ex)]
|
||
- fh[idx_fh_F_ord2(iF, jF + 2, kF - 2, ex)] );
|
||
|
||
const double t_km1 =
|
||
( fh[idx_fh_F_ord2(iF, jF - 2, kF - 1, ex)]
|
||
-F8*fh[idx_fh_F_ord2(iF, jF - 1, kF - 1, ex)]
|
||
+F8*fh[idx_fh_F_ord2(iF, jF + 1, kF - 1, ex)]
|
||
- fh[idx_fh_F_ord2(iF, jF + 2, kF - 1, ex)] );
|
||
|
||
const double t_kp1 =
|
||
( fh[idx_fh_F_ord2(iF, jF - 2, kF + 1, ex)]
|
||
-F8*fh[idx_fh_F_ord2(iF, jF - 1, kF + 1, ex)]
|
||
+F8*fh[idx_fh_F_ord2(iF, jF + 1, kF + 1, ex)]
|
||
- fh[idx_fh_F_ord2(iF, jF + 2, kF + 1, ex)] );
|
||
|
||
const double t_kp2 =
|
||
( fh[idx_fh_F_ord2(iF, jF - 2, kF + 2, ex)]
|
||
-F8*fh[idx_fh_F_ord2(iF, jF - 1, kF + 2, ex)]
|
||
+F8*fh[idx_fh_F_ord2(iF, jF + 1, kF + 2, ex)]
|
||
- fh[idx_fh_F_ord2(iF, jF + 2, kF + 2, ex)] );
|
||
|
||
fyz[p] = Fdydz * ( t_km2 - F8 * t_km1 + F8 * t_kp1 - t_kp2 );
|
||
}
|
||
}
|
||
/* 二阶分支：i±1,j±1,k±1 在范围内 */
|
||
else if ((iF + 1) <= imaxF && (iF - 1) >= iminF &&
|
||
(jF + 1) <= jmaxF && (jF - 1) >= jminF &&
|
||
(kF + 1) <= kmaxF && (kF - 1) >= kminF)
|
||
{
|
||
fxx[p] = Sdxdx * (
|
||
fh[idx_fh_F_ord2(iF - 1, jF, kF, ex)] -
|
||
TWO * fh[idx_fh_F_ord2(iF, jF, kF, ex)] +
|
||
fh[idx_fh_F_ord2(iF + 1, jF, kF, ex)]
|
||
);
|
||
|
||
fyy[p] = Sdydy * (
|
||
fh[idx_fh_F_ord2(iF, jF - 1, kF, ex)] -
|
||
TWO * fh[idx_fh_F_ord2(iF, jF, kF, ex)] +
|
||
fh[idx_fh_F_ord2(iF, jF + 1, kF, ex)]
|
||
);
|
||
|
||
fzz[p] = Sdzdz * (
|
||
fh[idx_fh_F_ord2(iF, jF, kF - 1, ex)] -
|
||
TWO * fh[idx_fh_F_ord2(iF, jF, kF, ex)] +
|
||
fh[idx_fh_F_ord2(iF, jF, kF + 1, ex)]
|
||
);
|
||
|
||
fxy[p] = Sdxdy * (
|
||
fh[idx_fh_F_ord2(iF - 1, jF - 1, kF, ex)] -
|
||
fh[idx_fh_F_ord2(iF + 1, jF - 1, kF, ex)] -
|
||
fh[idx_fh_F_ord2(iF - 1, jF + 1, kF, ex)] +
|
||
fh[idx_fh_F_ord2(iF + 1, jF + 1, kF, ex)]
|
||
);
|
||
|
||
fxz[p] = Sdxdz * (
|
||
fh[idx_fh_F_ord2(iF - 1, jF, kF - 1, ex)] -
|
||
fh[idx_fh_F_ord2(iF + 1, jF, kF - 1, ex)] -
|
||
fh[idx_fh_F_ord2(iF - 1, jF, kF + 1, ex)] +
|
||
fh[idx_fh_F_ord2(iF + 1, jF, kF + 1, ex)]
|
||
);
|
||
|
||
fyz[p] = Sdydz * (
|
||
fh[idx_fh_F_ord2(iF, jF - 1, kF - 1, ex)] -
|
||
fh[idx_fh_F_ord2(iF, jF + 1, kF - 1, ex)] -
|
||
fh[idx_fh_F_ord2(iF, jF - 1, kF + 1, ex)] +
|
||
fh[idx_fh_F_ord2(iF, jF + 1, kF + 1, ex)]
|
||
);
|
||
}
|
||
else {
|
||
fxx[p] = 0.0; fyy[p] = 0.0; fzz[p] = 0.0;
|
||
fxy[p] = 0.0; fxz[p] = 0.0; fyz[p] = 0.0;
|
||
}
|
||
} |