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

Timestamp:

Jul 7, 2025, 1:39:13 PM (20 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/mapmeanwinds.py (modified) (4 diffs)

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

@@ -8 +8 @@
 from matplotlib import ticker
 import matplotlib.colors as colors
-import datetime
 from mpl_toolkits.basemap import Basemap, shiftgrid
 from FV3_utils import *
+from input import *
 
-############################
 fa='sans-serif'
 hfont = {'fontname':'Arial'}
@@ -25 +24 @@
 lvls=np.linspace(37,57,21)
 
+altitude=1 # in km
+
 ### Data
-name2=name+'.nc'
-# name=name+'_A.nc'
-name=name+'.nc'
-print("Plot "+name)
-nc1=Dataset(name)
-nc2=Dataset(name2)
+filename1=name+'_A.nc'
+# name=name+'.nc'
+filename2=name+'.nc'
+print("Plot "+filename1)
+nc1=Dataset(filename1)
+nc2=Dataset(filename2)
 ts=nc2.variables["temperature"][:,0,:,:]
 # ts=nc2.variables["tsurf"][:,:,:]
 u=nc1.variables["u"][:,:,:,:]
 v=nc1.variables["v"][:,:,:,:]
-lat=nc1.variables["lat"][:]
-alt=nc1.variables["altitude"][:]
-lon=nc1.variables["lon"][:]
+lat=getvar(nc1,"latitude")
+alt=getvar(nc1,"altitude")
+lon=getvar(nc1,"longitude")
 ps=nc2.variables["ps"][:,:]
 # ps=nc2.variables["phisinit"][:,:]/0.6169/1000.  # altitude km
 
-numalt=getind(1,alt)
-# numalt=getind(1000,alt)
+numalt=getind(altitude,alt)
 print('numalt =',numalt,'altitude=',alt[numalt])
 u=u[:,numalt,:,:]
@@ -52 +52 @@
 ps=np.mean(ps,axis=0)
 
-ts=switchlon(ts)
-u=switchlon(u)
-v=switchlon(v)
-# ps=switchlon(ps)
-# topo=switchlon(topo)
-# lon=lon+180.
+if lon[0]<0:
+    ts=switchlon(ts)
+    u=switchlon(u)
+    v=switchlon(v)
+    ps=switchlon(ps)
+    # topo=switchlon(topo)
+    lon=lon+180.
 
 ### Figure
@@ -96 +97 @@
 mpl.show()
 
-#######################
-