Changeset 2364 in lmdz_wrf

Timestamp:

Feb 22, 2019, 7:33:44 PM (6 years ago)

Author:

lfita

Message:

Adding:

• `draw_MerPara': Function to add the parallels and meridians into a figure with a panel of plots
• Enhancing `NcolNrow_figure' with how plot meridians and parallels

File:

• trunk/tools/drawing_tools.py (modified) (4 diffs)

trunk/tools/drawing_tools.py

@@ -125 +125 @@
 # check_colorBar: Check if the given colorbar exists in matplotlib
 # colorbar_vals: Function to provide the colorbar values for a figure
+# draw_MerPara: Function to add the parallels and meridians into a figure with a 
+#   panel of plots
 # figureMap_size: Function to determine the real size of a figure adjusted to the 
 #   ratio of the maps
@@ -160 +162 @@
 # multi_plot_2D_shadow: plot of multiple 2D shadowing variables sharing colorbar
 # NcolNrow_figure: Function to provide the number of rows and columns for a panel of 
-#   figures always more rows than columns
+#   figures always more rows than columns and to provide where to draw meridians and 
+#   parallels
 # pixelgrid: Function to provide the kind of grid distribution for a plot
 # plot_points:
@@ -3107 +3110 @@
     return figsizev, txtlineh
 
-def NcolNrow_figure(Npanels):
+def NcolNrow_figure(Npanels, kind='squared'):
     """ Function to provide the number of rows and columns for a panel of figures
-        always more rows than columns
+        and to provide where to draw meridians and parallels
       Npanels: number of panels to be included in the figure
+      kind: kind of panel distribution
+        'squared': equal columns and rows as: Ncol=sqrt(Npanels), Nrow=Npanels/Ncol+1
+           (default). Always more rows than columns
+        'colfix',[Ncol]: fixing [Ncol] number of columns 
+        'rowfix',[Nrow]: fixing [Nrow] number of rows 
+      Returns:
+        [Ncol], [Nrow], [NOpanels], [meridians], [parallels]
+        Ncol: number of columns
+        Nrow: number of rows
+        NOpanels: list with coordinates from axis matrix without assigned plot
+        meridians: list of values wether meridians should be labelled left, right, 
+          or not labelled at all
+        parallels: list of values wether parallels should be labelled bottom, top, 
+          or not labelled at all
     >>> NcolNrow_figure(10)
-    3, 4, [[3, 1], [3, 2]]
+    3, 4, [[3, 1], [3, 2]], ['top', 'top', 'top', 'top', 'no', 'no', 'no', 'no', 
+      'bottom', 'bottom', 'bottom', 'bottom'], ['left', 'no', 'no', 'right', 'left', 
+      'no', 'no', 'right', 'left', 'no', 'no', 'right']
     """
     fname = 'NcolNrow_figure'
 
-    Ncol = int(np.sqrt(Npanels))
-    Nrow = Npanels / Ncol + 1
-
-    if Ncol > Nrow:
-        Ncol0 = Ncol + 0
-        Ncol = Nrow + 0
-        Nrow = Ncol0
+    availkind = ['squared', "'colfix',[Ncol]", "'rowfix',[Nrow]"]
+
+    if kind == 'squared':
+        Ncol = int(np.sqrt(Npanels))
+        Nrow = Npanels / Ncol + 1
+        if Ncol > Nrow:
+            Ncol0 = Ncol + 0
+            Ncol = Nrow + 0
+            Nrow = Ncol0
+    elif kind[0:6] == 'colfix':
+        Ncol = int(kind.split(',')[1])
+        Nrow = Npanels/Ncol
+        if Ncol*Nrow < Npanels: Nrow = Nrow+1
+    elif kind[0:6] == 'rowfix':
+        Nrow = int(kind.split(',')[1])
+        Ncol = Npanels/Nrow
+        if Ncol*Nrow < Npanels: Ncol = Ncol+1
+    else:
+        print errormsg
+        print '  ' +fname+ ": kind of panel distribution '" + kind + "' not ready !!"
+        print '    available ones:', availkind
+        quit(-1)
 
     # panel array without draw
     NOpanels = []
+    meridians = []
+    parallels = []
     panels = np.arange(Ncol*Nrow).reshape(Nrow,Ncol)
     ipanel = 0
@@ -3133 +3169 @@
             ipanel = ipanel + 1
 
-    return Ncol, Nrow, NOpanels
+            if ir == 0: parallels.append('left')
+            elif ir == Nrow-1: parallels.append('right')
+            else: parallels.append('no')
+
+            if ic == 0: meridians.append('top')
+            elif ic == Ncol-1: meridians.append('bottom')
+            else: meridians.append('no')
+
+    return Ncol, Nrow, NOpanels, meridians, parallels
 
 #print NcolNrow_figure(10)
+
+def draw_MerPara(mape, mers, pars, ip, Ncol, Nrow, fsize=8, rotmer=0, rotpar=0,      \
+  col='#080808'):
+    """ Function to add the parallels and meridians into a figure with a panel of 
+       plots
+     map: basemap map environtment
+     mers: list of meridians to plot
+     pars: list of parallels to plot
+     ip: plot number (0, starting)
+     Ncol: number of columns
+     Nrow: number of rows
+     fsize: size of labels (8 default)
+     rotmer: angle of ratation of the labels of the meridians
+     rotpar: angle of ratation of the labels of the parallels
+     col: color of lines ('#080808' default)
+    """
+    fname = 'draw_MerPara'
+
+    # Parallels
+    if np.mod(ip,Ncol) == 0:
+        mape.drawparallels(pars,labels=[True,False,False,True], fontsize=fsize,      \
+          color=col, rotation=rotpar)
+    elif np.mod(ip+1,Ncol) == 0:
+        mape.drawparallels(pars,labels=[False,True,True,False], fontsize=fsize,      \
+          color=col, rotation=rotpar)
+    else:
+        mape.drawparallels(pars,labels=[False,False,False,False], fontsize=fsize,    \
+          color=col, rotation=rotpar)
+
+    if ip+1 > (Nrow-1)*Ncol:
+        mape.drawmeridians(mers, labels=[True,False,False,True], fontsize=fsize,     \
+          color=col, rotation=rotmer)
+# NO on top meridians
+#    elif ip+1 < Ncol:
+#        mape.drawmeridians(mers, labels=[False,True,True,False], fontsize=fsize,     \
+#          color=col, rotation=rotmer)
+    else:
+        mape.drawmeridians(mers, labels=[False,False,False,False], fontsize=fsize,   \
+          color=col, rotation=rotmer)
+
+    return
 
 ####### ###### ##### #### ### ## # END of GENERIC functions