Changeset 827 in lmdz_wrf

Timestamp:

Jun 16, 2016, 12:10:20 AM (9 years ago)

Author:

lfita

Message:

Working version of `draw_subbasin'

File:

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

trunk/tools/drawing_tools.py

@@ -6979 +6979 @@
     ddx = np.abs(lons[dy/2+1,dx/2] - lons[dy/2,dx/2])
     ddy = np.abs(lats[dy/2,dx/2+1] - lats[dy/2,dx/2])
-    fontcharac = {'family': 'serif', 'weight': 'normal', 'size': 3}
-
-    print '  ' + fname + ' Lluis; rsf:', rsf, type(rsf)
+    fontcharac = {'family': 'serif', 'weight': 'normal', 'size': 10}
+
     subhr = gen.chainSnumHierarchy(rsf)
 
@@ -7003 +7002 @@
             collev[gen.num_chainSnum(isub+1)] = rcolors[isub]
 
-    print ' Lluis collev:', collev
-
 # All the colors should finish white...
     endcol = [1., 1., 1.]
@@ -7018 +7015 @@
         levs = gen.num_split(isub)
         Nlevs = len(levs)
-        begcol = collev[levs[0]]
+        if Nlevs == 1:
+            begcol = [0., 0., 0.]
+        else:
+            begcol = collev[levs[1]]
         for ilev in range(2,Nlevs):
 # We do not want a last white value...
@@ -7028 +7028 @@
             begcol = colors[1:4,gen.chainSnum_num(levs[ilev])-1]
 
-#        colorsub[isub] = gen.coldec_hex(begcolors)
         colorsub[isub] = tuple(begcol)
         Llfs = rLlf[isub]
         outfs = rof[isub]
-        print '  Lluis; ' + fname + ': isub:', isub,' outfs:', outfs
         for igrid in range(len(outfs)):
             Llf = Llfs[igrid]
@@ -7063 +7061 @@
     plt.rc('text', usetex=True)
 
-    latlonss = rLlf.values()
-    nlon = np.min(latlonss[1])
-    xlon = np.max(latlonss[1])
-    nlat = np.min(latlonss[0])
-    xlat = np.max(latlonss[0])
+    nlon = np.min(xlabpos)
+    xlon = np.max(xlabpos)
+    nlat = np.min(ylabpos)
+    xlat = np.max(ylabpos)
 
     dlon = xlon - nlon
@@ -7073 +7070 @@
 
 # Making bigger the area to map
-    nlon = nlon-dlon*0.1
-    xlon = xlon+dlon*0.1
-    nlat = nlat-dlat*0.1
-    xlat = xlat+dlat*0.1 
+    nlon = nlon-dlon-lengthtrac2
+    xlon = xlon+dlon+lengthtrac2
+    nlat = nlat-dlat-lengthtrac2
+    xlat = xlat+dlat+lengthtrac2
 
     plt.xlim(nlon,xlon)
@@ -7118 +7115 @@
 
     j = 0
-    for i in range(len(rsf)):
+    for i in range(len(xtrack)/3):
         plt.plot(xtrack[j:j+2], ytrack[j:j+2], color=colortrack[i])
         j = j + 3
-
-    print 'Lluis : xtrack shape', len(xtrack)/3, 'xlabpos:', len(xlabpos)
 
 # Sea-flow