Plotting changes

flood.py

@@ -299,7 +299,7 @@ def main():
     #print("u(A) = ")
     #pprint(u)
     
-    Q = (A_sym*u).subs([("g", 9.81), ("alpha", 0.001), ("f", 0.05), (a, 100)])
+    Q = (A_sym*u).subs([("g", 9.81), ("alpha", 0.001), ("f", 0.05), (a, 500)])
 
     delta_x = 0.5
     L = 300
@@ -334,8 +334,18 @@ def main():
     t = [i*delta_t for i in range(int(T/delta_t))]
     
     c_sln = godunov_solve(Q_lamb, A_init_data, A_inflow, delta_x, t)
+
+    n_xticks = 21
+    n_tticks = 11
+    x_ticks = [int(i*L/delta_x/(n_xticks-1)) for i in range(n_xticks)]
+    x_tick_labels = [delta_x*i for i in x_ticks]
+    t_ticks = [int(i*T/delta_t/(n_tticks-1)) for i in range(n_tticks)]
+    t_tick_labels = [delta_t*i for i in t_ticks]
     
+    plt.colorbar(plt.pcolor(c_sln))
     ax1.imshow(c_sln, origin="lower", aspect="auto")
+    ax1.set_xticks(x_ticks, x_tick_labels)
+    ax1.set_yticks(t_ticks, t_tick_labels)
 
     plt.show()