Fixed a bug

burgers_filtered.py

@@ -80,7 +80,7 @@ def burgers(t, u_hat, N, M, filter, a=1):
     kk = freqs(NN)
     nonlinear = -1 * fft(u**2/2) * 1j * kk
     nonlinear *= a
-#    nonlinear *= filter
+    nonlinear *= filter
     return unpad(mask(nonlinear), M//2)
 def semigroup_heat(dt, k, eps):
     S_half = sparse.diags(np.exp(-1 * eps * k**2 * dt / 2))
@@ -104,7 +104,7 @@ def semigroup_none(dt, k, eps):
 fig.tight_layout()
 god = np.loadtxt("burg3_GOD_3.txt").transpose()
 # ax[0].plot(god[0] * 2 * np.pi, god[1], 'ko', markersize=0.1, label="Godunov flux")
-for i in range(6):
+for i in range(8):
     N = np.power(2, i + 4);
     M = 3 * N // 2;
     NN = (2 * M//2) + N

common.py

@@ -2,7 +2,7 @@
 
 def elrk4(SemiGroup,Nonlinear,y0,tinterval,dt,args):
     y = y0
-    t, tfinal = tinterval
+    t, tf = tinterval
     time = [t,]
     solution = [y,]
     flag = True
@@ -48,14 +48,14 @@ def elrk4(SemiGroup,Nonlinear,y0,tinterval,dt,args):
         # solution.append(y)
 
         time.append(t)
-        if t + dt > tfinal:
-            dt = tfinal - t
-        elif (t >= tfinal):
+        if t + dt > tf:
+            dt = tf - t
+        elif (t >= tf):
             flag = False
         elif np.any(np.isnan(y)):
             flag = False
         print("t, dt = ", t, dt)
 
     times = np.array(time)
-#   solutions = np.array(solution)
+    solution.append(y)
     return times, solution