Changeset 3868 for trunk/LMDZ.PLUTO/util

Timestamp:

Jul 28, 2025, 3:43:42 PM (8 days ago)

Author:

afalco

Message:

Pluto: update figure scripts.
AF

Location:

trunk/LMDZ.PLUTO/util/script_figures

Files:

• FV3_utils.py (modified) (2 diffs)
• meanmeridwind.py (modified) (3 diffs)
• meanzonalwind.py (modified) (3 diffs)
• movie_winds.py (moved) (moved from trunk/LMDZ.PLUTO/util/script_figures/movie_winds/tsurf_winds.py) (7 diffs)
• movie_winds.sh (added)
• movie_winds/anim.sh (deleted)
• temp_anomaly.py (added)
• temp_section.py (modified) (4 diffs)
• temp_time_zoom.py (modified) (3 diffs)
• tempglobmean.py (modified) (1 diff)
• waves/temp_anomaly.py (deleted)

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

@@ -60 +60 @@
  return alpha,airetot
 
-def switchlon(arr):
+def switchlon(arr, lon):
 # changer les longitudes pour mettre TR au centre
  vec=np.shape(arr)
  myvar=np.zeros(vec,dtype='f')
+ if (lon is not None and lon[0]>=0):
+     return arr
  # i lat : pas de changement
  # j lon :
@@ -1260 +1262 @@
                 tmp_var = find_coord_var(nc1, dim)
                 file_values = nc1.variables[tmp_var][:]
+                if dim is LON_DIMS and file_values[0]<0:
+                    file_values+=180 # start longitudes at 0 rather than -180
                 i = getind(values, file_values)
-                print(dim, i)
-                if len(i)==2: # range of indices
+                print("Getting",dim[0], i, file_values[i])
+                if len(np.atleast_1d(i))==2: # range of indices
                     slicer[dim_idx] = slice(min(i),max(i)+1)
                 else: # specific index
-                    slicer[dim_idx] = i
+                    slicer[dim_idx] = slice(i,i+1)
             except:
                 continue

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

@@ -15 +15 @@
 var="v" #variable
 tint=[30,31] #Time must be as written in the input file
-xarea=[-180,179]
+xarea=[0,359] # omit repetition of longitude
+xarea=None
 
 nc1=Dataset(filename1)
@@ -64 +65 @@
 rvb1=diverge_map(h,l)
 
-myvar=getvar(nc1,var,tint,xarea,lon,t_mean=True,l_mean=True)
+myvar=getvar(nc1,var,times=tint,longitudes=xarea,t_mean=True,l_mean=True)
 
 mpl.figure(figsize=(20, 10))
@@ -97 +98 @@
 mpl.yticks(fontsize=font)
 pylab.ylim([-4,250])
+mpl.tight_layout()
 
 mpl.savefig('meanmeridwind.eps',dpi=200)

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

@@ -16 +16 @@
 var="u" #variable
 tint=[30,32] #Time must be as written in the input file
-xarea=[-180,179]
+xarea=[0,359]
+# xarea=None
 
 nc1=Dataset(filename1)
@@ -65 +66 @@
 rvb1=diverge_map(h,l)
 
-myvar=getvar(nc1,var,tint,xarea,lon,t_mean=True,l_mean=True)
+myvar=getvar(nc1,var,times=tint,longitudes=xarea,t_mean=True,l_mean=True)
 
 mpl.figure(figsize=(20, 10))
@@ -98 +99 @@
 mpl.yticks(fontsize=font)
 pylab.ylim([-4,250])
-
+mpl.tight_layout()
 mpl.savefig('meanzonalwind.eps',dpi=200)
 mpl.savefig('meanzonalwind.png',dpi=200)

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

@@ -1 +1 @@
 #! /usr/bin/env python
+import os, sys
 from    netCDF4               import    Dataset
 from    numpy                 import    *
@@ -13 +14 @@
 
 ############################
-# basename="../../Xhistins2072"
-filename1="../"+name+"_A.nc"
-filename2="../"+name+".nc"
-filename3="../"+name+".nc"
+filename1=name+"_A.nc"
+filename2=name+".nc"
+filename3=name+".nc"
 # filename3="../../phisinit.nc"
 var="tsurf" #variable
@@ -23 +23 @@
 varj="v"
 tint=[30,32] #Time must be as written in the input file
+# tint=None
 #tintstr=["03:00","09:00","15:00","21:00"] #Time must be as written in the input file
+
+altitude = 10 # default value, in km ! WARNING: only integers accepted for animation script yet
+if len(sys.argv)>1:
+    altitude=float(sys.argv[1])
+    print("Setting altitude to ",altitude, "km")
 
 font=26
@@ -86 +92 @@
 
 def getfigvar(i):
-    pal=get_cmap(name="jet")
+    pal=get_cmap(name="OrRd")
     lev=np.linspace(37,51,15)
     newlon=lon+180
-    CF=mpl.contourf(newlon, lat, myvarbis,lev,cmap=pal,extend='both')
+    CF=mpl.contourf(newlon, lat, myvarbis,lev,cmap=pal,extend='both',alpha=0.7)
     yticks=[-90,-60,-30,0,30,60,90]
     xticks=[0,60,120,180,240,300,360]
     cbar=mpl.colorbar(CF,shrink=1, format="%1.0f")
-    cbar.ax.set_title("Tsurf [K]",y=1.04,fontsize=font)
+    cbar.ax.set_title("Tsurf [K]",y=1.04,fontsize=font, pad=20)
 
     for t in cbar.ax.get_yticklabels():
@@ -100 +106 @@
     c=mpl.contour(newlon, lat, myvar2bis, 10,levels=np.linspace(-4,4,8), colors = 'k', linewidths = 3.5)
     mpl.clabel(c, fmt='%2.1f',inline=1, colors='k', fontsize=23,inline_spacing=1)
-    #mpl.title('Local Time at Sputnik Planum='+str(i*3)+'H00',fontsize=font)
+    mpl.title('Winds & Surface Temperature',fontsize=font, pad=20)
+    # mpl.title('Local Time at Sputnik Planum='+str(i*3)+'H00',fontsize=font)
     mpl.ylabel('Latitude (deg)',labelpad=10,fontsize=font)
     mpl.xlabel('West Longitude (deg)',labelpad=10, fontsize=font)
@@ -113 +120 @@
 
 #######################
-numalt=getnumalt(30,alt)
+numalt=getnumalt(altitude,alt)
 print(('numalt =',numalt,'altitude=',alt[numalt]))
-uini=getvar(nc1,vari,tint,tim)[:,numalt]
-vini=getvar(nc1,varj,tint,tim)[:,numalt]
-myvar=getvar(nc2,var,tint,tim)
+uini=getvar(nc1,vari,times=tint)[:,numalt]
+vini=getvar(nc1,varj,times=tint)[:,numalt]
+myvar=getvar(nc2,var,times=tint)
 myvar2=getvar(nc3,phisinit) # phisinitmyvar2=myvar2/0.6169/1000.  # altitude km
 nbfig=uini.shape[0]
 print(("nbfig=",nbfig))
-myvar2bis=switchlon(myvar2)
+myvar2bis=switchlon(myvar2, lon)
+
+os.makedirs("movie_winds", exist_ok=True)
 
 for i in range(nbfig):
@@ -128 +137 @@
    v2=vini[i,:,:]
    myv=myvar[i,:,:]
-   u=switchlon(u2)
-   v=switchlon(v2)
-   myvarbis=switchlon(myv)
+   u=switchlon(u2, lon)
+   v=switchlon(v2, lon)
+   myvarbis=switchlon(myv, lon)
    print(i,"/",nbfig)
    getfigvar(i)
-   mpl.savefig('mapwinds'+str('{0:03}'.format(i))+'.eps',dpi=200)
-   mpl.savefig('mapwinds'+str('{0:03}'.format(i))+'.png',dpi=200)
+#    mpl.tight_layout()
+   mpl.savefig(f"movie_winds/mapwinds_{altitude:.0f}km_{i:03}.eps",dpi=200)
+   mpl.savefig(f"movie_winds/mapwinds_{altitude:.0f}km_{i:03}.png",dpi=200)
 
 left  = None  # the left side of the subplots of the figure

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

@@ -24 +24 @@
 nc1=Dataset(name)
 
+# max altitude in plot (maxed to max of alt)
+max_alt = 230 
+
 
 lat=getvar(nc1,"latitude")
@@ -48 +51 @@
 CF=mpl.contourf(lat,alt,myvar,lev,cmap=pal,extend='both')
 cbar=mpl.colorbar(CF,shrink=1, format="%1.0f")
-cbar.ax.set_title("Tsurf [K]",y=1.04,fontsize=font)
+cbar.ax.set_title("T [K]",y=1.04,fontsize=font,pad=20)
 for t in cbar.ax.get_yticklabels():
       t.set_fontsize(font)
@@ -58 +61 @@
 mpl.clabel(CS, inline=1, fontsize=20, fmt='%1.0f',inline_spacing=1)
 
+mpl.title('Zonal mean temperature',fontsize=font, pad=20)
 #mpl.title('Latitude ='+str(tintstr[i]),fontsize=font)
 mpl.ylabel('Altitude (km)',labelpad=10,fontsize=font)
@@ -65 +69 @@
 mpl.yticks(fontsize=font)
 if type(ma.getmask(myvar)) in (bool,np.bool_):
-    max_alt = max(alt)
+    max_alt = min(max_alt,max(alt))
 else:
-    max_alt = max(alt[~ma.getmask(myvar)[:,0]][:-1])
-# max_alt = 230
+    max_alt = min(max_alt,max(alt[~ma.getmask(myvar)[:,0]][:-1]))
 pylab.ylim([-4,max_alt])
 

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

@@ -18 +18 @@
 # longitude=-165
 latitude=[-19,-15]
-# latitude=-19
+latitude=-18
 
 # local time de la longitude consideree a t=0
@@ -39 +39 @@
 myvar=getvar(nc1,var,times=tint,longitudes=longitude,latitudes=latitude)[:,:,0,0]
 font=23
-# tim=np.linspace(0,24,9)
 print(("tim=",tim))
 print(('on prend les premiers indice, shape (tmps, alt, var) =',shape(tim), shape(alt), shape(myvar)))
@@ -46 +45 @@
 lev=np.linspace(40,50,10)
 # xticks=[0,2,4,6,8,10,12,14,16,18,20,22,24] 
-print(('hello:',np.linspace(0,24,13)))
 #yticks=np.linspace(0,240,9)
 

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

@@ -65 +65 @@
 #xticks=[-90,-60,-30,0,30,60,90]
 #yticks=np.linspace(0,240,9)
-alt=alt/1000.
-#alt2=alt2/1000.
 
 mpl.plot(meantemp,alt,'r')