jyang526843 / Simple-Wave-Equation-solver

An example of solving a wave equation using finite difference

Finite differences for the wave equation

Implementation of a simple numerical schemes for the wave equation.

Applying the second-order centered differences to approximate the derivatives,

It can be arrange to obtain

where

It is known that the scheme needs to satisfy for stability. Copyright 2015-2016 The MathWorks, Inc.

Proboem Setup

N = 101;
L = 4*pi;
x = linspace(0,L,N);

% It has three data set; 1: past, 2: current, 3: future.
u = zeros(N,3);
s = 0.5;

Initial Condition

% Gaussian Pulse
y = 2*exp(-(x-L/2).^2);
u(:,1) = y;
u(:,2) = y;

% Plot the initial condition.
handle_line = plot(x,u(:,2),'LineWidth',2);
axis([0,L,-2,2]);
xlabel('x'); ylabel('u');
title('Wave equation');

Boundary condition

% Dirichet Boundary conditions
u(1,:) = 0;
u(end,:) = 0;

Simulation

filename = 'wave.gif';
for ii=1:100
    disp(['at ii= ', num2str(ii)]);
    u(2:end-1,3) = s*(u(3:end,2)+u(1:end-2,2)) ...
        + 2*(1-s)*u(2:end-1,2) ...
        - u(2:end-1,1);
    u(:,1) = u(:,2);
    u(:,2) = u(:,3);
    handle_line.YData = u(:,2);
    drawnow;
    frame = getframe(gcf);
    im = frame2im(frame);
    [A,map] = rgb2ind(im,256);
    if ii==1
           imwrite(A,map,filename,'gif','LoopCount',Inf,'DelayTime',0.05);
    else
        imwrite(A,map,filename,'gif','WriteMode','append','DelayTime',0.05);
    end
end

Copyright 2015-2016 The MathWorks, Inc.