近一个月来,我一直在解决一个非常奇怪的错误。问你们是我的最后希望。我用C语言编写了一个程序,该程序使用傅立叶(或倒数)空间中的隐式欧拉(IE)方案集成了2d Cahn–Hilliard equation:
这里的“帽子”表示我们处于傅立叶空间:h_q(t_n + 1)和h_q(t_n)是在时间t_n和t_(n + 1)时h(x,y)的FT。 h_q]是在傅立叶空间中应用于h_q的非线性算子,而L_q是在傅立叶空间中也是线性算子。我不想过多地介绍正在使用的数值方法,因为我确信问题并非出自此(我尝试使用其他方案)。
我的代码实际上非常简单。这是开始,基本上是我声明变量,分配内存并为FFTW例程创建计划。
# include <stdlib.h>
# include <stdio.h>
# include <time.h>
# include <math.h>
# include <fftw3.h>
# define pi M_PI
int main(){
// define lattice size and spacing
int Nx = 150; // n of points on x
int Ny = 150; // n of points on y
double dx = 0.5; // bin size on x and y
// define simulation time and time step
long int Nt = 1000; // n of time steps
double dt = 0.5; // time step size
// number of frames to plot (at denominator)
long int nframes = Nt/100;
// define the noise
double rn, drift = 0.05; // punctual drift of h(x)
srand(666); // seed the RNG
// other variables
int i, j, nt; // variables for space and time loops
// declare FFTW3 routine
fftw_plan FT_h_hft; // routine to perform fourier transform
fftw_plan FT_Nonl_Nonlft;
fftw_plan IFT_hft_h; // routine to perform inverse fourier transform
// declare and allocate memory for real variables
double *Linft = fftw_alloc_real(Nx*Ny);
double *Q2 = fftw_alloc_real(Nx*Ny);
double *qx = fftw_alloc_real(Nx);
double *qy = fftw_alloc_real(Ny);
// declare and allocate memory for complex variables
fftw_complex *dh = fftw_alloc_complex(Nx*Ny);
fftw_complex *dhft = fftw_alloc_complex(Nx*Ny);
fftw_complex *Nonl = fftw_alloc_complex(Nx*Ny);
fftw_complex *Nonlft = fftw_alloc_complex(Nx*Ny);
// create the FFTW plans
FT_h_hft = fftw_plan_dft_2d ( Nx, Ny, dh, dhft, FFTW_FORWARD, FFTW_ESTIMATE );
FT_Nonl_Nonlft = fftw_plan_dft_2d ( Nx, Ny, Nonl, Nonlft, FFTW_FORWARD, FFTW_ESTIMATE );
IFT_hft_h = fftw_plan_dft_2d ( Nx, Ny, dhft, dh, FFTW_BACKWARD, FFTW_ESTIMATE );
// open file to store the data
char acstr[160];
FILE *fp;
sprintf(acstr, "CH2d_IE_dt%.2f_dx%.3f_Nt%ld_Nx%d_Ny%d_#f%.ld.dat",dt,dx,Nt,Nx,Ny,Nt/nframes);
在此序言之后,我用均匀的随机噪声初始化函数h(x,y),并对其进行FT运算。我将h(x,y)的虚部(在代码中为dh[i*Ny+j][1]
)设置为0,因为它是实函数。然后,我计算波向量qx
和qy
,并用它们在傅立叶空间中计算方程的线性算子,该线性算子在代码中为Linft
。我仅将h的-四阶导数作为线性项,因此线性项的FT只是-q ^ 4 ...,但我不想再详细说明我的积分方法。问题不关乎。
// generate h(x,y) at initial time
for ( i = 0; i < Nx; i++ ) {
for ( j = 0; j < Ny; j++ ) {
rn = (double) rand()/RAND_MAX; // extract a random number between 0 and 1
dh[i*Ny+j][0] = drift-2.0*drift*rn; // shift of +-drift
dh[i*Ny+j][1] = 0.0;
}
}
// execute plan for the first time
fftw_execute (FT_h_hft);
// calculate wavenumbers
for (i = 0; i < Nx; i++) { qx[i] = 2.0*i*pi/(Nx*dx); }
for (i = 0; i < Ny; i++) { qy[i] = 2.0*i*pi/(Ny*dx); }
for (i = 1; i < Nx/2; i++) { qx[Nx-i] = -qx[i]; }
for (i = 1; i < Ny/2; i++) { qy[Ny-i] = -qy[i]; }
// calculate the FT of the linear operator
for ( i = 0; i < Nx; i++ ) {
for ( j = 0; j < Ny; j++ ) {
Q2[i*Ny+j] = qx[i]*qx[i] + qy[j]*qy[j];
Linft[i*Ny+j] = -Q2[i*Ny+j]*Q2[i*Ny+j];
}
}
然后,最后是时间循环。本质上,我要做的是以下事情:
每隔一段时间,我会将数据保存到文件中,并在终端上打印一些信息。特别是,我打印了非线性项FT的最大值。我还要检查h(x,y)是否发散到无穷大(不应该发生!),
计算直接空间中的h ^ 3(即dh[i*Ny+j][0]*dh[i*Ny+j][0]*dh[i*Ny+j][0]
)。同样,虚部设置为0,
进行h ^ 3的傅立叶变换
通过计算-q ^ 2 *(FT [h ^ 3]-FT [h]),在倒数空间中获得完整的非线性项(即上述IE算法中的N [h_q])。在代码中,对于虚构部分,我指的是Nonlft[i*Ny+j][0] = -Q2[i*Ny+j]*(Nonlft[i*Ny+j][0] -dhft[i*Ny+j][0])
行和下面的行。我这样做是因为:
代码如下:
for(nt = 0; nt < Nt; nt++) {
if((nt % nframes)== 0) {
printf("%.0f %%\n",((double)nt/(double)Nt)*100);
printf("Nonlft %.15f \n",Nonlft[(Nx/2)*(Ny/2)][0]);
// write data to file
fp = fopen(acstr,"a");
for ( i = 0; i < Nx; i++ ) {
for ( j = 0; j < Ny; j++ ) {
fprintf(fp, "%4d %4d %.6f\n", i, j, dh[i*Ny+j][0]);
}
}
fclose(fp);
}
// check if h is going to infinity
if (isnan(dh[1][0])!=0) {
printf("crashed!\n");
return 0;
}
// calculate nonlinear term h^3 in direct space
for ( i = 0; i < Nx; i++ ) {
for ( j = 0; j < Ny; j++ ) {
Nonl[i*Ny+j][0] = dh[i*Ny+j][0]*dh[i*Ny+j][0]*dh[i*Ny+j][0];
Nonl[i*Ny+j][1] = 0.0;
}
}
// Fourier transform of nonlinear term
fftw_execute (FT_Nonl_Nonlft);
// second derivative in Fourier space is just multiplication by -q^2
for ( i = 0; i < Nx; i++ ) {
for ( j = 0; j < Ny; j++ ) {
Nonlft[i*Ny+j][0] = -Q2[i*Ny+j]*(Nonlft[i*Ny+j][0] -dhft[i*Ny+j][0]);
Nonlft[i*Ny+j][1] = -Q2[i*Ny+j]*(Nonlft[i*Ny+j][1] -dhft[i*Ny+j][1]);
}
}
// Implicit Euler scheme in Fourier space
for ( i = 0; i < Nx; i++ ) {
for ( j = 0; j < Ny; j++ ) {
dhft[i*Ny+j][0] = (dhft[i*Ny+j][0] + dt*Nonlft[i*Ny+j][0])/(1.0 - dt*Linft[i*Ny+j]);
dhft[i*Ny+j][1] = (dhft[i*Ny+j][1] + dt*Nonlft[i*Ny+j][1])/(1.0 - dt*Linft[i*Ny+j]);
}
}
// transform h back in direct space
fftw_execute (IFT_hft_h);
// normalize
for ( i = 0; i < Nx; i++ ) {
for ( j = 0; j < Ny; j++ ) {
dh[i*Ny+j][0] = dh[i*Ny+j][0] / (double) (Nx*Ny);
dh[i*Ny+j][1] = dh[i*Ny+j][1] / (double) (Nx*Ny);
}
}
}
代码的最后一部分:清空内存并销毁FFTW计划。
// terminate the FFTW3 plan and free memory
fftw_destroy_plan (FT_h_hft);
fftw_destroy_plan (FT_Nonl_Nonlft);
fftw_destroy_plan (IFT_hft_h);
fftw_cleanup();
fftw_free(dh);
fftw_free(Nonl);
fftw_free(qx);
fftw_free(qy);
fftw_free(Q2);
fftw_free(Linft);
fftw_free(dhft);
fftw_free(Nonlft);
return 0;
}
如果运行此代码,则会获得以下输出:
0 %
Nonlft 0.0000000000000000000
1 %
Nonlft -0.0000000000001353512
2 %
Nonlft -0.0000000000000115539
3 %
Nonlft 0.0000000001376379599
...
69 %
Nonlft -12.1987455309071730625
70 %
Nonlft -70.1631962517720353389
71 %
Nonlft -252.4941743351609204637
72 %
Nonlft 347.5067875825179726235
73 %
Nonlft 109.3351142318568633982
74 %
Nonlft 39933.1054502610786585137
crashed!
代码在到达末尾之前便崩溃了,我们可以看到非线性项在发散。
现在,对我来说没有意义的是,如果我以以下方式更改计算非线性项的FT的行,则该行:
// calculate nonlinear term h^3 -h in direct space
for ( i = 0; i < Nx; i++ ) {
for ( j = 0; j < Ny; j++ ) {
Nonl[i*Ny+j][0] = dh[i*Ny+j][0]*dh[i*Ny+j][0]*dh[i*Ny+j][0] -dh[i*Ny+j][0];
Nonl[i*Ny+j][1] = 0.0;
}
}
// Fourier transform of nonlinear term
fftw_execute (FT_Nonl_Nonlft);
// second derivative in Fourier space is just multiplication by -q^2
for ( i = 0; i < Nx; i++ ) {
for ( j = 0; j < Ny; j++ ) {
Nonlft[i*Ny+j][0] = -Q2[i*Ny+j]* Nonlft[i*Ny+j][0];
Nonlft[i*Ny+j][1] = -Q2[i*Ny+j]* Nonlft[i*Ny+j][1];
}
}
这意味着我正在使用此定义:
代替这个:
然后,代码是完全稳定的,不会发生差异!即使是数十亿的时间步长! 为什么会这样,因为计算Nonlft
的两种方法应该等效?
非常感谢任何愿意花时间阅读所有内容并给我一些帮助的人!
编辑:为了使事情变得更奇怪,我应该指出,此错误不会在一维的同一系统上发生。在1D中,两种计算Nonlft
的方法都是稳定的。
编辑:我添加了一个简短的动画,说明崩溃前函数h(x,y)发生了什么。另外:我很快在MATLAB中重新编写了代码,该代码使用了基于FFTW库的快速傅立叶变换函数,并且该错误没有发生……谜底加深了。
答案 0 :(得分:22)
我解决了!!
问题在于Nonl
项的计算:
Nonl[i*Ny+j][0] = dh[i*Ny+j][0]*dh[i*Ny+j][0]*dh[i*Ny+j][0];
Nonl[i*Ny+j][1] = 0.0;
需要更改为:
Nonl[i*Ny+j][0] = dh[i*Ny+j][0]*dh[i*Ny+j][0]*dh[i*Ny+j][0] -3.0*dh[i*Ny+j][0]*dh[i*Ny+j][1]*dh[i*Ny+j][1];
Nonl[i*Ny+j][1] = -dh[i*Ny+j][1]*dh[i*Ny+j][1]*dh[i*Ny+j][1] +3.0*dh[i*Ny+j][0]*dh[i*Ny+j][0]*dh[i*Ny+j][1];
换句话说:我需要将dh
视为一个复杂的函数(即使它应该是实函数)。
基本上,由于愚蠢的舍入错误,实函数的FT的IFT (在我的情况下为dh
),不是纯实的,但只有很少的假想部分。通过设置Nonl[i*Ny+j][1] = 0.0
,我完全忽略了这个虚构部分。
那么,问题是我递归求和FT({dh
),dhft
和使用IFT(FT({dh
))获得的对象,这就是{{1} },但忽略了剩余的虚部!
Nonlft
很明显,将Nonlft[i*Ny+j][0] = -Q2[i*Ny+j]*(Nonlft[i*Ny+j][0] -dhft[i*Ny+j][0]);
Nonlft[i*Ny+j][1] = -Q2[i*Ny+j]*(Nonlft[i*Ny+j][1] -dhft[i*Ny+j][1]);
计算为Nonlft
^ 3 dh
,然后进行
-dh
避免了执行此“混合”总和的问题。
哎呀,真是太好了!我希望我可以将赏金分配给我自己! :P
编辑:我想补充一点,在使用Nonlft[i*Ny+j][0] = -Q2[i*Ny+j]* Nonlft[i*Ny+j][0];
Nonlft[i*Ny+j][1] = -Q2[i*Ny+j]* Nonlft[i*Ny+j][1];
函数之前,我先使用过fftw_plan_dft_2d
和fftw_plan_dft_r2c_2d
(从实到复杂和从实到实) ,而我也看到了相同的错误。但是,我想如果不切换到fftw_plan_dft_c2r_2d
就无法解决问题,因为fftw_plan_dft_2d
函数会自动“切除”来自IFT的剩余虚部。 如果是这种情况,但我没有遗漏任何东西,我认为这应该写在FFTW网站上的某处,以防止用户遇到此类问题。类似“ {{1} }和c2r
转换不利于实现伪光谱方法”。
编辑:我发现another SO question解决了完全相同的问题。