b.h

来自「二维热能方程的求解」· C头文件 代码 · 共 43 行

ATYPE *K4,*A3,*A5,*b,*x,*src,*ws;

x  =new ATYPE[nx*ny*2]; assert(x  != 0);
b  =new ATYPE[nx*ny];   assert(b  != 0);
src=new ATYPE[nx*ny*2]; assert(src!= 0);

if(idsol<=2)
   {K4 =new ATYPE[nx*ny*4]; assert(K4 != 0);}
else
   {K4 =new ATYPE[1]; assert(K4 != 0);}

if(idsol==1){
   A3=new ATYPE[1]; assert(A3!= 0);
   A5=new ATYPE[nx*ny*(2*nx+1)]; assert(A5!= 0);
   ws=new ATYPE[nx*ny]; assert(ws!= 0);
  }
else if(idsol==2){
   A3=new ATYPE[1]; assert(A3!= 0);
   A5=new ATYPE[(nx*ny*5)*2]; assert(A5!= 0);
  }
else if(idsol==3){
   A3=new ATYPE[(nx*ny*3)*2]; assert(A3!= 0);
   A5=new ATYPE[1]; assert(A5!= 0);
   ws=new ATYPE[nx*ny*2]; assert(ws!= 0);
  }
else if(idsol==4 || idsol==5){ // ADI-II || ADI-II-SOR
   A3=new ATYPE[(nx*ny*3)*2]; assert(A3!= 0);
   A5=new ATYPE[nx*ny*5]; assert(A5!= 0);
   ws=new ATYPE[nx*ny*3]; assert(ws!= 0);
  }
else{
   cout<<"No solver defined for idsol="<<idsol<<endl;
   exit(1);
  }

heat_trans(nt,nx,ny,itmax,idELL,idsol,id_nl,idpcond,level,maxADI,maxPCG,maxGS,
      idSGS,ndiff,nonlin,nreact,nbc,ntrue,idpsrc,n1,n2,interpol,id_hsrc,idsymm,
      &at,&bt,&ax,&bx,&ay,&by,&tol,&wavet,&wavex,&wavey,&tsbgn,&tsend,&eta,
      K4,A3,A5,A,x,b,src,wksp,ws,HEATSRC,&ierr);

// #include C.h