Grad_water_level.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 90,
   "metadata": {
    "collapsed": true
   },
   "outputs": [],
   "source": [
    "import matplotlib.pyplot as plt\n",
    "import netCDF4\n",
    "import numpy as np "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 92,
   "metadata": {
    "collapsed": true
   },
   "outputs": [],
   "source": [
    "def read(his_file,sim_num,fname): \n",
    "\n",
    "# Middle of the vegetation patch  \n",
    "    jmid=50; imid=45; \n",
    "# detect the presence of vegetation patch to know its extents (Currently hardwired\n",
    "    imin_plant=41 ; imax_plant=50 ; jmin_plant=45 ; jmax_plant=54   \n",
    "    \n",
    "    nc = netCDF4.Dataset(his_file)\n",
    "    mask = nc.variables['mask_rho'][:]\n",
    "\n",
    "# read x,y\n",
    "    x = nc.variables['x_rho'][:,:]\n",
    "    y = nc.variables['y_rho'][:,:] \n",
    "    tot_time = nc.variables['ocean_time'] \n",
    "\n",
    "# Remember all arrays are (time,kmax,jmax,imax) contrary to Matlab     \n",
    "    imax=size(x[1,:])\n",
    "    jmax=size(y[:,1])\n",
    "     \n",
    "# read water depth\n",
    "    depth = nc.variables['h'][:,:]\n",
    "    theta_s = nc.variables['theta_s'][:]\n",
    "    theta_b = nc.variables['theta_b'][:]\n",
    "    depth_c = nc.variables['hc'][:]\n",
    "\n",
    "#Get Stretching function \n",
    "    N=10 # Number of vertical levels, kgrid=1 MATLAB code copy \n",
    "    Np=N+1 \n",
    "    ds=1.0/N\n",
    "    s=np.zeros(11)\n",
    "    Csur=np.zeros(11)\n",
    "    for i in range(0,Np):\n",
    "        lev=i\n",
    "        s[i]=(lev-N)*ds\n",
    "        Csur[i]=-s[i]**2\n",
    "        \n",
    "# Reshape 1D vertical variables to 3D so we can broadcast\n",
    "    s.shape=(np.size(s),1,1)\n",
    "    Csur.shape=(np.size(Csur),1,1)\n",
    "    \n",
    "# read 3D water level at all time steps\n",
    "    eta = nc.variables['zeta'] \n",
    "    \n",
    "# calculate the 4D field of z values (vertical coordinate)\n",
    "    z0=(depth_c*s+depth*Csur)/(depth_c+depth)  \n",
    "    \n",
    "# Extract the time frame for the last tidal cycle\n",
    "    t_tidal=[]\n",
    "    for t in range(1,size(tot_time)):\n",
    "        if(eta[t,jmid,imid]<0.0 and eta[t+1,jmid,imid]>0.0):\n",
    "            t_tidal.append(t)\n",
    "\n",
    "    t_tidal_srt=t_tidal[-2]\n",
    "    t_tidal_end=t_tidal[-1]\n",
    "    print \"Last tidal cycle starts and ends at this instance\" \n",
    "    print t_tidal_srt,t_tidal_end   \n",
    "    t_slice=t_tidal_end-t_tidal_srt+1\n",
    "\n",
    "# Find water level for all locations at last tidal cycle \n",
    "    z=np.zeros(shape=(t_slice,Np,jmax,imax))\n",
    "    for t in range(t_tidal_srt,t_tidal_end+1): \n",
    "        eta = nc.variables['zeta'][t,:,:] \n",
    "        z[t-t_tidal_srt,:,:,:] = eta+(eta+depth)*z0\n",
    "    \n",
    "# Calculate maximum water level change in a tidal cycle before and after the patch\n",
    "# Probe the region infront of the patch by adding 5 points in vertical \n",
    "# Probe the region behind the patch by adding 5 points in vertical \n",
    "# Water level change in the middle transect before and after the patch\n",
    "\n",
    "    grad_wl=0.0\n",
    "    for t in range(0,t_slice):\n",
    "        for k in range(0,Np):\n",
    "            grad=abs(z[t,k,jmin_plant-5,imid]-z[t,k,jmax_plant+5,imid])\n",
    "            grad_wl=max(grad_wl,grad)\n",
    "    \n",
    "    print \"Maximum gradient water level\"\n",
    "    print grad_wl \n",
    "    \n",
    "    f.write(\"%s\\t, %s\\t \\n\" % (sim_num, grad_wl))\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 93,
   "metadata": {
    "collapsed": false
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "/media/gadar/DATADRIVE1/sensitivity_history/veg_test_his_1.nc\n",
      "Last tidal cycle starts and ends at this instance\n",
      "186 258\n",
      "Maximum gradient water level\n",
      "0.0906050857157\n",
      "/media/gadar/DATADRIVE1/sensitivity_history/veg_test_his_2.nc\n",
      "Last tidal cycle starts and ends at this instance\n",
      "186 258\n",
      "Maximum gradient water level\n",
      "0.0776969827712\n",
      "/media/gadar/DATADRIVE1/sensitivity_history/veg_test_his_3.nc\n",
      "Last tidal cycle starts and ends at this instance\n",
      "186 258\n",
      "Maximum gradient water level\n",
      "0.0864368518814\n",
      "/media/gadar/DATADRIVE1/sensitivity_history/veg_test_his_4.nc\n",
      "Last tidal cycle starts and ends at this instance\n",
      "186 258\n",
      "Maximum gradient water level\n",
      "0.0932615157217\n",
      "/media/gadar/DATADRIVE1/sensitivity_history/veg_test_his_5.nc\n",
      "Last tidal cycle starts and ends at this instance\n",
      "186 258\n",
      "Maximum gradient water level\n",
      "0.0892242211848\n",
      "/media/gadar/DATADRIVE1/sensitivity_history/veg_test_his_6.nc\n",
      "Last tidal cycle starts and ends at this instance\n",
      "186 258\n",
      "Maximum gradient water level\n",
      "0.0767011023127\n",
      "/media/gadar/DATADRIVE1/sensitivity_history/veg_test_his_7.nc\n",
      "Last tidal cycle starts and ends at this instance\n",
      "186 258\n",
      "Maximum gradient water level\n",
      "0.0866207387298\n",
      "/media/gadar/DATADRIVE1/sensitivity_history/veg_test_his_8.nc\n",
      "Last tidal cycle starts and ends at this instance\n",
      "186 258\n",
      "Maximum gradient water level\n",
      "0.0977708660066\n",
      "/media/gadar/DATADRIVE1/sensitivity_history/veg_test_his_9.nc\n",
      "Last tidal cycle starts and ends at this instance\n",
      "186 258\n",
      "Maximum gradient water level\n",
      "0.0861118221655\n",
      "/media/gadar/DATADRIVE1/sensitivity_history/veg_test_his_10.nc\n",
      "Last tidal cycle starts and ends at this instance\n",
      "186 258\n",
      "Maximum gradient water level\n",
      "0.0963280722499\n",
      "/media/gadar/DATADRIVE1/sensitivity_history/veg_test_his_11.nc\n",
      "Last tidal cycle starts and ends at this instance\n",
      "186 258\n",
      "Maximum gradient water level\n",
      "0.0774124991149\n",
      "/media/gadar/DATADRIVE1/sensitivity_history/veg_test_his_12.nc\n",
      "Last tidal cycle starts and ends at this instance\n",
      "186 258\n",
      "Maximum gradient water level\n",
      "0.087565776892\n",
      "/media/gadar/DATADRIVE1/sensitivity_history/veg_test_his_13.nc\n",
      "Last tidal cycle starts and ends at this instance\n",
      "186 258\n",
      "Maximum gradient water level\n",
      "0.0768618592992\n",
      "/media/gadar/DATADRIVE1/sensitivity_history/veg_test_his_14.nc\n",
      "Last tidal cycle starts and ends at this instance\n",
      "186 258\n",
      "Maximum gradient water level\n",
      "0.0870198933408\n",
      "/media/gadar/DATADRIVE1/sensitivity_history/veg_test_his_15.nc\n",
      "Last tidal cycle starts and ends at this instance\n",
      "186 258\n",
      "Maximum gradient water level\n",
      "0.0866443598643\n"
     ]
    }
   ],
   "source": [
    " '''Output from vegetation module COAWST \n",
    "Mean gradient of water level around vegetation p)[mts.]\n",
    "'''    \n",
    "fname='grad_water_level_output.txt'\n",
    "f = open(fname, 'w')\n",
    "total_sim_list=16\n",
    "f.write(\"Case_no, Gradient of water level(mts.): \\n\")\n",
    "\n",
    "for sim_num in range(1,total_sim_list):\n",
    "    path1='/media/gadar/DATADRIVE1/sensitivity_history/'\n",
    "    path2='veg_test_his_'+ str(sim_num)+'.nc'\n",
    "    url=path1+path2\n",
    "    print url \n",
    "    read(url,sim_num,fname) "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {
    "collapsed": true
   },
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 2",
   "language": "python",
   "name": "python2"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 2
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython2",
   "version": "2.7.11"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 0
}