From 49c6adb0db70ffc30eaac33b66eacf7574b34e26 Mon Sep 17 00:00:00 2001 From: Julian T Date: Wed, 10 Feb 2021 23:06:48 +0100 Subject: Fixed most clippy warnings --- src/world/mod.rs | 23 +++++++++++++++ src/world/scene.rs | 52 ++++++++++++++++++++++++++++++++++ src/world/shapes/mod.rs | 4 +++ src/world/shapes/sphere.rs | 70 ++++++++++++++++++++++++++++++++++++++++++++++ 4 files changed, 149 insertions(+) create mode 100644 src/world/mod.rs create mode 100644 src/world/scene.rs create mode 100644 src/world/shapes/mod.rs create mode 100644 src/world/shapes/sphere.rs (limited to 'src/world') diff --git a/src/world/mod.rs b/src/world/mod.rs new file mode 100644 index 0000000..ef239db --- /dev/null +++ b/src/world/mod.rs @@ -0,0 +1,23 @@ +//! Manages world objects, and implements intersection +pub mod shapes; + +mod scene; +pub use scene::*; + +use std::rc::Rc; +use crate::core::Hittable; +use crate::material::Material; + +pub struct Object { + pub shape: Box, + pub mat: Rc, +} + +impl Object { + pub fn new(mat: Rc, shape: Box) -> Self { + Object { + mat, + shape, + } + } +} diff --git a/src/world/scene.rs b/src/world/scene.rs new file mode 100644 index 0000000..444e915 --- /dev/null +++ b/src/world/scene.rs @@ -0,0 +1,52 @@ +use crate::core::{Ray, Intersection}; +use crate::material::Material; + +use super::Object; + +pub struct Scene { + objs: Vec, +} + +pub struct SceneIntersect<'a> { + pub mat: &'a dyn Material, + pub i: Intersection, +} + +impl Scene { + pub fn new() -> Self { + Self::default() + } + + pub fn add_object(&mut self, obj: Object) { + self.objs.push(obj); + } + + pub fn add_objects(&mut self, objs: Vec) { + for obj in objs { + self.add_object(obj); + } + } + + pub fn intersect(&self, ray: &Ray) -> Option { + let mut min: Option = None; + + for obj in self.objs.iter() { + if let Some(i) = obj.shape.intersect(&ray) { + match min { + Some(ref si) if si.i.t < i.t => (), + _ => min = Some(SceneIntersect {i, mat: obj.mat.as_ref() }), + } + } + } + + min + } +} + +impl Default for Scene { + fn default() -> Self { + Self { + objs: Vec::new(), + } + } +} diff --git a/src/world/shapes/mod.rs b/src/world/shapes/mod.rs new file mode 100644 index 0000000..d7583ad --- /dev/null +++ b/src/world/shapes/mod.rs @@ -0,0 +1,4 @@ +mod sphere; + +pub use sphere::Sphere; + diff --git a/src/world/shapes/sphere.rs b/src/world/shapes/sphere.rs new file mode 100644 index 0000000..8e0816b --- /dev/null +++ b/src/world/shapes/sphere.rs @@ -0,0 +1,70 @@ +//! Implements sphere +//! +//! Spheres are relatively easy to calculate intersections between +use crate::Float; +use crate::core::{Ray, Vector3f, Hittable, Intersection}; + +pub struct Sphere { + radius: Float, + center: Vector3f, +} + +impl Sphere { + pub fn new(radius: Float, center: Vector3f) -> Sphere { + Sphere { + radius, + center, + } + } + + fn norm_at(&self, point: &Vector3f) -> Vector3f { + let mut v = *point - self.center; + v /= self.radius; + v + } +} + +impl Hittable for Sphere { + // Implementation from ray tracing in a weekend + fn intersect(&self, ray: &Ray) -> Option { + let oc = ray.origin - self.center; + let a = ray.direction.len_squared(); + let half_b = oc.dot(&ray.direction); + let c = oc.len_squared() - self.radius * self.radius; + let disc = half_b*half_b - a*c; + + if disc < 0.0 { + None + } else { + let distance = (-half_b - disc.sqrt()) / a; + if distance < 0.0 { + return None + } + let w = ray.at(distance); + Some(Intersection { + n: self.norm_at(&w), + p: w, + t: distance, + }) + } + + } +} + +#[cfg(test)] +mod tests { + use super::*; + + #[test] + fn sphere_intersect() { + let sph = Sphere::new(2.0, Vector3f::new_xyz(2.0, 3.0, 4.0)); + + let ray = Ray { + origin: Vector3f::new_xyz(1.0, 0.0, 0.0), + direction: Vector3f::new_xyz(0.0, 1.0, 1.5).norm(), + }; + + let dist = sph.intersect(&ray).unwrap(); + assert!((dist.t - 3.28).abs() < 0.01); + } +} -- cgit v1.2.3