ruby-freenect 0.0.0

data/lib/freenect/freenect.rb

@@ -0,0 +1,149 @@
+module Freenect
+  
+  FREENECT_COUNTS_PER_G     = Driver::FREENECT_COUNTS_PER_G
+  FREENECT_DEVICE_FLAGS     = Driver::FREENECT_DEVICE_FLAGS
+  FREENECT_RESOLUTION       = Driver::FREENECT_RESOLUTION
+  FREENECT_VIDEO_FORMAT     = Driver::FREENECT_VIDEO_FORMAT
+  FREENECT_DEPTH_FORMAT     = Driver::FREENECT_DEPTH_FORMAT
+  FREENECT_LED_OPTIONS      = Driver::FREENECT_LED_OPTIONS
+  FREENECT_TILT_STATUS_CODE = Driver::FREENECT_TILT_STATUS_CODE
+  FREENECT_LOGLEVEL         = Driver::FREENECT_LOGLEVEL
+  
+  class << self
+    def method_missing(meth, *args, &blk)
+      Interface.send(meth, *args, &blk)
+    end
+    def init(*args)
+      Context.new(*args)
+    end
+  end
+  
+  class Context        
+    include Freenect::Driver
+    attr_reader :context
+    def initialize
+      context_p = FFI::MemoryPointer.new(:pointer)
+      freenect_init(context_p, nil)
+      @context = context_p.read_pointer
+    end
+    def get_device_count
+      freenect_num_devices(context)
+    end
+    def close
+      freenect_shutdown(context)
+    end    
+  end
+  
+  class Interface
+    extend Freenect::Driver
+
+    @video_mode, @depth_mode = {}, {}
+    
+    class << self
+            
+      def get_device_count
+        context = Freenect::Context.new
+        count = context.get_device_count
+        context.close; count
+      end
+                  
+      def get_video_mode_count
+        freenect_get_video_mode_count
+      end
+      
+      def get_video_mode(mode_id)
+        freenect_get_video_mode(mode_id)
+      end
+      
+      def get_current_video_mode(idx=0)
+        (@video_mode || {})[idx]
+      end
+      
+      def find_video_mode(format=:freenect_video_rgb, resolution=:freenect_resolution_medium)
+        freenect_find_video_mode(resolution, format)
+      end
+
+      def get_video(video_mode, idx=0)
+        video_p, timestamp_p = pointer(:pointer, 1), pointer(:uint32)
+        set_current_video_mode(video_mode, idx)
+        freenect_sync_get_video(video_p, timestamp_p, idx, Freenect::FREENECT_VIDEO_FORMAT[video_mode[:format][:video_format]])
+        video_p.read_pointer.get_bytes(0, video_mode[:bytes])
+      end
+
+      def get_depth_mode_count
+        freenect_get_depth_mode_count
+      end
+
+      def get_depth_mode(mode_id)
+        freenect_get_depth_mode(mode_id)
+      end
+
+      def get_current_depth_mode(idx=0)
+        (@depth_mode || {})[idx]
+      end
+
+      def find_depth_mode(format=:freenect_depth_11bit, resolution=:freenect_resolution_medium)
+        freenect_find_depth_mode(resolution, format)
+      end
+        
+      def get_depth(depth_mode, idx=0)
+        depth_p, timestamp_p = pointer(:pointer, 1), pointer(:uint32)
+        set_current_depth_mode(depth_mode, idx)
+        freenect_sync_get_depth(depth_p, timestamp_p, idx, Freenect::FREENECT_DEPTH_FORMAT[depth_mode[:format][:depth_format]])
+        depth_p.read_pointer.get_bytes(0, depth_mode[:bytes])
+      end
+
+      def set_tilt(angle, idx=0)
+        freenect_sync_set_tilt_degs(angle, idx)
+      end
+
+      def get_tilt(tilt_state=nil, idx=0)
+        tilt_state ||= get_tilt_state(idx)
+        freenect_get_tilt_degs(tilt_state)
+      end
+      
+      def get_tilt_status(tilt_state=nil, idx=0)
+        tilt_state ||= get_tilt_state(idx)
+        freenect_get_tilt_status(tilt_state)
+      end
+      
+      def get_tilt_state(idx=0)
+        state_ptr = pointer(:pointer)
+        freenect_sync_get_tilt_state(state_ptr, idx)
+        FreenectRawTiltState.new(state_ptr.get_pointer(0))
+      end
+
+      def get_acceleration(tilt_state=nil, idx=0)
+        tilt_state ||= get_tilt_state(idx)
+        xacc = pointer(:double)
+        yacc = pointer(:double)
+        zacc = pointer(:double)
+        freenect_get_mks_accel(tilt_state, xacc, yacc, zacc)
+        {:x=>xacc.read_double, :y=>yacc.read_double, :z=>zacc.read_double}
+      end
+
+      def set_led(led_option, idx=0)
+        freenect_sync_set_led(Freenect::FREENECT_LED_OPTIONS[led_option], idx)
+      end
+      
+      def stop
+        freenect_sync_stop()
+      end
+
+      private
+      
+        def set_current_video_mode(video_mode, idx)
+          (@video_mode ||={})[idx] = video_mode
+        end
+
+        def set_current_depth_mode(depth_mode, idx)
+          (@depth_mode ||= {})[idx] = depth_mode
+        end
+         
+        def pointer(*args)
+          FFI::MemoryPointer.new(*args)
+        end
+      
+    end
+  end
+end

data/ruby-freenect.gemspec

@@ -0,0 +1,56 @@
+# Generated by jeweler
+# DO NOT EDIT THIS FILE DIRECTLY
+# Instead, edit Jeweler::Tasks in Rakefile, and run 'rake gemspec'
+# -*- encoding: utf-8 -*-
+
+Gem::Specification.new do |s|
+  s.name = "ruby-freenect"
+  s.version = "0.0.0"
+
+  s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
+  s.authors = ["Troy Stribling"]
+  s.date = "2012-01-02"
+  s.description = "Ruby bindings for the libfreenect Kinect driver"
+  s.extra_rdoc_files = [
+    "LICENSE.txt",
+    "README.rdoc"
+  ]
+  s.files = [
+    ".gems",
+    ".rvmrc",
+    "LICENSE.txt",
+    "README.rdoc",
+    "Rakefile",
+    "VERSION",
+    "examples/depth.rb",
+    "examples/ir.rb",
+    "examples/kinect.rb",
+    "examples/led.rb",
+    "examples/tilt.rb",
+    "examples/tools.rb",
+    "examples/video.rb",
+    "freenect4r.gemspec",
+    "lib/freenect.rb",
+    "lib/freenect/driver.rb",
+    "lib/freenect/freenect.rb",
+    "ruby-freenect.gemspec"
+  ]
+  s.homepage = "https://github.com/troystribling/freenect4r"
+  s.rdoc_options = ["--title", "FFI Freenect", "--main", "README.rdoc", "--line-numbers"]
+  s.require_paths = ["lib"]
+  s.rubygems_version = "1.8.10"
+  s.summary = "Ruby bindings for the libfreenect Kinect driver"
+
+  if s.respond_to? :specification_version then
+    s.specification_version = 3
+
+    if Gem::Version.new(Gem::VERSION) >= Gem::Version.new('1.2.0') then
+      s.add_runtime_dependency(%q<ffi>, [">= 1.0.11"])
+    else
+      s.add_dependency(%q<ffi>, [">= 1.0.11"])
+    end
+  else
+    s.add_dependency(%q<ffi>, [">= 1.0.11"])
+  end
+end
+

metadata

@@ -0,0 +1,80 @@
+--- !ruby/object:Gem::Specification
+name: ruby-freenect
+version: !ruby/object:Gem::Version
+  version: 0.0.0
+  prerelease: 
+platform: ruby
+authors:
+- Troy Stribling
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-01-02 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: ffi
+  requirement: &81131700 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 1.0.11
+  type: :runtime
+  prerelease: false
+  version_requirements: *81131700
+description: Ruby bindings for the libfreenect Kinect driver
+email: 
+executables: []
+extensions: []
+extra_rdoc_files:
+- LICENSE.txt
+- README.rdoc
+files:
+- .gems
+- .rvmrc
+- LICENSE.txt
+- README.rdoc
+- Rakefile
+- VERSION
+- examples/depth.rb
+- examples/ir.rb
+- examples/kinect.rb
+- examples/led.rb
+- examples/tilt.rb
+- examples/tools.rb
+- examples/video.rb
+- freenect4r.gemspec
+- lib/freenect.rb
+- lib/freenect/driver.rb
+- lib/freenect/freenect.rb
+- ruby-freenect.gemspec
+homepage: https://github.com/troystribling/freenect4r
+licenses: []
+post_install_message: 
+rdoc_options:
+- --title
+- FFI Freenect
+- --main
+- README.rdoc
+- --line-numbers
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.10
+signing_key: 
+specification_version: 3
+summary: Ruby bindings for the libfreenect Kinect driver
+test_files: []