get_matrix.m

来自「五点差分型多重网格方法：各种插值算子的比较）」· M 代码 · 共 46 行

% GET_MATRIX Generate a discrete linear opererator
%
%       [A,N] = GET_MATRIX(NX,NY) generates the matrix A of order N for
%       the problem defined by the global flag "problem_flag" and global
%       parameters "prob_args".

function [A,N] = get_matrix(nx,ny)

include_flags
include_globals

tic

%fprintf('Generating %g by %g matrix...',nx*ny,nx*ny)

if (problem_flag == POISSON)

   [A,N] = sp_laplace(nx,ny);

elseif (problem_flag == CUT_SQUARE)

   val=prob_args(1);
   [A,N] = sp_cutsq2d (nx,ny,val);

elseif (problem_flag == HELMHOLTZ)

   lambda=0;
   sigma=prob_args(1);
   [A,N] = sp_convdiff (nx,ny,lambda,sigma);

elseif (problem_flag == POISSON_BOLTZMAN)

   disp(sprintf('POISSON_BOLTZMAN not implemented yet'));

elseif (problem_flag == CONVECT_DIFFUSE)

   lambda=prob_args(1);
   sigma=prob_args(2);
   [A,N] = sp_convdiff (nx,ny,lambda,sigma);

end

toc1=toc;

%fprintf(' %g seconds.\n',toc1)