Changeset 930 for trunk/UTIL/PYTHON

Timestamp:

Apr 15, 2013, 12:55:29 PM (12 years ago)

Author:

aslmd

Message:

UTIL PYTHON planetoplot_v2. Added attribute allfield so that the user can easily access values. See examples/easy_get_field.py. Added attribute includedate to select option for naming files. Corrected index bug in meanarea mode. Added an example intercompare.py which shows how to easily compare results in two quite different netCDF files.

Location:

trunk/UTIL/PYTHON/planetoplot_v2

Files:

• examples/easy_get_field.py (added)
• examples/intercompare.py (added)
• exapi.py (added)
• ppclass.py (modified) (11 diffs)
• ppplot.py (modified) (2 diffs)
• set_area.txt (modified) (1 diff)

trunk/UTIL/PYTHON/planetoplot_v2/ppclass.py

@@ -100 +100 @@
                 zesaxis[i,1] = float(a[1])          
     return zesaxis
+
+# look for comas in the input name to separate different names (files, variables,etc ..)
 # (needed by readslices)
-# look for comas in the input name to separate different names (files, variables,etc ..)
 def separatenames (name):
     if name is None: names = None
@@ -141 +142 @@
                       filename="myplot",\
                       folder="./",\
+                      includedate=True,\
                       xlabel=None,ylabel=None,\
                       xcoeff=None,ycoeff=None,\
@@ -162 +164 @@
         self.p = None
         self.customplot = False
+        self.allfield = None
         ## what could be defined by the user
         self.file = file
@@ -180 +183 @@
         self.filename = filename
         self.folder = folder
+        self.includedate = includedate
         ## here are user-defined plot settings 
         ## -- if not None, valid on all plots in the pp() objects
@@ -238 +242 @@
             self.label = other.label
             self.title = other.title
+            self.includedate = other.includedate
         else:
             print "!! ERROR !! argument must be a pp object." ; exit()
@@ -560 +565 @@
         # check if things were done OK before
         if self.status != "defined": print "!! ERROR !! Please use .define() to define your pp object." ; exit()
+        # create the list of allfield() objects
+        # --> so that the user can easily access values
+        self.allfield = [[[[[[ \
+                        [] \
+                        for x in range(self.nplotx)] for y in range(self.nploty)] \
+                        for z in range(self.nplotz)] for t in range(self.nplott)] \
+                        for j in range(self.nvin)]   for i in range(self.nfin)]
         ## first get fields
         ## ... only what is needed is extracted from the files
@@ -572 +584 @@
               obj.getfield()
               obj.computations()
+              self.allfield[i][j][t][z][y][x] = obj.field
+        # get rid of useless dimensions in allfield
+        self.allfield = np.squeeze(self.allfield)
         # change status
         self.status = "retrieved"
@@ -860 +875 @@
         print "**** PPCLASS. Done step: makeplot"
         if (self.n == self.howmanyplots):
-            ppplot.save(mode=self.out,filename=self.filename,folder=self.folder,custom=self.customplot)
+            ppplot.save(mode=self.out,filename=self.filename,folder=self.folder,custom=self.customplot,includedate=self.includedate)
             mpl.close()
         # SAVE A PICKLE FILE WITH THE self.p ARRAY OF OBJECTS
@@ -1380 +1395 @@
           pass
         elif test in ["fixedfree","fixedcomp"] or test in ["freefixed","compfixed"]: 
-          time0 = timelib.time()  
+          time0 = timelib.time()
           # now have to obtain the new indexes which correspond to the extracted self.field
           # for it to work with unique index, ensure that any index_* is a numpy array
@@ -1518 +1533 @@
                 totarea = np.tile(totarea,(1,self.index_x.size))
         elif self.method_x == "comp" and self.method_y == "comp":
+            aire.field = aire.field[np.ix_(self.index_y, self.index_x)] # reduce to requested indexes only
             totarea = ppcompute.sum(ppcompute.sum(aire.field,axis=1),axis=0)
         else:
@@ -1527 +1543 @@
         aire.field = np.tile(aire.field,(self.index_t.size,self.index_z.size,1,1))
         if self.field.shape != aire.field.shape:
-            print "!! ERROR !! Problem in area(). Check array shapes." ; exit()
+            print "!! ERROR !! Problem in area(). Check array shapes."
+            print "Field vs. aire:",self.field.shape,aire.field.shape ; exit()
         else:
             self.field = self.field*aire.field

trunk/UTIL/PYTHON/planetoplot_v2/ppplot.py

@@ -418 +418 @@
                 mpl.legend(loc="best",fancybox=True)
         ## TBD: set with .div the number of ticks
-        ## TBD: be able to control plot limits
         #ticks = self.div + 1
         #ax = mpl.gca()
         #ax.get_xaxis().set_ticks(np.linspace(ppcompute.min(x),ppcompute.max(x),ticks/2+1))
+        # control plot limits (TBD: be able to control those)
+        ax = mpl.gca() ; ax.set_xbound(lower=ppcompute.min(x), upper=ppcompute.max(x))
+
 
 ################################
@@ -660 +662 @@
                               angles='xy',color=self.colorvec,pivot='middle',\
                               scale=zescale*reducevec,width=widthvec )
-                # make vector key. default is on upper left corner.
-                keyh = 1.025 ; keyv = 1.05
-                #keyh = -0.03 ; keyv = 1.08
+                # make vector key.
+                keyh = 1.025 ; keyv = 1.05 # upper right corner over colorbar
+                #keyh = -0.03 ; keyv = 1.08 # upper left corner
                 p = mpl.quiverkey(q,keyh,keyv,\
                                   zescale,str(int(zescale)),\

trunk/UTIL/PYTHON/planetoplot_v2/set_area.txt

@@ -24 +24 @@
 Xanadu ; -40 ; 20 ; 40 ; 120
 Hyperboreae ; 80 ; 87 ; -70 ; -10
+VM ; -15 ; 0 ; -90 ; -60
 VM_alt ; -15 ; 0 ; 270 ; 300