Rendering of advection-diffusion with pillow

.gitignore

@@ -28,3 +28,8 @@ demos/NavierStokes/data/*
 !demos/NavierStokes/data/README.md
 !demos/NavierStokes/data/Comparison
 !demos/NavierStokes/data/TempFolder/*
+
+demos/Advection/data/*
+!demos/Advection/data/README.md
+!demos/Advection/data/Comparison
+!demos/Advection/data/TempFolder/*

demos/Advection/00_advection_with_rbf.py

@@ -28,15 +28,15 @@
 MAX_DEGREE = 1
 
 DT = 1e-4
-NB_TIMESTEPS = 10
+NB_TIMESTEPS = 100
 PLOT_EVERY = 10
 
 VEL = jnp.array([100.0, 0.0])
 ## Diffusive constant
 K = 0.08
 
-Nx = 30
-Ny = Nx
+Nx = 40
+Ny = 20
 SUPPORT_SIZE = "max"
 
 facet_types={"South":"d", "West":"d", "North":"d", "East":"n"}
@@ -64,8 +64,8 @@ def my_rhs_operator(x, centers=None, rbf=None, fields=None):
 
 ## u0 is zero everywhere except at a point in the middle
 u0 = jnp.zeros(cloud.N)
-source_id = int(cloud.N*0.05)
-source_neighbors = jnp.array(cloud.local_supports[source_id][:cloud.N//50])
+source_id = int(cloud.N*0.01)
+source_neighbors = jnp.array(cloud.local_supports[source_id][:cloud.N//60])
 u0 = u0.at[source_neighbors].set(0.95)
 
 ## Begin timestepping for 100 steps
@@ -112,8 +112,8 @@ def my_rhs_operator(x, centers=None, rbf=None, fields=None):
 # %%
 
 ax = plt.gca()
-filename = DATAFOLDER + "advection_diffusion_rbf.png"
-cloud.animate_fields([ulist], cmaps="jet", filename=filename, figsize=(6,3),colorbar=False, title=["Implicit Advection-Diffusion with RBFs"])
+filename = DATAFOLDER + "advection_diffusion_rbf.gif"
+cloud.animate_fields([ulist], cmaps="jet", filename=filename, figsize=(6,3),colorbar=False, titles=["Implicit scheme for Advection-Diffusion with RBFs"])
 
 
 

updec/cloud.py

@@ -306,7 +306,8 @@ def animate(frame):
         ### Save the video
         if filename:
             fps = step_count / duration
-            anim.save(filename, writer='ffmpeg', fps=fps)
+            # anim.save(filename, writer='ffmpeg', fps=fps)
+            anim.save(filename, writer='pillow', fps=fps)
             os.system("open "+filename)
 
         return ax