rec.h.txt

来自「FDTD的并行算法」· 文本 代码 · 共 58 行

void rec_data(float **edata, FILE *fpgnu,int rank,float **eframe,int nxo,MPI_Status status,int size, int nx, int nx_size[], int n)
{
	int 	i,j,k;
	FILE	*fp_snap;
	char	filename[32];

	
	for(k=1;k<=(size-1);k++)
	{
		nx=nx_size[k];
		int s;
		s=nx*ny;
		float *recbuf;
		recbuf=(float*)calloc(s,sizeof(float));
		MPI_Recv(&recbuf,s,MPI_FLOAT,k,22,MPI_COMM_WORLD,&status);			
	
	s=0;		
	for(i=0;i<k;i++)
		s+=nx_size[i];

	for(i=1;i<=nx;i++)
	{
	for(j=1;j<=ny;j++)
		{
			eframe[(i-1)+s][j-1]=recbuf[(i-1)+(j-1)*nx];
		}
	}
	}

	for(i=1;i<=nx_size[0];i++)
	{
	for(j=1;j<=ny;j++)
	{
		eframe[i-1][j-1]=edata[i][j];
	}	
	}
	
		sprintf(filename,"ef-%04d.fd",n);
		fprintf(fpgnu,"set view 10,60,,\nspl '%s\n",filename);
		
		fp_snap=fopen(filename,"w");
		
		for(i=1;i<=nxo;i++)
		{
			for(j=1;j<=ny;j++)
			{
				fprintf(fp_snap,"%d %d %e\n",i,j,eframe[i-1][j-1]);		
			}
			fprintf(fp_snap,"\n");
		}
		
		fclose(fp_snap);

	

}