change parameters in HW example

examples/10_HasegawaWakatani/10_hasegawa_wakatani.cpp

@@ -353,7 +353,7 @@ class HasegawaWakatani {
   double m_norm_coef;
   double m_dt = 0.0, m_time = 0.0;
   int m_nkx2, m_nkyh, m_nx, m_ny;
-  int m_diag_it = 0, m_diag_steps = 1000;
+  int m_diag_it = 0, m_diag_steps = 5000;
 
   //! The directory to output diagnostic data.
   std::string m_out_dir;
@@ -735,8 +735,8 @@ int main(int argc, char* argv[]) {
   Kokkos::ScopeGuard guard(argc, argv);
   auto kwargs         = IO::parse_args(argc, argv);
   std::string out_dir = IO::get_arg(kwargs, "out_dir", "data_kokkos");
-  int nx = 1024, nbiter = 100000;
-  double lx = 10.0, dt = 0.0005;
+  int nx = 1024, nbiter = 1000000;
+  double lx = 10.0, dt = 0.00025;
   HasegawaWakatani model(nx, lx, nbiter, dt, out_dir);
   Kokkos::Timer timer;
   model.run();

examples/10_HasegawaWakatani/postscript.py

@@ -222,7 +222,7 @@ def save_fig(iter: int) -> None:
             if iter >= nb_iters:
                 return
             var = data_loader.load(var_name=var_name, iter=iter)
-            fig, ax = plt.subplots(figsize=(12,10))
+            fig, ax = plt.subplots(figsize=(12,10), subplot_kw={'xticks':[], 'yticks':[]})
             pmesh = ax.pcolormesh(x, y, var, cmap='twilight', vmin=-vmax, vmax=vmax)
             ax.set_title(r'Time $t = {:03d}$'.format(iter), **title_font)
             ax.set_xlabel(r'$x$', **axis_font)
@@ -246,7 +246,7 @@ def save_fig(iter: int) -> None:
     parser.add_argument('-n_jobs', nargs='?', type=int, default=8)
     args = parser.parse_args()
 
-    var_dict = dict(phi= 0.0005, density=0.0005, vorticity=0.0005)
+    var_dict = dict(phi= 2.e-5, density=2.e-5, vorticity=2.e-5)
 
     # Plot settings
     fontname = 'Times New Roman'