Update ash_animator/basemaps.py

ash_animator/basemaps.py

@@ -5,14 +5,15 @@ import contextily as ctx
 from mpl_toolkits.basemap import Basemap
 import cartopy.crs as ccrs
 import cartopy.feature as cfeature
+import cartopy.io.shapereader as shpreader
 from PIL import Image
 import matplotlib.pyplot as plt
 
-# Determine platform and fallback cache path
+# Determine platform-appropriate, writable cache directory
 def get_cache_dir(app_name):
-    if os.name == 'nt':
+    if os.name == 'nt':  # Windows
         return os.path.join(os.getenv('LOCALAPPDATA', tempfile.gettempdir()), f"{app_name}_cache")
-    elif os.name == 'posix':
+    elif os.name == 'posix':  # Linux/macOS
         home_dir = os.path.expanduser("~")
         if os.path.isdir(home_dir) and os.access(home_dir, os.W_OK):
             return os.path.join(home_dir, f".{app_name}_cache")
@@ -21,13 +22,18 @@ def get_cache_dir(app_name):
     else:
         return os.path.join(tempfile.gettempdir(), f"{app_name}_cache")
 
-# Define cache directories
+# Define and create cache directories
 CTX_TILE_CACHE_DIR = get_cache_dir("contextily")
 BASEMAP_TILE_CACHE_DIR = get_cache_dir("basemap")
+CARTOPY_CACHE_DIR = get_cache_dir("cartopy")
 
 os.environ["XDG_CACHE_HOME"] = CTX_TILE_CACHE_DIR
+os.environ["CARTOPY_USER_BACKGROUNDS"] = CARTOPY_CACHE_DIR
+os.environ["CARTOPY_CACHE_DIR"] = CARTOPY_CACHE_DIR
+
 os.makedirs(CTX_TILE_CACHE_DIR, exist_ok=True)
 os.makedirs(BASEMAP_TILE_CACHE_DIR, exist_ok=True)
+os.makedirs(CARTOPY_CACHE_DIR, exist_ok=True)
 
 def draw_etopo_basemap(ax, mode="basemap", zoom=11):
     """
@@ -46,13 +52,7 @@ def draw_etopo_basemap(ax, mode="basemap", zoom=11):
         Default is "basemap".
 
     zoom : int, optional
-        Tile zoom level (only for "contextily"). Higher = more detail. Default is 7.
-
-    Notes
-    -----
-    - Uses high resolution for Basemap (resolution='h') and saves figure at 300 DPI.
-    - Cached images are reused using extent-based hashing to avoid re-rendering.
-    - Basemap is deprecated; Cartopy with web tiles is recommended for new projects.
+        Tile zoom level (only for "contextily"). Higher = more detail. Default is 11.
     """
     try:
         if mode == "stock":