Changeset 1078 in lmdz_wrf for trunk

Timestamp:

Sep 1, 2016, 6:48:08 PM (9 years ago)

Author:

lfita

Message:

Working version with 2lines', 2linesTime'

File:

• trunk/tools/model_graphics.py (modified) (7 diffs)

trunk/tools/model_graphics.py

@@ -791 +791 @@
         tvar: temporal variance values (Taylor's turbulence)
         xmean: x-axis mean values
+        xvar: x-axis variance values
         ymean: y-axis mean values
         zsum: vertical aggregated values
@@ -1152 +1153 @@
             varkeep.append('lonstats')
 
+        elif op == 'xvar':
+            # x-axis variance values
+            vals='lon|-1,lon,var,' + ':'.join(varnCFs) + ':' + vdnz
+            dims = gen.dictvar_listS(varnCFs,CFdimvardict,',','@') + dnz+'@'+vdnz
+            if dofile:
+                try:
+                    with gen.Capturing() as output:
+                        ncvar.file_oper_alongdims(vals,prevfile,CFvarnp)
+                except:
+                    print errmsg
+                    print 'file_oper_alongdims('+vals+',', prevfile, ','+CFvarnp+')'
+                    for s1out in output: print s1out
+                    quit(-1)
+
+                if db:
+                    for s1out in output: print s1out
+
+                sout = sub.call('mv file_oper_alongdims_var.nc '+fileon, shell=True)
+
+                # Keeping the operations
+                pyins=pyH + "/nc_var.py -o file_oper_alongdims -S '" + vals +        \
+                  "' -f " + prevfile + " -v " + CFvarnp
+                otrackf.write("\n")
+                otrackf.write("# " + CFvarn + " " + Fopers + "\n")
+                otrackf.write(pyins + "\n")
+
+            # removing dimension variable-dimension 'lon'
+            varnCFs.remove('lon')
+
+            varkeep.append('lonstats')
+
         elif op == 'ymean':
             # y-axis mean values
@@ -1570 +1602 @@
             if gen.searchInlist(dplot,'lon'):
                 spacedim ='lon'
-                figmid = 'longitudinal|evolution|of'
-                figtit = '|'.join(tfig.split('!')[0:2]) + '|' + figmid + '|' +       \
-                  '|'.join(tfig.split('!')[2:])
-                reverse= 'None'
-                dims= spacedim+'|-1,time|-1;'+spacedim+'|-1,time|-1;'+spacedim+';time'
+                axisvals = 'y'
+                figmid = 'longitudinal'
+            elif gen.searchInlist(dplot,'lat'):
+                spacedim = 'lat'
+                axisvals = 'x'
+                figmid = 'meridional'
+            elif gen.searchInlist(dplot,'pres'):
+                spacedim = 'pres'
+                axisvals = 'x'
+                figmid = 'vertical'
             else:
-                spacedim='lat'
-                figmid = 'meridional|evolution|of'
-                figtit = '|'.join(tfig.split('!')[0:2]) + '|' + figmid + '|' +       \
-                  '|'.join(tfig.split('!')[2:])
-                reverse='None'
-                dims= spacedim+'|-1,time|-1;'+spacedim+'|-1,time|-1;time;'+spacedim
-
-            graphvals = ','.join(CFvplot)
-            graphvals = graphvals + ';'+ dims +';'+ cbar +  \
-              ';' + ckind + ',' + cline + ';' + cfmt + ';' + srange + ';' + crange + \
-              ',9;' + figtit + ';' + kfig + ';' + reverse + ';' + 'time|' + tunits + \
-              '|'+ timekind + '|' + timefmt + '|' + timelabel
+                print errmsg
+                print '  ' + fname + ": in '2lines' only ready for: 'lon', 'lat'," + \
+                 " 'pres' as common dimension !!"
+                print '  dimensions in plot:', dplot
+                quit(-1)
+
+            figtit = '|'.join(tfig.split('!')[0:2]) + '|' + figmid + '|' +           \
+              '|'.join(tfig.split('!')[2:])
+
+            graphvals = spacedim + ':' + Arange + ':' + Brange + ':Extrs:' +         \
+              axisvals + ':' + ','.join(CFvplot) + ':' + Aline + ',' + Bline +       \
+              ':2.,2.:' + Akind + ',' + Bkind + ':,:' + figtit + ':' + spacedim +    \
+              ':0:' +  finame + ':' + kfig
 
             fvarS = ','.join(vplot)
 
-            drwins = 'draw_2D_shad_cont_time'
+            drwins = 'draw_2lines'
             plotins = 'python ' + pyH + '/drawing.py -f ' + figfs +' -o ' + drwins + \
               " -S '" + graphvals + "' -v " + fvarS
-
-
-      values= [dimvname]:[valuesaxis]:[dimtit]:[leglabels]:[vtit]:[title]:[locleg]:[colns]:[lines]
-       [points]:[lwdths]:[psizes]:[freqv]:[figname][graphk]
-        [dimvname]: ',' list of names of the variable with he values of the common dimension
-        [valuesaxis]: which axis will be used for the values ('x', or 'y')
-        [dimtit]: title for the common dimension
-        [leglabels]: ',' separated list of names for the legend
-        [vartit]: name of the variable in the graph
-        [title]: title of the plot ('|' for spaces)
-        [locleg]: location of the legend (0, autmoatic)
-          1: 'upper right', 2: 'upper left', 3: 'lower left', 4: 'lower right',
-          5: 'right', 6: 'center left', 7: 'center right', 8: 'lower center',
-          9: 'upper center', 10: 'center'
-        [colns]= ',' list of color names ('None' for automatic, single value for all the same)
-        [lines]= ',' list of style of lines ('None' for automatic, single value for all the same)
-        [points]= ',' list of style of points ('None' for automatic, single value for all the same)
-        [lwdths]= ',' list of withs of lines ('None' for automatic, single value for all the same)
-        [psizes]= ',' list of size of points ('None' for automatic, single value for all the same)
-        [freqv]= frequency of values ('all' for all values)
-        [figname]= name of the figure
-        [graphk]: kind of the graphic
-      varname= variable to plot
- 
 
             try:
@@ -1629 +1643 @@
                 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 kplot == '2linesTime':
+            figtit = tfig.replace('!','|')
+
+            lineAstdn = CFvplot[0]
+            lineBstdn = CFvplot[1]
+            figfs = ','.join(fplot)
+
+            lineA = pplot[0]
+            Arange = str(lineA[0]) + ',' + str(lineA[1])
+            # Values are changed from specific values in `model_graphics.dat'
+            if lineA[3].find('%') == -1:
+                Aline = lineA[2]
+                Akind = lineA[3]
+                Amark = lineA[4]
+                Asize = lineA[5]
+            else:
+                Aline = 'red'
+                Akind = '-'
+                Amark = ','
+                Asize = 2.
+
+            lineB = pplot[1]
+            Brange = str(lineB[0]) + ',' + str(lineB[1])
+            # Values are changed from specific values in `model_graphics.dat'
+            if lineB[3].find('%') == -1:
+                Bline = lineB[2]
+                Bkind = lineB[3]
+                Bmark = lineB[4]
+                Bsize = lineB[5]
+            else:
+                Bline = 'red'
+                Bkind = '-'
+                Bmark = ','
+                Bsize = 2.
+
+            timeaxis = 'x'
+            figmid = 'evolution'
+
+            figtit = '!'.join(tfig.split('!')[0:2]) + '!' + figmid + '!' +           \
+              '!'.join(tfig.split('!')[2:])
+
+            graphvals = 'time:' + Arange + ':' + Brange + ':' + timekind + ';' +     \
+              timefmt + ':' + timeaxis + ':' + ','.join(CFvplot) + ':' + Aline + ',' \
+              + Bline + ':2.,2.:' + Akind + ',' + Bkind + ':2.,2.:,:' + figtit + ':' \
+              + timelabel + ':0:' +  finame + ':' + kfig
+
+            fvarS = ','.join(vplot)
+
+            drwins = 'draw_2lines_time'
+            plotins = 'python ' + pyH + '/drawing.py -f ' + figfs +' -o ' + drwins + \
+              " -S '" + graphvals + "' -v " + fvarS
+
+            try:
+                with gen.Capturing() as output:
+                    sout = sub.call(plotins, shell=True)
+            except:
+                print errmsg
+                print drwins + '(' + graphvals + ',' + figfs + ',' + fvarS + ')'
+                print sout
+                for s1out in output: print s1out
+                quit(-1)
+
+            # keeping all figures
+            trkobjf.write('\n')
+            trkobjf.write("#" + tfig.replace('!',' ') + '\n')
+            trkobjf.write(plotins + '\n')
+
         elif kplot == 'shadcont2Dsfc':
             figtit = tfig.replace('!','|')
@@ -1985 +2065 @@
                 for op1 in opvals:
                     if not opexplained.has_key(op1):
-                        print errormsg
+                        print errmsg
                         print '  '+fname+": no explanation for operation '"+op1+"' !!"
                         print '    provided:', opexplained.keys()
@@ -2344 +2424 @@
         tvar: temporal variance values (Taylor's turbulence)
         xmean: x-axis mean values
+        xvar: x-axis variance values
         ymean: y-axis mean values
         zsum: vertical aggregated values
@@ -2610 +2691 @@
             varkeep.append('lonstats')
 
+        elif op == 'xvar':
+            # x-axis variance values
+            vals='lon|-1,lon,var,' + ':'.join(varnCFs) + ':' + vdnz
+            dims = gen.dictvar_listS(varnCFs,CFdimvardict,',','@') + dnz+'@'+vdnz
+            if dofile:
+                try:
+                    with gen.Capturing() as output:
+                        ncvar.file_oper_alongdims(vals,prevfile,CFvarnp)
+                except:
+                    print errmsg
+                    print 'file_oper_alongdims('+vals+',', prevfile, ','+CFvarnp+')'
+                    for s1out in output: print s1out
+                    quit(-1)
+
+                if db:
+                    for s1out in output: print s1out
+
+                sout = sub.call('mv file_oper_alongdims_var.nc '+fileon, shell=True)
+
+                # Keeping the operations
+                pyins=pyH + "/nc_var.py -o file_oper_alongdims -S '" + vals +        \
+                  "' -f " + prevfile + " -v " + CFvarnp
+                otrackf.write("\n")
+                otrackf.write("# " + CFvarn + " " + Fopers + "\n")
+                otrackf.write(pyins + "\n")
+
+            # removing dimension variable-dimension 'lon'
+            varnCFs.remove('lon')
+
+            varkeep.append('lonstats')
+
         elif op == 'ymean':
             # y-axis mean values