Changeset 1945 in lmdz_wrf for trunk

Timestamp:

Jul 17, 2018, 11:45:19 PM (7 years ago)

Author:

lfita

Message:

Fixing `draw_multi_SkewT'

Location:

trunk/tools

Files:

• drawing.py (modified) (2 diffs)
• drawing_tools.py (modified) (8 diffs)

trunk/tools/drawing.py

@@ -9078 +9078 @@
             quit(-1)
 
+        if  not os.path.isfile(ncfile):
+            print errormsg
+            print '  ' + fname + ": file '" + ncfile + "' does not exist !!"
+            quit(-1)
+
         onc = NetCDFFile(ncfile, 'r')
+        print '  ' + fname + ": getting from file '" + ncfile + "' variables:" ,     \
+          varns, 'with slice:', dimvals
         for varn in varns:
             if not onc.variables.has_key(varn):
@@ -9089 +9096 @@
         ov = onc.variables[varns[0]]
         opres = onc.variables[varns[1]]
+        otime = onc.variables['time']
 
         # Getting values
         vals, valdims = drw.slice_variable(ov, dimvals.replace(',','|'))
         pvals, presdims = drw.slice_variable(opres, dimvals.replace(',','|'))
+        tvals, tdims = drw.slice_variable(otime, dimvals.replace(',','|'))
 
         # Expected masked values

trunk/tools/drawing_tools.py

@@ -11406 +11406 @@
 
 def SaturationPressure(temp):
-    """ Provides de saturation pressure at a given temperature following The August-Roche-Magnus equation
+    """ Provides de saturation pressure [hPa] at a given temperature following The 
+      August-Roche-Magnus equation
       temp: temperature [K]
+    >>> SaturationPressure(293.15)
+    23.334406231
     """
     fname = 'SaturationPressure'
@@ -11416 +11419 @@
     
     tempC = temp - 273.15  
-    satpres = 100.*ARM1*(ARM2*tempC)/(tempC+ARM3)
+    satpres = ARM1*exp(ARM2*tempC/(tempC+ARM3))
 
     return satpres
@@ -11423 +11426 @@
 
 def SatMixingRatio(ta,pres):
-    """ Provides de saturation mixing ratio
+    """ Provides de saturation mixing ratio [kgkg-1]
       ta: temperature [K]
       pres: pressure [Pa]
+    >>> SatMixingRatio(293.,101300.)
+    0.014531424608
     """
     fname = 'SatMixingRatio'
 
     es = SaturationPressure(ta)
-    rs = (0.6222*es)/(pres-es)
+    rs = (0.6222*es)/(0.01*pres-es)
 
     return rs
@@ -11463 +11468 @@
     for iz in range(dz):
         ethetapotv[iz,0] = psfc-dp*iz
-        thetapotv = (tsfc+273.15)*((psfc-dp*iz)/p0)**(R/Cp) 
-#        print iz,psfc-dp*iz,thetapotv,SatMixingRatio(tsfc+273.15,psfc),np.exp(L*1000.*SatMixingRatio(tsfc+273.15,psfc)/Cp*(tsfc+273.15))
-        ethetapotv[iz,1] = thetapotv*np.exp(L*1000.*SatMixingRatio(tsfc+273.15,psfc)/(Cp*(tsfc+273.15))) - 273.15
-
-#    quit()
+        thetapotv = (tsfc+273.15)*((psfc-dp*iz)/p0)**(R/Cp)
+        ws = SatMixingRatio(tsfc+273.15,psfc)
+        ethetapotv[iz,1] = thetapotv*np.exp(L*1000.*ws/(Cp*(tsfc+273.15))) -273.15
+        print iz,psfc-dp*iz,thetapotv-273.15,ethetapotv[iz,1]
+
+    #quit()
     return ethetapotv
 
@@ -11533 +11539 @@
     xmax = taxtrm[1]
     xmin = taxtrm[0]
-    print 'len:', np.arange(xmin,xmax,10.)
     if len(np.arange(xmin,xmax,10.)) < 10: xfrqtk = 10.
     else: xfrqtk = 20.
@@ -11620 +11625 @@
     xmax = taxtrm[1]
     xmin = taxtrm[0]
-    print 'len:', np.arange(xmin,xmax,10.)
     if len(np.arange(xmin,xmax,10.)) < 10: xfrqtk = 10.
     else: xfrqtk = 20.
@@ -11639 +11643 @@
     l = ax.axvline(0, color='C0')
 
-    Nlines = len(kindgv['labels'])
+    Nlines = len(tvals)
+    NLlines = len(kindgv['labels'])
+    if Nlines != NLlines:
+        print errormsg
+        print '  ' + fname + ': number of lines:', Nlines,                           \
+          ' and number of labels:', NLlines, 'do not coincide !!'
+        print '    provided labels:', kindgv['labels']
+        quit(-1)
+
+    print '  ' + fname + ': plotting ', Nlines, 'lines ...'
 
     for iline in range(Nlines):
@@ -11694 +11707 @@
     plt.legend(loc=legv[0], prop={'size':legv[1]})
 
+    ax.set_xlabel('t ($^{\circ}$C)')
+    ax.set_ylabel('pres (hPa)')
+
     ax.set_title(gen.latex_text(figtitle))