Changeset 3851 for trunk/LMDZ.MARS/util/display_netcdf.py

Timestamp:

Jul 16, 2025, 3:25:48 PM (9 days ago)

Author:

jbclement

Message:

PEM:

• Making the computation of CO2 mass balance more robust, especially regarding 'CO2cond_ps'.
• Small correction about the dust tendency management for the layering algorithm.
• Small improvement of the visualization in "visu_evol_layering.py".
• File "log_launchPEM.txt" renamed into "launchPEM.log".

JBC

File:

• trunk/LMDZ.MARS/util/display_netcdf.py (modified) (5 diffs)

trunk/LMDZ.MARS/util/display_netcdf.py

@@ -84 +84 @@
 
 # Attempt to load MOLA topography
-try:
-    MOLA = np.load('MOLA_1px_per_deg.npy') # shape (nlat, nlon) at 1° per pixel: lat from -90 to 90, lon from 0 to 360
+if os.path.isfile(MOLA_NPY): # shape (nlat, nlon) at 1° per pixel: lat from -90 to 90, lon from 0 to 360
+    MOLA = np.load(MOLA_NPY) 
     nlat, nlon = MOLA.shape
     topo_lats = np.linspace(90 - 0.5, -90 + 0.5, nlat)
@@ -91 +91 @@
     topo_lon2d, topo_lat2d = np.meshgrid(topo_lons, topo_lats)
     topo_loaded = True
-except Exception as e:
-    print(f"Warning: '{MOLA_NPY}' not found: {e}")
+else:
+    print(f"Warning: '{MOLA_NPY}' not found! Topography contours disabled.")
     topo_loaded = False
 
@@ -101 +101 @@
     csv_loaded = True
 else:
-    print(f"Warning: '{MOLA_CSV}' not found. 3D view colors disabled.")
+    print(f"Warning: '{MOLA_CSV}' not found! 3D view colors disabled.")
     csv_loaded = False
 
@@ -585 +585 @@
         ax.set_ylabel(f"Latitude ({getattr(dataset.variables[latv], 'units', 'deg')})")
         # Prompt for polar-stereo views if interactive
-        if input("Display polar-stereo views? [y/n]: ").strip().lower() == "y":
+        if input("Display polar-stereo views? [y = yes, anything else = no]: ").strip().lower() == "y":
             units = getattr(var, 'units', None)
             plot_polar_views(lon2d, lat2d, grid, colormap, varname, units)
         # Prompt for 3D globe view if interactive
-        if input("Display 3D globe view? [y/n]: ").strip().lower() == "y":
+        if input("Display 3D globe view? [y = yes, anything else = no]: ").strip().lower() == "y":
             units = getattr(var, 'units', None)
             plot_3D_globe(lon2d, lat2d, grid, colormap, varname, units)
@@ -659 +659 @@
         ax.grid(True)
         # Prompt for polar-stereo views if interactive
-        if sys.stdin.isatty() and input("Display polar-stereo views? [y/n]: ").strip().lower() == "y":
+        if sys.stdin.isatty() and input("Display polar-stereo views? [y = yes, anything else = no]: ").strip().lower() == "y":
             units = getattr(dataset.variables[varname], "units", None)
             plot_polar_views(x_coords, y_coords, plot_data, colormap, varname, units)
         # Prompt for 3D globe view if interactive
-        if sys.stdin.isatty() and input("Display 3D globe view? [y/n]: ").strip().lower() == "y":
+        if sys.stdin.isatty() and input("Display 3D globe view? [y = yes, anything else = no]: ").strip().lower() == "y":
             units = getattr(dataset.variables[varname], "units", None)
             plot_3D_globe(x_coords, y_coords, plot_data, colormap, varname, units)