diff --git a/Inst_eddies/Analysis/plot_eddy_contours.py b/Inst_eddies/Analysis/plot_eddy_contours.py index be897c375705719b2c3475a19394e0efbe560075..d8bf4375f65f35928ccbd9f6652fbc852eb84574 100755 --- a/Inst_eddies/Analysis/plot_eddy_contours.py +++ b/Inst_eddies/Analysis/plot_eddy_contours.py @@ -18,6 +18,7 @@ mn for a 180° by 180° window. import shapefile import numpy as np +from cartopy.mpl import geoaxes import cartopy.crs as ccrs from os import path from matplotlib import patches @@ -25,7 +26,8 @@ import f90nml import sys import util_eddies -def select_ishapes(d, SHPC, i_slice, orientation, window = None): +def select_ishapes(d: int, SHPC: util_eddies.SHPC_class, i_slice: int, + orientation: str, window: list = None): """Select ishapes at date d and in window. We assume date d is in slice i_slice of the SHPC. @@ -47,8 +49,9 @@ def select_ishapes(d, SHPC, i_slice, orientation, window = None): return ishape_list_filt -def snapshot(ax, ishape_list, SHPC, i_slice, orientation, *, dashed = False, - light = False, src_crs = ccrs.PlateCarree()): +def snapshot(ax: geoaxes.GeoAxesSubplot, ishape_list, SHPC, i_slice, + orientation, *, dashed = False, light = False, + src_crs = ccrs.PlateCarree()): """Plots extrema, outermost contours and max-speed contours from a given SHPC, for slice i_slice, for a given orientation, for ishapes given by ishape_list.