feat(raytracer): add minimal phong lighting demo

2024-11-17 23:50:05 +01:00
parent e8a2b0f059
commit b458b99c82
11 changed files with 1192 additions and 51 deletions
--- a/src/bin/rt_demo.rs
+++ b/src/bin/rt_demo.rs
@@ -1,5 +1,86 @@
-use lispers::raytracer::{scene::Scene, types::Light};
+use lispers::raytracer::{
+    camera::Camera,
+    plane::Plane,
+    scene::Scene,
+    sphere::Sphere,
+    types::{Color, Light, Material, Point3, Vector3},
+};
+extern crate nalgebra as na;

 fn main() {
-    let scene = Scene::new();
+    let mut scene = Scene::new();
+
+    scene.set_ambient(Color::new(0.2, 0.2, 0.2));
+
+    scene.add_light(Light {
+        position: Point3::new(4.0, 7.0, 10.0),
+        color: Color::new(1.0, 1.0, 1.0),
+    });
+    scene.add_light(Light {
+        position: Point3::new(-2.0, 7.0, 10.0),
+        color: Color::new(1.0, 1.0, 1.0),
+    });
+
+    scene.add_object(Box::new(Plane::new(
+        Point3::new(0.0, -1.0, 0.0),
+        Vector3::new(0.0, 1.0, 0.0),
+        Material::new(
+            Color::new(0.5, 0.5, 0.5),
+            Color::new(0.5, 0.5, 0.5),
+            Color::new(0.0, 0.0, 0.0),
+            0.0,
+            0.6,
+        ),
+    )));
+
+    scene.add_object(Box::new(Sphere::new(
+        Point3::new(-2.0, 0.0, 1.0),
+        1.0,
+        Material::new(
+            Color::new(0.0, 1.0, 0.0),
+            Color::new(0.0, 1.0, 0.0),
+            Color::new(0.6, 0.6, 0.6),
+            20.0,
+            0.3,
+        ),
+    )));
+
+    scene.add_object(Box::new(Sphere::new(
+        Point3::new(0.2, -0.5, -0.2),
+        0.5,
+        Material::new(
+            Color::new(0.0, 0.0, 1.0),
+            Color::new(0.0, 0.0, 1.0),
+            Color::new(0.6, 0.6, 0.6),
+            20.0,
+            0.3,
+        ),
+    )));
+
+    scene.add_object(Box::new(Sphere::new(
+        Point3::new(-0.5, 0.5, -2.0),
+        1.5,
+        Material::new(
+            Color::new(1.0, 0.0, 0.0),
+            Color::new(1.0, 0.0, 0.0),
+            Color::new(0.6, 0.6, 0.6),
+            20.0,
+            0.3,
+        ),
+    )));
+
+    let camera = Camera::new(
+        Point3::new(0.0, 0.7, 5.0),
+        Point3::new(-1.0, -0.5, 0.0),
+        Vector3::new(0.0, 1.0, 0.0),
+        60.0,
+        400,
+        300,
+    );
+
+    let fname = "demo-scene.png";
+    match camera.render(&scene, 5, 2).save(fname) {
+        Ok(_) => println!("Image saved to {}", fname),
+        Err(e) => println!("Error saving image: {}", e),
+    }
 }