quadtree 1.0.2 → 1.0.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 78274ec810f20ddb345e5af2f239011c45bb493005acb132cacb1ca23146b463
-  data.tar.gz: cac53ba45cdb796ea599d820d2c8c8f9eee3a41b3f73f2d7ec3b0eedf0542e8a
+  metadata.gz: d5db2d5026956693fb12005b6f413baec2df1c55336d54612c69c2d2ea5daee7
+  data.tar.gz: 64fad915c832732dce1b3d8f5d4e09153d6c2a5d0eacbe341ae9e36432ae910f
 SHA512:
-  metadata.gz: 7fbebccb0db1cb2aff816ff0fb9dd2de44bdb8fce97f7f0abe25d158ea192b95639e80a2aee49734df8f08b0aa7717e0f2bb18dfad590e57f2b26105fed51298
-  data.tar.gz: 1d1107de6c8641e61f605c552aefb5b54bedaeabb0a3f4e644913a8f61846bd91079a5b2e20dafc8b14a45301b71c494eecf8288881b2a96cf2dc7541affd1bb
+  metadata.gz: ba095f44c64f53fe6c708a55e59b2da9e43f4317c309fb8677ca24d0e9538c6104eec212c3ce05e092b17816a0b226e2a114ee86b16b5ce92cfc1a8dd582a681
+  data.tar.gz: 3d2ea5d459b276877dae1e42300f5b457c88b557cd88dc3240c4f6864cb430c0044c5ee27ceac0d0d6b1b2a1ee573f627548d12373a66eb4760f0b9c09f91712

data/lib/quadtree/point.rb

@@ -8,16 +8,17 @@ module Quadtree
       # @return [Float] Y coordinate
       attr_accessor :y
 
-      # Payload attached to this {Point}.
+      # @return [Object] Payload attached to this {Point}.
       attr_accessor :data
 
       # Create a new {Point}.
       #
-      # @param [Float] x X coordinate
-      # @param [Float] y Y coordinate
+      # @param [Float, Numeric] x X coordinate.
+      # @param [Float, Numeric] y Y coordinate.
+      # @param [Object] data {Point} payload (optional).
       def initialize(x, y, data=nil)
-        @x = x
-        @y = y
+        @x = x.to_f
+        @y = y.to_f
         @data = data unless data.nil?
       end
 
@@ -48,17 +49,17 @@ module Quadtree
         # earth's radius
         r = 6371 * 1000.0
         # coverting degrees to radians
-        lat1 = self.y * (Math::PI / 180)
-        lat2 = other.y * (Math::PI / 180)
-        dlat = (other.y - self.y) * (Math::PI / 180)
-        dlon = (other.x - self.x) * (Math::PI / 180)
+        lat1 = self.y * (Math::PI / 180.0)
+        lat2 = other.y * (Math::PI / 180.0)
+        dlat = (other.y - self.y) * (Math::PI / 180.0)
+        dlon = (other.x - self.x) * (Math::PI / 180.0)
 
         # a = sin²(Δφ/2) + cos φ_1 ⋅ cos φ_2 ⋅ sin²(Δλ/2)
-        a = Math.sin(dlat / 2) * Math.sin(dlat / 2) +
+        a = Math.sin(dlat / 2.0) * Math.sin(dlat / 2.0) +
             Math.cos(lat1) * Math.cos(lat2) *
-            Math.sin(dlon / 2) * Math.sin(dlon / 2)
+            Math.sin(dlon / 2.0) * Math.sin(dlon / 2.0)
         # c = 2 ⋅ atan2( √a, √(1−a) )
-        c = 2 * Math.atan2(Math.sqrt(a), Math.sqrt(1-a))
+        c = 2.0 * Math.atan2(Math.sqrt(a), Math.sqrt(1-a))
         # d = R ⋅ c
         return r * c
       end

data/lib/quadtree/quadtree.rb

@@ -28,29 +28,29 @@ module Quadtree
     attr_accessor :south_east
 
     def initialize(boundary)
-      @boundary = boundary
-      @points = []
-      @north_west = nil
-      @north_east = nil
-      @south_west = nil
-      @south_east = nil
+      self.boundary = boundary
+      self.points = []
+      self.north_west = nil
+      self.north_east = nil
+      self.south_west = nil
+      self.south_east = nil
     end
 
     # @param [Point] point
     # @return [Boolean]
     def insert!(point)
-      return false unless @boundary.contains_point?(point)
+      return false unless self.boundary.contains_point?(point)
 
-      if points.size < NODE_CAPACITY
-        @points << point
+      if self.points.size < NODE_CAPACITY
+        self.points << point
         return true
       end
 
-      subdivide! if @north_west.nil?
-      return true if @north_west.insert!(point)
-      return true if @north_east.insert!(point)
-      return true if @south_west.insert!(point)
-      return true if @south_east.insert!(point)
+      subdivide! if self.north_west.nil?
+      return true if self.north_west.insert!(point)
+      return true if self.north_east.insert!(point)
+      return true if self.south_west.insert!(point)
+      return true if self.south_east.insert!(point)
 
       false
     end
@@ -64,21 +64,21 @@ module Quadtree
       points_in_range = []
 
       # Automatically abort if the range does not intersect this quad
-      return points_in_range unless @boundary.intersects?(range)
+      return points_in_range unless self.boundary.intersects?(range)
 
       # Check objects at this quad level
-      @points.each do |point|
+      self.points.each do |point|
         points_in_range << point if range.contains_point?(point)
       end
 
       # Terminate here, if there are no children
-      return points_in_range if @north_west.nil?
+      return points_in_range if self.north_west.nil?
 
       # Otherwise, add the points from the children
-      points_in_range += @north_west.query_range(range)
-      points_in_range += @north_east.query_range(range)
-      points_in_range += @south_west.query_range(range)
-      points_in_range += @south_east.query_range(range)
+      points_in_range += self.north_west.query_range(range)
+      points_in_range += self.north_east.query_range(range)
+      points_in_range += self.south_west.query_range(range)
+      points_in_range += self.south_east.query_range(range)
 
       points_in_range
     end
@@ -87,29 +87,30 @@ module Quadtree
 
     # @return [Boolean]
     def subdivide!
-      left_edge = @boundary.left
-      right_edge = @boundary.right
-      top_edge = @boundary.top
-      bottom_edge = @boundary.bottom
-      quad_half_dimension = @boundary.half_dimension / 2
-
-      north_west_center = Quadtree::Point.new left_edge + quad_half_dimension, top_edge - quad_half_dimension
-      north_east_center = Quadtree::Point.new right_edge - quad_half_dimension, top_edge - quad_half_dimension
-      south_east_center = Quadtree::Point.new left_edge + quad_half_dimension, bottom_edge + quad_half_dimension
-      south_west_center = Quadtree::Point.new right_edge - quad_half_dimension, bottom_edge + quad_half_dimension
-
-      north_west_boundary = Quadtree::AxisAlignedBoundingBox.new north_west_center, quad_half_dimension
-      north_east_boundary = Quadtree::AxisAlignedBoundingBox.new north_east_center, quad_half_dimension
-      south_west_boundary = Quadtree::AxisAlignedBoundingBox.new south_west_center, quad_half_dimension
-      south_east_boundary = Quadtree::AxisAlignedBoundingBox.new south_east_center, quad_half_dimension
-
-      @north_west = Quadtree::Quadtree.new north_west_boundary
-      @north_east = Quadtree::Quadtree.new north_east_boundary
-      @south_west = Quadtree::Quadtree.new south_west_boundary
-      @south_east = Quadtree::Quadtree.new south_east_boundary
+      left_edge = self.boundary.left
+      right_edge = self.boundary.right
+      top_edge = self.boundary.top
+      bottom_edge = self.boundary.bottom
+      quad_half_dimension = self.boundary.half_dimension / 2
+
+      north_west_center = Point.new left_edge + quad_half_dimension, top_edge - quad_half_dimension
+      north_east_center = Point.new right_edge - quad_half_dimension, top_edge - quad_half_dimension
+      south_east_center = Point.new left_edge + quad_half_dimension, bottom_edge + quad_half_dimension
+      south_west_center = Point.new right_edge - quad_half_dimension, bottom_edge + quad_half_dimension
+
+      north_west_boundary = AxisAlignedBoundingBox.new north_west_center, quad_half_dimension
+      north_east_boundary = AxisAlignedBoundingBox.new north_east_center, quad_half_dimension
+      south_west_boundary = AxisAlignedBoundingBox.new south_west_center, quad_half_dimension
+      south_east_boundary = AxisAlignedBoundingBox.new south_east_center, quad_half_dimension
+
+      self.north_west = Quadtree.new north_west_boundary
+      self.north_east = Quadtree.new north_east_boundary
+      self.south_west = Quadtree.new south_west_boundary
+      self.south_east = Quadtree.new south_east_boundary
 
       true
-    rescue
+    rescue => error
+      puts "Something went wrong: #{error}"
       false
     end
   end

data/lib/quadtree/version.rb

@@ -1,3 +1,3 @@
 module Quadtree
-  VERSION = "1.0.2"
+  VERSION = "1.0.3"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: quadtree
 version: !ruby/object:Gem::Version
-  version: 1.0.2
+  version: 1.0.3
 platform: ruby
 authors:
 - Jan Lindblom