rubytracer 0.1.0

data/.gitignore

@@ -0,0 +1,5 @@
+pkg/*
+*.gem
+.bundle
+Gemfile.lock
+.rvmrc

data/Gemfile

@@ -0,0 +1,4 @@
+source "http://rubygems.org"
+
+# Specify your gem's dependencies in Geom3d.gemspec
+gemspec

data/Rakefile

@@ -0,0 +1,7 @@
+require 'bundler'
+Bundler::GemHelper.install_tasks
+
+require 'rspec/core/rake_task'
+RSpec::Core::RakeTask.new(:spec)
+
+task :default => :spec

data/lib/rubytracer/camera.rb

@@ -0,0 +1,20 @@
+module Rubytracer
+  class Camera
+    def initialize(view, scene)
+      @view = view
+      @scene = scene
+    end
+
+    def colour_of_pixel(row, col)
+      ray = @view.eye_ray(row, col)
+      obj, t = @scene.intersect(ray)
+      if t == Float::INFINITY
+        Colour.new(0.6,0.6,0.6)
+      else
+        pos = ray.pos(t)
+        normal = obj.normal(pos)
+        obj.material.lit_colour(@scene, normal, -ray.dir, pos)
+      end
+    end
+  end
+end

data/lib/rubytracer/colour.rb

@@ -0,0 +1,88 @@
+module Rubytracer
+  class Colour
+    attr_reader :r, :g, :b
+
+    def initialize(r,g,b)
+      @r = r
+      @g = g
+      @b = b
+    end
+
+    def - other
+      Colour.new(@r - other.r, @g - other.g, @b - other.b)
+    end
+
+    def + other
+      Colour.new(@r + other.r, @g + other.g, @b + other.b)
+    end
+
+    def * other
+      case other
+      when Colour
+        Colour.new(@r * other.r, @g * other.g, @b * other.b)
+      else
+        Colour.new(@r * other, @g * other, @b * other)
+      end
+    end
+
+    def ** power
+      Colour.new(@r ** power, @g ** power, @b ** power)
+    end
+
+    def / scale
+      self * (1.0/scale)
+    end
+
+    def +@
+      self
+    end
+
+    def -@
+      Colour.new(-@r, -@g, -@b)
+    end
+
+    def to_int
+      [[[0, @r * 256].max, 255].min, [[0, @g * 256].max, 255].min, [[0, @b * 256].max, 255].min]
+    end
+  end
+end
+
+
+class Fixnum
+  alias :times_without_rubytracer :'*'
+
+  def * other
+    case other
+    when Rubytracer::Colour
+      Rubytracer::Colour.new(self * other.r, self * other.g, self * other.b)
+    else
+      self.times_without_rubytracer(other)
+    end
+  end
+end
+
+class Bignum
+  alias :times_without_rubytracer :'*'
+
+  def * other
+    case other
+    when Rubytracer::Colour
+      Rubytracer::Colour.new(self * other.r, self * other.g, self * other.b)
+    else
+      self.times_without_rubytracer(other)
+    end
+  end
+end
+
+class Float
+  alias :times_without_rubytracer :'*'
+
+  def * other
+    case other
+    when Rubytracer::Colour
+      Rubytracer::Colour.new(self * other.r, self * other.g, self * other.b)
+    else
+      self.times_without_rubytracer(other)
+    end
+  end
+end

data/lib/rubytracer/lights/ambient.rb

@@ -0,0 +1,17 @@
+require 'rubytracer/colour'
+
+module Rubytracer
+  class AmbientLight
+    def initialize value
+      @value = value
+    end
+
+    def specular(normal, view_vector, point, scene)
+      return Colour.new(0,0,0)
+    end
+
+    def diffuse(normal, point, scene)
+      return @value
+    end
+  end
+end

data/lib/rubytracer/lights/point.rb

@@ -0,0 +1,46 @@
+require 'geom3d'
+
+module Rubytracer
+  class PointLight
+    def initialize(value, point)
+      @value = value
+      @point = point
+    end
+
+    def check_shadow(point, direction, distance, scene)
+      ray = Geom3d::Ray.new(point + direction * 0.00001, direction)
+      obj, alpha = scene.intersect(ray)
+      pos = ray.pos(alpha)
+      alpha < distance
+    end
+
+    def distance_loss(distance)
+      ([0, 8 - distance].max / 8) ** 3
+    end
+
+    def specular(normal, view_vector, point, scene)
+      vector = @point - point
+      dir = vector.unit
+      dis = vector.length
+
+      if check_shadow(point, dir, dis, scene)
+        Colour.new(0,0,0)
+      else
+        d = (2 * view_vector.dot(normal) * normal - view_vector).unit
+        @value * distance_loss(dis) * [0, d.dot(dir)].max
+      end
+    end
+
+    def diffuse(normal, point, scene)
+      vector = @point - point
+      dir = vector.unit
+      dis = vector.length
+
+      if check_shadow(point, dir, dis, scene)
+        Colour.new(0,0,0)
+      else
+        @value * distance_loss(dis) * [0, dir.dot(normal)].max
+      end
+    end
+  end
+end

data/lib/rubytracer/lights.rb

@@ -0,0 +1,6 @@
+%w{
+  rubytracer/lights/ambient
+  rubytracer/lights/point
+}.each do |file|
+  require file
+end

data/lib/rubytracer/material.rb

@@ -0,0 +1,43 @@
+module Rubytracer
+  class Material
+    def initialize(options)
+      @d_colour = options[:diffuse_colour]
+      @s_colour = options[:specular_colour]
+      @shininess = options[:shininess]
+    end
+
+    def diffuse_colour(point, normal)
+      @d_colour || Colour.new(0,0,0)
+    end
+
+    def specular_colour(point, normal)
+      @s_colour || Colour.new(0,0,0)
+    end
+
+    def lit_colour(scene, normal, view_vector, point, n=0)
+      reflection = Colour.new(0,0,0)
+
+      #reflection += self.reflection(
+      #reflection += self.refraction(
+      
+      scene.lighting.each do |light|
+        reflection += self.diffuse(point, normal, scene, light)
+        reflection += self.specular(point, normal, scene, view_vector, light)
+      end
+
+      return reflection
+    end
+
+    def diffuse(point, normal, scene, light)
+      self.diffuse_colour(point, normal) * light.diffuse(normal, point, scene)
+    end
+
+    def specular(point, normal, scene, view_vector, light)
+      if @shininess
+        self.specular_colour(point, normal) * (light.specular(normal, view_vector, point, scene) ** @shininess)
+      else
+        Colour.new(0,0,0)
+      end
+    end
+  end
+end

data/lib/rubytracer/scene.rb

@@ -0,0 +1,26 @@
+module Rubytracer
+  class Scene
+    attr_reader :lighting
+
+    def initialize(objects=[], lighting=[])
+      @objects = objects
+      @lighting = lighting
+    end
+
+    def add_object(object)
+      @objects << object
+    end
+
+    def add_light(light)
+      @lighting << light
+    end
+
+    def intersect(ray)
+      min = [nil, Float::INFINITY]
+      @objects.map { |object| [object, object.intersect(ray)[0]] }.each do |object, intersection|
+        min = [object, intersection] if intersection > 0 && intersection < min[1]
+      end
+      min
+    end
+  end
+end

data/lib/rubytracer/shapes/sphere.rb

@@ -0,0 +1,30 @@
+module Rubytracer
+  class Sphere
+    attr_reader :material
+
+    def initialize(center, radius, material)
+      @center = center
+      @radius = radius
+      @material = material
+    end
+
+    def normal(point)
+      (point - @center).unit
+    end
+
+    def intersect(ray)
+      q = @center - ray.start
+      v_dot_q = ray.dir.dot(q)
+      square_diffs = q.dot(q) - @radius ** 2
+      discrim = v_dot_q ** 2 - square_diffs
+      if discrim >= 0
+        root = Math.sqrt(discrim)
+        t0 = v_dot_q - root
+        t1 = v_dot_q + root
+        [t0, t1]
+      else
+        [Float::INFINITY, -Float::INFINITY] # May not be portable
+      end
+    end
+  end
+end

data/lib/rubytracer/shapes.rb

@@ -0,0 +1,5 @@
+%w{
+  rubytracer/shapes/sphere
+}.each do |file|
+  require file
+end

data/lib/rubytracer/version.rb

@@ -0,0 +1,3 @@
+module Rubytracer
+  VERSION = "0.1.0"
+end

data/lib/rubytracer/view.rb

@@ -0,0 +1,24 @@
+require 'geom3d'
+
+module Rubytracer
+  class View
+    def initialize(view_point, look_at_point, view_up, h_fov, height, width, multi)
+      view_direction = (look_at_point - view_point).unit
+      view_up = view_up - view_direction * view_up.dot(view_direction)
+      spacing = 2 * Math.tan(h_fov * Math::PI / 360) / width
+
+      @point = view_point
+      @multi = multi
+      @y_vector = -spacing * view_up.unit
+      @x_vector = spacing * view_up.unit.cross(view_direction.unit).unit
+      @top_left = (view_point + view_direction) - (@y_vector * height / 2  + @x_vector * width / 2)
+    end
+
+    def eye_ray(row, col)
+      dy = @y_vector * (row + 0.5)
+      dx = @x_vector * (col + 0.5)
+      point = @top_left + dy + dx
+      Geom3d::Ray.new(@point, point - @point)
+    end
+  end
+end   

data/lib/rubytracer.rb

@@ -0,0 +1,11 @@
+%w{
+  rubytracer/scene
+  rubytracer/view
+  rubytracer/camera
+  rubytracer/colour
+  rubytracer/material
+  rubytracer/shapes
+  rubytracer/lights
+}.each do |file|
+  require file
+end

data/rubytracer.gemspec

@@ -0,0 +1,26 @@
+# -*- encoding: utf-8 -*-
+$:.push File.expand_path("../lib", __FILE__)
+require "rubytracer/version"
+
+Gem::Specification.new do |s|
+  s.name        = "rubytracer"
+  s.version     = Rubytracer::VERSION
+  s.platform    = Gem::Platform::RUBY
+  s.authors     = ["Nemo157 (Wim Looman)"]
+  s.email       = ["ghostunderscore@gmail.com"]
+  s.homepage    = ""
+  s.summary     = %q{A simple pure ruby raytracer library}
+  s.description = %q{A simple pure ruby raytracer library}
+  s.licenses    = ["Apache License 2.0"]
+
+  s.rubyforge_project = "rubytracer"
+
+  s.files         = `git ls-files`.split("\n")
+  s.test_files    = `git ls-files -- {test,spec,features}/*`.split("\n")
+  s.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
+  s.require_paths = ["lib"]
+
+  s.add_runtime_dependency 'geom3d'
+
+  s.add_development_dependency 'rspec', '~>2.3.0'
+end

data/spec/spec_helper.rb

@@ -0,0 +1 @@
+require 'rspec'

metadata

@@ -0,0 +1,108 @@
+--- !ruby/object:Gem::Specification 
+name: rubytracer
+version: !ruby/object:Gem::Version 
+  prerelease: false
+  segments: 
+  - 0
+  - 1
+  - 0
+  version: 0.1.0
+platform: ruby
+authors: 
+- Nemo157 (Wim Looman)
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2011-01-13 00:00:00 +13:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: geom3d
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        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 
+        segments: 
+        - 2
+        - 3
+        - 0
+        version: 2.3.0
+  type: :development
+  version_requirements: *id002
+description: A simple pure ruby raytracer library
+email: 
+- ghostunderscore@gmail.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- .gitignore
+- Gemfile
+- Rakefile
+- lib/rubytracer.rb
+- lib/rubytracer/camera.rb
+- lib/rubytracer/colour.rb
+- lib/rubytracer/lights.rb
+- lib/rubytracer/lights/ambient.rb
+- lib/rubytracer/lights/point.rb
+- lib/rubytracer/material.rb
+- lib/rubytracer/scene.rb
+- lib/rubytracer/shapes.rb
+- lib/rubytracer/shapes/sphere.rb
+- lib/rubytracer/version.rb
+- lib/rubytracer/view.rb
+- rubytracer.gemspec
+- spec/spec_helper.rb
+has_rdoc: true
+homepage: ""
+licenses: 
+- Apache License 2.0
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: rubytracer
+rubygems_version: 1.3.7
+signing_key: 
+specification_version: 3
+summary: A simple pure ruby raytracer library
+test_files: 
+- spec/spec_helper.rb