Changeset 4137 for spike/wktraster/scripts/rtreader.py

Timestamp:

06/05/09 09:10:47 (3 years ago)

Author:

mloskot

Message:

[wktraster] Added query_pixel_types() function to rtreader.RasterReader? class.

Files:

• spike/wktraster/scripts/rtreader.py (modified) (6 diffs)

spike/wktraster/scripts/rtreader.py

@@ -54 +54 @@
         self._where = where
         self._sizes = None
+        self._types = None
         # Connect and read RASTER header
         self._setup()
@@ -65 +66 @@
     height = property(fget = lambda self: self._get_height())
     num_bands = property(fget = lambda self: self._get_num_bands())
+    pixel_types = property(fget = lambda self: self._get_pixel_types())
 
     # Public methods 
@@ -92 +94 @@
         return self._query_raster_size(2)
 
+    def _get_pixel_types(self):
+        return self._query_pixel_types()
+
     def _setup(self):
         self._connect()
@@ -116 +121 @@
 
     def _query_value(self, band, x, y):
-        sql = """SELECT st_value(%s, %d, %d, %d) FROM %s""" % \
+        sql = 'SELECT st_value(%s, %d, %d, %d) FROM %s' % \
                  (self._column, band, x, y, self._table)
         if len(self._where) > 0:
-            sql += """ WHERE %s""" % self._where
+            sql += ' WHERE %s' % self._where
         row = self._query_single_row(sql)
         if row is None:
@@ -127 +132 @@
     def _query_raster_size(self, dim, force = False):
         if self._sizes is None or force is True:
-            sql = """SELECT st_width(%s), st_height(%s), st_numbands(%s) FROM %s""" % \
+            sql = 'SELECT st_width(%s), st_height(%s), st_numbands(%s) FROM %s' % \
                      (self._column, self._column, self._column, self._table)
             if len(self._where) > 0:
-                sql += """ WHERE %s""" % self._where
+                sql += ' WHERE %s' % self._where
             self._sizes = self._query_single_row(sql)
 
@@ -137 +142 @@
         return self._sizes[dim]
 
+    def _query_pixel_types(self):
+
+        types = []
+        sql = 'SELECT '
+        for i in range(0, self.num_bands):
+            if i != 0:
+                sql += ','
+            nband = i + 1
+            sql += ' st_bandpixeltype(%s, %d) ' % (self._column, nband)
+        sql += ' FROM ' + self._table
+        
+        return self._query_single_row(sql)