Changeset 1072 in lmdz_wrf

Timestamp:

Aug 31, 2016, 6:12:53 PM (9 years ago)

Author:

lfita

Message:

Starting to add plot `2lines'

File:

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

trunk/tools/model_graphics.py

@@ -118 +118 @@
         scr = True
         print warnmsg
-        print "  " + main + ": starting from the SCRATCH !!"
+        print "  " + fname + ": starting from the SCRATCH !!"
         print "    10 seconds left!!"
         filescr = True
@@ -183 +183 @@
             figmoddifscr = False
 
-    if config['addfiles'] == 'true':
+    if config['addall'] == 'true':
+        print warnmsg
+        print "  " + fname + ": adding new values everywhere !!"
         addfils = True
-    else:
-        addfils = False
-
-    if config['addfigures'] == 'true':
         addfigs = True
-    else:
-        addfigs = False
-
-    if config['adddiffs'] == 'true':
         adddiffs = True
-    else:
-        adddiffs = False
-
-    if config['adddifffigures'] == 'true':
         adddifffigs = True
-    else:
-        adddifffigs = False
-
-    if config['addmoddiffs'] == 'true':
         addmoddiffs = True
-    else:
-        addmoddiffs = False
-
-    if config['addmoddifffigures'] == 'true':
         addmoddifffigs = True
     else:
-        addmoddifffigs = False
+        if config['addfiles'] == 'true':
+            addfils = True
+        else:
+            addfils = False
+
+        if config['addfigures'] == 'true':
+            addfigs = True
+        else:
+            addfigs = False
+
+        if config['adddiffs'] == 'true':
+            adddiffs = True
+        else:
+            adddiffs = False
+
+        if config['adddifffigures'] == 'true':
+            adddifffigs = True
+        else:
+            adddifffigs = False
+
+        if config['addmoddiffs'] == 'true':
+            addmoddiffs = True
+        else:
+            addmoddiffs = False
+
+        if config['addmoddifffigures'] == 'true':
+            addmoddifffigs = True
+        else:
+            addmoddifffigs = False
 
     if config['debug'] == 'true':
@@ -1539 +1549 @@
             print "  " + fname + ": kind of plot '" + kplot + "' not ready !!"
             quit(-1)
-        elif kplot == 'shadcont2Dsfc':
+        elif kplot == '2lines':
             figtit = tfig.replace('!','|')
-            shdstdn = CFvplot[0]
-            cntstdn = CFvplot[1]
+
+            lineAstdn = CFvplot[0]
+            lineBstdn = 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]
-            ckind = cnt[4]
-            cline = cnt[5]
-
-            graphvals = ','.join(CFvplot)
-            graphvals = graphvals + ':lon|-1,lat|-1:lon|-1,lat|-1:lon:lat:' + cbar + \
-              ':' + ckind + ',' + cline + ':' + cfmt + ':' + srange + ':' + crange + \
-              ',9:' + figtit + ':' + kfig + ':None:' + mapval
-
-            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 errmsg
-                print drwins + '(' + graphvals + ',' + figfs + ',' + fvarS + ')'
-                print sout0
-                for s1out in output: print s1out
-                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 == 'shadconthovmsfc':
-            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]
-            ckind = cnt[4]
-            cline = cnt[5]
+            lineA = pplot[0]
+            Arange = str(lineA[0]) + ',' + str(lineA[1])
+            Akind = lineA[4]
+            Aline = lineA[5]
+
+            lineB = pplot[0]
+            Brange = str(lineB[0]) + ',' + str(lineB[1])
+            Bkind = lineB[4]
+            Bline = lineB[5]
+
             # It is assumed that if the space variable is 'lon': is desired a 
-            #   (lon, time) plot it it is 'lat': then (time, lat) plot
+            #   (lon, vals) plot if it is 'lat': then (vals, lat) plot
             if gen.searchInlist(dplot,'lon'):
                 spacedim ='lon'
@@ -1624 +1595 @@
               " -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:
                 with gen.Capturing() as output:
-                    sout0 = sub.call(plotins, shell=True)
+                    sout = sub.call(plotins, shell=True)
             except:
                 print errmsg
                 print drwins + '(' + graphvals + ',' + figfs + ',' + fvarS + ')'
+                print sout
+                for s1out in output: print s1out
+                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 == 'shadcont2Dsfc':
+            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]
+            ckind = cnt[4]
+            cline = cnt[5]
+
+            graphvals = ','.join(CFvplot)
+            graphvals = graphvals + ':lon|-1,lat|-1:lon|-1,lat|-1:lon:lat:' + cbar + \
+              ':' + ckind + ',' + cline + ':' + cfmt + ':' + srange + ':' + crange + \
+              ',9:' + figtit + ':' + kfig + ':None:' + mapval
+
+            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:
+                    sout = sub.call(plotins, shell=True)
+            except:
+                print errmsg
+                print drwins + '(' + graphvals + ',' + figfs + ',' + fvarS + ')'
+                print sout
+                for s1out in output: print s1out
+                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 == 'shadconthovmsfc':
+            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]
+            ckind = cnt[4]
+            cline = cnt[5]
+            # It is assumed that if the space variable is 'lon': is desired a 
+            #   (lon, time) plot if it is 'lat': then (time, lat) plot
+            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'
+            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
+
+            fvarS = ','.join(vplot)
+
+            drwins = 'draw_2D_shad_cont_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)
@@ -3549 +3646 @@
             cdovars = CDOremapvar[cdoop]
             for cdovar in cdovars:
-                cdofs = gen.files_folder_HMT(folder=cdofdir, head=cdovar, tail='.nc')
+                cdofs= gen.files_folder_HMT(folder=cdofdir,head=cdovar+'_',tail='.nc')
                 for cdof in cdofs:
                     # Getting direct files as [var]_[fhead].nc