Example 3: Section Plot
=======================

This example demonstrates how to import file and draw a section plot of one or several points with GINCCO_lib
using :func:`GINCCO_lib.heatmap_plot.plot_heatmap`.


Code Example
------------

Now we will import the library and the grid

.. code-block:: python

    # =========================
    # IMPORTS
    # =========================
    import numpy as np
    from netCDF4 import Dataset
    from datetime import *
    import GINCCO_lib as gc

    # =========================
    # CONFIGURATION
    # =========================
    tstart = datetime(2010, 1, 1)
    tend   = datetime(2010, 10, 30)

    path = '/work/users/tungnd/GOT271/GOT_REF5/OFFLINE/'

    # =========================
    # LOAD GRID AND DEPTH
    # =========================
    fgrid   = Dataset(path + 'grid.nc', 'r')
    lat_t   = fgrid.variables['latitude_t'][:]
    lon_t   = fgrid.variables['longitude_t'][:]
    mask_t_var = fgrid.variables['mask_t']
    if mask_t_var.ndim == 3:
        mask_t = mask_t_var[0, :, :]
    elif mask_t_var.ndim == 2:
        mask_t = mask_t_var[:, :]

    depth_t = fgrid.variables['depth_t'][:, :, :]

    # Apply mask: set land values to NaN
    mask_t_3d = np.repeat(mask_t[np.newaxis, :, :], depth_t.shape[0], axis=0)
    depth_t[mask_t_3d == 0] = np.nan


Now define the point and see where it is on map 

.. code-block:: python

    # ============================================================
    # EXAMPLE: Plot salinity profiles at 3 points
    # ============================================================

    # Step 1: Define three points along latitude 19N
    lon_p = [106, 107, 108]
    lat_p = [19, 19, 19]

    # Step 2: Plot locations of the points with bathymetry
    gc.map_draw_point(
        lon_min=105, lon_max=111,
        lat_min=16.5, lat_max=22,
        title="Locations of points and bathymetry (in m)",
        lon_data=lon_t,
        lat_data=lat_t,
        data_draw=depth_t[0, :, :], 
        lat_point=lat_p,
        lon_point=lon_p,
        path_save="/prod/projects/data/tungnd/figure/",
        name_save="example3.1"
    )


.. image:: ../_static/example3.1_82511.png
   :width: 500px
   :align: center



Now import data and plot the results

.. code-block:: python

    # Step 3: Import salinity profiles for the defined points
    # data_draw shape = (number of points, number of days, number of depths)
    data_draw    = np.zeros((len(lon_p), (tend - tstart).days + 1, np.size(depth_t, 0)))
    point_index  = np.zeros((len(lon_p), 2), dtype=int)

    for i in range(len(lon_p)):
        data_draw[i, :], point_index[i, :] = gc.import_profile(
            path, 'sal', tstart, tend, lat_p[i], lon_p[i],
            ji='False', ignore_missing='False'
        )

    # Step 4: Plot heatmaps for each point
    for i in range(len(lon_p)):
        print(data_draw[i, :, :].shape, depth_t[:, point_index[i, 0], point_index[i, 1]].shape)
        gc.plot_heatmap(
            title="Salinity profile at %sE %sN" % (lon_p[i], lat_p[i]),
            tstart=tstart,
            tend=tend,
            data_draw=data_draw[i, :, :],
            depth=depth_t[:, point_index[i, 0], point_index[i, 1]],
            path_save="/prod/projects/data/tungnd/figure/",
            name_save="example3.2",
        )


.. image:: ../_static/example3.2_66236.png
   :width: 700px
   :align: center


.. image:: ../_static/example3.2_23402.png
   :width: 700px
   :align: center


.. image:: ../_static/example3.2_69289.png
   :width: 700px
   :align: center