Changeset 932 for trunk

Timestamp:

Apr 16, 2013, 3:06:48 PM (12 years ago)

Author:

aslmd

Message:

UTIL PYTHON planetoplot_v2. fixed small bugs.

Location:

trunk/UTIL/PYTHON/planetoplot_v2

Files:

• ppclass.py (modified) (2 diffs)
• ppplot.py (modified) (3 diffs)

trunk/UTIL/PYTHON/planetoplot_v2/ppclass.py

@@ -508 +508 @@
         mz = findmethod(sz) ; mt = findmethod(st)
         # get number of plots to be done
-        if mx == "fixed": self.nplotx = sx.size/2
-        else:             self.nplotx = 1
-        if my == "fixed": self.nploty = sy.size/2
-        else:             self.nploty = 1
-        if mz == "fixed": self.nplotz = sz.size/2
-        else:             self.nplotz = 1
-        if mt == "fixed": self.nplott = st.size/2
-        else:             self.nplott = 1
+        if mx in ["fixed","comp"]: self.nplotx = sx.size/2
+        else:                      self.nplotx = 1
+        if my in ["fixed","comp"]: self.nploty = sy.size/2
+        else:                      self.nploty = 1
+        if mz in ["fixed","comp"]: self.nplotz = sz.size/2
+        else:                      self.nplotz = 1
+        if mt in ["fixed","comp"]: self.nplott = st.size/2
+        else:                      self.nplott = 1
         if self.verbose:  print "**** OK. Plots over x,y,z,t -->",self.nplotx,self.nploty,self.nplotz,self.nplott
         # create the list of onerequest() objects
@@ -1362 +1362 @@
         else:
             print "!! ERROR !! field would have more than four dimensions ?" ; exit()
+        # dirty hack (AS) ref1_dirty_hack
+        # waiting for more fundamental modifications. case when self.index_y is a bool array.
+        # ... be careful if no point...
+        if type(self.index_x[0]) == np.bool_: nx = np.sum(self.index_x) ## gives the size of the True part!
+        if type(self.index_y[0]) == np.bool_: ny = np.sum(self.index_y) ## gives the size of the True part!
         # NB: ... always 4D array but possibly with "size 1" dimensions 
         #     ... if one dimension is missing because 1D 2D or 3D requests, make it appear again
-        try: 
-            self.field = np.reshape(self.field,(nt,nz,ny,nx))
-        except:
-            # dirty hack (AS) ref1_dirty_hack
-            # waiting for more fundamental modifications. case when self.index_y is a bool array.
-            # ... be careful if no point...
-            nx = np.sum(self.index_x) ## gives the size of the True part!
-            ny = np.sum(self.index_y)
-            self.field = np.reshape(self.field,(nt,nz,ny,nx))
+        self.field = np.reshape(self.field,(nt,nz,ny,nx))
         if self.verbose: print "**** OK. I got %7.1e values. This took me %6.4f seconds" % (nx*ny*nz*nt,timelib.time() - time0)
         if self.verbose: print "**** OK. I got var "+self.var+" with shape",self.field.shape

trunk/UTIL/PYTHON/planetoplot_v2/ppplot.py

@@ -602 +602 @@
                 # (this actually serves for the default setting just above)
                 if self.proj == "spstere" and self.blat > 0: self.blat = -self.blat
+                # labels
+                mertab = np.r_[-360.:360.:15.]
+                partab = np.r_[-90.:90.:5.]
             # ... unsupported projections
             else:
@@ -614 +617 @@
             # draw meridians and parallels
             ft = int(mpl.rcParams['font.size']*3./4.)
-            m.drawmeridians(mertab,labels=merlab,color='grey',linewidth=0.75,fontsize=ft,fmt=format)
-            m.drawparallels(partab,labels=parlab,color='grey',linewidth=0.75,fontsize=ft,fmt=format)
+            zelatmax = 85.
+            m.drawmeridians(mertab,labels=merlab,color='grey',linewidth=0.75,fontsize=ft,fmt=format,latmax=zelatmax)
+            m.drawparallels(partab,labels=parlab,color='grey',linewidth=0.75,fontsize=ft,fmt=format,latmax=zelatmax)
             # define background (see set_back.txt)
             if self.back is not None:
@@ -653 +657 @@
         stride = 3
         if self.addvecx is not None and self.addvecy is not None and self.mapmode:
-                ## for metwinds only ???
+                ### for metwinds only ???
                 [vecx,vecy] = m.rotate_vector(self.addvecx,self.addvecy,self.absc,self.ordi) 
+                #vecx,vecy = self.addvecx,self.addvecy
                 # reference vector is scaled
                 zescale = ppcompute.mean(np.sqrt(self.addvecx*self.addvecx+self.addvecy*self.addvecy))