我一直在努力编写一些小的测试代码,以将数据写入netcdf格式文件。该代码应写入两个具有相同值的变量,但内存分配不同。代码如下:
int main(int argc, char** argv)
{
int i, j, k;
char filen2[80];
int ncid, err, cmode;
int dimid[2],varid[2];
printf("Creating Netcdf Output file\n");
int mx=5, my=5, mz=5;
printf("mx my mz: %d %d %d %i \n", mx , my, mz, rank);
sprintf(filen2, "test.nc");
printf("Writing file: %s \n",filen2);
cmode = NC_CLOBBER;
err = nc_create(filen2, cmode, &ncid); ERR
/* free info object */
if (MInfo != MPI_INFO_NULL) MPI_Info_free(&MInfo);
/* define dimensions */
err = nc_def_dim(ncid, "MZ", mz, &dimid[0]); ERR
err = nc_def_dim(ncid, "MY", my, &dimid[1]); ERR
err = nc_def_var(ncid, "Z", NC_DOUBLE, 2, dimid, &varid[0]); ERR
err = nc_def_var(ncid, "Y", NC_DOUBLE, 2, dimid, &varid[1]); ERR
/* exit define mode */
err = nc_enddef(ncid); ERR
size_t start2[2], count2[2];
start2[0] = 0;
start2[1] = 0;
count2[0] = mz;
count2[1] = my;
printf("Start: %d %d %i \n", start2[0], start2[1], rank);
printf("Count: %d %d %i \n", count[0], count[1], rank);
double **buffertmp2=NULL;
double buffertmp3[mz][my];
buffertmp2 = (double **) malloc( sizeof(double *) * mz);
for(k=0; k<mz; k++){
buffertmp2[k] = (double *) malloc( sizeof(double) * my);
for(j=0; j<my; j++){
buffertmp2[k][j] = k;
buffertmp3[k][j] = k;
}
}
err = nc_put_vara_double(ncid, varid[0], start2, count2, &buffertmp2[0][0]); ERR
err = nc_put_vara_double(ncid, varid[1], start2, count2, &buffertmp3[0][0]); ERR
err = nc_close(ncid); ERR
for(k=0; k<mz; k++){
for(j=0; j<my; j++){
printf("bff2 %le \n", buffertmp2[k][j]);
}
}
for(k=0; k<mz; k++){
for(j=0; j<my; j++){
printf("bff3 %le \n", buffertmp3[k][j]);
} }
printf("The End \n");
}
在MATLAB上读取的输出给出:
X =
0 0 1.0000 0.0000 2.0000
0 0 1.0000 2.0000 0
0 0.0000 1.0000 2.0000 0
0 1.0000 0 2.0000 0.0000
0 1.0000 0 2.0000 3.0000
Y =
0 1 2 3 4
0 1 2 3 4
0 1 2 3 4
0 1 2 3 4
0 1 2 3 4
Y的输出实际上是我所期望的,但是X的输出是完全错误的。我想知道是否有人可以指出内存分配如何更改整个输出。如果使用与for buffertmp2一样的for循环进行内存分配,而不必创建中间变量,该怎么办才能将数据写入netcdf文件。
谢谢克里斯
我不确定这是正确的答案,但是您应该注意,buffertmp2
和buffertmp3
是两个完全不同的数据结构。
buffertmp3
是包含二维数组的内存的一大块。
buffertmp2
是指向行数据的指针(行)的数组。
[将这些不同的数据结构传递给Matlab(或netcdf函数)时,应该知道您正在传递不同的数据结构。