wave/transport.py

"""
This code aims to solve the transport equation through the finite difference
method.
"""
import numpy as np

# Parameters
E=7.8e10
rho=2700
c=np.sqrt(E/rho)
T=1.0
L=1.0
# Discretization
x_step=e3
t_step=e3
dx=L/x_step
dt=T/t_step
r=c*dt/dx

# Grid
x=np.linspace(0,L,x_step)
t=np.linspace(0,T,t_step)

# Initial conditions
u=np.zeros((x_step,t_step))
u[:,0]=np.exp(-((x-0.5)**2)/0.1)