lib/terraformer/multi_point.rb in terraformer-0.0.1 vs lib/terraformer/multi_point.rb in terraformer-0.0.2

@@ -1,10 +1,73 @@
 module Terraformer
 
   class MultiPoint < Geometry
 
+    def initialize *args
+      case
+      when Point === args[0]
+        self.coordinates = args.map &:coordinates
+      else
+        super *args
+      end
+    end
+
     def first_coordinate
       coordinates[0]
+    end
+
+    def points
+      coordinates.map {|p| Point.new p}
+    end
+
+    def == obj
+      super obj do |o|
+        self.coordinates.sort == obj.coordinates.sort
+      end
+    end
+
+    def contains? obj
+      points.any? {|p| p.contains? obj}
+    end
+
+    def within? obj
+      case obj
+      when MultiPoint
+        points.all? {|p| obj.contains? p}
+      when LineString
+        points.all? {|p| obj.contains? p}
+      when MultiLineString
+        points.all? {|p| obj.contains? p}
+      when Polygon
+        obj.contains? self
+      when MultiPolygon
+        points.all? {|p| obj.polygons.any? {|polygon| polygon.contains? p}}
+      else
+        raise ArgumentError.new "unsupported type: #{obj.type rescue obj.class}"
+      end
+    end
+
+    def add_point p
+      p = p.coordinates if Point === p
+      raise ArugmentError unless Coordinate === p
+      coordinates << p
+    end
+    alias_method :<<, :add_point
+
+    def insert_point idx, p
+      p = p.coordinates if Point === p
+      raise ArugmentError unless Coordinate === p
+      coordinates.insert idx, p
+    end
+
+    def remove_point p
+      p = p.coordinates if Point === p
+      raise ArugmentError unless Coordinate === p
+      coordinates.delete p
+    end
+
+    def remove_point_at idx
+      coordinates.delete_at idx
     end
 
   end
 
 end