Skip to content

Commit 6d062cf

Browse files
committed
Adding missing files
1 parent 4c28a0b commit 6d062cf

File tree

5 files changed

+23
-0
lines changed

5 files changed

+23
-0
lines changed

cython/heat-equation/bottle.dat

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
../../numpy/heat-equation/bottle.dat
Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
../../numpy/heat-equation/bottle_large.dat
Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
../../numpy/heat-equation/bottle_medium.dat

cython/heat-equation/evolve.py

Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,20 @@
1+
import numpy as np
2+
3+
def evolve(u, u_previous, a, dt, dx2, dy2):
4+
"""Explicit time evolution.
5+
u: new temperature field
6+
u_previous: previous field
7+
a: diffusion constant
8+
dt: time step. """
9+
10+
n, m = u.shape
11+
12+
for i in range(1, n-1):
13+
for j in range(1, m-1):
14+
u[i, j] = u_previous[i, j] + a * dt * ( \
15+
(u_previous[i+1, j] - 2*u_previous[i, j] + \
16+
u_previous[i-1, j]) / dx2 + \
17+
(u_previous[i, j+1] - 2*u_previous[i, j] + \
18+
u_previous[i, j-1]) / dy2 )
19+
u_previous[:] = u[:]
20+

0 commit comments

Comments
 (0)