☕ ray class

Author: michalbe

src/main.rs

@@ -1,16 +1,34 @@
 extern crate vec3D;
 
+mod ray;
+
 use vec3D::Vec3D;
+use ray::Ray;
+
+fn color(ray: Ray) -> Vec3D {
+    let unit_direction = ray.direction.unit();
+    let t = 0.5 * (unit_direction.y + 1.0);
+    (Vec3D::new(1.0, 1.0, 1.0) * (1.0 - t)) + (Vec3D::new(0.5, 0.7, 1.0) * t)
+}
 
 fn main() {
-    let max_i = 200;
-    let max_j = 200;
+    let nx = 200;
+    let ny = 100;
+
+    println!("P3\n{} {} 255", nx, ny);
 
-    println!("P3\n{} {} 255", max_i, max_j);
-    for i in 0..max_i {
-        for j in 0..max_j {
+    let lower_left_corner = Vec3D::new(-2.0, -1.0, -1.0);
+    let horizontal = Vec3D::new(4.0, 0.0, 0.0);
+    let vertical = Vec3D::new(0.0, 2.0, 0.0);
+    let origin = Vec3D::new(0.0, 0.0, 0.0);
 
-            let col = Vec3D::new(i as f64 / max_i as f64, j as f64 / max_j as f64, 0.2);
+    for j in (0..ny).rev() {
+        for i in 0..nx {
+            let u = i as f64 / nx as f64;
+            let v = j as f64 / ny as f64;
+            let r = Ray::new(origin, lower_left_corner + (horizontal * u) + (vertical * v));
+            let col = color(r);
+            // println!("{}, t", col);
             let r = (255.00 * col.x).round();
             let g = (255.00 * col.y).round();
             let b = (255.00 * col.z).round();

src/ray.rs

@@ -0,0 +1,19 @@
+use vec3D::Vec3D;
+
+pub struct Ray {
+    pub origin: Vec3D,
+    pub direction: Vec3D,
+}
+
+impl Ray {
+    pub fn new(origin: Vec3D, direction: Vec3D) -> Ray {
+        Ray {
+            origin,
+            direction,
+        }
+    }
+
+    pub fn point_at_parameter(self, t: f64) -> Vec3D {
+        self.origin + self.direction * t
+    }
+}