Changeset 1080 in lmdz_wrf for trunk

Timestamp:

Sep 1, 2016, 8:14:41 PM (9 years ago)

Author:

lfita

Message:

Adding:

• `optitle': Function to creat the operation section in the title of a figure ('!' for spaces)
• `create_figure_title': Function to create the title of a figure ('!' for spaces)

File:

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

trunk/tools/model_graphics.py

@@ -1625 +1625 @@
               axisvals + ':' + ','.join(CFvplot) + ':' + Aline + ',' + Bline +       \
               ':2.,2.:' + Akind + ',' + Bkind + ':,:' + figtit + ':' + spacedim +    \
-              ':0:' +  finame + ':' + kfig
+              ':0:' +  finame.replace('.'+kfig, '') + ':' + kfig
 
             fvarS = ','.join(vplot)
@@ -1692 +1692 @@
               timefmt + ':' + timeaxis + ':' + ','.join(CFvplot) + ':' + Aline + ',' \
               + Bline + ':2.,2.:' + Akind + ',' + Bkind + ':2.,2.:,:' + figtit + ':' \
-              + timelabel + ':0:' +  finame + ':' + kfig
+              + timelabel + ':0:' +  finame.replace('.'+kfig, '') + ':' + kfig
 
             fvarS = ','.join(vplot)
@@ -1885 +1885 @@
     return
 
+def optitle(op,opexp):
+    """ Function to creat the operation section in the title of a figure ('!' for spaces)
+      op= '+' separated list of operations
+      opexp= dictionary with the `explanation'(text in title) to appear for each operation
+    >>> optitle('pinterp+xmean+tmean',{'pinterp':'pinterp', 'xmean':'xmean', 'tmean':'tmean'})
+    $_{[pinterp\!xmean\!tmean]}$
+    """
+    fname = 'optitle'
+
+    opvals = op.split('+')
+    for op1 in opvals:
+        if not opexp.has_key(op1):
+            print errmsg
+            print '  '+fname+": no explanation for operation '"+op1+"' !!"
+            print '    provided:', opexp.keys()
+            quit(-1)
+        if op1 == opvals[0]:
+            titop = '$_{[' + opexp[op1].replace(' ','!')
+            if len(opvals) == 1: titop = titop + ']}$'
+        elif op1 == opvals[len(opvals)-1]:
+            titop = titop + '\!' + opexp[op1].replace(' ','!') + ']}$'
+        else:
+            titop = titop + '\!' + opexp[op1].replace(' ','!')
+
+    return titop
+
+def create_figure_title(mod,exp,varops,explop):
+    """ Function to create the title of a figure ('!' for spaces)
+      mod: model name
+      exp: experiment name
+      varops: list with [var]|[op] values from wich create the title
+      explop: dictionary with the `explanation'(text in title) to appear for each operation
+    >>> expops = {'pinterp':'pinterp', 'xmean':'xmean', 'tmean':'tmean', 'last':'last'}
+    >>> create_figure_title('WRF','current',['ua|pinterp+xmean+tmean', 'va|pinterp+xmean+tmean'], expops)
+    WRF!current!ua!&!va$_{[pinterp\!xmean\!tmean]}$
+    >>> create_figure_title('WRF','current',['ua|pinterp+xmean+tmean', 'ua|pinterp+xmean+last'], expops)
+    WRF!current!ua$_{[pinterp\!xmean\!tmean]}$!&!!$_{[pinterp\!xmean\!last]}$
+    >>> create_figure_title('WRF','current',['ua|pinterp+xmean+tmean', 'va|pinterp+xmean+last'], expops)
+    WRF!current!ua$_{[pinterp\!xmean\!tmean]}$!&!va$_{[pinterp\!xmean\!last]}$
+    """
+    fname = 'create_figure_title'
+
+    titfig = mod + '!' + exp + '!'
+
+    varns = []
+    opns = []
+    for varop in varops:
+        vn = varop.split('|')[0]
+        op = varop.split('|')[1]
+        varns.append(vn)
+        opns.append(op)
+        if varop == varops[0]:
+            allvarequal = True
+            allopequal = True
+            varinfig = vn
+            opinfig = op
+        else:
+            if vn != varinfig: allvarequal = False
+            if op != opinfig: allopequal = False
+
+    Nvars = len(varns)
+    Nops = len(opns)
+
+    if allvarequal:
+        titfig = titfig + '!' + varns[0]
+        if allopequal:
+            opS = optitle(op,explop)
+            titfig = titfig + opS
+        else:
+            for opn in opns:
+                opS = optitle(opn,explop)
+                if opn == opns[0]:
+                    titfig = titfig + opS 
+                elif opn == opns[Nops-1]:
+                    titfig = titfig + '!&!' + opS
+                else:
+                    titfig = titfig + ',!' + opS
+    else: 
+        if allopequal:
+            opS = optitle(op,explop)
+            for vn in varns:
+                if vn == varns[0]:
+                    titfig = titfig + '!' + vn
+                elif vn == varns[Nvars-1]:
+                    titfig = titfig + '!&!' + vn
+                else:
+                    titfig = titfig + ',!' + vn
+            titfig = titfig + opS
+        else:
+            for ivop in range(Nvars):
+                vn = varns[ivop]
+                op = opns[ivop]
+                vnop = vn + optitle(op,explop)
+                
+                if vn == varns[0]:
+                    titfig = titfig + '!' + vnop
+                elif vn == varns[Nvars-1]:
+                    titfig = titfig + '!&!' + vnop
+                else:
+                    titfig = titfig + ',!' + vnop
+
+    return titfig
+
 def draw_plots(config, plots, mod, exp, odir, allvarcomp, figscr, debug):
     """ Function to draw all plots
@@ -2057 +2160 @@
                     figname = figname + '-'+vn+'_'+headf+'_'+op.replace('+','-')
 
-                # Title of the figure:
-                if ivp == 0:
-                    titfigure = mod + '!' + exp + "!'" + vn + "'"
-                else:
-                    titfigure = titfigure + "!&!'" + vn + "'"
-                opvals = op.split('+')
-                for op1 in opvals:
-                    if not opexplained.has_key(op1):
-                        print errmsg
-                        print '  '+fname+": no explanation for operation '"+op1+"' !!"
-                        print '    provided:', opexplained.keys()
-                        quit(-1)
-                    if op1 == opvals[0]:
-                        titfigure = titfigure + '$_{[' + opexplained[op1]
-                        if len(opvals) == 1: titfigure = titfigure + ']}$'
-                    elif op1 == opvals[len(opvals)-1]:
-                        titfigure = titfigure + '\!' + opexplained[op1] + ']}$'
-                    else:
-                        titfigure = titfigure + '\!' + opexplained[op1]
-
                 ivp = ivp + 1
             # End of variable-operation
             figname = figname + '.' + kindfigure
+
+            # Title of figure
+            titfigure = create_figure_title(mod, exp, varops, opexplained)
 
             draw_plot(kplot, CFvarsplot, filesplot, varsplot, dimsplot, pictplot,    \
@@ -2693 +2779 @@
         elif op == 'xvar':
             # x-axis variance values
-            vals='lon|-1,lon,var,' + ':'.join(varnCFs) + ':' + vdnz
-            dims = gen.dictvar_listS(varnCFs,CFdimvardict,',','@') + dnz+'@'+vdnz
+            vals='lon|-1,lon,var,' + ':'.join(varnCFs)
+            dims = gen.dictvar_listS(varnCFs,CFdimvardict,',','@')
             if dofile:
                 try:
@@ -2714 +2800 @@
                   "' -f " + prevfile + " -v " + CFvarnp
                 otrackf.write("\n")
-                otrackf.write("# " + CFvarn + " " + Fopers + "\n")
+                otrackf.write("# " + varn + " " + Fopers + "\n")
                 otrackf.write(pyins + "\n")
 
@@ -3084 +3170 @@
                 headf = vopvals[4]
 
-                modexpdiff = mod2 + '/' + exp2 + ' - ' + mod1 + '/' + exp1
+                modexpdiff = mod2 + '/' + exp2 + '-' + mod1 + '/' + exp1
                 modexpdiffS = mod2 + '-' + exp2 + '_' + mod1 + '-' + exp1
 
@@ -3137 +3223 @@
                     modexptit = modexpdiff
 
-                if ivp == 0:
-                    titfigure = modexptit + "!" + kdiff + "!'" + vn + "'"
-                    vnS = vn
-                else:
-                    titfigure = titfigure + "!&!'" + vn + "'"
-                    vnS = vnS + '!&!' + vn
-
-                opvals = op.split('+')
-                for op1 in opvals:
-                    if not opexplained.has_key(op1):
-                        print errormsg
-                        print '  '+fname+": no explanation for operation '"+op1+"' !!"
-                        print '    provided:', opexplained.keys()
-                        quit(-1)
-                    if op1 == opvals[0]:
-                        titfigure = titfigure + '$_{[' + opexplained[op1]
-                        if len(opvals) == 1: titfigure = titfigure + ']}$'
-                    elif op1 == opvals[len(opvals)-1]:
-                        titfigure = titfigure + '\!' + opexplained[op1] + ']}$'
-                    else:
-                        titfigure = titfigure + '\!' + opexplained[op1]
-
                 ivp = ivp + 1
             # End of variable-operation
             figname = figname + '.' + kindfigure
+
+            # Title of figure
+            titfigure = create_figure_title(modexptit, kdiff, varops, opexplained)
 
             if len(titfigure) > 80: