add validation to swm_numpy.ipynb

Author: cenamiller

swm_python/swm_numpy.ipynb

@@ -35,6 +35,7 @@
         "L_OUT = True # args.L_OUT\n",
         "VIS = True\n",
         "VIS_DT=100\n",
+        "VAL = True\n",
         "ITMAX = 1000\n",
         "dt = 90.\n",
         "tdt = dt\n",
@@ -98,6 +99,28 @@
         "    fig.suptitle(title)\n",
         "    #plt.xlabel('x')\n",
         "    #plt.ylabel('y')\n",
+        "    plt.show()\n",
+        "\n",
+        "def live_plot_val(fu, fv, fp, title=''):\n",
+        "    mxu = fu.max()\n",
+        "    mxv = fv.max()\n",
+        "    mxp = fp.max()\n",
+        "    clear_output(wait=True)\n",
+        "    fig, (ax1, ax2, ax3) = plt.subplots(figsize=(13, 3), ncols=3)\n",
+        "\n",
+        "    pos1 = ax1.imshow(fp, cmap='Blues', vmin=-mxp, vmax=mxp,interpolation='none')\n",
+        "    ax1.set_title('p')\n",
+        "    plt.colorbar(pos1,ax=ax1)\n",
+        "    pos2 = ax2.imshow(fu, cmap='Reds', vmin=-mxu, vmax=mxu,interpolation='none')\n",
+        "    ax2.set_title('u')\n",
+        "    plt.colorbar(pos2,ax=ax2)\n",
+        "    pos3 = ax3.imshow(fv, cmap='Greens',vmin=-mxv, vmax=mxv,interpolation='none')\n",
+        "    ax3.set_title('v')\n",
+        "    plt.colorbar(pos3, ax=ax3)\n",
+        "\n",
+        "    fig.suptitle(title)\n",
+        "    #plt.xlabel('x')\n",
+        "    #plt.ylabel('y')\n",
         "    plt.show()\n"
       ]
     },
@@ -293,7 +316,35 @@
       "id": "78ef5751-4589-4fbc-abfc-a316f672b3b9",
       "metadata": {},
       "outputs": [],
-      "source": []
+      "source": [
+        "if VAL:\n",
+        "\n",
+        "    u_val_f = 'ref/u.64.64.IT4000.txt'\n",
+        "    v_val_f = 'ref/v.64.64.IT4000.txt'\n",
+        "    p_val_f = 'ref/p.64.64.IT4000.txt'\n",
+        "    uval = np.zeros((M_LEN, N_LEN))\n",
+        "    vval = np.zeros((M_LEN, N_LEN))\n",
+        "    pval = np.zeros((M_LEN, N_LEN))\n",
+        "\n",
+        "    uref, vref, pref = read_arrays(v_val_f, u_val_f, p_val_f)\n",
+        "    uval = uref-unew\n",
+        "    vval = vref-vnew\n",
+        "    pval = pref-pnew\n",
+        "    \n",
+        "    uLinfN= np.linalg.norm(uval, np.inf)\n",
+        "    vLinfN= np.linalg.norm(vval, np.inf)\n",
+        "    pLinfN= np.linalg.norm(pval, np.inf)\n",
+        "\n",
+        "    \n",
+        "\n",
+        "    live_plot_val(uval, vval, pval, \"Val\")\n",
+        "    print(\"uLinfN: \", uLinfN)\n",
+        "    print(\"vLinfN: \", vLinfN)\n",
+        "    print(\"pLinfN: \", pLinfN)\n",
+        "    print(\"udiff max: \",uval.max())\n",
+        "    print(\"vdiff max: \",vval.max())\n",
+        "    print(\"pdiff max: \",pval.max())"
+      ]
     }
   ],
   "metadata": {