aboutsummaryrefslogtreecommitdiff
path: root/lib/hittable.rb
blob: facdd14ae1fa2e596e70c61d0cfec6208ad0d7ab (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
class HitRecord
  def initialize(point, t, ray, out_normal, material)
    @point = point
    @t = t
    front_face = ray.direction.dot(out_normal) < 0
    @normal = front_face ? out_normal : -out_normal
    @material = material
  end

  attr_accessor :point, :normal, :t, :material
end

class Hittable
  def hit(ray, trange)
    nil
  end
end

class Hittables < Hittable
  def initialize
    clear
  end

  def clear
    @objects = []
  end

  def <<(object)
    @objects << object
  end

  def hit(ray, trange)
    rec = nil
    closest = trange.max

    @objects.each do |object|
      if trec = object.hit(ray, Interval.new(trange.min, closest))
        rec = trec
        closest = trec.t
      end
    end

    rec
  end
end

class Sphere < Hittable
  def initialize(ox, oy, oz, radius = 1, material)
    @centre = Point.new(ox, oy, oz)
    @radius = radius
    @material = material
  end

  attr_reader :centre, :radius

  def hit(ray, trange)
    oc = @centre - ray.origin
    a = ray.direction.mag_sqr
    h = ray.direction.dot(oc)
    c = oc.mag_sqr - @radius ** 2
    disc = h ** 2 - a * c
    
    return nil if disc < 0

    sqrtd = disc ** 0.5
    root = (h - sqrtd) / a
    if !trange.surround?(root)
      root = (h + sqrtd) / a
      if !trange.surround?(root)
        return nil
      end
    end

    t = root
    p = ray.at(t)
    o_n = (p - @centre) / @radius
    HitRecord.new(p, t, ray, o_n, @material)
  end
end