Changeset 5569

Timestamp:

04/21/10 10:13:25 (2 years ago)

Author:

jorgearevalo

Message:

Small bug fixed: numpy.zeros uses (height x width) syntax for input parameters,

not (width x height)

Files:

• spike/wktraster/scripts/rtrowdump.py (modified) (3 diffs)

spike/wktraster/scripts/rtrowdump.py

@@ -103 +103 @@
 
     logit("Connected to %s" % opts.db)
-    logit("Source WKT rasteri:")
+    logit("Source WKT raster:")
     logit("\trow=%s" % opts.where)
     logit("\twidth=%d, height=%d, bands=%d, pixel types=%s" \
@@ -109 +109 @@
     logit("Target GeoTIFF: %s" % opts.output)
 
-    raster_size = ( rt.width, rt.height )
-
     out_format = "GTiff"
     out_driver = gdal.GetDriverByName(out_format)
     out_data_type = pt2gdt(rt.pixel_types[0])
-    out_ds = out_driver.Create(opts.output, raster_size[0], raster_size[1], rt.num_bands, out_data_type)
+    out_ds = out_driver.Create(opts.output, rt.width, rt.height, rt.num_bands, out_data_type)
     
 
@@ -120 +118 @@
         logit("--- BAND %d ---------------------------------" % b)
 
-        raster = numpy.zeros(raster_size, pt2numpy(out_data_type))
-        for y in range(0, raster_size[1]):
-            for x in range(0, raster_size[0]):
-                pixel = rt.get_value(b, x + 1, y + 1)
-                raster[y, x] = pixel
+        ### Be careful!!
+        ### Zeros function's input parameter can be a (height x width) array,
+        ### not (width x height): http://docs.scipy.org/doc/numpy/reference/generated/numpy.zeros.html?highlight=zeros#numpy.zeros
+        raster = numpy.zeros((rt.height, rt.width), pt2numpy(out_data_type))
+        for width_index in range(0, rt.width):
+            for height_index in range(0, rt.height):
+                pixel = rt.get_value(b, width_index + 1, height_index + 1)
+                raster[height_index, width_index] = pixel
 
         logit(str(raster))