shapely 0.1

data/lib/shapefile.rb

@@ -0,0 +1,66 @@
+require "dbf"
+
+class Shapefile
+  HEADER_BYTES = 100
+  
+  attr_reader :shape_class
+  
+  def initialize(shp, shx = nil, dbf = nil)
+    # Open files.
+    @shp = File.open(shp)
+    @dbf = DBF::Table.new(dbf) if dbf
+     # Read and verify the header.
+    hdr = @shp.read(HEADER_BYTES).unpack("NNNNNNNVVEEEEEEEE")
+    unless hdr[0] == 9994
+      raise RuntimeError, "#{shp} isn't a shapefile"
+    end
+    unless hdr[7] == 1000
+      raise RuntimeError, "#{shp} isn't a supported version of shapefile"
+    end
+    unless @shape_class = Shape.class_for_code(hdr[8])
+      raise RuntimeError, "#{shp} is composed of an unsupported shape type"
+    end
+  end
+  
+  def self.open(shp, shx = nil, dbf = nil)
+    new(shp, shx, dbf)
+  end
+  
+  def shapes
+    result = []
+    each_shape {|shape| result << shape}
+    result
+  end
+  
+  def each_shape(&block)
+    @shp.seek(HEADER_BYTES, IO::SEEK_SET)
+    until @shp.eof?
+      # Read the next shape record.
+      nbr, len, shp_type = @shp.read(12).unpack("NNV")
+      record = @shp.read(len * 2 - 4)
+      next if shp_type == Shape::Null.code
+      shape = @shape_class.new(nbr, record)
+      # Load the shape's attributes. DBF's record method is ridiculously slow
+      # (it rereads every single row each time you call it), so we cheat a bit
+      # and use some of its private methods.
+      if @dbf
+        @dbf.send(:seek_to_record, nbr - 1)
+        unless @dbf.send(:deleted_record?)
+          shape.attributes = DBF::Record.new(@dbf).attributes
+        end
+      end
+      # Run the shape through the block.
+      block.call(shape)
+    end
+  end
+  
+  def attribute_names
+    if @dbf
+      @dbf.columns.collect {|c| c.name.downcase}
+    else
+      nil
+    end
+  end
+end
+
+require "shapefile/shape"

data/lib/shapefile/shape.rb

@@ -0,0 +1,38 @@
+class Shapefile::Shape
+  attr_reader :id
+  attr_accessor :attributes
+  
+  def initialize(id, raw_record) # :nodoc:
+    @id = id
+    self.attributes = {}
+  end
+  
+  def self.enumerate_subclass_codes # :nodoc:
+    @codes_classes = {}
+    constants.each do |const|
+      subclass = eval("Shapefile::Shape::#{const}") rescue next
+      @codes_classes[subclass.code] = subclass
+    end
+  end
+  
+  def self.code
+    nil
+  end
+  
+  def self.class_for_code(code)
+    @codes_classes[code]
+  end
+  
+  def [](key)
+    attributes[key]
+  end
+end
+
+require "shapefile/shape/multi_patch.rb"
+require "shapefile/shape/multi_point.rb"
+require "shapefile/shape/null.rb"
+require "shapefile/shape/point.rb"
+require "shapefile/shape/poly_line.rb"
+require "shapefile/shape/polygon.rb"
+
+Shapefile::Shape.enumerate_subclass_codes

data/lib/shapefile/shape/multi_patch.rb

@@ -0,0 +1,3 @@
+class Shapefile::Shape::MultiPatch < Shapefile::Shape
+  def self.code; 31 end
+end

data/lib/shapefile/shape/multi_point.rb

@@ -0,0 +1,11 @@
+class Shapefile::Shape::MultiPoint < Shapefile::Shape
+  def self.code; 8 end
+end
+
+class Shapefile::Shape::MultiPointM < Shapefile::Shape::MultiPoint
+  def self.code; 28 end
+end
+
+class Shapefile::Shape::MultiPointZ < Shapefile::Shape::MultiPointM
+  def self.code; 18 end
+end

data/lib/shapefile/shape/null.rb

@@ -0,0 +1,3 @@
+class Shapefile::Shape::Null < Shapefile::Shape
+  def self.code; 0 end
+end

data/lib/shapefile/shape/point.rb

@@ -0,0 +1,32 @@
+class Shapefile::Shape::Point < Shapefile::Shape
+  attr_reader :x, :y
+  
+  def self.code; 1 end
+  
+  def initialize(nbr, record)
+    super
+    @x, @y = record.unpack("EE")
+  end
+end
+
+class Shapefile::Shape::PointM < Shapefile::Shape::Point
+  attr_reader :m
+  
+  def self.code; 21 end
+  
+  def initialize(nbr, record)
+    super
+    @m = record[16, 8].unpack("E")
+  end
+end
+
+class Shapefile::Shape::PointZ < Shapefile::Shape::PointM
+  attr_reader :z
+  
+  def self.code; 11 end
+  
+  def initialize(nbr, record)
+    super
+    @z, @m = record[16, 16].unpack("EE")
+  end
+end

data/lib/shapefile/shape/poly_line.rb

@@ -0,0 +1,11 @@
+class Shapefile::Shape::PolyLine < Shapefile::Shape
+  def self.code; 3 end
+end
+
+class Shapefile::Shape::PolyLineM < Shapefile::Shape::PolyLine
+  def self.code; 23 end
+end
+
+class Shapefile::Shape::PolyLineZ < Shapefile::Shape::PolyLineM
+  def self.code; 13 end
+end

data/lib/shapefile/shape/polygon.rb

@@ -0,0 +1,11 @@
+class Shapefile::Shape::Polygon < Shapefile::Shape
+  def self.code; 5 end
+end
+
+class Shapefile::Shape::PolygonM < Shapefile::Shape::Polygon
+  def self.code; 25 end
+end
+
+class Shapefile::Shape::PolygonZ < Shapefile::Shape::PolygonM
+  def self.code; 15 end
+end

data/lib/shapely.rb

@@ -0,0 +1 @@
+require "shapefile"

metadata

@@ -0,0 +1,82 @@
+--- !ruby/object:Gem::Specification 
+name: shapely
+version: !ruby/object:Gem::Version 
+  prerelease: false
+  segments: 
+  - 0
+  - 1
+  version: "0.1"
+platform: ruby
+authors: 
+- Dana Contreras
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2010-03-27 00:00:00 -04:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: dbf
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        segments: 
+        - 1
+        - 0
+        - 0
+        version: 1.0.0
+  type: :runtime
+  version_requirements: *id001
+description: 
+email: 
+executables: []
+
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- lib/shapely.rb
+- lib/shapefile.rb
+- lib/shapefile/shape.rb
+- lib/shapefile/shape/multi_patch.rb
+- lib/shapefile/shape/multi_point.rb
+- lib/shapefile/shape/null.rb
+- lib/shapefile/shape/point.rb
+- lib/shapefile/shape/poly_line.rb
+- lib/shapefile/shape/polygon.rb
+has_rdoc: true
+homepage: http://github.com/DanaDanger/shapely
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.6
+signing_key: 
+specification_version: 3
+summary: A gem for reading ESRI Shapefiles.
+test_files: []
+