Update pde.py

LuisGMM · May 6, 2022 · 4df50e0 · 4df50e0
1 parent 412f5ba
commit 4df50e0
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/numericalmethods/pde.py b/src/numericalmethods/pde.py
@@ -326,5 +326,11 @@ def hyperbolic(h: float, k: float, x0: float, xf: float, t0: float, tf: float, u
     sol[1:-1, 0] = u0(x)
 
     sol[1:-1, 1] = 1/2 * rho * (u0(x+h)[1:-1] - u0(x-h)[1:-1]) + (1 - rho) * u0(x)
+
+    for ti in range(2, LEN_T):
+        sol[1:-1, ti] = rho*sol[2:, ti] + 2*(1 - rho)*sol[:, ti] + rho*sol[:-2, ti]  - sol[:, ti-1]
+
+    return sol, x, t
+
 def elliptic(h: float, k: float, x0: float, xf: float, t0: float, tf: float, u0: Callable) -> Tuple[np.ndarray]:
     raise NotImplementedError()