rgeo 0.3.3 → 0.3.4

data/lib/rgeo/geos/ffi_classes.rb

@@ -1,15 +1,15 @@
 # -----------------------------------------------------------------------------
-# 
+#
 # FFI-GEOS geometry implementation
-# 
+#
 # -----------------------------------------------------------------------------
-# Copyright 2010 Daniel Azuma
-# 
+# Copyright 2010-2012 Daniel Azuma
+#
 # All rights reserved.
-# 
+#
 # Redistribution and use in source and binary forms, with or without
 # modification, are permitted provided that the following conditions are met:
-# 
+#
 # * Redistributions of source code must retain the above copyright notice,
 #   this list of conditions and the following disclaimer.
 # * Redistributions in binary form must reproduce the above copyright notice,
@@ -18,7 +18,7 @@
 # * Neither the name of the copyright holder, nor the names of any other
 #   contributors to this software, may be used to endorse or promote products
 #   derived from this software without specific prior written permission.
-# 
+#
 # THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
 # AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 # IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
@@ -35,15 +35,15 @@
 
 
 module RGeo
-  
+
   module Geos
-    
-    
+
+
     module FFIUtils  # :nodoc:
-      
+
       class << self
-        
-        
+
+
         def coord_seqs_equal?(cs1_, cs2_, check_z_)
           len1_ = cs1_.length
           len2_ = cs2_.length
@@ -58,8 +58,8 @@ module RGeo
             false
           end
         end
-        
-        
+
+
         def compute_dimension(geom_)
           result_ = -1
           case geom_.type_id
@@ -83,62 +83,102 @@ module RGeo
           end
           result_
         end
-        
-        
+
+
+        def _init
+          @supports_prepared_level_1 = ::Geos::FFIGeos.respond_to?(:GEOSPreparedContains_r)
+          @supports_prepared_level_2 = ::Geos::FFIGeos.respond_to?(:GEOSPreparedDisjoint_r)
+        end
+
+        attr_reader :supports_prepared_level_1
+        attr_reader :supports_prepared_level_2
+
+
       end
-      
+
+
     end
-    
-    
+
+
     class FFIGeometryImpl  # :nodoc:
-      
+
       include Feature::Instance
-      
+
       Feature::MixinCollection::GLOBAL.for_type(Feature::Geometry).include_in_class(self)
-      
-      
+
+
       def initialize(factory_, fg_geom_, klasses_)
         @factory = factory_
         @fg_geom = fg_geom_
+        @_fg_prep = factory_._auto_prepare ? 1 : 0
         @_klasses = klasses_
         fg_geom_.srid = factory_.srid
       end
-      
-      
+
+
       def inspect
         "#<#{self.class}:0x#{object_id.to_s(16)} #{as_text.inspect}>"
       end
-      
-      
+
+
       attr_reader :factory
       attr_reader :fg_geom
-      
+
       attr_reader :_klasses  # :nodoc:
-      
-      
+
+
       def initialize_copy(orig_)
         @factory = orig_.factory
         @fg_geom = orig_.fg_geom.clone
         @fg_geom.srid = orig_.fg_geom.srid
+        @_fg_prep = @factory._auto_prepare ? 1 : 0
         @_klasses = orig_._klasses
       end
-      
-      
+
+
       def srid
         @fg_geom.srid
       end
-      
-      
+
+
       def dimension
         FFIUtils.compute_dimension(@fg_geom)
       end
-      
-      
+
+
       def geometry_type
         Feature::Geometry
       end
-      
-      
+
+
+      def prepared?
+        !@_fg_prep.is_a?(::Integer)
+      end
+
+
+      def prepare!
+        if @_fg_prep.is_a?(::Integer)
+          @_fg_prep = ::Geos::PreparedGeometry.new(@fg_geom)
+        end
+        self
+      end
+
+
+      def _request_prepared  # :nodoc:
+        case @_fg_prep
+        when 0
+          nil
+        when 1
+          @_fg_prep = 2
+          nil
+        when 2
+          @_fg_prep = ::Geos::PreparedGeometry.new(@fg_geom)
+        else
+          @_fg_prep
+        end
+      end
+
+
       def envelope
         fg_geom_ = @fg_geom.envelope
         # GEOS returns an "empty" point for an empty collection's envelope.
@@ -148,8 +188,8 @@ module RGeo
         end
         @factory.wrap_fg_geom(fg_geom_)
       end
-      
-      
+
+
       def boundary
         if self.class == FFIGeometryCollectionImpl
           nil
@@ -157,29 +197,29 @@ module RGeo
           @factory.wrap_fg_geom(@fg_geom.boundary)
         end
       end
-      
-      
+
+
       def as_text
         @factory._generate_wkt(self)
       end
       alias_method :to_s, :as_text
-      
-      
+
+
       def as_binary
         @factory._generate_wkb(self)
       end
-      
-      
+
+
       def is_empty?
         @fg_geom.empty?
       end
-      
-      
+
+
       def is_simple?
         @fg_geom.simple?
       end
-      
-      
+
+
       def equals?(rhs_)
         return false unless rhs_.kind_of?(::RGeo::Feature::Instance)
         fg_ = factory._convert_to_fg_geometry(rhs_)
@@ -194,121 +234,156 @@ module RGeo
         end
       end
       alias_method :==, :equals?
-      
-      
+
+
       def disjoint?(rhs_)
         fg_ = factory._convert_to_fg_geometry(rhs_)
-        fg_ ? @fg_geom.disjoint?(fg_) : false
+        if fg_
+          prep_ = _request_prepared if FFIUtils.supports_prepared_level_2
+          prep_ ? prep_.disjoint?(fg_) : @fg_geom.disjoint?(fg_)
+        else
+          false
+        end
       end
-      
-      
+
+
       def intersects?(rhs_)
         fg_ = factory._convert_to_fg_geometry(rhs_)
-        fg_ ? @fg_geom.intersects?(fg_) : false
+        if fg_
+          prep_ = _request_prepared if FFIUtils.supports_prepared_level_1
+          prep_ ? prep_.intersects?(fg_) : @fg_geom.intersects?(fg_)
+        else
+          false
+        end
       end
-      
-      
+
+
       def touches?(rhs_)
         fg_ = factory._convert_to_fg_geometry(rhs_)
-        fg_ ? @fg_geom.touches?(fg_) : false
+        if fg_
+          prep_ = _request_prepared if FFIUtils.supports_prepared_level_2
+          prep_ ? prep_.touches?(fg_) : @fg_geom.touches?(fg_)
+        else
+          false
+        end
       end
-      
-      
+
+
       def crosses?(rhs_)
         fg_ = factory._convert_to_fg_geometry(rhs_)
-        fg_ ? @fg_geom.crosses?(fg_) : false
+        if fg_
+          prep_ = _request_prepared if FFIUtils.supports_prepared_level_2
+          prep_ ? prep_.crosses?(fg_) : @fg_geom.crosses?(fg_)
+        else
+          false
+        end
       end
-      
-      
+
+
       def within?(rhs_)
         fg_ = factory._convert_to_fg_geometry(rhs_)
-        fg_ ? @fg_geom.within?(fg_) : false
+        if fg_
+          prep_ = _request_prepared if FFIUtils.supports_prepared_level_2
+          prep_ ? prep_.within?(fg_) : @fg_geom.within?(fg_)
+        else
+          false
+        end
       end
-      
-      
+
+
       def contains?(rhs_)
         fg_ = factory._convert_to_fg_geometry(rhs_)
-        fg_ ? @fg_geom.contains?(fg_) : false
+        if fg_
+          prep_ = _request_prepared if FFIUtils.supports_prepared_level_1
+          prep_ ? prep_.contains?(fg_) : @fg_geom.contains?(fg_)
+        else
+          false
+        end
       end
-      
-      
+
+
       def overlaps?(rhs_)
         fg_ = factory._convert_to_fg_geometry(rhs_)
-        fg_ ? @fg_geom.overlaps?(fg_) : false
+        if fg_
+          prep_ = _request_prepared if FFIUtils.supports_prepared_level_2
+          prep_ ? prep_.overlaps?(fg_) : @fg_geom.overlaps?(fg_)
+        else
+          false
+        end
       end
-      
-      
+
+
       def relate(rhs_, pattern_)
         fg_ = factory._convert_to_fg_geometry(rhs_)
         fg_ ? @fg_geom.relate_pattern(fg_, pattern_) : nil
       end
-      
-      
+
+
       def distance(rhs_)
         fg_ = factory._convert_to_fg_geometry(rhs_)
         fg_ ? @fg_geom.distance(fg_) : nil
       end
-      
-      
+
+
       def buffer(distance_)
         @factory.wrap_fg_geom(@fg_geom.buffer(distance_, @factory.buffer_resolution))
       end
-      
-      
+
+
       def convex_hull
         @factory.wrap_fg_geom(@fg_geom.convex_hull)
       end
-      
-      
+
+
       def intersection(rhs_)
         fg_ = factory._convert_to_fg_geometry(rhs_)
         fg_ ? @factory.wrap_fg_geom(@fg_geom.intersection(fg_)) : nil
       end
-      
-      
+
+
       def union(rhs_)
         fg_ = factory._convert_to_fg_geometry(rhs_)
         fg_ ? @factory.wrap_fg_geom(@fg_geom.union(fg_)) : nil
       end
-      
-      
+
+
       def difference(rhs_)
         fg_ = factory._convert_to_fg_geometry(rhs_)
         fg_ ? @factory.wrap_fg_geom(@fg_geom.difference(fg_)) : nil
       end
-      
-      
+
+
       def sym_difference(rhs_)
         fg_ = factory._convert_to_fg_geometry(rhs_)
         fg_ ? @factory.wrap_fg_geom(@fg_geom.sym_difference(fg_)) : nil
       end
-      
-      
+
+
       def _detach_fg_geom  # :nodoc:
         fg_ = @fg_geom
         @fg_geom = nil
         fg_
       end
-      
-      
+
+
     end
-    
-    
+
+
     class FFIPointImpl < FFIGeometryImpl  # :nodoc:
-      
+
       Feature::MixinCollection::GLOBAL.for_type(Feature::Point).include_in_class(self)
-      
-      
+
+
       def x
         @fg_geom.coord_seq.get_x(0)
       end
-      
-      
+
+
       def y
         @fg_geom.coord_seq.get_y(0)
       end
-      
-      
+
+
       def z
         if @factory.property(:has_z_coordinate)
           @fg_geom.coord_seq.get_z(0)
@@ -316,8 +391,8 @@ module RGeo
           nil
         end
       end
-      
-      
+
+
       def m
         if @factory.property(:has_m_coordinate)
           @fg_geom.coord_seq.get_z(0)
@@ -325,44 +400,44 @@ module RGeo
           nil
         end
       end
-      
-      
+
+
       def geometry_type
         Feature::Point
       end
-      
-      
+
+
       def eql?(rhs_)
         rhs_.class == self.class && rhs_.factory.eql?(@factory) &&
           FFIUtils.coord_seqs_equal?(rhs_.fg_geom.coord_seq, @fg_geom.coord_seq, @factory._has_3d)
       end
-      
-      
+
+
     end
-    
-    
+
+
     class FFILineStringImpl < FFIGeometryImpl  # :nodoc:
-      
-      
+
+
       Feature::MixinCollection::GLOBAL.for_type(Feature::Curve).include_in_class(self)
       Feature::MixinCollection::GLOBAL.for_type(Feature::LineString).include_in_class(self)
-      
-      
+
+
       def geometry_type
         Feature::LineString
       end
-      
-      
+
+
       def length
         @fg_geom.length
       end
-      
-      
+
+
       def num_points
         @fg_geom.num_points
       end
-      
-      
+
+
       def point_n(n_)
         if n_ >= 0 && n_ < @fg_geom.num_points
           coord_seq_ = @fg_geom.coord_seq
@@ -374,18 +449,18 @@ module RGeo
           nil
         end
       end
-      
-      
+
+
       def start_point
         point_n(0)
       end
-      
-      
+
+
       def end_point
         point_n(@fg_geom.num_points - 1)
       end
-      
-      
+
+
       def points
         coord_seq_ = @fg_geom.coord_seq
         has_3d_ = @factory._has_3d
@@ -396,92 +471,92 @@ module RGeo
           @factory.point(x_, y_, *extra_)
         end
       end
-      
-      
+
+
       def is_closed?
         @fg_geom.closed?
       end
-      
-      
+
+
       def is_ring?
         @fg_geom.ring?
       end
-      
-      
+
+
       def eql?(rhs_)
         rhs_.class == self.class && rhs_.factory.eql?(@factory) &&
           FFIUtils.coord_seqs_equal?(rhs_.fg_geom.coord_seq, @fg_geom.coord_seq, @factory._has_3d)
       end
-      
-      
+
+
     end
-    
-    
+
+
     class FFILinearRingImpl < FFILineStringImpl  # :nodoc:
-      
-      
+
+
       Feature::MixinCollection::GLOBAL.for_type(Feature::LinearRing).include_in_class(self)
-      
-      
+
+
       def geometry_type
         Feature::LinearRing
       end
-      
-      
+
+
     end
-    
-    
+
+
     class FFILineImpl < FFILineStringImpl  # :nodoc:
-      
-      
+
+
       Feature::MixinCollection::GLOBAL.for_type(Feature::Line).include_in_class(self)
-      
-      
+
+
       def geometry_type
         Feature::Line
       end
-      
-      
+
+
     end
-    
-    
+
+
     class FFIPolygonImpl < FFIGeometryImpl  # :nodoc:
-      
-      
+
+
       Feature::MixinCollection::GLOBAL.for_type(Feature::Surface).include_in_class(self)
       Feature::MixinCollection::GLOBAL.for_type(Feature::Polygon).include_in_class(self)
-      
-      
+
+
       def geometry_type
         Feature::Polygon
       end
-      
-      
+
+
       def area
         @fg_geom.area
       end
-      
-      
+
+
       def centroid
         @factory.wrap_fg_geom(@fg_geom.centroid, FFIPointImpl)
       end
-      
-      
+
+
       def point_on_surface
         @factory.wrap_fg_geom(@fg_geom.point_on_surface, FFIPointImpl)
       end
-      
-      
+
+
       def exterior_ring
         @factory.wrap_fg_geom(@fg_geom.exterior_ring, FFILinearRingImpl)
       end
-      
-      
+
+
       def num_interior_rings
         @fg_geom.num_interior_rings
       end
-      
-      
+
+
       def interior_ring_n(n_)
         if n_ >= 0 && n_ < @fg_geom.num_interior_rings
           @factory.wrap_fg_geom(@fg_geom.interior_ring_n(n_), FFILinearRingImpl)
@@ -489,15 +564,15 @@ module RGeo
           nil
         end
       end
-      
-      
+
+
       def interior_rings
         ::Array.new(@fg_geom.num_interior_rings) do |n_|
           @factory.wrap_fg_geom(@fg_geom.interior_ring_n(n_), FFILinearRingImpl)
         end
       end
-      
-      
+
+
       def eql?(rhs_)
         if rhs_.class == self.class && rhs_.factory.eql?(@factory) &&
           rhs_.exterior_ring.eql?(self.exterior_ring)
@@ -513,22 +588,22 @@ module RGeo
         end
         false
       end
-      
-      
+
+
     end
-    
-    
+
+
     class FFIGeometryCollectionImpl < FFIGeometryImpl  # :nodoc:
-      
-      
+
+
       Feature::MixinCollection::GLOBAL.for_type(Feature::GeometryCollection).include_in_class(self)
-      
-      
+
+
       def geometry_type
         Feature::GeometryCollection
       end
-      
-      
+
+
       def eql?(rhs_)
         if rhs_.class == self.class && rhs_.factory.eql?(@factory)
           size_ = @fg_geom.num_geometries
@@ -541,14 +616,14 @@ module RGeo
         end
         false
       end
-      
-      
+
+
       def num_geometries
         @fg_geom.num_geometries
       end
       alias_method :size, :num_geometries
-      
-      
+
+
       def geometry_n(n_)
         if n_ >= 0 && n_ < @fg_geom.num_geometries
           @factory.wrap_fg_geom(@fg_geom.get_geometry_n(n_),
@@ -557,8 +632,8 @@ module RGeo
           nil
         end
       end
-      
-      
+
+
       def [](n_)
         n_ += @fg_geom.num_geometries if n_ < 0
         if n_ >= 0 && n_ < @fg_geom.num_geometries
@@ -568,52 +643,52 @@ module RGeo
           nil
         end
       end
-      
-      
+
+
       def each
         @fg_geom.num_geometries.times do |n_|
           yield @factory.wrap_fg_geom(@fg_geom.get_geometry_n(n_),
             @_klasses ? @_klasses[n_] : nil)
         end
       end
-      
+
       include ::Enumerable
-      
-      
+
+
     end
-    
-    
+
+
     class FFIMultiPointImpl < FFIGeometryCollectionImpl  # :nodoc:
-      
-      
+
+
       Feature::MixinCollection::GLOBAL.for_type(Feature::MultiPoint).include_in_class(self)
-      
-      
+
+
       def geometry_type
         Feature::MultiPoint
       end
-      
-      
+
+
     end
-    
-    
+
+
     class FFIMultiLineStringImpl < FFIGeometryCollectionImpl  # :nodoc:
-      
-      
+
+
       Feature::MixinCollection::GLOBAL.for_type(Feature::MultiCurve).include_in_class(self)
       Feature::MixinCollection::GLOBAL.for_type(Feature::MultiLineString).include_in_class(self)
-      
-      
+
+
       def geometry_type
         Feature::MultiLineString
       end
-      
-      
+
+
       def length
         @fg_geom.length
       end
-      
-      
+
+
       def is_closed?
         size_ = num_geometries
         size_.times do |n_|
@@ -621,41 +696,41 @@ module RGeo
         end
         true
       end
-      
-      
+
+
     end
-    
-    
+
+
     class FFIMultiPolygonImpl < FFIGeometryCollectionImpl  # :nodoc:
-      
-      
+
+
       Feature::MixinCollection::GLOBAL.for_type(Feature::MultiSurface).include_in_class(self)
       Feature::MixinCollection::GLOBAL.for_type(Feature::MultiPolygon).include_in_class(self)
-      
-      
+
+
       def geometry_type
         Feature::MultiPolygon
       end
-      
-      
+
+
       def area
         @fg_geom.area
       end
-      
-      
+
+
       def centroid
         @factory.wrap_fg_geom(@fg_geom.centroid, FFIPointImpl)
       end
-      
-      
+
+
       def point_on_surface
         @factory.wrap_fg_geom(@fg_geom.point_on_surface, FFIPointImpl)
       end
-      
-      
+
+
     end
-    
-    
+
+
   end
-  
+
 end