misc: Noodling

Author: JDBetteridge

examples/timestepping/acoustic_superstep_2d.py

@@ -224,8 +224,8 @@ def acoustic_model(model, step=1, snapshots=1):
                 ax.set_title(f't = {time[idx]:0.3f}')
                 idx += 1
                 if ii > 0:
-                    ax.set_xticks([])
-                    ax.set_yticks([])
+                    ax.set_xticklabels([])
+                    ax.set_yticklabels([])
                 else:
                     xticks = ax.get_xticks()
                     ax.set_xticks(np.array((

examples/timestepping/superstep_1d.py

@@ -75,6 +75,7 @@ def wave_on_string(step=1):
 
 if __name__ == '__main__':
     fig, ax = plt.subplots(1, 1)
+    fig.set_size_inches(8, 4)
     x = np.linspace(0, 1, *shape)
     ax.plot(
         x, gaussian(x, mu=mu, sigma_sq=sigma_sq),
@@ -88,4 +89,6 @@ def wave_on_string(step=1):
     ax.set_xlim(*xlim)
     ax.set_ylim(-ylim, ylim)
     ax.legend()
+
+    fig.savefig('1d_example.png', dpi=300)
     plt.show()

examples/timestepping/superstep_2d.py

@@ -89,6 +89,8 @@ def ripple_on_pond(step=1, snapshots=1):
     # Snapshots
     m = 13
     fig, axes = plt.subplots(len(k), m)
+    fig.set_size_inches(16, 5)
+
     for step, ax_row in zip(k, axes, strict=True):
         data = ripple_on_pond(step=step, snapshots=m)
         idx = 0
@@ -101,14 +103,22 @@ def ripple_on_pond(step=1, snapshots=1):
                     cmap='seismic'
                 )
                 idx += 1
+                if step == 1:
+                    ax.set_title(f't = {(ii*t1)/(m - 1) :0.3f}')
+                    if ii != 0:
+                        ax.set_yticklabels([])
+                    if ii % 2 == 1:
+                        ax.set_xticklabels([])
             else:
                 ax.remove()
     fig.subplots_adjust(
         left=0.05,
-        bottom=0.025,
+        bottom=0.02,
         right=0.99,
-        top=0.97,
-        wspace=0.06,
-        hspace=0.06
+        top=0.96,
+        wspace=0.23,
+        hspace=0.0
     )
+
+    fig.savefig('2d_example.png', dpi=300)
     plt.show()