gdal-helper 0.0.1.1 → 0.0.1.2

data/examples/basic_example_using_each_line.rb

@@ -0,0 +1,31 @@
+#!/usr/bin/env ruby
+require "rubygems"
+require "pp"
+require "gdal_helper"
+
+#basic plan - open image,dim image by 25%, set the projection and geo_trans, then quit, job done.
+# Open a tiff to write to, with default create options (TILED=YES, COMPRESS=LZW) to write to..
+if (ARGV.length != 2)
+  puts("Usage: ./basic_example_using_each_line.rb (infile) (outfile)")
+  return -1
+end
+
+infile = GdalFile.new(ARGV[0])
+outfile = GdalFile.new(ARGV[1], "w", infile.xsize,infile.ysize,infile.number_of_bands,"GTiff", infile.data_type, ["COMPRESS=DEFLATE", "TILED=YES"])
+y = 0
+infile.each_line do |data|
+  data.each do |band|
+    band.each_index do |x_index|
+      band[x_index] = band[x_index] - band[x_index]/4
+    end
+  end
+  outfile.write_bands(0,y,infile.xsize,1,data)
+  y += 1
+end
+# Set the projection
+outfile.set_projection(infile.get_projection)
+# set the geo transform (world file)
+outfile.set_geo_transform(infile.get_geo_transform)
+
+
+

data/examples/basic_example_using_each_line_with_index.rb

@@ -0,0 +1,30 @@
+#!/usr/bin/env ruby
+require "rubygems"
+require "pp"
+require "gdal_helper"
+
+#basic plan - open image,dim image by 25%, set the projection and geo_trans, then quit, job done.
+# Open a tiff to write to, with default create options (TILED=YES, COMPRESS=LZW) to write to..
+if (ARGV.length != 2)
+  puts("Usage: ./basic_example_using_each_line_with_index.rb (infile) (outfile)")
+  return -1
+end
+
+infile = GdalFile.new(ARGV[0])
+outfile = GdalFile.new(ARGV[1], "w", infile.xsize,infile.ysize,infile.number_of_bands,"GTiff", infile.data_type, ["COMPRESS=DEFLATE", "TILED=YES"])
+y = 0
+infile.each_line_with_index do |y_index,data|
+  data.each do |band|
+    band.each_index do |x_index|
+      band[x_index] = band[x_index] - band[x_index]/4
+    end
+  end
+  outfile.write_bands(0,y_index,infile.xsize,1,data)
+end
+# Set the projection
+outfile.set_projection(infile.get_projection)
+# set the geo transform (world file)
+outfile.set_geo_transform(infile.get_geo_transform)
+
+
+

data/examples/ndvi_color.rb

@@ -0,0 +1,108 @@
+#!/usr/bin/env ruby
+require "rubygems"
+require "gdal_helper"
+require "pp"
+
+##
+# This is a very basic example using the helper library to do some real work - in this case taking a MODIS ndvi product
+# that is scaled from 10,000 to -1000 and creating a nice color presentation of it.
+# This is just an example, so YMMV.  Have fun - jc@alaska.edu
+
+##
+# Finds the correct item in a colormap for a particular value
+def find_color_range ( value, colormap )
+        ##
+        # Keep last, incase it is needed again - speed up 
+        if (@last && @last[0] <= value && value <= @last[1])
+                return @last
+        end
+        
+        colormap.each do |color_set|
+                if ( color_set[0] <= value && value <= color_set[1] )
+                        @last = color_set
+                        return color_set
+                end
+        end
+	
+	# outside the range of the color map, this is an error..
+	raise(ArgumentError, "value requested is not in colormap..", caller)
+end
+
+
+##
+# Maps v to a range, given a min and max source and desitancation values.. wow, I can't spell..
+def map_color(color_set, value)
+	#give the color table some more reasonable names..
+	# [min, max, [min r,g,b], [max r,g,b]]
+	min = color_set[0]
+	max = color_set[1]
+	min_c = color_set[2]
+	max_c = color_set[3]
+	rgb=[] #requested color
+	perc = (value.to_f - min.to_f)/(max - min) #percent between
+	0.upto(2) { |zang| rgb[zang] = ((max_c[zang] - min_c[zang]) * perc + min_c[zang] ).to_i }
+	rgb
+end
+
+##
+# Maps a value to a color set
+def colorize(value, colormap)
+        @cache={} if (!@cache)
+        return @cache[value] if (@cache[value])
+        color_set = find_color_range(value, colormap)
+
+        rgb = map_color(color_set, value)
+        @cache[value]=rgb if (@cache.length < 10000) #cache first x items - deleting from the hash is slow
+	rgb
+end
+        
+##
+# A color map for ndvi, adapted from GRASS. the format is [min value, max value, [starting r,g,b], [ending r,g,b]]
+# this could of course be improved..
+ndvi_colormap = [
+        [-1.0000, -0.3000, [255,255,255], [0,0,255]],
+        [-0.3000, -0.2000, [0,0,255],  [205,193, 173]],
+        [-0.2000, 0.0000, [205,193, 173], [150,150,150]],
+        [0.0000, 0.1000,[150,150,150], [120,100,51]],
+        [0.1000, 0.3000,[120,100,51], [120, 200, 100]],
+        [0.3000, 0.4000,[120,200,100],[28, 144, 3]],
+        [0.4000, 0.6000,[28, 144, 3],[ 6, 55,  0]],
+        [0.6000, 0.8000,[ 6, 55,  0],[10, 30, 25]],
+        [0.8000, 1.0000,[10, 30, 25],[ 6, 27, 7]]
+        ]
+
+if (ARGV.length != 2)
+  puts("Usage: ./ndvi_color.rb (infile) (outfile)")
+  return -1
+end
+
+#input file
+infile = GdalFile.new(ARGV[0])
+#output file
+outfile = GdalFile.new(ARGV[1], "w", infile.xsize,infile.ysize,3,"GTiff", String, ["COMPRESS=DEFLATE", "TILED=YES"])
+#set the projection related details on the output file
+outfile.set_projection(infile.get_projection)
+outfile.set_geo_transform(infile.get_geo_transform)
+
+# Loop though each line, colorizing the data
+infile.each_line_with_index do |y_index, data|
+	out_data = [[],[],[]] #rbg array..
+	
+	# print some status details..
+	if (y_index%100 == 0)
+		STDOUT.write(".")
+		STDOUT.flush
+	end
+	
+	# loop though each sample, colorzing them.
+	data[0].each_index do |x_index|
+		# take each value, map it to a rgb, then put this in the bands to be output
+		rgb  = colorize(data[0][x_index].to_f*0.0001, ndvi_colormap)
+		out_data[0][x_index] = rgb[0]
+		out_data[1][x_index] = rgb[1]
+		out_data[2][x_index] = rgb[2]
+	end
+	outfile.write_bands(0,y_index, infile.xsize, 1, out_data)
+end
+
+STDOUT.write("Done!\n")

data/lib/gdal_helper.rb

@@ -302,16 +302,16 @@ end
 class GdalFile < GdalStuff
   def initialize ( name, mode="r", xsize=nil, ysize=nil,bands=3, driver="GTiff", data_type=String, options=['TILED=YES','COMPRESS=DEFLATE'] )
     if ( mode == "r" )
-      @Gdalfile = Gdal::Gdal.open(name)
+      @gdalfile = Gdal::Gdal.open(name)
     else
       if ( mode == "w")
         if (File.exists?(name))
-          @Gdalfile = Gdal::Gdal.open(name,Gdal::Gdalconst::GA_UPDATE )
+          @gdalfile = Gdal::Gdal.open(name,Gdal::Gdalconst::GA_UPDATE )
         else
           driver = Gdal::Gdal.get_driver_by_name(driver)
           #puts(driver.class)
           #puts("Creating create(#{name}, #{xsize}, #{ysize}, #{bands}, #{data_type_to_gdal(data_type).to_s})")
-          @Gdalfile = driver.create(name, xsize, ysize, bands, data_type_to_gdal(data_type), options)
+          @gdalfile = driver.create(name, xsize, ysize, bands, data_type_to_gdal(data_type), options)
         end
       else
         raise ArgumentError, "mode of \"#{mode}\" is not useful (not r|w) not sure what to do here folks", caller
@@ -320,7 +320,7 @@ class GdalFile < GdalStuff
     
     @bands=[]
     #1 is not a mistake - the raster bands start at 1 no 0. just a fyi.
-    1.upto(@Gdalfile.RasterCount).each {|x| @bands << GdalBand.new(@Gdalfile.get_raster_band(x))}
+    1.upto(@gdalfile.RasterCount).each {|x| @bands << GdalBand.new(@gdalfile.get_raster_band(x))}
   end
   
   ###
@@ -351,20 +351,20 @@ class GdalFile < GdalStuff
   
   #returns basic size info as a hash
   def size()
-    { "x"=> @Gdalfile.RasterXSize,
-      "y" => @Gdalfile.RasterYSize,
+    { "x"=> @gdalfile.RasterXSize,
+      "y" => @gdalfile.RasterYSize,
       "bands" => @bands.length,
       "data_type" => @bands[0].data_type()}
   end
   
   #x dimention size
   def xsize()
-    @Gdalfile.RasterXSize
+    @gdalfile.RasterXSize
   end
   
   #y dim size
   def ysize()
-    @Gdalfile.RasterYSize
+    @gdalfile.RasterYSize
   end
   
   #number of bands
@@ -384,29 +384,39 @@ class GdalFile < GdalStuff
   
   # gets the projection
   def get_projection
-    @Gdalfile.get_projection
+    @gdalfile.get_projection
   end
   
   #sets the projection
   def set_projection(proj_str)
-    @Gdalfile.set_projection(proj_str)
+    @gdalfile.set_projection(proj_str)
   end
   
   #looks up the projection in the epsg database, give it a number like 102006.
   def set_projection_epsg(epsg)
     srs =  Gdal::Osr::SpatialReference.new()
     srs.import_from_epsg(epsg)
-    @Gdalfile.set_projection(srs.export_to_wkt)
+    @gdalfile.set_projection(srs.export_to_wkt)
   end
   
   #sets the geo_transform, the wld file generally.
   def set_geo_transform(srs) 
-    @Gdalfile.set_geo_transform(srs)
+    @gdalfile.set_geo_transform(srs)
   end
   
   #gets the geo transform (wld file traditionally)
   def get_geo_transform()
-     @Gdalfile.get_geo_transform
+     @gdalfile.get_geo_transform
+  end
+  
+  #iterator over each line..
+  def each_line( )
+    0.upto(ysize-1){|y| yield(read_bands(0,y,xsize,1))}
+  end
+  
+  #iterator over each line, with index
+  def each_line_with_index( )
+    0.upto(ysize-1){|y| yield(y,read_bands(0,y,xsize,1))}
   end
   
 end

data/lib/version.rb

@@ -1,4 +1,4 @@
-module Gdal_Helper
-  Version = "0.0.1.1"
+module GdalHelper
+  Version = "0.0.1.2"
 end
 

metadata

@@ -6,8 +6,8 @@ version: !ruby/object:Gem::Version
   - 0
   - 0
   - 1
-  - 1
-  version: 0.0.1.1
+  - 2
+  version: 0.0.1.2
 platform: ruby
 authors: 
 - JC
@@ -15,7 +15,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2010-04-20 00:00:00 -08:00
+date: 2010-04-21 00:00:00 -08:00
 default_executable: 
 dependencies: []
 
@@ -31,8 +31,11 @@ files:
 - lib/version.rb
 - lib/gdal_helper.rb
 - test/ruby_gem_test.rb
+- examples/basic_example_using_each_line_with_index.rb
 - examples/basic_example.rb
 - examples/print_info_example.rb
+- examples/ndvi_color.rb
+- examples/basic_example_using_each_line.rb
 - examples/gdal_helper_test.rb
 has_rdoc: true
 homepage: http://www.gina.alaska.edu