Changeset 1029 for trunk/UTIL/PYTHON

Timestamp:

Sep 5, 2013, 2:28:49 PM (11 years ago)

Author:

aslmd

Message:

UTIL PYTHON planetoplot_v2

• added keywords nxticks nyticks xp yp missing for improved plot settings
• added function to print contents of pp object
• added to ppplot a function figureref to easily set a figure of given size
• bug fix affecting .ppobj objects containing fields with missing values
• bug fix when size of z coordinate do not match field
• bug fix for Ls time coordinates (added changetime=correctls)
• bug fix if not connected to internet
• tidying up examples

Location:

trunk/UTIL/PYTHON/planetoplot_v2

Files:

• examples/additional (added)
• examples/additional/add_text.txt_ (added)
• examples/additional/easy_get_field.py (moved) (moved from trunk/UTIL/PYTHON/planetoplot_v2/examples/easy_get_field.py)
• examples/additional/histo.py (moved) (moved from trunk/UTIL/PYTHON/planetoplot_v2/examples/histo.py)
• examples/additional/intercompare.py (moved) (moved from trunk/UTIL/PYTHON/planetoplot_v2/examples/intercompare.py)
• examples/additional/just_color_and_title.py (moved) (moved from trunk/UTIL/PYTHON/planetoplot_v2/examples/just_color_and_title.py) (1 diff)
• examples/additional/les_surf.py (added)
• examples/additional/listvar.py (moved) (moved from trunk/UTIL/PYTHON/planetoplot_v2/examples/listvar.py)
• examples/additional/madeleineplot.py (moved) (moved from trunk/UTIL/PYTHON/planetoplot_v2/examples/madeleineplot.py)
• examples/additional/maktext (added)
• examples/additional/minimal_field.py (moved) (moved from trunk/UTIL/PYTHON/planetoplot_v2/examples/minimal_field.py)
• examples/additional/minimal_test.py (added)
• examples/additional/myhisto.py (moved) (moved from trunk/UTIL/PYTHON/planetoplot_v2/examples/myhisto.py)
• examples/additional/tes.py (added)
• examples/additional/vertpress.py (moved) (moved from trunk/UTIL/PYTHON/planetoplot_v2/examples/vertpress.py) (1 diff)
• examples/additional/vertpress1d.py (moved) (moved from trunk/UTIL/PYTHON/planetoplot_v2/examples/vertpress1d.py)
• examples/aire.py (deleted)
• examples/anomaly.py (deleted)
• examples/commandline (added)
• examples/commandline/chain_pp.sh (moved) (moved from trunk/UTIL/PYTHON/planetoplot_v2/examples/chain_pp.sh)
• examples/commandline/pp.py (added)
• examples/function.py (deleted)
• examples/gw.py (deleted)
• examples/gw_v2.py (deleted)
• examples/hodograph.py (deleted)
• examples/les_psfc.py (deleted)
• examples/meso_profile.py (deleted)
• examples/minimal.py (deleted)
• examples/ref_all.sh (added)
• examples/reference (added)
• examples/reference/aire.py (added)
• examples/reference/anomaly.py (added)
• examples/reference/function.py (added)
• examples/reference/gw.py (added)
• examples/reference/gw_v2.py (added)
• examples/reference/hodograph.py (added)
• examples/reference/les_psfc.py (added)
• examples/reference/meso_profile.py (added)
• examples/reference/minimal.py (added)
• examples/reference/scatter.py (added)
• examples/reference/simple.py (added)
• examples/reference/tide.py (added)
• examples/reference/vector.py (added)
• examples/reference/windspeed.py (added)
• examples/reference/zonalcontour.py (added)
• examples/reference/zonalmean.py (added)
• examples/scatter.py (deleted)
• examples/simple.py (deleted)
• examples/tide.py (deleted)
• examples/vector.py (deleted)
• examples/windspeed.py (deleted)
• examples/zonalcontour.py (deleted)
• examples/zonalmean.py (deleted)
• pp.py (modified) (1 diff)
• ppclass.py (modified) (17 diffs)
• ppplot.py (modified) (7 diffs)
• set_back.txt (modified) (2 diffs)
• set_var.txt (modified) (1 diff)

trunk/UTIL/PYTHON/planetoplot_v2/examples/additional/just_color_and_title.py

@@ -3 +3 @@
 
 anewversion = pp()
-anewversion.defineplot(loadfile="../demo_data/gw.ppobj")
+anewversion.defineplot(loadfile="../../demo_data/gw.ppobj")
 for plot in anewversion.p:
     plot.colorb = "Paired"
-    plot.title = "This is a psychedelic plot"
+    plot.title = "Psychedelic plot!"
 anewversion.makeplot()

trunk/UTIL/PYTHON/planetoplot_v2/examples/additional/vertpress.py

@@ -4 +4 @@
 prof = pp()
 prof.verbose = True
-prof.file = "/home/aymeric/Big_Data/case_night.nc"
+#prof.file = "/home/aymeric/Big_Data/case_night.nc"
+prof.file = "/home/aymeric/Big_Data/GALE/wrfout_d03_2024-06-09_00:00:00_z"
 prof.var = "tk"
 prof.t = [10.,20.]
-prof.x = 30
-prof.y = 30
+prof.x = -5. #30
+prof.y = 138. #30
 prof.filename = "vertpress"
 

trunk/UTIL/PYTHON/planetoplot_v2/pp.py

@@ -54 +54 @@
 parser.add_option('-d','--directory',action='store',dest='folder',type="string",default="./",help="directory of output files")
 parser.add_option('-s','--changetime',action='store',dest='changetime',type="string",default=None,\
-                  help="transformation on time axis : [None] | mars_sol2ls | mars_dayini | mars_meso_ls | mars_meso_sol | mars_meso_utc | mars_meso_lt ")
+                  help="transformation on time axis : [None] | correctls | mars_sol2ls | mars_dayini | mars_meso_ls | mars_meso_sol | mars_meso_utc | mars_meso_lt ")
 parser.add_option('-p','--print',action='store_true',dest='savtxt',default=False,help="[1D] output field+coord in an ASCII file")
 parser.add_option('--stridex',action='store',dest='stridex',type="int",default=1,help="Load data every stridex grid points over x dimension")

trunk/UTIL/PYTHON/planetoplot_v2/ppclass.py

@@ -151 +151 @@
                       xlabel=None,ylabel=None,\
                       xcoeff=None,ycoeff=None,\
+                      nxticks=10,nyticks=10,\
                       proj=None,\
                       vmin=None,vmax=None,\
@@ -163 +164 @@
                       savtxt=False,\
                       modx=None,\
+                      xp=16,yp=8,\
+                      missing=1.e25,\
                       title=None):
         self.request = None
@@ -203 +206 @@
         self.savtxt = savtxt
         self.modx = modx
+        self.missing = missing
         ## here are user-defined plot settings 
         ## -- if not None, valid on all plots in the pp() objects
@@ -217 +221 @@
         self.units = units
         self.title = title
+        self.xp = xp ; self.yp = yp
+        self.nxticks = nxticks ; self.nyticks = nyticks
 
     # print status
@@ -225 +231 @@
         else:
             print "**** PPCLASS. Done step: " + self.status
+
+    # print attributes
+    def printme(self):
+        for k, v in vars(self).items():
+            print k,v
 
     #####################################################
@@ -268 +279 @@
             self.savtxt = other.savtxt
             self.modx = other.modx
+            self.xp = other.xp ; self.yp = other.yp
+            self.missing = other.missing
+            self.nxticks = other.nxticks ; self.nyticks = other.nyticks
         else:
             print "!! ERROR !! argument must be a pp object." ; exit()
@@ -589 +603 @@
               obj.getindexvert(dalist=sz,ind=z)
               obj.getindexhori(dalistx=sx,dalisty=sy,indx=x,indy=y)
+              # missing value
+              obj.missing = self.missing
         # change status
         self.status = "defined"
@@ -715 +731 @@
             self.status = "definedplot" ; self.plotin = None
             self.nplot = len(self.p) ; self.howmanyplots = self.nplot
-            return
+            ## [BUG FIX: apparently info about missing values is not saved correctly]
+            for count in range(self.nplot):
+              pl = self.p[count]
+              masked = np.ma.masked_where(np.abs(pl.field) > self.missing,pl.field)
+              pl.field = masked ; pl.field[pl.field.mask] = np.NaN
+            return #self?
         # -----------------------------------------------------
         # REGULAR MODE
@@ -802 +823 @@
                     if self.colorb is not None: plobj.colorb = self.colorb
                     if self.modx is not None: plobj.modx = self.modx
+                    if self.nxticks is not None: plobj.nxticks = self.nxticks
+                    if self.nyticks is not None: plobj.nyticks = self.nyticks
                     # -- 1D specific
                     if dp == 1:
@@ -896 +919 @@
         if self.plotin is None:  
             # start from scratch
-            self.fig = mpl.figure(figsize=(16,8))
+            self.fig = ppplot.figuref(x=self.xp,y=self.yp)
             self.subv,self.subh = ppplot.definesubplot(self.howmanyplots,self.fig) 
             self.n = 0
@@ -1220 +1243 @@
         self.changetime = None
         self.stridex = 1 ; self.stridey = 1 ; self.stridez = 1 ; self.stridet = 1
+        self.missing = '!! missing value: I am not set, damned !!'
 
     # open a file. for now it is netcdf. TBD for other formats.
@@ -1309 +1333 @@
                if self.verbose: print "**** OK. y axis %4.0f values [%5.1f,%5.1f]" % (self.dim_y,self.field_y.min(),self.field_y.max())
           # ALTITUDE. Try preset fields. If not present set grid points axis.
-          # WARNING: how do we do if several are available?
+          # WARNING: how do we do if several are available? the last one is chosen.
           self.name_z = "nothing"
           for c in glob_listz:
@@ -1329 +1353 @@
             else: 
                 self.field_z = self.f.variables[self.name_z][:] # specify dimension
+            # TBD: problems when self.dim_z != self.field_z.size
+            if self.field_z.size != self.dim_z:
+                if self.verbose: print "!! WARNING !! Cannot use this z coordinate. Not enough points. Use simple z axis."
+                self.field_z = np.array(range(self.dim_z))
+                self.name_z = "z grid points"
           if self.dim_z > 1: 
                if self.verbose: print "**** OK. z axis %4.0f values [%5.1f,%5.1f]" % (self.dim_z,self.field_z.min(),self.field_z.max())
+
           # TIME. Try preset fields.
           self.name_t = "nothing"
@@ -1351 +1381 @@
             else:
                 daint = tabtime[1] - dafirst
-                dalast = dafirst + (self.dim_t-1)*daint 
-                if dalast != tabtime[self.dim_t-1] and self.verbose:
-                    print "!! WARNING !! Time axis has been recast to be monotonic",dalast,tabtime[self.dim_t-1]
+                dalast = dafirst + (self.dim_t-1)*daint
                 self.field_t = np.linspace(dafirst,dalast,num=self.dim_t)
+                if self.verbose:
+                    print "!! WARNING !! WARNING !! Time axis is supposed to be equally spaced !!             
+                    if dalast != tabtime[self.dim_t-1]:
+                        print "!! WARNING !! Time axis has been recast to be monotonic",dalast,tabtime[self.dim_t-1]
           except:
             # ... or if a problem encountered, define a simple time axis
@@ -1369 +1401 @@
         if self.changetime is not None:
             if self.verbose: print "**** OK. Converting time axis:",self.changetime
-            ### option added by T. Navarro
+            ### options added by T. Navarro
             if self.changetime == "mars_sol2ls":
                 if "controle" in self.f.variables: 
@@ -1381 +1413 @@
                               + self.f.variables['controle'][26]
             ### options added by A. Spiga
+            elif self.changetime == "correctls":
+                # not regularly spaced + handle modulo 360. in files
+                dafirst = tabtime[0] + 0.
+                daint = tabtime[1] - dafirst
+                dalast = dafirst + (self.dim_t-1)*daint
+                year = 0.
+                add = np.linspace(dafirst,dalast,num=self.dim_t) ; add[0] = 0.
+                for iii in range(1,self.dim_t):
+                  if tabtime[iii] - tabtime[iii-1] < 0: year = year+1.
+                  add[iii] = year*360.
+                self.field_t = add + tabtime
             elif "mars_meso" in self.changetime:
                 if 'Times' not in self.f.variables.keys():
@@ -1671 +1714 @@
         # make a mask in case there are non-NaN missing values. (what about NaN missing values?)
         # ... this is important for computations below (see ppcompute)
-        masked = np.ma.masked_where(np.abs(self.field) > 1e25,self.field)
+        masked = np.ma.masked_where(np.abs(self.field) > self.missing,self.field)
         if masked.mask.any() == True:
-             if self.verbose: print "!! WARNING !! Values over +-1e25 are considered missing values."
+             if self.verbose: print "!! WARNING !! Values over %5.3e are considered missing values." % self.missing
              self.field = masked
              self.field.set_fill_value([np.NaN])

trunk/UTIL/PYTHON/planetoplot_v2/ppplot.py

@@ -187 +187 @@
 # a function for predefined figure sizes
 def figuref(x=16,y=9):
-    mpl.figure(figsize=(x,y))
+    fig = mpl.figure(figsize=(x,y))
+    return fig
 
 # a function to change color lines according to a color map (idea by A. Pottier)
@@ -208 +209 @@
         subv, subh = subv_t[numplot],subh_t[numplot]
     except: 
-        print "!! WARNING !! no preset found from set_multiplot.txt, or this setting file was not found."
+        print "!! WARNING !! (ignore if superpose mode) no preset found from set_multiplot.txt, or this setting file was not found."
         subv = 1 ; subh = numplot
     return subv,subh
@@ -355 +356 @@
                  xmax=None,\
                  ymax=None,\
+                 nxticks=10,\
+                 nyticks=10,\
                  title=""):
         ## what could be defined by the user
@@ -379 +382 @@
         self.xmax = xmax
         self.ymax = ymax
+        self.nxticks = nxticks
+        self.nyticks = nyticks
         ## other useful arguments
         ## ... not used here in ppplot but need to be attached to plot object
@@ -532 +537 @@
         ## set with .div the number of ticks. (is it better than automatic?)
         if not self.logx:
-            ax.xaxis.set_major_locator(MaxNLocator(self.div/2))
+            ax.xaxis.set_major_locator(MaxNLocator(self.div/2)) #TBD: with nxticks?
         else:
             print "!! WARNING. in logx mode, ticks are set automatically."
@@ -684 +689 @@
             if self.ymin is not None: ax.set_ybound(lower=self.ymin)
             if self.ymax is not None: ax.set_ybound(upper=self.ymax)
-            # set the number of ticks (hardcoded)
-            ax.xaxis.set_major_locator(MaxNLocator(10))
+            # set the number of ticks
+            if not self.logx:
+                ax.xaxis.set_major_locator(MaxNLocator(self.nxticks))
+            else:
+                print "!! WARNING. in logx mode, ticks are set automatically."
             ## specific modulo labels
             if self.modx is not None:
@@ -773 +781 @@
               if self.back in back.keys():
                  print "**** info: loading a background, please wait.",self.back
-                 if self.back not in ["coast","sea"]:   m.warpimage(back[self.back],scale=0.75)
-                 elif self.back == "coast":             m.drawcoastlines()
-                 elif self.back == "sea":               m.drawlsmask(land_color='white',ocean_color='aqua')
+                 if self.back not in ["coast","sea"]:
+                    try: m.warpimage(back[self.back],scale=0.75)
+                    except: print "!! ERROR !! no background image could be loaded. probably not connected to the internet?"
+                 elif self.back == "coast":
+                    m.drawcoastlines()
+                 elif self.back == "sea":
+                    m.drawlsmask(land_color='white',ocean_color='aqua')
               else:
                  print "!! ERROR !! requested background not defined. change name or fill in set_back.txt" ; exit()

trunk/UTIL/PYTHON/planetoplot_v2/set_back.txt

@@ -7 +7 @@
 tesalbedo ; http://www.lmd.jussieu.fr/~aslmd/maps/tesalbedo.jpg
 vis ; http://www.lmd.jussieu.fr/~aslmd/maps/mar0kuu2.jpg
-vishires ; http://www.lmd.jussieu.fr/~aslmd/maps/MarsMap_2500x1250.jpg
+vishires ; /home/aymeric/Dropbox/Public/MarsMap_2500x1250.jpg
 geolocal ; http://www.lmd.jussieu.fr/~aslmd/maps/geolocal.jpg
 mola ; http://www.lmd.jussieu.fr/~aslmd/maps/mars-mola-2k.jpg
-molabw ; http://www.lmd.jussieu.fr/~aslmd/maps/MarsElevation_2500x1250.jpg
+molabw ; /home/aymeric/Dropbox/Public/MarsElevation_2500x1250.jpg
 bw ; http://www.lmd.jussieu.fr/~aslmd/maps/EarthElevation_2500x1250.jpg
 contrast ; http://www.lmd.jussieu.fr/~aslmd/maps/EarthMapAtmos_2500x1250.jpg
@@ -24 +24 @@
 titan ; http://laps.noaa.gov/albers/sos/saturn/titan/titan_rgb_cyl_www.jpg
 titanuni ; http://maps.jpl.nasa.gov/pix/sat6fss1.jpg
+titanreg ; http://laps.noaa.gov/albers/sos/features/combined_titan_lon_zero_center.png
 venus ; http://laps.noaa.gov/albers/sos/venus/venus4/venus4_rgb_cyl_www.jpg
 cosmic ; http://laps.noaa.gov/albers/sos/universe/wmap/wmap_rgb_cyl_www.jpg

trunk/UTIL/PYTHON/planetoplot_v2/set_var.txt

@@ -4 +4 @@
 ##################################
 T ; %.0f ; gist_heat ; Temperature ; K
-TEMP ; %0.f ; gist_heat ; Temperature ; K 
+TEMP ; %.1f ; gist_heat ; Temperature ; K 
 TPOT ; %.0f ; Paired ; Potential temperature ; K  
 TSURF ; %.0f ; RdBu_r ; Surface temperature ; K