1 files changed, 15 insertions, 10 deletions

diff --git a/src/camera.rs b/src/camera.rs
index 8e9cf68..ec37c70 100644
--- a/src/camera.rs
+++ b/src/camera.rs
@@ -13,6 +13,7 @@ pub struct Camera {
     delta_pixel_u: Vec3,
     delta_pixel_v: Vec3,
     pixel_samples_scale: f32,
+    max_depth: usize,
 }
 
 impl Camera {
@@ -45,6 +46,8 @@ impl Camera {
             &delta_pixel_u.add(&delta_pixel_u).scalar_mul(0.5)
         );
 
+        let max_depth = 50;
+
         Self {
             aspect_ratio,
             image_width,
@@ -55,6 +58,7 @@ impl Camera {
             delta_pixel_v,
             samples_per_pixel,
             pixel_samples_scale,
+            max_depth
         }
     }
 
@@ -73,18 +77,18 @@ impl Camera {
         }
     }
 
-    fn ray_color(&self, ray: &Ray, world: &HittableList) -> Color {
+    fn ray_color(&self, ray: &Ray, world: &HittableList, depth: usize) -> Color {
+        if depth <= 0 {
+            return Color::new(0.0, 0.0, 0.0);
+        }
+
         let mut rec = HitRecord::default();
         let ray_t = Interval::new(0.0, f32::INFINITY);
         if world.hit(ray, ray_t, &mut rec) {
-            let color = Color::new(1.0, 1.0, 1.0)
-                .add(&Color::new(
-                    rec.normal.x,
-                    rec.normal.y,
-                    rec.normal.z)
-                )
-                .mul_scalar(0.5);
-            return color;
+            let direction = Vec3::random_on_hemisphere(&rec.normal);
+            let origin = rec.position.clone();
+            let r = Ray { origin, direction };
+            return self.ray_color(&r, world, depth - 1).mul_scalar(0.5);
         }
 
         let unit_direction = ray.direction.unit();
@@ -97,10 +101,11 @@ impl Camera {
     pub fn render(&self, display_buffer: &mut DisplayBuffer, world: &HittableList) {
         (0..self.image_height).for_each(|j| {
             (0..self.image_width).for_each(|i| {
+                println!("Processing line {} col {}", j, i);
                 let mut pixel_color = Color::new(0.0, 0.0, 0.0);
                 (0..self.samples_per_pixel).for_each(|_| {
                     let r = self.get_ray(i, j);
-                    let ray_color = self.ray_color(&r, world);
+                    let ray_color = self.ray_color(&r, world, self.max_depth);
                     pixel_color = pixel_color.add(&ray_color);
                 });
                 pixel_color = pixel_color.mul_scalar(self.pixel_samples_scale);