Changeset 1029 in lmdz_wrf

Timestamp:

Aug 17, 2016, 1:10:00 PM (9 years ago)

Author:

lfita

Message:

Working for plot vertical cross section plot. Issue on axis labels

Location:

trunk/tools

Files:

• model_graphics.py (modified) (5 diffs)
• model_graphics_template.dat (modified) (2 diffs)

trunk/tools/model_graphics.py ¶

@@ -125 +125 @@
     for headerf in ftests.keys():
         filen=ftests[headerf]
-        vmod, vdiag = ncvar.computevar_model(var, idir + '/' + filen)
+        try:
+            with gen.Capturing() as output:
+                vmod, vdiag = ncvar.computevar_model(var, idir + '/' + filen)
+        except:
+            print 'ncvar.computevar_model(' + var + ', ' + idir + '/' + filen + ')'
+            for sout in output: print sout
+            quit(-1)
 
         if vmod is None and vdiag is None:
@@ -298 +304 @@
             globalP = pinterp_var(op)
             if vcomp.model is not None:
-                Smodel = ':'.join(vcomp.model)
+                if type(vcomp.model) == type(list([1, 2])):
+                    Smodel = ':'.join(vcomp.model)
+                elif type(vcomp.model) == type('A'):
+                    Smodel = vcomp.model
             else:
                 Smodel = 'None'
             if vcomp.diag is not None:
-                Sdiag = ':'.join(vcomp.diag)
+                if type(vcomp.diag) == type(list([1, 2])):
+                    Sdiag = ':'.join(vcomp.diag)
+                elif type(vcomp.diag) == type('A'):
+                    Sdiag = vcomp.diag
             else:
                 Sdiag = 'None'
@@ -1227 +1239 @@
             fvarS = ','.join(vplot)
 
-            plotins = "python "+pyH+"/drawing.py -f "+figfs+" -o draw_2D_shad_cont "+\
-              "-S '" + graphvals + "' -v " + fvarS
+            drwins = 'draw_2D_shad_cont'
+            plotins = "python " + pyH + "/drawing.py -f " + figfs + " -o " + drwins +\
+              " -S '" + graphvals + "' -v " + fvarS
             try:
                 with gen.Capturing() as output:
                     sout0 = sub.call(plotins, shell=True)
             except:
-                print 'drawing.draw_2D_shad_cont(' + graphvals + ',' + figfs + ',' + \
-                  fvarS + ')'
+                print drwins + '(' + graphvals + ',' + figfs + ',' + fvarS + ')'
                 print sout0
                 for sout in output: print sout
@@ -1292 +1304 @@
                     sout0 = sub.call(plotins, shell=True)
             except:
-                print 'draw_2D_shad_cont_time(' + graphvals + ',' + figfs + ',' +    \
-                  fvarS + ')'
+                print drwins + '(' + graphvals + ',' + figfs + ',' + fvarS + ')'
+                print sout0
+                for sout in output: print sout
+                quit(-1)
+
+            sout = sub.call('mv 2Dfields_shadow-contour.'+kfig+' '+finame, shell=True)
+
+            # keeping all figures
+            trkobjf.write('\n')
+            trkobjf.write("#" + tfig.replace('!',' ') + '\n')
+            trkobjf.write(plotins + '\n')
+
+        elif 'shadcont2Dzsec':
+            figtit = tfig.replace('!','|')
+            shdstdn = CFvplot[0]
+            cntstdn = CFvplot[1]
+            figfs = ','.join(fplot)
+
+            shad = pplot[0]
+            srange = str(shad[0]) + ',' + str(shad[1])
+            cbar = shad[2]
+            cnt = pplot[1]
+            crange = str(cnt[0]) + ',' + str(cnt[1])
+            cfmt = cnt[3]
+            cline = cnt[4]
+
+            # It is assumed that if the space variable is 'lon': is desired a 
+            #   (lon, pres) plot it it is 'lat': then (pres, lat) plot
+            if gen.searchInlist(dplot, 'lon'):
+                spacedim='lon'
+                figmid = 'long.|vert.|cross|sec.|of'
+                figtit = '|'.join(tfig.split('!')[0:2]) + '|' + figmid + '|' +       \
+                  '|'.join(tfig.split('!')[2:])          
+                dims= spacedim+'|-1,pres|-1:'+spacedim+'|-1,pres|-1:'+spacedim+':pres'
+            else:
+                spacedim='lat'
+                figmid = 'mer.|vert.|cross|sec.|of'
+                figtit = '|'.join(tfig.split('!')[0:2]) + '|' + figmid + '|' +       \
+                  '|'.join(tfig.split('!')[2:])
+# NOT WORKING?   dims= spacedim+'|-1,pres|-1:'+spacedim+'|-1,pres|-1:'+'pres:'+spacedim
+                dims= spacedim+'|-1,pres|-1:'+spacedim+'|-1,pres|-1:'+spacedim+':pres'
+
+            reverse = 'flip@y'
+
+            graphvals = ','.join(CFvplot)
+            graphvals = graphvals + ':' + dims + ':' + cbar + ':fixc,' + cline +     \
+              ':' + cfmt + ':' + srange + ':' + crange + ',9:' + figtit + ':' +      \
+              kfig + ':' + reverse + ':None'
+ 
+            fvarS = ','.join(vplot)
+
+            drwins = 'draw_2D_shad_cont'
+            plotins = 'python ' + pyH + '/drawing.py -f ' + figfs + ' -o ' + drwins + \
+              " -S '" + graphvals + "' -v " + fvarS
+            try:
+                with gen.Capturing() as output:
+                    sout0 = sub.call(plotins, shell=True)
+            except:
+                print drwins + '(' + graphvals + ',' + figfs + ',' + fvarS + ')'
                 print sout0
                 for sout in output: print sout
@@ -1443 +1512 @@
                 gP = pinterpS(globalP)
 
-                filen = vn + '_' + headf + gP + '_' + op + '.nc'
+                filen = vn + '_' + headf + gP + '_' + op.replace('+','_') + '.nc'
                 filesplot.append(filen)
                 # Do we have processed the given variable?

trunk/tools/model_graphics_template.dat ¶

@@ -112 +112 @@
 # Text as title for each operation
 #   as ':' separated list of [op]|[explanation '!' as spaces]
-titleoperations = acc|tacc:diff|diff:direct|!:last|last:Lmean|latmean:Lsec|latsec:lmean|lonmean:lsec|lonsec:pinterp|pressure!interpolation:tmean|temporal!mean:turb|turbulence:xmean|xmean:ymean|ymean:zsum|zsum
+titleoperations = acc|tacc:diff|diff:direct|!:last|last:Lmean|latmean:Lsec|latsec:lmean|lonmean:lsec|lonsec:pinterp|pinterp:tmean|tmean:turb|turbulence:xmean|xmean:ymean|ymean:zsum|zsum
 
 # Pressure levels
@@ -139 +139 @@
 DIRPLT_shadcont2Dsfc = uas|last#vas|last:hfls|last#hfss|last:wss|last#tas|last
 DIRPLT_shadconthovmsfc = wss|xmean#tas|xmean
-#DIRPLT_shadcont2Dzsec = ua|pinterp+tmean+xmean#va|pinterp+tmean+xmean:hus|pinterp+tmean+xmean#ta|pinterp+tmean+xmean
+DIRPLT_shadcont2Dzsec = ua|pinterp+tmean+xmean#va|pinterp+tmean+xmean:hus|pinterp+tmean+xmean#ta|pinterp+tmean+xmean
 
 #pltshadcont2Dsfc = huss|tmean#tas|tmean:vas|tmean#uas|tmean:ps|tmean#wss|tmean:pr|tmean#rsds|tmean:prw|tmean#clt|tmean:hfls|tmean#hfss|tmean:evspsbl|tmean#bils|tmean:zmaxth|tmean#stherm|tmean:wakeh|tmean#wakes|tmean:zmaxth|tmean#stherm|tmean:prls|tmean#prc|tmean:clt|tmean#cll|tmean:clm|tmean#clh|tmean:hurs|tmean#zmla|tmean:huss|last#tas|last:vas|last#uas|last:ps|last#wss|tmean:pr|last#rsds|last:prw|last#clt|last:hfls|last#hfss|last:evspsbl|last#bils|last:zmaxth|last#stherm|last:wakeh|last#wakes|last:zmaxth|last#stherm|last:prls|last#prc|last:clt|last#cll|last:clm|last#clh|last:hurs|last#zmla|last