still having nan problems

This commit is contained in:
Michael Zhang 2023-03-25 05:05:53 -05:00
parent da58ee9824
commit d1d7b59081
Signed by: michael
GPG key ID: BDA47A31A3C8EE6B
11 changed files with 73 additions and 36 deletions

View file

@ -0,0 +1,2 @@
[registries.crates-io]
protocol = "sparse"

View file

@ -42,6 +42,7 @@ dependencies = [
"anyhow",
"base64",
"clap",
"contracts",
"derivative",
"generator",
"itertools",
@ -142,6 +143,17 @@ dependencies = [
"os_str_bytes",
]
[[package]]
name = "contracts"
version = "0.6.3"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "f1d1429e3bd78171c65aa010eabcdf8f863ba3254728dbfb0ad4b1545beac15c"
dependencies = [
"proc-macro2",
"quote",
"syn",
]
[[package]]
name = "crossbeam-channel"
version = "0.5.6"

View file

@ -22,6 +22,7 @@ path = "src/main.rs"
anyhow = { version = "1.0.68", features = ["backtrace"] }
base64 = "0.21.0"
clap = { version = "4.1.4", features = ["cargo", "derive"] }
contracts = "0.6.3"
derivative = "2.2.0"
generator = "0.7.2"
itertools = "0.10.5"

View file

@ -2,6 +2,13 @@
.PRECIOUS: $(EXAMPLES_PPM)
DEBUG :=
RELEASE_FLAG := --release
ifeq ($(DEBUG),1)
RELEASE_FLAG :=
endif
RAYTRACER_FLAGS :=
DOCKER := docker
ZIP := zip
@ -35,7 +42,7 @@ $(HANDIN): $(BINARY) Makefile Cargo.toml Cargo.lock README.md $(EXAMPLES_PNG) $(
$(ZIP) -r $@ src examples $^
examples/%.ppm: examples/%.txt $(SOURCES)
cargo run --release -- -o $@ $(RAYTRACER_FLAGS) $<
cargo run $(RELEASE_FLAG) -- -o $@ $(RAYTRACER_FLAGS) $<
examples/%.png: examples/%.ppm
convert $< $@

View file

@ -3,6 +3,8 @@ use nalgebra::{Vector2, Vector3};
#[macro_use]
extern crate anyhow;
#[macro_use]
extern crate contracts;
#[macro_use]
extern crate derivative;
#[macro_use]
extern crate tracing;

View file

@ -28,4 +28,14 @@ impl Ray {
pub fn eval(&self, time: f64) -> Point {
self.origin + self.direction * time
}
/// Check if any of the components is NaN
pub fn has_nan(&self) -> bool {
self.origin.x.is_nan()
|| self.origin.y.is_nan()
|| self.origin.z.is_nan()
|| self.direction.x.is_nan()
|| self.direction.y.is_nan()
|| self.direction.z.is_nan()
}
}

View file

@ -137,24 +137,24 @@ impl Scene {
fresnel_coefficient * r_lambda
};
let transparency = {
let n = intersection_context.normal;
let i = incident_ray.direction;
let (eta_i, eta_t) = match intersection_context.exiting {
true => (material.eta, 1.0),
false => (1.0, material.eta),
};
let transparency = if eta_t < 1.0 {
Vector::default()
} else {
let n = intersection_context.normal.normalize();
let i = incident_ray.direction.normalize();
assert!(eta_t != 0.0, "wtf eta_t is 0");
// Slide 69
let cos_theta_i = dot(i, n);
assert!(cos_theta_i != std::f64::NAN);
let sin_theta_i = (1.0 - cos_theta_i.powi(2)).sqrt();
assert!(sin_theta_i != std::f64::NAN);
let sin_theta_t = (eta_i / eta_t) * sin_theta_i;
assert!(sin_theta_t != std::f64::NAN);
let cos_theta_t = (1.0 - sin_theta_t.powi(2)).sqrt();
assert!(cos_theta_t != std::f64::NAN);
let fresnel_coefficient =
self.compute_fresnel_coefficient(&material, i, n);
@ -172,6 +172,12 @@ impl Scene {
let origin = origin + JITTER_CONST * t;
let ray = Ray::new(origin, t);
assert!(
!ray.has_nan(),
"ray is nan WTF {cos_theta_i} {sin_theta_i} ({}) {sin_theta_t} {cos_theta_t} | normal: {:?}",
eta_i / eta_t,
intersection_context.normal,
);
let t_lambda = self.trace_single_ray(ray, depth + 1)?;
(1.0 - fresnel_coefficient) * (1.0 - material.alpha) * t_lambda

View file

@ -1,6 +0,0 @@
use super::data::Material;
pub struct MaterialStack<'a> {
stack: Vec<&'a f64>,
}

View file

@ -5,16 +5,15 @@ pub mod input_file;
pub mod object;
pub mod sphere;
pub mod texture;
pub mod triangle;
pub mod tracing;
pub mod materials;
pub mod triangle;
use crate::image::Color;
use crate::{Point, Point2, Vector};
use self::data::{Attenuation, DepthCueing, Light, Material};
use self::object::Object;
use self::texture::{Texture, NormalMap};
use self::texture::{NormalMap, Texture};
#[derive(Debug, Default)]
pub struct Scene {

View file

@ -6,11 +6,11 @@
"rust-analyzer-src": "rust-analyzer-src"
},
"locked": {
"lastModified": 1674240251,
"narHash": "sha256-AVMmf/CtcGensTZmMicToDpOwySEGNKYgRPC7lu3m8w=",
"lastModified": 1679638953,
"narHash": "sha256-OFATLLft/9vXHf+zqa6OQGTQ2rrCuD97KRzsrPsJRdY=",
"owner": "nix-community",
"repo": "fenix",
"rev": "d8067f4d1d3d30732703209bec5ca7d62aaececc",
"rev": "1336152ce31c0f781ae121aa297a5699c642a9fe",
"type": "github"
},
"original": {
@ -35,11 +35,11 @@
},
"nixpkgs": {
"locked": {
"lastModified": 1673796341,
"narHash": "sha256-1kZi9OkukpNmOaPY7S5/+SlCDOuYnP3HkXHvNDyLQcc=",
"lastModified": 1679437018,
"narHash": "sha256-vOuiDPLHSEo/7NkiWtxpHpHgoXoNmrm+wkXZ6a072Fc=",
"owner": "nixos",
"repo": "nixpkgs",
"rev": "6dccdc458512abce8d19f74195bb20fdb067df50",
"rev": "19cf008bb18e47b6e3b4e16e32a9a4bdd4b45f7e",
"type": "github"
},
"original": {
@ -51,16 +51,17 @@
},
"nixpkgs_2": {
"locked": {
"lastModified": 1667629849,
"narHash": "sha256-P+v+nDOFWicM4wziFK9S/ajF2lc0N2Rg9p6Y35uMoZI=",
"lastModified": 1679721255,
"narHash": "sha256-R199ew5XtM/fXy0P9Ls+sOtYWc9Cubz2c2Ept0csUVg=",
"owner": "nixos",
"repo": "nixpkgs",
"rev": "3bacde6273b09a21a8ccfba15586fb165078fb62",
"rev": "798e23beab9b5cba4d6f05e8b243e1d4535770f3",
"type": "github"
},
"original": {
"id": "nixpkgs",
"type": "indirect"
"owner": "nixos",
"repo": "nixpkgs",
"type": "github"
}
},
"root": {
@ -73,11 +74,11 @@
"rust-analyzer-src": {
"flake": false,
"locked": {
"lastModified": 1674162026,
"narHash": "sha256-iY0bxoVE7zAZmp0BB/m5hZW5pWHUfgntDvc1m2zyt/U=",
"lastModified": 1679520343,
"narHash": "sha256-AJGSGWRfoKWD5IVTu1wEsR990wHbX0kIaolPqNMEh0c=",
"owner": "rust-lang",
"repo": "rust-analyzer",
"rev": "6e52c64031825920983515b9e975e93232739f7f",
"rev": "eb791f31e688ae00908eb75d4c704ef60c430a92",
"type": "github"
},
"original": {

View file

@ -1,5 +1,8 @@
{
inputs = { fenix.url = "github:nix-community/fenix"; };
inputs = {
nixpkgs.url = "github:nixos/nixpkgs";
fenix.url = "github:nix-community/fenix";
};
outputs = { self, nixpkgs, flake-utils, fenix }:
flake-utils.lib.eachDefaultSystem (system: