ffi-kdtree 0.0.2

data/spec/find_nearest_spec.rb

@@ -0,0 +1,40 @@
+require File.expand_path(File.join(File.dirname(__FILE__), 'spec_helper'))
+
+describe KdTree::Tree do
+  include TestHelpers
+
+  it "should find neighbors in a 2D tree" do
+    tree = KdTree::Tree.new(2, :string)
+    tree.add([-1,   -1  ], "top-left"         )
+    tree.add([-0.5, -0.5], "mid-upper-left"   )
+    tree.add([ 1,   -1  ], "top-right"        )
+    tree.add([ 0.5, -0.5], "mid-upper-right"  )
+    tree.add([ 0,    0  ], "center"           )
+    tree.add([-0.5,  0.5], "mid-lower-left"   )
+    tree.add([-1,    1  ], "bottom-left"      )
+    tree.add([ 0.5,  0.5], "mid-lower-right"  )
+    tree.add([ 1,    1  ], "bottom-right"     )
+
+    where, data = tree.find_nearest([0, 0])
+    where.should == [0, 0]
+    data.should == "center"
+
+    where, data = tree.find_nearest([0.3, 0.3])
+    where.should == [0.5, 0.5]
+    data.should == "mid-lower-right"
+
+    where, data = tree.find_nearest([-0.3, -0.3])
+    where.should == [-0.5, -0.5]
+    data.should == "mid-upper-left"
+
+    where, data = tree.find_nearest([0.1,0.1])
+    where.should == [0, 0]
+    data.should == "center"
+
+    locations = tree.find_nearest_range([0, 0], 2)
+    locations.size.should == 9 # all of them
+
+    locations = tree.find_nearest_range([0, 0], 1)
+    locations.size.should == 5 # center + mids
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,15 @@
+require 'rubygems'
+require 'spec'
+require File.join(File.dirname(__FILE__), '..', 'lib', 'ffi-kdtree')
+
+module TestHelpers
+  def generate_tree(dim_size, num_entries, type = :object, &block)
+    block ||= proc.new { |i| i }
+    tree = KdTree::Tree.new(dim_size, type)
+    1.upto(num_entries).each do |i|
+      tree.add(1.upto(dim_size).collect { rand }, block.call(i))
+    end
+  end
+end
+
+

data/spec/tree_spec.rb

@@ -0,0 +1,12 @@
+require File.expand_path(File.join(File.dirname(__FILE__), 'spec_helper'))
+
+describe KdTree::Tree do
+  include TestHelpers
+
+  it "should create a 2D tree with 1000 random entries" do
+    tree = generate_tree(2, 1000, :integer) do
+      rand
+    end
+    tree.should_not be_nil
+  end
+end

metadata

@@ -0,0 +1,130 @@
+--- !ruby/object:Gem::Specification 
+name: ffi-kdtree
+version: !ruby/object:Gem::Version 
+  hash: 27
+  prerelease: false
+  segments: 
+  - 0
+  - 0
+  - 2
+  version: 0.0.2
+platform: ruby
+authors: 
+- Scott Deming
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2012-03-01 00:00:00 -06:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: ffi
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :runtime
+  version_requirements: *id001
+- !ruby/object:Gem::Dependency 
+  name: rspec
+  prerelease: false
+  requirement: &id002 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        hash: 15
+        segments: 
+        - 2
+        - 6
+        version: "2.6"
+  type: :development
+  version_requirements: *id002
+- !ruby/object:Gem::Dependency 
+  name: rake
+  prerelease: false
+  requirement: &id003 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :development
+  version_requirements: *id003
+description: |-
+  This is a simple FFI wrapped version of libkdtree for Ruby. 
+    It's a pure ruby wrapper so this gem can be loaded and run by any ruby runtime 
+    that supports FFI. That's all of them: MRI 1.9.x, Rubinius and JRuby.
+email: 
+- sdeming@makefile.com
+executables: []
+
+extensions: 
+- ext/ffi-kdtree/extconf.rb
+extra_rdoc_files: []
+
+files: 
+- .gitignore
+- README.md
+- examples/test.rb
+- ext/ffi-kdtree/extconf.rb
+- ext/ffi-kdtree/kdtree.c
+- ext/ffi-kdtree/kdtree.h
+- ffi-kdtree.gemspec
+- lib/ffi-kdtree.rb
+- lib/ffi-kdtree/data_conversion.rb
+- lib/ffi-kdtree/dimensions.rb
+- lib/ffi-kdtree/ffi/libkdtree.rb
+- lib/ffi-kdtree/finders.rb
+- lib/ffi-kdtree/helpers.rb
+- lib/ffi-kdtree/tree.rb
+- spec/data/zips.txt
+- spec/find_nearest_spec.rb
+- spec/spec_helper.rb
+- spec/tree_spec.rb
+has_rdoc: true
+homepage: https://github.com/blakesmith/ffi-kdtree
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.7
+signing_key: 
+specification_version: 3
+summary: This is a simple FFI wrapped version of libkdtree for Ruby
+test_files: []
+