use adcp gridded file for plots if it exists

Author: callumrollo

complete_mission_plots.py

@@ -1,9 +1,8 @@
 import sys
 import os
-import pathlib
 import logging
-
-script_dir = pathlib.Path(__file__).parent.absolute()
+from pathlib import Path
+script_dir = Path(__file__).parent.absolute()
 sys.path.append(str(script_dir))
 os.chdir(script_dir)
 from gridded_plots import create_plots, make_map
@@ -17,13 +16,18 @@ def complete_plots(glider, mission):
                         format='%(asctime)s %(levelname)-8s %(message)s',
                         level=logging.INFO,
                         datefmt='%Y-%m-%d %H:%M:%S')
-    mission_dir = f'/data/data_l0_pyglider/complete_mission/SEA{glider}/M{mission}/gridfiles/'
-    try:
-        netcdf = list(pathlib.Path(mission_dir).glob('*.nc'))[0]
-    except IndexError:
-        _log.error(f"nc file in {mission_dir} not found")
-        return
-    outdir = pathlib.Path(f'/data/plots/complete_mission/SEA{glider}/M{mission}/')
+    mission_dir = Path(f'/data/data_l0_pyglider/complete_mission/SEA{glider}/M{mission}/gridfiles/')
+    netcdf = None
+    for fn in ['gridded_adcp_gliderad2cp.nc', 'gridded_gliderad2cp.nc']:
+        if (mission_dir / fn).exists():
+            netcdf = mission_dir / fn
+    if not netcdf:
+        try:
+            netcdf = list(mission_dir.glob('*.nc'))[0]
+        except IndexError:
+            _log.error(f"nc file in {mission_dir} not found")
+            return
+    outdir = Path(f'/data/plots/complete_mission/SEA{glider}/M{mission}/')
     if not outdir.exists():
         outdir.mkdir(parents=True)
     if 'scatter' in sys.argv:

gridded_plots.py

@@ -46,7 +46,9 @@ def sort_by_priority_list(values, priority):
     #    'DOWN_IRRADIANCE380',
     #    'DOWN_IRRADIANCE490',
     #    'DOWN_IRRADIANCE532',
-    'molar_nitrate'
+    'molar_nitrate',
+    'velocity_N_DAC_reference_sb_corrected',
+    'velocity_E_DAC_reference_sb_corrected',
 )
 
 # create dictionary to match each variable in glider_variables to a colourmap
@@ -66,6 +68,8 @@ def sort_by_priority_list(values, priority):
 cmap_dict['potential_density'] = cmo.dense
 cmap_dict['chlorophyll'] = cmo.algae
 cmap_dict['cdom'] = cmo.turbid
+cmap_dict['velocity_N_DAC_reference_sb_corrected'] = cmo.balance
+cmap_dict['velocity_E_DAC_reference_sb_corrected'] = cmo.balance
 
 labels_dict = {'Celsius': '$^{\\circ}$C',
                'None': '',
@@ -75,6 +79,7 @@ def sort_by_priority_list(values, priority):
                'μE/m^2/s': 'µE m$^{-1}$ s$^{-1}$',
                '1': '',
                'kg m-3': 'kg m$^{-3}$',
+               'm.s-1': 'm s$^{-1}$',
                'mmol m-3': 'mmol m$^{-3}$',
                'mg m-3': 'mg m$^{-3}$',
                'g kg^-1': 'g kg$^{-1}$',
@@ -85,7 +90,9 @@ def sort_by_priority_list(values, priority):
                'cdom': 'colored dissolved organic matter',
                'potential_density': 'potential density',
                'potential_temperature': 'potential temperature',
-               'absolute_salinity': 'absolute salinity'
+               'absolute_salinity': 'absolute salinity',
+               'velocity_E_DAC_reference_sb_corrected': 'Eastward velocity referenced to DAC',
+               'velocity_N_DAC_reference_sb_corrected': 'Northward velocity referenced to DAC',
                }
 
 
@@ -267,7 +274,7 @@ def multiplotter(dataset, variables, plots_dir, glider='', mission='', grid=True
             days = np.arange(1, 32)
         ax.xaxis.set_major_locator(mdates.MonthLocator())
         ax.xaxis.set_minor_locator(mdates.DayLocator(days))
-        ax.xaxis.set_major_formatter(mdates.DateFormatter("\n%b %Y"))
+        ax.xaxis.set_major_formatter(mdates.DateFormatter("\n %d %b %Y"))
         ax.xaxis.set_minor_formatter(mdates.DateFormatter("%d"))
         ax.tick_params(axis="x", which="both", length=4)
         plt.setp(ax.get_xticklabels(), rotation=0, ha="center")
@@ -366,4 +373,4 @@ def make_map(nc, filename):
 
 
 if __name__ == '__main__':
-    create_plots(Path("/data/data_l0_pyglider/nrt/SEA76/M25/gridfiles/mission_grid.nc"), Path("/home/callum/Downloads"), True)
+    create_plots(Path("/data/data_l0_pyglider/complete_mission/SEA55/M85/gridfiles/gridded_adcp_gliderad2cp.nc"), Path("/home/callum/Downloads"), True)