Changeset 3868 for trunk/LMDZ.PLUTO/util/script_figures/movie_winds.py

Timestamp:

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

Author:

afalco

Message:

Pluto: update figure scripts.
AF

File:

• trunk/LMDZ.PLUTO/util/script_figures/movie_winds.py (moved) (moved from trunk/LMDZ.PLUTO/util/script_figures/movie_winds/tsurf_winds.py) (7 diffs)

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