Changeset 990 for trunk/UTIL/PYTHON/planetoplot_v2/ppclass.py

Timestamp:

Jun 14, 2013, 9:52:31 AM (12 years ago)

Author:

aslmd

Message:

UTIL PYTHON planetoplot_v2. ASCII output. 4D altitude coordinate is possible.

File:

• trunk/UTIL/PYTHON/planetoplot_v2/ppclass.py (modified) (5 diffs)

trunk/UTIL/PYTHON/planetoplot_v2/ppclass.py

@@ -161 +161 @@
                       changetime=None,\
                       units=None,\
+                      savtxt=False,\
                       title=None):
         self.request = None
@@ -199 +200 @@
         self.includedate = includedate
         self.changetime = changetime
+        self.savtxt = savtxt
         ## here are user-defined plot settings 
         ## -- if not None, valid on all plots in the pp() objects
@@ -262 +264 @@
             self.includedate = other.includedate
             self.changetime = other.changetime
+            self.savtxt = other.savtxt
         else:
             print "!! ERROR !! argument must be a pp object." ; exit()
@@ -943 +946 @@
             # finally make the plot
             pl.make()
+            # possibly print results in a text file
+            if self.savtxt:
+                if self.verbose: print "**** Printing results in a text file"
+                name = pl.var + "%04d" % self.n
+                ppplot.writeascii(field=pl.field,absc=pl.absc,name=name) 
             # increment plot count (and propagate this in plotin)
             self.n = self.n+1
@@ -1299 +1307 @@
             self.name_z = "z grid points"
           else:
-            self.field_z = self.f.variables[self.name_z][:] # specify dimension
-                                                            # TBD: have to check that this is not a 3D field
+            tabalt = self.f.variables[self.name_z]
+            # (consider the case where tabtime is not dim 1) TBD: 2D and 3D cases
+            if tabalt.ndim == 4: 
+                try:
+                    self.field_z = tabalt[1,:,0,0] # 4D case. alt is usually third dimension.
+                                                   # 1 for time to avoid initial 0s
+                except:
+                    self.field_z = tabalt[0,:,0,0]
+                if self.verbose: print "!! WARNING !! "+self.name_z+" is 4D var. We made it 1D."
+            else: 
+                self.field_z = self.f.variables[self.name_z][:] # specify dimension
           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())