cranknicolson.m

《实用化工计算机模拟：MATLAB在化学工程中的应用 》这本书光盘里的程序~

function U = CrankNicolson(f,c1,c2,a,b,alpha,n,m)
% 使用Crank-Nicolson有限差分方法求解一维动态传热模型
% Crank-Nicolson Method for the heat equation: du(x,t)/dt = alpha * d2u(x,t)/dx2 over R={(x,t):0<=x<=a,
%   0<=t<=b} with u(x,0)=f(x), for 0<=x<=a, and u(0,t)=c1,u(a,t)=c2,for 0<=t<=b.
%
% Input     - f=u(x,0) as a string 'f'
%           - c1=u(0,t) and c2=u(a,t)
%           - a and b right end points of [0,a] and [0,b]
%           - c the constant in the heat equation
%           - n and m number of grid points over [0,a] and [0,b]
% Output    - U solution matrix

% Initialize parameters and U
h = a/(n-1);
k = b/(m-1);
r = alpha*k/h^2;
s1 = 2+2/r;
s2 = 2/r-2;
U = zeros(n,m);

% Boundary conditions
U(1,1:m) = c1;
U(n,1:m) = c2;

% Generate first row (that is, the first column in matrix U)
U(2:n-1,1) = feval(f, h:h:(n-2)*h)';

% Form the diagonal and off-diagonal elements of A and
% the constant vector B and solve tridiagonal system AX=B
Vd(1,1:n) = s1*ones(1,n);
Vd(1) = 1;
Vd(n) = 1;
Va = -ones(1,n-1);
Va(n-1) = 0;
Vc = -ones(1,n-1);
Vc(1) = 0;
Vb(1) = c1;
Vb(n) = c2;
for j=2:m
    for i=2:n-1
        Vb(i) = U(i-1,j-1)+U(i+1,j-1)+s2*U(i,j-1);
    end
    X = TDMA(Va,Vd,Vc,Vb);
    U(1:n,j)=X';
end
U = U'