Navier-Stokes plots and next release

README.md

@@ -7,7 +7,7 @@
     <th><img src="docs/assets/laplace.png" width="180"></th>
     <th><img src="docs/assets/adv_diff_periodic.gif" width="350"></th>
     <!-- <th><img src="docs/assets/burgers_u.gif" width="188"></th> -->
-    <th><img src="docs/assets/ns_norm.gif" width="188"></th>
+    <th><img src="docs/assets/ns_norm.gif" width="150"></th>
   </tr>
 </table>
 

demos/NavierStokes/001_unit_square_rotation.py

@@ -29,7 +29,7 @@
 # RBF = gaussian
 MAX_DEGREE = 1
 
-DT = 1e-3
+DT = 4e-3
 NB_TIMESTEPS = 200
 PLOT_EVERY = 20
 
@@ -38,8 +38,8 @@
 
 ## Diffusive constant
 
-Nx = 20
-Ny = 20
+Nx = 40
+Ny = 40
 SUPPORT_SIZE = "max"
 
 facet_types_vel={"South":"d", "North":"d", "West":"d", "East":"d"}
@@ -213,8 +213,9 @@ def Phi_x_Phi_y(x, y):
 # cloud_vel.animate_fields([ulist], cmaps="coolwarm", filename=filename, levels=200, duration=10, figsize=(7.5,6), titles=["Burgers with RBFs - u"]);
 
 normlist = [jnp.sqrt(sol[0]**2 + sol[1]**2) for sol in sol_list]
+# normlist = [sol[0] for sol in sol_list]
 filename = DATAFOLDER + "ns_norm.gif"
-cloud_vel.animate_fields([normlist[::10]], cmaps="coolwarm", filename=filename, levels=200, duration=10, figsize=(7.5,6), titles=["Navier-Stokes with RBFs"]);
+cloud_vel.animate_fields([normlist[::10]], cmaps="coolwarm", filename=filename, levels=200, duration=10, figsize=(6,6), titles=["Navier-Stokes with RBFs"], colorbar=False);
 
 
 # %%

docs/assets/NextRelease.md

@@ -0,0 +1,2 @@
+For the next release v1.1.0
+- [X] Added colorbar to animate fields

updes/cloud.py

@@ -269,7 +269,7 @@ def visualize_field(self, field, projection="2d", title="Field", xlabel=r'$x$',
         return ax, img
 
 
-    def animate_fields(self, fields, filename=None, titles="Field", xlabel=r'$x$', ylabel=r'$y$', levels=50, figsize=(6,5), cmaps="jet", cbarsplit=50, duration=5, **kwargs):
+    def animate_fields(self, fields, filename=None, titles="Field", xlabel=r'$x$', ylabel=r'$y$', levels=50, figsize=(6,5), cmaps="jet", cbarsplit=50, duration=5, colorbar=True, **kwargs):
         import matplotlib as mpl
         mpl.use('Agg')
         import matplotlib.pyplot as plt
@@ -307,15 +307,16 @@ def animate_fields(self, fields, filename=None, titles="Field", xlabel=r'$x$', y
                 minmax = minmax[0], minmax[0]+1e-3
 
             minmaxs.append(minmax)
-            boundaries = jnp.linspace(minmax[0], minmax[1], cbarsplit)
 
             imgs.append(ax[i].tricontourf(x, y, signals[i][0], levels=levels, vmin=minmax[0], vmax=minmax[1], cmap=cmaps[i], **kwargs))
 
-            m = plt.cm.ScalarMappable(cmap=cmaps[i])
-            m.set_array(signals[i])
-            m.set_clim(minmax[0], minmax[1])
-            # m.set_norm(plt.Normalize(vmin=minmax[0], vmax=minmax[1]))
-            plt.colorbar(m, boundaries=boundaries, shrink=1.0, aspect=10, ax=ax[i])
+            if colorbar == True:
+                boundaries = jnp.linspace(minmax[0], minmax[1], cbarsplit)
+                m = plt.cm.ScalarMappable(cmap=cmaps[i])
+                m.set_array(signals[i])
+                m.set_clim(minmax[0], minmax[1])
+                # m.set_norm(plt.Normalize(vmin=minmax[0], vmax=minmax[1]))
+                plt.colorbar(m, boundaries=boundaries, shrink=1.0, aspect=10, ax=ax[i])
 
             try:
                 title = titles[i]