Changeset 3833 for trunk/LMDZ.PLUTO/util/script_figures/maptemp.py

Timestamp:

Jul 7, 2025, 1:39:13 PM (8 hours ago)

Author:

afalco

Message:

Pluto: updated plots scripts.
Fixed some issues with reading XIOS, etc.
Included display_netcdf.py tool from Mars PCM.
AF

File:

• trunk/LMDZ.PLUTO/util/script_figures/maptemp.py (modified) (5 diffs)

trunk/LMDZ.PLUTO/util/script_figures/maptemp.py

@@ -1 +1 @@
 #! /usr/bin/env python
+import os
 from    netCDF4               import    Dataset
 from    numpy                 import    *
@@ -11 +12 @@
 from mpl_toolkits.basemap import Basemap, shiftgrid
 from matplotlib.cm import get_cmap
-from FV3_utils import * # import name
+from FV3_utils import *
+from input import * # import name
+print("Running "+os.path.basename(__file__))
 
 ############################
@@ -26 +29 @@
 
 ### Data
-# name='../diagfi2015.nc' # read from FV3_util
-print(name)
 try:
+    print("Plotting "+name+"_A.nc")
     nc1=Dataset(name+"_A.nc")
 except:
+    print("Plotting "+name+".nc")
     nc1=Dataset(name+".nc")
-alt=nc1.variables["altitude"][:]
-lat=nc1.variables["lat"][:]
-lon=nc1.variables["lon"][:]
+alt=getvar(nc1,"altitude")
+lat=getvar(nc1,"latitude")
+lon=getvar(nc1,"longitude")
 # temp=switchlon(temp)
-lon=lon+180.
 
-def plot_alt(altitude = 1000):
+if lon[0]<0:
+    lon=lon+180.
+
+def plot_alt(altitude = 1):
     temp=nc1.variables["temperature"][:,:,:,:]
     numalt=getind(altitude,alt)
@@ -63 +68 @@
 
     mpl.grid()
-    mpl.title(f"Temperatures @ z={altitude/1000}km",fontsize=font)
+    mpl.title(f"Temperatures @ z={altitude}km",fontsize=font)
     mpl.ylabel(r'Latitude',labelpad=10,fontsize=font)
     mpl.xlabel('Longitude',labelpad=10, fontsize=font)
@@ -72 +77 @@
     mpl.yticks(yticks,fontsize=font)
     mpl.xticks(xticks,fontsize=font)
-    mpl.savefig(f"maptemp{altitude}",bbox_inches='tight',dpi=70)
+    output=f"maptemp{altitude}"
+    mpl.savefig(output,bbox_inches='tight',dpi=70)
+    print(f"Saved {output}")
     #mpl.show()
 
 
-plot_alt(1000)
-plot_alt(5000)
-plot_alt(20000)
-plot_alt(50000)
-plot_alt(100000)
+plot_alt(1)
+plot_alt(5)
+plot_alt(20)
+plot_alt(50)
+plot_alt(100)