Compare commits
202 Commits
5f3bfd744e
...
master
| Author | SHA1 | Date | |
|---|---|---|---|
| f93d215fc2 | |||
| 593632a9e3 | |||
| ed2ee749cd | |||
| 42f3daefaa | |||
| 9430a1e7da | |||
| d3153032b1 | |||
| 35071b06ac | |||
| 5f0e7a26dd | |||
| 6fbdb49ce1 | |||
| 23bc5b0bf0 | |||
| 37137ac9ca | |||
| 9353ff675e | |||
| 4b8bd84a84 | |||
| e19ec20c7b | |||
| deb46acb5a | |||
| 1076e6dcaf | |||
| 7d9750b9d0 | |||
| 88b8c547e0 | |||
| ac555beafc | |||
| 0158f9ea15 | |||
| 450342c3d4 | |||
| 41f9fa2742 | |||
| 7ec30d8557 | |||
| f51d3396f4 | |||
| a2f9166b5a | |||
| f73a471cb6 | |||
| cd149755cb | |||
| 9188ce17fa | |||
| 63975bad96 | |||
| df928e1779 | |||
| 3c28466d68 | |||
| a0b79ee2fa | |||
| 4e62975d56 | |||
| eea5c7c61e | |||
| 188b550fb7 | |||
| f7c5f29e67 | |||
| 6ab3021403 | |||
| d213e04c11 | |||
| 739b38b4ed | |||
| 5ba5aa5f5d | |||
| 4e6e9bf78a | |||
| beeb5e479b | |||
| fc1bfa419e | |||
| 95827a4a52 | |||
| d3dd002883 | |||
| ef737c6df9 | |||
| 2c490b7e83 | |||
| 63f8fba6a4 | |||
| 5c2786a54d | |||
| 2d696932e3 | |||
| 27d6c1280b | |||
| 4506418706 | |||
| 1d8aff7905 | |||
| 585ad4805c | |||
| b7f163c5a9 | |||
| 4ab9425a97 | |||
| 468cba97b3 | |||
| b9ebc186fa | |||
| 3e9d900f1e | |||
| 9e81acfda9 | |||
| f8ec874d13 | |||
| a8756debb8 | |||
| a0fb4637b5 | |||
| 6069bf9a65 | |||
| eeb7813243 | |||
| c644299726 | |||
| e6db61543b | |||
| 39eeb79409 | |||
| 54e72cd81d | |||
| 2d91f781f3 | |||
| 24e8b4f9cf | |||
| a12938db95 | |||
| 4066bf4b85 | |||
| b432e9a6dd | |||
| 62317d57ae | |||
| 8d92cc861e | |||
| f2ade1eee2 | |||
| a4baedefec | |||
| 8adf1bcadb | |||
| aea437785a | |||
| 91fd65259c | |||
| e5ffe87192 | |||
| ac73d13fb0 | |||
| 6b4be0ed1e | |||
| 85b87a6854 | |||
| d15a9e6c3e | |||
| a2012e6742 | |||
| 6d7998ad9f | |||
| 58646e4142 | |||
| 94b0f8355e | |||
| f4d3129d5a | |||
| e31f5e0a3a | |||
| 84a0ba2ec6 | |||
| 5043a7e526 | |||
| 622c23d5ed | |||
| f2c68e0b6f | |||
| 287344c272 | |||
| 8bc5e347cc | |||
| f0da916a22 | |||
| 386daf5876 | |||
| a4adefdb23 | |||
| f3aace486b | |||
| 93bfeb9125 | |||
| 55af087d69 | |||
| 78f7ca8956 | |||
| 51185e9e84 | |||
| 1ca903c64b | |||
| 5e7139f0ba | |||
| 665ae244d7 | |||
| e574cdb592 | |||
| 270a7ec349 | |||
| de6cd0da4d | |||
| 7a80179f41 | |||
| 926fffa29f | |||
| 9befbd9ad2 | |||
| c882fc81e5 | |||
| 1c2caf2cc5 | |||
| 9006671a26 | |||
| 3838efd134 | |||
| e3d8988658 | |||
| e4846de25b | |||
| 4352c03d20 | |||
| 1d5e5a164b | |||
| f476822bcd | |||
| 135a519526 | |||
| 5d6b3e6d57 | |||
| 3aea76b35c | |||
| cd2a4770ca | |||
| 5debb16d10 | |||
| 42e8ebe3bd | |||
| 1d61f59935 | |||
| 7f36aecf5e | |||
| 0c7bbae4a3 | |||
| eaae65712b | |||
| 68709da6c2 | |||
| 77215193fa | |||
| 74fe69188a | |||
| bdcee49d5a | |||
| 2e4e8b3dcd | |||
| b9f2c3f0ec | |||
| 8b79876aee | |||
| bfa3282a37 | |||
| 3e383c4dbd | |||
| c158d92252 | |||
| 62ad827507 | |||
| 363f15fb00 | |||
| be2041285c | |||
| 958c4c3ee8 | |||
| 62cb5e4ec4 | |||
| 4680c97adc | |||
| 0965ac9ddf | |||
| 4936839723 | |||
| c500d56d4d | |||
| c058d043e0 | |||
| 166c87dfe5 | |||
| 9389fed84c | |||
| 28fe6fe982 | |||
| 44b46187a0 | |||
| de898f0b0a | |||
| e041fd1f6a | |||
| 952ed8bf81 | |||
| a553786807 | |||
| b42adcebfc | |||
| d999e8196b | |||
| fc5ef09cc3 | |||
| e419994fae | |||
| ee8ef4e2c5 | |||
| 41f3b63ad0 | |||
| b2dc0e9509 | |||
| 2d8a3927f4 | |||
| 5600d6c561 | |||
| 0e8a0e4163 | |||
| 2f85697b88 | |||
| c0e422a7eb | |||
| 7741766635 | |||
| 3799f93393 | |||
| 1629b2cbfa | |||
| 839642b886 | |||
| c4f10126e3 | |||
| 7a8ed15017 | |||
| 5d8024a485 | |||
| ecf7cd7bdc | |||
| 1c06833658 | |||
| 95de5863cc | |||
| 5d57304d95 | |||
| 599f484dff | |||
| ce998c68eb | |||
| 37048ddd52 | |||
| 19b2ef6ded | |||
| 7e450d664e | |||
| 09047eb713 | |||
| 1065702a5d | |||
| efdc963a48 | |||
| 02ef8634c3 | |||
| 7e19d7e61b | |||
| dbf5451070 | |||
| 4f88d2c101 | |||
| 94ea724344 | |||
| 967920e1fa | |||
| 2eeeb2013b | |||
| 538b8ad364 | |||
| 2395c96e01 |
@@ -10,6 +10,14 @@ steps:
|
|||||||
- env | sort
|
- env | sort
|
||||||
- find $PWD
|
- find $PWD
|
||||||
|
|
||||||
|
- name: rtchallenge
|
||||||
|
image: registry.z.xinu.tv/drone/omnibus
|
||||||
|
commands:
|
||||||
|
- (cd rtchallenge && cargo build --examples)
|
||||||
|
- (cd rtchallenge && cargo test)
|
||||||
|
- (cd rtchallenge && cargo build --examples --no-default-features)
|
||||||
|
- (cd rtchallenge && cargo test --no-default-features)
|
||||||
|
|
||||||
- name: rtiow
|
- name: rtiow
|
||||||
image: registry.z.xinu.tv/drone/omnibus
|
image: registry.z.xinu.tv/drone/omnibus
|
||||||
commands:
|
commands:
|
||||||
|
|||||||
2
.gitignore
vendored
2
.gitignore
vendored
@@ -1,2 +1,4 @@
|
|||||||
**/target
|
**/target
|
||||||
**/*.rs.bk
|
**/*.rs.bk
|
||||||
|
**/zig-out
|
||||||
|
**/zig-cache
|
||||||
|
|||||||
@@ -5,3 +5,5 @@ Raytracers
|
|||||||
|
|
||||||
Random collection of ray tracing experiments.
|
Random collection of ray tracing experiments.
|
||||||
|
|
||||||
|
# TODO
|
||||||
|
Implement http://www.kevinbeason.com/smallpt/
|
||||||
|
|||||||
@@ -1,5 +1,7 @@
|
|||||||
#!/usr/bin/env bash
|
#!/usr/bin/env bash
|
||||||
set -e
|
set -e
|
||||||
|
SCRIPT_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" &> /dev/null && pwd )"
|
||||||
|
|
||||||
pushd () {
|
pushd () {
|
||||||
command pushd "$@" > /dev/null
|
command pushd "$@" > /dev/null
|
||||||
}
|
}
|
||||||
@@ -18,9 +20,14 @@ RT=rtchallenge
|
|||||||
AFFECTED="$(git diff-index --cached --name-only HEAD)"
|
AFFECTED="$(git diff-index --cached --name-only HEAD)"
|
||||||
#echo "AFFECTED $AFFECTED"
|
#echo "AFFECTED $AFFECTED"
|
||||||
RT_AFFECTED=$(echo "${AFFECTED:?}" | grep ${RT:?} || true)
|
RT_AFFECTED=$(echo "${AFFECTED:?}" | grep ${RT:?} || true)
|
||||||
|
NO_FEATURES_TARGET_DIR="${SCRIPT_DIR:?}/../../rtchallenge/target/no-default-features"
|
||||||
if [ ! -z "$RT_AFFECTED" ]; then
|
if [ ! -z "$RT_AFFECTED" ]; then
|
||||||
echo "Files for the rt challenge were touched, running tests"
|
echo "Files for the rt challenge were touched, running tests"
|
||||||
cd ${RT:?} && cargo build --examples && cargo test
|
cd ${RT:?} && \
|
||||||
|
cargo build --examples && \
|
||||||
|
cargo test && \
|
||||||
|
cargo build --target-dir=${NO_FEATURES_TARGET_DIR:?} --no-default-features --examples && \
|
||||||
|
cargo test --target-dir=${NO_FEATURES_TARGET_DIR:?} --no-default-features
|
||||||
fi
|
fi
|
||||||
# Uncomment to debug presubmit.
|
# Uncomment to debug presubmit.
|
||||||
#exit 1
|
#exit 1
|
||||||
|
|||||||
408
rtchallenge/Cargo.lock
generated
408
rtchallenge/Cargo.lock
generated
@@ -1,11 +1,37 @@
|
|||||||
# This file is automatically @generated by Cargo.
|
# This file is automatically @generated by Cargo.
|
||||||
# It is not intended for manual editing.
|
# It is not intended for manual editing.
|
||||||
|
version = 3
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "addr2line"
|
||||||
|
version = "0.15.2"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "e7a2e47a1fbe209ee101dd6d61285226744c6c8d3c21c8dc878ba6cb9f467f3a"
|
||||||
|
dependencies = [
|
||||||
|
"gimli",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "adler"
|
||||||
|
version = "1.0.2"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "f26201604c87b1e01bd3d98f8d5d9a8fcbb815e8cedb41ffccbeb4bf593a35fe"
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "adler32"
|
name = "adler32"
|
||||||
version = "1.2.0"
|
version = "1.2.0"
|
||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "aae1277d39aeec15cb388266ecc24b11c80469deae6067e17a1a7aa9e5c1f234"
|
checksum = "aae1277d39aeec15cb388266ecc24b11c80469deae6067e17a1a7aa9e5c1f234"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "ansi_term"
|
||||||
|
version = "0.11.0"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "ee49baf6cb617b853aa8d93bf420db2383fab46d314482ca2803b40d5fde979b"
|
||||||
|
dependencies = [
|
||||||
|
"winapi 0.3.9",
|
||||||
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "anyhow"
|
name = "anyhow"
|
||||||
version = "1.0.41"
|
version = "1.0.41"
|
||||||
@@ -20,7 +46,7 @@ checksum = "d9b39be18770d11421cdb1b9947a45dd3f37e93092cbf377614828a319d5fee8"
|
|||||||
dependencies = [
|
dependencies = [
|
||||||
"hermit-abi",
|
"hermit-abi",
|
||||||
"libc",
|
"libc",
|
||||||
"winapi",
|
"winapi 0.3.9",
|
||||||
]
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
@@ -29,6 +55,21 @@ version = "1.0.1"
|
|||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "cdb031dd78e28731d87d56cc8ffef4a8f36ca26c38fe2de700543e627f8a464a"
|
checksum = "cdb031dd78e28731d87d56cc8ffef4a8f36ca26c38fe2de700543e627f8a464a"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "backtrace"
|
||||||
|
version = "0.3.60"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "b7815ea54e4d821e791162e078acbebfd6d8c8939cd559c9335dceb1c8ca7282"
|
||||||
|
dependencies = [
|
||||||
|
"addr2line",
|
||||||
|
"cc",
|
||||||
|
"cfg-if",
|
||||||
|
"libc",
|
||||||
|
"miniz_oxide 0.4.4",
|
||||||
|
"object",
|
||||||
|
"rustc-demangle",
|
||||||
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "bitflags"
|
name = "bitflags"
|
||||||
version = "1.2.1"
|
version = "1.2.1"
|
||||||
@@ -68,6 +109,12 @@ dependencies = [
|
|||||||
"rustc_version",
|
"rustc_version",
|
||||||
]
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "cc"
|
||||||
|
version = "1.0.69"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "e70cc2f62c6ce1868963827bd677764c62d07c3d9a3e1fb1177ee1a9ab199eb2"
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "cfg-if"
|
name = "cfg-if"
|
||||||
version = "1.0.0"
|
version = "1.0.0"
|
||||||
@@ -80,9 +127,25 @@ version = "2.33.3"
|
|||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "37e58ac78573c40708d45522f0d80fa2f01cc4f9b4e2bf749807255454312002"
|
checksum = "37e58ac78573c40708d45522f0d80fa2f01cc4f9b4e2bf749807255454312002"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
|
"ansi_term",
|
||||||
|
"atty",
|
||||||
"bitflags",
|
"bitflags",
|
||||||
|
"strsim 0.8.0",
|
||||||
"textwrap",
|
"textwrap",
|
||||||
"unicode-width",
|
"unicode-width",
|
||||||
|
"vec_map",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "core_affinity"
|
||||||
|
version = "0.5.10"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "7f8a03115cc34fb0d7c321dd154a3914b3ca082ccc5c11d91bf7117dbbe7171f"
|
||||||
|
dependencies = [
|
||||||
|
"kernel32-sys",
|
||||||
|
"libc",
|
||||||
|
"num_cpus",
|
||||||
|
"winapi 0.2.8",
|
||||||
]
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
@@ -196,6 +259,41 @@ dependencies = [
|
|||||||
"memchr",
|
"memchr",
|
||||||
]
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "darling"
|
||||||
|
version = "0.12.4"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "5f2c43f534ea4b0b049015d00269734195e6d3f0f6635cb692251aca6f9f8b3c"
|
||||||
|
dependencies = [
|
||||||
|
"darling_core",
|
||||||
|
"darling_macro",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "darling_core"
|
||||||
|
version = "0.12.4"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "8e91455b86830a1c21799d94524df0845183fa55bafd9aa137b01c7d1065fa36"
|
||||||
|
dependencies = [
|
||||||
|
"fnv",
|
||||||
|
"ident_case",
|
||||||
|
"proc-macro2",
|
||||||
|
"quote",
|
||||||
|
"strsim 0.10.0",
|
||||||
|
"syn",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "darling_macro"
|
||||||
|
version = "0.12.4"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "29b5acf0dea37a7f66f7b25d2c5e93fd46f8f6968b1a5d7a3e02e97768afc95a"
|
||||||
|
dependencies = [
|
||||||
|
"darling_core",
|
||||||
|
"quote",
|
||||||
|
"syn",
|
||||||
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "deflate"
|
name = "deflate"
|
||||||
version = "0.8.6"
|
version = "0.8.6"
|
||||||
@@ -206,18 +304,114 @@ dependencies = [
|
|||||||
"byteorder",
|
"byteorder",
|
||||||
]
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "derive_builder"
|
||||||
|
version = "0.10.2"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "d13202debe11181040ae9063d739fa32cfcaaebe2275fe387703460ae2365b30"
|
||||||
|
dependencies = [
|
||||||
|
"derive_builder_macro",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "derive_builder_core"
|
||||||
|
version = "0.10.2"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "66e616858f6187ed828df7c64a6d71720d83767a7f19740b2d1b6fe6327b36e5"
|
||||||
|
dependencies = [
|
||||||
|
"darling",
|
||||||
|
"proc-macro2",
|
||||||
|
"quote",
|
||||||
|
"syn",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "derive_builder_macro"
|
||||||
|
version = "0.10.2"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "58a94ace95092c5acb1e97a7e846b310cfbd499652f72297da7493f618a98d73"
|
||||||
|
dependencies = [
|
||||||
|
"derive_builder_core",
|
||||||
|
"syn",
|
||||||
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "either"
|
name = "either"
|
||||||
version = "1.6.1"
|
version = "1.6.1"
|
||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "e78d4f1cc4ae33bbfc157ed5d5a5ef3bc29227303d595861deb238fcec4e9457"
|
checksum = "e78d4f1cc4ae33bbfc157ed5d5a5ef3bc29227303d595861deb238fcec4e9457"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "enum-utils"
|
||||||
|
version = "0.1.2"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "ed327f716d0d351d86c9fd3398d20ee39ad8f681873cc081da2ca1c10fed398a"
|
||||||
|
dependencies = [
|
||||||
|
"enum-utils-from-str",
|
||||||
|
"failure",
|
||||||
|
"proc-macro2",
|
||||||
|
"quote",
|
||||||
|
"serde_derive_internals",
|
||||||
|
"syn",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "enum-utils-from-str"
|
||||||
|
version = "0.1.2"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "d49be08bad6e4ca87b2b8e74146987d4e5cb3b7512efa50ef505b51a22227ee1"
|
||||||
|
dependencies = [
|
||||||
|
"proc-macro2",
|
||||||
|
"quote",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "failure"
|
||||||
|
version = "0.1.8"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "d32e9bd16cc02eae7db7ef620b392808b89f6a5e16bb3497d159c6b92a0f4f86"
|
||||||
|
dependencies = [
|
||||||
|
"backtrace",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "fnv"
|
||||||
|
version = "1.0.7"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "3f9eec918d3f24069decb9af1554cad7c880e2da24a9afd88aca000531ab82c1"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "getrandom"
|
||||||
|
version = "0.2.3"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "7fcd999463524c52659517fe2cea98493cfe485d10565e7b0fb07dbba7ad2753"
|
||||||
|
dependencies = [
|
||||||
|
"cfg-if",
|
||||||
|
"libc",
|
||||||
|
"wasi",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "gimli"
|
||||||
|
version = "0.24.0"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "0e4075386626662786ddb0ec9081e7c7eeb1ba31951f447ca780ef9f5d568189"
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "half"
|
name = "half"
|
||||||
version = "1.7.1"
|
version = "1.7.1"
|
||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "62aca2aba2d62b4a7f5b33f3712cb1b0692779a56fb510499d5c0aa594daeaf3"
|
checksum = "62aca2aba2d62b4a7f5b33f3712cb1b0692779a56fb510499d5c0aa594daeaf3"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "heck"
|
||||||
|
version = "0.3.3"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "6d621efb26863f0e9924c6ac577e8275e5e6b77455db64ffa6c65c904e9e132c"
|
||||||
|
dependencies = [
|
||||||
|
"unicode-segmentation",
|
||||||
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "hermit-abi"
|
name = "hermit-abi"
|
||||||
version = "0.1.19"
|
version = "0.1.19"
|
||||||
@@ -227,6 +421,12 @@ dependencies = [
|
|||||||
"libc",
|
"libc",
|
||||||
]
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "ident_case"
|
||||||
|
version = "1.0.1"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "b9e0384b61958566e926dc50660321d12159025e767c18e043daf26b70104c39"
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "itertools"
|
name = "itertools"
|
||||||
version = "0.9.0"
|
version = "0.9.0"
|
||||||
@@ -260,6 +460,16 @@ dependencies = [
|
|||||||
"wasm-bindgen",
|
"wasm-bindgen",
|
||||||
]
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "kernel32-sys"
|
||||||
|
version = "0.2.2"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "7507624b29483431c0ba2d82aece8ca6cdba9382bff4ddd0f7490560c056098d"
|
||||||
|
dependencies = [
|
||||||
|
"winapi 0.2.8",
|
||||||
|
"winapi-build",
|
||||||
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "lazy_static"
|
name = "lazy_static"
|
||||||
version = "1.4.0"
|
version = "1.4.0"
|
||||||
@@ -305,6 +515,16 @@ dependencies = [
|
|||||||
"adler32",
|
"adler32",
|
||||||
]
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "miniz_oxide"
|
||||||
|
version = "0.4.4"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "a92518e98c078586bc6c934028adcca4c92a53d6a958196de835170a01d84e4b"
|
||||||
|
dependencies = [
|
||||||
|
"adler",
|
||||||
|
"autocfg",
|
||||||
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "num-traits"
|
name = "num-traits"
|
||||||
version = "0.2.14"
|
version = "0.2.14"
|
||||||
@@ -324,6 +544,15 @@ dependencies = [
|
|||||||
"libc",
|
"libc",
|
||||||
]
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "object"
|
||||||
|
version = "0.25.3"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "a38f2be3697a57b4060074ff41b44c16870d916ad7877c17696e063257482bc7"
|
||||||
|
dependencies = [
|
||||||
|
"memchr",
|
||||||
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "oorandom"
|
name = "oorandom"
|
||||||
version = "11.1.3"
|
version = "11.1.3"
|
||||||
@@ -367,7 +596,37 @@ dependencies = [
|
|||||||
"bitflags",
|
"bitflags",
|
||||||
"crc32fast",
|
"crc32fast",
|
||||||
"deflate",
|
"deflate",
|
||||||
"miniz_oxide",
|
"miniz_oxide 0.3.7",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "ppv-lite86"
|
||||||
|
version = "0.2.10"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "ac74c624d6b2d21f425f752262f42188365d7b8ff1aff74c82e45136510a4857"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "proc-macro-error"
|
||||||
|
version = "1.0.4"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "da25490ff9892aab3fcf7c36f08cfb902dd3e71ca0f9f9517bea02a73a5ce38c"
|
||||||
|
dependencies = [
|
||||||
|
"proc-macro-error-attr",
|
||||||
|
"proc-macro2",
|
||||||
|
"quote",
|
||||||
|
"syn",
|
||||||
|
"version_check",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "proc-macro-error-attr"
|
||||||
|
version = "1.0.4"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "a1be40180e52ecc98ad80b184934baf3d0d29f979574e439af5a55274b35f869"
|
||||||
|
dependencies = [
|
||||||
|
"proc-macro2",
|
||||||
|
"quote",
|
||||||
|
"version_check",
|
||||||
]
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
@@ -388,6 +647,46 @@ dependencies = [
|
|||||||
"proc-macro2",
|
"proc-macro2",
|
||||||
]
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "rand"
|
||||||
|
version = "0.8.4"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "2e7573632e6454cf6b99d7aac4ccca54be06da05aca2ef7423d22d27d4d4bcd8"
|
||||||
|
dependencies = [
|
||||||
|
"libc",
|
||||||
|
"rand_chacha",
|
||||||
|
"rand_core",
|
||||||
|
"rand_hc",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "rand_chacha"
|
||||||
|
version = "0.3.1"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "e6c10a63a0fa32252be49d21e7709d4d4baf8d231c2dbce1eaa8141b9b127d88"
|
||||||
|
dependencies = [
|
||||||
|
"ppv-lite86",
|
||||||
|
"rand_core",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "rand_core"
|
||||||
|
version = "0.6.3"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "d34f1408f55294453790c48b2f1ebbb1c5b4b7563eb1f418bcfcfdbb06ebb4e7"
|
||||||
|
dependencies = [
|
||||||
|
"getrandom",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "rand_hc"
|
||||||
|
version = "0.3.1"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "d51e9f596de227fda2ea6c84607f5558e196eeaf43c986b724ba4fb8fdf497e7"
|
||||||
|
dependencies = [
|
||||||
|
"rand_core",
|
||||||
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "rayon"
|
name = "rayon"
|
||||||
version = "1.5.1"
|
version = "1.5.1"
|
||||||
@@ -439,11 +738,26 @@ name = "rtchallenge"
|
|||||||
version = "0.1.0"
|
version = "0.1.0"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"anyhow",
|
"anyhow",
|
||||||
|
"core_affinity",
|
||||||
"criterion",
|
"criterion",
|
||||||
|
"derive_builder",
|
||||||
|
"enum-utils",
|
||||||
|
"num_cpus",
|
||||||
"png",
|
"png",
|
||||||
|
"rand",
|
||||||
|
"rayon",
|
||||||
|
"serde",
|
||||||
|
"serde_json",
|
||||||
|
"structopt",
|
||||||
"thiserror",
|
"thiserror",
|
||||||
]
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "rustc-demangle"
|
||||||
|
version = "0.1.20"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "dead70b0b5e03e9c814bcb6b01e03e68f7c57a80aa48c72ec92152ab3e818d49"
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "rustc_version"
|
name = "rustc_version"
|
||||||
version = "0.4.0"
|
version = "0.4.0"
|
||||||
@@ -485,6 +799,9 @@ name = "serde"
|
|||||||
version = "1.0.126"
|
version = "1.0.126"
|
||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "ec7505abeacaec74ae4778d9d9328fe5a5d04253220a85c4ee022239fc996d03"
|
checksum = "ec7505abeacaec74ae4778d9d9328fe5a5d04253220a85c4ee022239fc996d03"
|
||||||
|
dependencies = [
|
||||||
|
"serde_derive",
|
||||||
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "serde_cbor"
|
name = "serde_cbor"
|
||||||
@@ -507,6 +824,17 @@ dependencies = [
|
|||||||
"syn",
|
"syn",
|
||||||
]
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "serde_derive_internals"
|
||||||
|
version = "0.25.0"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "1dbab34ca63057a1f15280bdf3c39f2b1eb1b54c17e98360e511637aef7418c6"
|
||||||
|
dependencies = [
|
||||||
|
"proc-macro2",
|
||||||
|
"quote",
|
||||||
|
"syn",
|
||||||
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "serde_json"
|
name = "serde_json"
|
||||||
version = "1.0.64"
|
version = "1.0.64"
|
||||||
@@ -518,6 +846,42 @@ dependencies = [
|
|||||||
"serde",
|
"serde",
|
||||||
]
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "strsim"
|
||||||
|
version = "0.8.0"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "8ea5119cdb4c55b55d432abb513a0429384878c15dde60cc77b1c99de1a95a6a"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "strsim"
|
||||||
|
version = "0.10.0"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "73473c0e59e6d5812c5dfe2a064a6444949f089e20eec9a2e5506596494e4623"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "structopt"
|
||||||
|
version = "0.3.22"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "69b041cdcb67226aca307e6e7be44c8806423d83e018bd662360a93dabce4d71"
|
||||||
|
dependencies = [
|
||||||
|
"clap",
|
||||||
|
"lazy_static",
|
||||||
|
"structopt-derive",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "structopt-derive"
|
||||||
|
version = "0.4.15"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "7813934aecf5f51a54775e00068c237de98489463968231a51746bbbc03f9c10"
|
||||||
|
dependencies = [
|
||||||
|
"heck",
|
||||||
|
"proc-macro-error",
|
||||||
|
"proc-macro2",
|
||||||
|
"quote",
|
||||||
|
"syn",
|
||||||
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "syn"
|
name = "syn"
|
||||||
version = "1.0.73"
|
version = "1.0.73"
|
||||||
@@ -568,6 +932,12 @@ dependencies = [
|
|||||||
"serde_json",
|
"serde_json",
|
||||||
]
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "unicode-segmentation"
|
||||||
|
version = "1.8.0"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "8895849a949e7845e06bd6dc1aa51731a103c42707010a5b591c0038fb73385b"
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "unicode-width"
|
name = "unicode-width"
|
||||||
version = "0.1.8"
|
version = "0.1.8"
|
||||||
@@ -580,6 +950,18 @@ version = "0.2.2"
|
|||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "8ccb82d61f80a663efe1f787a51b16b5a51e3314d6ac365b08639f52387b33f3"
|
checksum = "8ccb82d61f80a663efe1f787a51b16b5a51e3314d6ac365b08639f52387b33f3"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "vec_map"
|
||||||
|
version = "0.8.2"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "f1bddf1187be692e79c5ffeab891132dfb0f236ed36a43c7ed39f1165ee20191"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "version_check"
|
||||||
|
version = "0.9.3"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "5fecdca9a5291cc2b8dcf7dc02453fee791a280f3743cb0905f8822ae463b3fe"
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "walkdir"
|
name = "walkdir"
|
||||||
version = "2.3.2"
|
version = "2.3.2"
|
||||||
@@ -587,10 +969,16 @@ source = "registry+https://github.com/rust-lang/crates.io-index"
|
|||||||
checksum = "808cf2735cd4b6866113f648b791c6adc5714537bc222d9347bb203386ffda56"
|
checksum = "808cf2735cd4b6866113f648b791c6adc5714537bc222d9347bb203386ffda56"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"same-file",
|
"same-file",
|
||||||
"winapi",
|
"winapi 0.3.9",
|
||||||
"winapi-util",
|
"winapi-util",
|
||||||
]
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "wasi"
|
||||||
|
version = "0.10.2+wasi-snapshot-preview1"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "fd6fbd9a79829dd1ad0cc20627bf1ed606756a7f77edff7b66b7064f9cb327c6"
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "wasm-bindgen"
|
name = "wasm-bindgen"
|
||||||
version = "0.2.74"
|
version = "0.2.74"
|
||||||
@@ -655,6 +1043,12 @@ dependencies = [
|
|||||||
"wasm-bindgen",
|
"wasm-bindgen",
|
||||||
]
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "winapi"
|
||||||
|
version = "0.2.8"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "167dc9d6949a9b857f3451275e911c3f44255842c1f7a76f33c55103a909087a"
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "winapi"
|
name = "winapi"
|
||||||
version = "0.3.9"
|
version = "0.3.9"
|
||||||
@@ -665,6 +1059,12 @@ dependencies = [
|
|||||||
"winapi-x86_64-pc-windows-gnu",
|
"winapi-x86_64-pc-windows-gnu",
|
||||||
]
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "winapi-build"
|
||||||
|
version = "0.1.1"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "2d315eee3b34aca4797b2da6b13ed88266e6d612562a0c46390af8299fc699bc"
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "winapi-i686-pc-windows-gnu"
|
name = "winapi-i686-pc-windows-gnu"
|
||||||
version = "0.4.0"
|
version = "0.4.0"
|
||||||
@@ -677,7 +1077,7 @@ version = "0.1.5"
|
|||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "70ec6ce85bb158151cae5e5c87f95a8e97d2c0c4b001223f33a334e3ce5de178"
|
checksum = "70ec6ce85bb158151cae5e5c87f95a8e97d2c0c4b001223f33a334e3ce5de178"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"winapi",
|
"winapi 0.3.9",
|
||||||
]
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
|
|||||||
@@ -6,13 +6,29 @@ edition = "2018"
|
|||||||
|
|
||||||
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
|
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
|
||||||
|
|
||||||
|
[features]
|
||||||
|
default = [ "float-as-double" ]
|
||||||
|
float-as-double = []
|
||||||
|
|
||||||
[dependencies]
|
[dependencies]
|
||||||
anyhow = "1.0.41"
|
anyhow = "1.0.41"
|
||||||
|
core_affinity = "0.5.10"
|
||||||
criterion = "0.3.4"
|
criterion = "0.3.4"
|
||||||
|
derive_builder = "0.10.2"
|
||||||
|
enum-utils = "0.1.2"
|
||||||
|
num_cpus = "1.13.0"
|
||||||
png = "0.16.8"
|
png = "0.16.8"
|
||||||
|
rand = "0.8.4"
|
||||||
|
rayon = "1.5.1"
|
||||||
|
serde = { version = "1.0", features = ["derive"] }
|
||||||
|
serde_json = "1.0.64"
|
||||||
|
structopt = "0.3.22"
|
||||||
thiserror = "1.0.25"
|
thiserror = "1.0.25"
|
||||||
|
|
||||||
|
|
||||||
[[bench]]
|
[[bench]]
|
||||||
name = "matrices"
|
name = "matrices"
|
||||||
harness = false
|
harness = false
|
||||||
|
|
||||||
|
[profile.release]
|
||||||
|
debug = true
|
||||||
|
|||||||
119
rtchallenge/examples/balls.rs
Normal file
119
rtchallenge/examples/balls.rs
Normal file
@@ -0,0 +1,119 @@
|
|||||||
|
use std::time::Instant;
|
||||||
|
|
||||||
|
use anyhow::Result;
|
||||||
|
use structopt::StructOpt;
|
||||||
|
|
||||||
|
use rtchallenge::{
|
||||||
|
camera::{Camera, RenderStrategy},
|
||||||
|
float::consts::PI,
|
||||||
|
lights::PointLightBuilder,
|
||||||
|
materials::MaterialBuilder,
|
||||||
|
matrices::Matrix4x4,
|
||||||
|
shapes::{Geometry, ShapeBuilder},
|
||||||
|
transformations::view_transform,
|
||||||
|
tuples::Tuple,
|
||||||
|
world::WorldBuilder,
|
||||||
|
Float, WHITE,
|
||||||
|
};
|
||||||
|
|
||||||
|
/// Experimenting with balls.
|
||||||
|
#[derive(StructOpt, Debug)]
|
||||||
|
#[structopt(name = "balls")]
|
||||||
|
struct Opt {
|
||||||
|
#[structopt(long, default_value = "rayon")]
|
||||||
|
render_strategy: RenderStrategy,
|
||||||
|
/// Number of samples per pixel. 0 renders from the center of the pixel, 1 or more samples N
|
||||||
|
/// times randomly across the pixel.
|
||||||
|
#[structopt(short, long, default_value = "0")]
|
||||||
|
samples: usize,
|
||||||
|
}
|
||||||
|
|
||||||
|
fn main() -> Result<()> {
|
||||||
|
let start = Instant::now();
|
||||||
|
let opt = Opt::from_args();
|
||||||
|
let width = 2560;
|
||||||
|
let height = 1440;
|
||||||
|
|
||||||
|
let light1 = PointLightBuilder::default()
|
||||||
|
.position(Tuple::point(-5., 5., -5.))
|
||||||
|
.intensity(WHITE)
|
||||||
|
.build()?;
|
||||||
|
let light2 = PointLightBuilder::default()
|
||||||
|
.position(Tuple::point(5., 5., -5.))
|
||||||
|
.intensity([0.2, 0.2, 0.6])
|
||||||
|
.build()?;
|
||||||
|
let light3 = PointLightBuilder::default()
|
||||||
|
.position(Tuple::point(0., 2., -5.))
|
||||||
|
.intensity([0.2, 0.2, 0.1])
|
||||||
|
.build()?;
|
||||||
|
|
||||||
|
let ball_size = 0.5;
|
||||||
|
let num_per_axis = 3;
|
||||||
|
let center_to_center = 4. * ball_size;
|
||||||
|
let from = Tuple::point(
|
||||||
|
-5. * center_to_center,
|
||||||
|
5. * center_to_center,
|
||||||
|
-4. * center_to_center,
|
||||||
|
);
|
||||||
|
let to = Tuple::point(
|
||||||
|
num_per_axis as Float * center_to_center / 2.,
|
||||||
|
0., //num_per_axis as Float * center_to_center / 2.,
|
||||||
|
num_per_axis as Float * center_to_center / 2.,
|
||||||
|
);
|
||||||
|
let up = Tuple::point(0., 1., 0.);
|
||||||
|
let mut camera = Camera::new(width, height, PI / 6.);
|
||||||
|
camera.set_transform(view_transform(from, to, up));
|
||||||
|
camera.render_strategy = opt.render_strategy;
|
||||||
|
camera.samples_per_pixel = opt.samples;
|
||||||
|
|
||||||
|
let floor = ShapeBuilder::default()
|
||||||
|
.geometry(Geometry::Plane)
|
||||||
|
.transform(Matrix4x4::translation(0., -ball_size, 0.))
|
||||||
|
.material(
|
||||||
|
MaterialBuilder::default()
|
||||||
|
.color([0.2, 0.2, 0.2])
|
||||||
|
.specular(0.)
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.build()?;
|
||||||
|
let mut objects = vec![floor];
|
||||||
|
|
||||||
|
for z in 0..num_per_axis {
|
||||||
|
for y in 0..num_per_axis {
|
||||||
|
for x in 0..num_per_axis {
|
||||||
|
objects.push(
|
||||||
|
ShapeBuilder::default()
|
||||||
|
.transform(
|
||||||
|
Matrix4x4::translation(
|
||||||
|
x as Float * center_to_center,
|
||||||
|
y as Float * center_to_center,
|
||||||
|
z as Float * center_to_center,
|
||||||
|
) * Matrix4x4::scaling(ball_size, ball_size, ball_size),
|
||||||
|
)
|
||||||
|
.material(
|
||||||
|
MaterialBuilder::default()
|
||||||
|
.color([0.1, 1., 0.5])
|
||||||
|
.ambient(y as Float / 4.)
|
||||||
|
.diffuse(x as Float / 4.)
|
||||||
|
.specular(z as Float / 4.)
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.build()?,
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
let world = WorldBuilder::default()
|
||||||
|
.lights(vec![light1, light2, light3])
|
||||||
|
.objects(objects)
|
||||||
|
.build()?;
|
||||||
|
|
||||||
|
let image = camera.render(&world);
|
||||||
|
|
||||||
|
let path = "/tmp/balls.png";
|
||||||
|
println!("saving output to {}", path);
|
||||||
|
image.write_to_file(path)?;
|
||||||
|
println!("Render time {:.3} seconds", start.elapsed().as_secs_f32());
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
171
rtchallenge/examples/eoc10.rs
Normal file
171
rtchallenge/examples/eoc10.rs
Normal file
@@ -0,0 +1,171 @@
|
|||||||
|
use std::time::Instant;
|
||||||
|
|
||||||
|
use anyhow::Result;
|
||||||
|
use structopt::StructOpt;
|
||||||
|
|
||||||
|
use rtchallenge::prelude::*;
|
||||||
|
|
||||||
|
use rtchallenge::{
|
||||||
|
camera::RenderStrategy,
|
||||||
|
float::consts::PI,
|
||||||
|
patterns::{test_pattern, BLACK_PAT, WHITE_PAT},
|
||||||
|
WHITE,
|
||||||
|
};
|
||||||
|
|
||||||
|
/// End of chapter 10 challenge.
|
||||||
|
#[derive(StructOpt, Debug)]
|
||||||
|
#[structopt(name = "eoc10")]
|
||||||
|
struct Opt {
|
||||||
|
/// Strategy for casting rays into image.
|
||||||
|
#[structopt(long, default_value = "rayon")]
|
||||||
|
render_strategy: RenderStrategy,
|
||||||
|
/// Number of samples per pixel. 0 renders from the center of the pixel, 1 or more samples N
|
||||||
|
/// times randomly across the pixel.
|
||||||
|
#[structopt(short, long, default_value = "0")]
|
||||||
|
samples: usize,
|
||||||
|
/// Rendered image width in pixels.
|
||||||
|
#[structopt(short, long, default_value = "2560")]
|
||||||
|
width: usize,
|
||||||
|
/// Rendered image height in pixels.
|
||||||
|
#[structopt(short, long, default_value = "1440")]
|
||||||
|
height: usize,
|
||||||
|
}
|
||||||
|
|
||||||
|
fn main() -> Result<()> {
|
||||||
|
let start = Instant::now();
|
||||||
|
let opt = Opt::from_args();
|
||||||
|
|
||||||
|
let light1 = PointLightBuilder::default()
|
||||||
|
.position(point(-5., 5., -5.))
|
||||||
|
.intensity(WHITE)
|
||||||
|
.build()?;
|
||||||
|
let light2 = PointLightBuilder::default()
|
||||||
|
.position(point(5., 5., -5.))
|
||||||
|
.intensity([0.2, 0.2, 0.6])
|
||||||
|
.build()?;
|
||||||
|
let light3 = PointLightBuilder::default()
|
||||||
|
.position(point(0., 2., -5.))
|
||||||
|
.intensity([0.2, 0.2, 0.1])
|
||||||
|
.build()?;
|
||||||
|
|
||||||
|
let from = point(2., 2., -10.);
|
||||||
|
let to = point(2., 1., 0.);
|
||||||
|
let up = point(0., 1., 0.);
|
||||||
|
let camera = CameraBuilder::default()
|
||||||
|
.hsize(opt.width)
|
||||||
|
.vsize(opt.height)
|
||||||
|
.field_of_view(PI / 4.)
|
||||||
|
.transform(view_transform(from, to, up))
|
||||||
|
.render_strategy(opt.render_strategy)
|
||||||
|
.samples_per_pixel(opt.samples)
|
||||||
|
.build()?;
|
||||||
|
|
||||||
|
let floor = plane()
|
||||||
|
.material(
|
||||||
|
MaterialBuilder::default()
|
||||||
|
.color(
|
||||||
|
checkers_pattern(WHITE_PAT, BLACK_PAT)
|
||||||
|
.transform(translation(1., 0., 0.) * scaling(2., 2., 2.))
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.specular(0.)
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.build()?;
|
||||||
|
let sphere_size = scaling(0.5, 0.5, 0.5);
|
||||||
|
|
||||||
|
let x1y1 = sphere()
|
||||||
|
.transform(translation(1., 1., 0.) * sphere_size)
|
||||||
|
.material(
|
||||||
|
MaterialBuilder::default()
|
||||||
|
.color(
|
||||||
|
gradient_pattern([0., 0., 1.].into(), [1., 1., 0.].into())
|
||||||
|
.transform(scaling(2., 1., 1.) * translation(-0.5, 0., 0.))
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.diffuse(0.7)
|
||||||
|
.specular(0.3)
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.build()?;
|
||||||
|
|
||||||
|
let x2y1 = sphere()
|
||||||
|
.transform(translation(2., 1., 0.) * sphere_size)
|
||||||
|
.material(
|
||||||
|
MaterialBuilder::default()
|
||||||
|
.color(stripe_pattern(WHITE_PAT, BLACK_PAT).build()?)
|
||||||
|
.diffuse(0.7)
|
||||||
|
.specular(0.3)
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.build()?;
|
||||||
|
|
||||||
|
let x3y1 = sphere()
|
||||||
|
.transform(translation(3., 1., 0.) * sphere_size)
|
||||||
|
.material(
|
||||||
|
MaterialBuilder::default()
|
||||||
|
.color(
|
||||||
|
stripe_pattern(WHITE_PAT, BLACK_PAT)
|
||||||
|
.transform(scaling(0.2, 1., 1.))
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.diffuse(0.7)
|
||||||
|
.specular(0.0)
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.build()?;
|
||||||
|
|
||||||
|
let x1y2 = sphere()
|
||||||
|
.transform(translation(1., 2., 0.) * sphere_size)
|
||||||
|
.material(
|
||||||
|
MaterialBuilder::default()
|
||||||
|
.color(test_pattern().build()?)
|
||||||
|
.diffuse(0.7)
|
||||||
|
.specular(0.3)
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.build()?;
|
||||||
|
|
||||||
|
let x2y2 = sphere()
|
||||||
|
.transform(translation(2., 2., 0.) * sphere_size)
|
||||||
|
.material(
|
||||||
|
MaterialBuilder::default()
|
||||||
|
.color(
|
||||||
|
ring_pattern(WHITE_PAT, BLACK_PAT)
|
||||||
|
.transform(scaling(0.2, 0.2, 0.2))
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.diffuse(0.7)
|
||||||
|
.specular(0.3)
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.build()?;
|
||||||
|
|
||||||
|
let x3y2 = sphere()
|
||||||
|
.transform(translation(3., 2., 0.) * sphere_size)
|
||||||
|
.material(
|
||||||
|
MaterialBuilder::default()
|
||||||
|
.color(
|
||||||
|
checkers_pattern(WHITE_PAT, BLACK_PAT)
|
||||||
|
.transform(scaling(0.5, 0.5, 0.5))
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.diffuse(0.7)
|
||||||
|
.specular(0.3)
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.build()?;
|
||||||
|
|
||||||
|
let world = WorldBuilder::default()
|
||||||
|
.lights(vec![light1, light2, light3])
|
||||||
|
.objects(vec![floor, x1y1, x2y1, x3y1, x1y2, x2y2, x3y2])
|
||||||
|
.build()?;
|
||||||
|
|
||||||
|
let image = camera.render(&world);
|
||||||
|
|
||||||
|
let path = "/tmp/eoc10.png";
|
||||||
|
println!("saving output to {}", path);
|
||||||
|
image.write_to_file(path)?;
|
||||||
|
println!("Render time {:.3} seconds", start.elapsed().as_secs_f32());
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
218
rtchallenge/examples/eoc11.rs
Normal file
218
rtchallenge/examples/eoc11.rs
Normal file
@@ -0,0 +1,218 @@
|
|||||||
|
use std::time::Instant;
|
||||||
|
|
||||||
|
use anyhow::Result;
|
||||||
|
use structopt::StructOpt;
|
||||||
|
|
||||||
|
use rtchallenge::prelude::*;
|
||||||
|
|
||||||
|
use rtchallenge::{
|
||||||
|
camera::RenderStrategy,
|
||||||
|
float::consts::PI,
|
||||||
|
patterns::{test_pattern, BLACK_PAT, WHITE_PAT},
|
||||||
|
WHITE,
|
||||||
|
};
|
||||||
|
|
||||||
|
/// End of chapter 11 challenge.
|
||||||
|
#[derive(StructOpt, Debug)]
|
||||||
|
#[structopt(name = "eoc11")]
|
||||||
|
struct Opt {
|
||||||
|
/// Strategy for casting rays into image.
|
||||||
|
#[structopt(long, default_value = "rayon")]
|
||||||
|
render_strategy: RenderStrategy,
|
||||||
|
/// Number of samples per pixel. 0 renders from the center of the pixel, 1 or more samples N
|
||||||
|
/// times randomly across the pixel.
|
||||||
|
#[structopt(short, long, default_value = "0")]
|
||||||
|
samples: usize,
|
||||||
|
/// Rendered image width in pixels.
|
||||||
|
#[structopt(short, long, default_value = "2560")]
|
||||||
|
width: usize,
|
||||||
|
/// Rendered image height in pixels.
|
||||||
|
#[structopt(short, long, default_value = "1440")]
|
||||||
|
height: usize,
|
||||||
|
}
|
||||||
|
|
||||||
|
fn main() -> Result<()> {
|
||||||
|
let start = Instant::now();
|
||||||
|
let opt = Opt::from_args();
|
||||||
|
|
||||||
|
let light1 = PointLightBuilder::default()
|
||||||
|
.position(point(-5., 5., -5.))
|
||||||
|
.intensity(WHITE)
|
||||||
|
.build()?;
|
||||||
|
let light2 = PointLightBuilder::default()
|
||||||
|
.position(point(5., 5., -5.))
|
||||||
|
.intensity([0.2, 0.2, 0.6])
|
||||||
|
.build()?;
|
||||||
|
let light3 = PointLightBuilder::default()
|
||||||
|
.position(point(0., 2., -5.))
|
||||||
|
.intensity([0.2, 0.2, 0.1])
|
||||||
|
.build()?;
|
||||||
|
|
||||||
|
let from = point(2., 8., -10.);
|
||||||
|
let to = point(2., 1., -1.);
|
||||||
|
let up = point(0., 1., 0.);
|
||||||
|
let camera = CameraBuilder::default()
|
||||||
|
.hsize(opt.width)
|
||||||
|
.vsize(opt.height)
|
||||||
|
.field_of_view(PI / 6.)
|
||||||
|
.transform(view_transform(from, to, up))
|
||||||
|
.render_strategy(opt.render_strategy)
|
||||||
|
.samples_per_pixel(opt.samples)
|
||||||
|
.build()?;
|
||||||
|
|
||||||
|
let floor = plane()
|
||||||
|
.material(
|
||||||
|
MaterialBuilder::default()
|
||||||
|
.color(
|
||||||
|
checkers_pattern(
|
||||||
|
ring_pattern([0.8, 0.8, 0.2].into(), [0.8, 0.2, 0.8].into())
|
||||||
|
.transform(scaling(1. / 8., 1. / 8., 1. / 8.))
|
||||||
|
.build()?,
|
||||||
|
ring_pattern([0.2, 0.8, 0.2].into(), [0.8, 0.2, 0.2].into())
|
||||||
|
.transform(scaling(1. / 4., 1. / 4., 1. / 4.))
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.transform(translation(0., 1., 0.) * scaling(2., 2., 2.))
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.specular(0.)
|
||||||
|
.reflective(0.5)
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.build()?;
|
||||||
|
let sphere_size = scaling(0.5, 0.5, 0.5);
|
||||||
|
|
||||||
|
let x1y1 = sphere()
|
||||||
|
.transform(translation(1., 1., 0.) * sphere_size)
|
||||||
|
.material(
|
||||||
|
MaterialBuilder::default()
|
||||||
|
.color(
|
||||||
|
gradient_pattern([0., 0., 1.].into(), [1., 1., 0.].into())
|
||||||
|
.transform(scaling(2., 1., 1.) * translation(-0.5, 0., 0.))
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.diffuse(0.7)
|
||||||
|
.specular(0.3)
|
||||||
|
.reflective(0.5)
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.build()?;
|
||||||
|
|
||||||
|
let x2y1 = sphere()
|
||||||
|
.transform(translation(2., 1., 0.) * sphere_size)
|
||||||
|
.material(
|
||||||
|
MaterialBuilder::default()
|
||||||
|
.color(stripe_pattern(WHITE_PAT, BLACK_PAT).build()?)
|
||||||
|
.diffuse(0.7)
|
||||||
|
.specular(0.3)
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.build()?;
|
||||||
|
|
||||||
|
let x3y1 = sphere()
|
||||||
|
.transform(translation(3., 1., 0.) * sphere_size)
|
||||||
|
.material(
|
||||||
|
MaterialBuilder::default()
|
||||||
|
.color(
|
||||||
|
stripe_pattern(WHITE_PAT, BLACK_PAT)
|
||||||
|
.transform(scaling(0.2, 1., 1.))
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.diffuse(0.7)
|
||||||
|
.specular(0.0)
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.build()?;
|
||||||
|
|
||||||
|
let x1y2 = sphere()
|
||||||
|
.transform(translation(1., 2., 0.) * sphere_size)
|
||||||
|
.material(
|
||||||
|
MaterialBuilder::default()
|
||||||
|
.color(test_pattern().build()?)
|
||||||
|
.diffuse(0.7)
|
||||||
|
.specular(0.3)
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.build()?;
|
||||||
|
|
||||||
|
let x2y2 = sphere()
|
||||||
|
.transform(translation(2., 2., 0.) * sphere_size)
|
||||||
|
.material(
|
||||||
|
MaterialBuilder::default()
|
||||||
|
.color(
|
||||||
|
ring_pattern(WHITE_PAT, BLACK_PAT)
|
||||||
|
.transform(scaling(0.2, 0.2, 0.2))
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.diffuse(0.7)
|
||||||
|
.specular(0.3)
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.build()?;
|
||||||
|
|
||||||
|
let x3y2 = sphere()
|
||||||
|
.transform(translation(3., 2., 0.) * sphere_size)
|
||||||
|
.material(
|
||||||
|
MaterialBuilder::default()
|
||||||
|
.color(
|
||||||
|
checkers_pattern(WHITE_PAT, BLACK_PAT)
|
||||||
|
.transform(scaling(0.5, 0.5, 0.5))
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.diffuse(0.7)
|
||||||
|
.specular(0.3)
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.build()?;
|
||||||
|
|
||||||
|
let x1y2z1 = sphere()
|
||||||
|
.transform(translation(1., 2., -1.) * sphere_size)
|
||||||
|
.material(
|
||||||
|
MaterialBuilder::default()
|
||||||
|
.color([0.8, 0.2, 0.2])
|
||||||
|
.diffuse(0.7)
|
||||||
|
.specular(0.3)
|
||||||
|
.transparency(0.5)
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.build()?;
|
||||||
|
|
||||||
|
let x2y2z1 = sphere()
|
||||||
|
.transform(translation(2., 2., -1.) * sphere_size)
|
||||||
|
.material(
|
||||||
|
MaterialBuilder::default()
|
||||||
|
.color([0.8, 0.2, 0.2])
|
||||||
|
.diffuse(0.7)
|
||||||
|
.specular(0.3)
|
||||||
|
.transparency(0.9)
|
||||||
|
.refractive_index(1.5)
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.build()?;
|
||||||
|
|
||||||
|
let x2y4z1 = sphere()
|
||||||
|
.transform(translation(2., 4., 0.) * sphere_size)
|
||||||
|
.material(
|
||||||
|
MaterialBuilder::default()
|
||||||
|
.color([0.2, 0.2, 0.8])
|
||||||
|
.diffuse(0.7)
|
||||||
|
.specular(0.3)
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.build()?;
|
||||||
|
|
||||||
|
let world = WorldBuilder::default()
|
||||||
|
.lights(vec![light1, light2, light3])
|
||||||
|
.objects(vec![
|
||||||
|
floor, x1y1, x2y1, x3y1, x1y2, x2y2, x3y2, x1y2z1, x2y2z1, x2y4z1,
|
||||||
|
])
|
||||||
|
.build()?;
|
||||||
|
|
||||||
|
let image = camera.render(&world);
|
||||||
|
|
||||||
|
let path = "/tmp/eoc11.png";
|
||||||
|
println!("saving output to {}", path);
|
||||||
|
image.write_to_file(path)?;
|
||||||
|
println!("Render time {:.3} seconds", start.elapsed().as_secs_f32());
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
232
rtchallenge/examples/eoc12.rs
Normal file
232
rtchallenge/examples/eoc12.rs
Normal file
@@ -0,0 +1,232 @@
|
|||||||
|
use std::time::Instant;
|
||||||
|
|
||||||
|
use anyhow::Result;
|
||||||
|
use structopt::StructOpt;
|
||||||
|
|
||||||
|
use rtchallenge::prelude::*;
|
||||||
|
|
||||||
|
use rtchallenge::{
|
||||||
|
camera::RenderStrategy,
|
||||||
|
float::consts::PI,
|
||||||
|
patterns::{test_pattern, BLACK_PAT, WHITE_PAT},
|
||||||
|
WHITE,
|
||||||
|
};
|
||||||
|
|
||||||
|
/// End of chapter 12 challenge.
|
||||||
|
#[derive(StructOpt, Debug)]
|
||||||
|
#[structopt(name = "eoc12")]
|
||||||
|
struct Opt {
|
||||||
|
/// Strategy for casting rays into image.
|
||||||
|
#[structopt(long, default_value = "rayon")]
|
||||||
|
render_strategy: RenderStrategy,
|
||||||
|
/// Number of samples per pixel. 0 renders from the center of the pixel, 1 or more samples N
|
||||||
|
/// times randomly across the pixel.
|
||||||
|
#[structopt(short, long, default_value = "0")]
|
||||||
|
samples: usize,
|
||||||
|
/// Rendered image width in pixels.
|
||||||
|
#[structopt(short, long, default_value = "2560")]
|
||||||
|
width: usize,
|
||||||
|
/// Rendered image height in pixels.
|
||||||
|
#[structopt(short, long, default_value = "1440")]
|
||||||
|
height: usize,
|
||||||
|
}
|
||||||
|
|
||||||
|
fn main() -> Result<()> {
|
||||||
|
let start = Instant::now();
|
||||||
|
let opt = Opt::from_args();
|
||||||
|
|
||||||
|
let light1 = PointLightBuilder::default()
|
||||||
|
.position(point(-5., 5., -5.))
|
||||||
|
.intensity(WHITE)
|
||||||
|
.build()?;
|
||||||
|
let light2 = PointLightBuilder::default()
|
||||||
|
.position(point(5., 5., -5.))
|
||||||
|
.intensity([0.2, 0.2, 0.6])
|
||||||
|
.build()?;
|
||||||
|
let light3 = PointLightBuilder::default()
|
||||||
|
.position(point(0., 2., -5.))
|
||||||
|
.intensity([0.2, 0.2, 0.1])
|
||||||
|
.build()?;
|
||||||
|
|
||||||
|
let from = point(2., 8., -10.);
|
||||||
|
let to = point(2., 1., -1.);
|
||||||
|
let up = point(0., 1., 0.);
|
||||||
|
let camera = CameraBuilder::default()
|
||||||
|
.hsize(opt.width)
|
||||||
|
.vsize(opt.height)
|
||||||
|
.field_of_view(PI / 6.)
|
||||||
|
.transform(view_transform(from, to, up))
|
||||||
|
.render_strategy(opt.render_strategy)
|
||||||
|
.samples_per_pixel(opt.samples)
|
||||||
|
.build()?;
|
||||||
|
|
||||||
|
let floor = plane()
|
||||||
|
.material(
|
||||||
|
MaterialBuilder::default()
|
||||||
|
.color(
|
||||||
|
checkers_pattern(
|
||||||
|
ring_pattern([0.8, 0.8, 0.2].into(), [0.8, 0.2, 0.8].into())
|
||||||
|
.transform(scaling(1. / 8., 1. / 8., 1. / 8.))
|
||||||
|
.build()?,
|
||||||
|
ring_pattern([0.2, 0.8, 0.2].into(), [0.8, 0.2, 0.2].into())
|
||||||
|
.transform(scaling(1. / 4., 1. / 4., 1. / 4.))
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.transform(translation(0., 1., 0.) * scaling(2., 2., 2.))
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.specular(0.)
|
||||||
|
.reflective(0.5)
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.build()?;
|
||||||
|
let sphere_size = scaling(0.5, 0.5, 0.5);
|
||||||
|
|
||||||
|
let x1y1 = sphere()
|
||||||
|
.transform(translation(1., 1., 0.) * sphere_size)
|
||||||
|
.material(
|
||||||
|
MaterialBuilder::default()
|
||||||
|
.color(
|
||||||
|
gradient_pattern([0., 0., 1.].into(), [1., 1., 0.].into())
|
||||||
|
.transform(scaling(2., 1., 1.) * translation(-0.5, 0., 0.))
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.diffuse(0.7)
|
||||||
|
.specular(0.3)
|
||||||
|
.reflective(0.5)
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.build()?;
|
||||||
|
|
||||||
|
let x2y1 = sphere()
|
||||||
|
.transform(translation(2., 1., 0.) * sphere_size)
|
||||||
|
.material(
|
||||||
|
MaterialBuilder::default()
|
||||||
|
.color(stripe_pattern(WHITE_PAT, BLACK_PAT).build()?)
|
||||||
|
.diffuse(0.7)
|
||||||
|
.specular(0.3)
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.build()?;
|
||||||
|
|
||||||
|
let x3y1 = sphere()
|
||||||
|
.transform(translation(3., 1., 0.) * sphere_size)
|
||||||
|
.material(
|
||||||
|
MaterialBuilder::default()
|
||||||
|
.color(
|
||||||
|
stripe_pattern(WHITE_PAT, BLACK_PAT)
|
||||||
|
.transform(scaling(0.2, 1., 1.))
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.diffuse(0.7)
|
||||||
|
.specular(0.0)
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.build()?;
|
||||||
|
|
||||||
|
let x1y2 = sphere()
|
||||||
|
.transform(translation(1., 2., 0.) * sphere_size)
|
||||||
|
.material(
|
||||||
|
MaterialBuilder::default()
|
||||||
|
.color(test_pattern().build()?)
|
||||||
|
.diffuse(0.7)
|
||||||
|
.specular(0.3)
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.build()?;
|
||||||
|
|
||||||
|
let x2y2 = sphere()
|
||||||
|
.transform(translation(2., 2., 0.) * sphere_size)
|
||||||
|
.material(
|
||||||
|
MaterialBuilder::default()
|
||||||
|
.color(
|
||||||
|
ring_pattern(WHITE_PAT, BLACK_PAT)
|
||||||
|
.transform(scaling(0.2, 0.2, 0.2))
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.diffuse(0.7)
|
||||||
|
.specular(0.3)
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.build()?;
|
||||||
|
|
||||||
|
let x3y2 = sphere()
|
||||||
|
.transform(translation(3., 2., 0.) * sphere_size)
|
||||||
|
.material(
|
||||||
|
MaterialBuilder::default()
|
||||||
|
.color(
|
||||||
|
checkers_pattern(WHITE_PAT, BLACK_PAT)
|
||||||
|
.transform(scaling(0.5, 0.5, 0.5))
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.diffuse(0.7)
|
||||||
|
.specular(0.3)
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.build()?;
|
||||||
|
|
||||||
|
let x1y2z1 = sphere()
|
||||||
|
.transform(translation(1., 2., -1.) * sphere_size)
|
||||||
|
.material(
|
||||||
|
MaterialBuilder::default()
|
||||||
|
.color([0.8, 0.2, 0.2])
|
||||||
|
.diffuse(0.7)
|
||||||
|
.specular(0.3)
|
||||||
|
.transparency(0.5)
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.build()?;
|
||||||
|
|
||||||
|
let x2y2z1 = sphere()
|
||||||
|
.transform(translation(2., 2., -1.) * sphere_size)
|
||||||
|
.material(
|
||||||
|
MaterialBuilder::default()
|
||||||
|
.color([0.8, 0.2, 0.2])
|
||||||
|
.diffuse(0.7)
|
||||||
|
.specular(0.3)
|
||||||
|
.transparency(0.9)
|
||||||
|
.refractive_index(1.5)
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.build()?;
|
||||||
|
|
||||||
|
let x3y2z1 = cube()
|
||||||
|
.transform(
|
||||||
|
translation(3., 2., -1.) * (sphere_size * scaling(0.5, 0.5, 0.5)) * rotation_y(PI / 4.),
|
||||||
|
)
|
||||||
|
.material(
|
||||||
|
MaterialBuilder::default()
|
||||||
|
.color([0.8, 0.8, 0.2])
|
||||||
|
.diffuse(0.7)
|
||||||
|
.specular(0.3)
|
||||||
|
.reflective(0.5)
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.build()?;
|
||||||
|
|
||||||
|
let x2y4z1 = sphere()
|
||||||
|
.transform(translation(2., 4., 0.) * sphere_size)
|
||||||
|
.material(
|
||||||
|
MaterialBuilder::default()
|
||||||
|
.color([0.2, 0.2, 0.8])
|
||||||
|
.diffuse(0.7)
|
||||||
|
.specular(0.3)
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.build()?;
|
||||||
|
|
||||||
|
let world = WorldBuilder::default()
|
||||||
|
.lights(vec![light1, light2, light3])
|
||||||
|
.objects(vec![
|
||||||
|
floor, x1y1, x2y1, x3y1, x1y2, x2y2, x3y2, x1y2z1, x2y2z1, x2y4z1, x3y2z1,
|
||||||
|
])
|
||||||
|
.build()?;
|
||||||
|
|
||||||
|
let image = camera.render(&world);
|
||||||
|
|
||||||
|
let path = "/tmp/eoc12.png";
|
||||||
|
println!("saving output to {}", path);
|
||||||
|
image.write_to_file(path)?;
|
||||||
|
println!("Render time {:.3} seconds", start.elapsed().as_secs_f32());
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
232
rtchallenge/examples/eoc14.rs
Normal file
232
rtchallenge/examples/eoc14.rs
Normal file
@@ -0,0 +1,232 @@
|
|||||||
|
use std::time::Instant;
|
||||||
|
|
||||||
|
use anyhow::Result;
|
||||||
|
use structopt::StructOpt;
|
||||||
|
|
||||||
|
use rtchallenge::prelude::*;
|
||||||
|
|
||||||
|
use rtchallenge::{
|
||||||
|
camera::RenderStrategy,
|
||||||
|
float::consts::PI,
|
||||||
|
patterns::{test_pattern, BLACK_PAT, WHITE_PAT},
|
||||||
|
WHITE,
|
||||||
|
};
|
||||||
|
|
||||||
|
/// End of chapter 14 challenge.
|
||||||
|
#[derive(StructOpt, Debug)]
|
||||||
|
#[structopt(name = "eoc14")]
|
||||||
|
struct Opt {
|
||||||
|
/// Strategy for casting rays into image.
|
||||||
|
#[structopt(long, default_value = "rayon")]
|
||||||
|
render_strategy: RenderStrategy,
|
||||||
|
/// Number of samples per pixel. 0 renders from the center of the pixel, 1 or more samples N
|
||||||
|
/// times randomly across the pixel.
|
||||||
|
#[structopt(short, long, default_value = "0")]
|
||||||
|
samples: usize,
|
||||||
|
/// Rendered image width in pixels.
|
||||||
|
#[structopt(short, long, default_value = "2560")]
|
||||||
|
width: usize,
|
||||||
|
/// Rendered image height in pixels.
|
||||||
|
#[structopt(short, long, default_value = "1440")]
|
||||||
|
height: usize,
|
||||||
|
}
|
||||||
|
|
||||||
|
fn main() -> Result<()> {
|
||||||
|
let start = Instant::now();
|
||||||
|
let opt = Opt::from_args();
|
||||||
|
|
||||||
|
let light1 = PointLightBuilder::default()
|
||||||
|
.position(point(-5., 5., -5.))
|
||||||
|
.intensity(WHITE)
|
||||||
|
.build()?;
|
||||||
|
let light2 = PointLightBuilder::default()
|
||||||
|
.position(point(5., 5., -5.))
|
||||||
|
.intensity([0.2, 0.2, 0.6])
|
||||||
|
.build()?;
|
||||||
|
let light3 = PointLightBuilder::default()
|
||||||
|
.position(point(0., 2., -5.))
|
||||||
|
.intensity([0.2, 0.2, 0.1])
|
||||||
|
.build()?;
|
||||||
|
|
||||||
|
let from = point(2., 8., -10.);
|
||||||
|
let to = point(2., 2., -1.);
|
||||||
|
let up = point(0., 1., 0.);
|
||||||
|
let camera = CameraBuilder::default()
|
||||||
|
.hsize(opt.width)
|
||||||
|
.vsize(opt.height)
|
||||||
|
.field_of_view(PI / 12.)
|
||||||
|
.transform(view_transform(from, to, up))
|
||||||
|
.render_strategy(opt.render_strategy)
|
||||||
|
.samples_per_pixel(opt.samples)
|
||||||
|
.build()?;
|
||||||
|
|
||||||
|
let floor = plane()
|
||||||
|
.material(
|
||||||
|
MaterialBuilder::default()
|
||||||
|
.color(
|
||||||
|
checkers_pattern(
|
||||||
|
ring_pattern([0.8, 0.8, 0.2].into(), [0.8, 0.2, 0.8].into())
|
||||||
|
.transform(scaling(1. / 8., 1. / 8., 1. / 8.))
|
||||||
|
.build()?,
|
||||||
|
ring_pattern([0.2, 0.8, 0.2].into(), [0.8, 0.2, 0.2].into())
|
||||||
|
.transform(scaling(1. / 4., 1. / 4., 1. / 4.))
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.transform(translation(0., 1., 0.) * scaling(2., 2., 2.))
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.specular(0.)
|
||||||
|
.reflective(0.5)
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.build()?;
|
||||||
|
let sphere_size = scaling(0.5, 0.5, 0.5);
|
||||||
|
|
||||||
|
let x1y1 = sphere()
|
||||||
|
.transform(translation(1., 1., 0.) * sphere_size)
|
||||||
|
.material(
|
||||||
|
MaterialBuilder::default()
|
||||||
|
.color(
|
||||||
|
gradient_pattern([0., 0., 1.].into(), [1., 1., 0.].into())
|
||||||
|
.transform(scaling(2., 1., 1.) * translation(-0.5, 0., 0.))
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.diffuse(0.7)
|
||||||
|
.specular(0.3)
|
||||||
|
.reflective(0.5)
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.build()?;
|
||||||
|
|
||||||
|
let x2y1 = sphere()
|
||||||
|
.transform(translation(2., 1., 0.) * sphere_size)
|
||||||
|
.material(
|
||||||
|
MaterialBuilder::default()
|
||||||
|
.color(stripe_pattern(WHITE_PAT, BLACK_PAT).build()?)
|
||||||
|
.diffuse(0.7)
|
||||||
|
.specular(0.3)
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.build()?;
|
||||||
|
|
||||||
|
let x3y1 = sphere()
|
||||||
|
.transform(translation(3., 1., 0.) * sphere_size)
|
||||||
|
.material(
|
||||||
|
MaterialBuilder::default()
|
||||||
|
.color(
|
||||||
|
stripe_pattern(WHITE_PAT, BLACK_PAT)
|
||||||
|
.transform(scaling(0.2, 1., 1.))
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.diffuse(0.7)
|
||||||
|
.specular(0.0)
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.build()?;
|
||||||
|
|
||||||
|
let x1y2 = sphere()
|
||||||
|
.transform(translation(1., 2., 0.) * sphere_size)
|
||||||
|
.material(
|
||||||
|
MaterialBuilder::default()
|
||||||
|
.color(test_pattern().build()?)
|
||||||
|
.diffuse(0.7)
|
||||||
|
.specular(0.3)
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.build()?;
|
||||||
|
|
||||||
|
let x2y2 = sphere()
|
||||||
|
.transform(translation(2., 2., 0.) * sphere_size)
|
||||||
|
.material(
|
||||||
|
MaterialBuilder::default()
|
||||||
|
.color(
|
||||||
|
ring_pattern(WHITE_PAT, BLACK_PAT)
|
||||||
|
.transform(scaling(0.2, 0.2, 0.2))
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.diffuse(0.7)
|
||||||
|
.specular(0.3)
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.build()?;
|
||||||
|
|
||||||
|
let x3y2 = sphere()
|
||||||
|
.transform(translation(3., 2., 0.) * sphere_size)
|
||||||
|
.material(
|
||||||
|
MaterialBuilder::default()
|
||||||
|
.color(
|
||||||
|
checkers_pattern(WHITE_PAT, BLACK_PAT)
|
||||||
|
.transform(scaling(0.5, 0.5, 0.5))
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.diffuse(0.7)
|
||||||
|
.specular(0.3)
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.build()?;
|
||||||
|
|
||||||
|
let x1y2z1 = sphere()
|
||||||
|
.transform(translation(1., 2., -1.) * sphere_size)
|
||||||
|
.material(
|
||||||
|
MaterialBuilder::default()
|
||||||
|
.color([0.8, 0.2, 0.2])
|
||||||
|
.diffuse(0.7)
|
||||||
|
.specular(0.3)
|
||||||
|
.transparency(0.5)
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.build()?;
|
||||||
|
|
||||||
|
let x2y2z1 = sphere()
|
||||||
|
.transform(translation(2., 2., -1.) * sphere_size)
|
||||||
|
.material(
|
||||||
|
MaterialBuilder::default()
|
||||||
|
.color([0.8, 0.2, 0.2])
|
||||||
|
.diffuse(0.7)
|
||||||
|
.specular(0.3)
|
||||||
|
.transparency(0.9)
|
||||||
|
.refractive_index(1.5)
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.build()?;
|
||||||
|
|
||||||
|
let x3y2z1 = cube()
|
||||||
|
.transform(
|
||||||
|
translation(3., 2., -1.) * (sphere_size * scaling(0.5, 0.5, 0.5)) * rotation_y(PI / 4.),
|
||||||
|
)
|
||||||
|
.material(
|
||||||
|
MaterialBuilder::default()
|
||||||
|
.color([0.8, 0.8, 0.2])
|
||||||
|
.diffuse(0.7)
|
||||||
|
.specular(0.3)
|
||||||
|
.reflective(0.5)
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.build()?;
|
||||||
|
|
||||||
|
let x2y4z1 = sphere()
|
||||||
|
.transform(translation(2., 4., 0.) * sphere_size)
|
||||||
|
.material(
|
||||||
|
MaterialBuilder::default()
|
||||||
|
.color([0.2, 0.2, 0.8])
|
||||||
|
.diffuse(0.7)
|
||||||
|
.specular(0.3)
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.build()?;
|
||||||
|
|
||||||
|
let world = WorldBuilder::default()
|
||||||
|
.lights(vec![light1, light2, light3])
|
||||||
|
.objects(vec![
|
||||||
|
floor, x1y1, x2y1, x3y1, x1y2, x2y2, x3y2, x1y2z1, x2y2z1, x2y4z1, x3y2z1,
|
||||||
|
])
|
||||||
|
.build()?;
|
||||||
|
|
||||||
|
let image = camera.render(&world);
|
||||||
|
|
||||||
|
let path = "/tmp/eoc14.png";
|
||||||
|
println!("saving output to {}", path);
|
||||||
|
image.write_to_file(path)?;
|
||||||
|
println!("Render time {:.3} seconds", start.elapsed().as_secs_f32());
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
@@ -3,6 +3,7 @@ use anyhow::{bail, Result};
|
|||||||
use rtchallenge::{
|
use rtchallenge::{
|
||||||
canvas::Canvas,
|
canvas::Canvas,
|
||||||
tuples::{Color, Tuple},
|
tuples::{Color, Tuple},
|
||||||
|
Float,
|
||||||
};
|
};
|
||||||
|
|
||||||
#[derive(Debug)]
|
#[derive(Debug)]
|
||||||
@@ -47,8 +48,8 @@ fn main() -> Result<()> {
|
|||||||
let bg = Color::new(0.2, 0.2, 0.2);
|
let bg = Color::new(0.2, 0.2, 0.2);
|
||||||
let mut c = Canvas::new(w, h, bg);
|
let mut c = Canvas::new(w, h, bg);
|
||||||
let mut i = 0;
|
let mut i = 0;
|
||||||
let w = w as f32;
|
let w = w as Float;
|
||||||
let h = h as f32;
|
let h = h as Float;
|
||||||
while p.position.y > 0. {
|
while p.position.y > 0. {
|
||||||
p = tick(&e, &p);
|
p = tick(&e, &p);
|
||||||
println!("tick {}: proj {:?}", i, p);
|
println!("tick {}: proj {:?}", i, p);
|
||||||
|
|||||||
@@ -1,11 +1,11 @@
|
|||||||
use std::f32::consts::PI;
|
|
||||||
|
|
||||||
use anyhow::Result;
|
use anyhow::Result;
|
||||||
|
|
||||||
use rtchallenge::{
|
use rtchallenge::{
|
||||||
canvas::Canvas,
|
canvas::Canvas,
|
||||||
|
float::consts::PI,
|
||||||
matrices::Matrix4x4,
|
matrices::Matrix4x4,
|
||||||
tuples::{Color, Tuple},
|
tuples::{Color, Tuple},
|
||||||
|
Float,
|
||||||
};
|
};
|
||||||
|
|
||||||
fn draw_dot(c: &mut Canvas, x: usize, y: usize) {
|
fn draw_dot(c: &mut Canvas, x: usize, y: usize) {
|
||||||
@@ -30,13 +30,13 @@ fn main() -> Result<()> {
|
|||||||
let rot_hour = Matrix4x4::rotation_z(-PI / 6.);
|
let rot_hour = Matrix4x4::rotation_z(-PI / 6.);
|
||||||
|
|
||||||
let mut p = t * p;
|
let mut p = t * p;
|
||||||
let w = w as f32;
|
let w = w as Float;
|
||||||
let h = h as f32;
|
let h = h as Float;
|
||||||
// The 'world' exists between -0.5 - 0.5 in X-Y plane.
|
// The 'world' exists between -0.5 - 0.5 in X-Y plane.
|
||||||
// To convert to screen space, we translate by 0.5, scale to canvas size,
|
// To convert to screen space, we translate by 0.5, scale to canvas size,
|
||||||
// and invert the Y-axis.
|
// and invert the Y-axis.
|
||||||
let world_to_screen =
|
let world_to_screen =
|
||||||
Matrix4x4::scaling(w as f32, -h as f32, 1.0) * Matrix4x4::translation(0.5, -0.5, 0.);
|
Matrix4x4::scaling(w as Float, -h as Float, 1.0) * Matrix4x4::translation(0.5, -0.5, 0.);
|
||||||
for _ in 0..12 {
|
for _ in 0..12 {
|
||||||
let canvas_pixel = world_to_screen * p;
|
let canvas_pixel = world_to_screen * p;
|
||||||
draw_dot(&mut c, canvas_pixel.x as usize, canvas_pixel.y as usize);
|
draw_dot(&mut c, canvas_pixel.x as usize, canvas_pixel.y as usize);
|
||||||
|
|||||||
@@ -4,8 +4,9 @@ use rtchallenge::{
|
|||||||
canvas::Canvas,
|
canvas::Canvas,
|
||||||
matrices::Matrix4x4,
|
matrices::Matrix4x4,
|
||||||
rays::Ray,
|
rays::Ray,
|
||||||
spheres::{intersect, Sphere},
|
shapes::{intersect, Shape},
|
||||||
tuples::{Color, Tuple},
|
tuples::{Color, Tuple},
|
||||||
|
Float,
|
||||||
};
|
};
|
||||||
|
|
||||||
fn main() -> Result<()> {
|
fn main() -> Result<()> {
|
||||||
@@ -16,17 +17,18 @@ fn main() -> Result<()> {
|
|||||||
let ray_origin = Tuple::point(0., 0., -5.);
|
let ray_origin = Tuple::point(0., 0., -5.);
|
||||||
let wall_z = 10.;
|
let wall_z = 10.;
|
||||||
let wall_size = 7.;
|
let wall_size = 7.;
|
||||||
let pixel_size = wall_size / w as f32;
|
let pixel_size = wall_size / w as Float;
|
||||||
let half = wall_size / 2.;
|
let half = wall_size / 2.;
|
||||||
let color = Color::new(1., 0., 0.);
|
let color = Color::new(1., 0., 0.);
|
||||||
let mut shape = Sphere::default();
|
let mut shape = Shape::sphere();
|
||||||
shape.transform =
|
shape.set_transform(
|
||||||
Matrix4x4::shearing(1., 0., 0., 0., 0., 0.) * Matrix4x4::scaling(0.5, 1., 1.0);
|
Matrix4x4::shearing(1., 0., 0., 0., 0., 0.) * Matrix4x4::scaling(0.5, 1., 1.0),
|
||||||
|
);
|
||||||
|
|
||||||
for y in 0..h {
|
for y in 0..h {
|
||||||
let world_y = half - pixel_size * y as f32;
|
let world_y = half - pixel_size * y as Float;
|
||||||
for x in 0..w {
|
for x in 0..w {
|
||||||
let world_x = -half + pixel_size * x as f32;
|
let world_x = -half + pixel_size * x as Float;
|
||||||
let position = Tuple::point(world_x, world_y, wall_z);
|
let position = Tuple::point(world_x, world_y, wall_z);
|
||||||
let r = Ray::new(ray_origin, (position - ray_origin).normalize());
|
let r = Ray::new(ray_origin, (position - ray_origin).normalize());
|
||||||
let xs = intersect(&shape, &r);
|
let xs = intersect(&shape, &r);
|
||||||
|
|||||||
@@ -5,9 +5,9 @@ use rtchallenge::{
|
|||||||
lights::PointLight,
|
lights::PointLight,
|
||||||
materials::{lighting, Material},
|
materials::{lighting, Material},
|
||||||
rays::Ray,
|
rays::Ray,
|
||||||
spheres::{intersect, Sphere},
|
shapes::{intersect, Shape},
|
||||||
tuples::{Color, Tuple},
|
tuples::{Color, Tuple},
|
||||||
WHITE,
|
Float, WHITE,
|
||||||
};
|
};
|
||||||
|
|
||||||
fn main() -> Result<()> {
|
fn main() -> Result<()> {
|
||||||
@@ -18,11 +18,11 @@ fn main() -> Result<()> {
|
|||||||
let ray_origin = Tuple::point(0., 0., -5.);
|
let ray_origin = Tuple::point(0., 0., -5.);
|
||||||
let wall_z = 10.;
|
let wall_z = 10.;
|
||||||
let wall_size = 7.;
|
let wall_size = 7.;
|
||||||
let pixel_size = wall_size / w as f32;
|
let pixel_size = wall_size / w as Float;
|
||||||
let half = wall_size / 2.;
|
let half = wall_size / 2.;
|
||||||
let mut shape = Sphere::default();
|
let mut shape = Shape::sphere();
|
||||||
shape.material = Material {
|
shape.material = Material {
|
||||||
color: Color::new(1., 0.2, 1.),
|
color: Color::new(1., 0.2, 1.).into(),
|
||||||
specular: 0.5,
|
specular: 0.5,
|
||||||
diffuse: 0.7,
|
diffuse: 0.7,
|
||||||
shininess: 30.,
|
shininess: 30.,
|
||||||
@@ -31,11 +31,11 @@ fn main() -> Result<()> {
|
|||||||
let light_position = Tuple::point(-10., 10., -10.);
|
let light_position = Tuple::point(-10., 10., -10.);
|
||||||
let light_color = WHITE;
|
let light_color = WHITE;
|
||||||
let light = PointLight::new(light_position, light_color);
|
let light = PointLight::new(light_position, light_color);
|
||||||
|
let in_shadow = false;
|
||||||
for y in 0..h {
|
for y in 0..h {
|
||||||
let world_y = half - pixel_size * y as f32;
|
let world_y = half - pixel_size * y as Float;
|
||||||
for x in 0..w {
|
for x in 0..w {
|
||||||
let world_x = -half + pixel_size * x as f32;
|
let world_x = -half + pixel_size * x as Float;
|
||||||
let position = Tuple::point(world_x, world_y, wall_z);
|
let position = Tuple::point(world_x, world_y, wall_z);
|
||||||
let direction = (position - ray_origin).normalize();
|
let direction = (position - ray_origin).normalize();
|
||||||
let r = Ray::new(ray_origin, direction);
|
let r = Ray::new(ray_origin, direction);
|
||||||
@@ -44,7 +44,15 @@ fn main() -> Result<()> {
|
|||||||
let point = r.position(hit.t);
|
let point = r.position(hit.t);
|
||||||
let normal = hit.object.normal_at(point);
|
let normal = hit.object.normal_at(point);
|
||||||
let eye = -r.direction;
|
let eye = -r.direction;
|
||||||
let color = lighting(&hit.object.material, &light, point, eye, normal);
|
let color = lighting(
|
||||||
|
&hit.object.material,
|
||||||
|
&hit.object,
|
||||||
|
&light,
|
||||||
|
point,
|
||||||
|
eye,
|
||||||
|
normal,
|
||||||
|
in_shadow,
|
||||||
|
);
|
||||||
c.set(x, y, color);
|
c.set(x, y, color);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,23 +1,34 @@
|
|||||||
use std::{f32::consts::PI, time::Instant};
|
use std::time::Instant;
|
||||||
|
|
||||||
use anyhow::Result;
|
use anyhow::Result;
|
||||||
|
use structopt::StructOpt;
|
||||||
|
|
||||||
use rtchallenge::{
|
use rtchallenge::{
|
||||||
camera::Camera,
|
camera::{Camera, RenderStrategy},
|
||||||
|
float::consts::PI,
|
||||||
lights::PointLight,
|
lights::PointLight,
|
||||||
materials::Material,
|
materials::Material,
|
||||||
matrices::Matrix4x4,
|
matrices::Matrix4x4,
|
||||||
spheres::Sphere,
|
shapes::Shape,
|
||||||
transformations::view_transform,
|
transformations::view_transform,
|
||||||
tuples::{Color, Tuple},
|
tuples::{Color, Tuple},
|
||||||
world::World,
|
world::World,
|
||||||
WHITE,
|
WHITE,
|
||||||
};
|
};
|
||||||
|
|
||||||
|
/// End of chapter 7 challenge.
|
||||||
|
#[derive(StructOpt, Debug)]
|
||||||
|
#[structopt(name = "eoc7")]
|
||||||
|
struct Opt {
|
||||||
|
#[structopt(long, default_value = "rayon")]
|
||||||
|
render_strategy: RenderStrategy,
|
||||||
|
}
|
||||||
|
|
||||||
fn main() -> Result<()> {
|
fn main() -> Result<()> {
|
||||||
let start = Instant::now();
|
let start = Instant::now();
|
||||||
let width = 640;
|
let opt = Opt::from_args();
|
||||||
let height = 480;
|
let width = 1024;
|
||||||
|
let height = 1024;
|
||||||
let light_position = Tuple::point(-10., 10., -10.);
|
let light_position = Tuple::point(-10., 10., -10.);
|
||||||
let light_color = WHITE;
|
let light_color = WHITE;
|
||||||
let light = PointLight::new(light_position, light_color);
|
let light = PointLight::new(light_position, light_color);
|
||||||
@@ -25,60 +36,66 @@ fn main() -> Result<()> {
|
|||||||
let from = Tuple::point(0., 1.5, -5.);
|
let from = Tuple::point(0., 1.5, -5.);
|
||||||
let to = Tuple::point(0., 1., 0.);
|
let to = Tuple::point(0., 1., 0.);
|
||||||
let up = Tuple::point(0., 1., 0.);
|
let up = Tuple::point(0., 1., 0.);
|
||||||
camera.transform = view_transform(from, to, up);
|
camera.set_transform(view_transform(from, to, up));
|
||||||
|
camera.render_strategy = opt.render_strategy;
|
||||||
|
|
||||||
let mut floor = Sphere::default();
|
let mut floor = Shape::sphere();
|
||||||
floor.transform = Matrix4x4::scaling(10., 0.01, 10.);
|
floor.set_transform(Matrix4x4::scaling(10., 0.01, 10.));
|
||||||
floor.material = Material {
|
floor.material = Material {
|
||||||
color: Color::new(1., 0.9, 0.9),
|
color: Color::new(1., 0.9, 0.9).into(),
|
||||||
specular: 0.,
|
specular: 0.,
|
||||||
..Material::default()
|
..Material::default()
|
||||||
};
|
};
|
||||||
|
|
||||||
let mut left_wall = Sphere::default();
|
let mut left_wall = Shape::sphere();
|
||||||
left_wall.transform = Matrix4x4::translation(0., 0., 5.)
|
left_wall.set_transform(
|
||||||
* Matrix4x4::rotation_y(-PI / 4.)
|
Matrix4x4::translation(0., 0., 5.)
|
||||||
* Matrix4x4::rotation_x(PI / 2.)
|
* Matrix4x4::rotation_y(-PI / 4.)
|
||||||
* Matrix4x4::scaling(10., 0.01, 10.);
|
* Matrix4x4::rotation_x(PI / 2.)
|
||||||
|
* Matrix4x4::scaling(10., 0.01, 10.),
|
||||||
|
);
|
||||||
left_wall.material = floor.material.clone();
|
left_wall.material = floor.material.clone();
|
||||||
|
|
||||||
let mut right_wall = Sphere::default();
|
let mut right_wall = Shape::sphere();
|
||||||
right_wall.transform = Matrix4x4::translation(0., 0., 5.)
|
right_wall.set_transform(
|
||||||
* Matrix4x4::rotation_y(PI / 4.)
|
Matrix4x4::translation(0., 0., 5.)
|
||||||
* Matrix4x4::rotation_x(PI / 2.)
|
* Matrix4x4::rotation_y(PI / 4.)
|
||||||
* Matrix4x4::scaling(10., 0.01, 10.);
|
* Matrix4x4::rotation_x(PI / 2.)
|
||||||
|
* Matrix4x4::scaling(10., 0.01, 10.),
|
||||||
|
);
|
||||||
right_wall.material = floor.material.clone();
|
right_wall.material = floor.material.clone();
|
||||||
|
|
||||||
let mut middle = Sphere::default();
|
let mut middle = Shape::sphere();
|
||||||
middle.transform = Matrix4x4::translation(-0.5, 1., 0.5);
|
middle.set_transform(Matrix4x4::translation(-0.5, 1., 0.5));
|
||||||
middle.material = Material {
|
middle.material = Material {
|
||||||
color: Color::new(0.1, 1., 0.5),
|
color: Color::new(0.1, 1., 0.5).into(),
|
||||||
diffuse: 0.7,
|
diffuse: 0.7,
|
||||||
specular: 0.3,
|
specular: 0.3,
|
||||||
..Material::default()
|
..Material::default()
|
||||||
};
|
};
|
||||||
|
|
||||||
let mut right = Sphere::default();
|
let mut right = Shape::sphere();
|
||||||
right.transform = Matrix4x4::translation(1.5, 0.5, -0.5) * Matrix4x4::scaling(0.5, 0.5, 0.5);
|
right.set_transform(Matrix4x4::translation(1.5, 0.5, -0.5) * Matrix4x4::scaling(0.5, 0.5, 0.5));
|
||||||
right.material = Material {
|
right.material = Material {
|
||||||
color: Color::new(0.5, 1., 0.1),
|
color: Color::new(0.5, 1., 0.1).into(),
|
||||||
diffuse: 0.7,
|
diffuse: 0.7,
|
||||||
specular: 0.3,
|
specular: 0.3,
|
||||||
..Material::default()
|
..Material::default()
|
||||||
};
|
};
|
||||||
|
|
||||||
let mut left = Sphere::default();
|
let mut left = Shape::sphere();
|
||||||
left.transform =
|
left.set_transform(
|
||||||
Matrix4x4::translation(-1.5, 0.33, -0.75) * Matrix4x4::scaling(0.33, 0.33, 0.33);
|
Matrix4x4::translation(-1.5, 0.33, -0.75) * Matrix4x4::scaling(0.33, 0.33, 0.33),
|
||||||
|
);
|
||||||
left.material = Material {
|
left.material = Material {
|
||||||
color: Color::new(1., 0.8, 0.1),
|
color: Color::new(1., 0.8, 0.1).into(),
|
||||||
diffuse: 0.7,
|
diffuse: 0.7,
|
||||||
specular: 0.3,
|
specular: 0.3,
|
||||||
..Material::default()
|
..Material::default()
|
||||||
};
|
};
|
||||||
|
|
||||||
let mut world = World::default();
|
let mut world = World::default();
|
||||||
world.light = Some(light);
|
world.lights = vec![light];
|
||||||
world.objects = vec![floor, left_wall, right_wall, middle, right, left];
|
world.objects = vec![floor, left_wall, right_wall, middle, right, left];
|
||||||
|
|
||||||
let image = camera.render(&world);
|
let image = camera.render(&world);
|
||||||
@@ -86,6 +103,6 @@ fn main() -> Result<()> {
|
|||||||
let path = "/tmp/eoc7.png";
|
let path = "/tmp/eoc7.png";
|
||||||
println!("saving output to {}", path);
|
println!("saving output to {}", path);
|
||||||
image.write_to_file(path)?;
|
image.write_to_file(path)?;
|
||||||
println!("Render time {:.2} seconds", start.elapsed().as_secs_f32());
|
println!("Render time {:.3} seconds", start.elapsed().as_secs_f32());
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|||||||
120
rtchallenge/examples/eoc8.rs
Normal file
120
rtchallenge/examples/eoc8.rs
Normal file
@@ -0,0 +1,120 @@
|
|||||||
|
use std::time::Instant;
|
||||||
|
|
||||||
|
use anyhow::Result;
|
||||||
|
use structopt::StructOpt;
|
||||||
|
|
||||||
|
use rtchallenge::{
|
||||||
|
camera::{Camera, RenderStrategy},
|
||||||
|
float::consts::PI,
|
||||||
|
lights::PointLight,
|
||||||
|
materials::Material,
|
||||||
|
matrices::Matrix4x4,
|
||||||
|
shapes::Shape,
|
||||||
|
transformations::view_transform,
|
||||||
|
tuples::{Color, Tuple},
|
||||||
|
world::World,
|
||||||
|
WHITE,
|
||||||
|
};
|
||||||
|
|
||||||
|
/// End of chapter 8 challenge.
|
||||||
|
#[derive(StructOpt, Debug)]
|
||||||
|
#[structopt(name = "eoc8")]
|
||||||
|
struct Opt {
|
||||||
|
#[structopt(long, default_value = "rayon")]
|
||||||
|
render_strategy: RenderStrategy,
|
||||||
|
/// Number of samples per pixel. 0 renders from the center of the pixel, 1 or more samples N
|
||||||
|
/// times randomly across the pixel.
|
||||||
|
#[structopt(short, long, default_value = "0")]
|
||||||
|
samples: usize,
|
||||||
|
}
|
||||||
|
|
||||||
|
fn main() -> Result<()> {
|
||||||
|
let start = Instant::now();
|
||||||
|
let opt = Opt::from_args();
|
||||||
|
let width = 2560;
|
||||||
|
let height = 1440;
|
||||||
|
|
||||||
|
let light_position = Tuple::point(-10., 10., -10.);
|
||||||
|
let light_color = WHITE;
|
||||||
|
let light1 = PointLight::new(light_position, light_color);
|
||||||
|
let light_position = Tuple::point(10., 10., -10.);
|
||||||
|
let light_color = Color::new(0.2, 0.2, 0.2);
|
||||||
|
let light2 = PointLight::new(light_position, light_color);
|
||||||
|
let light_position = Tuple::point(0., 10., -10.);
|
||||||
|
let light3 = PointLight::new(light_position, light_color);
|
||||||
|
|
||||||
|
let mut camera = Camera::new(width, height, PI / 4.);
|
||||||
|
let from = Tuple::point(0., 1.5, -5.);
|
||||||
|
let to = Tuple::point(0., 1., 0.);
|
||||||
|
let up = Tuple::point(0., 1., 0.);
|
||||||
|
camera.set_transform(view_transform(from, to, up));
|
||||||
|
camera.render_strategy = opt.render_strategy;
|
||||||
|
camera.samples_per_pixel = opt.samples;
|
||||||
|
|
||||||
|
let mut floor = Shape::sphere();
|
||||||
|
floor.set_transform(Matrix4x4::scaling(10., 0.01, 10.));
|
||||||
|
floor.material = Material {
|
||||||
|
color: Color::new(1., 0.9, 0.9).into(),
|
||||||
|
specular: 0.,
|
||||||
|
..Material::default()
|
||||||
|
};
|
||||||
|
|
||||||
|
let mut left_wall = Shape::sphere();
|
||||||
|
left_wall.set_transform(
|
||||||
|
Matrix4x4::translation(0., 0., 5.)
|
||||||
|
* Matrix4x4::rotation_y(-PI / 4.)
|
||||||
|
* Matrix4x4::rotation_x(PI / 2.)
|
||||||
|
* Matrix4x4::scaling(10., 0.01, 10.),
|
||||||
|
);
|
||||||
|
left_wall.material = floor.material.clone();
|
||||||
|
|
||||||
|
let mut right_wall = Shape::sphere();
|
||||||
|
right_wall.set_transform(
|
||||||
|
Matrix4x4::translation(0., 0., 5.)
|
||||||
|
* Matrix4x4::rotation_y(PI / 4.)
|
||||||
|
* Matrix4x4::rotation_x(PI / 2.)
|
||||||
|
* Matrix4x4::scaling(10., 0.00001, 10.),
|
||||||
|
);
|
||||||
|
right_wall.material = floor.material.clone();
|
||||||
|
|
||||||
|
let mut middle = Shape::sphere();
|
||||||
|
middle.set_transform(Matrix4x4::translation(-0.5, 1., 0.5));
|
||||||
|
middle.material = Material {
|
||||||
|
color: Color::new(0.1, 1., 0.5).into(),
|
||||||
|
diffuse: 0.7,
|
||||||
|
specular: 0.3,
|
||||||
|
..Material::default()
|
||||||
|
};
|
||||||
|
|
||||||
|
let mut right = Shape::sphere();
|
||||||
|
right.set_transform(Matrix4x4::translation(1.5, 0.5, -0.5) * Matrix4x4::scaling(0.5, 0.5, 0.5));
|
||||||
|
right.material = Material {
|
||||||
|
color: Color::new(1., 1., 1.).into(),
|
||||||
|
diffuse: 0.7,
|
||||||
|
specular: 0.0,
|
||||||
|
..Material::default()
|
||||||
|
};
|
||||||
|
|
||||||
|
let mut left = Shape::sphere();
|
||||||
|
left.set_transform(
|
||||||
|
Matrix4x4::translation(-1.5, 0.33, -0.75) * Matrix4x4::scaling(0.33, 0.33, 0.33),
|
||||||
|
);
|
||||||
|
left.material = Material {
|
||||||
|
color: Color::new(1., 0.8, 0.1).into(),
|
||||||
|
diffuse: 0.7,
|
||||||
|
specular: 0.3,
|
||||||
|
..Material::default()
|
||||||
|
};
|
||||||
|
|
||||||
|
let mut world = World::default();
|
||||||
|
world.lights = vec![light1, light2, light3];
|
||||||
|
world.objects = vec![floor, left_wall, right_wall, middle, right, left];
|
||||||
|
|
||||||
|
let image = camera.render(&world);
|
||||||
|
|
||||||
|
let path = "/tmp/eoc8.png";
|
||||||
|
println!("saving output to {}", path);
|
||||||
|
image.write_to_file(path)?;
|
||||||
|
println!("Render time {:.3} seconds", start.elapsed().as_secs_f32());
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
122
rtchallenge/examples/eoc9-prelude.rs
Normal file
122
rtchallenge/examples/eoc9-prelude.rs
Normal file
@@ -0,0 +1,122 @@
|
|||||||
|
use std::time::Instant;
|
||||||
|
|
||||||
|
use anyhow::Result;
|
||||||
|
use structopt::StructOpt;
|
||||||
|
|
||||||
|
use rtchallenge::prelude::*;
|
||||||
|
|
||||||
|
use rtchallenge::{camera::RenderStrategy, float::consts::PI, WHITE};
|
||||||
|
|
||||||
|
/// End of chapter 9 challenge.
|
||||||
|
#[derive(StructOpt, Debug)]
|
||||||
|
#[structopt(name = "eoc9")]
|
||||||
|
struct Opt {
|
||||||
|
/// Strategy for casting rays into image.
|
||||||
|
#[structopt(long, default_value = "rayon")]
|
||||||
|
render_strategy: RenderStrategy,
|
||||||
|
/// Number of samples per pixel. 0 renders from the center of the pixel, 1 or more samples N
|
||||||
|
/// times randomly across the pixel.
|
||||||
|
#[structopt(short, long, default_value = "0")]
|
||||||
|
samples: usize,
|
||||||
|
/// Rendered image width in pixels.
|
||||||
|
#[structopt(short, long, default_value = "2560")]
|
||||||
|
width: usize,
|
||||||
|
/// Rendered image height in pixels.
|
||||||
|
#[structopt(short, long, default_value = "1440")]
|
||||||
|
height: usize,
|
||||||
|
}
|
||||||
|
|
||||||
|
fn main() -> Result<()> {
|
||||||
|
let start = Instant::now();
|
||||||
|
let opt = Opt::from_args();
|
||||||
|
|
||||||
|
let light1 = PointLightBuilder::default()
|
||||||
|
.position(point(-5., 5., -5.))
|
||||||
|
.intensity(WHITE)
|
||||||
|
.build()?;
|
||||||
|
let light2 = PointLightBuilder::default()
|
||||||
|
.position(point(5., 5., -5.))
|
||||||
|
.intensity([0.2, 0.2, 0.6])
|
||||||
|
.build()?;
|
||||||
|
let light3 = PointLightBuilder::default()
|
||||||
|
.position(point(0., 2., -5.))
|
||||||
|
.intensity([0.2, 0.2, 0.1])
|
||||||
|
.build()?;
|
||||||
|
|
||||||
|
let from = point(0., 1.5, -5.);
|
||||||
|
let to = point(0., 1., 0.);
|
||||||
|
let up = point(0., 1., 0.);
|
||||||
|
let camera = CameraBuilder::default()
|
||||||
|
.hsize(opt.width)
|
||||||
|
.vsize(opt.height)
|
||||||
|
.field_of_view(PI / 4.)
|
||||||
|
.transform(view_transform(from, to, up))
|
||||||
|
.render_strategy(opt.render_strategy)
|
||||||
|
.samples_per_pixel(opt.samples)
|
||||||
|
.build()?;
|
||||||
|
|
||||||
|
let floor = plane()
|
||||||
|
.material(
|
||||||
|
MaterialBuilder::default()
|
||||||
|
.color([1., 0.2, 0.2])
|
||||||
|
.specular(0.)
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.build()?;
|
||||||
|
|
||||||
|
let ceiling = plane()
|
||||||
|
.transform(translation(0., 6., 0.) * rotation_x(PI))
|
||||||
|
.material(
|
||||||
|
MaterialBuilder::default()
|
||||||
|
.color([0.6, 0.6, 0.8])
|
||||||
|
.specular(0.2)
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.build()?;
|
||||||
|
|
||||||
|
let middle = sphere()
|
||||||
|
.transform(translation(-0.5, 0.5, 0.5))
|
||||||
|
.material(
|
||||||
|
MaterialBuilder::default()
|
||||||
|
.color([0.1, 1., 0.5])
|
||||||
|
.diffuse(0.7)
|
||||||
|
.specular(0.3)
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.build()?;
|
||||||
|
|
||||||
|
let right = sphere()
|
||||||
|
.transform(translation(1.5, 0.5, -0.5) * scaling(0.5, 0.5, 0.5))
|
||||||
|
.material(
|
||||||
|
MaterialBuilder::default()
|
||||||
|
.color([1., 1., 1.])
|
||||||
|
.diffuse(0.7)
|
||||||
|
.specular(0.0)
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.build()?;
|
||||||
|
|
||||||
|
let left = sphere()
|
||||||
|
.transform(translation(-1.5, 0.33, -0.75) * scaling(0.33, 0.33, 0.33))
|
||||||
|
.material(
|
||||||
|
MaterialBuilder::default()
|
||||||
|
.color([1., 0.8, 0.1])
|
||||||
|
.diffuse(0.7)
|
||||||
|
.specular(0.3)
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.build()?;
|
||||||
|
|
||||||
|
let world = WorldBuilder::default()
|
||||||
|
.lights(vec![light1, light2, light3])
|
||||||
|
.objects(vec![floor, ceiling, middle, right, left])
|
||||||
|
.build()?;
|
||||||
|
|
||||||
|
let image = camera.render(&world);
|
||||||
|
|
||||||
|
let path = "/tmp/eoc9.png";
|
||||||
|
println!("saving output to {}", path);
|
||||||
|
image.write_to_file(path)?;
|
||||||
|
println!("Render time {:.3} seconds", start.elapsed().as_secs_f32());
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
114
rtchallenge/examples/eoc9.rs
Normal file
114
rtchallenge/examples/eoc9.rs
Normal file
@@ -0,0 +1,114 @@
|
|||||||
|
use std::time::Instant;
|
||||||
|
|
||||||
|
use anyhow::Result;
|
||||||
|
use structopt::StructOpt;
|
||||||
|
|
||||||
|
use rtchallenge::{
|
||||||
|
camera::{Camera, RenderStrategy},
|
||||||
|
float::consts::PI,
|
||||||
|
lights::PointLight,
|
||||||
|
materials::Material,
|
||||||
|
matrices::Matrix4x4,
|
||||||
|
shapes::Shape,
|
||||||
|
transformations::view_transform,
|
||||||
|
tuples::{Color, Tuple},
|
||||||
|
world::World,
|
||||||
|
WHITE,
|
||||||
|
};
|
||||||
|
|
||||||
|
/// End of chapter 9 challenge.
|
||||||
|
#[derive(StructOpt, Debug)]
|
||||||
|
#[structopt(name = "eoc9")]
|
||||||
|
struct Opt {
|
||||||
|
/// Strategy for casting rays into image.
|
||||||
|
#[structopt(long, default_value = "rayon")]
|
||||||
|
render_strategy: RenderStrategy,
|
||||||
|
/// Number of samples per pixel. 0 renders from the center of the pixel, 1 or more samples N
|
||||||
|
/// times randomly across the pixel.
|
||||||
|
#[structopt(short, long, default_value = "0")]
|
||||||
|
samples: usize,
|
||||||
|
/// Rendered image width in pixels.
|
||||||
|
#[structopt(short, long, default_value = "2560")]
|
||||||
|
width: usize,
|
||||||
|
/// Rendered image height in pixels.
|
||||||
|
#[structopt(short, long, default_value = "1440")]
|
||||||
|
height: usize,
|
||||||
|
}
|
||||||
|
|
||||||
|
fn main() -> Result<()> {
|
||||||
|
let start = Instant::now();
|
||||||
|
let opt = Opt::from_args();
|
||||||
|
|
||||||
|
let light_position = Tuple::point(-5., 5., -5.);
|
||||||
|
let light_color = WHITE;
|
||||||
|
let light1 = PointLight::new(light_position, light_color);
|
||||||
|
let light_position = Tuple::point(5., 5., -5.);
|
||||||
|
let light_color = Color::new(0.2, 0.2, 0.6);
|
||||||
|
let light2 = PointLight::new(light_position, light_color);
|
||||||
|
let light_position = Tuple::point(0., 2., -5.);
|
||||||
|
let light_color = Color::new(0.2, 0.2, 0.1);
|
||||||
|
let light3 = PointLight::new(light_position, light_color);
|
||||||
|
|
||||||
|
let mut camera = Camera::new(opt.width, opt.height, PI / 4.);
|
||||||
|
let from = Tuple::point(0., 1.5, -5.);
|
||||||
|
let to = Tuple::point(0., 1., 0.);
|
||||||
|
let up = Tuple::point(0., 1., 0.);
|
||||||
|
camera.set_transform(view_transform(from, to, up));
|
||||||
|
camera.render_strategy = opt.render_strategy;
|
||||||
|
camera.samples_per_pixel = opt.samples;
|
||||||
|
|
||||||
|
let mut floor = Shape::plane();
|
||||||
|
floor.material = Material {
|
||||||
|
color: Color::new(1., 0.2, 0.2).into(),
|
||||||
|
specular: 0.,
|
||||||
|
..Material::default()
|
||||||
|
};
|
||||||
|
let mut ceiling = Shape::plane();
|
||||||
|
ceiling.set_transform(Matrix4x4::translation(0., 6., 0.) * Matrix4x4::rotation_x(PI));
|
||||||
|
ceiling.material = Material {
|
||||||
|
color: Color::new(0.6, 0.6, 0.8).into(),
|
||||||
|
specular: 0.2,
|
||||||
|
..Material::default()
|
||||||
|
};
|
||||||
|
|
||||||
|
let mut middle = Shape::sphere();
|
||||||
|
middle.set_transform(Matrix4x4::translation(-0.5, 0.5, 0.5));
|
||||||
|
middle.material = Material {
|
||||||
|
color: Color::new(0.1, 1., 0.5).into(),
|
||||||
|
diffuse: 0.7,
|
||||||
|
specular: 0.3,
|
||||||
|
..Material::default()
|
||||||
|
};
|
||||||
|
|
||||||
|
let mut right = Shape::sphere();
|
||||||
|
right.set_transform(Matrix4x4::translation(1.5, 0.5, -0.5) * Matrix4x4::scaling(0.5, 0.5, 0.5));
|
||||||
|
right.material = Material {
|
||||||
|
color: Color::new(1., 1., 1.).into(),
|
||||||
|
diffuse: 0.7,
|
||||||
|
specular: 0.0,
|
||||||
|
..Material::default()
|
||||||
|
};
|
||||||
|
|
||||||
|
let mut left = Shape::sphere();
|
||||||
|
left.set_transform(
|
||||||
|
Matrix4x4::translation(-1.5, 0.33, -0.75) * Matrix4x4::scaling(0.33, 0.33, 0.33),
|
||||||
|
);
|
||||||
|
left.material = Material {
|
||||||
|
color: Color::new(1., 0.8, 0.1).into(),
|
||||||
|
diffuse: 0.7,
|
||||||
|
specular: 0.3,
|
||||||
|
..Material::default()
|
||||||
|
};
|
||||||
|
|
||||||
|
let mut world = World::default();
|
||||||
|
world.lights = vec![light1, light2, light3];
|
||||||
|
world.objects = vec![floor, ceiling, middle, right, left];
|
||||||
|
|
||||||
|
let image = camera.render(&world);
|
||||||
|
|
||||||
|
let path = "/tmp/eoc9.png";
|
||||||
|
println!("saving output to {}", path);
|
||||||
|
image.write_to_file(path)?;
|
||||||
|
println!("Render time {:.3} seconds", start.elapsed().as_secs_f32());
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
115
rtchallenge/examples/glass.rs
Normal file
115
rtchallenge/examples/glass.rs
Normal file
@@ -0,0 +1,115 @@
|
|||||||
|
use std::time::Instant;
|
||||||
|
|
||||||
|
use anyhow::Result;
|
||||||
|
use structopt::StructOpt;
|
||||||
|
|
||||||
|
use rtchallenge::prelude::*;
|
||||||
|
|
||||||
|
use rtchallenge::{
|
||||||
|
camera::RenderStrategy,
|
||||||
|
float::consts::PI,
|
||||||
|
patterns::{BLACK_PAT, WHITE_PAT},
|
||||||
|
WHITE,
|
||||||
|
};
|
||||||
|
|
||||||
|
/// Experiment with transparency.
|
||||||
|
#[derive(StructOpt, Debug)]
|
||||||
|
#[structopt(name = "glass")]
|
||||||
|
struct Opt {
|
||||||
|
/// Strategy for casting rays into image.
|
||||||
|
#[structopt(long, default_value = "rayon")]
|
||||||
|
render_strategy: RenderStrategy,
|
||||||
|
/// Number of samples per pixel. 0 renders from the center of the pixel, 1 or more samples N
|
||||||
|
/// times randomly across the pixel.
|
||||||
|
#[structopt(short, long, default_value = "0")]
|
||||||
|
samples: usize,
|
||||||
|
/// Rendered image width in pixels.
|
||||||
|
#[structopt(short, long, default_value = "2560")]
|
||||||
|
width: usize,
|
||||||
|
/// Rendered image height in pixels.
|
||||||
|
#[structopt(short, long, default_value = "1440")]
|
||||||
|
height: usize,
|
||||||
|
}
|
||||||
|
|
||||||
|
fn main() -> Result<()> {
|
||||||
|
let start = Instant::now();
|
||||||
|
let opt = Opt::from_args();
|
||||||
|
|
||||||
|
let light1 = PointLightBuilder::default()
|
||||||
|
.position(point(-5., 5., -5.))
|
||||||
|
.intensity(WHITE)
|
||||||
|
.build()?;
|
||||||
|
let light2 = PointLightBuilder::default()
|
||||||
|
.position(point(5., 5., -5.))
|
||||||
|
.intensity([0.2, 0.2, 0.6])
|
||||||
|
.build()?;
|
||||||
|
let light3 = PointLightBuilder::default()
|
||||||
|
.position(point(0., 2., -5.))
|
||||||
|
.intensity([0.2, 0.2, 0.1])
|
||||||
|
.build()?;
|
||||||
|
|
||||||
|
let from = point(0., 0., -10.);
|
||||||
|
let to = point(0., 0., 0.);
|
||||||
|
let up = point(0., 1., 0.);
|
||||||
|
let camera = CameraBuilder::default()
|
||||||
|
.hsize(opt.width)
|
||||||
|
.vsize(opt.height)
|
||||||
|
.field_of_view(PI / 4.)
|
||||||
|
.transform(view_transform(from, to, up))
|
||||||
|
.render_strategy(opt.render_strategy)
|
||||||
|
.samples_per_pixel(opt.samples)
|
||||||
|
.build()?;
|
||||||
|
|
||||||
|
let floor = plane()
|
||||||
|
.transform(rotation_x(PI / 2.))
|
||||||
|
.material(
|
||||||
|
MaterialBuilder::default()
|
||||||
|
.color(checkers_pattern(BLACK_PAT, WHITE_PAT).build()?)
|
||||||
|
.reflective(0.5)
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.build()?;
|
||||||
|
let ceiling = plane()
|
||||||
|
.transform(translation(0., 0., -20.) * rotation_x(PI / 2.))
|
||||||
|
.material(MaterialBuilder::default().color([0.2, 0.2, 0.8]).build()?)
|
||||||
|
.build()?;
|
||||||
|
|
||||||
|
let mut objects = Vec::new();
|
||||||
|
for x in 0..5 {
|
||||||
|
for y in 0..5 {
|
||||||
|
let trans = y as Float / 5.;
|
||||||
|
let index = 1. + x as Float / 5.;
|
||||||
|
dbg!(x, y, trans, index);
|
||||||
|
objects.push(
|
||||||
|
sphere()
|
||||||
|
.transform(
|
||||||
|
translation(x as Float - 2., y as Float - 2., -2.) * scaling(0.5, 0.5, 0.5),
|
||||||
|
)
|
||||||
|
.material(
|
||||||
|
MaterialBuilder::default()
|
||||||
|
.color([0.5, 0.5, 0.5])
|
||||||
|
.transparency(trans)
|
||||||
|
.refractive_index(index)
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.build()?,
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
objects.push(floor);
|
||||||
|
objects.push(ceiling);
|
||||||
|
|
||||||
|
let world = WorldBuilder::default()
|
||||||
|
.lights(vec![light1, light2, light3])
|
||||||
|
.objects(objects)
|
||||||
|
.build()?;
|
||||||
|
|
||||||
|
let image = camera.render(&world);
|
||||||
|
|
||||||
|
let path = "/tmp/glass.png";
|
||||||
|
println!("saving output to {}", path);
|
||||||
|
image.write_to_file(path)?;
|
||||||
|
println!("Render time {:.3} seconds", start.elapsed().as_secs_f32());
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
@@ -1 +1,2 @@
|
|||||||
|
imports_granularity = "Crate"
|
||||||
format_code_in_doc_comments = true
|
format_code_in_doc_comments = true
|
||||||
|
|||||||
@@ -1,58 +1,143 @@
|
|||||||
use crate::{canvas::Canvas, matrices::Matrix4x4, rays::Ray, tuples::Tuple, world::World, BLACK};
|
use std::{
|
||||||
|
str::FromStr,
|
||||||
|
sync::{
|
||||||
|
mpsc::{sync_channel, Receiver, SyncSender},
|
||||||
|
Arc, Mutex,
|
||||||
|
},
|
||||||
|
thread,
|
||||||
|
};
|
||||||
|
|
||||||
|
use derive_builder::Builder;
|
||||||
|
use rand::Rng;
|
||||||
|
use rayon::iter::{IntoParallelIterator, ParallelIterator};
|
||||||
|
use serde::Deserialize;
|
||||||
|
use structopt::StructOpt;
|
||||||
|
|
||||||
|
use crate::{
|
||||||
|
canvas::Canvas,
|
||||||
|
matrices::Matrix4x4,
|
||||||
|
rays::Ray,
|
||||||
|
tuples::{Color, Tuple},
|
||||||
|
world::World,
|
||||||
|
Float, BLACK,
|
||||||
|
};
|
||||||
|
|
||||||
|
const MAX_DEPTH_RECURSION: usize = 10;
|
||||||
|
|
||||||
|
#[derive(Copy, Clone, StructOpt, Debug, Deserialize)]
|
||||||
|
#[serde(rename_all = "kebab-case")]
|
||||||
|
pub enum RenderStrategy {
|
||||||
|
Serial,
|
||||||
|
Rayon,
|
||||||
|
WorkerPool,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Default for RenderStrategy {
|
||||||
|
fn default() -> RenderStrategy {
|
||||||
|
RenderStrategy::Rayon
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl FromStr for RenderStrategy {
|
||||||
|
type Err = serde_json::error::Error;
|
||||||
|
fn from_str(s: &str) -> Result<RenderStrategy, serde_json::error::Error> {
|
||||||
|
serde_json::from_str(&format!("\"{}\"", s))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Builder, Clone, Debug, Default)]
|
||||||
|
#[builder(setter(skip), build_fn(skip))]
|
||||||
pub struct Camera {
|
pub struct Camera {
|
||||||
|
#[builder(setter(skip = "false"))]
|
||||||
hsize: usize,
|
hsize: usize,
|
||||||
|
#[builder(setter(skip = "false"))]
|
||||||
vsize: usize,
|
vsize: usize,
|
||||||
field_of_view: f32,
|
#[builder(setter(skip = "false"))]
|
||||||
pub transform: Matrix4x4,
|
field_of_view: Float,
|
||||||
pixel_size: f32,
|
#[builder(setter(skip = "false"))]
|
||||||
half_width: f32,
|
transform: Matrix4x4,
|
||||||
half_height: f32,
|
inverse_transform: Matrix4x4,
|
||||||
|
pixel_size: Float,
|
||||||
|
half_width: Float,
|
||||||
|
half_height: Float,
|
||||||
|
#[builder(setter(skip = "false"))]
|
||||||
|
pub render_strategy: RenderStrategy,
|
||||||
|
/// 0 renders from the center of the pixel, 1 or higher is random sampling of the pixel.
|
||||||
|
#[builder(setter(skip = "false"))]
|
||||||
|
pub samples_per_pixel: usize,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl CameraBuilder {
|
||||||
|
pub fn build(&self) -> Result<Camera, CameraBuilderError> {
|
||||||
|
let hsize = match self.hsize {
|
||||||
|
Some(ref value) => Clone::clone(value),
|
||||||
|
None => {
|
||||||
|
return Err(Into::into(::derive_builder::UninitializedFieldError::from(
|
||||||
|
"hsize",
|
||||||
|
)))
|
||||||
|
}
|
||||||
|
};
|
||||||
|
let vsize = match self.vsize {
|
||||||
|
Some(ref value) => Clone::clone(value),
|
||||||
|
None => {
|
||||||
|
return Err(Into::into(::derive_builder::UninitializedFieldError::from(
|
||||||
|
"vsize",
|
||||||
|
)))
|
||||||
|
}
|
||||||
|
};
|
||||||
|
let field_of_view = match self.field_of_view {
|
||||||
|
Some(ref value) => Clone::clone(value),
|
||||||
|
None => {
|
||||||
|
return Err(Into::into(::derive_builder::UninitializedFieldError::from(
|
||||||
|
"field_of_view",
|
||||||
|
)))
|
||||||
|
}
|
||||||
|
};
|
||||||
|
let mut c = Camera::new(hsize, vsize, field_of_view);
|
||||||
|
if let Some(transform) = self.transform {
|
||||||
|
c.set_transform(transform);
|
||||||
|
}
|
||||||
|
if let Some(render_strategy) = self.render_strategy {
|
||||||
|
c.render_strategy = render_strategy;
|
||||||
|
}
|
||||||
|
if let Some(samples_per_pixel) = self.samples_per_pixel {
|
||||||
|
c.samples_per_pixel = samples_per_pixel;
|
||||||
|
}
|
||||||
|
Ok(c)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
enum Request {
|
||||||
|
Line { width: usize, y: usize },
|
||||||
|
}
|
||||||
|
|
||||||
|
enum Response {
|
||||||
|
Line { y: usize, pixels: Canvas },
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Camera {
|
impl Camera {
|
||||||
/// Create a camera with a canvas of pixel hsize (height) and vsize (width)
|
/// Create a camera with a canvas of pixel hsize (height) and vsize (width)
|
||||||
/// with the given field of view (in radians).
|
/// with the given field of view (in radians).
|
||||||
///
|
pub fn new(hsize: usize, vsize: usize, field_of_view: Float) -> Camera {
|
||||||
/// # Examples
|
|
||||||
/// ```
|
|
||||||
/// use std::f32::consts::PI;
|
|
||||||
///
|
|
||||||
/// use rtchallenge::{camera::Camera, matrices::Matrix4x4};
|
|
||||||
///
|
|
||||||
/// let hsize = 160;
|
|
||||||
/// let vsize = 120;
|
|
||||||
/// let field_of_view = PI / 2.;
|
|
||||||
/// let c = Camera::new(hsize, vsize, field_of_view);
|
|
||||||
/// assert_eq!(c.hsize(), 160);
|
|
||||||
/// assert_eq!(c.vsize(), 120);
|
|
||||||
/// assert_eq!(c.transform(), Matrix4x4::identity());
|
|
||||||
///
|
|
||||||
/// // Pixel size for a horizontal canvas.
|
|
||||||
/// let c = Camera::new(200, 150, PI / 2.);
|
|
||||||
/// assert_eq!(c.pixel_size(), 0.01);
|
|
||||||
///
|
|
||||||
/// // Pixel size for a horizontal canvas.
|
|
||||||
/// let c = Camera::new(150, 200, PI / 2.);
|
|
||||||
/// assert_eq!(c.pixel_size(), 0.01);
|
|
||||||
/// ```
|
|
||||||
pub fn new(hsize: usize, vsize: usize, field_of_view: f32) -> Camera {
|
|
||||||
let half_view = (field_of_view / 2.).tan();
|
let half_view = (field_of_view / 2.).tan();
|
||||||
let aspect = hsize as f32 / vsize as f32;
|
let aspect = hsize as Float / vsize as Float;
|
||||||
let (half_width, half_height) = if aspect >= 1. {
|
let (half_width, half_height) = if aspect >= 1. {
|
||||||
(half_view, half_view / aspect)
|
(half_view, half_view / aspect)
|
||||||
} else {
|
} else {
|
||||||
(half_view * aspect, half_view)
|
(half_view * aspect, half_view)
|
||||||
};
|
};
|
||||||
let pixel_size = 2. * half_width / hsize as f32;
|
let pixel_size = 2. * half_width / hsize as Float;
|
||||||
Camera {
|
Camera {
|
||||||
hsize,
|
hsize,
|
||||||
vsize,
|
vsize,
|
||||||
field_of_view,
|
field_of_view,
|
||||||
transform: Matrix4x4::identity(),
|
transform: Matrix4x4::identity(),
|
||||||
|
inverse_transform: Matrix4x4::identity(),
|
||||||
pixel_size,
|
pixel_size,
|
||||||
half_height,
|
half_height,
|
||||||
half_width,
|
half_width,
|
||||||
|
render_strategy: RenderStrategy::Rayon,
|
||||||
|
samples_per_pixel: 0,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
pub fn hsize(&self) -> usize {
|
pub fn hsize(&self) -> usize {
|
||||||
@@ -61,99 +146,291 @@ impl Camera {
|
|||||||
pub fn vsize(&self) -> usize {
|
pub fn vsize(&self) -> usize {
|
||||||
self.vsize
|
self.vsize
|
||||||
}
|
}
|
||||||
pub fn field_of_view(&self) -> f32 {
|
pub fn field_of_view(&self) -> Float {
|
||||||
self.field_of_view
|
self.field_of_view
|
||||||
}
|
}
|
||||||
pub fn transform(&self) -> Matrix4x4 {
|
pub fn transform(&self) -> Matrix4x4 {
|
||||||
self.transform
|
self.transform
|
||||||
}
|
}
|
||||||
pub fn pixel_size(&self) -> f32 {
|
pub fn set_transform(&mut self, t: Matrix4x4) {
|
||||||
|
self.transform = t;
|
||||||
|
self.inverse_transform = t.inverse();
|
||||||
|
}
|
||||||
|
pub fn pixel_size(&self) -> Float {
|
||||||
self.pixel_size
|
self.pixel_size
|
||||||
}
|
}
|
||||||
|
pub fn supersample_rays_for_pixel(&self, px: usize, py: usize, samples: usize) -> Vec<Ray> {
|
||||||
|
let mut rng = rand::thread_rng();
|
||||||
|
|
||||||
|
(0..samples)
|
||||||
|
.map(|_| {
|
||||||
|
// The offset from the edge of the canvas to the pixel's corner.
|
||||||
|
let xoffset = (px as Float + rng.gen::<Float>()) * self.pixel_size;
|
||||||
|
let yoffset = (py as Float + rng.gen::<Float>()) * self.pixel_size;
|
||||||
|
|
||||||
|
// The untransformed coordinates of the pixle in world space.
|
||||||
|
// (Remember that the camera looks toward -z, so +x is to the left.)
|
||||||
|
let world_x = self.half_width - xoffset;
|
||||||
|
let world_y = self.half_height - yoffset;
|
||||||
|
|
||||||
|
// Using the camera matrix, transofmrm the canvas point and the origin,
|
||||||
|
// and then compute the ray's direction vector.
|
||||||
|
// (Remember that the canvas is at z>=-1).
|
||||||
|
let pixel = self.inverse_transform * Tuple::point(world_x, world_y, -1.);
|
||||||
|
let origin = self.inverse_transform * Tuple::point(0., 0., 0.);
|
||||||
|
let direction = (pixel - origin).normalize();
|
||||||
|
|
||||||
|
Ray::new(origin, direction)
|
||||||
|
})
|
||||||
|
.collect()
|
||||||
|
}
|
||||||
|
|
||||||
/// Calculate ray that starts at the camera and passes through the (x,y)
|
/// Calculate ray that starts at the camera and passes through the (x,y)
|
||||||
/// pixel on the canvas.
|
/// pixel on the canvas.
|
||||||
///
|
|
||||||
/// # Examples
|
|
||||||
/// ```
|
|
||||||
/// use std::f32::consts::PI;
|
|
||||||
///
|
|
||||||
/// use rtchallenge::{camera::Camera, matrices::Matrix4x4, tuples::Tuple};
|
|
||||||
///
|
|
||||||
/// // Constructing a ray through the center of the canvas.
|
|
||||||
/// let c = Camera::new(201, 101, PI / 2.);
|
|
||||||
/// let r = c.ray_for_pixel(100, 50);
|
|
||||||
/// assert_eq!(r.origin, Tuple::point(0., 0., 0.));
|
|
||||||
/// assert_eq!(r.direction, Tuple::vector(0., 0., -1.));
|
|
||||||
///
|
|
||||||
/// // Constructing a ray through the corner of the canvas.
|
|
||||||
/// let c = Camera::new(201, 101, PI / 2.);
|
|
||||||
/// let r = c.ray_for_pixel(0, 0);
|
|
||||||
/// assert_eq!(r.origin, Tuple::point(0., 0., 0.));
|
|
||||||
/// assert_eq!(r.direction, Tuple::vector(0.66519, 0.33259, -0.66851));
|
|
||||||
///
|
|
||||||
/// // Constructing a ray when the camera is transformed.
|
|
||||||
/// let mut c = Camera::new(201, 101, PI / 2.);
|
|
||||||
/// c.transform = Matrix4x4::rotation_y(PI / 4.) * Matrix4x4::translation(0., -2., 5.);
|
|
||||||
/// let r = c.ray_for_pixel(100, 50);
|
|
||||||
/// assert_eq!(r.origin, Tuple::point(0., 2., -5.));
|
|
||||||
/// assert_eq!(
|
|
||||||
/// r.direction,
|
|
||||||
/// Tuple::vector(2_f32.sqrt() / 2., 0., -2_f32.sqrt() / 2.)
|
|
||||||
/// );
|
|
||||||
/// ```
|
|
||||||
pub fn ray_for_pixel(&self, px: usize, py: usize) -> Ray {
|
pub fn ray_for_pixel(&self, px: usize, py: usize) -> Ray {
|
||||||
// The offset from the edge of the canvas to the pixel's corner.
|
// The offset from the edge of the canvas to the pixel's corner.
|
||||||
let xoffset = (px as f32 + 0.5) * self.pixel_size;
|
let xoffset = (px as Float + 0.5) * self.pixel_size;
|
||||||
let yoffset = (py as f32 + 0.5) * self.pixel_size;
|
let yoffset = (py as Float + 0.5) * self.pixel_size;
|
||||||
|
|
||||||
// The untransformed coordinates of the pixle in world space.
|
// The untransformed coordinates of the pixel in world space.
|
||||||
// (Remember that the camera looks toward -z, so +x is to the left.)
|
// (Remember that the camera looks toward -z, so +x is to the left.)
|
||||||
let world_x = self.half_width - xoffset;
|
let world_x = self.half_width - xoffset;
|
||||||
let world_y = self.half_height - yoffset;
|
let world_y = self.half_height - yoffset;
|
||||||
|
|
||||||
// Using the camera matrix, transofmrm the canvas point and the origin,
|
// Using the camera matrix, transform the canvas point and the origin,
|
||||||
// and then compute the ray's direction vector.
|
// and then compute the ray's direction vector.
|
||||||
// (Remember that the canves is at z>=-1).
|
// (Remember that the canvas is at z>=-1).
|
||||||
let pixel = self.transform.inverse() * Tuple::point(world_x, world_y, -1.);
|
let pixel = self.inverse_transform * Tuple::point(world_x, world_y, -1.);
|
||||||
let origin = self.transform.inverse() * Tuple::point(0., 0., 0.);
|
let origin = self.inverse_transform * Tuple::point(0., 0., 0.);
|
||||||
let direction = (pixel - origin).normalize();
|
let direction = (pixel - origin).normalize();
|
||||||
|
|
||||||
Ray::new(origin, direction)
|
Ray::new(origin, direction)
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Use camera to render an image of the given world.
|
/// Use camera to render an image of the given world.
|
||||||
///
|
|
||||||
/// # Examples
|
|
||||||
/// ```
|
|
||||||
/// use std::f32::consts::PI;
|
|
||||||
///
|
|
||||||
/// use rtchallenge::{
|
|
||||||
/// camera::Camera,
|
|
||||||
/// transformations::view_transform,
|
|
||||||
/// tuples::{Color, Tuple},
|
|
||||||
/// world::World,
|
|
||||||
/// };
|
|
||||||
///
|
|
||||||
/// // Rendering a world with a camera.
|
|
||||||
/// let w = World::test_world();
|
|
||||||
/// let mut c = Camera::new(11, 11, PI / 2.);
|
|
||||||
/// let from = Tuple::point(0., 0., -5.);
|
|
||||||
/// let to = Tuple::point(0., 0., 0.);
|
|
||||||
/// let up = Tuple::vector(0., 1., 0.);
|
|
||||||
/// c.transform = view_transform(from, to, up);
|
|
||||||
/// let image = c.render(&w);
|
|
||||||
/// assert_eq!(image.get(5, 5), Color::new(0.38066, 0.47583, 0.2855));
|
|
||||||
/// ```
|
|
||||||
pub fn render(&self, w: &World) -> Canvas {
|
pub fn render(&self, w: &World) -> Canvas {
|
||||||
|
use RenderStrategy::*;
|
||||||
|
|
||||||
|
match self.render_strategy {
|
||||||
|
Serial => self.render_serial(w),
|
||||||
|
Rayon => self.render_parallel_rayon(w),
|
||||||
|
WorkerPool => self.render_parallel_one_tread_per_core(w),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// This render function spins up one thread per core, and pins the thread
|
||||||
|
/// to the core. It then sends work requests to the worker threads,
|
||||||
|
/// requesting a full line of the image by rendered. The main thread
|
||||||
|
/// collects results and stores them in the canvas returned to the user.
|
||||||
|
fn render_parallel_one_tread_per_core(&self, world: &World) -> Canvas {
|
||||||
|
let mut image = Canvas::new(self.hsize, self.vsize, BLACK);
|
||||||
|
let num_threads = num_cpus::get();
|
||||||
|
let (pixel_req_tx, pixel_req_rx) = sync_channel(2 * num_threads);
|
||||||
|
let (pixel_resp_tx, pixel_resp_rx) = sync_channel(2 * num_threads);
|
||||||
|
let pixel_req_rx = Arc::new(Mutex::new(pixel_req_rx));
|
||||||
|
|
||||||
|
// Create copy of world and camera we can share with all workers.
|
||||||
|
// It's probably okay to clone camera, but world could get large (think
|
||||||
|
// textures and high poly count models).
|
||||||
|
// TODO(wathiede): prevent second copy of world when they start getting
|
||||||
|
// large.
|
||||||
|
let world = Arc::new(world.clone());
|
||||||
|
let camera = Arc::new(self.clone());
|
||||||
|
|
||||||
|
let core_ids = core_affinity::get_core_ids().unwrap();
|
||||||
|
println!("Creating {} render threads", core_ids.len());
|
||||||
|
// Create a worker thread for each CPU core and pin the thread to the core.
|
||||||
|
let mut handles = core_ids
|
||||||
|
.into_iter()
|
||||||
|
.map(|id| {
|
||||||
|
let w = Arc::clone(&world);
|
||||||
|
let c = Arc::clone(&camera);
|
||||||
|
let pixel_req_rx = pixel_req_rx.clone();
|
||||||
|
let pixel_resp_tx = pixel_resp_tx.clone();
|
||||||
|
thread::spawn(move || {
|
||||||
|
core_affinity::set_for_current(id);
|
||||||
|
render_worker_task(&c, &w, pixel_req_rx, &pixel_resp_tx);
|
||||||
|
})
|
||||||
|
})
|
||||||
|
.collect::<Vec<_>>();
|
||||||
|
drop(pixel_req_rx);
|
||||||
|
drop(pixel_resp_tx);
|
||||||
|
|
||||||
|
// Send render requests over channels to worker threads.
|
||||||
|
let (w, h) = (camera.hsize, camera.vsize);
|
||||||
|
handles.push(thread::spawn(move || {
|
||||||
|
for y in 0..h {
|
||||||
|
pixel_req_tx
|
||||||
|
.send(Request::Line { width: w, y })
|
||||||
|
.expect("failed to send line request");
|
||||||
|
}
|
||||||
|
drop(pixel_req_tx);
|
||||||
|
}));
|
||||||
|
|
||||||
|
// Read responses from channel and blit image data.
|
||||||
|
for resp in pixel_resp_rx {
|
||||||
|
match resp {
|
||||||
|
Response::Line { y, pixels } => {
|
||||||
|
for x in 0..camera.hsize {
|
||||||
|
image.set(x, y, pixels.get(x, 0));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Wait for all the threads to exit.
|
||||||
|
for thr in handles {
|
||||||
|
thr.join().expect("thread join");
|
||||||
|
}
|
||||||
|
image
|
||||||
|
}
|
||||||
|
|
||||||
|
/// This renderer use rayon to split each row into a seperate thread. It
|
||||||
|
/// seems to have more consistent performance than worker pool, equally fast
|
||||||
|
/// as WP at WP's fastest. The downside is the flame graph looks a mess. A
|
||||||
|
/// strength over `render_parallel_one_tread_per_core` is that it doesn't
|
||||||
|
/// require `Camera` and `World` to be cloneable.
|
||||||
|
fn render_parallel_rayon(&self, w: &World) -> Canvas {
|
||||||
|
let image_mu = Mutex::new(Canvas::new(self.hsize, self.vsize, BLACK));
|
||||||
|
|
||||||
|
(0..self.vsize).into_par_iter().for_each(|y| {
|
||||||
|
let mut row_image = Canvas::new(self.hsize, 1, BLACK);
|
||||||
|
for x in 0..self.hsize {
|
||||||
|
let color = self.sample(w, x, y);
|
||||||
|
row_image.set(x, 0, color);
|
||||||
|
}
|
||||||
|
// TODO(wathiede): create a row based setter for memcpying the row as a whole.
|
||||||
|
let mut image = image_mu.lock().expect("failed to lock image mutex");
|
||||||
|
for x in 0..self.hsize {
|
||||||
|
image.set(x, y, row_image.get(x, 0));
|
||||||
|
}
|
||||||
|
});
|
||||||
|
image_mu
|
||||||
|
.into_inner()
|
||||||
|
.expect("failed to get image out of mutex")
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Reference render implementation from the book. Single threaded, nothing fancy.
|
||||||
|
fn render_serial(&self, w: &World) -> Canvas {
|
||||||
let mut image = Canvas::new(self.hsize, self.vsize, BLACK);
|
let mut image = Canvas::new(self.hsize, self.vsize, BLACK);
|
||||||
for y in 0..self.vsize {
|
for y in 0..self.vsize {
|
||||||
for x in 0..self.hsize {
|
for x in 0..self.hsize {
|
||||||
let ray = self.ray_for_pixel(x, y);
|
let color = self.sample(w, x, y);
|
||||||
let color = w.color_at(&ray);
|
|
||||||
image.set(x, y, color);
|
image.set(x, y, color);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
image
|
image
|
||||||
}
|
}
|
||||||
|
|
||||||
|
fn sample(&self, w: &World, x: usize, y: usize) -> Color {
|
||||||
|
if self.samples_per_pixel > 0 {
|
||||||
|
let color = self
|
||||||
|
.supersample_rays_for_pixel(x, y, self.samples_per_pixel)
|
||||||
|
.iter()
|
||||||
|
.map(|ray| w.color_at(ray, MAX_DEPTH_RECURSION))
|
||||||
|
.fold(BLACK, |acc, c| acc + c);
|
||||||
|
color / self.samples_per_pixel as Float
|
||||||
|
} else {
|
||||||
|
let ray = self.ray_for_pixel(x, y);
|
||||||
|
w.color_at(&ray, MAX_DEPTH_RECURSION)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
fn render_worker_task(
|
||||||
|
c: &Camera,
|
||||||
|
w: &World,
|
||||||
|
input_chan: Arc<Mutex<Receiver<Request>>>,
|
||||||
|
output_chan: &SyncSender<Response>,
|
||||||
|
) {
|
||||||
|
loop {
|
||||||
|
let job = { input_chan.lock().unwrap().recv() };
|
||||||
|
match job {
|
||||||
|
Err(_) => {
|
||||||
|
// From the docs:
|
||||||
|
// "The recv operation can only fail if the sending half of a
|
||||||
|
// channel (or sync_channel) is disconnected, implying that no
|
||||||
|
// further messages will ever be received."
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
Ok(req) => match req {
|
||||||
|
Request::Line { width, y } => {
|
||||||
|
let mut pixels = Canvas::new(width, 1, BLACK);
|
||||||
|
for x in 0..width {
|
||||||
|
let color = c.sample(w, x, y);
|
||||||
|
pixels.set(x, 0, color);
|
||||||
|
}
|
||||||
|
output_chan
|
||||||
|
.send(Response::Line { y, pixels })
|
||||||
|
.expect("failed to send pixel response");
|
||||||
|
}
|
||||||
|
},
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
mod tests {
|
||||||
|
use crate::{
|
||||||
|
camera::Camera,
|
||||||
|
float::consts::PI,
|
||||||
|
matrices::Matrix4x4,
|
||||||
|
transformations::view_transform,
|
||||||
|
tuples::{point, vector},
|
||||||
|
world::World,
|
||||||
|
Float, EPSILON,
|
||||||
|
};
|
||||||
|
#[test]
|
||||||
|
fn new() {
|
||||||
|
let hsize = 160;
|
||||||
|
let vsize = 120;
|
||||||
|
let field_of_view = PI / 2.;
|
||||||
|
let c = Camera::new(hsize, vsize, field_of_view);
|
||||||
|
assert_eq!(c.hsize(), 160);
|
||||||
|
assert_eq!(c.vsize(), 120);
|
||||||
|
assert_eq!(c.transform(), Matrix4x4::identity());
|
||||||
|
|
||||||
|
// Pixel size for a horizontal canvas.
|
||||||
|
let c = Camera::new(200, 150, PI / 2.);
|
||||||
|
assert!((c.pixel_size() - 0.010).abs() < EPSILON);
|
||||||
|
|
||||||
|
// Pixel size for a horizontal canvas.
|
||||||
|
let c = Camera::new(150, 200, PI / 2.);
|
||||||
|
assert!((c.pixel_size() - 0.010).abs() < EPSILON);
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn ray_for_pixel() {
|
||||||
|
// Constructing a ray through the center of the canvas.
|
||||||
|
let c = Camera::new(201, 101, PI / 2.);
|
||||||
|
let r = c.ray_for_pixel(100, 50);
|
||||||
|
assert_eq!(r.origin, point(0., 0., 0.));
|
||||||
|
assert_eq!(r.direction, vector(0., 0., -1.));
|
||||||
|
|
||||||
|
// Constructing a ray through the corner of the canvas.
|
||||||
|
let c = Camera::new(201, 101, PI / 2.);
|
||||||
|
let r = c.ray_for_pixel(0, 0);
|
||||||
|
assert_eq!(r.origin, point(0., 0., 0.));
|
||||||
|
assert_eq!(r.direction, vector(0.66519, 0.33259, -0.66851));
|
||||||
|
|
||||||
|
// Constructing a ray when the camera is transformed.
|
||||||
|
let mut c = Camera::new(201, 101, PI / 2.);
|
||||||
|
c.set_transform(Matrix4x4::rotation_y(PI / 4.) * Matrix4x4::translation(0., -2., 5.));
|
||||||
|
let r = c.ray_for_pixel(100, 50);
|
||||||
|
assert_eq!(r.origin, point(0., 2., -5.));
|
||||||
|
assert_eq!(
|
||||||
|
r.direction,
|
||||||
|
vector((2. as Float).sqrt() / 2., 0., -(2. as Float).sqrt() / 2.)
|
||||||
|
);
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn render() {
|
||||||
|
// Rendering a world with a camera.
|
||||||
|
let w = World::test_world();
|
||||||
|
let mut c = Camera::new(11, 11, PI / 2.);
|
||||||
|
let from = point(0., 0., -5.);
|
||||||
|
let to = point(0., 0., 0.);
|
||||||
|
let up = vector(0., 1., 0.);
|
||||||
|
c.set_transform(view_transform(from, to, up));
|
||||||
|
let image = c.render(&w);
|
||||||
|
assert_eq!(image.get(5, 5), [0.38066, 0.47583, 0.2855].into());
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,6 +1,4 @@
|
|||||||
use std::fs::File;
|
use std::{fs::File, io::BufWriter, path::Path};
|
||||||
use std::io::BufWriter;
|
|
||||||
use std::path::Path;
|
|
||||||
|
|
||||||
use png;
|
use png;
|
||||||
use thiserror::Error;
|
use thiserror::Error;
|
||||||
@@ -48,7 +46,7 @@ impl Canvas {
|
|||||||
{
|
{
|
||||||
let path = Path::new(path.as_ref());
|
let path = Path::new(path.as_ref());
|
||||||
let file = File::create(path)?;
|
let file = File::create(path)?;
|
||||||
let ref mut w = BufWriter::new(file);
|
let w = &mut BufWriter::new(file);
|
||||||
|
|
||||||
let mut encoder = png::Encoder::new(w, self.width as u32, self.height as u32);
|
let mut encoder = png::Encoder::new(w, self.width as u32, self.height as u32);
|
||||||
encoder.set_color(png::ColorType::RGB);
|
encoder.set_color(png::ColorType::RGB);
|
||||||
|
|||||||
@@ -2,114 +2,51 @@ use std::ops::Index;
|
|||||||
|
|
||||||
use crate::{
|
use crate::{
|
||||||
rays::Ray,
|
rays::Ray,
|
||||||
spheres::Sphere,
|
shapes::Shape,
|
||||||
tuples::{dot, Tuple},
|
tuples::{dot, reflect, Tuple},
|
||||||
|
Float, EPSILON,
|
||||||
};
|
};
|
||||||
|
|
||||||
#[derive(Debug, Clone, PartialEq)]
|
#[derive(Debug, Clone)]
|
||||||
pub struct Intersection<'i> {
|
pub struct Intersection<'i> {
|
||||||
pub t: f32,
|
pub t: Float,
|
||||||
pub object: &'i Sphere,
|
pub object: &'i Shape,
|
||||||
|
}
|
||||||
|
impl<'i> PartialEq for Intersection<'i> {
|
||||||
|
fn eq(&self, rhs: &Intersection) -> bool {
|
||||||
|
((self.t - rhs.t).abs() < EPSILON) && (self.object == rhs.object)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl<'i> Intersection<'i> {
|
impl<'i> Intersection<'i> {
|
||||||
/// Create new `Intersection` at the given `t` that hits the given `object`.
|
/// Create new `Intersection` at the given `t` that hits the given `object`.
|
||||||
///
|
pub fn new(t: Float, object: &Shape) -> Intersection {
|
||||||
/// # Examples
|
|
||||||
/// ```
|
|
||||||
/// use rtchallenge::{intersections::Intersection, spheres::Sphere};
|
|
||||||
///
|
|
||||||
/// // An intersection ecapsulates t and object.
|
|
||||||
/// let s = Sphere::default();
|
|
||||||
/// let i = Intersection::new(3.5, &s);
|
|
||||||
/// assert_eq!(i.t, 3.5);
|
|
||||||
/// assert_eq!(i.object, &s);
|
|
||||||
/// ```
|
|
||||||
pub fn new(t: f32, object: &Sphere) -> Intersection {
|
|
||||||
Intersection { t, object }
|
Intersection { t, object }
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Aggregates `Intersection`s.
|
/// Aggregates `Intersection`s.
|
||||||
///
|
|
||||||
/// # Examples
|
|
||||||
/// ```
|
|
||||||
/// use rtchallenge::{
|
|
||||||
/// intersections::{Intersection, Intersections},
|
|
||||||
/// rays::Ray,
|
|
||||||
/// spheres::{intersect, Sphere},
|
|
||||||
/// tuples::Tuple,
|
|
||||||
/// };
|
|
||||||
///
|
|
||||||
/// let s = Sphere::default();
|
|
||||||
/// let i1 = Intersection::new(1., &s);
|
|
||||||
/// let i2 = Intersection::new(2., &s);
|
|
||||||
/// let xs = Intersections::new(vec![i1, i2]);
|
|
||||||
/// assert_eq!(xs.len(), 2);
|
|
||||||
/// assert_eq!(xs[0].t, 1.);
|
|
||||||
/// assert_eq!(xs[1].t, 2.);
|
|
||||||
///
|
|
||||||
/// let r = Ray::new(Tuple::point(0., 0., -5.), Tuple::vector(0., 0., 1.));
|
|
||||||
/// let xs = intersect(&s, &r);
|
|
||||||
/// assert_eq!(xs.len(), 2);
|
|
||||||
/// assert_eq!(xs[0].object, &s);
|
|
||||||
/// assert_eq!(xs[1].object, &s);
|
|
||||||
/// ```
|
|
||||||
#[derive(Debug, Default, PartialEq)]
|
#[derive(Debug, Default, PartialEq)]
|
||||||
pub struct Intersections<'i>(Vec<Intersection<'i>>);
|
pub struct Intersections<'i>(Vec<Intersection<'i>>);
|
||||||
|
|
||||||
|
/// Create an [Intersections] from a single [Intersection] to aid in tests.
|
||||||
|
impl<'i> From<Intersection<'i>> for Intersections<'i> {
|
||||||
|
fn from(i: Intersection<'i>) -> Intersections<'i> {
|
||||||
|
Intersections::new(vec![i])
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
impl<'i> Intersections<'i> {
|
impl<'i> Intersections<'i> {
|
||||||
pub fn new(xs: Vec<Intersection<'i>>) -> Intersections {
|
pub fn new(xs: Vec<Intersection<'i>>) -> Intersections {
|
||||||
Intersections(xs)
|
Intersections(xs)
|
||||||
}
|
}
|
||||||
|
pub fn is_empty(&self) -> bool {
|
||||||
|
self.0.is_empty()
|
||||||
|
}
|
||||||
pub fn len(&self) -> usize {
|
pub fn len(&self) -> usize {
|
||||||
self.0.len()
|
self.0.len()
|
||||||
}
|
}
|
||||||
/// Finds nearest hit for this collection of intersections.
|
/// Finds nearest hit for this collection of intersections.
|
||||||
///
|
|
||||||
/// # Examples
|
|
||||||
/// ```
|
|
||||||
/// use rtchallenge::{
|
|
||||||
/// intersections::{Intersection, Intersections},
|
|
||||||
/// rays::Ray,
|
|
||||||
/// spheres::{intersect, Sphere},
|
|
||||||
/// tuples::Tuple,
|
|
||||||
/// };
|
|
||||||
///
|
|
||||||
/// // The hit, when all intersections have positive t.
|
|
||||||
/// let s = Sphere::default();
|
|
||||||
/// let i1 = Intersection::new(1., &s);
|
|
||||||
/// let i2 = Intersection::new(2., &s);
|
|
||||||
/// let xs = Intersections::new(vec![i2, i1.clone()]);
|
|
||||||
/// let i = xs.hit();
|
|
||||||
/// assert_eq!(i, Some(&i1));
|
|
||||||
///
|
|
||||||
/// // The hit, when some intersections have negative t.
|
|
||||||
/// let s = Sphere::default();
|
|
||||||
/// let i1 = Intersection::new(-1., &s);
|
|
||||||
/// let i2 = Intersection::new(1., &s);
|
|
||||||
/// let xs = Intersections::new(vec![i2.clone(), i1]);
|
|
||||||
/// let i = xs.hit();
|
|
||||||
/// assert_eq!(i, Some(&i2));
|
|
||||||
///
|
|
||||||
/// // The hit, when all intersections have negative t.
|
|
||||||
/// let s = Sphere::default();
|
|
||||||
/// let i1 = Intersection::new(-2., &s);
|
|
||||||
/// let i2 = Intersection::new(-1., &s);
|
|
||||||
/// let xs = Intersections::new(vec![i2, i1]);
|
|
||||||
/// let i = xs.hit();
|
|
||||||
/// assert_eq!(i, None);
|
|
||||||
///
|
|
||||||
/// // The hit is always the lowest nonnegative intersection.
|
|
||||||
/// let s = Sphere::default();
|
|
||||||
/// let i1 = Intersection::new(5., &s);
|
|
||||||
/// let i2 = Intersection::new(7., &s);
|
|
||||||
/// let i3 = Intersection::new(-3., &s);
|
|
||||||
/// let i4 = Intersection::new(2., &s);
|
|
||||||
/// let xs = Intersections::new(vec![i1, i2, i3, i4.clone()]);
|
|
||||||
/// let i = xs.hit();
|
|
||||||
/// assert_eq!(i, Some(&i4));
|
|
||||||
/// ```
|
|
||||||
pub fn hit(&self) -> Option<&Intersection> {
|
pub fn hit(&self) -> Option<&Intersection> {
|
||||||
self.0.iter().filter(|i| i.t > 0.).min_by(|i1, i2| {
|
self.0.iter().filter(|i| i.t > 0.).min_by(|i1, i2| {
|
||||||
i1.t.partial_cmp(&i2.t)
|
i1.t.partial_cmp(&i2.t)
|
||||||
@@ -134,70 +71,363 @@ impl<'i> Index<usize> for Intersections<'i> {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[derive(Debug)]
|
||||||
pub struct PrecomputedData<'i> {
|
pub struct PrecomputedData<'i> {
|
||||||
pub t: f32,
|
pub t: Float,
|
||||||
pub object: &'i Sphere,
|
pub object: &'i Shape,
|
||||||
pub point: Tuple,
|
pub point: Tuple,
|
||||||
|
pub under_point: Tuple,
|
||||||
|
pub over_point: Tuple,
|
||||||
pub eyev: Tuple,
|
pub eyev: Tuple,
|
||||||
pub normalv: Tuple,
|
pub normalv: Tuple,
|
||||||
|
pub reflectv: Tuple,
|
||||||
pub inside: bool,
|
pub inside: bool,
|
||||||
|
pub n1: Float,
|
||||||
|
pub n2: Float,
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Precomputes data common to all intersections.
|
/// Precomputes data common to all intersections.
|
||||||
///
|
pub fn prepare_computations<'i>(
|
||||||
/// # Examples
|
hit: &'i Intersection,
|
||||||
/// ```
|
r: &Ray,
|
||||||
/// use rtchallenge::{
|
xs: &Intersections,
|
||||||
/// intersections::{prepare_computations, Intersection, Intersections},
|
) -> PrecomputedData<'i> {
|
||||||
/// rays::Ray,
|
let point = r.position(hit.t);
|
||||||
/// spheres::{intersect, Sphere},
|
let normalv = hit.object.normal_at(point);
|
||||||
/// tuples::Tuple,
|
|
||||||
/// };
|
|
||||||
///
|
|
||||||
/// // Precomputing the state of an intersection.
|
|
||||||
/// let r = Ray::new(Tuple::point(0., 0., -5.), Tuple::vector(0., 0., 1.));
|
|
||||||
/// let shape = Sphere::default();
|
|
||||||
/// let i = Intersection::new(4., &shape);
|
|
||||||
/// let comps = prepare_computations(&i, &r);
|
|
||||||
/// assert_eq!(comps.t, i.t);
|
|
||||||
/// assert_eq!(comps.object, i.object);
|
|
||||||
/// assert_eq!(comps.point, Tuple::point(0., 0., -1.));
|
|
||||||
/// assert_eq!(comps.eyev, Tuple::vector(0., 0., -1.));
|
|
||||||
/// assert_eq!(comps.normalv, Tuple::vector(0., 0., -1.));
|
|
||||||
///
|
|
||||||
/// // The hit, when an intersection occurs on the outside.
|
|
||||||
/// let r = Ray::new(Tuple::point(0., 0., -5.), Tuple::vector(0., 0., 1.));
|
|
||||||
/// let shape = Sphere::default();
|
|
||||||
/// let i = Intersection::new(4., &shape);
|
|
||||||
/// let comps = prepare_computations(&i, &r);
|
|
||||||
/// assert_eq!(comps.inside, false);
|
|
||||||
///
|
|
||||||
/// // The hit, when an intersection occurs on the inside.
|
|
||||||
/// let r = Ray::new(Tuple::point(0., 0., 0.), Tuple::vector(0., 0., 1.));
|
|
||||||
/// let shape = Sphere::default();
|
|
||||||
/// let i = Intersection::new(1., &shape);
|
|
||||||
/// let comps = prepare_computations(&i, &r);
|
|
||||||
/// assert_eq!(comps.point, Tuple::point(0., 0., 1.));
|
|
||||||
/// assert_eq!(comps.eyev, Tuple::vector(0., 0., -1.));
|
|
||||||
/// assert_eq!(comps.inside, true);
|
|
||||||
//// // Normal would have been (0, 0, 1), but is inverted when inside.
|
|
||||||
/// assert_eq!(comps.normalv, Tuple::vector(0., 0., -1.));
|
|
||||||
/// ```
|
|
||||||
pub fn prepare_computations<'i>(i: &'i Intersection, r: &Ray) -> PrecomputedData<'i> {
|
|
||||||
let point = r.position(i.t);
|
|
||||||
let normalv = i.object.normal_at(point);
|
|
||||||
let eyev = -r.direction;
|
let eyev = -r.direction;
|
||||||
let (inside, normalv) = if dot(normalv, eyev) < 0. {
|
let (inside, normalv) = if dot(normalv, eyev) < 0. {
|
||||||
(true, -normalv)
|
(true, -normalv)
|
||||||
} else {
|
} else {
|
||||||
(false, normalv)
|
(false, normalv)
|
||||||
};
|
};
|
||||||
|
let reflectv = reflect(r.direction, normalv);
|
||||||
|
|
||||||
|
let mut n1 = -1.;
|
||||||
|
let mut n2 = -1.;
|
||||||
|
let mut containers: Vec<&Shape> = Vec::new();
|
||||||
|
for i in xs.0.iter() {
|
||||||
|
if hit == i {
|
||||||
|
if containers.is_empty() {
|
||||||
|
n1 = 1.;
|
||||||
|
} else {
|
||||||
|
n1 = containers.last().unwrap().material.refractive_index;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
match containers.iter().position(|o| o == &i.object) {
|
||||||
|
Some(idx) => {
|
||||||
|
containers.remove(idx);
|
||||||
|
}
|
||||||
|
None => containers.push(i.object),
|
||||||
|
}
|
||||||
|
|
||||||
|
if hit == i {
|
||||||
|
if containers.is_empty() {
|
||||||
|
n2 = 1.;
|
||||||
|
} else {
|
||||||
|
n2 = containers.last().unwrap().material.refractive_index;
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
let over_point = point + normalv * EPSILON;
|
||||||
|
let under_point = point - normalv * EPSILON;
|
||||||
PrecomputedData {
|
PrecomputedData {
|
||||||
t: i.t,
|
t: hit.t,
|
||||||
object: i.object,
|
object: hit.object,
|
||||||
point,
|
point,
|
||||||
|
over_point,
|
||||||
|
under_point,
|
||||||
normalv,
|
normalv,
|
||||||
|
reflectv,
|
||||||
inside,
|
inside,
|
||||||
eyev,
|
eyev,
|
||||||
|
n1,
|
||||||
|
n2,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Compute Schlick approximation to Fresnel's equations.
|
||||||
|
pub fn schlick(comps: &PrecomputedData) -> Float {
|
||||||
|
// Find the cosine of the angle between the eye and normal vectors.
|
||||||
|
let mut cos = dot(comps.eyev, comps.normalv);
|
||||||
|
// Total internal reflection can only occur if n1 > n2.
|
||||||
|
if comps.n1 > comps.n2 {
|
||||||
|
let n = comps.n1 / comps.n2;
|
||||||
|
let sin2_t = n * n * (1. - cos * cos);
|
||||||
|
if sin2_t > 1. {
|
||||||
|
return 1.;
|
||||||
|
}
|
||||||
|
// Compute cosine of theta_t using trig identity.
|
||||||
|
let cos_t = (1. - sin2_t).sqrt();
|
||||||
|
// When n1 > n2 use cos(theta_t) instead.
|
||||||
|
cos = cos_t;
|
||||||
|
}
|
||||||
|
|
||||||
|
let r0 = (comps.n1 - comps.n2) / (comps.n1 + comps.n2);
|
||||||
|
let r0 = r0 * r0;
|
||||||
|
r0 + (1. - r0) * (1. - cos) * (1. - cos) * (1. - cos) * (1. - cos) * (1. - cos)
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
mod tests {
|
||||||
|
use crate::{
|
||||||
|
intersections::{prepare_computations, schlick, Intersection, Intersections},
|
||||||
|
materials::MaterialBuilder,
|
||||||
|
matrices::{scaling, translation},
|
||||||
|
rays::Ray,
|
||||||
|
shapes::{glass_sphere, intersect, Shape},
|
||||||
|
tuples::{point, vector},
|
||||||
|
Float, EPSILON,
|
||||||
|
};
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn intersection() {
|
||||||
|
// An intersection ecapsulates t and object.
|
||||||
|
let s = Shape::sphere();
|
||||||
|
let i = Intersection::new(3.5, &s);
|
||||||
|
assert_eq!(i.t, 3.5);
|
||||||
|
assert_eq!(i.object, &s);
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn intersections() {
|
||||||
|
let s = Shape::sphere();
|
||||||
|
let i1 = Intersection::new(1., &s);
|
||||||
|
let i2 = Intersection::new(2., &s);
|
||||||
|
let xs = Intersections::new(vec![i1, i2]);
|
||||||
|
assert_eq!(xs.len(), 2);
|
||||||
|
assert_eq!(xs[0].t, 1.);
|
||||||
|
assert_eq!(xs[1].t, 2.);
|
||||||
|
|
||||||
|
let r = Ray::new(point(0., 0., -5.), vector(0., 0., 1.));
|
||||||
|
let xs = intersect(&s, &r);
|
||||||
|
assert_eq!(xs.len(), 2);
|
||||||
|
assert_eq!(xs[0].object, &s);
|
||||||
|
assert_eq!(xs[1].object, &s);
|
||||||
|
}
|
||||||
|
|
||||||
|
mod hit {
|
||||||
|
use super::*;
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn when_all_intersections_have_positive_t() {
|
||||||
|
let s = Shape::sphere();
|
||||||
|
let i1 = Intersection::new(1., &s);
|
||||||
|
let i2 = Intersection::new(2., &s);
|
||||||
|
let xs = Intersections::new(vec![i2, i1.clone()]);
|
||||||
|
let i = xs.hit();
|
||||||
|
assert_eq!(i, Some(&i1));
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn when_some_intersections_have_negative_t() {
|
||||||
|
let s = Shape::sphere();
|
||||||
|
let i1 = Intersection::new(-1., &s);
|
||||||
|
let i2 = Intersection::new(1., &s);
|
||||||
|
let xs = Intersections::new(vec![i2.clone(), i1]);
|
||||||
|
let i = xs.hit();
|
||||||
|
assert_eq!(i, Some(&i2));
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn when_all_intersections_have_negative_t() {
|
||||||
|
let s = Shape::sphere();
|
||||||
|
let i1 = Intersection::new(-2., &s);
|
||||||
|
let i2 = Intersection::new(-1., &s);
|
||||||
|
let xs = Intersections::new(vec![i2, i1]);
|
||||||
|
let i = xs.hit();
|
||||||
|
assert_eq!(i, None);
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn always_the_lowest_nonnegative_intersection() {
|
||||||
|
let s = Shape::sphere();
|
||||||
|
let i1 = Intersection::new(5., &s);
|
||||||
|
let i2 = Intersection::new(7., &s);
|
||||||
|
let i3 = Intersection::new(-3., &s);
|
||||||
|
let i4 = Intersection::new(2., &s);
|
||||||
|
let xs = Intersections::new(vec![i1, i2, i3, i4.clone()]);
|
||||||
|
let i = xs.hit();
|
||||||
|
assert_eq!(i, Some(&i4));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
mod prepare_computations {
|
||||||
|
use super::*;
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn precomputing_the_state_of_intersection() -> Result<(), Box<dyn std::error::Error>> {
|
||||||
|
let r = Ray::new(point(0., 0., -5.), vector(0., 0., 1.));
|
||||||
|
let shape = Shape::sphere();
|
||||||
|
let xs = Intersections::from(Intersection::new(4., &shape));
|
||||||
|
let comps = prepare_computations(&xs[0], &r, &xs);
|
||||||
|
assert_eq!(comps.t, xs[0].t);
|
||||||
|
assert_eq!(comps.object, xs[0].object);
|
||||||
|
assert_eq!(comps.point, point(0., 0., -1.));
|
||||||
|
assert_eq!(comps.eyev, vector(0., 0., -1.));
|
||||||
|
assert_eq!(comps.normalv, vector(0., 0., -1.));
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn hit_when_intersection_occurs_outside() -> Result<(), Box<dyn std::error::Error>> {
|
||||||
|
// The hit, when an intersection occurs on the outside.
|
||||||
|
let r = Ray::new(point(0., 0., -5.), vector(0., 0., 1.));
|
||||||
|
let shape = Shape::sphere();
|
||||||
|
let xs = Intersections::from(Intersection::new(4., &shape));
|
||||||
|
let comps = prepare_computations(&xs[0], &r, &xs);
|
||||||
|
assert_eq!(comps.inside, false);
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn hit_when_intersection_occurs_inside() -> Result<(), Box<dyn std::error::Error>> {
|
||||||
|
// The hit, when an intersection occurs on the inside.
|
||||||
|
let r = Ray::new(point(0., 0., 0.), vector(0., 0., 1.));
|
||||||
|
let shape = Shape::sphere();
|
||||||
|
let xs = Intersections::from(Intersection::new(1., &shape));
|
||||||
|
let comps = prepare_computations(&xs[0], &r, &xs);
|
||||||
|
assert_eq!(comps.point, point(0., 0., 1.));
|
||||||
|
assert_eq!(comps.eyev, vector(0., 0., -1.));
|
||||||
|
assert_eq!(comps.inside, true);
|
||||||
|
// Normal would have been (0, 0, 1), but is inverted when inside.
|
||||||
|
assert_eq!(comps.normalv, vector(0., 0., -1.));
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn hit_should_offset_the_point() -> Result<(), Box<dyn std::error::Error>> {
|
||||||
|
// The hit should offset the point.
|
||||||
|
let r = Ray::new(point(0., 0., -5.), vector(0., 0., 1.));
|
||||||
|
let mut shape = Shape::sphere();
|
||||||
|
shape.set_transform(translation(0., 0., 1.));
|
||||||
|
let xs = Intersections::from(Intersection::new(5., &shape));
|
||||||
|
let comps = prepare_computations(&xs[0], &r, &xs);
|
||||||
|
assert!(comps.over_point.z < -EPSILON / 2.);
|
||||||
|
assert!(comps.point.z > comps.over_point.z);
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn precomputing_the_reflection_vector() -> Result<(), Box<dyn std::error::Error>> {
|
||||||
|
// Precomputing the reflection vector.
|
||||||
|
let shape = Shape::plane();
|
||||||
|
let r = Ray::new(
|
||||||
|
point(0., 1., -1.),
|
||||||
|
vector(0., -(2 as Float).sqrt() / 2., (2 as Float).sqrt() / 2.),
|
||||||
|
);
|
||||||
|
let xs = Intersections::from(Intersection::new((2 as Float).sqrt(), &shape));
|
||||||
|
let comps = prepare_computations(&xs[0], &r, &xs);
|
||||||
|
assert_eq!(
|
||||||
|
comps.reflectv,
|
||||||
|
vector(0., (2 as Float).sqrt() / 2., (2 as Float).sqrt() / 2.)
|
||||||
|
);
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn finding_n1_and_n2() -> Result<(), Box<dyn std::error::Error>> {
|
||||||
|
// Finding n1 and n2 at various intersections.
|
||||||
|
let a = glass_sphere()
|
||||||
|
.transform(scaling(2., 2., 2.))
|
||||||
|
.material(
|
||||||
|
MaterialBuilder::default()
|
||||||
|
.transparency(1.)
|
||||||
|
.refractive_index(1.5)
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.build()?;
|
||||||
|
let b = glass_sphere()
|
||||||
|
.transform(translation(0., 0., -0.25))
|
||||||
|
.material(
|
||||||
|
MaterialBuilder::default()
|
||||||
|
.transparency(1.)
|
||||||
|
.refractive_index(2.)
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.build()?;
|
||||||
|
let c = glass_sphere()
|
||||||
|
.transform(translation(0., 0., 0.25))
|
||||||
|
.material(
|
||||||
|
MaterialBuilder::default()
|
||||||
|
.transparency(1.)
|
||||||
|
.refractive_index(2.5)
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.build()?;
|
||||||
|
let r = Ray::new(point(0., 0., -4.), vector(0., 0., 1.));
|
||||||
|
let xs = Intersections::new(vec![
|
||||||
|
Intersection::new(2., &a),
|
||||||
|
Intersection::new(2.75, &b),
|
||||||
|
Intersection::new(3.25, &c),
|
||||||
|
Intersection::new(4.75, &b),
|
||||||
|
Intersection::new(5.25, &c),
|
||||||
|
Intersection::new(6., &a),
|
||||||
|
]);
|
||||||
|
for (index, n1, n2) in &[
|
||||||
|
(0, 1.0, 1.5),
|
||||||
|
(1, 1.5, 2.0),
|
||||||
|
(2, 2.0, 2.5),
|
||||||
|
(3, 2.5, 2.5),
|
||||||
|
(4, 2.5, 1.5),
|
||||||
|
(5, 1.5, 1.0),
|
||||||
|
] {
|
||||||
|
let comps = prepare_computations(&xs[*index], &r, &xs);
|
||||||
|
assert_eq!(comps.n1, *n1);
|
||||||
|
assert_eq!(comps.n2, *n2);
|
||||||
|
}
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn under_point_offset_below_surface() -> Result<(), Box<dyn std::error::Error>> {
|
||||||
|
// The under point is offset below the surface.
|
||||||
|
let r = Ray::new(point(0., 0., -5.), vector(0., 0., 1.));
|
||||||
|
let shape = glass_sphere().transform(translation(0., 0., 1.)).build()?;
|
||||||
|
let xs = Intersections::from(Intersection::new(5., &shape));
|
||||||
|
let comps = prepare_computations(&xs[0], &r, &xs);
|
||||||
|
assert!(comps.under_point.z > EPSILON / 2.);
|
||||||
|
assert!(comps.point.z < comps.under_point.z);
|
||||||
|
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
mod schlick {
|
||||||
|
use super::*;
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn under_total_reflection() -> Result<(), Box<dyn std::error::Error>> {
|
||||||
|
let shape = glass_sphere().build()?;
|
||||||
|
let r = Ray::new(point(0., 0., (2 as Float).sqrt() / 2.), vector(0., 1., 0.));
|
||||||
|
let xs = Intersections::new(vec![
|
||||||
|
Intersection::new(-(2 as Float).sqrt() / 2., &shape),
|
||||||
|
Intersection::new((2 as Float).sqrt() / 2., &shape),
|
||||||
|
]);
|
||||||
|
let comps = prepare_computations(&xs[1], &r, &xs);
|
||||||
|
let reflectance = schlick(&comps);
|
||||||
|
assert_eq!(reflectance, 1.);
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn perpendicular_viewing_angle() -> Result<(), Box<dyn std::error::Error>> {
|
||||||
|
let shape = glass_sphere().build()?;
|
||||||
|
let r = Ray::new(point(0., 0., 0.), vector(0., 1., 0.));
|
||||||
|
let xs = Intersections::new(vec![
|
||||||
|
Intersection::new(-1., &shape),
|
||||||
|
Intersection::new(1., &shape),
|
||||||
|
]);
|
||||||
|
let comps = prepare_computations(&xs[1], &r, &xs);
|
||||||
|
let reflectance = schlick(&comps);
|
||||||
|
assert!((reflectance - 0.04).abs() < EPSILON);
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn small_angle_n2_greater_n1() -> Result<(), Box<dyn std::error::Error>> {
|
||||||
|
let shape = glass_sphere().build()?;
|
||||||
|
let r = Ray::new(point(0., 0.99, -2.), vector(0., 0., 1.));
|
||||||
|
let xs = Intersections::new(vec![Intersection::new(1.8589, &shape)]);
|
||||||
|
let comps = prepare_computations(&xs[0], &r, &xs);
|
||||||
|
let reflectance = schlick(&comps);
|
||||||
|
assert!((reflectance - 0.48873).abs() < EPSILON);
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -4,14 +4,52 @@ pub mod intersections;
|
|||||||
pub mod lights;
|
pub mod lights;
|
||||||
pub mod materials;
|
pub mod materials;
|
||||||
pub mod matrices;
|
pub mod matrices;
|
||||||
|
pub mod patterns;
|
||||||
pub mod rays;
|
pub mod rays;
|
||||||
pub mod spheres;
|
pub mod shapes;
|
||||||
pub mod transformations;
|
pub mod transformations;
|
||||||
pub mod tuples;
|
pub mod tuples;
|
||||||
pub mod world;
|
pub mod world;
|
||||||
|
|
||||||
/// Value considered close enough for PartialEq implementations.
|
/// Value considered close enough for PartialEq implementations.
|
||||||
pub const EPSILON: f32 = 0.00001;
|
pub const EPSILON: Float = 0.00001;
|
||||||
|
|
||||||
pub const BLACK: tuples::Color = tuples::Color::new(0., 0., 0.);
|
pub const BLACK: tuples::Color = tuples::Color::new(0., 0., 0.);
|
||||||
pub const WHITE: tuples::Color = tuples::Color::new(1., 1., 1.);
|
pub const WHITE: tuples::Color = tuples::Color::new(1., 1., 1.);
|
||||||
|
|
||||||
|
#[cfg(feature = "float-as-double")]
|
||||||
|
/// submodule to defined types, constants and methods when `Float` is defined as a `f64` using the
|
||||||
|
/// "float-as-double" cargo feature.
|
||||||
|
pub mod float {
|
||||||
|
pub use std::f64::*;
|
||||||
|
/// Alias of the `f64` type, to be used through out the codebase anywhere a default sized
|
||||||
|
/// `Float` is necessary.
|
||||||
|
pub type Float = f64;
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(not(feature = "float-as-double"))]
|
||||||
|
/// submodule to defined types, constants and methods when `Float` is defined as a `f32` when not using the
|
||||||
|
/// "float-as-double" cargo feature.
|
||||||
|
pub mod float {
|
||||||
|
pub use std::f32::*;
|
||||||
|
/// Alias of the `f32` type, to be used through out the codebase anywhere a default sized
|
||||||
|
/// `Float` is necessary.
|
||||||
|
pub type Float = f32;
|
||||||
|
}
|
||||||
|
|
||||||
|
pub use float::Float;
|
||||||
|
|
||||||
|
pub mod prelude {
|
||||||
|
pub use crate::{
|
||||||
|
camera::{Camera, CameraBuilder},
|
||||||
|
lights::{PointLight, PointLightBuilder},
|
||||||
|
materials::{Material, MaterialBuilder},
|
||||||
|
matrices::{identity, rotation_x, rotation_y, rotation_z, scaling, shearing, translation},
|
||||||
|
patterns::{checkers_pattern, gradient_pattern, ring_pattern, stripe_pattern},
|
||||||
|
shapes::{cube, plane, sphere, test_shape},
|
||||||
|
transformations::view_transform,
|
||||||
|
tuples::{point, vector, Color},
|
||||||
|
world::{World, WorldBuilder},
|
||||||
|
Float,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|||||||
@@ -1,33 +1,39 @@
|
|||||||
|
use derive_builder::Builder;
|
||||||
|
|
||||||
use crate::tuples::{Color, Tuple};
|
use crate::tuples::{Color, Tuple};
|
||||||
|
|
||||||
#[derive(Debug, PartialEq)]
|
#[derive(Builder, Clone, Debug, Default, PartialEq)]
|
||||||
|
#[builder(default)]
|
||||||
pub struct PointLight {
|
pub struct PointLight {
|
||||||
pub position: Tuple,
|
pub position: Tuple,
|
||||||
|
#[builder(setter(into))]
|
||||||
pub intensity: Color,
|
pub intensity: Color,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl PointLight {
|
impl PointLight {
|
||||||
/// Creates a new `PositionLight` at the given `position` and with the given
|
/// Creates a new `PositionLight` at the given `position` and with the given
|
||||||
/// `intensity`.
|
/// `intensity`.
|
||||||
///
|
pub fn new<C>(position: Tuple, intensity: C) -> PointLight
|
||||||
/// # Examples
|
where
|
||||||
/// ```
|
C: Into<Color>,
|
||||||
/// use rtchallenge::{
|
{
|
||||||
/// lights::PointLight,
|
|
||||||
/// tuples::{Color, Tuple},
|
|
||||||
/// WHITE,
|
|
||||||
/// };
|
|
||||||
///
|
|
||||||
/// let intensity = WHITE;
|
|
||||||
/// let position = Tuple::point(0., 0., 0.);
|
|
||||||
/// let light = PointLight::new(position, intensity);
|
|
||||||
/// assert_eq!(light.position, position);
|
|
||||||
/// assert_eq!(light.intensity, intensity);
|
|
||||||
/// ```
|
|
||||||
pub fn new(position: Tuple, intensity: Color) -> PointLight {
|
|
||||||
PointLight {
|
PointLight {
|
||||||
position,
|
position,
|
||||||
intensity,
|
intensity: intensity.into(),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
mod tests {
|
||||||
|
use crate::{lights::PointLight, tuples::point, WHITE};
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn new() {
|
||||||
|
let intensity = WHITE;
|
||||||
|
let position = point(0., 0., 0.);
|
||||||
|
let light = PointLight::new(position, intensity);
|
||||||
|
assert_eq!(light.position, position);
|
||||||
|
assert_eq!(light.intensity, intensity);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|||||||
@@ -1,106 +1,56 @@
|
|||||||
|
use derive_builder::Builder;
|
||||||
|
|
||||||
use crate::{
|
use crate::{
|
||||||
lights::PointLight,
|
lights::PointLight,
|
||||||
tuples::Color,
|
patterns::Pattern,
|
||||||
tuples::{dot, reflect, Tuple},
|
shapes::Shape,
|
||||||
BLACK, WHITE,
|
tuples::{dot, reflect, Color, Tuple},
|
||||||
|
Float, BLACK, WHITE,
|
||||||
};
|
};
|
||||||
#[derive(Debug, PartialEq, Clone)]
|
|
||||||
|
#[derive(Builder, Debug, PartialEq, Clone)]
|
||||||
|
#[builder(default)]
|
||||||
pub struct Material {
|
pub struct Material {
|
||||||
pub color: Color,
|
#[builder(setter(into))]
|
||||||
pub ambient: f32,
|
pub color: Pattern,
|
||||||
pub diffuse: f32,
|
pub ambient: Float,
|
||||||
pub specular: f32,
|
pub diffuse: Float,
|
||||||
pub shininess: f32,
|
pub specular: Float,
|
||||||
|
pub shininess: Float,
|
||||||
|
pub reflective: Float,
|
||||||
|
pub transparency: Float,
|
||||||
|
pub refractive_index: Float,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Default for Material {
|
impl Default for Material {
|
||||||
/// Creates the default material.
|
/// Creates the default material.
|
||||||
///
|
|
||||||
/// # Examples
|
|
||||||
/// ```
|
|
||||||
/// use rtchallenge::{materials::Material, tuples::Color, WHITE};
|
|
||||||
///
|
|
||||||
/// let m = Material::default();
|
|
||||||
/// assert_eq!(
|
|
||||||
/// m,
|
|
||||||
/// Material {
|
|
||||||
/// color: WHITE,
|
|
||||||
/// ambient: 0.1,
|
|
||||||
/// diffuse: 0.9,
|
|
||||||
/// specular: 0.9,
|
|
||||||
/// shininess: 200.,
|
|
||||||
/// }
|
|
||||||
/// );
|
|
||||||
/// ```
|
|
||||||
fn default() -> Material {
|
fn default() -> Material {
|
||||||
Material {
|
Material {
|
||||||
color: WHITE,
|
color: WHITE.into(),
|
||||||
ambient: 0.1,
|
ambient: 0.1,
|
||||||
diffuse: 0.9,
|
diffuse: 0.9,
|
||||||
specular: 0.9,
|
specular: 0.9,
|
||||||
shininess: 200.,
|
shininess: 200.,
|
||||||
|
reflective: 0.0,
|
||||||
|
transparency: 0.0,
|
||||||
|
refractive_index: 1.0,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Compute lighting contributions using the Phong reflection model.
|
/// Compute lighting contributions using the Phong reflection model.
|
||||||
///
|
|
||||||
/// # Examples
|
|
||||||
/// ```
|
|
||||||
/// use rtchallenge::{
|
|
||||||
/// lights::PointLight,
|
|
||||||
/// materials::{lighting, Material},
|
|
||||||
/// tuples::{Color, Tuple},
|
|
||||||
/// WHITE,
|
|
||||||
/// };
|
|
||||||
///
|
|
||||||
/// let m = Material::default();
|
|
||||||
/// let position = Tuple::point(0., 0., 0.);
|
|
||||||
///
|
|
||||||
/// // Lighting with the eye between the light and the surface.
|
|
||||||
/// let eyev = Tuple::vector(0., 0., -1.);
|
|
||||||
/// let normalv = Tuple::vector(0., 0., -1.);
|
|
||||||
/// let light = PointLight::new(Tuple::point(0., 0., -10.), WHITE);
|
|
||||||
/// let result = lighting(&m, &light, position, eyev, normalv);
|
|
||||||
/// assert_eq!(result, Color::new(1.9, 1.9, 1.9));
|
|
||||||
///
|
|
||||||
/// // Lighting with the eye between the light and the surface, eye offset 45°.
|
|
||||||
/// let eyev = Tuple::vector(0., 2_f32.sqrt() / 2., -2_f32.sqrt() / 2.);
|
|
||||||
/// let normalv = Tuple::vector(0., 0., -1.);
|
|
||||||
/// let light = PointLight::new(Tuple::point(0., 0., -10.), WHITE);
|
|
||||||
/// let result = lighting(&m, &light, position, eyev, normalv);
|
|
||||||
/// assert_eq!(result, WHITE);
|
|
||||||
///
|
|
||||||
/// // Lighting with the eye opposite surface, light offset 45°.
|
|
||||||
/// let eyev = Tuple::vector(0., 0., -1.);
|
|
||||||
/// let normalv = Tuple::vector(0., 0., -1.);
|
|
||||||
/// let light = PointLight::new(Tuple::point(0., 10., -10.), WHITE);
|
|
||||||
/// let result = lighting(&m, &light, position, eyev, normalv);
|
|
||||||
/// assert_eq!(result, Color::new(0.7364, 0.7364, 0.7364));
|
|
||||||
///
|
|
||||||
/// // Lighting with the eye in the path of the reflection vector.
|
|
||||||
/// let eyev = Tuple::vector(0., -2_f32.sqrt() / 2., -2_f32.sqrt() / 2.);
|
|
||||||
/// let normalv = Tuple::vector(0., 0., -1.);
|
|
||||||
/// let light = PointLight::new(Tuple::point(0., 10., -10.), WHITE);
|
|
||||||
/// let result = lighting(&m, &light, position, eyev, normalv);
|
|
||||||
/// assert_eq!(result, Color::new(1.6363853, 1.6363853, 1.6363853));
|
|
||||||
///
|
|
||||||
/// // Lighting with the light behind the surface.
|
|
||||||
/// let eyev = Tuple::vector(0., 0., -1.);
|
|
||||||
/// let normalv = Tuple::vector(0., 0., -1.);
|
|
||||||
/// let light = PointLight::new(Tuple::point(0., 0., 10.), WHITE);
|
|
||||||
/// let result = lighting(&m, &light, position, eyev, normalv);
|
|
||||||
/// assert_eq!(result, Color::new(0.1, 0.1, 0.1));
|
|
||||||
/// ```
|
|
||||||
pub fn lighting(
|
pub fn lighting(
|
||||||
material: &Material,
|
material: &Material,
|
||||||
|
object: &Shape,
|
||||||
light: &PointLight,
|
light: &PointLight,
|
||||||
point: Tuple,
|
point: Tuple,
|
||||||
eyev: Tuple,
|
eyev: Tuple,
|
||||||
normalv: Tuple,
|
normalv: Tuple,
|
||||||
|
in_shadow: bool,
|
||||||
) -> Color {
|
) -> Color {
|
||||||
// Combine the surface color with the light's color.
|
// Combine the surface color with the light's color.
|
||||||
let effective_color = material.color * light.intensity;
|
let color = material.color.pattern_at_object(object, point);
|
||||||
|
let effective_color = color * light.intensity;
|
||||||
// Find the direciton of the light source.
|
// Find the direciton of the light source.
|
||||||
let lightv = (light.position - point).normalize();
|
let lightv = (light.position - point).normalize();
|
||||||
// Compute the ambient distribution.
|
// Compute the ambient distribution.
|
||||||
@@ -128,5 +78,159 @@ pub fn lighting(
|
|||||||
};
|
};
|
||||||
(diffuse, specular)
|
(diffuse, specular)
|
||||||
};
|
};
|
||||||
ambient + diffuse + specular
|
if in_shadow {
|
||||||
|
ambient
|
||||||
|
} else {
|
||||||
|
ambient + diffuse + specular
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
mod tests {
|
||||||
|
use crate::{materials::Material, WHITE};
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn default() {
|
||||||
|
let m = Material::default();
|
||||||
|
assert_eq!(
|
||||||
|
m,
|
||||||
|
Material {
|
||||||
|
color: WHITE.into(),
|
||||||
|
ambient: 0.1,
|
||||||
|
diffuse: 0.9,
|
||||||
|
specular: 0.9,
|
||||||
|
shininess: 200.,
|
||||||
|
reflective: 0.0,
|
||||||
|
transparency: 0.0,
|
||||||
|
refractive_index: 1.0,
|
||||||
|
}
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
mod lighting {
|
||||||
|
use crate::{
|
||||||
|
lights::PointLight,
|
||||||
|
materials::{lighting, Material},
|
||||||
|
patterns::{Pattern, BLACK_PAT, WHITE_PAT},
|
||||||
|
shapes::Shape,
|
||||||
|
tuples::{point, vector, Color},
|
||||||
|
Float, BLACK, WHITE,
|
||||||
|
};
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn eye_between_light_and_surface() {
|
||||||
|
let in_shadow = false;
|
||||||
|
let m = Material::default();
|
||||||
|
let position = point(0., 0., 0.);
|
||||||
|
let object = Shape::sphere();
|
||||||
|
|
||||||
|
// Lighting with the eye between the light and the surface.
|
||||||
|
let eyev = vector(0., 0., -1.);
|
||||||
|
let normalv = vector(0., 0., -1.);
|
||||||
|
let light = PointLight::new(point(0., 0., -10.), WHITE);
|
||||||
|
let result = lighting(&m, &object, &light, position, eyev, normalv, in_shadow);
|
||||||
|
assert_eq!(result, Color::new(1.9, 1.9, 1.9));
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn eye_between_light_and_surface_offset_45() {
|
||||||
|
// Lighting with the eye between the light and the surface, eye offset 45°.
|
||||||
|
let in_shadow = false;
|
||||||
|
let m = Material::default();
|
||||||
|
let position = point(0., 0., 0.);
|
||||||
|
let object = Shape::sphere();
|
||||||
|
let eyev = vector(0., (2. as Float).sqrt() / 2., -(2. as Float).sqrt() / 2.);
|
||||||
|
let normalv = vector(0., 0., -1.);
|
||||||
|
let light = PointLight::new(point(0., 0., -10.), WHITE);
|
||||||
|
let result = lighting(&m, &object, &light, position, eyev, normalv, in_shadow);
|
||||||
|
assert_eq!(result, WHITE);
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn eye_opposite_surface_light_offset_45() {
|
||||||
|
// Lighting with the eye opposite surface, light offset 45°.
|
||||||
|
let in_shadow = false;
|
||||||
|
let m = Material::default();
|
||||||
|
let position = point(0., 0., 0.);
|
||||||
|
let object = Shape::sphere();
|
||||||
|
let eyev = vector(0., 0., -1.);
|
||||||
|
let normalv = vector(0., 0., -1.);
|
||||||
|
let light = PointLight::new(point(0., 10., -10.), WHITE);
|
||||||
|
let result = lighting(&m, &object, &light, position, eyev, normalv, in_shadow);
|
||||||
|
assert_eq!(result, Color::new(0.7364, 0.7364, 0.7364));
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn eye_in_path_of_reflection_vector() {
|
||||||
|
// Lighting with the eye in the path of the reflection vector.
|
||||||
|
let in_shadow = false;
|
||||||
|
let m = Material::default();
|
||||||
|
let position = point(0., 0., 0.);
|
||||||
|
let object = Shape::sphere();
|
||||||
|
let eyev = vector(0., -(2.0 as Float).sqrt() / 2., -(2.0 as Float).sqrt() / 2.);
|
||||||
|
let normalv = vector(0., 0., -1.);
|
||||||
|
let light = PointLight::new(point(0., 10., -10.), WHITE);
|
||||||
|
let result = lighting(&m, &object, &light, position, eyev, normalv, in_shadow);
|
||||||
|
assert_eq!(result, Color::new(1.63639, 1.63639, 1.63639));
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn light_behind_surface() {
|
||||||
|
// Lighting with the light behind the surface.
|
||||||
|
let in_shadow = false;
|
||||||
|
let m = Material::default();
|
||||||
|
let position = point(0., 0., 0.);
|
||||||
|
let object = Shape::sphere();
|
||||||
|
let eyev = vector(0., 0., -1.);
|
||||||
|
let normalv = vector(0., 0., -1.);
|
||||||
|
let light = PointLight::new(point(0., 0., 10.), WHITE);
|
||||||
|
let result = lighting(&m, &object, &light, position, eyev, normalv, in_shadow);
|
||||||
|
assert_eq!(result, Color::new(0.1, 0.1, 0.1));
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn surface_in_shadow() {
|
||||||
|
// Lighting with the surface in shadow.
|
||||||
|
let m = Material::default();
|
||||||
|
let position = point(0., 0., 0.);
|
||||||
|
let object = Shape::sphere();
|
||||||
|
let in_shadow = true;
|
||||||
|
let eyev = vector(0., 0., -1.);
|
||||||
|
let normalv = vector(0., 0., -1.);
|
||||||
|
let light = PointLight::new(point(0., 0., -10.), WHITE);
|
||||||
|
let result = lighting(&m, &object, &light, position, eyev, normalv, in_shadow);
|
||||||
|
assert_eq!(result, Color::new(0.1, 0.1, 0.1));
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn pattern_applied() {
|
||||||
|
// Lighting with a pattern applied.
|
||||||
|
let object = Shape::sphere();
|
||||||
|
let m = Material {
|
||||||
|
color: Pattern::stripe(WHITE_PAT, BLACK_PAT),
|
||||||
|
ambient: 1.,
|
||||||
|
diffuse: 0.,
|
||||||
|
specular: 0.,
|
||||||
|
..Material::default()
|
||||||
|
};
|
||||||
|
let eyev = vector(0., 0., -1.);
|
||||||
|
let normalv = vector(0., 0., -1.);
|
||||||
|
let light = PointLight::new(point(0., 0., -10.), WHITE);
|
||||||
|
let c1 = lighting(
|
||||||
|
&m,
|
||||||
|
&object,
|
||||||
|
&light,
|
||||||
|
point(0.9, 0., 0.),
|
||||||
|
eyev,
|
||||||
|
normalv,
|
||||||
|
false,
|
||||||
|
);
|
||||||
|
let c2 = lighting(
|
||||||
|
&m,
|
||||||
|
&object,
|
||||||
|
&light,
|
||||||
|
point(1.1, 0., 0.),
|
||||||
|
eyev,
|
||||||
|
normalv,
|
||||||
|
false,
|
||||||
|
);
|
||||||
|
assert_eq!(c1, WHITE);
|
||||||
|
assert_eq!(c2, BLACK);
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
File diff suppressed because it is too large
Load Diff
444
rtchallenge/src/patterns.rs
Normal file
444
rtchallenge/src/patterns.rs
Normal file
@@ -0,0 +1,444 @@
|
|||||||
|
use derive_builder::Builder;
|
||||||
|
|
||||||
|
use crate::{
|
||||||
|
matrices::Matrix4x4,
|
||||||
|
shapes::Shape,
|
||||||
|
tuples::{Color, Tuple},
|
||||||
|
BLACK, WHITE,
|
||||||
|
};
|
||||||
|
|
||||||
|
pub const BLACK_PAT: Pattern = Pattern {
|
||||||
|
color: ColorMapper::Constant(BLACK),
|
||||||
|
transform: Matrix4x4::identity(),
|
||||||
|
inverse_transform: Matrix4x4::identity(),
|
||||||
|
};
|
||||||
|
pub const WHITE_PAT: Pattern = Pattern {
|
||||||
|
color: ColorMapper::Constant(WHITE),
|
||||||
|
transform: Matrix4x4::identity(),
|
||||||
|
inverse_transform: Matrix4x4::identity(),
|
||||||
|
};
|
||||||
|
|
||||||
|
#[derive(Debug, PartialEq, Clone)]
|
||||||
|
pub enum ColorMapper {
|
||||||
|
/// TestPattern the color returned is the pattern space point after going through world->object and object->pattern space translation.
|
||||||
|
TestPattern,
|
||||||
|
/// Solid color, the same sampled every where.
|
||||||
|
Constant(Color),
|
||||||
|
/// Pattern that alternates between the given patterns along each unit of the X-axis. The
|
||||||
|
/// strip extends infinitely in the positive and negative Y and Z axes.
|
||||||
|
Stripe { a: Box<Pattern>, b: Box<Pattern> },
|
||||||
|
/// Linear blend between `a` and `b` along the X-axis.
|
||||||
|
Gradient { a: Box<Pattern>, b: Box<Pattern> },
|
||||||
|
/// Bullseye pattern in the XZ plane.
|
||||||
|
Ring { a: Box<Pattern>, b: Box<Pattern> },
|
||||||
|
/// Traditional ray tracer tile floor pattern.
|
||||||
|
Checkers { a: Box<Pattern>, b: Box<Pattern> },
|
||||||
|
}
|
||||||
|
|
||||||
|
impl From<Color> for ColorMapper {
|
||||||
|
fn from(c: Color) -> ColorMapper {
|
||||||
|
ColorMapper::Constant(c)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl From<Color> for Box<Pattern> {
|
||||||
|
fn from(c: Color) -> Box<Pattern> {
|
||||||
|
Box::new(Pattern {
|
||||||
|
color: ColorMapper::Constant(c),
|
||||||
|
..Pattern::default()
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Builder, Debug, PartialEq, Clone)]
|
||||||
|
#[builder(default, pattern = "immutable")]
|
||||||
|
pub struct Pattern {
|
||||||
|
pub color: ColorMapper,
|
||||||
|
transform: Matrix4x4,
|
||||||
|
#[builder(private, default = "self.default_inverse_transform()?")]
|
||||||
|
inverse_transform: Matrix4x4,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl PatternBuilder {
|
||||||
|
fn default_inverse_transform(&self) -> Result<Matrix4x4, String> {
|
||||||
|
Ok(self.transform.unwrap_or(Matrix4x4::identity()).inverse())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Default for Pattern {
|
||||||
|
fn default() -> Pattern {
|
||||||
|
Pattern {
|
||||||
|
color: ColorMapper::Constant(WHITE),
|
||||||
|
transform: Matrix4x4::identity(),
|
||||||
|
inverse_transform: Matrix4x4::identity(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Creates a [Pattern] with a color type of [ColorMapper::Constant] from the given [Color]
|
||||||
|
impl<C> From<C> for Pattern
|
||||||
|
where
|
||||||
|
C: Into<Color>,
|
||||||
|
{
|
||||||
|
fn from(c: C) -> Self {
|
||||||
|
Pattern {
|
||||||
|
color: ColorMapper::Constant(c.into()),
|
||||||
|
..Pattern::default()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Builder for creating a material pattern used for testing. The color returned is the pattern space point
|
||||||
|
/// after going through world->object and object->pattern space translation.
|
||||||
|
pub fn test_pattern() -> PatternBuilder {
|
||||||
|
PatternBuilder::default().color(ColorMapper::TestPattern)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Builder for creating a material pattern that alternates between the given colors along each unit of the
|
||||||
|
/// X-axis. The strip extends infinitely in the positive and negative Y and Z axes.
|
||||||
|
pub fn stripe_pattern(a: Pattern, b: Pattern) -> PatternBuilder {
|
||||||
|
PatternBuilder::default().color(ColorMapper::Stripe {
|
||||||
|
a: a.into(),
|
||||||
|
b: b.into(),
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Builder for creating a material pattern that gradually blends between the given colors along
|
||||||
|
/// the X-axis.
|
||||||
|
pub fn gradient_pattern(a: Pattern, b: Pattern) -> PatternBuilder {
|
||||||
|
PatternBuilder::default().color(ColorMapper::Gradient {
|
||||||
|
a: a.into(),
|
||||||
|
b: b.into(),
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Builder for creating a material pattern that alternates between the given colors in a ring
|
||||||
|
/// shape in the XZ plane.
|
||||||
|
pub fn ring_pattern(a: Pattern, b: Pattern) -> PatternBuilder {
|
||||||
|
PatternBuilder::default().color(ColorMapper::Ring {
|
||||||
|
a: a.into(),
|
||||||
|
b: b.into(),
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Builder for creating a material pattern that alternates between the given colors along the X, Y
|
||||||
|
/// and Z pattern. Creates traditional ray tracer tile floor pattern.
|
||||||
|
pub fn checkers_pattern(a: Pattern, b: Pattern) -> PatternBuilder {
|
||||||
|
PatternBuilder::default().color(ColorMapper::Checkers {
|
||||||
|
a: a.into(),
|
||||||
|
b: b.into(),
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Generic implementation for mapping points to colors according to the given [ColorMapper].
|
||||||
|
impl Pattern {
|
||||||
|
/// Create a pattern used for testing. The color returned is the pattern space point
|
||||||
|
/// after going through world->object and object->pattern space translation.
|
||||||
|
pub fn test() -> Pattern {
|
||||||
|
Pattern {
|
||||||
|
color: ColorMapper::TestPattern,
|
||||||
|
..Pattern::default()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Create a pattern that alternates between the given Patterns along each unit of the
|
||||||
|
/// X-axis. The strip extends infinitely in the positive and negative Y and Z axes.
|
||||||
|
pub fn stripe(a: Pattern, b: Pattern) -> Pattern {
|
||||||
|
Pattern {
|
||||||
|
color: ColorMapper::Stripe {
|
||||||
|
a: a.into(),
|
||||||
|
b: b.into(),
|
||||||
|
},
|
||||||
|
..Pattern::default()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Create a pattern that gradually blends between the given Patterns along the X-axis.
|
||||||
|
pub fn gradient(a: Pattern, b: Pattern) -> Pattern {
|
||||||
|
Pattern {
|
||||||
|
color: ColorMapper::Gradient {
|
||||||
|
a: a.into(),
|
||||||
|
b: b.into(),
|
||||||
|
},
|
||||||
|
..Pattern::default()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Create a pattern that alternates between the given Patterns in a ring in the XZ plane.
|
||||||
|
pub fn ring(a: Pattern, b: Pattern) -> Pattern {
|
||||||
|
Pattern {
|
||||||
|
color: ColorMapper::Ring {
|
||||||
|
a: a.into(),
|
||||||
|
b: b.into(),
|
||||||
|
},
|
||||||
|
..Pattern::default()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Create a pattern that alternates between the given Patterns along the X, Y and Z axis.
|
||||||
|
pub fn checkers(a: Pattern, b: Pattern) -> Pattern {
|
||||||
|
Pattern {
|
||||||
|
color: ColorMapper::Checkers {
|
||||||
|
a: a.into(),
|
||||||
|
b: b.into(),
|
||||||
|
},
|
||||||
|
..Pattern::default()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Sample the color at the given point in untranslated object space.
|
||||||
|
pub fn pattern_at(&self, object_point: Tuple) -> Color {
|
||||||
|
let point = self.inverse_transform * object_point;
|
||||||
|
match &self.color {
|
||||||
|
ColorMapper::TestPattern => [point.x, point.y, point.z].into(),
|
||||||
|
ColorMapper::Constant(c) => *c,
|
||||||
|
ColorMapper::Stripe { a, b } => {
|
||||||
|
let x = point.x.floor();
|
||||||
|
if x % 2. == 0. {
|
||||||
|
a.pattern_at(point)
|
||||||
|
} else {
|
||||||
|
b.pattern_at(point)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
ColorMapper::Gradient { a, b } => {
|
||||||
|
let a = a.pattern_at(point);
|
||||||
|
let b = b.pattern_at(point);
|
||||||
|
let distance = b - a;
|
||||||
|
let fraction = point.x - point.x.floor();
|
||||||
|
a + distance * fraction
|
||||||
|
}
|
||||||
|
ColorMapper::Ring { a, b } => {
|
||||||
|
let a = a.pattern_at(point);
|
||||||
|
let b = b.pattern_at(point);
|
||||||
|
let px = point.x;
|
||||||
|
let pz = point.z;
|
||||||
|
if (px * px + pz * pz).sqrt().floor() % 2. == 0. {
|
||||||
|
a
|
||||||
|
} else {
|
||||||
|
b
|
||||||
|
}
|
||||||
|
}
|
||||||
|
ColorMapper::Checkers { a, b } => {
|
||||||
|
let a = a.pattern_at(point);
|
||||||
|
let b = b.pattern_at(point);
|
||||||
|
let d = point.x.floor() + point.y.floor() + point.z.floor();
|
||||||
|
if d % 2. == 0. {
|
||||||
|
a
|
||||||
|
} else {
|
||||||
|
b
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
/// Sample the color at the given world point on the given object.
|
||||||
|
/// This function respects the object and the pattern's transform matrix.
|
||||||
|
pub fn pattern_at_object(&self, object: &Shape, world_point: Tuple) -> Color {
|
||||||
|
let object_point = object.inverse_transform() * world_point;
|
||||||
|
self.pattern_at(object_point)
|
||||||
|
}
|
||||||
|
pub fn transform(&self) -> Matrix4x4 {
|
||||||
|
self.transform
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn inverse_transform(&self) -> Matrix4x4 {
|
||||||
|
self.inverse_transform
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn set_transform(&mut self, t: Matrix4x4) {
|
||||||
|
self.transform = t;
|
||||||
|
self.inverse_transform = t.inverse();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
mod tests {
|
||||||
|
use crate::{
|
||||||
|
matrices::identity,
|
||||||
|
patterns::{ColorMapper, Pattern, BLACK_PAT, WHITE_PAT},
|
||||||
|
BLACK, WHITE,
|
||||||
|
};
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_create() {
|
||||||
|
let pattern = Pattern::test();
|
||||||
|
assert_eq!(pattern.transform(), identity());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn stripe_create() {
|
||||||
|
let pattern = Pattern::stripe(BLACK_PAT, WHITE_PAT);
|
||||||
|
assert_eq!(
|
||||||
|
pattern.color,
|
||||||
|
ColorMapper::Stripe {
|
||||||
|
a: BLACK.into(),
|
||||||
|
b: WHITE.into(),
|
||||||
|
}
|
||||||
|
);
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn gradient_create() {
|
||||||
|
println!("SHOE ME SOMETHING");
|
||||||
|
println!("* * * 1");
|
||||||
|
let pattern = Pattern::gradient(BLACK_PAT, WHITE_PAT);
|
||||||
|
println!("* * * 2");
|
||||||
|
assert_eq!(
|
||||||
|
pattern.color,
|
||||||
|
ColorMapper::Gradient {
|
||||||
|
a: BLACK.into(),
|
||||||
|
b: WHITE.into(),
|
||||||
|
}
|
||||||
|
);
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn ring_create() {
|
||||||
|
let pattern = Pattern::ring(BLACK_PAT, WHITE_PAT);
|
||||||
|
assert_eq!(
|
||||||
|
pattern.color,
|
||||||
|
ColorMapper::Ring {
|
||||||
|
a: BLACK.into(),
|
||||||
|
b: WHITE.into()
|
||||||
|
}
|
||||||
|
);
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn checkers_create() {
|
||||||
|
let pattern = Pattern::checkers(BLACK_PAT, WHITE_PAT);
|
||||||
|
assert_eq!(
|
||||||
|
pattern.color,
|
||||||
|
ColorMapper::Checkers {
|
||||||
|
a: BLACK.into(),
|
||||||
|
b: WHITE.into()
|
||||||
|
}
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
mod pattern_at {
|
||||||
|
use super::*;
|
||||||
|
use crate::tuples::point;
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_returns_coordinates() {
|
||||||
|
// A test returns the pattern space coordinates of the point.
|
||||||
|
let pattern = Pattern::test();
|
||||||
|
let p = point(1., 2., 3.);
|
||||||
|
assert_eq!(pattern.pattern_at(p), [p.x, p.y, p.z].into());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn stripe_alternates_between_two_colors() {
|
||||||
|
// A stripe alternates between two colors.
|
||||||
|
let pattern = Pattern::stripe(WHITE_PAT, BLACK_PAT);
|
||||||
|
for (p, want) in &[
|
||||||
|
// A stripe pattern is constant in y.
|
||||||
|
(point(0., 0., 0.), WHITE),
|
||||||
|
(point(0., 1., 0.), WHITE),
|
||||||
|
(point(0., 2., 0.), WHITE),
|
||||||
|
// A stripe pattern is constant in z.
|
||||||
|
(point(0., 0., 0.), WHITE),
|
||||||
|
(point(0., 0., 1.), WHITE),
|
||||||
|
(point(0., 0., 2.), WHITE),
|
||||||
|
// A stripe pattern alternates in z.
|
||||||
|
(point(0., 0., 0.), WHITE),
|
||||||
|
(point(0.9, 0., 0.), WHITE),
|
||||||
|
(point(1., 0., 0.), BLACK),
|
||||||
|
(point(-0.1, 0., 0.), BLACK),
|
||||||
|
(point(-1., 0., 0.), BLACK),
|
||||||
|
(point(-1.1, 0., 0.), WHITE),
|
||||||
|
] {
|
||||||
|
assert_eq!(pattern.pattern_at(*p), *want, "{:?}", p);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn gradient_linearly_interpolates_between_colors() {
|
||||||
|
// A gradient linearly interpolates between two colors.
|
||||||
|
let pattern = Pattern::gradient(WHITE_PAT, BLACK_PAT);
|
||||||
|
assert_eq!(pattern.pattern_at(point(0., 0., 0.)), WHITE);
|
||||||
|
assert_eq!(
|
||||||
|
pattern.pattern_at(point(0.25, 0., 0.)),
|
||||||
|
[0.75, 0.75, 0.75].into()
|
||||||
|
);
|
||||||
|
assert_eq!(
|
||||||
|
pattern.pattern_at(point(0.5, 0., 0.)),
|
||||||
|
[0.5, 0.5, 0.5].into()
|
||||||
|
);
|
||||||
|
assert_eq!(
|
||||||
|
pattern.pattern_at(point(0.75, 0., 0.)),
|
||||||
|
[0.25, 0.25, 0.25].into()
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn ring_extend_in_x_and_z() {
|
||||||
|
// A ring should extend both in x and z.
|
||||||
|
let pattern = Pattern::ring(WHITE_PAT, BLACK_PAT);
|
||||||
|
assert_eq!(pattern.pattern_at(point(0., 0., 0.)), WHITE);
|
||||||
|
assert_eq!(pattern.pattern_at(point(1., 0., 0.)), BLACK);
|
||||||
|
assert_eq!(pattern.pattern_at(point(0., 0., 1.)), BLACK);
|
||||||
|
// 0.708 is slight more than 2.sqrt()/2.
|
||||||
|
assert_eq!(pattern.pattern_at(point(0.708, 0., 0.708)), BLACK);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn checkers_repeat_along_x_axis() {
|
||||||
|
// Checkers should repeat along X-axis.
|
||||||
|
let pattern = Pattern::checkers(WHITE_PAT, BLACK_PAT);
|
||||||
|
assert_eq!(pattern.pattern_at(point(0., 0., 0.)), WHITE);
|
||||||
|
assert_eq!(pattern.pattern_at(point(0.99, 0., 0.)), WHITE);
|
||||||
|
assert_eq!(pattern.pattern_at(point(1.01, 0., 0.)), BLACK);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn checkers_repeat_along_y_axis() {
|
||||||
|
// Checkers should repeat along Y-axis.
|
||||||
|
let pattern = Pattern::checkers(WHITE_PAT, BLACK_PAT);
|
||||||
|
assert_eq!(pattern.pattern_at(point(0., 0., 0.)), WHITE);
|
||||||
|
assert_eq!(pattern.pattern_at(point(0., 0.99, 0.)), WHITE);
|
||||||
|
assert_eq!(pattern.pattern_at(point(0., 1.01, 0.)), BLACK);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn checkers_repeat_along_z_axis() {
|
||||||
|
// Checkers should repeat along Z-axis.
|
||||||
|
let pattern = Pattern::checkers(WHITE_PAT, BLACK_PAT);
|
||||||
|
assert_eq!(pattern.pattern_at(point(0., 0., 0.)), WHITE);
|
||||||
|
assert_eq!(pattern.pattern_at(point(0., 0., 0.99)), WHITE);
|
||||||
|
assert_eq!(pattern.pattern_at(point(0., 0., 1.01)), BLACK);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
mod pattern_at_object {
|
||||||
|
use std::error::Error;
|
||||||
|
|
||||||
|
use crate::{
|
||||||
|
matrices::scaling,
|
||||||
|
patterns::{stripe_pattern, BLACK_PAT, WHITE_PAT},
|
||||||
|
shapes::{Shape, ShapeBuilder},
|
||||||
|
tuples::point,
|
||||||
|
WHITE,
|
||||||
|
};
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn stripes_with_an_object_transformation() -> Result<(), Box<dyn Error>> {
|
||||||
|
// Stripes with an object transformation.
|
||||||
|
let object = ShapeBuilder::sphere()
|
||||||
|
.transform(scaling(2., 2., 2.))
|
||||||
|
.build()?;
|
||||||
|
let pattern = stripe_pattern(WHITE_PAT, BLACK_PAT).build()?;
|
||||||
|
let c = pattern.pattern_at_object(&object, point(1.5, 0., 0.));
|
||||||
|
assert_eq!(c, WHITE);
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn stripes_with_a_pattern_transformation() -> Result<(), Box<dyn Error>> {
|
||||||
|
// Stripes with a pattern transformation.
|
||||||
|
let object = Shape::sphere();
|
||||||
|
let mut pattern = stripe_pattern(WHITE_PAT, BLACK_PAT).build()?;
|
||||||
|
pattern.set_transform(scaling(2., 2., 2.));
|
||||||
|
let c = pattern.pattern_at_object(&object, point(1.5, 0., 0.));
|
||||||
|
assert_eq!(c, WHITE);
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -1,6 +1,7 @@
|
|||||||
use crate::{matrices::Matrix4x4, tuples::Tuple};
|
use crate::{matrices::Matrix4x4, tuples::Tuple, Float};
|
||||||
|
|
||||||
/// Rays have an origin and a direction. This datatype is the 'ray' in 'raytracer'.
|
/// Rays have an origin and a direction. This datatype is the 'ray' in 'raytracer'.
|
||||||
|
#[derive(Debug, Default, Clone, PartialEq)]
|
||||||
pub struct Ray {
|
pub struct Ray {
|
||||||
pub origin: Tuple,
|
pub origin: Tuple,
|
||||||
pub direction: Tuple,
|
pub direction: Tuple,
|
||||||
@@ -9,17 +10,6 @@ pub struct Ray {
|
|||||||
impl Ray {
|
impl Ray {
|
||||||
/// Create a ray with the given origin point and direction vector.
|
/// Create a ray with the given origin point and direction vector.
|
||||||
/// Will panic if origin not a point or direction not a vector.
|
/// Will panic if origin not a point or direction not a vector.
|
||||||
///
|
|
||||||
/// # Examples
|
|
||||||
/// ```
|
|
||||||
/// use rtchallenge::{rays::Ray, tuples::Tuple};
|
|
||||||
///
|
|
||||||
/// let origin = Tuple::point(1., 2., 3.);
|
|
||||||
/// let direction = Tuple::vector(4., 5., 6.);
|
|
||||||
/// let r = Ray::new(origin, direction);
|
|
||||||
/// assert_eq!(r.origin, origin);
|
|
||||||
/// assert_eq!(r.direction, direction);
|
|
||||||
/// ```
|
|
||||||
pub fn new(origin: Tuple, direction: Tuple) -> Ray {
|
pub fn new(origin: Tuple, direction: Tuple) -> Ray {
|
||||||
assert!(origin.is_point(), "Ray origin must be a point");
|
assert!(origin.is_point(), "Ray origin must be a point");
|
||||||
assert!(direction.is_vector(), "Ray direction must be a vector");
|
assert!(direction.is_vector(), "Ray direction must be a vector");
|
||||||
@@ -27,41 +17,11 @@ impl Ray {
|
|||||||
}
|
}
|
||||||
|
|
||||||
/// Compute a point from the given distance along the `Ray`.
|
/// Compute a point from the given distance along the `Ray`.
|
||||||
///
|
pub fn position(&self, t: Float) -> Tuple {
|
||||||
/// # Examples
|
|
||||||
/// ```
|
|
||||||
/// use rtchallenge::{rays::Ray, tuples::Tuple};
|
|
||||||
///
|
|
||||||
/// let r = Ray::new(Tuple::point(2., 3., 4.), Tuple::vector(1., 0., 0.));
|
|
||||||
/// assert_eq!(r.position(0.), Tuple::point(2., 3., 4.));
|
|
||||||
/// assert_eq!(r.position(1.), Tuple::point(3., 3., 4.));
|
|
||||||
/// assert_eq!(r.position(-1.), Tuple::point(1., 3., 4.));
|
|
||||||
/// assert_eq!(r.position(2.5), Tuple::point(4.5, 3., 4.));
|
|
||||||
/// ```
|
|
||||||
pub fn position(&self, t: f32) -> Tuple {
|
|
||||||
self.origin + self.direction * t
|
self.origin + self.direction * t
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Apply Matrix4x4 transforms to Ray.
|
/// Apply Matrix4x4 transforms to Ray.
|
||||||
///
|
|
||||||
/// # Examples
|
|
||||||
/// ```
|
|
||||||
/// use rtchallenge::{matrices::Matrix4x4, rays::Ray, tuples::Tuple};
|
|
||||||
///
|
|
||||||
/// // Translating a ray
|
|
||||||
/// let r = Ray::new(Tuple::point(1., 2., 3.), Tuple::vector(0., 1., 0.));
|
|
||||||
/// let m = Matrix4x4::translation(3., 4., 5.);
|
|
||||||
/// let r2 = r.transform(m);
|
|
||||||
/// assert_eq!(r2.origin, Tuple::point(4., 6., 8.));
|
|
||||||
/// assert_eq!(r2.direction, Tuple::vector(0., 1., 0.));
|
|
||||||
///
|
|
||||||
/// // Scaling a ray
|
|
||||||
/// let r = Ray::new(Tuple::point(1., 2., 3.), Tuple::vector(0., 1., 0.));
|
|
||||||
/// let m = Matrix4x4::scaling(2., 3., 4.);
|
|
||||||
/// let r2 = r.transform(m);
|
|
||||||
/// assert_eq!(r2.origin, Tuple::point(2., 6., 12.));
|
|
||||||
/// assert_eq!(r2.direction, Tuple::vector(0., 3., 0.));
|
|
||||||
/// ```
|
|
||||||
pub fn transform(&self, m: Matrix4x4) -> Ray {
|
pub fn transform(&self, m: Matrix4x4) -> Ray {
|
||||||
Ray {
|
Ray {
|
||||||
origin: m * self.origin,
|
origin: m * self.origin,
|
||||||
@@ -69,3 +29,46 @@ impl Ray {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
mod tests {
|
||||||
|
use crate::{
|
||||||
|
matrices::Matrix4x4,
|
||||||
|
rays::Ray,
|
||||||
|
tuples::{point, vector},
|
||||||
|
};
|
||||||
|
#[test]
|
||||||
|
fn create() {
|
||||||
|
let origin = point(1., 2., 3.);
|
||||||
|
let direction = vector(4., 5., 6.);
|
||||||
|
let r = Ray::new(origin, direction);
|
||||||
|
assert_eq!(r.origin, origin);
|
||||||
|
assert_eq!(r.direction, direction);
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn position() {
|
||||||
|
let r = Ray::new(point(2., 3., 4.), vector(1., 0., 0.));
|
||||||
|
assert_eq!(r.position(0.), point(2., 3., 4.));
|
||||||
|
assert_eq!(r.position(1.), point(3., 3., 4.));
|
||||||
|
assert_eq!(r.position(-1.), point(1., 3., 4.));
|
||||||
|
assert_eq!(r.position(2.5), point(4.5, 3., 4.));
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn transform_translating_ray() {
|
||||||
|
// Translating a ray
|
||||||
|
let r = Ray::new(point(1., 2., 3.), vector(0., 1., 0.));
|
||||||
|
let m = Matrix4x4::translation(3., 4., 5.);
|
||||||
|
let r2 = r.transform(m);
|
||||||
|
assert_eq!(r2.origin, point(4., 6., 8.));
|
||||||
|
assert_eq!(r2.direction, vector(0., 1., 0.));
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn transform_scaling_ray() {
|
||||||
|
// Scaling a ray
|
||||||
|
let r = Ray::new(point(1., 2., 3.), vector(0., 1., 0.));
|
||||||
|
let m = Matrix4x4::scaling(2., 3., 4.);
|
||||||
|
let r2 = r.transform(m);
|
||||||
|
assert_eq!(r2.origin, point(2., 6., 12.));
|
||||||
|
assert_eq!(r2.direction, vector(0., 3., 0.));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|||||||
825
rtchallenge/src/shapes.rs
Normal file
825
rtchallenge/src/shapes.rs
Normal file
@@ -0,0 +1,825 @@
|
|||||||
|
use std::sync::{Arc, Mutex};
|
||||||
|
|
||||||
|
use derive_builder::Builder;
|
||||||
|
|
||||||
|
use crate::{
|
||||||
|
intersections::Intersections,
|
||||||
|
materials::{Material, MaterialBuilder},
|
||||||
|
matrices::Matrix4x4,
|
||||||
|
rays::Ray,
|
||||||
|
tuples::Tuple,
|
||||||
|
};
|
||||||
|
|
||||||
|
#[derive(Default, PartialEq, Debug, Clone)]
|
||||||
|
pub struct TestData {
|
||||||
|
pub saved_ray: Option<Ray>,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug, Clone)]
|
||||||
|
pub enum Geometry {
|
||||||
|
/// Shape with predictable normals useful for debugging.
|
||||||
|
TestShape(Arc<Mutex<TestData>>),
|
||||||
|
/// Sphere represents the unit-sphere (radius of unit 1.) at the origin 0., 0., 0.
|
||||||
|
Sphere,
|
||||||
|
/// Flat surface that extends infinitely in the XZ axes.
|
||||||
|
Plane,
|
||||||
|
/// AABB cube at origin from -1,1 in each direction.
|
||||||
|
Cube,
|
||||||
|
/// Takes shape from children held within.
|
||||||
|
Group(Vec<Shape>),
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Default for Geometry {
|
||||||
|
fn default() -> Geometry {
|
||||||
|
Geometry::Sphere
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl PartialEq for Geometry {
|
||||||
|
fn eq(&self, rhs: &Geometry) -> bool {
|
||||||
|
use Geometry::*;
|
||||||
|
match (self, rhs) {
|
||||||
|
(TestShape(l), TestShape(r)) => *l.lock().unwrap() == *r.lock().unwrap(),
|
||||||
|
(Sphere, Sphere) => true,
|
||||||
|
(Plane, Plane) => true,
|
||||||
|
(Cube, Cube) => true,
|
||||||
|
(Group(v1), Group(v2)) => v1 == v2,
|
||||||
|
_ => false,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Shape represents visible objects. A signal instance of Shape can generically represent one of
|
||||||
|
/// many different shapes based on the value of it's geometry field. Users chose the shape by
|
||||||
|
/// calling the appropriate constructor, i.e. [Shape::sphere].
|
||||||
|
#[derive(Builder, Debug, Clone, PartialEq)]
|
||||||
|
#[builder(default, pattern = "owned", build_fn(skip))]
|
||||||
|
pub struct Shape {
|
||||||
|
transform: Matrix4x4,
|
||||||
|
inverse_transform: Matrix4x4,
|
||||||
|
pub material: Material,
|
||||||
|
geometry: Geometry,
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Short hand for creating a ShapeBuilder with a plane geometry.
|
||||||
|
pub fn plane() -> ShapeBuilder {
|
||||||
|
ShapeBuilder::plane()
|
||||||
|
}
|
||||||
|
/// Short hand for creating a ShapeBuilder with a sphere geometry.
|
||||||
|
pub fn sphere() -> ShapeBuilder {
|
||||||
|
ShapeBuilder::sphere()
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Short hand for creating a ShapeBuilder with a test shape geometry.
|
||||||
|
pub fn test_shape() -> ShapeBuilder {
|
||||||
|
ShapeBuilder::test_shape()
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Short hand for creating a ShapeBuilder with a cube geometry.
|
||||||
|
pub fn cube() -> ShapeBuilder {
|
||||||
|
ShapeBuilder::cube()
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Short hand for creating a ShapeBuilder with a group geometry.
|
||||||
|
pub fn group() -> ShapeBuilder {
|
||||||
|
ShapeBuilder::group()
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Helper for producing a sphere with a glassy material.
|
||||||
|
pub fn glass_sphere() -> ShapeBuilder {
|
||||||
|
ShapeBuilder::sphere().material(
|
||||||
|
MaterialBuilder::default()
|
||||||
|
.transparency(1.)
|
||||||
|
.refractive_index(1.5)
|
||||||
|
.build()
|
||||||
|
.unwrap(),
|
||||||
|
)
|
||||||
|
}
|
||||||
|
impl ShapeBuilder {
|
||||||
|
/// Short hand for creating a ShapeBuilder with a plane geometry.
|
||||||
|
pub fn plane() -> ShapeBuilder {
|
||||||
|
ShapeBuilder::default().geometry(Geometry::Plane)
|
||||||
|
}
|
||||||
|
/// Short hand for creating a ShapeBuilder with a sphere geometry.
|
||||||
|
pub fn sphere() -> ShapeBuilder {
|
||||||
|
ShapeBuilder::default().geometry(Geometry::Sphere)
|
||||||
|
}
|
||||||
|
/// Short hand for creating a ShapeBuilder with a test shape geometry.
|
||||||
|
pub fn test_shape() -> ShapeBuilder {
|
||||||
|
ShapeBuilder::default().geometry(Geometry::TestShape(Arc::new(Mutex::new(
|
||||||
|
TestData::default(),
|
||||||
|
))))
|
||||||
|
}
|
||||||
|
/// Short hand for creating a ShapeBuilder with a cube geometry.
|
||||||
|
pub fn cube() -> ShapeBuilder {
|
||||||
|
ShapeBuilder::default().geometry(Geometry::Cube)
|
||||||
|
}
|
||||||
|
/// Short hand for creating a ShapeBuilder with a group geometry.
|
||||||
|
pub fn group() -> ShapeBuilder {
|
||||||
|
ShapeBuilder::default().geometry(Geometry::Group(Vec::new()))
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Add child shapes, only valid for Group::Geometry.
|
||||||
|
pub fn add_child(mut self, child: Shape) -> ShapeBuilder {
|
||||||
|
if let Some(Geometry::Group(ref mut children)) = &mut self.geometry {
|
||||||
|
children.push(child);
|
||||||
|
}
|
||||||
|
self
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn build(&self) -> Result<Shape, ShapeBuilderError> {
|
||||||
|
let mut s = Shape::default();
|
||||||
|
|
||||||
|
if let Some(transform) = &self.transform {
|
||||||
|
s.set_transform(*transform);
|
||||||
|
}
|
||||||
|
if let Some(material) = &self.material {
|
||||||
|
s.material = material.clone();
|
||||||
|
};
|
||||||
|
if let Some(geometry) = &self.geometry {
|
||||||
|
s.geometry = geometry.clone();
|
||||||
|
};
|
||||||
|
let transform = s.transform();
|
||||||
|
if let Geometry::Group(ref mut children) = s.geometry {
|
||||||
|
children
|
||||||
|
.iter_mut()
|
||||||
|
.for_each(|c| c.set_transform(transform * c.transform()));
|
||||||
|
}
|
||||||
|
Ok(s)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Default for Shape {
|
||||||
|
fn default() -> Shape {
|
||||||
|
Shape {
|
||||||
|
transform: Matrix4x4::identity(),
|
||||||
|
inverse_transform: Matrix4x4::identity(),
|
||||||
|
material: Material::default(),
|
||||||
|
geometry: Geometry::default(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Shape {
|
||||||
|
/// Create a test shape useful for debugging.
|
||||||
|
pub fn test_shape() -> Shape {
|
||||||
|
Shape {
|
||||||
|
transform: Matrix4x4::identity(),
|
||||||
|
inverse_transform: Matrix4x4::identity(),
|
||||||
|
material: Material::default(),
|
||||||
|
geometry: Geometry::TestShape(Arc::new(Mutex::new(TestData::default()))),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
pub fn sphere() -> Shape {
|
||||||
|
Shape {
|
||||||
|
transform: Matrix4x4::identity(),
|
||||||
|
inverse_transform: Matrix4x4::identity(),
|
||||||
|
material: Material::default(),
|
||||||
|
geometry: Geometry::Sphere,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
pub fn plane() -> Shape {
|
||||||
|
Shape {
|
||||||
|
transform: Matrix4x4::identity(),
|
||||||
|
inverse_transform: Matrix4x4::identity(),
|
||||||
|
material: Material::default(),
|
||||||
|
geometry: Geometry::Plane,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
pub fn cube() -> Shape {
|
||||||
|
Shape {
|
||||||
|
transform: Matrix4x4::identity(),
|
||||||
|
inverse_transform: Matrix4x4::identity(),
|
||||||
|
material: Material::default(),
|
||||||
|
geometry: Geometry::Cube,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
pub fn group() -> Shape {
|
||||||
|
Shape {
|
||||||
|
transform: Matrix4x4::identity(),
|
||||||
|
inverse_transform: Matrix4x4::identity(),
|
||||||
|
material: Material::default(),
|
||||||
|
geometry: Geometry::Group(Vec::new()),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
/// Find the normal at the point on the sphere.
|
||||||
|
pub fn normal_at(&self, world_point: Tuple) -> Tuple {
|
||||||
|
let object_point = self.inverse_transform * world_point;
|
||||||
|
let object_normal = match self.geometry {
|
||||||
|
Geometry::Sphere => object_point - Tuple::point(0., 0., 0.),
|
||||||
|
Geometry::Plane => Tuple::vector(0., 1., 0.),
|
||||||
|
Geometry::TestShape(_) => object_point,
|
||||||
|
Geometry::Cube => cube::local_normal_at(object_point),
|
||||||
|
Geometry::Group(_) => todo!("normal_at"),
|
||||||
|
};
|
||||||
|
let mut world_normal = self.inverse_transform.transpose() * object_normal;
|
||||||
|
world_normal.w = 0.;
|
||||||
|
world_normal.normalize()
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn transform(&self) -> Matrix4x4 {
|
||||||
|
self.transform
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn inverse_transform(&self) -> Matrix4x4 {
|
||||||
|
self.inverse_transform
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn set_transform(&mut self, t: Matrix4x4) {
|
||||||
|
self.transform = t;
|
||||||
|
self.inverse_transform = t.inverse();
|
||||||
|
}
|
||||||
|
pub fn geometry(&self) -> &Geometry {
|
||||||
|
&self.geometry
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Intersect a ray with a shapes.
|
||||||
|
pub fn intersect<'s>(shape: &'s Shape, ray: &Ray) -> Intersections<'s> {
|
||||||
|
let local_ray = ray.transform(shape.inverse_transform);
|
||||||
|
match shape.geometry {
|
||||||
|
Geometry::Sphere => sphere::intersect(shape, &local_ray),
|
||||||
|
Geometry::Plane => plane::intersect(shape, &local_ray),
|
||||||
|
Geometry::TestShape(_) => test_shape::intersect(shape, &local_ray),
|
||||||
|
Geometry::Cube => cube::intersect(shape, &local_ray),
|
||||||
|
Geometry::Group(_) => group::intersect(shape, ray),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
mod test_shape {
|
||||||
|
use crate::{
|
||||||
|
intersections::Intersections,
|
||||||
|
rays::Ray,
|
||||||
|
shapes::{Geometry, Shape},
|
||||||
|
};
|
||||||
|
pub fn intersect<'s>(shape: &'s Shape, ray: &Ray) -> Intersections<'s> {
|
||||||
|
if let Geometry::TestShape(s) = &shape.geometry {
|
||||||
|
s.lock()
|
||||||
|
.expect("couldn't grab mutex for TestData")
|
||||||
|
.saved_ray = Some(ray.clone());
|
||||||
|
}
|
||||||
|
Intersections::default()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
mod sphere {
|
||||||
|
use crate::{
|
||||||
|
intersections::{Intersection, Intersections},
|
||||||
|
rays::Ray,
|
||||||
|
shapes::Shape,
|
||||||
|
tuples::{dot, Tuple},
|
||||||
|
};
|
||||||
|
pub fn intersect<'s>(shape: &'s Shape, ray: &Ray) -> Intersections<'s> {
|
||||||
|
let sphere_to_ray = ray.origin - Tuple::point(0., 0., 0.);
|
||||||
|
let a = dot(ray.direction, ray.direction);
|
||||||
|
let b = 2. * dot(ray.direction, sphere_to_ray);
|
||||||
|
let c = dot(sphere_to_ray, sphere_to_ray) - 1.;
|
||||||
|
let discriminant = b * b - 4. * a * c;
|
||||||
|
if discriminant < 0. {
|
||||||
|
return Intersections::default();
|
||||||
|
}
|
||||||
|
Intersections::new(vec![
|
||||||
|
Intersection::new((-b - discriminant.sqrt()) / (2. * a), shape),
|
||||||
|
Intersection::new((-b + discriminant.sqrt()) / (2. * a), shape),
|
||||||
|
])
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
mod plane {
|
||||||
|
use crate::{
|
||||||
|
intersections::{Intersection, Intersections},
|
||||||
|
rays::Ray,
|
||||||
|
shapes::Shape,
|
||||||
|
EPSILON,
|
||||||
|
};
|
||||||
|
pub fn intersect<'s>(shape: &'s Shape, ray: &Ray) -> Intersections<'s> {
|
||||||
|
if (ray.direction.y).abs() < EPSILON {
|
||||||
|
return Intersections::default();
|
||||||
|
}
|
||||||
|
Intersections::new(vec![Intersection::new(
|
||||||
|
-ray.origin.y / ray.direction.y,
|
||||||
|
shape,
|
||||||
|
)])
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
mod cube {
|
||||||
|
use crate::{
|
||||||
|
intersections::{Intersection, Intersections},
|
||||||
|
rays::Ray,
|
||||||
|
shapes::Shape,
|
||||||
|
tuples::{vector, Tuple},
|
||||||
|
Float, EPSILON,
|
||||||
|
};
|
||||||
|
|
||||||
|
fn check_axis(origin: Float, direction: Float) -> (Float, Float) {
|
||||||
|
let tmin_numerator = -1. - origin;
|
||||||
|
let tmax_numerator = 1. - origin;
|
||||||
|
|
||||||
|
let (tmin, tmax) = if direction.abs() >= EPSILON {
|
||||||
|
(tmin_numerator / direction, tmax_numerator / direction)
|
||||||
|
} else {
|
||||||
|
(
|
||||||
|
tmin_numerator * Float::INFINITY,
|
||||||
|
tmax_numerator * Float::INFINITY,
|
||||||
|
)
|
||||||
|
};
|
||||||
|
|
||||||
|
if tmin > tmax {
|
||||||
|
return (tmax, tmin);
|
||||||
|
}
|
||||||
|
(tmin, tmax)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn intersect<'s>(shape: &'s Shape, ray: &Ray) -> Intersections<'s> {
|
||||||
|
let (xtmin, xtmax) = check_axis(ray.origin.x, ray.direction.x);
|
||||||
|
let (ytmin, ytmax) = check_axis(ray.origin.y, ray.direction.y);
|
||||||
|
let (ztmin, ztmax) = check_axis(ray.origin.z, ray.direction.z);
|
||||||
|
|
||||||
|
let tmin = xtmin.max(ytmin).max(ztmin);
|
||||||
|
let tmax = xtmax.min(ytmax).min(ztmax);
|
||||||
|
|
||||||
|
if tmin > tmax {
|
||||||
|
return Intersections::default();
|
||||||
|
}
|
||||||
|
Intersections::new(vec![
|
||||||
|
Intersection::new(tmin, shape),
|
||||||
|
Intersection::new(tmax, shape),
|
||||||
|
])
|
||||||
|
}
|
||||||
|
pub fn local_normal_at(point: Tuple) -> Tuple {
|
||||||
|
let x = point.x.abs();
|
||||||
|
let y = point.y.abs();
|
||||||
|
let z = point.z.abs();
|
||||||
|
let maxc = x.max(y).max(z);
|
||||||
|
if maxc == x {
|
||||||
|
return vector(point.x, 0., 0.);
|
||||||
|
}
|
||||||
|
if maxc == y {
|
||||||
|
return vector(0., point.y, 0.);
|
||||||
|
}
|
||||||
|
vector(0., 0., point.z)
|
||||||
|
}
|
||||||
|
#[cfg(test)]
|
||||||
|
mod tests {
|
||||||
|
use crate::{
|
||||||
|
rays::Ray,
|
||||||
|
shapes::Shape,
|
||||||
|
tuples::{point, vector},
|
||||||
|
EPSILON,
|
||||||
|
};
|
||||||
|
|
||||||
|
use super::{intersect, local_normal_at};
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn ray_intersects_cube() {
|
||||||
|
let c = Shape::cube();
|
||||||
|
for (name, o, d, t1, t2) in [
|
||||||
|
("+x", point(5., 0.5, 0.), vector(-1., 0., 0.), 4., 6.),
|
||||||
|
("-x", point(-5., 0.5, 0.), vector(1., 0., 0.), 4., 6.),
|
||||||
|
("+y", point(0.5, 5., 0.), vector(0., -1., 0.), 4., 6.),
|
||||||
|
("-y", point(0.5, -5., 0.), vector(0., 1., 0.), 4., 6.),
|
||||||
|
("+z", point(0.5, 0., 5.), vector(0., 0., -1.), 4., 6.),
|
||||||
|
("-z", point(0.5, 0., -5.), vector(0., 0., 1.), 4., 6.),
|
||||||
|
("inside", point(0., 0.5, 0.), vector(0., 0., 1.), -1., 1.),
|
||||||
|
] {
|
||||||
|
let r = Ray::new(o, d);
|
||||||
|
let xs = intersect(&c, &r);
|
||||||
|
assert_eq!(xs.len(), 2, "{}", name);
|
||||||
|
assert!((xs[0].t - t1).abs() < EPSILON, "{} t1 {}", name, xs[0].t);
|
||||||
|
assert!((xs[1].t - t2).abs() < EPSILON, "{} t2 {}", name, xs[1].t);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn ray_misses_cube() {
|
||||||
|
let c = Shape::cube();
|
||||||
|
for (o, d) in [
|
||||||
|
(point(-2., 0., 0.), vector(0.2673, 0.5345, 0.8018)),
|
||||||
|
(point(0., -2., 0.), vector(0.8018, 0.2673, 0.5345)),
|
||||||
|
(point(0., 0., -2.), vector(0.5345, 0.8018, 0.2673)),
|
||||||
|
(point(2., 0., 2.), vector(0., 0., -1.)),
|
||||||
|
(point(0., 2., 2.), vector(0., -1., 0.)),
|
||||||
|
(point(2., 2., -5.), vector(-1., 0., 0.)),
|
||||||
|
] {
|
||||||
|
let r = Ray::new(o, d);
|
||||||
|
let xs = intersect(&c, &r);
|
||||||
|
assert_eq!(xs.len(), 0, "({:?}, {:?})", o, d);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn normal_cube_surface() {
|
||||||
|
for (p, n) in [
|
||||||
|
(point(1., 0.8, -0.8), vector(1., 0., 0.)),
|
||||||
|
(point(-1., -0.2, 0.9), vector(-1., 0., 0.)),
|
||||||
|
(point(-0.4, 1., -0.1), vector(0., 1., 0.)),
|
||||||
|
(point(0.3, -1., -0.7), vector(0., -1., 0.)),
|
||||||
|
(point(-0.6, 0.3, 1.), vector(0., 0., 1.)),
|
||||||
|
(point(0.4, 0.4, -1.), vector(0., 0., -1.)),
|
||||||
|
(point(1., 1., 1.), vector(1., 0., 0.)),
|
||||||
|
(point(-1., -1., -1.), vector(-1., 0., 0.)),
|
||||||
|
] {
|
||||||
|
let normal = local_normal_at(p);
|
||||||
|
assert_eq!(n, normal);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
mod group {
|
||||||
|
use crate::{intersections::Intersections, rays::Ray, shapes::Shape};
|
||||||
|
|
||||||
|
use super::Geometry;
|
||||||
|
|
||||||
|
pub fn intersect<'s>(shape: &'s Shape, ray: &Ray) -> Intersections<'s> {
|
||||||
|
if let Geometry::Group(children) = &shape.geometry {
|
||||||
|
let mut intersections: Vec<_> = children
|
||||||
|
.iter()
|
||||||
|
.flat_map(|c| super::intersect(c, ray))
|
||||||
|
.collect();
|
||||||
|
intersections.sort_by(|a, b| {
|
||||||
|
a.t.partial_cmp(&b.t)
|
||||||
|
.expect("an intersection has a t value that is NaN")
|
||||||
|
});
|
||||||
|
return Intersections::new(intersections);
|
||||||
|
}
|
||||||
|
unreachable!();
|
||||||
|
}
|
||||||
|
#[cfg(test)]
|
||||||
|
mod tests {
|
||||||
|
use crate::{
|
||||||
|
matrices::{scaling, translation},
|
||||||
|
rays::Ray,
|
||||||
|
shapes::{intersect, Shape, ShapeBuilder},
|
||||||
|
tuples::{point, vector},
|
||||||
|
};
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn intersecting_empty_group() {
|
||||||
|
let g = Shape::group();
|
||||||
|
let r = Ray::new(point(0., 0., 0.), vector(0., 0., 1.));
|
||||||
|
let xs = intersect(&g, &r);
|
||||||
|
assert_eq!(xs.len(), 0);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn intersecting_nonempty_group() -> Result<(), Box<dyn std::error::Error>> {
|
||||||
|
let s1 = Shape::sphere();
|
||||||
|
let s2 = ShapeBuilder::sphere()
|
||||||
|
.transform(translation(0., 0., -3.))
|
||||||
|
.build()?;
|
||||||
|
let s3 = ShapeBuilder::sphere()
|
||||||
|
.transform(translation(5., 0., 0.))
|
||||||
|
.build()?;
|
||||||
|
let g = ShapeBuilder::group()
|
||||||
|
.add_child(s1.clone())
|
||||||
|
.add_child(s2.clone())
|
||||||
|
.add_child(s3.clone())
|
||||||
|
.build()?;
|
||||||
|
let r = Ray::new(point(0., 0., -5.), vector(0., 0., 1.));
|
||||||
|
let xs = intersect(&g, &r);
|
||||||
|
assert_eq!(xs.len(), 4);
|
||||||
|
assert_eq!(xs[0].object, &s2);
|
||||||
|
assert_eq!(xs[1].object, &s2);
|
||||||
|
assert_eq!(xs[2].object, &s1);
|
||||||
|
assert_eq!(xs[3].object, &s1);
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn intersecting_transformed_group() -> Result<(), Box<dyn std::error::Error>> {
|
||||||
|
let g = ShapeBuilder::group()
|
||||||
|
.transform(scaling(2., 2., 2.))
|
||||||
|
.add_child(
|
||||||
|
ShapeBuilder::sphere()
|
||||||
|
.transform(translation(5., 0., 0.))
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.build()?;
|
||||||
|
let r = Ray::new(point(10., 0., -10.), vector(0., 0., 1.));
|
||||||
|
let xs = intersect(&g, &r);
|
||||||
|
assert_eq!(xs.len(), 2);
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
#[cfg(test)]
|
||||||
|
mod tests {
|
||||||
|
mod shape_builder {
|
||||||
|
use std::error::Error;
|
||||||
|
|
||||||
|
use crate::shapes::{plane, sphere, test_shape, Shape};
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn plane_builder() -> Result<(), Box<dyn Error>> {
|
||||||
|
assert_eq!(plane().build()?, Shape::plane());
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn sphere_builder() -> Result<(), Box<dyn Error>> {
|
||||||
|
assert_eq!(sphere().build()?, Shape::sphere());
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn test_shape_builder() -> Result<(), Box<dyn Error>> {
|
||||||
|
assert_eq!(test_shape().build()?, Shape::test_shape());
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
mod shape {
|
||||||
|
use crate::{
|
||||||
|
materials::Material,
|
||||||
|
matrices::{identity, translation},
|
||||||
|
shapes::Shape,
|
||||||
|
};
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_shape() {
|
||||||
|
let mut s = Shape::test_shape();
|
||||||
|
// The default transform.
|
||||||
|
assert_eq!(s.transform(), identity());
|
||||||
|
// The default material.
|
||||||
|
assert_eq!(s.material, Material::default());
|
||||||
|
// Assigning a material.
|
||||||
|
let m = Material {
|
||||||
|
ambient: 1.,
|
||||||
|
..Material::default()
|
||||||
|
};
|
||||||
|
s.material = m.clone();
|
||||||
|
assert_eq!(s.material, m);
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn sphere() {
|
||||||
|
// A sphere's default transform is the identity matrix.
|
||||||
|
let s = Shape::sphere();
|
||||||
|
assert_eq!(s.transform(), identity());
|
||||||
|
|
||||||
|
// It can be changed by directly setting the transform member.
|
||||||
|
let mut s = Shape::sphere();
|
||||||
|
let t = translation(2., 3., 4.);
|
||||||
|
s.set_transform(t.clone());
|
||||||
|
assert_eq!(s.transform(), t);
|
||||||
|
|
||||||
|
// Default Sphere has the default material.
|
||||||
|
assert_eq!(s.material, Material::default());
|
||||||
|
// It can be overridden.
|
||||||
|
let mut s = Shape::sphere();
|
||||||
|
let mut m = Material::default();
|
||||||
|
m.ambient = 1.;
|
||||||
|
s.material = m.clone();
|
||||||
|
assert_eq!(s.material, m);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
mod normal_at {
|
||||||
|
use crate::{float::consts::PI, matrices::Matrix4x4, shapes::Shape, tuples::Tuple, Float};
|
||||||
|
#[test]
|
||||||
|
fn compute_normal_on_translated_shape() {
|
||||||
|
// Computing the normal on a translated shape.
|
||||||
|
let mut s = Shape::test_shape();
|
||||||
|
s.set_transform(Matrix4x4::translation(0., 1., 0.));
|
||||||
|
let n = s.normal_at(Tuple::point(0., 1.70711, -0.70711));
|
||||||
|
assert_eq!(n, Tuple::vector(0., 0.70711, -0.70711));
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn compute_normal_on_scaled_shape() {
|
||||||
|
// Computing the normal on a scaled shape.
|
||||||
|
let mut s = Shape::test_shape();
|
||||||
|
s.set_transform(Matrix4x4::scaling(1., 0.5, 1.) * Matrix4x4::rotation_z(PI / 5.));
|
||||||
|
let n = s.normal_at(Tuple::point(
|
||||||
|
0.,
|
||||||
|
(2. as Float).sqrt() / 2.,
|
||||||
|
-(2. as Float).sqrt() / 2.,
|
||||||
|
));
|
||||||
|
assert_eq!(n, Tuple::vector(0., 0.97014, -0.24254));
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn sphere_normal_on_x_axis() {
|
||||||
|
// Normal on X-axis
|
||||||
|
let s = Shape::sphere();
|
||||||
|
let n = s.normal_at(Tuple::point(1., 0., 0.));
|
||||||
|
assert_eq!(n, Tuple::vector(1., 0., 0.));
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn sphere_normal_on_y_axis() {
|
||||||
|
// Normal on Y-axis
|
||||||
|
let s = Shape::sphere();
|
||||||
|
let n = s.normal_at(Tuple::point(0., 1., 0.));
|
||||||
|
assert_eq!(n, Tuple::vector(0., 1., 0.));
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn sphere_normal_on_z_axis() {
|
||||||
|
// Normal on Z-axis
|
||||||
|
let s = Shape::sphere();
|
||||||
|
let n = s.normal_at(Tuple::point(0., 0., 1.));
|
||||||
|
assert_eq!(n, Tuple::vector(0., 0., 1.));
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn sphere_normal_non_axial() {
|
||||||
|
// Normal on a sphere at a nonaxial point.
|
||||||
|
let s = Shape::sphere();
|
||||||
|
let n = s.normal_at(Tuple::point(
|
||||||
|
(3. as Float).sqrt() / 3.,
|
||||||
|
(3. as Float).sqrt() / 3.,
|
||||||
|
(3. as Float).sqrt() / 3.,
|
||||||
|
));
|
||||||
|
assert_eq!(
|
||||||
|
n,
|
||||||
|
Tuple::vector(
|
||||||
|
(3. as Float).sqrt() / 3.,
|
||||||
|
(3. as Float).sqrt() / 3.,
|
||||||
|
(3. as Float).sqrt() / 3.,
|
||||||
|
)
|
||||||
|
);
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn normals_are_normalized() {
|
||||||
|
// Normals returned are normalized.
|
||||||
|
let s = Shape::sphere();
|
||||||
|
let n = s.normal_at(Tuple::point(
|
||||||
|
(3. as Float).sqrt() / 3.,
|
||||||
|
(3. as Float).sqrt() / 3.,
|
||||||
|
(3. as Float).sqrt() / 3.,
|
||||||
|
));
|
||||||
|
assert_eq!(n, n.normalize());
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn compute_normal_on_translated_sphere() {
|
||||||
|
// Compute the normal on a translated sphere.
|
||||||
|
let mut s = Shape::sphere();
|
||||||
|
s.set_transform(Matrix4x4::translation(0., 1., 0.));
|
||||||
|
let n = s.normal_at(Tuple::point(0., 1.70711, -0.70711));
|
||||||
|
assert_eq!(n, Tuple::vector(0., 0.70711, -0.70711));
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn compute_normal_on_scaled_sphere() {
|
||||||
|
// Compute the normal on a transformed sphere.
|
||||||
|
let mut s = Shape::sphere();
|
||||||
|
s.set_transform(Matrix4x4::scaling(1., 0.5, 1.) * Matrix4x4::rotation_z(PI / 5.));
|
||||||
|
let n = s.normal_at(Tuple::point(
|
||||||
|
0.,
|
||||||
|
(2. as Float).sqrt() / 2.,
|
||||||
|
-(2. as Float).sqrt() / 2.,
|
||||||
|
));
|
||||||
|
assert_eq!(n, Tuple::vector(0., 0.97014, -0.24254));
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn nomal_of_plane_constant() {
|
||||||
|
// Normal of a plane is constant everywhere.
|
||||||
|
let p = Shape::plane();
|
||||||
|
assert_eq!(
|
||||||
|
p.normal_at(Tuple::point(0., 0., 0.)),
|
||||||
|
Tuple::vector(0., 1., 0.)
|
||||||
|
);
|
||||||
|
assert_eq!(
|
||||||
|
p.normal_at(Tuple::point(10., 0., -10.)),
|
||||||
|
Tuple::vector(0., 1., 0.)
|
||||||
|
);
|
||||||
|
assert_eq!(
|
||||||
|
p.normal_at(Tuple::point(-5., 0., 150.)),
|
||||||
|
Tuple::vector(0., 1., 0.)
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
mod intersect {
|
||||||
|
use crate::{
|
||||||
|
intersections::{Intersection, Intersections},
|
||||||
|
matrices::Matrix4x4,
|
||||||
|
rays::Ray,
|
||||||
|
shapes::{intersect, Geometry, Shape},
|
||||||
|
tuples::Tuple,
|
||||||
|
};
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn scaled_shape_with_ray() {
|
||||||
|
// Intersecting a scaled shape with a ray.
|
||||||
|
let r = Ray::new(Tuple::point(0., 0., -5.), Tuple::vector(0., 0., 1.));
|
||||||
|
let mut s = Shape::test_shape();
|
||||||
|
s.set_transform(Matrix4x4::scaling(2., 2., 2.));
|
||||||
|
let _xs = intersect(&s, &r);
|
||||||
|
if let Geometry::TestShape(data) = s.geometry() {
|
||||||
|
if let Some(ray) = &data.lock().unwrap().saved_ray {
|
||||||
|
assert_eq!(ray.origin, Tuple::point(0., 0., -2.5));
|
||||||
|
assert_eq!(ray.direction, Tuple::vector(0., 0., 0.5));
|
||||||
|
} else {
|
||||||
|
panic!("ray wasn't set");
|
||||||
|
};
|
||||||
|
} else {
|
||||||
|
panic!("test_shape returned a non-TestShape geometry")
|
||||||
|
};
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn translated_shape_with_ray() {
|
||||||
|
// Intersecting a translated shape with a ray.
|
||||||
|
let r = Ray::new(Tuple::point(0., 0., -5.), Tuple::vector(0., 0., 1.));
|
||||||
|
let mut s = Shape::test_shape();
|
||||||
|
s.set_transform(Matrix4x4::translation(5., 0., 0.));
|
||||||
|
let _xs = intersect(&s, &r);
|
||||||
|
if let Geometry::TestShape(data) = s.geometry() {
|
||||||
|
if let Some(ray) = &data.lock().unwrap().saved_ray {
|
||||||
|
assert_eq!(ray.origin, Tuple::point(-5., 0., -5.));
|
||||||
|
assert_eq!(ray.direction, Tuple::vector(0., 0., 1.));
|
||||||
|
} else {
|
||||||
|
panic!("ray wasn't set");
|
||||||
|
};
|
||||||
|
} else {
|
||||||
|
panic!("test_shape returned a non-TestShape geometry")
|
||||||
|
};
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn ray_intersects_sphere_two_points() {
|
||||||
|
// A ray intersects a sphere in two points.
|
||||||
|
let r = Ray::new(Tuple::point(0., 0., -5.), Tuple::vector(0., 0., 1.));
|
||||||
|
let s = Shape::sphere();
|
||||||
|
let xs = intersect(&s, &r);
|
||||||
|
assert_eq!(
|
||||||
|
xs,
|
||||||
|
Intersections::new(vec![Intersection::new(4., &s), Intersection::new(6., &s)])
|
||||||
|
);
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn ray_intersects_at_tangent() {
|
||||||
|
// A ray intersects a sphere at a tangent.
|
||||||
|
let r = Ray::new(Tuple::point(0., 2., -5.), Tuple::vector(0., 0., 1.));
|
||||||
|
let s = Shape::sphere();
|
||||||
|
let xs = intersect(&s, &r);
|
||||||
|
assert_eq!(xs, Intersections::default());
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn ray_originates_inside_sphere() {
|
||||||
|
// A ray originates inside a sphere.
|
||||||
|
let r = Ray::new(Tuple::point(0., 0., 0.), Tuple::vector(0., 0., 1.));
|
||||||
|
let s = Shape::sphere();
|
||||||
|
let xs = intersect(&s, &r);
|
||||||
|
assert_eq!(
|
||||||
|
xs,
|
||||||
|
Intersections::new(vec![Intersection::new(-1., &s), Intersection::new(1., &s)])
|
||||||
|
);
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn sphere_behind_ray() {
|
||||||
|
// A sphere is behind a ray.
|
||||||
|
let r = Ray::new(Tuple::point(0., 0., 5.), Tuple::vector(0., 0., 1.));
|
||||||
|
let s = Shape::sphere();
|
||||||
|
let xs = intersect(&s, &r);
|
||||||
|
assert_eq!(
|
||||||
|
xs,
|
||||||
|
Intersections::new(vec![Intersection::new(-6., &s), Intersection::new(-4., &s)])
|
||||||
|
);
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn ray_intersects_scaled_sphere() {
|
||||||
|
// Intersect a scaled sphere with a ray.
|
||||||
|
let r = Ray::new(Tuple::point(0., 0., -5.), Tuple::vector(0., 0., 1.));
|
||||||
|
let mut s = Shape::sphere();
|
||||||
|
s.set_transform(Matrix4x4::scaling(2., 2., 2.));
|
||||||
|
let xs = intersect(&s, &r);
|
||||||
|
assert_eq!(xs.len(), 2, "xs {:?}", xs);
|
||||||
|
assert_eq!(xs[0].t, 3., "xs {:?}", xs);
|
||||||
|
assert_eq!(xs[1].t, 7., "xs {:?}", xs);
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn ray_intersects_translated_sphere() {
|
||||||
|
// Intersect a translated sphere with a ray.
|
||||||
|
let r = Ray::new(Tuple::point(0., 0., -5.), Tuple::vector(0., 0., 1.));
|
||||||
|
let mut s = Shape::sphere();
|
||||||
|
s.set_transform(Matrix4x4::translation(5., 0., 0.));
|
||||||
|
let xs = intersect(&s, &r);
|
||||||
|
assert_eq!(xs.len(), 0);
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn ray_parallel_to_plane() {
|
||||||
|
// Intersect with a ray parallel to the plane.
|
||||||
|
let p = Shape::plane();
|
||||||
|
let r = Ray::new(Tuple::point(0., 10., 0.), Tuple::vector(0., 0., 1.));
|
||||||
|
let xs = intersect(&p, &r);
|
||||||
|
assert_eq!(xs.len(), 0);
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn ray_coplanar_to_plane() {
|
||||||
|
// Intersect with a coplanar.
|
||||||
|
let p = Shape::plane();
|
||||||
|
let r = Ray::new(Tuple::point(0., 0., 0.), Tuple::vector(0., 0., 1.));
|
||||||
|
let xs = intersect(&p, &r);
|
||||||
|
assert_eq!(xs.len(), 0);
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn ray_intersects_plane_from_above() {
|
||||||
|
// A ray intersecting a plane from above.
|
||||||
|
let p = Shape::plane();
|
||||||
|
let r = Ray::new(Tuple::point(0., 1., 0.), Tuple::vector(0., -1., 0.));
|
||||||
|
let xs = intersect(&p, &r);
|
||||||
|
assert_eq!(xs.len(), 1);
|
||||||
|
assert_eq!(xs[0].t, 1.);
|
||||||
|
assert_eq!(xs[0].object, &p);
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn ray_intersects_plane_from_below() {
|
||||||
|
// A ray intersecting a plane from below.
|
||||||
|
let p = Shape::plane();
|
||||||
|
let r = Ray::new(Tuple::point(0., -1., 0.), Tuple::vector(0., 1., 0.));
|
||||||
|
let xs = intersect(&p, &r);
|
||||||
|
assert_eq!(xs.len(), 1);
|
||||||
|
assert_eq!(xs[0].t, 1.);
|
||||||
|
assert_eq!(xs[0].object, &p);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -1,191 +0,0 @@
|
|||||||
use crate::{
|
|
||||||
intersections::{Intersection, Intersections},
|
|
||||||
materials::Material,
|
|
||||||
matrices::Matrix4x4,
|
|
||||||
rays::Ray,
|
|
||||||
tuples::{dot, Tuple},
|
|
||||||
};
|
|
||||||
|
|
||||||
#[derive(Debug, PartialEq)]
|
|
||||||
/// Sphere represents the unit-sphere (radius of unit 1.) at the origin 0., 0., 0.
|
|
||||||
pub struct Sphere {
|
|
||||||
// TODO(wathiede): cache inverse to speed up intersect.
|
|
||||||
pub transform: Matrix4x4,
|
|
||||||
pub material: Material,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl Default for Sphere {
|
|
||||||
/// # Examples
|
|
||||||
/// ```
|
|
||||||
/// use rtchallenge::{materials::Material, matrices::Matrix4x4, spheres::Sphere};
|
|
||||||
///
|
|
||||||
/// // A sphere's default transform is the identity matrix.
|
|
||||||
/// let s = Sphere::default();
|
|
||||||
/// assert_eq!(s.transform, Matrix4x4::identity());
|
|
||||||
///
|
|
||||||
/// // It can be changed by directly setting the transform member.
|
|
||||||
/// let mut s = Sphere::default();
|
|
||||||
/// let t = Matrix4x4::translation(2., 3., 4.);
|
|
||||||
/// s.transform = t.clone();
|
|
||||||
/// assert_eq!(s.transform, t);
|
|
||||||
///
|
|
||||||
/// // Default Sphere has the default material.
|
|
||||||
/// assert_eq!(s.material, Material::default());
|
|
||||||
/// // It can be overridden.
|
|
||||||
/// let mut s = Sphere::default();
|
|
||||||
/// let mut m = Material::default();
|
|
||||||
/// m.ambient = 1.;
|
|
||||||
/// s.material = m.clone();
|
|
||||||
/// assert_eq!(s.material, m);
|
|
||||||
|
|
||||||
/// ```
|
|
||||||
fn default() -> Sphere {
|
|
||||||
Sphere {
|
|
||||||
transform: Matrix4x4::identity(),
|
|
||||||
material: Material::default(),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl Sphere {
|
|
||||||
/// Find the normal at the point on the sphere.
|
|
||||||
///
|
|
||||||
/// # Examples
|
|
||||||
/// ```
|
|
||||||
/// use rtchallenge::{matrices::Matrix4x4, spheres::Sphere, tuples::Tuple};
|
|
||||||
///
|
|
||||||
/// // Normal on X-axis
|
|
||||||
/// let s = Sphere::default();
|
|
||||||
/// let n = s.normal_at(Tuple::point(1., 0., 0.));
|
|
||||||
/// assert_eq!(n, Tuple::vector(1., 0., 0.));
|
|
||||||
///
|
|
||||||
/// // Normal on Y-axis
|
|
||||||
/// let s = Sphere::default();
|
|
||||||
/// let n = s.normal_at(Tuple::point(0., 1., 0.));
|
|
||||||
/// assert_eq!(n, Tuple::vector(0., 1., 0.));
|
|
||||||
///
|
|
||||||
/// // Normal on Z-axis
|
|
||||||
/// let s = Sphere::default();
|
|
||||||
/// let n = s.normal_at(Tuple::point(0., 0., 1.));
|
|
||||||
/// assert_eq!(n, Tuple::vector(0., 0., 1.));
|
|
||||||
///
|
|
||||||
/// // Normal on a sphere at a nonaxial point.
|
|
||||||
/// let s = Sphere::default();
|
|
||||||
/// let n = s.normal_at(Tuple::point(
|
|
||||||
/// 3_f32.sqrt() / 3.,
|
|
||||||
/// 3_f32.sqrt() / 3.,
|
|
||||||
/// 3_f32.sqrt() / 3.,
|
|
||||||
/// ));
|
|
||||||
/// assert_eq!(
|
|
||||||
/// n,
|
|
||||||
/// Tuple::vector(3_f32.sqrt() / 3., 3_f32.sqrt() / 3., 3_f32.sqrt() / 3.,)
|
|
||||||
/// );
|
|
||||||
/// // Normals returned are normalized.
|
|
||||||
/// let s = Sphere::default();
|
|
||||||
/// let n = s.normal_at(Tuple::point(
|
|
||||||
/// 3_f32.sqrt() / 3.,
|
|
||||||
/// 3_f32.sqrt() / 3.,
|
|
||||||
/// 3_f32.sqrt() / 3.,
|
|
||||||
/// ));
|
|
||||||
/// assert_eq!(n, n.normalize());
|
|
||||||
///
|
|
||||||
/// // Compute the normal on a translated sphere.
|
|
||||||
/// let mut s = Sphere::default();
|
|
||||||
/// s.transform = Matrix4x4::translation(0., 1., 0.);
|
|
||||||
/// let n = s.normal_at(Tuple::point(0., 1.70711, -0.70711));
|
|
||||||
/// assert_eq!(n, Tuple::vector(0., 0.70711, -0.70711));
|
|
||||||
|
|
||||||
/// // Compute the normal on a transformed sphere.
|
|
||||||
/// use std::f32::consts::PI;
|
|
||||||
///
|
|
||||||
/// let mut s = Sphere::default();
|
|
||||||
/// s.transform = Matrix4x4::scaling(1.,0.5,1.) * Matrix4x4::rotation_z(PI/5.);
|
|
||||||
/// let n = s.normal_at(Tuple::point(0., 2_f32.sqrt()/2., -2_f32.sqrt()/2.));
|
|
||||||
/// assert_eq!(n, Tuple::vector(0., 0.97014, -0.24254));
|
|
||||||
/// ```
|
|
||||||
pub fn normal_at(&self, world_point: Tuple) -> Tuple {
|
|
||||||
let object_point = self.transform.inverse() * world_point;
|
|
||||||
let object_normal = object_point - Tuple::point(0., 0., 0.);
|
|
||||||
let mut world_normal = self.transform.inverse().transpose() * object_normal;
|
|
||||||
world_normal.w = 0.;
|
|
||||||
world_normal.normalize()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Intersect a ray with a sphere.
|
|
||||||
///
|
|
||||||
/// # Examples
|
|
||||||
/// ```
|
|
||||||
/// use rtchallenge::{
|
|
||||||
/// intersections::{Intersection, Intersections},
|
|
||||||
/// matrices::Matrix4x4,
|
|
||||||
/// rays::Ray,
|
|
||||||
/// spheres::{intersect, Sphere},
|
|
||||||
/// tuples::Tuple,
|
|
||||||
/// };
|
|
||||||
///
|
|
||||||
/// // A ray intersects a sphere in two points.
|
|
||||||
/// let r = Ray::new(Tuple::point(0., 0., -5.), Tuple::vector(0., 0., 1.));
|
|
||||||
/// let s = Sphere::default();
|
|
||||||
/// let xs = intersect(&s, &r);
|
|
||||||
/// assert_eq!(
|
|
||||||
/// xs,
|
|
||||||
/// Intersections::new(vec![Intersection::new(4., &s), Intersection::new(6., &s)])
|
|
||||||
/// );
|
|
||||||
///
|
|
||||||
/// // A ray intersects a sphere at a tangent.
|
|
||||||
/// let r = Ray::new(Tuple::point(0., 2., -5.), Tuple::vector(0., 0., 1.));
|
|
||||||
/// let s = Sphere::default();
|
|
||||||
/// let xs = intersect(&s, &r);
|
|
||||||
/// assert_eq!(xs, Intersections::default());
|
|
||||||
///
|
|
||||||
/// // A ray originates inside a sphere.
|
|
||||||
/// let r = Ray::new(Tuple::point(0., 0., 0.), Tuple::vector(0., 0., 1.));
|
|
||||||
/// let s = Sphere::default();
|
|
||||||
/// let xs = intersect(&s, &r);
|
|
||||||
/// assert_eq!(
|
|
||||||
/// xs,
|
|
||||||
/// Intersections::new(vec![Intersection::new(-1., &s), Intersection::new(1., &s)])
|
|
||||||
/// );
|
|
||||||
///
|
|
||||||
/// // A sphere is behind a ray.
|
|
||||||
/// let r = Ray::new(Tuple::point(0., 0., 5.), Tuple::vector(0., 0., 1.));
|
|
||||||
/// let s = Sphere::default();
|
|
||||||
/// let xs = intersect(&s, &r);
|
|
||||||
/// assert_eq!(
|
|
||||||
/// xs,
|
|
||||||
/// Intersections::new(vec![Intersection::new(-6., &s), Intersection::new(-4., &s)])
|
|
||||||
/// );
|
|
||||||
///
|
|
||||||
/// // Intersect a scaled sphere with a ray.
|
|
||||||
/// let r = Ray::new(Tuple::point(0., 0., -5.), Tuple::vector(0., 0., 1.));
|
|
||||||
/// let mut s = Sphere::default();
|
|
||||||
/// s.transform = Matrix4x4::scaling(2., 2., 2.);
|
|
||||||
/// let xs = intersect(&s, &r);
|
|
||||||
/// assert_eq!(xs.len(), 2, "xs {:?}", xs);
|
|
||||||
/// assert_eq!(xs[0].t, 3., "xs {:?}", xs);
|
|
||||||
/// assert_eq!(xs[1].t, 7., "xs {:?}", xs);
|
|
||||||
///
|
|
||||||
/// // Intersect a translated sphere with a ray.
|
|
||||||
/// let r = Ray::new(Tuple::point(0., 0., -5.), Tuple::vector(0., 0., 1.));
|
|
||||||
/// let mut s = Sphere::default();
|
|
||||||
/// s.transform = Matrix4x4::translation(5., 0., 0.);
|
|
||||||
/// let xs = intersect(&s, &r);
|
|
||||||
/// assert_eq!(xs.len(), 0);
|
|
||||||
/// ```
|
|
||||||
pub fn intersect<'s>(sphere: &'s Sphere, ray: &Ray) -> Intersections<'s> {
|
|
||||||
let ray = ray.transform(sphere.transform.inverse());
|
|
||||||
let sphere_to_ray = ray.origin - Tuple::point(0., 0., 0.);
|
|
||||||
let a = dot(ray.direction, ray.direction);
|
|
||||||
let b = 2. * dot(ray.direction, sphere_to_ray);
|
|
||||||
let c = dot(sphere_to_ray, sphere_to_ray) - 1.;
|
|
||||||
let discriminant = b * b - 4. * a * c;
|
|
||||||
if discriminant < 0. {
|
|
||||||
Intersections::default()
|
|
||||||
} else {
|
|
||||||
Intersections::new(vec![
|
|
||||||
Intersection::new((-b - discriminant.sqrt()) / (2. * a), &sphere),
|
|
||||||
Intersection::new((-b + discriminant.sqrt()) / (2. * a), &sphere),
|
|
||||||
])
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -5,47 +5,6 @@ use crate::{
|
|||||||
|
|
||||||
/// Create a matrix representing a eye at `from` looking at `to`, with an `up`
|
/// Create a matrix representing a eye at `from` looking at `to`, with an `up`
|
||||||
/// as the up vector.
|
/// as the up vector.
|
||||||
///
|
|
||||||
/// # Examples
|
|
||||||
/// ```
|
|
||||||
/// use rtchallenge::{matrices::Matrix4x4, transformations::view_transform, tuples::Tuple};
|
|
||||||
///
|
|
||||||
/// // The transofrmation matrix for the default orientation.
|
|
||||||
/// let from = Tuple::point(0., 0., 0.);
|
|
||||||
/// let to = Tuple::point(0., 0., -1.);
|
|
||||||
/// let up = Tuple::vector(0., 1., 0.);
|
|
||||||
/// let t = view_transform(from, to, up);
|
|
||||||
/// assert_eq!(t, Matrix4x4::identity());
|
|
||||||
///
|
|
||||||
/// // A view transformation matrix looking in positive z direction.
|
|
||||||
/// let from = Tuple::point(0., 0., 0.);
|
|
||||||
/// let to = Tuple::point(0., 0., 1.);
|
|
||||||
/// let up = Tuple::vector(0., 1., 0.);
|
|
||||||
/// let t = view_transform(from, to, up);
|
|
||||||
/// assert_eq!(t, Matrix4x4::scaling(-1., 1., -1.));
|
|
||||||
///
|
|
||||||
/// // The view tranformation moves the world.
|
|
||||||
/// let from = Tuple::point(0., 0., 8.);
|
|
||||||
/// let to = Tuple::point(0., 0., 0.);
|
|
||||||
/// let up = Tuple::vector(0., 1., 0.);
|
|
||||||
/// let t = view_transform(from, to, up);
|
|
||||||
/// assert_eq!(t, Matrix4x4::translation(0., 0., -8.));
|
|
||||||
///
|
|
||||||
/// // An arbitrary view transformation.
|
|
||||||
/// let from = Tuple::point(1., 3., 2.);
|
|
||||||
/// let to = Tuple::point(4., -2., 8.);
|
|
||||||
/// let up = Tuple::vector(1., 1., 0.);
|
|
||||||
/// let t = view_transform(from, to, up);
|
|
||||||
/// assert_eq!(
|
|
||||||
/// t,
|
|
||||||
/// Matrix4x4::new(
|
|
||||||
/// [-0.50709, 0.50709, 0.67612, -2.36643],
|
|
||||||
/// [0.76772, 0.60609, 0.12122, -2.82843],
|
|
||||||
/// [-0.35857, 0.59761, -0.71714, 0.],
|
|
||||||
/// [0., 0., 0., 1.],
|
|
||||||
/// )
|
|
||||||
/// );
|
|
||||||
/// ```
|
|
||||||
pub fn view_transform(from: Tuple, to: Tuple, up: Tuple) -> Matrix4x4 {
|
pub fn view_transform(from: Tuple, to: Tuple, up: Tuple) -> Matrix4x4 {
|
||||||
let forward = (to - from).normalize();
|
let forward = (to - from).normalize();
|
||||||
let left = cross(forward, up.normalize());
|
let left = cross(forward, up.normalize());
|
||||||
@@ -57,3 +16,56 @@ pub fn view_transform(from: Tuple, to: Tuple, up: Tuple) -> Matrix4x4 {
|
|||||||
[0., 0., 0., 1.],
|
[0., 0., 0., 1.],
|
||||||
) * Matrix4x4::translation(-from.x, -from.y, -from.z)
|
) * Matrix4x4::translation(-from.x, -from.y, -from.z)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
mod tests {
|
||||||
|
use crate::{
|
||||||
|
matrices::{identity, scaling, translation, Matrix4x4},
|
||||||
|
transformations::view_transform,
|
||||||
|
tuples::{point, vector},
|
||||||
|
};
|
||||||
|
#[test]
|
||||||
|
fn default_orientation() {
|
||||||
|
// The transformation matrix for the default orientation.
|
||||||
|
let from = point(0., 0., 0.);
|
||||||
|
let to = point(0., 0., -1.);
|
||||||
|
let up = vector(0., 1., 0.);
|
||||||
|
let t = view_transform(from, to, up);
|
||||||
|
assert_eq!(t, identity());
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn looking_positive_z() {
|
||||||
|
// A view transformation matrix looking in positive z direction.
|
||||||
|
let from = point(0., 0., 0.);
|
||||||
|
let to = point(0., 0., 1.);
|
||||||
|
let up = vector(0., 1., 0.);
|
||||||
|
let t = view_transform(from, to, up);
|
||||||
|
assert_eq!(t, scaling(-1., 1., -1.));
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn transformation_moves_world() {
|
||||||
|
// The view transformation moves the world.
|
||||||
|
let from = point(0., 0., 8.);
|
||||||
|
let to = point(0., 0., 0.);
|
||||||
|
let up = vector(0., 1., 0.);
|
||||||
|
let t = view_transform(from, to, up);
|
||||||
|
assert_eq!(t, translation(0., 0., -8.));
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn arbitrary_view() {
|
||||||
|
// An arbitrary view transformation.
|
||||||
|
let from = point(1., 3., 2.);
|
||||||
|
let to = point(4., -2., 8.);
|
||||||
|
let up = vector(1., 1., 0.);
|
||||||
|
let t = view_transform(from, to, up);
|
||||||
|
assert_eq!(
|
||||||
|
t,
|
||||||
|
Matrix4x4::new(
|
||||||
|
[-0.50709, 0.50709, 0.67612, -2.36643],
|
||||||
|
[0.76772, 0.60609, 0.12122, -2.82843],
|
||||||
|
[-0.35857, 0.59761, -0.71714, 0.],
|
||||||
|
[0., 0., 0., 1.],
|
||||||
|
)
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|||||||
@@ -1,25 +1,35 @@
|
|||||||
use std::ops::{Add, Div, Mul, Neg, Sub};
|
use std::ops::{Add, Div, Mul, Neg, Sub};
|
||||||
|
|
||||||
use crate::EPSILON;
|
use crate::{Float, EPSILON};
|
||||||
|
|
||||||
#[derive(Debug, Copy, Clone)]
|
/// Short hand for creating a Tuple that represents a point, w=1.
|
||||||
|
pub fn point(x: Float, y: Float, z: Float) -> Tuple {
|
||||||
|
Tuple::point(x, y, z)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Short hand for creating a Tuple that represents a vector, w=0.
|
||||||
|
pub fn vector(x: Float, y: Float, z: Float) -> Tuple {
|
||||||
|
Tuple::vector(x, y, z)
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug, Default, Copy, Clone)]
|
||||||
pub struct Tuple {
|
pub struct Tuple {
|
||||||
pub x: f32,
|
pub x: Float,
|
||||||
pub y: f32,
|
pub y: Float,
|
||||||
pub z: f32,
|
pub z: Float,
|
||||||
pub w: f32,
|
pub w: Float,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Tuple {
|
impl Tuple {
|
||||||
pub fn point(x: f32, y: f32, z: f32) -> Tuple {
|
pub fn point(x: Float, y: Float, z: Float) -> Tuple {
|
||||||
Tuple::new(x, y, z, 1.0)
|
Tuple::new(x, y, z, 1.0)
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn vector(x: f32, y: f32, z: f32) -> Tuple {
|
pub fn vector(x: Float, y: Float, z: Float) -> Tuple {
|
||||||
Tuple::new(x, y, z, 0.0)
|
Tuple::new(x, y, z, 0.0)
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn new(x: f32, y: f32, z: f32, w: f32) -> Tuple {
|
pub fn new(x: Float, y: Float, z: Float, w: Float) -> Tuple {
|
||||||
Tuple { x, y, z, w }
|
Tuple { x, y, z, w }
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -31,7 +41,7 @@ impl Tuple {
|
|||||||
self.w == 0.0
|
self.w == 0.0
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn magnitude(&self) -> f32 {
|
pub fn magnitude(&self) -> Float {
|
||||||
(self.x * self.x + self.y * self.y + self.z * self.z + self.w * self.w).sqrt()
|
(self.x * self.x + self.y * self.y + self.z * self.z + self.w * self.w).sqrt()
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -47,23 +57,6 @@ impl Tuple {
|
|||||||
}
|
}
|
||||||
|
|
||||||
/// Reflects vector v across normal n.
|
/// Reflects vector v across normal n.
|
||||||
///
|
|
||||||
/// # Examples
|
|
||||||
/// ```
|
|
||||||
/// use rtchallenge::tuples::{reflect, Tuple};
|
|
||||||
///
|
|
||||||
/// // Reflecting a vector approaching at 45°
|
|
||||||
/// let v = Tuple::vector(1., -1., 0.);
|
|
||||||
/// let n = Tuple::vector(0., 1., 0.);
|
|
||||||
/// let r = reflect(v, n);
|
|
||||||
/// assert_eq!(r, Tuple::vector(1., 1., 0.));
|
|
||||||
///
|
|
||||||
/// // Reflecting off a slanted surface.
|
|
||||||
/// let v = Tuple::vector(0., -1., 0.);
|
|
||||||
/// let n = Tuple::vector(2_f32.sqrt() / 2., 2_f32.sqrt() / 2., 0.);
|
|
||||||
/// let r = reflect(v, n);
|
|
||||||
/// assert_eq!(r, Tuple::vector(1., 0., 0.));
|
|
||||||
/// ```
|
|
||||||
pub fn reflect(v: Tuple, n: Tuple) -> Tuple {
|
pub fn reflect(v: Tuple, n: Tuple) -> Tuple {
|
||||||
v - n * 2. * dot(v, n)
|
v - n * 2. * dot(v, n)
|
||||||
}
|
}
|
||||||
@@ -80,9 +73,9 @@ impl Add for Tuple {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Div<f32> for Tuple {
|
impl Div<Float> for Tuple {
|
||||||
type Output = Self;
|
type Output = Self;
|
||||||
fn div(self, rhs: f32) -> Self::Output {
|
fn div(self, rhs: Float) -> Self::Output {
|
||||||
Self::Output {
|
Self::Output {
|
||||||
x: self.x / rhs,
|
x: self.x / rhs,
|
||||||
y: self.y / rhs,
|
y: self.y / rhs,
|
||||||
@@ -92,9 +85,9 @@ impl Div<f32> for Tuple {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Mul<f32> for Tuple {
|
impl Mul<Float> for Tuple {
|
||||||
type Output = Self;
|
type Output = Self;
|
||||||
fn mul(self, rhs: f32) -> Self::Output {
|
fn mul(self, rhs: Float) -> Self::Output {
|
||||||
Self::Output {
|
Self::Output {
|
||||||
x: self.x * rhs,
|
x: self.x * rhs,
|
||||||
y: self.y * rhs,
|
y: self.y * rhs,
|
||||||
@@ -104,7 +97,7 @@ impl Mul<f32> for Tuple {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Mul<Tuple> for f32 {
|
impl Mul<Tuple> for Float {
|
||||||
type Output = Tuple;
|
type Output = Tuple;
|
||||||
fn mul(self, rhs: Tuple) -> Self::Output {
|
fn mul(self, rhs: Tuple) -> Self::Output {
|
||||||
Self::Output {
|
Self::Output {
|
||||||
@@ -147,7 +140,7 @@ impl PartialEq for Tuple {
|
|||||||
&& ((self.w - rhs.w).abs() < EPSILON)
|
&& ((self.w - rhs.w).abs() < EPSILON)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
pub fn dot(a: Tuple, b: Tuple) -> f32 {
|
pub fn dot(a: Tuple, b: Tuple) -> Float {
|
||||||
a.x * b.x + a.y * b.y + a.z * b.z + a.w * b.w
|
a.x * b.x + a.y * b.y + a.z * b.z + a.w * b.w
|
||||||
}
|
}
|
||||||
pub fn cross(a: Tuple, b: Tuple) -> Tuple {
|
pub fn cross(a: Tuple, b: Tuple) -> Tuple {
|
||||||
@@ -158,17 +151,29 @@ pub fn cross(a: Tuple, b: Tuple) -> Tuple {
|
|||||||
)
|
)
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Copy, Clone, Debug)]
|
#[derive(Copy, Clone, Debug, Default)]
|
||||||
pub struct Color {
|
pub struct Color {
|
||||||
pub red: f32,
|
pub red: Float,
|
||||||
pub green: f32,
|
pub green: Float,
|
||||||
pub blue: f32,
|
pub blue: Float,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Color {
|
impl Color {
|
||||||
pub const fn new(red: f32, green: f32, blue: f32) -> Color {
|
pub const fn new(red: Float, green: Float, blue: Float) -> Color {
|
||||||
Color { red, green, blue }
|
Color { red, green, blue }
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
impl From<[Float; 3]> for Color {
|
||||||
|
fn from(rgb: [Float; 3]) -> Self {
|
||||||
|
Color {
|
||||||
|
red: rgb[0],
|
||||||
|
green: rgb[1],
|
||||||
|
blue: rgb[2],
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
impl PartialEq for Color {
|
impl PartialEq for Color {
|
||||||
fn eq(&self, rhs: &Color) -> bool {
|
fn eq(&self, rhs: &Color) -> bool {
|
||||||
((self.red - rhs.red).abs() < EPSILON)
|
((self.red - rhs.red).abs() < EPSILON)
|
||||||
@@ -176,6 +181,7 @@ impl PartialEq for Color {
|
|||||||
&& ((self.blue - rhs.blue).abs() < EPSILON)
|
&& ((self.blue - rhs.blue).abs() < EPSILON)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Add for Color {
|
impl Add for Color {
|
||||||
type Output = Self;
|
type Output = Self;
|
||||||
fn add(self, other: Self) -> Self {
|
fn add(self, other: Self) -> Self {
|
||||||
@@ -187,9 +193,9 @@ impl Add for Color {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Div<f32> for Color {
|
impl Div<Float> for Color {
|
||||||
type Output = Self;
|
type Output = Self;
|
||||||
fn div(self, rhs: f32) -> Self::Output {
|
fn div(self, rhs: Float) -> Self::Output {
|
||||||
Self::Output {
|
Self::Output {
|
||||||
red: self.red / rhs,
|
red: self.red / rhs,
|
||||||
green: self.green / rhs,
|
green: self.green / rhs,
|
||||||
@@ -198,9 +204,9 @@ impl Div<f32> for Color {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Mul<f32> for Color {
|
impl Mul<Float> for Color {
|
||||||
type Output = Self;
|
type Output = Self;
|
||||||
fn mul(self, rhs: f32) -> Self::Output {
|
fn mul(self, rhs: Float) -> Self::Output {
|
||||||
Self::Output {
|
Self::Output {
|
||||||
red: self.red * rhs,
|
red: self.red * rhs,
|
||||||
green: self.green * rhs,
|
green: self.green * rhs,
|
||||||
@@ -209,7 +215,7 @@ impl Mul<f32> for Color {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Mul<Color> for f32 {
|
impl Mul<Color> for Float {
|
||||||
type Output = Color;
|
type Output = Color;
|
||||||
fn mul(self, rhs: Color) -> Self::Output {
|
fn mul(self, rhs: Color) -> Self::Output {
|
||||||
Self::Output {
|
Self::Output {
|
||||||
@@ -219,6 +225,7 @@ impl Mul<Color> for f32 {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Mul<Color> for Color {
|
impl Mul<Color> for Color {
|
||||||
type Output = Color;
|
type Output = Color;
|
||||||
fn mul(self, rhs: Color) -> Self::Output {
|
fn mul(self, rhs: Color) -> Self::Output {
|
||||||
@@ -240,6 +247,7 @@ impl Neg for Color {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Sub for Color {
|
impl Sub for Color {
|
||||||
type Output = Self;
|
type Output = Self;
|
||||||
fn sub(self, other: Self) -> Self {
|
fn sub(self, other: Self) -> Self {
|
||||||
@@ -253,7 +261,7 @@ impl Sub for Color {
|
|||||||
|
|
||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
mod tests {
|
mod tests {
|
||||||
use super::{cross, dot, Color, Tuple, EPSILON};
|
use super::{cross, dot, reflect, Color, Float, Tuple, EPSILON};
|
||||||
#[test]
|
#[test]
|
||||||
fn is_point() {
|
fn is_point() {
|
||||||
// A tuple with w = 1 is a point
|
// A tuple with w = 1 is a point
|
||||||
@@ -341,8 +349,11 @@ mod tests {
|
|||||||
assert_eq!(1., Tuple::vector(1., 0., 0.).magnitude());
|
assert_eq!(1., Tuple::vector(1., 0., 0.).magnitude());
|
||||||
assert_eq!(1., Tuple::vector(0., 1., 0.).magnitude());
|
assert_eq!(1., Tuple::vector(0., 1., 0.).magnitude());
|
||||||
assert_eq!(1., Tuple::vector(0., 0., 1.).magnitude());
|
assert_eq!(1., Tuple::vector(0., 0., 1.).magnitude());
|
||||||
assert_eq!(14_f32.sqrt(), Tuple::vector(1., 2., 3.).magnitude());
|
assert_eq!((14. as Float).sqrt(), Tuple::vector(1., 2., 3.).magnitude());
|
||||||
assert_eq!(14_f32.sqrt(), Tuple::vector(-1., -2., -3.).magnitude());
|
assert_eq!(
|
||||||
|
(14. as Float).sqrt(),
|
||||||
|
Tuple::vector(-1., -2., -3.).magnitude()
|
||||||
|
);
|
||||||
}
|
}
|
||||||
#[test]
|
#[test]
|
||||||
fn vector_normalize() {
|
fn vector_normalize() {
|
||||||
@@ -351,7 +362,11 @@ mod tests {
|
|||||||
Tuple::vector(4., 0., 0.).normalize()
|
Tuple::vector(4., 0., 0.).normalize()
|
||||||
);
|
);
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
Tuple::vector(1. / 14_f32.sqrt(), 2. / 14_f32.sqrt(), 3. / 14_f32.sqrt()),
|
Tuple::vector(
|
||||||
|
1. / (14. as Float).sqrt(),
|
||||||
|
2. / (14. as Float).sqrt(),
|
||||||
|
3. / (14. as Float).sqrt()
|
||||||
|
),
|
||||||
Tuple::vector(1., 2., 3.).normalize()
|
Tuple::vector(1., 2., 3.).normalize()
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
@@ -405,4 +420,20 @@ mod tests {
|
|||||||
let c2 = Color::new(0.9, 1., 0.1);
|
let c2 = Color::new(0.9, 1., 0.1);
|
||||||
assert_eq!(c1 * c2, Color::new(1.0 * 0.9, 0.2 * 1., 0.4 * 0.1));
|
assert_eq!(c1 * c2, Color::new(1.0 * 0.9, 0.2 * 1., 0.4 * 0.1));
|
||||||
}
|
}
|
||||||
|
#[test]
|
||||||
|
fn reflect_approaching_at_45() {
|
||||||
|
// Reflecting a vector approaching at 45°
|
||||||
|
let v = Tuple::vector(1., -1., 0.);
|
||||||
|
let n = Tuple::vector(0., 1., 0.);
|
||||||
|
let r = reflect(v, n);
|
||||||
|
assert_eq!(r, Tuple::vector(1., 1., 0.));
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn reflect_slanted_surface() {
|
||||||
|
// Reflecting off a slanted surface.
|
||||||
|
let v = Tuple::vector(0., -1., 0.);
|
||||||
|
let n = Tuple::vector((2. as Float).sqrt() / 2., (2. as Float).sqrt() / 2., 0.);
|
||||||
|
let r = reflect(v, n);
|
||||||
|
assert_eq!(r, Tuple::vector(1., 0., 0.));
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,82 +1,46 @@
|
|||||||
|
use derive_builder::Builder;
|
||||||
|
|
||||||
use crate::{
|
use crate::{
|
||||||
intersections::{prepare_computations, Intersections, PrecomputedData},
|
intersections::{prepare_computations, schlick, Intersections, PrecomputedData},
|
||||||
lights::PointLight,
|
lights::PointLight,
|
||||||
materials::{lighting, Material},
|
materials::{lighting, Material},
|
||||||
matrices::Matrix4x4,
|
matrices::Matrix4x4,
|
||||||
rays::Ray,
|
rays::Ray,
|
||||||
spheres::{intersect, Sphere},
|
shapes::{intersect, Shape},
|
||||||
tuples::{Color, Tuple},
|
tuples::{dot, Color, Tuple},
|
||||||
BLACK, WHITE,
|
BLACK, WHITE,
|
||||||
};
|
};
|
||||||
|
|
||||||
/// World holds all drawable objects and the light(s) that illuminate them.
|
/// World holds all drawable objects and the light(s) that illuminate them.
|
||||||
///
|
#[derive(Builder, Clone, Debug, Default)]
|
||||||
/// # Examples
|
#[builder(default)]
|
||||||
/// ```
|
|
||||||
/// use rtchallenge::world::World;
|
|
||||||
///
|
|
||||||
/// let w = World::default();
|
|
||||||
/// assert!(w.objects.is_empty());
|
|
||||||
/// assert_eq!(w.light, None);
|
|
||||||
/// ```
|
|
||||||
|
|
||||||
#[derive(Debug, Default)]
|
|
||||||
pub struct World {
|
pub struct World {
|
||||||
// TODO(wathiede): make this a list of abstract Light traits.
|
pub lights: Vec<PointLight>,
|
||||||
pub light: Option<PointLight>,
|
pub objects: Vec<Shape>,
|
||||||
pub objects: Vec<Sphere>,
|
|
||||||
}
|
}
|
||||||
|
|
||||||
impl World {
|
impl World {
|
||||||
/// Creates a world suitable for use across multiple tests in from the book.
|
/// Creates a world suitable for use across multiple tests in from the book.
|
||||||
///
|
|
||||||
/// # Examples
|
|
||||||
/// ```
|
|
||||||
/// use rtchallenge::world::World;
|
|
||||||
///
|
|
||||||
/// let w = World::test_world();
|
|
||||||
/// assert_eq!(w.objects.len(), 2);
|
|
||||||
/// assert!(w.light.is_some());
|
|
||||||
/// ```
|
|
||||||
pub fn test_world() -> World {
|
pub fn test_world() -> World {
|
||||||
let light = PointLight::new(Tuple::point(-10., 10., -10.), WHITE);
|
let light = PointLight::new(Tuple::point(-10., 10., -10.), WHITE);
|
||||||
let mut s1 = Sphere::default();
|
let mut s1 = Shape::sphere();
|
||||||
s1.material = Material {
|
s1.material = Material {
|
||||||
color: Color::new(0.8, 1., 0.6),
|
color: [0.8, 1., 0.6].into(),
|
||||||
diffuse: 0.7,
|
diffuse: 0.7,
|
||||||
specular: 0.2,
|
specular: 0.2,
|
||||||
..Material::default()
|
..Material::default()
|
||||||
};
|
};
|
||||||
let mut s2 = Sphere::default();
|
let mut s2 = Shape::sphere();
|
||||||
s2.transform = Matrix4x4::scaling(0.5, 0.5, 0.5);
|
s2.set_transform(Matrix4x4::scaling(0.5, 0.5, 0.5));
|
||||||
World {
|
World {
|
||||||
light: Some(light),
|
lights: vec![light],
|
||||||
objects: vec![s1, s2],
|
objects: vec![s1, s2],
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Intesects the ray with this world.
|
/// Intersects the ray with this world.
|
||||||
///
|
|
||||||
/// # Examples
|
|
||||||
/// ```
|
|
||||||
/// use rtchallenge::{rays::Ray, tuples::Tuple, world::World};
|
|
||||||
///
|
|
||||||
/// let w = World::test_world();
|
|
||||||
/// let r = Ray::new(Tuple::point(0., 0., -5.), Tuple::vector(0., 0., 1.));
|
|
||||||
/// let xs = w.intersect(&r);
|
|
||||||
/// assert_eq!(xs.len(), 4);
|
|
||||||
/// assert_eq!(xs[0].t, 4.);
|
|
||||||
/// assert_eq!(xs[1].t, 4.5);
|
|
||||||
/// assert_eq!(xs[2].t, 5.5);
|
|
||||||
/// assert_eq!(xs[3].t, 6.);
|
|
||||||
/// ```
|
|
||||||
pub fn intersect(&self, r: &Ray) -> Intersections {
|
pub fn intersect(&self, r: &Ray) -> Intersections {
|
||||||
let mut xs: Vec<_> = self
|
let mut xs: Vec<_> = self.objects.iter().flat_map(|o| intersect(o, r)).collect();
|
||||||
.objects
|
|
||||||
.iter()
|
|
||||||
.map(|o| intersect(&o, &r))
|
|
||||||
.flatten()
|
|
||||||
.collect();
|
|
||||||
xs.sort_by(|i1, i2| {
|
xs.sort_by(|i1, i2| {
|
||||||
i1.t.partial_cmp(&i2.t)
|
i1.t.partial_cmp(&i2.t)
|
||||||
.expect("an intersection has a t value that is NaN")
|
.expect("an intersection has a t value that is NaN")
|
||||||
@@ -85,94 +49,507 @@ impl World {
|
|||||||
}
|
}
|
||||||
|
|
||||||
/// Compute shaded value for given precomputation.
|
/// Compute shaded value for given precomputation.
|
||||||
///
|
pub fn shade_hit(&self, comps: &PrecomputedData, remaining: usize) -> Color {
|
||||||
/// # Examples
|
let surface = self
|
||||||
/// ```
|
.lights
|
||||||
/// use rtchallenge::{
|
.iter()
|
||||||
/// intersections::{prepare_computations, Intersection},
|
.fold(Color::new(0., 0., 0.), |acc, light| {
|
||||||
/// lights::PointLight,
|
let shadowed = self.is_shadowed(comps.over_point, light);
|
||||||
/// rays::Ray,
|
let surface = lighting(
|
||||||
/// tuples::{Color, Tuple},
|
&comps.object.material,
|
||||||
/// world::World,
|
comps.object,
|
||||||
/// WHITE,
|
light,
|
||||||
/// };
|
comps.over_point,
|
||||||
///
|
comps.eyev,
|
||||||
/// // Shading an intersection.
|
comps.normalv,
|
||||||
/// let w = World::test_world();
|
shadowed,
|
||||||
/// let r = Ray::new(Tuple::point(0., 0., -5.), Tuple::vector(0., 0., 1.));
|
);
|
||||||
/// let s = &w.objects[0];
|
acc + surface
|
||||||
/// let i = Intersection::new(4., &s);
|
});
|
||||||
/// let comps = prepare_computations(&i, &r);
|
let reflected = self.reflected_color(comps, remaining);
|
||||||
/// let c = w.shade_hit(&comps);
|
let refracted = self.refracted_color(comps, remaining);
|
||||||
/// assert_eq!(c, Color::new(0.38066, 0.47583, 0.2855));
|
let material = &comps.object.material;
|
||||||
///
|
if material.reflective > 0. && material.transparency > 0. {
|
||||||
/// // Shading an intersection from the inside.
|
let reflectance = schlick(comps);
|
||||||
/// let mut w = World::test_world();
|
surface + reflected * reflectance + refracted * (1. - reflectance)
|
||||||
/// w.light = Some(PointLight::new(Tuple::point(0., 0.25, 0.), WHITE));
|
} else {
|
||||||
/// let r = Ray::new(Tuple::point(0., 0., 0.), Tuple::vector(0., 0., 1.));
|
surface + reflected + refracted
|
||||||
/// let s = &w.objects[1];
|
}
|
||||||
/// let i = Intersection::new(0.5, &s);
|
|
||||||
/// let comps = prepare_computations(&i, &r);
|
|
||||||
/// let c = w.shade_hit(&comps);
|
|
||||||
/// assert_eq!(c, Color::new(0.90498, 0.90498, 0.90498));
|
|
||||||
/// ```
|
|
||||||
pub fn shade_hit(&self, comps: &PrecomputedData) -> Color {
|
|
||||||
// TODO(wathiede): support multiple light sources by iterating over all
|
|
||||||
// the light sources and summing the calls to lighting.
|
|
||||||
lighting(
|
|
||||||
&comps.object.material,
|
|
||||||
&self.light.as_ref().expect("World has no lights"),
|
|
||||||
comps.point,
|
|
||||||
comps.eyev,
|
|
||||||
comps.normalv,
|
|
||||||
)
|
|
||||||
}
|
}
|
||||||
/// Compute color for given ray fired at the world.
|
/// Compute color for given ray fired at the world.
|
||||||
///
|
pub fn color_at(&self, r: &Ray, remaining: usize) -> Color {
|
||||||
/// # Examples
|
let xs = self.intersect(r);
|
||||||
/// ```
|
match xs.hit() {
|
||||||
/// use rtchallenge::{
|
|
||||||
/// intersections::{prepare_computations, Intersection},
|
|
||||||
/// lights::PointLight,
|
|
||||||
/// rays::Ray,
|
|
||||||
/// tuples::{Color, Tuple},
|
|
||||||
/// world::World,
|
|
||||||
/// BLACK,
|
|
||||||
/// };
|
|
||||||
///
|
|
||||||
/// // The color when a ray misses.
|
|
||||||
/// let w = World::test_world();
|
|
||||||
/// let r = Ray::new(Tuple::point(0., 0., -5.), Tuple::vector(0., 1., 0.));
|
|
||||||
/// let c = w.color_at(&r);
|
|
||||||
/// assert_eq!(c, BLACK);
|
|
||||||
///
|
|
||||||
/// // The color when a ray hits.
|
|
||||||
/// let w = World::test_world();
|
|
||||||
/// let r = Ray::new(Tuple::point(0., 0., -5.), Tuple::vector(0., 0., 1.));
|
|
||||||
/// let c = w.color_at(&r);
|
|
||||||
/// assert_eq!(c, Color::new(0.38066, 0.47583, 0.2855));
|
|
||||||
///
|
|
||||||
/// // The color with an intersection behind the ray.
|
|
||||||
/// let w = {
|
|
||||||
/// let mut w = World::test_world();
|
|
||||||
/// let mut outer = &mut w.objects[0];
|
|
||||||
/// outer.material.ambient = 1.;
|
|
||||||
/// let inner = &mut w.objects[1];
|
|
||||||
/// inner.material.ambient = 1.;
|
|
||||||
/// w
|
|
||||||
/// };
|
|
||||||
/// let inner = &w.objects[1];
|
|
||||||
/// let r = Ray::new(Tuple::point(0., 0., 0.75), Tuple::vector(0., 0., -1.));
|
|
||||||
/// let c = w.color_at(&r);
|
|
||||||
/// assert_eq!(c, inner.material.color);
|
|
||||||
/// ```
|
|
||||||
pub fn color_at(&self, r: &Ray) -> Color {
|
|
||||||
match self.intersect(r).hit() {
|
|
||||||
Some(hit) => {
|
Some(hit) => {
|
||||||
let comps = prepare_computations(&hit, r);
|
let comps = prepare_computations(hit, r, &xs);
|
||||||
self.shade_hit(&comps)
|
self.shade_hit(&comps, remaining)
|
||||||
}
|
}
|
||||||
None => BLACK,
|
None => BLACK,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Determine if point in world is in a shadow.
|
||||||
|
pub fn is_shadowed(&self, point: Tuple, light: &PointLight) -> bool {
|
||||||
|
let v = light.position - point;
|
||||||
|
let distance = v.magnitude();
|
||||||
|
let direction = v.normalize();
|
||||||
|
|
||||||
|
let r = Ray::new(point, direction);
|
||||||
|
let intersections = self.intersect(&r);
|
||||||
|
if let Some(h) = intersections.hit() {
|
||||||
|
return h.t < distance;
|
||||||
|
}
|
||||||
|
false
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Compute reflected color for the given precomputation.
|
||||||
|
pub fn reflected_color(&self, comps: &PrecomputedData, remaining: usize) -> Color {
|
||||||
|
if remaining == 0 {
|
||||||
|
return BLACK;
|
||||||
|
}
|
||||||
|
if comps.object.material.reflective == 0. {
|
||||||
|
return BLACK;
|
||||||
|
}
|
||||||
|
// Fire reflected ray to figure out color.
|
||||||
|
let reflect_ray = Ray::new(comps.over_point, comps.reflectv);
|
||||||
|
let color = self.color_at(&reflect_ray, remaining - 1);
|
||||||
|
color * comps.object.material.reflective
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Compute refracted color for the given precomputation.
|
||||||
|
pub fn refracted_color(&self, comps: &PrecomputedData, remaining: usize) -> Color {
|
||||||
|
if remaining == 0 {
|
||||||
|
return BLACK;
|
||||||
|
}
|
||||||
|
if comps.object.material.transparency == 0. {
|
||||||
|
return BLACK;
|
||||||
|
}
|
||||||
|
// Find the ratio of the first index of refraction to the secon
|
||||||
|
// (This is inverted from the definition of Snell's Law.)
|
||||||
|
let n_ratio = comps.n1 / comps.n2;
|
||||||
|
|
||||||
|
// cos(theta_i) is the same as the dot product of the two vectors
|
||||||
|
// TODO(wathiede): is cosine faster than doc productings?
|
||||||
|
let cos_i = dot(comps.eyev, comps.normalv);
|
||||||
|
|
||||||
|
// Find the sin(theta_t)^2 via trigonometric identity.
|
||||||
|
let sin2_t = n_ratio * n_ratio * (1. - cos_i * cos_i);
|
||||||
|
|
||||||
|
if sin2_t > 1. {
|
||||||
|
// Total internal reflection.
|
||||||
|
return BLACK;
|
||||||
|
}
|
||||||
|
// Find cos(theta_t) via trigonometric identity.
|
||||||
|
let cos_t = (1. - sin2_t).sqrt();
|
||||||
|
// Compute the direction of the refracted ray.
|
||||||
|
let direction = comps.normalv * (n_ratio * cos_i - cos_t) - comps.eyev * n_ratio;
|
||||||
|
// Create the refracted ray.
|
||||||
|
let refract_ray = Ray::new(comps.under_point, direction);
|
||||||
|
// Find he color of the refracted ray, making sure to multiply by the transparency value to
|
||||||
|
// account for any opacity.
|
||||||
|
self.color_at(&refract_ray, remaining - 1) * comps.object.material.transparency
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
mod tests {
|
||||||
|
use crate::{
|
||||||
|
float::consts::SQRT_2,
|
||||||
|
intersections::{prepare_computations, Intersection, Intersections},
|
||||||
|
lights::PointLight,
|
||||||
|
materials::MaterialBuilder,
|
||||||
|
matrices::translation,
|
||||||
|
patterns::test_pattern,
|
||||||
|
rays::Ray,
|
||||||
|
shapes::{plane, sphere, Shape, ShapeBuilder},
|
||||||
|
tuples::{point, vector},
|
||||||
|
world::{World, WorldBuilder},
|
||||||
|
Float, BLACK, WHITE,
|
||||||
|
};
|
||||||
|
|
||||||
|
mod intersect {
|
||||||
|
use super::*;
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn intersection_with_test_world() {
|
||||||
|
let w = World::test_world();
|
||||||
|
let r = Ray::new(point(0., 0., -5.), vector(0., 0., 1.));
|
||||||
|
let xs = w.intersect(&r);
|
||||||
|
assert_eq!(xs.len(), 4);
|
||||||
|
assert_eq!(xs[0].t, 4.);
|
||||||
|
assert_eq!(xs[1].t, 4.5);
|
||||||
|
assert_eq!(xs[2].t, 5.5);
|
||||||
|
assert_eq!(xs[3].t, 6.);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
mod world {
|
||||||
|
use super::*;
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn default_world() {
|
||||||
|
let w = World::default();
|
||||||
|
assert!(w.objects.is_empty());
|
||||||
|
assert_eq!(w.lights.len(), 0);
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn test_world() {
|
||||||
|
let w = World::test_world();
|
||||||
|
assert_eq!(w.objects.len(), 2);
|
||||||
|
assert!(!w.lights.is_empty());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
mod shade_hit {
|
||||||
|
use super::*;
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn shading_an_intersection() {
|
||||||
|
// Shading an intersection.
|
||||||
|
let w = World::test_world();
|
||||||
|
let r = Ray::new(point(0., 0., -5.), vector(0., 0., 1.));
|
||||||
|
let s = &w.objects[0];
|
||||||
|
let xs = Intersections::from(Intersection::new(4., &s));
|
||||||
|
let comps = prepare_computations(&xs[0], &r, &xs);
|
||||||
|
let c = w.shade_hit(&comps, 1);
|
||||||
|
assert_eq!(c, [0.38066, 0.47583, 0.2855].into());
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn shading_an_intersection_from_inside() {
|
||||||
|
// Shading an intersection from the inside.
|
||||||
|
let mut w = World::test_world();
|
||||||
|
w.lights = vec![PointLight::new(point(0., 0.25, 0.), WHITE)];
|
||||||
|
let r = Ray::new(point(0., 0., 0.), vector(0., 0., 1.));
|
||||||
|
let s = &w.objects[1];
|
||||||
|
let xs = Intersections::from(Intersection::new(0.5, &s));
|
||||||
|
let comps = prepare_computations(&xs[0], &r, &xs);
|
||||||
|
let c = w.shade_hit(&comps, 1);
|
||||||
|
assert_eq!(c, [0.90498, 0.90498, 0.90498].into());
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn shading_an_intersection_in_shadow() {
|
||||||
|
// Shading with an intersection in shadow.
|
||||||
|
let mut w = World::default();
|
||||||
|
w.lights = vec![PointLight::new(point(0., 0., -10.), WHITE)];
|
||||||
|
let s1 = Shape::sphere();
|
||||||
|
let mut s2 = Shape::sphere();
|
||||||
|
s2.set_transform(translation(0., 0., 10.));
|
||||||
|
w.objects = vec![s1, s2.clone()];
|
||||||
|
let r = Ray::new(point(0., 0., 5.), vector(0., 0., 1.));
|
||||||
|
let xs = Intersections::from(Intersection::new(4., &s2));
|
||||||
|
let comps = prepare_computations(&xs[0], &r, &xs);
|
||||||
|
let c = w.shade_hit(&comps, 1);
|
||||||
|
assert_eq!(c, [0.1, 0.1, 0.1].into());
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn shading_with_a_reflective_material() -> Result<(), Box<dyn std::error::Error>> {
|
||||||
|
// Shading with a reflective material.
|
||||||
|
let mut w = World::test_world();
|
||||||
|
let shape = ShapeBuilder::plane()
|
||||||
|
.material(MaterialBuilder::default().reflective(0.5).build()?)
|
||||||
|
.transform(translation(0., -1., 0.))
|
||||||
|
.build()?;
|
||||||
|
w.objects.push(shape.clone());
|
||||||
|
let r = Ray::new(
|
||||||
|
point(0., 0., -3.),
|
||||||
|
vector(0., -(2 as Float).sqrt() / 2., (2 as Float).sqrt() / 2.),
|
||||||
|
);
|
||||||
|
let xs = Intersections::from(Intersection::new((2 as Float).sqrt(), &shape));
|
||||||
|
let comps = prepare_computations(&xs[0], &r, &xs);
|
||||||
|
let c = w.shade_hit(&comps, 1);
|
||||||
|
assert_eq!(c, [0.87676, 0.92434, 0.82917].into());
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn shading_with_a_transparent_material() -> Result<(), Box<dyn std::error::Error>> {
|
||||||
|
// shade_hit() with a transparent material.
|
||||||
|
let mut w = World::test_world();
|
||||||
|
let floor = plane()
|
||||||
|
.transform(translation(0., -1., 0.))
|
||||||
|
.material(
|
||||||
|
MaterialBuilder::default()
|
||||||
|
.transparency(0.5)
|
||||||
|
.refractive_index(1.5)
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.build()?;
|
||||||
|
w.objects.push(floor.clone());
|
||||||
|
let ball = sphere()
|
||||||
|
.material(
|
||||||
|
MaterialBuilder::default()
|
||||||
|
.color([1., 0., 0.])
|
||||||
|
.ambient(0.5)
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.transform(translation(0., -3.5, -0.5))
|
||||||
|
.build()?;
|
||||||
|
w.objects.push(ball);
|
||||||
|
let r = Ray::new(
|
||||||
|
point(0., 0., -3.),
|
||||||
|
vector(0., -(2 as Float).sqrt() / 2., (2 as Float).sqrt() / 2.),
|
||||||
|
);
|
||||||
|
let xs = Intersections::new(vec![Intersection::new((2 as Float).sqrt(), &floor)]);
|
||||||
|
let comps = prepare_computations(&xs[0], &r, &xs);
|
||||||
|
let c = w.shade_hit(&comps, 5);
|
||||||
|
assert_eq!(c, [0.93642, 0.68642, 0.68643].into());
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn reflective_transparent_material() -> Result<(), Box<dyn std::error::Error>> {
|
||||||
|
let mut w = World::test_world();
|
||||||
|
let floor = plane()
|
||||||
|
.transform(translation(0., -1., 0.))
|
||||||
|
.material(
|
||||||
|
MaterialBuilder::default()
|
||||||
|
.reflective(0.5)
|
||||||
|
.transparency(0.5)
|
||||||
|
.refractive_index(1.5)
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.build()?;
|
||||||
|
w.objects.push(floor.clone());
|
||||||
|
let ball = sphere()
|
||||||
|
.transform(translation(0., -3.5, -0.5))
|
||||||
|
.material(
|
||||||
|
MaterialBuilder::default()
|
||||||
|
.color([1., 0., 0.])
|
||||||
|
.ambient(0.5)
|
||||||
|
.build()?,
|
||||||
|
)
|
||||||
|
.build()?;
|
||||||
|
w.objects.push(ball);
|
||||||
|
let r = Ray::new(point(0., 0., -3.), vector(0., -SQRT_2 / 2., SQRT_2 / 2.));
|
||||||
|
let xs = Intersections::new(vec![Intersection::new(SQRT_2, &floor)]);
|
||||||
|
let comps = prepare_computations(&xs[0], &r, &xs);
|
||||||
|
let color = w.shade_hit(&comps, 5);
|
||||||
|
assert_eq!(color, [0.93391, 0.69643, 0.69243].into());
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
mod color_at {
|
||||||
|
use super::*;
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn color_when_ray_misses() {
|
||||||
|
// The color when a ray misses.
|
||||||
|
let w = World::test_world();
|
||||||
|
let r = Ray::new(point(0., 0., -5.), vector(0., 1., 0.));
|
||||||
|
let c = w.color_at(&r, 1);
|
||||||
|
assert_eq!(c, BLACK);
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn color_when_ray_hits() {
|
||||||
|
// The color when a ray hits.
|
||||||
|
let w = World::test_world();
|
||||||
|
let r = Ray::new(point(0., 0., -5.), vector(0., 0., 1.));
|
||||||
|
let c = w.color_at(&r, 1);
|
||||||
|
assert_eq!(c, [0.38066, 0.47583, 0.2855].into());
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn color_with_an_intersection_behind_ray() {
|
||||||
|
// The color with an intersection behind the ray.
|
||||||
|
let w = {
|
||||||
|
let mut w = World::test_world();
|
||||||
|
let mut outer = &mut w.objects[0];
|
||||||
|
outer.material.ambient = 1.;
|
||||||
|
let inner = &mut w.objects[1];
|
||||||
|
inner.material.ambient = 1.;
|
||||||
|
w
|
||||||
|
};
|
||||||
|
let r = Ray::new(point(0., 0., 0.75), vector(0., 0., -1.));
|
||||||
|
let c = w.color_at(&r, 1);
|
||||||
|
// inner.material.color is WHITE
|
||||||
|
assert_eq!(c, WHITE);
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn ensure_mutually_reflective_surfaces_terminate() -> Result<(), Box<dyn std::error::Error>>
|
||||||
|
{
|
||||||
|
// Ensure mutually reflective surfaces don't infinitely recurse.
|
||||||
|
let lower = ShapeBuilder::plane()
|
||||||
|
.material(MaterialBuilder::default().reflective(1.).build()?)
|
||||||
|
.transform(translation(0., -1., 0.))
|
||||||
|
.build()?;
|
||||||
|
let upper = ShapeBuilder::plane()
|
||||||
|
.material(MaterialBuilder::default().reflective(1.).build()?)
|
||||||
|
.transform(translation(0., 1., 0.))
|
||||||
|
.build()?;
|
||||||
|
let w = WorldBuilder::default()
|
||||||
|
.lights(vec![PointLight::new(point(0., 0., 0.), WHITE)])
|
||||||
|
.objects(vec![lower, upper])
|
||||||
|
.build()?;
|
||||||
|
let r = Ray::new(point(0., 0., 0.), vector(0., 1., 0.));
|
||||||
|
// This should complete without stack overflow.
|
||||||
|
w.color_at(&r, 1);
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
mod is_shadowed {
|
||||||
|
use super::*;
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn no_shadow_when_nothing_collinear_with_point_and_light() {
|
||||||
|
// There is no shadow when nothing is collinear with point and light.
|
||||||
|
let w = World::test_world();
|
||||||
|
let light = &w.lights[0];
|
||||||
|
|
||||||
|
let p = point(0., 10., 0.);
|
||||||
|
assert_eq!(w.is_shadowed(p, light), false);
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn shadow_when_object_between_point_and_light() {
|
||||||
|
// The shadow when an object is between the point and the light.
|
||||||
|
let w = World::test_world();
|
||||||
|
let light = &w.lights[0];
|
||||||
|
|
||||||
|
let p = point(10., -10., 10.);
|
||||||
|
assert_eq!(w.is_shadowed(p, light), true);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn no_shadow_when_object_behind_light() {
|
||||||
|
// There is no shadow when an object is behind the light.
|
||||||
|
let w = World::test_world();
|
||||||
|
let light = &w.lights[0];
|
||||||
|
|
||||||
|
let p = point(-20., 20., -20.);
|
||||||
|
assert_eq!(w.is_shadowed(p, light), false);
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn no_shadow_when_object_behind_point() {
|
||||||
|
// There is no shadow when an object is behind the point.
|
||||||
|
let w = World::test_world();
|
||||||
|
let light = &w.lights[0];
|
||||||
|
|
||||||
|
let p = point(-2., 2., -2.);
|
||||||
|
assert_eq!(w.is_shadowed(p, light), false);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
mod reflected_color {
|
||||||
|
use super::*;
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn reflected_color_for_nonreflective_material() {
|
||||||
|
// The reflected color for a nonreflective material.
|
||||||
|
let r = Ray::new(point(0., 0., 0.), vector(0., 0., 1.));
|
||||||
|
let mut w = World::test_world();
|
||||||
|
w.objects[1].material.ambient = 1.;
|
||||||
|
let xs = Intersections::from(Intersection::new(1., &w.objects[1]));
|
||||||
|
let comps = prepare_computations(&xs[0], &r, &xs);
|
||||||
|
let c = w.reflected_color(&comps, 1);
|
||||||
|
assert_eq!(c, BLACK);
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn reflected_color_for_reflective_material() -> Result<(), Box<dyn std::error::Error>> {
|
||||||
|
// The reflected color for a reflective material.
|
||||||
|
let mut w = World::test_world();
|
||||||
|
let shape = ShapeBuilder::plane()
|
||||||
|
.material(MaterialBuilder::default().reflective(0.5).build()?)
|
||||||
|
.transform(translation(0., -1., 0.))
|
||||||
|
.build()?;
|
||||||
|
w.objects.push(shape.clone());
|
||||||
|
let r = Ray::new(
|
||||||
|
point(0., 0., -3.),
|
||||||
|
vector(0., -(2 as Float).sqrt() / 2., (2 as Float).sqrt() / 2.),
|
||||||
|
);
|
||||||
|
let xs = Intersections::from(Intersection::new((2 as Float).sqrt(), &shape));
|
||||||
|
let comps = prepare_computations(&xs[0], &r, &xs);
|
||||||
|
let c = w.reflected_color(&comps, 1);
|
||||||
|
assert_eq!(c, [0.19033, 0.23791, 0.14274].into());
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn reflected_color_at_maximum_recursion_depth() -> Result<(), Box<dyn std::error::Error>> {
|
||||||
|
// The reflected color at the maximum recursion depth.
|
||||||
|
let mut w = World::test_world();
|
||||||
|
let shape = ShapeBuilder::plane()
|
||||||
|
.material(MaterialBuilder::default().reflective(0.5).build()?)
|
||||||
|
.transform(translation(0., -1., 0.))
|
||||||
|
.build()?;
|
||||||
|
w.objects.push(shape.clone());
|
||||||
|
let r = Ray::new(
|
||||||
|
point(0., 0., -3.),
|
||||||
|
vector(0., -(2 as Float).sqrt() / 2., (2 as Float).sqrt() / 2.),
|
||||||
|
);
|
||||||
|
let xs = Intersections::from(Intersection::new((2 as Float).sqrt(), &shape));
|
||||||
|
let comps = prepare_computations(&xs[0], &r, &xs);
|
||||||
|
let _ = w.reflected_color(&comps, 0);
|
||||||
|
// Just needs to get here without infinite recursion.
|
||||||
|
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
mod refracted_color {
|
||||||
|
use super::*;
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn refracted_color_with_opaque_surface() {
|
||||||
|
// The refracted color with an opaque surface.
|
||||||
|
let w = World::test_world();
|
||||||
|
let shape = &w.objects[0];
|
||||||
|
let r = Ray::new(point(0., 0., -5.), vector(0., 0., 1.));
|
||||||
|
let xs = Intersections::new(vec![
|
||||||
|
Intersection::new(4., &shape),
|
||||||
|
Intersection::new(6., &shape),
|
||||||
|
]);
|
||||||
|
let comps = prepare_computations(&xs[0], &r, &xs);
|
||||||
|
let c = w.refracted_color(&comps, 5);
|
||||||
|
assert_eq!(c, BLACK);
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn refracted_color_at_maximum_recursion_depth() {
|
||||||
|
// The refracted color at the maximum recursive depth.
|
||||||
|
let mut w = World::test_world();
|
||||||
|
w.objects[0].material.transparency = 1.0;
|
||||||
|
w.objects[0].material.refractive_index = 1.5;
|
||||||
|
let shape = &w.objects[0];
|
||||||
|
let r = Ray::new(point(0., 0., -5.), vector(0., 0., 1.));
|
||||||
|
let xs = Intersections::new(vec![
|
||||||
|
Intersection::new(4., &shape),
|
||||||
|
Intersection::new(6., &shape),
|
||||||
|
]);
|
||||||
|
let comps = prepare_computations(&xs[0], &r, &xs);
|
||||||
|
let c = w.refracted_color(&comps, 0);
|
||||||
|
assert_eq!(c, BLACK);
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn refracted_color_under_total_internal_reflection() {
|
||||||
|
// The refracted color under total internal reflection.
|
||||||
|
let mut w = World::test_world();
|
||||||
|
w.objects[0].material.transparency = 1.0;
|
||||||
|
w.objects[0].material.refractive_index = 1.5;
|
||||||
|
let shape = &w.objects[0];
|
||||||
|
let r = Ray::new(point(0., 0., (2 as Float).sqrt() / 2.), vector(0., 1., 0.));
|
||||||
|
let xs = Intersections::new(vec![
|
||||||
|
Intersection::new(-(2 as Float).sqrt() / 2., &shape),
|
||||||
|
Intersection::new((2 as Float).sqrt() / 2., &shape),
|
||||||
|
]);
|
||||||
|
let comps = prepare_computations(&xs[1], &r, &xs);
|
||||||
|
let c = w.refracted_color(&comps, 5);
|
||||||
|
assert_eq!(c, BLACK);
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn refracted_color_with_a_refracted_ray() -> Result<(), Box<dyn std::error::Error>> {
|
||||||
|
// The refracted color with a refracted ray.
|
||||||
|
let mut w = World::test_world();
|
||||||
|
w.objects[0].material.ambient = 1.;
|
||||||
|
w.objects[0].material.color = test_pattern().build()?;
|
||||||
|
|
||||||
|
w.objects[1].material.transparency = 1.;
|
||||||
|
w.objects[1].material.refractive_index = 1.5;
|
||||||
|
|
||||||
|
let r = Ray::new(point(0., 0., 0.1), vector(0., 1., 0.));
|
||||||
|
let a = &w.objects[0];
|
||||||
|
let b = &w.objects[1];
|
||||||
|
let xs = Intersections::new(vec![
|
||||||
|
Intersection::new(-0.9899, &a),
|
||||||
|
Intersection::new(-0.4899, &b),
|
||||||
|
Intersection::new(0.4899, &b),
|
||||||
|
Intersection::new(0.9899, &a),
|
||||||
|
]);
|
||||||
|
let comps = prepare_computations(&xs[2], &r, &xs);
|
||||||
|
let c = w.refracted_color(&comps, 5);
|
||||||
|
assert_eq!(c, [0., 0.99887, 0.04721].into());
|
||||||
|
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
2299
rtiow/Cargo.lock
generated
2299
rtiow/Cargo.lock
generated
File diff suppressed because it is too large
Load Diff
@@ -1,13 +1,15 @@
|
|||||||
[workspace]
|
[workspace]
|
||||||
|
resolver = "2"
|
||||||
|
|
||||||
members = [
|
members = [
|
||||||
"noise_explorer",
|
"noise_explorer",
|
||||||
"noise_explorer_warp",
|
"noise_explorer_warp",
|
||||||
"renderer",
|
"renderer",
|
||||||
"tracer",
|
"tracer",
|
||||||
|
"vec3",
|
||||||
]
|
]
|
||||||
|
|
||||||
[profile]
|
|
||||||
[profile.release]
|
[profile.release]
|
||||||
debug = true
|
debug = true
|
||||||
|
|
||||||
|
[profile.dev]
|
||||||
|
opt-level = 3
|
||||||
|
|||||||
@@ -2,19 +2,19 @@
|
|||||||
name = "noise_explorer"
|
name = "noise_explorer"
|
||||||
version = "0.1.0"
|
version = "0.1.0"
|
||||||
authors = ["Bill Thiede <git@xinu.tv>"]
|
authors = ["Bill Thiede <git@xinu.tv>"]
|
||||||
edition = "2018"
|
edition = "2021"
|
||||||
|
|
||||||
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
|
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
|
||||||
|
|
||||||
[dependencies]
|
[dependencies]
|
||||||
actix-web = "0.7.8"
|
actix-web = "0.7.19"
|
||||||
askama = "0.7.1"
|
askama = "0.7.2"
|
||||||
image = "0.22.3"
|
image = "0.22.5"
|
||||||
log = "0.4.5"
|
log = "0.4.17"
|
||||||
rand = "0.8.3"
|
rand = "0.8.5"
|
||||||
rand_xorshift = "0.3.0"
|
rand_xorshift = "0.3.0"
|
||||||
renderer = { path = "../renderer" }
|
renderer = { path = "../renderer" }
|
||||||
serde = "1.0.79"
|
serde = "1.0.152"
|
||||||
serde_derive = "1.0.79"
|
serde_derive = "1.0.152"
|
||||||
stderrlog = "0.4.1"
|
stderrlog = "0.4.3"
|
||||||
structopt = "0.2.10"
|
structopt = "0.2.18"
|
||||||
|
|||||||
@@ -1 +1,2 @@
|
|||||||
|
imports_granularity = "Crate"
|
||||||
format_code_in_doc_comments = true
|
format_code_in_doc_comments = true
|
||||||
@@ -1,44 +1,26 @@
|
|||||||
use std::fmt;
|
use std::{fmt, time::SystemTime};
|
||||||
use std::time::SystemTime;
|
|
||||||
|
|
||||||
use actix_web::http;
|
use actix_web::{http, middleware, server, App, HttpRequest, HttpResponse, Path, Query, Result};
|
||||||
use actix_web::middleware;
|
|
||||||
use actix_web::server;
|
|
||||||
use actix_web::App;
|
|
||||||
use actix_web::HttpRequest;
|
|
||||||
use actix_web::HttpResponse;
|
|
||||||
use actix_web::Path;
|
|
||||||
use actix_web::Query;
|
|
||||||
use actix_web::Result;
|
|
||||||
use askama::Template;
|
use askama::Template;
|
||||||
use image;
|
|
||||||
use log::info;
|
use log::info;
|
||||||
use rand::SeedableRng;
|
use rand::SeedableRng;
|
||||||
use rand_xorshift::XorShiftRng;
|
use rand_xorshift::XorShiftRng;
|
||||||
use serde_derive::Deserialize;
|
use serde_derive::Deserialize;
|
||||||
use structopt::StructOpt;
|
use structopt::StructOpt;
|
||||||
|
|
||||||
use renderer::noise;
|
use renderer::{
|
||||||
use renderer::noise::lode::Lode;
|
noise,
|
||||||
use renderer::noise::perlin::Perlin;
|
noise::{lode::Lode, perlin::Perlin, NoiseType},
|
||||||
use renderer::noise::NoiseType;
|
texture::{NoiseTexture, Texture},
|
||||||
use renderer::texture::NoiseTexture;
|
vec3::Vec3,
|
||||||
use renderer::texture::Texture;
|
};
|
||||||
use renderer::vec3::Vec3;
|
|
||||||
|
|
||||||
#[derive(Debug, StructOpt)]
|
#[derive(StructOpt)]
|
||||||
#[structopt(name = "noise_explorer", about = "CLI for exploring Perlin noise")]
|
#[structopt(name = "noise_explorer", about = "CLI for exploring Perlin noise")]
|
||||||
struct Opt {
|
struct Opt {
|
||||||
/// HTTP listen address
|
/// HTTP listen address
|
||||||
#[structopt(long = "addr", default_value = "0.0.0.0:8889")]
|
#[structopt(long = "addr", default_value = "0.0.0.0:8889")]
|
||||||
pub addr: String,
|
pub addr: String,
|
||||||
|
|
||||||
/// Width of noise images
|
|
||||||
#[structopt(short = "w", long = "width", default_value = "128")]
|
|
||||||
pub width: u32,
|
|
||||||
/// Height of noise images
|
|
||||||
#[structopt(short = "h", long = "height", default_value = "128")]
|
|
||||||
pub height: u32,
|
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Copy, Clone, Debug, Deserialize, PartialEq)]
|
#[derive(Copy, Clone, Debug, Deserialize, PartialEq)]
|
||||||
@@ -63,11 +45,6 @@ struct OptionalParams {
|
|||||||
pixel_scale: f32,
|
pixel_scale: f32,
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, Deserialize)]
|
|
||||||
struct NoiseSourceParam {
|
|
||||||
noise_source: NoiseSource,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Debug, Deserialize)]
|
#[derive(Debug, Deserialize)]
|
||||||
struct NoiseParams {
|
struct NoiseParams {
|
||||||
width: u32,
|
width: u32,
|
||||||
@@ -466,16 +443,9 @@ fn main() -> Result<(), std::io::Error> {
|
|||||||
mod tests {
|
mod tests {
|
||||||
use std::str;
|
use std::str;
|
||||||
|
|
||||||
use actix_web::http::Method;
|
use actix_web::{http::Method, test::TestServer, HttpMessage, Path, Query};
|
||||||
use actix_web::test::TestServer;
|
|
||||||
use actix_web::HttpMessage;
|
|
||||||
use actix_web::Path;
|
|
||||||
use actix_web::Query;
|
|
||||||
|
|
||||||
use super::NoiseParamsMarble;
|
use super::{NoiseParamsMarble, NoiseParamsScale, NoiseParamsTurbulence, OptionalParams};
|
||||||
use super::NoiseParamsScale;
|
|
||||||
use super::NoiseParamsTurbulence;
|
|
||||||
use super::OptionalParams;
|
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn noise_param_from_req() {
|
fn noise_param_from_req() {
|
||||||
|
|||||||
@@ -2,18 +2,18 @@
|
|||||||
name = "noise_explorer_warp"
|
name = "noise_explorer_warp"
|
||||||
version = "0.1.0"
|
version = "0.1.0"
|
||||||
authors = ["Bill Thiede <git@xinu.tv>"]
|
authors = ["Bill Thiede <git@xinu.tv>"]
|
||||||
edition = "2018"
|
edition = "2021"
|
||||||
|
|
||||||
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
|
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
|
||||||
|
|
||||||
[dependencies]
|
[dependencies]
|
||||||
askama = "0.7.1"
|
askama = "0.7.2"
|
||||||
image = "0.22.3"
|
image = "0.22.5"
|
||||||
log = "0.4.5"
|
log = "0.4.17"
|
||||||
rand = "0.5.5"
|
rand = "0.5.6"
|
||||||
renderer = { path = "../renderer" }
|
renderer = { path = "../renderer" }
|
||||||
serde = "1.0.79"
|
serde = "1.0.152"
|
||||||
serde_derive = "1.0.79"
|
serde_derive = "1.0.152"
|
||||||
stderrlog = "0.4.1"
|
stderrlog = "0.4.3"
|
||||||
structopt = "0.2.10"
|
structopt = "0.2.18"
|
||||||
warp = "0.1.20"
|
warp = "0.1.23"
|
||||||
|
|||||||
2
rtiow/noise_explorer_warp/rustfmt.toml
Normal file
2
rtiow/noise_explorer_warp/rustfmt.toml
Normal file
@@ -0,0 +1,2 @@
|
|||||||
|
imports_granularity = "Crate"
|
||||||
|
format_code_in_doc_comments = true
|
||||||
@@ -2,29 +2,41 @@
|
|||||||
name = "renderer"
|
name = "renderer"
|
||||||
version = "0.1.0"
|
version = "0.1.0"
|
||||||
authors = ["Bill Thiede <git@xinu.tv>"]
|
authors = ["Bill Thiede <git@xinu.tv>"]
|
||||||
edition = "2018"
|
edition = "2021"
|
||||||
|
|
||||||
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
|
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
|
||||||
[[bench]]
|
[[bench]]
|
||||||
harness = false
|
harness = false
|
||||||
name = "spheres"
|
name = "spheres"
|
||||||
|
[[bench]]
|
||||||
|
harness = false
|
||||||
|
name = "aabb"
|
||||||
|
|
||||||
[dependencies]
|
[dependencies]
|
||||||
chrono = "*"
|
chrono = "*"
|
||||||
core_affinity = "0.5"
|
core_affinity = "0.5"
|
||||||
cpuprofiler = { version = "0.0.3", optional = true }
|
cpuprofiler = { version = "0.0.3", optional = true }
|
||||||
image = "0.22.3"
|
image = "0.22.5"
|
||||||
lazy_static = "1.1.0"
|
lazy_static = "1.4.0"
|
||||||
log = "0.4.5"
|
log = "0.4.17"
|
||||||
num_cpus = "1.8.0"
|
num_cpus = "1.15.0"
|
||||||
rand = "0.8.3"
|
rand = "0.8.5"
|
||||||
serde = "1.0.79"
|
serde = "1.0.152"
|
||||||
serde_derive = "1.0.79"
|
serde_derive = "1.0.152"
|
||||||
serde_json = "1.0.41"
|
serde_json = "1.0.93"
|
||||||
structopt = "0.2.10"
|
structopt = "0.2.18"
|
||||||
|
vec3 = {path = "../vec3"}
|
||||||
|
stl = {path = "../../../stl"}
|
||||||
|
strum = { version = "0.24.1", features = ["derive"] }
|
||||||
|
strum_macros = "0.24.3"
|
||||||
|
thiserror = "1.0.38"
|
||||||
|
tev_client = "0.5.2"
|
||||||
|
#stl = {git = "https://git-private.z.xinu.tv/wathiede/stl"}
|
||||||
|
|
||||||
[dev-dependencies]
|
[dev-dependencies]
|
||||||
criterion = "0.2"
|
criterion = "0.4"
|
||||||
|
pretty_assertions = "1.3.0"
|
||||||
|
proptest = "1.1.0"
|
||||||
|
|
||||||
[features]
|
[features]
|
||||||
profile = ["cpuprofiler"]
|
profile = ["cpuprofiler"]
|
||||||
|
|||||||
47
rtiow/renderer/benches/aabb.rs
Normal file
47
rtiow/renderer/benches/aabb.rs
Normal file
@@ -0,0 +1,47 @@
|
|||||||
|
use criterion::*;
|
||||||
|
use renderer::{aabb::AABB, ray::Ray};
|
||||||
|
|
||||||
|
fn bench(c: &mut Criterion) {
|
||||||
|
let bb = AABB::new([1., -1., -1.], [3., 1., 1.]);
|
||||||
|
let r_hit = Ray::new([0., 0., 0.], [1., 0., 0.], 0.);
|
||||||
|
let r_miss = Ray::new([0., 0., 0.], [-1., 0., 0.], 0.);
|
||||||
|
let t_min = 0.001;
|
||||||
|
let t_max = f32::MAX;
|
||||||
|
|
||||||
|
let mut group = c.benchmark_group("aabb");
|
||||||
|
group.throughput(Throughput::Elements(1));
|
||||||
|
group.bench_with_input(BenchmarkId::new("hit_naive", "r_hit"), &r_hit, |b, r| {
|
||||||
|
b.iter(|| bb.hit_naive(*r, t_min, t_max))
|
||||||
|
});
|
||||||
|
group.bench_with_input(BenchmarkId::new("hit2", "r_hit"), &r_hit, |b, r| {
|
||||||
|
b.iter(|| bb.hit2(*r, t_min, t_max))
|
||||||
|
});
|
||||||
|
//group.bench_with_input(BenchmarkId::new("hit_precompute", "r_hit"), &r_hit, |b, r| { b.iter(|| bb.hit_precompute(*r, t_min, t_max)) });
|
||||||
|
group.bench_with_input(BenchmarkId::new("hit_fast", "r_hit"), &r_hit, |b, r| {
|
||||||
|
b.iter(|| bb.hit_fast(*r, t_min, t_max))
|
||||||
|
});
|
||||||
|
#[cfg(target_arch = "x86_64")]
|
||||||
|
group.bench_with_input(BenchmarkId::new("hit_simd", "r_hit"), &r_hit, |b, r| {
|
||||||
|
b.iter(|| bb.hit_simd(*r, t_min, t_max))
|
||||||
|
});
|
||||||
|
|
||||||
|
group.bench_with_input(BenchmarkId::new("hit_naive", "r_miss"), &r_miss, |b, r| {
|
||||||
|
b.iter(|| bb.hit_naive(*r, t_min, t_max))
|
||||||
|
});
|
||||||
|
group.bench_with_input(BenchmarkId::new("hit2", "r_miss"), &r_miss, |b, r| {
|
||||||
|
b.iter(|| bb.hit2(*r, t_min, t_max))
|
||||||
|
});
|
||||||
|
//group.bench_with_input(BenchmarkId::new("hit_precompute", "r_miss"), &r_miss, |b, r| { b.iter(|| bb.hit_precompute(*r, t_min, t_max)) });
|
||||||
|
group.bench_with_input(BenchmarkId::new("hit_fast", "r_miss"), &r_miss, |b, r| {
|
||||||
|
b.iter(|| bb.hit_fast(*r, t_min, t_max))
|
||||||
|
});
|
||||||
|
#[cfg(target_arch = "x86_64")]
|
||||||
|
group.bench_with_input(BenchmarkId::new("hit_simd", "r_miss"), &r_miss, |b, r| {
|
||||||
|
b.iter(|| bb.hit_simd(*r, t_min, t_max))
|
||||||
|
});
|
||||||
|
|
||||||
|
group.finish();
|
||||||
|
}
|
||||||
|
|
||||||
|
criterion_group!(benches, bench);
|
||||||
|
criterion_main!(benches);
|
||||||
@@ -1,15 +1,9 @@
|
|||||||
#[macro_use]
|
use criterion::*;
|
||||||
extern crate criterion;
|
|
||||||
|
|
||||||
use criterion::Criterion;
|
use renderer::{
|
||||||
use criterion::ParameterizedBenchmark;
|
hitable::Hit, material::Lambertian, ray::Ray, sphere::Sphere, texture::ConstantTexture,
|
||||||
|
vec3::Vec3,
|
||||||
use rtiow::hitable::Hit;
|
};
|
||||||
use rtiow::material::Lambertian;
|
|
||||||
use rtiow::ray::Ray;
|
|
||||||
use rtiow::sphere::Sphere;
|
|
||||||
use rtiow::texture::ConstantTexture;
|
|
||||||
use rtiow::vec3::Vec3;
|
|
||||||
|
|
||||||
fn criterion_benchmark(c: &mut Criterion) {
|
fn criterion_benchmark(c: &mut Criterion) {
|
||||||
let sphere = Sphere::new(
|
let sphere = Sphere::new(
|
||||||
@@ -23,14 +17,15 @@ fn criterion_benchmark(c: &mut Criterion) {
|
|||||||
// Miss
|
// Miss
|
||||||
Ray::new([0., 0., -2.], [0., 0., -1.], 0.),
|
Ray::new([0., 0., -2.], [0., 0., -1.], 0.),
|
||||||
];
|
];
|
||||||
c.bench(
|
let mut group = c.benchmark_group("sphere");
|
||||||
"sphere",
|
group.throughput(Throughput::Elements(1));
|
||||||
ParameterizedBenchmark::new(
|
group.bench_with_input(BenchmarkId::new("Sphere", "hit"), &rays[0], |b, r| {
|
||||||
"Sphere",
|
b.iter(|| sphere.hit(*r, 0., 1.))
|
||||||
move |b, r| b.iter(|| sphere.hit(*r, 0., 1.)),
|
});
|
||||||
rays,
|
group.bench_with_input(BenchmarkId::new("Sphere", "miss"), &rays[1], |b, r| {
|
||||||
),
|
b.iter(|| sphere.hit(*r, 0., 1.))
|
||||||
);
|
});
|
||||||
|
group.finish()
|
||||||
}
|
}
|
||||||
|
|
||||||
criterion_group!(benches, criterion_benchmark);
|
criterion_group!(benches, criterion_benchmark);
|
||||||
|
|||||||
2
rtiow/renderer/rustfmt.toml
Normal file
2
rtiow/renderer/rustfmt.toml
Normal file
@@ -0,0 +1,2 @@
|
|||||||
|
imports_granularity = "Crate"
|
||||||
|
format_code_in_doc_comments = true
|
||||||
@@ -1,9 +1,8 @@
|
|||||||
use std::fmt;
|
use std::fmt;
|
||||||
|
|
||||||
use crate::ray::Ray;
|
use crate::{ray::Ray, vec3::Vec3};
|
||||||
use crate::vec3::Vec3;
|
|
||||||
|
|
||||||
#[derive(Debug, Clone, PartialEq)]
|
#[derive(Default, Copy, Clone, PartialEq)]
|
||||||
pub struct AABB {
|
pub struct AABB {
|
||||||
bounds: [Vec3; 2],
|
bounds: [Vec3; 2],
|
||||||
}
|
}
|
||||||
@@ -30,15 +29,65 @@ fn max(x: f32, y: f32) -> f32 {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
impl fmt::Debug for AABB {
|
||||||
|
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
|
||||||
|
write!(
|
||||||
|
f,
|
||||||
|
"AABB: <{} - {}> Vol: {} Area: {}",
|
||||||
|
self.bounds[0],
|
||||||
|
self.bounds[1],
|
||||||
|
self.volume(),
|
||||||
|
self.area()
|
||||||
|
)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
impl AABB {
|
impl AABB {
|
||||||
pub fn new(min: Vec3, max: Vec3) -> AABB {
|
// Create AABB with min = f32::MAX and max = -f32::MAX. It is expected the caller will use grow to create
|
||||||
|
// a vaild AABB.
|
||||||
|
pub fn infinite() -> AABB {
|
||||||
|
AABB {
|
||||||
|
bounds: [Vec3::from(f32::MAX), Vec3::from(f32::MIN)],
|
||||||
|
}
|
||||||
|
}
|
||||||
|
pub fn new<V: Into<Vec3>>(min: V, max: V) -> AABB {
|
||||||
|
let min: Vec3 = min.into();
|
||||||
|
let max: Vec3 = max.into();
|
||||||
|
assert!(min.x <= max.x);
|
||||||
|
assert!(min.y <= max.y);
|
||||||
|
assert!(min.z <= max.z);
|
||||||
AABB { bounds: [min, max] }
|
AABB { bounds: [min, max] }
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pub fn area(&self) -> f32 {
|
||||||
|
if self.max().x == f32::MIN || self.min().x == f32::MAX {
|
||||||
|
return 0.;
|
||||||
|
}
|
||||||
|
let e = self.max() - self.min();
|
||||||
|
let v = e.x * e.y + e.y * e.z + e.z * e.x;
|
||||||
|
if v.is_finite() {
|
||||||
|
v
|
||||||
|
} else {
|
||||||
|
0.
|
||||||
|
}
|
||||||
|
}
|
||||||
pub fn volume(&self) -> f32 {
|
pub fn volume(&self) -> f32 {
|
||||||
(self.min().x - self.max().x).abs()
|
if self.max().x == f32::MIN || self.min().x == f32::MAX {
|
||||||
|
return 0.;
|
||||||
|
}
|
||||||
|
let v = (self.min().x - self.max().x).abs()
|
||||||
* (self.min().y - self.max().y).abs()
|
* (self.min().y - self.max().y).abs()
|
||||||
* (self.min().z - self.max().z).abs()
|
* (self.min().z - self.max().z).abs();
|
||||||
|
if v.is_finite() {
|
||||||
|
v
|
||||||
|
} else {
|
||||||
|
0.
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn grow(&mut self, v: Vec3) {
|
||||||
|
self.bounds[0] = vec3::min(self.bounds[0], v);
|
||||||
|
self.bounds[1] = vec3::max(self.bounds[1], v);
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn longest_axis(&self) -> usize {
|
pub fn longest_axis(&self) -> usize {
|
||||||
@@ -62,11 +111,39 @@ impl AABB {
|
|||||||
pub fn min(&self) -> Vec3 {
|
pub fn min(&self) -> Vec3 {
|
||||||
self.bounds[0]
|
self.bounds[0]
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn max(&self) -> Vec3 {
|
pub fn max(&self) -> Vec3 {
|
||||||
self.bounds[1]
|
self.bounds[1]
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn hit2(&self, r: Ray, t_min: f32, t_max: f32) -> bool {
|
// TODO(wathiede): implement branchless https://tavianator.com/cgit/dimension.git/tree/libdimension/bvh/bvh.c#n194
|
||||||
|
|
||||||
|
pub fn hit(&self, r: Ray, t_min: f32, t_max: f32) -> bool {
|
||||||
|
self.hit_simd(r, t_min, t_max).is_some()
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn hit_distance(&self, r: Ray, t_min: f32, t_max: f32) -> Option<f32> {
|
||||||
|
self.hit_simd(r, t_min, t_max)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn hit_naive(&self, r: Ray, t_min: f32, t_max: f32) -> Option<f32> {
|
||||||
|
let mut t_min = t_min;
|
||||||
|
let mut t_max = t_max;
|
||||||
|
for axis in 0..3 {
|
||||||
|
let t0 = ((self.min()[axis] - r.origin[axis]) * r.inv_direction[axis])
|
||||||
|
.min((self.max()[axis] - r.origin[axis]) * r.inv_direction[axis]);
|
||||||
|
let t1 = ((self.min()[axis] - r.origin[axis]) * r.inv_direction[axis])
|
||||||
|
.max((self.max()[axis] - r.origin[axis]) * r.inv_direction[axis]);
|
||||||
|
t_min = t0.max(t_min);
|
||||||
|
t_max = t1.min(t_max);
|
||||||
|
if t_max <= t_min {
|
||||||
|
return None;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Some(t_min)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn hit2(&self, r: Ray, t_min: f32, t_max: f32) -> Option<f32> {
|
||||||
let mut t_min = t_min;
|
let mut t_min = t_min;
|
||||||
let mut t_max = t_max;
|
let mut t_max = t_max;
|
||||||
for axis in 0..3 {
|
for axis in 0..3 {
|
||||||
@@ -82,13 +159,13 @@ impl AABB {
|
|||||||
t_min = max(t0, t_min);
|
t_min = max(t0, t_min);
|
||||||
t_max = min(t1, t_max);
|
t_max = min(t1, t_max);
|
||||||
if t_max <= t_min {
|
if t_max <= t_min {
|
||||||
return false;
|
return None;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
true
|
Some(t_min)
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn hit_precompute(&self, r: Ray, t0: f32, t1: f32) -> bool {
|
pub fn hit_precompute(&self, r: Ray, t0: f32, t1: f32) -> Option<f32> {
|
||||||
// TODO(wathiede): this has bugs.
|
// TODO(wathiede): this has bugs.
|
||||||
let mut t_min = (self.bounds[r.sign[0]].x - r.origin.x) * r.inv_direction.x;
|
let mut t_min = (self.bounds[r.sign[0]].x - r.origin.x) * r.inv_direction.x;
|
||||||
let mut t_max = (self.bounds[1 - r.sign[0]].x - r.origin.x) * r.inv_direction.x;
|
let mut t_max = (self.bounds[1 - r.sign[0]].x - r.origin.x) * r.inv_direction.x;
|
||||||
@@ -96,7 +173,7 @@ impl AABB {
|
|||||||
let t_y_max = (self.bounds[1 - r.sign[0]].y - r.origin.y) * r.inv_direction.y;
|
let t_y_max = (self.bounds[1 - r.sign[0]].y - r.origin.y) * r.inv_direction.y;
|
||||||
|
|
||||||
if t_min > t_y_max || t_y_min > t_max {
|
if t_min > t_y_max || t_y_min > t_max {
|
||||||
return false;
|
return None;
|
||||||
}
|
}
|
||||||
|
|
||||||
if t_y_min > t_min {
|
if t_y_min > t_min {
|
||||||
@@ -109,7 +186,7 @@ impl AABB {
|
|||||||
let t_z_min = (self.bounds[r.sign[2]].z - r.origin.z) * r.inv_direction.z;
|
let t_z_min = (self.bounds[r.sign[2]].z - r.origin.z) * r.inv_direction.z;
|
||||||
let t_z_max = (self.bounds[1 - r.sign[2]].z - r.origin.z) * r.inv_direction.z;
|
let t_z_max = (self.bounds[1 - r.sign[2]].z - r.origin.z) * r.inv_direction.z;
|
||||||
if t_min > t_z_max || t_z_min > t_max {
|
if t_min > t_z_max || t_z_min > t_max {
|
||||||
return false;
|
return None;
|
||||||
}
|
}
|
||||||
if t_z_min > t_min {
|
if t_z_min > t_min {
|
||||||
t_min = t_z_min;
|
t_min = t_z_min;
|
||||||
@@ -117,27 +194,42 @@ impl AABB {
|
|||||||
if t_z_max < t_max {
|
if t_z_max < t_max {
|
||||||
t_max = t_z_max;
|
t_max = t_z_max;
|
||||||
}
|
}
|
||||||
t_min < t1 && t_max > t0
|
if t_min < t1 && t_max > t0 {
|
||||||
|
Some(t_min)
|
||||||
|
} else {
|
||||||
|
None
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn hit(&self, r: Ray, t_min: f32, t_max: f32) -> bool {
|
pub fn hit_simd(&self, r: Ray, t_min: f32, t_max: f32) -> Option<f32> {
|
||||||
let mut t_min = t_min;
|
#[cfg(target_arch = "x86_64")]
|
||||||
let mut t_max = t_max;
|
unsafe {
|
||||||
for axis in 0..3 {
|
use std::arch::x86_64::*;
|
||||||
let t0 = ((self.min()[axis] - r.origin[axis]) * r.inv_direction[axis])
|
let o4 = _mm_set_ps(0., r.origin.z, r.origin.y, r.origin.x);
|
||||||
.min((self.max()[axis] - r.origin[axis]) * r.inv_direction[axis]);
|
let d4 = _mm_set_ps(0., r.inv_direction.z, r.inv_direction.y, r.inv_direction.x);
|
||||||
let t1 = ((self.min()[axis] - r.origin[axis]) * r.inv_direction[axis])
|
let bmin4 = _mm_set_ps(0., self.min().z, self.min().y, self.min().x);
|
||||||
.max((self.max()[axis] - r.origin[axis]) * r.inv_direction[axis]);
|
let bmax4 = _mm_set_ps(0., self.max().z, self.max().y, self.max().x);
|
||||||
t_min = t0.max(t_min);
|
let mask4 = _mm_cmpeq_ps(_mm_setzero_ps(), _mm_set_ps(1., 0., 0., 0.));
|
||||||
t_max = t1.min(t_max);
|
let t1 = _mm_mul_ps(_mm_sub_ps(_mm_and_ps(bmin4, mask4), o4), d4);
|
||||||
if t_max <= t_min {
|
let t2 = _mm_mul_ps(_mm_sub_ps(_mm_and_ps(bmax4, mask4), o4), d4);
|
||||||
return false;
|
let vmax4 = _mm_max_ps(t1, t2);
|
||||||
|
let vmin4 = _mm_min_ps(t1, t2);
|
||||||
|
let vmax4: (f32, f32, f32, f32) = std::mem::transmute(vmax4);
|
||||||
|
let vmin4: (f32, f32, f32, f32) = std::mem::transmute(vmin4);
|
||||||
|
let tmax = min(min(vmax4.0, min(vmax4.1, vmax4.2)), t_max);
|
||||||
|
let tmin = max(max(vmin4.0, max(vmin4.1, vmin4.2)), t_min);
|
||||||
|
if tmin <= tmax {
|
||||||
|
Some(tmin)
|
||||||
|
} else {
|
||||||
|
None
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
true
|
#[cfg(target_arch = "aarch64")]
|
||||||
|
// TODO(wathiede): add NEON implementation.
|
||||||
|
self.hit2(r, t_min, t_max)
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn hit_fast(&self, r: Ray, _t_min: f32, _t_max: f32) -> bool {
|
pub fn hit_fast(&self, r: Ray, _t_min: f32, _t_max: f32) -> Option<f32> {
|
||||||
let b = self;
|
let b = self;
|
||||||
let mut t1 = (b.min()[0] - r.origin[0]) * 1. / r.direction[0];
|
let mut t1 = (b.min()[0] - r.origin[0]) * 1. / r.direction[0];
|
||||||
let mut t2 = (b.max()[0] - r.origin[0]) * 1. / r.direction[0];
|
let mut t2 = (b.max()[0] - r.origin[0]) * 1. / r.direction[0];
|
||||||
@@ -153,7 +245,11 @@ impl AABB {
|
|||||||
tmax = tmax.min(t1.max(t2));
|
tmax = tmax.min(t1.max(t2));
|
||||||
}
|
}
|
||||||
|
|
||||||
tmax > tmin.max(0.0)
|
if tmax > tmin.max(0.0) {
|
||||||
|
Some(tmin)
|
||||||
|
} else {
|
||||||
|
None
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -170,3 +266,112 @@ pub fn surrounding_box(box0: &AABB, box1: &AABB) -> AABB {
|
|||||||
);
|
);
|
||||||
AABB::new(min, max)
|
AABB::new(min, max)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
mod tests {
|
||||||
|
use super::*;
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn infinite() {
|
||||||
|
let bb = AABB::infinite();
|
||||||
|
assert_eq!(bb.area(), 0.);
|
||||||
|
assert_eq!(bb.volume(), 0.);
|
||||||
|
}
|
||||||
|
|
||||||
|
macro_rules! hit_test {
|
||||||
|
($($name:ident,)*) => {
|
||||||
|
$(
|
||||||
|
mod $name {
|
||||||
|
use super::*;
|
||||||
|
const T_MIN: f32 = 0.001;
|
||||||
|
const T_MAX: f32 = f32::MAX;
|
||||||
|
|
||||||
|
fn test_bb() -> AABB {
|
||||||
|
AABB::new([-1., -1., -1.], [1., 1., 1.])
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn hit_front() {
|
||||||
|
let bb = test_bb();
|
||||||
|
let r = Ray::new([-2., 0., 0.], [1., 0., 0.], 0.5);
|
||||||
|
assert!(bb.$name(r, T_MIN, T_MAX).is_some());
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn hit_back() {
|
||||||
|
let bb = test_bb();
|
||||||
|
let r = Ray::new([2., 0., 0.], [-1., 0., 0.], 0.5);
|
||||||
|
assert!(bb.$name(r, T_MIN, T_MAX).is_some());
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn hit_top() {
|
||||||
|
let bb = test_bb();
|
||||||
|
let r = Ray::new([0., 2., 0.], [0., -1., 0.], 0.5);
|
||||||
|
assert!(bb.$name(r, T_MIN, T_MAX).is_some());
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn hit_bottom() {
|
||||||
|
let bb = test_bb();
|
||||||
|
let r = Ray::new([0., -2., 0.], [0., 1., 0.], 0.5);
|
||||||
|
assert!(bb.$name(r, T_MIN, T_MAX).is_some());
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn hit_left() {
|
||||||
|
let bb = test_bb();
|
||||||
|
let r = Ray::new([0., 0., -2.], [0., 0., 1.], 0.5);
|
||||||
|
assert!(bb.$name(r, T_MIN, T_MAX).is_some());
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn hit_right() {
|
||||||
|
let bb = test_bb();
|
||||||
|
let r = Ray::new([0., 0., 2.], [0., 0., -1.], 0.5);
|
||||||
|
assert!(bb.$name(r, T_MIN, T_MAX).is_some());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn miss_front() {
|
||||||
|
let bb = test_bb();
|
||||||
|
let r = Ray::new([0., 1.1, -1.1], [0., -1., 0.], 0.5);
|
||||||
|
assert_eq!(bb.$name(r, T_MIN, T_MAX), None);
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn miss_back() {
|
||||||
|
let bb = test_bb();
|
||||||
|
let r = Ray::new([0., 1.1, 1.1], [0., -1., 0.], 0.5);
|
||||||
|
assert_eq!(bb.$name(r, T_MIN, T_MAX), None);
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn miss_top() {
|
||||||
|
let bb = test_bb();
|
||||||
|
let r = Ray::new([0., 1.1, -1.1], [0., 0., 1.], 0.5);
|
||||||
|
assert_eq!(bb.$name(r, T_MIN, T_MAX), None);
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn miss_bottom() {
|
||||||
|
let bb = test_bb();
|
||||||
|
let r = Ray::new([0., -1.1, -1.1], [0., 0., 1.], 0.5);
|
||||||
|
assert_eq!(bb.$name(r, T_MIN, T_MAX), None);
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn miss_left() {
|
||||||
|
let bb = test_bb();
|
||||||
|
let r = Ray::new([-1.1, 0., -1.1], [0., 0., 1.], 0.5);
|
||||||
|
assert_eq!(bb.$name(r, T_MIN, T_MAX), None);
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn miss_right() {
|
||||||
|
let bb = test_bb();
|
||||||
|
let r = Ray::new([1.1, 0., -1.1], [0., 0., 1.], 0.5);
|
||||||
|
assert_eq!(bb.$name(r, T_MIN, T_MAX), None);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
)*
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
hit_test! {
|
||||||
|
hit_naive,
|
||||||
|
hit_simd,
|
||||||
|
hit_fast,
|
||||||
|
hit2,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|||||||
@@ -1,17 +1,15 @@
|
|||||||
use std;
|
use std::{self, fmt, time::Instant};
|
||||||
use std::fmt;
|
|
||||||
use std::time::Instant;
|
|
||||||
|
|
||||||
use log::info;
|
use log::info;
|
||||||
use rand;
|
use rand::{self, Rng};
|
||||||
use rand::Rng;
|
|
||||||
|
|
||||||
use crate::aabb::surrounding_box;
|
use crate::{
|
||||||
use crate::aabb::AABB;
|
aabb::{surrounding_box, AABB},
|
||||||
use crate::hitable::Hit;
|
hitable::{Hit, HitRecord},
|
||||||
use crate::hitable::HitRecord;
|
ray::Ray,
|
||||||
use crate::ray::Ray;
|
};
|
||||||
|
|
||||||
|
#[derive(Debug)]
|
||||||
enum BVHNode {
|
enum BVHNode {
|
||||||
Leaf(Box<dyn Hit>),
|
Leaf(Box<dyn Hit>),
|
||||||
Branch {
|
Branch {
|
||||||
@@ -181,6 +179,7 @@ impl Hit for BVHNode {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[derive(Debug)]
|
||||||
pub struct BVH {
|
pub struct BVH {
|
||||||
root: BVHNode,
|
root: BVHNode,
|
||||||
}
|
}
|
||||||
@@ -232,12 +231,13 @@ impl Hit for BVH {
|
|||||||
|
|
||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
mod tests {
|
mod tests {
|
||||||
use crate::aabb::AABB;
|
use crate::{
|
||||||
use crate::material::Lambertian;
|
aabb::AABB,
|
||||||
use crate::material::Metal;
|
material::{Lambertian, Metal},
|
||||||
use crate::sphere::Sphere;
|
sphere::Sphere,
|
||||||
use crate::texture::ConstantTexture;
|
texture::ConstantTexture,
|
||||||
use crate::vec3::Vec3;
|
vec3::Vec3,
|
||||||
|
};
|
||||||
|
|
||||||
use super::*;
|
use super::*;
|
||||||
|
|
||||||
|
|||||||
711
rtiow/renderer/src/bvh_triangles.rs
Normal file
711
rtiow/renderer/src/bvh_triangles.rs
Normal file
@@ -0,0 +1,711 @@
|
|||||||
|
/// Implementation based on blog post @
|
||||||
|
/// https://jacco.ompf2.com/2022/04/13/how-to-build-a-bvh-part-1-basics/
|
||||||
|
use std::f32::EPSILON;
|
||||||
|
use std::fmt;
|
||||||
|
|
||||||
|
use log::info;
|
||||||
|
use stl::STL;
|
||||||
|
|
||||||
|
use crate::{
|
||||||
|
aabb::AABB,
|
||||||
|
hitable::{Hit, HitRecord},
|
||||||
|
material::Material,
|
||||||
|
ray::Ray,
|
||||||
|
vec3::{cross, dot, Vec3},
|
||||||
|
};
|
||||||
|
|
||||||
|
#[derive(Debug, PartialEq)]
|
||||||
|
struct BVHNode {
|
||||||
|
aabb: AABB,
|
||||||
|
// When tri_count==0, left_first holds the left child's index in bvh_nodes. When >0 left_first
|
||||||
|
// holds the index for the first triangle in triangles.
|
||||||
|
left_first: u32,
|
||||||
|
tri_count: u32,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl BVHNode {
|
||||||
|
fn is_leaf(&self) -> bool {
|
||||||
|
self.tri_count > 0
|
||||||
|
}
|
||||||
|
|
||||||
|
fn cost(&self) -> f32 {
|
||||||
|
let area = self.aabb.area();
|
||||||
|
self.tri_count as f32 * area
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Clone, PartialEq)]
|
||||||
|
pub struct Triangle {
|
||||||
|
centroid: Vec3,
|
||||||
|
verts: [Vec3; 3],
|
||||||
|
}
|
||||||
|
impl fmt::Debug for Triangle {
|
||||||
|
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
|
||||||
|
write!(
|
||||||
|
f,
|
||||||
|
"Tri: <{}, {}, {}> @ {}",
|
||||||
|
self.verts[0], self.verts[1], self.verts[2], self.centroid
|
||||||
|
)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub struct BVHTriangles<M>
|
||||||
|
where
|
||||||
|
M: Material,
|
||||||
|
{
|
||||||
|
pub triangles: Vec<Triangle>,
|
||||||
|
triangle_index: Vec<usize>,
|
||||||
|
material: M,
|
||||||
|
bvh_nodes: Vec<BVHNode>,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug, Default)]
|
||||||
|
struct Bin {
|
||||||
|
bounds: AABB,
|
||||||
|
tri_count: usize,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<M> fmt::Debug for BVHTriangles<M>
|
||||||
|
where
|
||||||
|
M: Material,
|
||||||
|
{
|
||||||
|
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
|
||||||
|
write!(f, "{} triangles", self.triangles.len())?;
|
||||||
|
if f.alternate() {
|
||||||
|
writeln!(f)?;
|
||||||
|
}
|
||||||
|
for (i, t) in self.triangles.iter().enumerate() {
|
||||||
|
if f.alternate() {
|
||||||
|
write!(f, "\t")?;
|
||||||
|
}
|
||||||
|
write!(f, "{i:>3} {:?} ", t)?;
|
||||||
|
if f.alternate() {
|
||||||
|
writeln!(f)?;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if f.alternate() {
|
||||||
|
writeln!(f)?;
|
||||||
|
}
|
||||||
|
for (i, n) in self.bvh_nodes.iter().enumerate() {
|
||||||
|
write!(f, "N[{i}] {n:?}")?;
|
||||||
|
if f.alternate() {
|
||||||
|
writeln!(f)?;
|
||||||
|
}
|
||||||
|
let n = &self.bvh_nodes[i];
|
||||||
|
if n.is_leaf() {
|
||||||
|
for t_idx in n.left_first..(n.left_first + n.tri_count) {
|
||||||
|
if f.alternate() {
|
||||||
|
write!(f, "\t")?;
|
||||||
|
}
|
||||||
|
write!(f, "{:?} ", self.triangles[t_idx as usize])?;
|
||||||
|
if f.alternate() {
|
||||||
|
writeln!(f)?;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug)]
|
||||||
|
struct SplitCost {
|
||||||
|
pos: f32,
|
||||||
|
axis: usize,
|
||||||
|
cost: f32,
|
||||||
|
}
|
||||||
|
|
||||||
|
const ROOT_NODE_IDX: usize = 0;
|
||||||
|
impl<M> BVHTriangles<M>
|
||||||
|
where
|
||||||
|
M: Material,
|
||||||
|
{
|
||||||
|
pub fn new(stl: &STL, material: M, scale_factor: f32) -> BVHTriangles<M> {
|
||||||
|
let now = std::time::Instant::now();
|
||||||
|
|
||||||
|
assert_eq!(std::mem::size_of::<BVHNode>(), 32);
|
||||||
|
let div3 = 1. / 3.;
|
||||||
|
let triangles: Vec<_> = stl
|
||||||
|
.triangles
|
||||||
|
.iter()
|
||||||
|
.map(|t| {
|
||||||
|
let v0 = t.verts[0] * scale_factor;
|
||||||
|
let v1 = t.verts[1] * scale_factor;
|
||||||
|
let v2 = t.verts[2] * scale_factor;
|
||||||
|
let centroid = (v0 + v1 + v2) * div3;
|
||||||
|
Triangle {
|
||||||
|
centroid,
|
||||||
|
verts: [v0, v1, v2],
|
||||||
|
}
|
||||||
|
})
|
||||||
|
.collect();
|
||||||
|
let triangle_index = (0..triangles.len()).collect();
|
||||||
|
|
||||||
|
let n = 2 * triangles.len() - 2;
|
||||||
|
let bvh_nodes = Vec::with_capacity(n);
|
||||||
|
let mut bvh = BVHTriangles {
|
||||||
|
triangles,
|
||||||
|
triangle_index,
|
||||||
|
bvh_nodes,
|
||||||
|
material,
|
||||||
|
};
|
||||||
|
bvh.build_bvh();
|
||||||
|
info!(
|
||||||
|
"BVHTriangles build time {:0.3}s",
|
||||||
|
now.elapsed().as_secs_f32()
|
||||||
|
);
|
||||||
|
struct Stats {
|
||||||
|
nodes: usize,
|
||||||
|
leafs: usize,
|
||||||
|
min_tris: u32,
|
||||||
|
max_tris: u32,
|
||||||
|
}
|
||||||
|
impl Default for Stats {
|
||||||
|
fn default() -> Self {
|
||||||
|
Stats {
|
||||||
|
nodes: 0,
|
||||||
|
leafs: 0,
|
||||||
|
min_tris: u32::MAX,
|
||||||
|
max_tris: u32::MIN,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
let stats = bvh.bvh_nodes.iter().fold(Stats::default(), |mut stats, n| {
|
||||||
|
stats.nodes += 1;
|
||||||
|
if n.is_leaf() {
|
||||||
|
stats.leafs += 1;
|
||||||
|
stats.min_tris = n.tri_count.min(stats.min_tris);
|
||||||
|
stats.max_tris = n.tri_count.max(stats.max_tris);
|
||||||
|
}
|
||||||
|
stats
|
||||||
|
});
|
||||||
|
info!("BVHTriangles build stats:");
|
||||||
|
info!(" Nodes: {}", stats.nodes);
|
||||||
|
info!(" Leaves: {}", stats.leafs);
|
||||||
|
info!(" Tris: {}", bvh.triangles.len());
|
||||||
|
info!(" Min Tri: {}", stats.min_tris);
|
||||||
|
info!(" Max Tri: {}", stats.max_tris);
|
||||||
|
info!(" Avg Tri: {}", bvh.triangles.len() / stats.leafs);
|
||||||
|
info!(" Predict: {}", bvh.bvh_nodes.capacity());
|
||||||
|
info!(" Actual: {}", bvh.bvh_nodes.len());
|
||||||
|
info!(
|
||||||
|
" Savings: {} bytes",
|
||||||
|
(bvh.bvh_nodes.capacity() - bvh.bvh_nodes.len()) * std::mem::size_of::<BVHNode>()
|
||||||
|
);
|
||||||
|
bvh.bvh_nodes.shrink_to_fit();
|
||||||
|
//dbg!(&bvh);
|
||||||
|
bvh
|
||||||
|
}
|
||||||
|
|
||||||
|
fn build_bvh(&mut self) {
|
||||||
|
// assign all triangles to root node
|
||||||
|
let root = BVHNode {
|
||||||
|
aabb: AABB::default(),
|
||||||
|
left_first: 0,
|
||||||
|
tri_count: self.triangles.len() as u32,
|
||||||
|
};
|
||||||
|
self.bvh_nodes.push(root);
|
||||||
|
self.update_node_bounds(ROOT_NODE_IDX);
|
||||||
|
// subdivide recursively
|
||||||
|
self.subdivide(ROOT_NODE_IDX);
|
||||||
|
}
|
||||||
|
fn update_node_bounds(&mut self, node_idx: usize) {
|
||||||
|
let node = &mut self.bvh_nodes[node_idx];
|
||||||
|
let mut aabb_min: Vec3 = f32::MAX.into();
|
||||||
|
let mut aabb_max: Vec3 = f32::MIN.into();
|
||||||
|
for i in node.left_first..(node.left_first + node.tri_count) {
|
||||||
|
let leaf_tri_idx = self.triangle_index[i as usize];
|
||||||
|
let leaf_tri = &self.triangles[leaf_tri_idx];
|
||||||
|
aabb_min = vec3::min(aabb_min, leaf_tri.verts[0]);
|
||||||
|
aabb_min = vec3::min(aabb_min, leaf_tri.verts[1]);
|
||||||
|
aabb_min = vec3::min(aabb_min, leaf_tri.verts[2]);
|
||||||
|
aabb_max = vec3::max(aabb_max, leaf_tri.verts[0]);
|
||||||
|
aabb_max = vec3::max(aabb_max, leaf_tri.verts[1]);
|
||||||
|
aabb_max = vec3::max(aabb_max, leaf_tri.verts[2]);
|
||||||
|
}
|
||||||
|
node.aabb = AABB::new(aabb_min, aabb_max);
|
||||||
|
}
|
||||||
|
|
||||||
|
fn find_best_split_plane(&self, node: &BVHNode) -> SplitCost {
|
||||||
|
let mut best = SplitCost {
|
||||||
|
pos: 0.,
|
||||||
|
cost: f32::MAX,
|
||||||
|
axis: usize::MAX,
|
||||||
|
};
|
||||||
|
|
||||||
|
for axis in 0..3 {
|
||||||
|
let mut bounds_min = f32::MAX;
|
||||||
|
let mut bounds_max = f32::MIN;
|
||||||
|
|
||||||
|
for i in 0..node.tri_count {
|
||||||
|
let triangle = &self.triangles[self.triangle_index[(node.left_first + i) as usize]];
|
||||||
|
bounds_min = bounds_min.min(triangle.centroid[axis]);
|
||||||
|
bounds_max = bounds_max.max(triangle.centroid[axis]);
|
||||||
|
}
|
||||||
|
|
||||||
|
if bounds_min == bounds_max {
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
const NUM_BINS: usize = 8;
|
||||||
|
let mut bins: Vec<_> = (0..NUM_BINS)
|
||||||
|
.map(|_| Bin {
|
||||||
|
bounds: AABB::infinite(),
|
||||||
|
tri_count: 0,
|
||||||
|
})
|
||||||
|
.collect();
|
||||||
|
let scale = bins.len() as f32 / (bounds_max - bounds_min);
|
||||||
|
// populate the bins
|
||||||
|
for i in 0..node.tri_count {
|
||||||
|
let triangle = &self.triangles[self.triangle_index[(node.left_first + i) as usize]];
|
||||||
|
let bin_idx = (triangle.centroid[axis] - bounds_min) * scale;
|
||||||
|
let bin_idx = (bins.len() - 1).min(bin_idx as usize);
|
||||||
|
|
||||||
|
bins[bin_idx].tri_count += 1;
|
||||||
|
bins[bin_idx].bounds.grow(triangle.verts[0]);
|
||||||
|
bins[bin_idx].bounds.grow(triangle.verts[1]);
|
||||||
|
bins[bin_idx].bounds.grow(triangle.verts[2]);
|
||||||
|
}
|
||||||
|
|
||||||
|
// gather data for the 7 planes between the 8 bins
|
||||||
|
let mut left_area: Vec<_> = (0..bins.len() - 1).map(|_| 0.).collect();
|
||||||
|
let mut right_area: Vec<_> = (0..bins.len() - 1).map(|_| 0.).collect();
|
||||||
|
let mut left_count: Vec<_> = (0..bins.len() - 1).map(|_| 0).collect();
|
||||||
|
let mut right_count: Vec<_> = (0..bins.len() - 1).map(|_| 0).collect();
|
||||||
|
let mut left_box = AABB::infinite();
|
||||||
|
let mut right_box = AABB::infinite();
|
||||||
|
let mut left_sum = 0;
|
||||||
|
let mut right_sum = 0;
|
||||||
|
for i in 0..(bins.len() - 1) {
|
||||||
|
left_sum += bins[i].tri_count;
|
||||||
|
left_count[i] = left_sum;
|
||||||
|
left_box.grow(bins[i].bounds.min());
|
||||||
|
left_box.grow(bins[i].bounds.max());
|
||||||
|
left_area[i] = left_box.area();
|
||||||
|
|
||||||
|
right_sum += bins[bins.len() - 1 - i].tri_count;
|
||||||
|
right_count[bins.len() - 2 - i] = right_sum;
|
||||||
|
right_box.grow(bins[bins.len() - 1 - i].bounds.min());
|
||||||
|
right_box.grow(bins[bins.len() - 1 - i].bounds.max());
|
||||||
|
right_area[bins.len() - 2 - i] = right_box.area();
|
||||||
|
}
|
||||||
|
|
||||||
|
let scale = (bounds_max - bounds_min) / bins.len() as f32;
|
||||||
|
// calculate SAH cost for the 7 planes
|
||||||
|
for i in 0..(bins.len() - 1) {
|
||||||
|
let plane_cost =
|
||||||
|
left_count[i] as f32 * left_area[i] + right_count[i] as f32 * right_area[i];
|
||||||
|
if plane_cost < best.cost {
|
||||||
|
best.axis = axis;
|
||||||
|
best.pos = bounds_min + scale * (i as f32 + 1.);
|
||||||
|
best.cost = plane_cost;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
best
|
||||||
|
}
|
||||||
|
|
||||||
|
fn subdivide(&mut self, idx: usize) {
|
||||||
|
let (left_first, tri_count, left_count, i) = {
|
||||||
|
let node = &self.bvh_nodes[idx];
|
||||||
|
let split = self.find_best_split_plane(&node);
|
||||||
|
let no_split_cost = node.cost();
|
||||||
|
// Stop subdividing if it isn't getting any better.
|
||||||
|
if split.cost >= no_split_cost {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Split the group in two halves.
|
||||||
|
let mut i = node.left_first as isize;
|
||||||
|
let mut j = i + node.tri_count as isize - 1;
|
||||||
|
while i <= j {
|
||||||
|
if self.triangles[self.triangle_index[i as usize]].centroid[split.axis] < split.pos
|
||||||
|
{
|
||||||
|
i += 1;
|
||||||
|
} else {
|
||||||
|
self.triangles.swap(
|
||||||
|
self.triangle_index[i as usize],
|
||||||
|
self.triangle_index[j as usize],
|
||||||
|
);
|
||||||
|
j -= 1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Create child nodes for each half.
|
||||||
|
let left_count = i as u32 - node.left_first;
|
||||||
|
if left_count == 0 || left_count == node.tri_count {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
(node.left_first, node.tri_count, left_count, i)
|
||||||
|
};
|
||||||
|
|
||||||
|
// create child nodes
|
||||||
|
let left_child_idx = self.bvh_nodes.len() as u32;
|
||||||
|
let right_child_idx = left_child_idx + 1 as u32;
|
||||||
|
let left = BVHNode {
|
||||||
|
aabb: AABB::default(),
|
||||||
|
left_first,
|
||||||
|
tri_count: left_count as u32,
|
||||||
|
};
|
||||||
|
let right = BVHNode {
|
||||||
|
aabb: AABB::default(),
|
||||||
|
left_first: i as u32,
|
||||||
|
tri_count: (tri_count - left_count) as u32,
|
||||||
|
};
|
||||||
|
self.bvh_nodes.push(left);
|
||||||
|
self.bvh_nodes.push(right);
|
||||||
|
let node = &mut self.bvh_nodes[idx];
|
||||||
|
node.left_first = left_child_idx;
|
||||||
|
node.tri_count = 0;
|
||||||
|
|
||||||
|
// Recurse
|
||||||
|
self.update_node_bounds(left_child_idx as usize);
|
||||||
|
self.subdivide(left_child_idx as usize);
|
||||||
|
self.update_node_bounds(right_child_idx as usize);
|
||||||
|
self.subdivide(right_child_idx as usize);
|
||||||
|
}
|
||||||
|
|
||||||
|
fn intersect_bvh(&self, r: Ray, t_min: f32, t_max: f32) -> Option<HitRecord> {
|
||||||
|
let mut node = &self.bvh_nodes[ROOT_NODE_IDX];
|
||||||
|
let mut stack = Vec::with_capacity(2);
|
||||||
|
let mut nearest = None;
|
||||||
|
loop {
|
||||||
|
if node.is_leaf() {
|
||||||
|
let canditate = (node.left_first..(node.left_first + node.tri_count))
|
||||||
|
// Map from idx to Triangle
|
||||||
|
.map(|idx| &self.triangles[self.triangle_index[idx as usize]])
|
||||||
|
// Try to hit all triangles for this node, filtering out misses.
|
||||||
|
.filter_map(|tri| intersect_tri(r, &tri))
|
||||||
|
// Find the nearest hit (if any).
|
||||||
|
.min_by(|a, b| a.t.partial_cmp(&b.t).unwrap());
|
||||||
|
|
||||||
|
// merge candidate with nearest.
|
||||||
|
nearest = match (&canditate, &nearest) {
|
||||||
|
(Some(_), None) => canditate,
|
||||||
|
(None, Some(_)) => nearest,
|
||||||
|
(Some(c), Some(n)) => {
|
||||||
|
//info!("merging {c:#?} and {n:#?}");
|
||||||
|
if c.t < n.t {
|
||||||
|
canditate
|
||||||
|
} else {
|
||||||
|
nearest
|
||||||
|
}
|
||||||
|
}
|
||||||
|
(None, None) => None,
|
||||||
|
};
|
||||||
|
if stack.is_empty() {
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
node = stack.pop().unwrap();
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
let child1 = &self.bvh_nodes[node.left_first as usize];
|
||||||
|
let child2 = &self.bvh_nodes[node.left_first as usize + 1];
|
||||||
|
let dist1 = child1.aabb.hit_distance(r, t_min, t_max);
|
||||||
|
let dist2 = child2.aabb.hit_distance(r, t_min, t_max);
|
||||||
|
// Swap c1/c2 & d1/d2 based on d1/d2.
|
||||||
|
let (child1, child2, dist1, dist2) = match (dist1, dist2) {
|
||||||
|
(Some(d1), Some(d2)) if d1 > d2 => (child2, child1, dist2, dist1),
|
||||||
|
(None, Some(_)) => (child2, child1, dist2, dist1),
|
||||||
|
_ => (child1, child2, dist1, dist2),
|
||||||
|
};
|
||||||
|
|
||||||
|
// dist1/child1 should now be the nearest hit.
|
||||||
|
|
||||||
|
// If we missed dist1/child1, then we implicitly missed dist2/child2, so pop a child
|
||||||
|
// from the stack or exit the function.
|
||||||
|
if dist1.is_none() {
|
||||||
|
if stack.is_empty() {
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
node = stack.pop().unwrap();
|
||||||
|
} else {
|
||||||
|
// We hit child1, so process it next.
|
||||||
|
node = child1;
|
||||||
|
// If we also hit child2 save it on the stack so we can process it later.
|
||||||
|
if dist2.is_some() {
|
||||||
|
stack.push(child2);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
nearest
|
||||||
|
.and_then(|rtr: RayTriangleResult| Some(rtr.hit_record_with_material(&self.material)))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Based on https://www.scratchapixel.com/lessons/3d-basic-rendering/ray-tracing-rendering-a-triangle/moller-trumbore-ray-triangle-intersection.html
|
||||||
|
fn intersect_tri(r: Ray, tri: &Triangle) -> Option<RayTriangleResult> {
|
||||||
|
// #ifdef MOLLER_TRUMBORE
|
||||||
|
// Vec3f v0v1 = v1 - v0;
|
||||||
|
// Vec3f v0v2 = v2 - v0;
|
||||||
|
// Vec3f pvec = dir.crossProduct(v0v2);
|
||||||
|
// float det = v0v1.dotProduct(pvec);
|
||||||
|
// #ifdef CULLING
|
||||||
|
// // if the determinant is negative, the triangle is 'back facing'
|
||||||
|
// // if the determinant is close to 0, the ray misses the triangle
|
||||||
|
// if (det < kEpsilon) return false;
|
||||||
|
// #else
|
||||||
|
// // ray and triangle are parallel if det is close to 0
|
||||||
|
// if (fabs(det) < kEpsilon) return false;
|
||||||
|
// #endif
|
||||||
|
// float invDet = 1 / det;
|
||||||
|
//
|
||||||
|
// Vec3f tvec = orig - v0;
|
||||||
|
// u = tvec.dotProduct(pvec) * invDet;
|
||||||
|
// if (u < 0 || u > 1) return false;
|
||||||
|
//
|
||||||
|
// Vec3f qvec = tvec.crossProduct(v0v1);
|
||||||
|
// v = dir.dotProduct(qvec) * invDet;
|
||||||
|
// if (v < 0 || u + v > 1) return false;
|
||||||
|
//
|
||||||
|
// t = v0v2.dotProduct(qvec) * invDet;
|
||||||
|
//
|
||||||
|
let v0 = tri.verts[0];
|
||||||
|
let v1 = tri.verts[1];
|
||||||
|
let v2 = tri.verts[2];
|
||||||
|
|
||||||
|
let v0v1 = v1 - v0;
|
||||||
|
let v0v2 = v2 - v0;
|
||||||
|
let p = cross(r.direction, v0v2);
|
||||||
|
let det = dot(v0v1, p);
|
||||||
|
if det.abs() < EPSILON {
|
||||||
|
return None;
|
||||||
|
}
|
||||||
|
|
||||||
|
let inv_det = 1. / det;
|
||||||
|
|
||||||
|
let t = r.origin - v0;
|
||||||
|
let u = dot(t, p) * inv_det;
|
||||||
|
if u < 0. || u > 1. {
|
||||||
|
return None;
|
||||||
|
}
|
||||||
|
|
||||||
|
let q = cross(t, v0v1);
|
||||||
|
let v = dot(r.direction, q) * inv_det;
|
||||||
|
if v < 0. || u + v > 1. {
|
||||||
|
return None;
|
||||||
|
}
|
||||||
|
let t = dot(v0v2, q) * inv_det;
|
||||||
|
|
||||||
|
if t > EPSILON {
|
||||||
|
return Some(RayTriangleResult {
|
||||||
|
t,
|
||||||
|
p: r.point_at_parameter(t),
|
||||||
|
tri: tri.clone(),
|
||||||
|
});
|
||||||
|
}
|
||||||
|
None
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<M> Hit for BVHTriangles<M>
|
||||||
|
where
|
||||||
|
M: Material,
|
||||||
|
{
|
||||||
|
fn hit(&self, r: Ray, t_min: f32, t_max: f32) -> Option<HitRecord> {
|
||||||
|
self.intersect_bvh(r, t_min, t_max)
|
||||||
|
}
|
||||||
|
|
||||||
|
fn bounding_box(&self, _t_min: f32, _t_max: f32) -> Option<AABB> {
|
||||||
|
Some(self.bvh_nodes[ROOT_NODE_IDX].aabb)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug)]
|
||||||
|
struct RayTriangleResult {
|
||||||
|
t: f32,
|
||||||
|
p: Vec3,
|
||||||
|
tri: Triangle,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl RayTriangleResult {
|
||||||
|
fn hit_record_with_material<'m>(self, material: &'m dyn Material) -> HitRecord<'m> {
|
||||||
|
// We don't support UV (yet?).
|
||||||
|
let uv = (0.5, 0.5);
|
||||||
|
let v0 = self.tri.verts[0];
|
||||||
|
let v1 = self.tri.verts[1];
|
||||||
|
let v2 = self.tri.verts[2];
|
||||||
|
|
||||||
|
let v0v1 = v1 - v0;
|
||||||
|
let v0v2 = v2 - v0;
|
||||||
|
let normal = cross(v0v1, v0v2).unit_vector();
|
||||||
|
//println!("hit triangle {tri:?}");
|
||||||
|
HitRecord {
|
||||||
|
t: self.t,
|
||||||
|
uv,
|
||||||
|
p: self.p,
|
||||||
|
normal,
|
||||||
|
material,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
mod tests {
|
||||||
|
use super::*;
|
||||||
|
use crate::{
|
||||||
|
bvh_triangles::BVHTriangles,
|
||||||
|
cuboid::Cuboid,
|
||||||
|
hitable::Hit,
|
||||||
|
material::{Dielectric, Lambertian},
|
||||||
|
ray::Ray,
|
||||||
|
texture::ConstantTexture,
|
||||||
|
};
|
||||||
|
use pretty_assertions::assert_eq;
|
||||||
|
use proptest::prelude::*;
|
||||||
|
use std::{
|
||||||
|
io::{BufReader, Cursor},
|
||||||
|
sync::Arc,
|
||||||
|
};
|
||||||
|
use stl::STL;
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn compare_cuboid() {
|
||||||
|
let c = Cuboid::new(
|
||||||
|
[0., 0., 0.].into(),
|
||||||
|
[20., 20., 20.].into(),
|
||||||
|
Arc::new(Dielectric::new(1.5)),
|
||||||
|
);
|
||||||
|
let stl_cube = STL::parse(
|
||||||
|
BufReader::new(Cursor::new(include_bytes!("../stls/cube.stl"))),
|
||||||
|
false,
|
||||||
|
)
|
||||||
|
.expect("failed to parse cube");
|
||||||
|
|
||||||
|
let s = BVHTriangles::new(
|
||||||
|
&stl_cube,
|
||||||
|
Dielectric::new(1.5),
|
||||||
|
//Metal::new(Vec3::new(0.8, 0.8, 0.8), 0.2),
|
||||||
|
//Lambertian::new(ConstantTexture::new(Vec3::new(1.0, 0.2, 0.2))),
|
||||||
|
);
|
||||||
|
//dbg!(&s);
|
||||||
|
let mut rays: Vec<_> = (1..20)
|
||||||
|
.flat_map(|y| {
|
||||||
|
(1..20).flat_map(move |x| {
|
||||||
|
let x = x as f32;
|
||||||
|
let y = y as f32;
|
||||||
|
vec![
|
||||||
|
// Outward in angle
|
||||||
|
Ray::new([-1., x, y], [1., 0., 0.], 0.),
|
||||||
|
Ray::new([21., x, y], [-1., 0., 0.], 0.),
|
||||||
|
Ray::new([x, -1., y], [0., 1., 0.], 0.),
|
||||||
|
Ray::new([x, 21., y], [0., -1., 0.], 0.),
|
||||||
|
Ray::new([x, y, -1.], [0., 0., 1.], 0.),
|
||||||
|
Ray::new([x, y, 21.], [0., 0., -1.], 0.),
|
||||||
|
// Inward out (
|
||||||
|
Ray::new([x, y, 10.], [1., 0., 0.], 0.),
|
||||||
|
Ray::new([x, y, 10.], [-1., 0., 0.], 0.),
|
||||||
|
Ray::new([x, 10., y], [1., 0., 0.], 0.),
|
||||||
|
Ray::new([x, 10., y], [-1., 0., 0.], 0.),
|
||||||
|
Ray::new([10., x, y], [1., 0., 0.], 0.),
|
||||||
|
Ray::new([10., x, y], [-1., 0., 0.], 0.),
|
||||||
|
Ray::new([x, y, 10.], [0., 1., 0.], 0.),
|
||||||
|
Ray::new([x, y, 10.], [0., -1., 0.], 0.),
|
||||||
|
Ray::new([x, 10., y], [0., 1., 0.], 0.),
|
||||||
|
Ray::new([x, 10., y], [0., -1., 0.], 0.),
|
||||||
|
Ray::new([10., x, y], [0., 1., 0.], 0.),
|
||||||
|
Ray::new([10., x, y], [0., -1., 0.], 0.),
|
||||||
|
Ray::new([x, y, 10.], [0., 0., 1.], 0.),
|
||||||
|
Ray::new([x, y, 10.], [0., 0., -1.], 0.),
|
||||||
|
Ray::new([x, 10., y], [0., 0., 1.], 0.),
|
||||||
|
Ray::new([x, 10., y], [0., 0., -1.], 0.),
|
||||||
|
Ray::new([10., x, y], [0., 0., 1.], 0.),
|
||||||
|
Ray::new([10., x, y], [0., 0., -1.], 0.),
|
||||||
|
]
|
||||||
|
.into_iter()
|
||||||
|
})
|
||||||
|
})
|
||||||
|
.collect();
|
||||||
|
if false {
|
||||||
|
// Outward in at an angle.
|
||||||
|
let sqrt2 = 2f32.sqrt();
|
||||||
|
rays.extend(vec![
|
||||||
|
Ray::new([-1., 10., 10.], [sqrt2, sqrt2, 0.], 0.),
|
||||||
|
Ray::new([-1., 10., 10.], [sqrt2, -sqrt2, 0.], 0.),
|
||||||
|
Ray::new([-1., 10., 10.], [sqrt2, 0., sqrt2], 0.),
|
||||||
|
Ray::new([-1., 10., 10.], [sqrt2, 0., -sqrt2], 0.),
|
||||||
|
]);
|
||||||
|
}
|
||||||
|
|
||||||
|
for r in rays.into_iter() {
|
||||||
|
let c_hit = c
|
||||||
|
.hit(r, 0., f32::MAX)
|
||||||
|
.expect(&format!("c_hit missed {r:#?}"));
|
||||||
|
let s_hit = s
|
||||||
|
.hit(r, 0., f32::MAX)
|
||||||
|
.expect(&format!("s_hit missed {r:#?}"));
|
||||||
|
assert!(
|
||||||
|
(c_hit.t - s_hit.t).abs() < EPSILON,
|
||||||
|
"{r:?} [t] c_hit: {c_hit:#?}, s_hit: {s_hit:#?}"
|
||||||
|
);
|
||||||
|
// uv isn't valid for BVHTriangles.
|
||||||
|
// assert_eq!( c_hit.uv, s_hit.uv, "{i}: [uv] c_hit: {c_hit:?}, s_hit: {s_hit:?}");
|
||||||
|
assert_eq!(
|
||||||
|
c_hit.p, s_hit.p,
|
||||||
|
"{r:?}: [p] c_hit: {c_hit:#?}, s_hit: {s_hit:#?}"
|
||||||
|
);
|
||||||
|
assert_eq!(
|
||||||
|
c_hit.normal, s_hit.normal,
|
||||||
|
"{r:?}: [normal] c_hit: {c_hit:?}, s_hit: {s_hit:?}"
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
proptest! {
|
||||||
|
// TODO(wathiede): make this work.
|
||||||
|
//#[test]
|
||||||
|
fn compare_cuboid_proptest(
|
||||||
|
ox in -20.0f32..40.0,
|
||||||
|
oy in -20.0f32..40.0,
|
||||||
|
oz in -20.0f32..40.0,
|
||||||
|
dx in -1.0f32..1.0,
|
||||||
|
dy in -1.0f32..1.0,
|
||||||
|
dz in -1.0f32..1.0,
|
||||||
|
) {
|
||||||
|
let r = Ray::new([ox,oy,oz].into(), Vec3::new(dx, dy, dz).unit_vector(), 0.5);
|
||||||
|
let c = Cuboid::new(
|
||||||
|
[0., 0., 0.].into(),
|
||||||
|
[20., 20., 20.].into(),
|
||||||
|
Arc::new(Dielectric::new(1.5)),
|
||||||
|
);
|
||||||
|
let stl_cube = STL::parse(
|
||||||
|
BufReader::new(Cursor::new(include_bytes!("../stls/cube.stl"))),
|
||||||
|
false,
|
||||||
|
)
|
||||||
|
.expect("failed to parse cube");
|
||||||
|
|
||||||
|
let s = BVHTriangles::new(
|
||||||
|
&stl_cube,
|
||||||
|
Dielectric::new(1.5),
|
||||||
|
//Metal::new(Vec3::new(0.8, 0.8, 0.8), 0.2),
|
||||||
|
//Lambertian::new(ConstantTexture::new(Vec3::new(1.0, 0.2, 0.2))),
|
||||||
|
);
|
||||||
|
|
||||||
|
let c_hit = c .hit(r, 0., f32::MAX);
|
||||||
|
let s_hit = s .hit(r, 0., f32::MAX);
|
||||||
|
|
||||||
|
match (c_hit, s_hit) {
|
||||||
|
(Some(_), None)=>assert!(false, "hit cuboid but not STL"),
|
||||||
|
(None, Some(_))=>assert!(false, "hit STL but not cuboid"),
|
||||||
|
(Some(c_hit), Some(s_hit))=> {
|
||||||
|
assert!(
|
||||||
|
(c_hit.t - s_hit.t).abs() < 0.00001,
|
||||||
|
"{r:?} [t] c_hit: {c_hit:#?}, s_hit: {s_hit:#?}"
|
||||||
|
);
|
||||||
|
// uv isn't valid for BVHTriangles.
|
||||||
|
// assert_eq!( c_hit.uv, s_hit.uv, "{i}: [uv] c_hit: {c_hit:?}, s_hit: {s_hit:?}");
|
||||||
|
assert_eq!(
|
||||||
|
c_hit.p, s_hit.p,
|
||||||
|
"{r:?}: [p] c_hit: {c_hit:#?}, s_hit: {s_hit:#?}"
|
||||||
|
);
|
||||||
|
assert_eq!(
|
||||||
|
c_hit.normal, s_hit.normal,
|
||||||
|
"{r:?}: [normal] c_hit: {c_hit:?}, s_hit: {s_hit:?}"
|
||||||
|
);
|
||||||
|
},
|
||||||
|
// It's okay if they both miss.
|
||||||
|
(None,None)=>(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -1,11 +1,11 @@
|
|||||||
use std::f32::consts::PI;
|
use std::f32::consts::PI;
|
||||||
|
|
||||||
use rand;
|
use rand::{self, Rng};
|
||||||
use rand::Rng;
|
|
||||||
|
|
||||||
use crate::ray::Ray;
|
use crate::{
|
||||||
use crate::vec3::cross;
|
ray::Ray,
|
||||||
use crate::vec3::Vec3;
|
vec3::{cross, Vec3},
|
||||||
|
};
|
||||||
|
|
||||||
fn random_in_unit_disk() -> Vec3 {
|
fn random_in_unit_disk() -> Vec3 {
|
||||||
let mut rng = rand::thread_rng();
|
let mut rng = rand::thread_rng();
|
||||||
@@ -18,6 +18,7 @@ fn random_in_unit_disk() -> Vec3 {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[derive(Debug)]
|
||||||
pub struct Camera {
|
pub struct Camera {
|
||||||
origin: Vec3,
|
origin: Vec3,
|
||||||
lower_left_corner: Vec3,
|
lower_left_corner: Vec3,
|
||||||
|
|||||||
46
rtiow/renderer/src/colors.rs
Normal file
46
rtiow/renderer/src/colors.rs
Normal file
@@ -0,0 +1,46 @@
|
|||||||
|
use rand::{self, Rng};
|
||||||
|
|
||||||
|
use crate::vec3::Vec3;
|
||||||
|
|
||||||
|
// HSV values in [0..1]
|
||||||
|
// returns [r, g, b] values from 0 to 255
|
||||||
|
//From https://martin.ankerl.com/2009/12/09/how-to-create-random-colors-programmatically/
|
||||||
|
pub fn hsv_to_rgb(h: f32, s: f32, v: f32) -> Vec3 {
|
||||||
|
let h_i = (h * 6.) as i32;
|
||||||
|
let f = h * 6. - h_i as f32;
|
||||||
|
let p = v * (1. - s);
|
||||||
|
let q = v * (1. - f * s);
|
||||||
|
let t = v * (1. - (1. - f) * s);
|
||||||
|
match h_i {
|
||||||
|
0 => Vec3::new(v, t, p),
|
||||||
|
1 => Vec3::new(q, v, p),
|
||||||
|
2 => Vec3::new(p, v, t),
|
||||||
|
3 => Vec3::new(p, q, v),
|
||||||
|
4 => Vec3::new(t, p, v),
|
||||||
|
5 => Vec3::new(v, p, q),
|
||||||
|
_ => panic!("Unknown H value {}", h_i),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn generate_rainbow(num: usize) -> Vec<Vec3> {
|
||||||
|
(0..num)
|
||||||
|
.map(|n| {
|
||||||
|
let h = n as f32 / num as f32;
|
||||||
|
hsv_to_rgb(h, 0.99, 0.99)
|
||||||
|
})
|
||||||
|
.collect()
|
||||||
|
}
|
||||||
|
pub fn generate_palette(num: usize) -> Vec<Vec3> {
|
||||||
|
let mut rng = rand::thread_rng();
|
||||||
|
let mut random = || rng.gen();
|
||||||
|
// use golden ratio
|
||||||
|
let golden_ratio_conjugate = 0.618_034;
|
||||||
|
let mut h = random();
|
||||||
|
(0..num)
|
||||||
|
.map(|_| {
|
||||||
|
h += golden_ratio_conjugate;
|
||||||
|
h %= 1.0;
|
||||||
|
hsv_to_rgb(h, 0.99, 0.99)
|
||||||
|
})
|
||||||
|
.collect()
|
||||||
|
}
|
||||||
@@ -1,16 +1,17 @@
|
|||||||
use std;
|
use std;
|
||||||
|
|
||||||
use rand;
|
use rand::{self, Rng};
|
||||||
use rand::Rng;
|
|
||||||
|
|
||||||
use crate::aabb::AABB;
|
use crate::{
|
||||||
use crate::hitable::Hit;
|
aabb::AABB,
|
||||||
use crate::hitable::HitRecord;
|
hitable::{Hit, HitRecord},
|
||||||
use crate::material::Isotropic;
|
material::Isotropic,
|
||||||
use crate::ray::Ray;
|
ray::Ray,
|
||||||
use crate::texture::Texture;
|
texture::Texture,
|
||||||
use crate::vec3::Vec3;
|
vec3::Vec3,
|
||||||
|
};
|
||||||
|
|
||||||
|
#[derive(Debug)]
|
||||||
pub struct ConstantMedium<H, T>
|
pub struct ConstantMedium<H, T>
|
||||||
where
|
where
|
||||||
H: Hit,
|
H: Hit,
|
||||||
|
|||||||
@@ -1,17 +1,17 @@
|
|||||||
use std::sync::Arc;
|
use std::sync::Arc;
|
||||||
|
|
||||||
use crate::aabb::AABB;
|
use crate::{
|
||||||
use crate::flip_normals::FlipNormals;
|
aabb::AABB,
|
||||||
use crate::hitable::Hit;
|
flip_normals::FlipNormals,
|
||||||
use crate::hitable::HitRecord;
|
hitable::{Hit, HitRecord},
|
||||||
use crate::hitable_list::HitableList;
|
hitable_list::HitableList,
|
||||||
use crate::material::Material;
|
material::Material,
|
||||||
use crate::ray::Ray;
|
ray::Ray,
|
||||||
use crate::rect::XYRect;
|
rect::{XYRect, XZRect, YZRect},
|
||||||
use crate::rect::XZRect;
|
vec3::Vec3,
|
||||||
use crate::rect::YZRect;
|
};
|
||||||
use crate::vec3::Vec3;
|
|
||||||
|
|
||||||
|
#[derive(Debug)]
|
||||||
pub struct Cuboid {
|
pub struct Cuboid {
|
||||||
p_min: Vec3,
|
p_min: Vec3,
|
||||||
p_max: Vec3,
|
p_max: Vec3,
|
||||||
@@ -22,6 +22,9 @@ impl Cuboid {
|
|||||||
// This clippy doesn't work right with Arc.
|
// This clippy doesn't work right with Arc.
|
||||||
#[allow(clippy::needless_pass_by_value)]
|
#[allow(clippy::needless_pass_by_value)]
|
||||||
pub fn new(p_min: Vec3, p_max: Vec3, material: Arc<dyn Material>) -> Cuboid {
|
pub fn new(p_min: Vec3, p_max: Vec3, material: Arc<dyn Material>) -> Cuboid {
|
||||||
|
assert!(p_min.x <= p_max.x);
|
||||||
|
assert!(p_min.y <= p_max.y);
|
||||||
|
assert!(p_min.z <= p_max.z);
|
||||||
Cuboid {
|
Cuboid {
|
||||||
p_min,
|
p_min,
|
||||||
p_max,
|
p_max,
|
||||||
|
|||||||
44
rtiow/renderer/src/debug_hit.rs
Normal file
44
rtiow/renderer/src/debug_hit.rs
Normal file
@@ -0,0 +1,44 @@
|
|||||||
|
use crate::{
|
||||||
|
aabb::AABB,
|
||||||
|
hitable::{Hit, HitRecord},
|
||||||
|
material::Lambertian,
|
||||||
|
ray::Ray,
|
||||||
|
};
|
||||||
|
|
||||||
|
#[derive(Debug)]
|
||||||
|
pub struct DebugHit<H>
|
||||||
|
where
|
||||||
|
H: Hit,
|
||||||
|
{
|
||||||
|
hitable: H,
|
||||||
|
material: Lambertian<[f32; 3]>,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<H> DebugHit<H>
|
||||||
|
where
|
||||||
|
H: Hit,
|
||||||
|
{
|
||||||
|
pub fn new(hitable: H) -> DebugHit<H>
|
||||||
|
where {
|
||||||
|
DebugHit {
|
||||||
|
hitable,
|
||||||
|
material: Lambertian::new([0.2, 0.2, 0.2]),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<H> Hit for DebugHit<H>
|
||||||
|
where
|
||||||
|
H: Hit,
|
||||||
|
{
|
||||||
|
fn hit(&self, r: Ray, t_min: f32, t_max: f32) -> Option<HitRecord> {
|
||||||
|
if let Some(hit) = self.hitable.hit(r, t_min, t_max) {
|
||||||
|
return Some(HitRecord { t: hit.t, ..hit });
|
||||||
|
}
|
||||||
|
None
|
||||||
|
}
|
||||||
|
|
||||||
|
fn bounding_box(&self, t_min: f32, t_max: f32) -> Option<AABB> {
|
||||||
|
self.hitable.bounding_box(t_min, t_max)
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -1,8 +1,10 @@
|
|||||||
use crate::aabb::AABB;
|
use crate::{
|
||||||
use crate::hitable::Hit;
|
aabb::AABB,
|
||||||
use crate::hitable::HitRecord;
|
hitable::{Hit, HitRecord},
|
||||||
use crate::ray::Ray;
|
ray::Ray,
|
||||||
|
};
|
||||||
|
|
||||||
|
#[derive(Debug)]
|
||||||
pub struct FlipNormals<H>
|
pub struct FlipNormals<H>
|
||||||
where
|
where
|
||||||
H: Hit,
|
H: Hit,
|
||||||
|
|||||||
139
rtiow/renderer/src/glowybox.rs
Normal file
139
rtiow/renderer/src/glowybox.rs
Normal file
@@ -0,0 +1,139 @@
|
|||||||
|
use std::sync::Arc;
|
||||||
|
|
||||||
|
use crate::{
|
||||||
|
aabb::AABB,
|
||||||
|
cuboid::Cuboid,
|
||||||
|
hitable::{Hit, HitRecord},
|
||||||
|
material::Material,
|
||||||
|
ray::Ray,
|
||||||
|
vec3::Vec3,
|
||||||
|
};
|
||||||
|
|
||||||
|
#[derive(Debug)]
|
||||||
|
pub struct Glowybox {
|
||||||
|
p_min: Vec3,
|
||||||
|
p_max: Vec3,
|
||||||
|
main: Cuboid,
|
||||||
|
edges: [Cuboid; 12],
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Glowybox {
|
||||||
|
// This clippy doesn't work right with Arc.
|
||||||
|
#[allow(clippy::needless_pass_by_value)]
|
||||||
|
pub fn new(
|
||||||
|
p_min: Vec3,
|
||||||
|
p_max: Vec3,
|
||||||
|
edge_thickness: f32,
|
||||||
|
main_material: Arc<dyn Material>,
|
||||||
|
edge_material: Arc<dyn Material>,
|
||||||
|
) -> Glowybox {
|
||||||
|
assert!(p_min.x < p_max.x);
|
||||||
|
assert!(p_min.y < p_max.y);
|
||||||
|
assert!(p_min.z < p_max.z);
|
||||||
|
let main = Cuboid::new(p_min, p_max, main_material);
|
||||||
|
// Top edges
|
||||||
|
let ht = edge_thickness / 2.;
|
||||||
|
let edges = [
|
||||||
|
// Top edges
|
||||||
|
Cuboid::new(
|
||||||
|
[p_min.x - ht, p_max.y - ht, p_min.z - ht].into(),
|
||||||
|
[p_min.x + ht, p_max.y + ht, p_max.z + ht].into(),
|
||||||
|
Arc::clone(&edge_material),
|
||||||
|
),
|
||||||
|
Cuboid::new(
|
||||||
|
[p_min.x - ht, p_max.y - ht, p_min.z - ht].into(),
|
||||||
|
[p_max.x + ht, p_max.y + ht, p_min.z + ht].into(),
|
||||||
|
Arc::clone(&edge_material),
|
||||||
|
),
|
||||||
|
Cuboid::new(
|
||||||
|
[p_max.x - ht, p_max.y - ht, p_min.z - ht].into(),
|
||||||
|
[p_max.x + ht, p_max.y + ht, p_max.z + ht].into(),
|
||||||
|
Arc::clone(&edge_material),
|
||||||
|
),
|
||||||
|
Cuboid::new(
|
||||||
|
[p_min.x - ht, p_max.y - ht, p_max.z - ht].into(),
|
||||||
|
[p_max.x + ht, p_max.y + ht, p_max.z + ht].into(),
|
||||||
|
Arc::clone(&edge_material),
|
||||||
|
),
|
||||||
|
// Bottom edges
|
||||||
|
Cuboid::new(
|
||||||
|
[p_min.x - ht, p_min.y - ht, p_min.z - ht].into(),
|
||||||
|
[p_min.x + ht, p_min.y + ht, p_max.z + ht].into(),
|
||||||
|
Arc::clone(&edge_material),
|
||||||
|
),
|
||||||
|
Cuboid::new(
|
||||||
|
[p_min.x - ht, p_min.y - ht, p_min.z - ht].into(),
|
||||||
|
[p_max.x + ht, p_min.y + ht, p_min.z + ht].into(),
|
||||||
|
Arc::clone(&edge_material),
|
||||||
|
),
|
||||||
|
Cuboid::new(
|
||||||
|
[p_max.x - ht, p_min.y - ht, p_min.z - ht].into(),
|
||||||
|
[p_max.x + ht, p_min.y + ht, p_max.z + ht].into(),
|
||||||
|
Arc::clone(&edge_material),
|
||||||
|
),
|
||||||
|
Cuboid::new(
|
||||||
|
[p_min.x - ht, p_min.y - ht, p_max.z - ht].into(),
|
||||||
|
[p_max.x + ht, p_min.y + ht, p_max.z + ht].into(),
|
||||||
|
Arc::clone(&edge_material),
|
||||||
|
),
|
||||||
|
// Middle edges
|
||||||
|
Cuboid::new(
|
||||||
|
[p_min.x - ht, p_min.y - ht, p_min.z - ht].into(),
|
||||||
|
[p_min.x + ht, p_max.y + ht, p_min.z + ht].into(),
|
||||||
|
Arc::clone(&edge_material),
|
||||||
|
),
|
||||||
|
Cuboid::new(
|
||||||
|
[p_min.x - ht, p_min.y - ht, p_max.z - ht].into(),
|
||||||
|
[p_min.x + ht, p_max.y + ht, p_max.z + ht].into(),
|
||||||
|
Arc::clone(&edge_material),
|
||||||
|
),
|
||||||
|
Cuboid::new(
|
||||||
|
[p_max.x - ht, p_min.y - ht, p_min.z - ht].into(),
|
||||||
|
[p_max.x + ht, p_max.y + ht, p_min.z + ht].into(),
|
||||||
|
Arc::clone(&edge_material),
|
||||||
|
),
|
||||||
|
Cuboid::new(
|
||||||
|
[p_max.x - ht, p_min.y - ht, p_max.z - ht].into(),
|
||||||
|
[p_max.x + ht, p_max.y + ht, p_max.z + ht].into(),
|
||||||
|
Arc::clone(&edge_material),
|
||||||
|
),
|
||||||
|
];
|
||||||
|
let p_min = [p_min.x - ht, p_min.y - ht, p_min.z - ht].into();
|
||||||
|
let p_max = [p_max.x + ht, p_max.y + ht, p_max.z + ht].into();
|
||||||
|
Glowybox {
|
||||||
|
p_min,
|
||||||
|
p_max,
|
||||||
|
main,
|
||||||
|
edges,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Hit for Glowybox {
|
||||||
|
fn hit(&self, r: Ray, t_min: f32, t_max: f32) -> Option<HitRecord> {
|
||||||
|
let mut edge_hit = None;
|
||||||
|
for edge in &self.edges {
|
||||||
|
if let Some(hit) = edge.hit(r, t_min, t_max) {
|
||||||
|
edge_hit = Some(hit);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
let main_hit = self.main.hit(r, t_min, t_max);
|
||||||
|
match (edge_hit, main_hit) {
|
||||||
|
(Some(ehit), Some(mhit)) => {
|
||||||
|
if mhit.t < ehit.t {
|
||||||
|
Some(mhit)
|
||||||
|
} else {
|
||||||
|
Some(ehit)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
(Some(ehit), None) => Some(ehit),
|
||||||
|
(None, Some(mhit)) => Some(mhit),
|
||||||
|
_ => None,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn bounding_box(&self, _t_min: f32, _t_max: f32) -> Option<AABB> {
|
||||||
|
Some(AABB::new(self.p_min, self.p_max))
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -1,10 +1,8 @@
|
|||||||
use std::sync::Arc;
|
use std::{fmt::Debug, sync::Arc};
|
||||||
|
|
||||||
use crate::aabb::AABB;
|
use crate::{aabb::AABB, material::Material, ray::Ray, vec3::Vec3};
|
||||||
use crate::material::Material;
|
|
||||||
use crate::ray::Ray;
|
|
||||||
use crate::vec3::Vec3;
|
|
||||||
|
|
||||||
|
#[derive(Debug)]
|
||||||
pub struct HitRecord<'m> {
|
pub struct HitRecord<'m> {
|
||||||
pub t: f32,
|
pub t: f32,
|
||||||
pub uv: (f32, f32),
|
pub uv: (f32, f32),
|
||||||
@@ -13,7 +11,7 @@ pub struct HitRecord<'m> {
|
|||||||
pub material: &'m dyn Material,
|
pub material: &'m dyn Material,
|
||||||
}
|
}
|
||||||
|
|
||||||
pub trait Hit: Send + Sync {
|
pub trait Hit: Send + Sync + Debug {
|
||||||
fn hit(&self, r: Ray, t_min: f32, t_max: f32) -> Option<HitRecord>;
|
fn hit(&self, r: Ray, t_min: f32, t_max: f32) -> Option<HitRecord>;
|
||||||
fn bounding_box(&self, t_min: f32, t_max: f32) -> Option<AABB>;
|
fn bounding_box(&self, t_min: f32, t_max: f32) -> Option<AABB>;
|
||||||
}
|
}
|
||||||
@@ -26,3 +24,12 @@ impl Hit for Arc<dyn Hit> {
|
|||||||
(**self).bounding_box(t_min, t_max)
|
(**self).bounding_box(t_min, t_max)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
impl Hit for Box<dyn Hit> {
|
||||||
|
fn hit(&self, r: Ray, t_min: f32, t_max: f32) -> Option<HitRecord> {
|
||||||
|
(**self).hit(r, t_min, t_max)
|
||||||
|
}
|
||||||
|
fn bounding_box(&self, t_min: f32, t_max: f32) -> Option<AABB> {
|
||||||
|
(**self).bounding_box(t_min, t_max)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|||||||
@@ -1,13 +1,13 @@
|
|||||||
use std;
|
use std;
|
||||||
|
|
||||||
use crate::aabb::surrounding_box;
|
use crate::{
|
||||||
use crate::aabb::AABB;
|
aabb::{surrounding_box, AABB},
|
||||||
use crate::hitable::Hit;
|
hitable::{Hit, HitRecord},
|
||||||
use crate::hitable::HitRecord;
|
ray::Ray,
|
||||||
use crate::ray::Ray;
|
vec3::Vec3,
|
||||||
use crate::vec3::Vec3;
|
};
|
||||||
|
|
||||||
#[derive(Default)]
|
#[derive(Debug, Default)]
|
||||||
pub struct HitableList {
|
pub struct HitableList {
|
||||||
list: Vec<Box<dyn Hit>>,
|
list: Vec<Box<dyn Hit>>,
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -130,6 +130,12 @@ impl Formatter {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
impl Default for Formatter {
|
||||||
|
fn default() -> Self {
|
||||||
|
Self::new()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
impl Scales {
|
impl Scales {
|
||||||
/// Instantiates a new `Scales` with SI keys
|
/// Instantiates a new `Scales` with SI keys
|
||||||
pub fn new() -> Self {
|
pub fn new() -> Self {
|
||||||
@@ -199,7 +205,7 @@ impl Scales {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
return 0.0;
|
0.0
|
||||||
}
|
}
|
||||||
|
|
||||||
fn to_scaled_value(&self, value: f64) -> ScaledValue {
|
fn to_scaled_value(&self, value: f64) -> ScaledValue {
|
||||||
@@ -221,3 +227,9 @@ impl Scales {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
impl Default for Scales {
|
||||||
|
fn default() -> Self {
|
||||||
|
Self::new()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|||||||
@@ -2,12 +2,14 @@ use std::fmt;
|
|||||||
|
|
||||||
use log::info;
|
use log::info;
|
||||||
|
|
||||||
use crate::aabb::surrounding_box;
|
use crate::{
|
||||||
use crate::aabb::AABB;
|
aabb::{surrounding_box, AABB},
|
||||||
use crate::hitable::Hit;
|
hitable::{Hit, HitRecord},
|
||||||
use crate::hitable::HitRecord;
|
hitable_list::HitableList,
|
||||||
use crate::ray::Ray;
|
ray::Ray,
|
||||||
|
};
|
||||||
|
|
||||||
|
#[derive(Debug)]
|
||||||
pub enum KDTree {
|
pub enum KDTree {
|
||||||
Leaf(Box<dyn Hit>),
|
Leaf(Box<dyn Hit>),
|
||||||
Branch {
|
Branch {
|
||||||
@@ -102,6 +104,14 @@ impl KDTree {
|
|||||||
}),
|
}),
|
||||||
_ => panic!("Unreachable"),
|
_ => panic!("Unreachable"),
|
||||||
};
|
};
|
||||||
|
//info!("left_half {:?}", left_half);
|
||||||
|
//info!("right_half {:?}", right_half);
|
||||||
|
if left_half.is_empty() {
|
||||||
|
return KDTree::Leaf(Box::new(HitableList::new(right_half)));
|
||||||
|
};
|
||||||
|
if right_half.is_empty() {
|
||||||
|
return KDTree::Leaf(Box::new(HitableList::new(left_half)));
|
||||||
|
};
|
||||||
KDTree::Branch {
|
KDTree::Branch {
|
||||||
left: Box::new(KDTree::new(left_half, t_min, t_max)),
|
left: Box::new(KDTree::new(left_half, t_min, t_max)),
|
||||||
right: Box::new(KDTree::new(right_half, t_min, t_max)),
|
right: Box::new(KDTree::new(right_half, t_min, t_max)),
|
||||||
@@ -146,7 +156,7 @@ fn print_tree(f: &mut fmt::Formatter, depth: usize, kdt: &KDTree) -> fmt::Result
|
|||||||
impl fmt::Display for KDTree {
|
impl fmt::Display for KDTree {
|
||||||
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
|
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
|
||||||
writeln!(f, "kd-tree")?;
|
writeln!(f, "kd-tree")?;
|
||||||
print_tree(f, 1, &self)
|
print_tree(f, 1, self)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
impl Hit for KDTree {
|
impl Hit for KDTree {
|
||||||
|
|||||||
@@ -1,9 +1,13 @@
|
|||||||
pub mod aabb;
|
pub mod aabb;
|
||||||
pub mod bvh;
|
pub mod bvh;
|
||||||
|
pub mod bvh_triangles;
|
||||||
pub mod camera;
|
pub mod camera;
|
||||||
|
pub mod colors;
|
||||||
pub mod constant_medium;
|
pub mod constant_medium;
|
||||||
pub mod cuboid;
|
pub mod cuboid;
|
||||||
|
pub mod debug_hit;
|
||||||
pub mod flip_normals;
|
pub mod flip_normals;
|
||||||
|
pub mod glowybox;
|
||||||
pub mod hitable;
|
pub mod hitable;
|
||||||
pub mod hitable_list;
|
pub mod hitable_list;
|
||||||
pub mod human;
|
pub mod human;
|
||||||
@@ -12,12 +16,15 @@ pub mod material;
|
|||||||
pub mod moving_sphere;
|
pub mod moving_sphere;
|
||||||
pub mod noise;
|
pub mod noise;
|
||||||
pub mod output;
|
pub mod output;
|
||||||
|
pub mod parser;
|
||||||
pub mod ray;
|
pub mod ray;
|
||||||
pub mod rect;
|
pub mod rect;
|
||||||
pub mod renderer;
|
pub mod renderer;
|
||||||
pub mod rotate;
|
pub mod rotate;
|
||||||
|
pub mod scale;
|
||||||
pub mod scenes;
|
pub mod scenes;
|
||||||
pub mod sphere;
|
pub mod sphere;
|
||||||
pub mod texture;
|
pub mod texture;
|
||||||
pub mod translate;
|
pub mod translate;
|
||||||
|
pub mod triangles;
|
||||||
pub mod vec3;
|
pub mod vec3;
|
||||||
|
|||||||
@@ -1,13 +1,13 @@
|
|||||||
use std::sync::Arc;
|
use std::{fmt::Debug, sync::Arc};
|
||||||
|
|
||||||
use rand;
|
use rand::{self, Rng};
|
||||||
use rand::Rng;
|
|
||||||
|
|
||||||
use crate::hitable::HitRecord;
|
use crate::{
|
||||||
use crate::ray::Ray;
|
hitable::HitRecord,
|
||||||
use crate::texture::Texture;
|
ray::Ray,
|
||||||
use crate::vec3::dot;
|
texture::Texture,
|
||||||
use crate::vec3::Vec3;
|
vec3::{dot, Vec3},
|
||||||
|
};
|
||||||
|
|
||||||
fn random_in_unit_sphere() -> Vec3 {
|
fn random_in_unit_sphere() -> Vec3 {
|
||||||
let mut rng = rand::thread_rng();
|
let mut rng = rand::thread_rng();
|
||||||
@@ -20,14 +20,14 @@ fn random_in_unit_sphere() -> Vec3 {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Default)]
|
#[derive(Default, Debug)]
|
||||||
pub struct ScatterResponse {
|
pub struct ScatterResponse {
|
||||||
pub scattered: Ray,
|
pub scattered: Ray,
|
||||||
pub attenutation: Vec3,
|
pub attenutation: Vec3,
|
||||||
pub reflected: bool,
|
pub reflected: bool,
|
||||||
}
|
}
|
||||||
|
|
||||||
pub trait Material: Send + Sync {
|
pub trait Material: Send + Sync + Debug {
|
||||||
fn scatter(&self, r_in: &Ray, rec: &HitRecord) -> ScatterResponse;
|
fn scatter(&self, r_in: &Ray, rec: &HitRecord) -> ScatterResponse;
|
||||||
fn emitted(&self, _u: f32, _v: f32, _p: Vec3) -> Vec3 {
|
fn emitted(&self, _u: f32, _v: f32, _p: Vec3) -> Vec3 {
|
||||||
Vec3::new(0., 0., 0.)
|
Vec3::new(0., 0., 0.)
|
||||||
@@ -52,6 +52,7 @@ impl Material for Box<dyn Material> {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[derive(Clone, Debug)]
|
||||||
pub struct Isotropic<T>
|
pub struct Isotropic<T>
|
||||||
where
|
where
|
||||||
T: Texture,
|
T: Texture,
|
||||||
@@ -82,6 +83,7 @@ where
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[derive(Clone, Debug)]
|
||||||
pub struct Lambertian<T>
|
pub struct Lambertian<T>
|
||||||
where
|
where
|
||||||
T: Texture,
|
T: Texture,
|
||||||
@@ -113,6 +115,7 @@ where
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[derive(Clone, Debug)]
|
||||||
pub struct Metal {
|
pub struct Metal {
|
||||||
albedo: Vec3,
|
albedo: Vec3,
|
||||||
fuzzy: f32,
|
fuzzy: f32,
|
||||||
@@ -167,6 +170,7 @@ fn schlick(cosine: f32, ref_idx: f32) -> f32 {
|
|||||||
r0 + (1. - r0) * (1. - cosine).powf(5.)
|
r0 + (1. - r0) * (1. - cosine).powf(5.)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[derive(Clone, Debug)]
|
||||||
pub struct Dielectric {
|
pub struct Dielectric {
|
||||||
ref_idx: f32,
|
ref_idx: f32,
|
||||||
}
|
}
|
||||||
@@ -213,6 +217,7 @@ impl Material for Dielectric {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[derive(Debug)]
|
||||||
pub struct DiffuseLight<T>
|
pub struct DiffuseLight<T>
|
||||||
where
|
where
|
||||||
T: Texture,
|
T: Texture,
|
||||||
@@ -246,6 +251,20 @@ where
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[derive(Debug)]
|
||||||
|
pub struct DebugMaterial {}
|
||||||
|
|
||||||
|
impl Material for DebugMaterial {
|
||||||
|
fn scatter(&self, _r_in: &Ray, rec: &HitRecord) -> ScatterResponse {
|
||||||
|
let dir = Vec3::new(0., -1., -1.).unit_vector();
|
||||||
|
ScatterResponse {
|
||||||
|
scattered: Ray::new(rec.p, dir, 0.),
|
||||||
|
attenutation: [1., 1., 1.].into(),
|
||||||
|
reflected: true,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
mod tests {
|
mod tests {
|
||||||
use super::*;
|
use super::*;
|
||||||
|
|||||||
@@ -1,13 +1,13 @@
|
|||||||
use crate::aabb::surrounding_box;
|
use crate::{
|
||||||
use crate::aabb::AABB;
|
aabb::{surrounding_box, AABB},
|
||||||
use crate::hitable::Hit;
|
hitable::{Hit, HitRecord},
|
||||||
use crate::hitable::HitRecord;
|
material::Material,
|
||||||
use crate::material::Material;
|
ray::Ray,
|
||||||
use crate::ray::Ray;
|
sphere::get_sphere_uv,
|
||||||
use crate::sphere::get_sphere_uv;
|
vec3::{dot, Vec3},
|
||||||
use crate::vec3::dot;
|
};
|
||||||
use crate::vec3::Vec3;
|
|
||||||
|
|
||||||
|
#[derive(Debug)]
|
||||||
pub struct MovingSphere<M>
|
pub struct MovingSphere<M>
|
||||||
where
|
where
|
||||||
M: Material,
|
M: Material,
|
||||||
|
|||||||
@@ -2,10 +2,10 @@ use log::trace;
|
|||||||
/// Implements the concepts from https://lodev.org/cgtutor/randomnoise.html
|
/// Implements the concepts from https://lodev.org/cgtutor/randomnoise.html
|
||||||
use rand;
|
use rand;
|
||||||
|
|
||||||
use crate::noise::NoiseSource;
|
use crate::{noise::NoiseSource, vec3::Vec3};
|
||||||
use crate::vec3::Vec3;
|
|
||||||
|
|
||||||
const NOISE_SIZE: usize = 128;
|
const NOISE_SIZE: usize = 128;
|
||||||
|
#[derive(Debug)]
|
||||||
pub struct Lode {
|
pub struct Lode {
|
||||||
// Using fixed array causes stack overflow.
|
// Using fixed array causes stack overflow.
|
||||||
noise: Vec<Vec<Vec<f32>>>, //[[[f32; NOISE_SIZE]; NOISE_SIZE]; NOISE_SIZE],
|
noise: Vec<Vec<Vec<f32>>>, //[[[f32; NOISE_SIZE]; NOISE_SIZE]; NOISE_SIZE],
|
||||||
|
|||||||
@@ -1,13 +1,13 @@
|
|||||||
pub mod lode;
|
pub mod lode;
|
||||||
pub mod perlin;
|
pub mod perlin;
|
||||||
|
|
||||||
use std::f32::consts::PI;
|
use std::{f32::consts::PI, fmt::Debug};
|
||||||
|
|
||||||
use serde_derive::Deserialize;
|
use serde_derive::Deserialize;
|
||||||
|
|
||||||
use crate::vec3::Vec3;
|
use crate::vec3::Vec3;
|
||||||
|
|
||||||
pub trait NoiseSource: Send + Sync {
|
pub trait NoiseSource: Send + Sync + Debug {
|
||||||
/// value returns noise on the interval [0., 1.).
|
/// value returns noise on the interval [0., 1.).
|
||||||
fn value(&self, p: Vec3) -> f32;
|
fn value(&self, p: Vec3) -> f32;
|
||||||
|
|
||||||
|
|||||||
@@ -1,13 +1,14 @@
|
|||||||
// There are many math functions in this file, so we allow single letter variable names.
|
// There are many math functions in this file, so we allow single letter variable names.
|
||||||
#![allow(clippy::many_single_char_names)]
|
#![allow(clippy::many_single_char_names)]
|
||||||
use log::trace;
|
use log::trace;
|
||||||
use rand::seq::SliceRandom;
|
use rand::{seq::SliceRandom, Rng};
|
||||||
use rand::Rng;
|
|
||||||
|
|
||||||
use crate::noise::NoiseSource;
|
use crate::{
|
||||||
use crate::vec3::dot;
|
noise::NoiseSource,
|
||||||
use crate::vec3::Vec3;
|
vec3::{dot, Vec3},
|
||||||
|
};
|
||||||
|
|
||||||
|
#[derive(Debug)]
|
||||||
pub struct Perlin {
|
pub struct Perlin {
|
||||||
ran_vec: Vec<Vec3>,
|
ran_vec: Vec<Vec3>,
|
||||||
perm_x: Vec<usize>,
|
perm_x: Vec<usize>,
|
||||||
@@ -35,7 +36,7 @@ fn perlin_generate_perm<R>(rng: &mut R) -> Vec<usize>
|
|||||||
where
|
where
|
||||||
R: Rng,
|
R: Rng,
|
||||||
{
|
{
|
||||||
let mut p: Vec<usize> = (0..256).map(|i| i).collect();
|
let mut p: Vec<usize> = (0..256).collect();
|
||||||
p.shuffle(rng);
|
p.shuffle(rng);
|
||||||
p
|
p
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,19 +1,20 @@
|
|||||||
use std::collections::HashMap;
|
use std::{
|
||||||
use std::fs::File;
|
collections::HashMap,
|
||||||
use std::io::BufWriter;
|
fs::File,
|
||||||
use std::path::Path;
|
io::BufWriter,
|
||||||
use std::sync::Arc;
|
net::TcpStream,
|
||||||
use std::sync::Mutex;
|
path::Path,
|
||||||
use std::time;
|
sync::{Arc, Mutex},
|
||||||
|
time,
|
||||||
|
};
|
||||||
|
|
||||||
use chrono::Local;
|
use chrono::Local;
|
||||||
use image;
|
use image;
|
||||||
use lazy_static::lazy_static;
|
|
||||||
use log::info;
|
use log::info;
|
||||||
use serde_derive::Serialize;
|
use serde_derive::Serialize;
|
||||||
|
use tev_client::{PacketCreateImage, PacketUpdateImage, TevClient};
|
||||||
|
|
||||||
use crate::renderer::Scene;
|
use crate::{renderer::Scene, vec3::Vec3};
|
||||||
use crate::vec3::Vec3;
|
|
||||||
|
|
||||||
// Main RGB image output from rendering the scene.
|
// Main RGB image output from rendering the scene.
|
||||||
pub const MAIN_IMAGE: &str = "@final";
|
pub const MAIN_IMAGE: &str = "@final";
|
||||||
@@ -24,10 +25,6 @@ pub const ADAPTIVE_DEPTH: &str = "adaptive_depth";
|
|||||||
// Grey scale showing rays cast per pixel.
|
// Grey scale showing rays cast per pixel.
|
||||||
pub const RAYS_PER_PIXEL: &str = "rays_per_pixel";
|
pub const RAYS_PER_PIXEL: &str = "rays_per_pixel";
|
||||||
|
|
||||||
lazy_static! {
|
|
||||||
static ref DEBUGGER: Arc<Mutex<Debugger>> = Arc::new(Mutex::new(Debugger::new()));
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Serialize)]
|
#[derive(Serialize)]
|
||||||
struct ImageMetadata {
|
struct ImageMetadata {
|
||||||
name: String,
|
name: String,
|
||||||
@@ -74,43 +71,176 @@ impl Image {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
struct Debugger {
|
pub struct OutputManager {
|
||||||
images: HashMap<String, (ImageType, Image)>,
|
images: Arc<Mutex<HashMap<String, (ImageType, Image)>>>,
|
||||||
|
tev_client: Option<Arc<Mutex<TevClient>>>,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Debugger {
|
impl OutputManager {
|
||||||
fn new() -> Debugger {
|
pub fn new(tev_addr: &Option<String>) -> std::io::Result<OutputManager> {
|
||||||
Debugger {
|
let tev_client = if let Some(addr) = tev_addr {
|
||||||
images: HashMap::new(),
|
Some(Arc::new(Mutex::new(TevClient::wrap(TcpStream::connect(
|
||||||
}
|
addr,
|
||||||
|
)?))))
|
||||||
|
} else {
|
||||||
|
None
|
||||||
|
};
|
||||||
|
Ok(OutputManager {
|
||||||
|
images: Arc::new(Mutex::new(HashMap::new())),
|
||||||
|
tev_client,
|
||||||
|
})
|
||||||
}
|
}
|
||||||
}
|
|
||||||
|
|
||||||
pub fn register_image(name: String, dimensions: (usize, usize), it: ImageType) {
|
pub fn register_image(&self, name: String, dimensions: (usize, usize), it: ImageType) {
|
||||||
let mut debugger = DEBUGGER.lock().unwrap();
|
let mut images = self.images.lock().unwrap();
|
||||||
debugger
|
images.insert(name.clone(), (it, Image::new(dimensions.0, dimensions.1)));
|
||||||
.images
|
self.tev_client.clone().map(|c| {
|
||||||
.insert(name, (it, Image::new(dimensions.0, dimensions.1)));
|
c.lock().unwrap().send(PacketCreateImage {
|
||||||
}
|
image_name: &name,
|
||||||
|
grab_focus: false,
|
||||||
|
width: dimensions.0 as u32,
|
||||||
|
height: dimensions.1 as u32,
|
||||||
|
channel_names: &["R", "G", "B"],
|
||||||
|
})
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
pub fn set_pixel(name: &str, x: usize, y: usize, pixel: Vec3) {
|
pub fn set_pixel(&self, name: &str, x: usize, y: usize, pixel: Vec3) {
|
||||||
let mut debugger = DEBUGGER.lock().unwrap();
|
let mut images = self.images.lock().unwrap();
|
||||||
let (_it, img) = debugger
|
let (_it, img) = images
|
||||||
.images
|
.get_mut(name)
|
||||||
.get_mut(name)
|
.unwrap_or_else(|| panic!("couldn't find image named '{}'", name));
|
||||||
.expect(&format!("couldn't find image named '{}'", name));
|
let y_inv = img.h - y - 1;
|
||||||
let y_inv = img.h - y - 1;
|
img.put_pixel(x, y_inv, pixel);
|
||||||
img.put_pixel(x, y_inv, pixel);
|
self.tev_client.clone().map(|c| {
|
||||||
}
|
c.lock().unwrap().send(PacketUpdateImage {
|
||||||
|
image_name: &name,
|
||||||
|
grab_focus: false,
|
||||||
|
channel_names: &["R", "G", "B"],
|
||||||
|
channel_offsets: &[0, 1, 2],
|
||||||
|
channel_strides: &[0, 0, 0],
|
||||||
|
x: x as u32,
|
||||||
|
y: y_inv as u32,
|
||||||
|
width: 1,
|
||||||
|
height: 1,
|
||||||
|
data: &[pixel.x, pixel.y, pixel.z],
|
||||||
|
})
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
pub fn set_pixel_grey(name: &str, x: usize, y: usize, grey: f32) {
|
pub fn set_pixel_grey(&self, name: &str, x: usize, y: usize, grey: f32) {
|
||||||
let mut debugger = DEBUGGER.lock().unwrap();
|
let mut images = self.images.lock().unwrap();
|
||||||
let (_it, img) = debugger
|
let (_it, img) = images
|
||||||
.images
|
.get_mut(name)
|
||||||
.get_mut(name)
|
.unwrap_or_else(|| panic!("couldn't find image named '{}'", name));
|
||||||
.expect(&format!("couldn't find image named '{}'", name));
|
let y_inv = img.h - y - 1;
|
||||||
let y_inv = img.h - y - 1;
|
img.put_pixel(x, y_inv, [grey, grey, grey].into());
|
||||||
img.put_pixel(x, y_inv, [grey, grey, grey].into());
|
}
|
||||||
|
pub fn write_images<P: AsRef<Path>>(
|
||||||
|
&self,
|
||||||
|
scene: &Scene,
|
||||||
|
render_time: time::Duration,
|
||||||
|
output_dir: P,
|
||||||
|
) -> std::io::Result<()> {
|
||||||
|
let output_dir: &Path = output_dir.as_ref();
|
||||||
|
let now = Local::now();
|
||||||
|
let images = self.images.lock().unwrap();
|
||||||
|
// Write out images in consistent order.
|
||||||
|
let mut names = images.keys().collect::<Vec<_>>();
|
||||||
|
names.sort();
|
||||||
|
let mut image_metadata = Vec::new();
|
||||||
|
for name in &names {
|
||||||
|
let (it, img) = images.get(*name).unwrap();
|
||||||
|
let image = format!("{}.png", name);
|
||||||
|
let binary = format!("{}.json", name);
|
||||||
|
let ratio = img.w as f32 / img.h as f32;
|
||||||
|
let size = (img.w, img.h);
|
||||||
|
let image_path = output_dir.join(&image);
|
||||||
|
let binary_path = output_dir.join(&binary);
|
||||||
|
image_metadata.push(ImageMetadata {
|
||||||
|
name: name.to_string(),
|
||||||
|
image,
|
||||||
|
binary,
|
||||||
|
ratio,
|
||||||
|
size,
|
||||||
|
format: *it,
|
||||||
|
});
|
||||||
|
info!("Saving {}", image_path.to_string_lossy());
|
||||||
|
match it {
|
||||||
|
ImageType::RGB01 => {
|
||||||
|
let mut out_img = image::RgbImage::new(img.w as u32, img.h as u32);
|
||||||
|
out_img
|
||||||
|
.enumerate_pixels_mut()
|
||||||
|
.enumerate()
|
||||||
|
.for_each(|(i, (_x, _y, p))| {
|
||||||
|
let pixel = img.pix[i];
|
||||||
|
*p = image::Rgb([
|
||||||
|
(pixel[0] * 255.).min(255.) as u8,
|
||||||
|
(pixel[1] * 255.).min(255.) as u8,
|
||||||
|
(pixel[2] * 255.).min(255.) as u8,
|
||||||
|
])
|
||||||
|
});
|
||||||
|
out_img.save(image_path)?;
|
||||||
|
}
|
||||||
|
ImageType::Grey01 => {
|
||||||
|
let mut out_img = image::GrayImage::new(img.w as u32, img.h as u32);
|
||||||
|
out_img
|
||||||
|
.enumerate_pixels_mut()
|
||||||
|
.enumerate()
|
||||||
|
.for_each(|(i, (_x, _y, p))| {
|
||||||
|
let pixel = img.pix[i];
|
||||||
|
*p = image::Luma([(pixel[0] * 255.).min(255.) as u8])
|
||||||
|
});
|
||||||
|
out_img.save(image_path)?;
|
||||||
|
}
|
||||||
|
ImageType::GreyNormalized => {
|
||||||
|
let mut out_img = image::GrayImage::new(img.w as u32, img.h as u32);
|
||||||
|
|
||||||
|
let max_val = img.pix.iter().map(|v| v.x).fold(0., f32::max);
|
||||||
|
out_img
|
||||||
|
.enumerate_pixels_mut()
|
||||||
|
.enumerate()
|
||||||
|
.for_each(|(i, (_x, _y, p))| {
|
||||||
|
let pixel = img.pix[i];
|
||||||
|
*p = image::Luma([(pixel[0] / max_val * 255.).min(255.) as u8])
|
||||||
|
});
|
||||||
|
out_img.save(image_path)?;
|
||||||
|
}
|
||||||
|
};
|
||||||
|
info!("Saving {}", binary_path.to_string_lossy());
|
||||||
|
let f = File::create(output_dir.join(binary_path))?;
|
||||||
|
let f = BufWriter::new(f);
|
||||||
|
match it {
|
||||||
|
ImageType::RGB01 => {
|
||||||
|
serde_json::ser::to_writer(
|
||||||
|
f,
|
||||||
|
&img.pix
|
||||||
|
.iter()
|
||||||
|
.map(|v| [v.x, v.y, v.z])
|
||||||
|
.collect::<Vec<[f32; 3]>>(),
|
||||||
|
)?;
|
||||||
|
}
|
||||||
|
ImageType::Grey01 | ImageType::GreyNormalized => {
|
||||||
|
serde_json::ser::to_writer(
|
||||||
|
f,
|
||||||
|
&img.pix.iter().map(|v| v.x).collect::<Vec<f32>>(),
|
||||||
|
)?;
|
||||||
|
}
|
||||||
|
};
|
||||||
|
}
|
||||||
|
let f = File::create(output_dir.join("data.json"))?;
|
||||||
|
let f = BufWriter::new(f);
|
||||||
|
serde_json::ser::to_writer(
|
||||||
|
f,
|
||||||
|
&Data {
|
||||||
|
timestamp: now.timestamp(),
|
||||||
|
render_time_seconds: render_time.as_secs_f32(),
|
||||||
|
scene,
|
||||||
|
image_metadata,
|
||||||
|
},
|
||||||
|
)?;
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
trait ImageSaver {
|
trait ImageSaver {
|
||||||
@@ -118,105 +248,3 @@ trait ImageSaver {
|
|||||||
where
|
where
|
||||||
Q: AsRef<Path> + Sized;
|
Q: AsRef<Path> + Sized;
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn write_images<P: AsRef<Path>>(
|
|
||||||
scene: &Scene,
|
|
||||||
render_time: time::Duration,
|
|
||||||
output_dir: P,
|
|
||||||
) -> std::io::Result<()> {
|
|
||||||
let output_dir: &Path = output_dir.as_ref();
|
|
||||||
let debugger = DEBUGGER.lock().unwrap();
|
|
||||||
let now = Local::now();
|
|
||||||
// Write out images in consistent order.
|
|
||||||
let mut names = debugger.images.keys().collect::<Vec<_>>();
|
|
||||||
names.sort();
|
|
||||||
let mut image_metadata = Vec::new();
|
|
||||||
for name in &names {
|
|
||||||
let (it, img) = debugger.images.get(*name).unwrap();
|
|
||||||
let image = format!("{}.png", name);
|
|
||||||
let binary = format!("{}.json", name);
|
|
||||||
let ratio = img.w as f32 / img.h as f32;
|
|
||||||
let size = (img.w, img.h);
|
|
||||||
let image_path = output_dir.join(&image);
|
|
||||||
let binary_path = output_dir.join(&binary);
|
|
||||||
image_metadata.push(ImageMetadata {
|
|
||||||
name: name.to_string(),
|
|
||||||
image,
|
|
||||||
binary,
|
|
||||||
ratio,
|
|
||||||
size,
|
|
||||||
format: *it,
|
|
||||||
});
|
|
||||||
info!("Saving {}", image_path.to_string_lossy());
|
|
||||||
match it {
|
|
||||||
ImageType::RGB01 => {
|
|
||||||
let mut out_img = image::RgbImage::new(img.w as u32, img.h as u32);
|
|
||||||
out_img
|
|
||||||
.enumerate_pixels_mut()
|
|
||||||
.enumerate()
|
|
||||||
.for_each(|(i, (_x, _y, p))| {
|
|
||||||
let pixel = img.pix[i];
|
|
||||||
*p = image::Rgb([
|
|
||||||
(pixel[0] * 255.).min(255.) as u8,
|
|
||||||
(pixel[1] * 255.).min(255.) as u8,
|
|
||||||
(pixel[2] * 255.).min(255.) as u8,
|
|
||||||
])
|
|
||||||
});
|
|
||||||
out_img.save(image_path)?;
|
|
||||||
}
|
|
||||||
ImageType::Grey01 => {
|
|
||||||
let mut out_img = image::GrayImage::new(img.w as u32, img.h as u32);
|
|
||||||
out_img
|
|
||||||
.enumerate_pixels_mut()
|
|
||||||
.enumerate()
|
|
||||||
.for_each(|(i, (_x, _y, p))| {
|
|
||||||
let pixel = img.pix[i];
|
|
||||||
*p = image::Luma([(pixel[0] * 255.).min(255.) as u8])
|
|
||||||
});
|
|
||||||
out_img.save(image_path)?;
|
|
||||||
}
|
|
||||||
ImageType::GreyNormalized => {
|
|
||||||
let mut out_img = image::GrayImage::new(img.w as u32, img.h as u32);
|
|
||||||
|
|
||||||
let max_val = img.pix.iter().map(|v| v.x).fold(0., f32::max);
|
|
||||||
out_img
|
|
||||||
.enumerate_pixels_mut()
|
|
||||||
.enumerate()
|
|
||||||
.for_each(|(i, (_x, _y, p))| {
|
|
||||||
let pixel = img.pix[i];
|
|
||||||
*p = image::Luma([(pixel[0] / max_val * 255.).min(255.) as u8])
|
|
||||||
});
|
|
||||||
out_img.save(image_path)?;
|
|
||||||
}
|
|
||||||
};
|
|
||||||
info!("Saving {}", binary_path.to_string_lossy());
|
|
||||||
let f = File::create(output_dir.join(binary_path))?;
|
|
||||||
let f = BufWriter::new(f);
|
|
||||||
match it {
|
|
||||||
ImageType::RGB01 => {
|
|
||||||
serde_json::ser::to_writer(
|
|
||||||
f,
|
|
||||||
&img.pix
|
|
||||||
.iter()
|
|
||||||
.map(|v| [v.x, v.y, v.z])
|
|
||||||
.collect::<Vec<[f32; 3]>>(),
|
|
||||||
)?;
|
|
||||||
}
|
|
||||||
ImageType::Grey01 | ImageType::GreyNormalized => {
|
|
||||||
serde_json::ser::to_writer(f, &img.pix.iter().map(|v| v.x).collect::<Vec<f32>>())?;
|
|
||||||
}
|
|
||||||
};
|
|
||||||
}
|
|
||||||
let f = File::create(output_dir.join("data.json"))?;
|
|
||||||
let f = BufWriter::new(f);
|
|
||||||
serde_json::ser::to_writer(
|
|
||||||
f,
|
|
||||||
&Data {
|
|
||||||
timestamp: now.timestamp(),
|
|
||||||
render_time_seconds: render_time.as_secs_f32(),
|
|
||||||
scene,
|
|
||||||
image_metadata,
|
|
||||||
},
|
|
||||||
)?;
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
|
|||||||
206
rtiow/renderer/src/parser.rs
Normal file
206
rtiow/renderer/src/parser.rs
Normal file
@@ -0,0 +1,206 @@
|
|||||||
|
use crate::{
|
||||||
|
bvh_triangles::BVHTriangles,
|
||||||
|
camera::Camera,
|
||||||
|
cuboid::Cuboid,
|
||||||
|
hitable::Hit,
|
||||||
|
hitable_list::HitableList,
|
||||||
|
material::{Dielectric, DiffuseLight, Isotropic, Lambertian, Material, Metal},
|
||||||
|
renderer::Scene,
|
||||||
|
sphere::Sphere,
|
||||||
|
texture::{EnvMap, Texture},
|
||||||
|
};
|
||||||
|
use chrono::IsoWeek;
|
||||||
|
use serde::Deserialize;
|
||||||
|
use std::{collections::HashMap, fs::File, io::BufReader, path::PathBuf, sync::Arc};
|
||||||
|
use stl::STL;
|
||||||
|
use thiserror::Error;
|
||||||
|
use vec3::Vec3;
|
||||||
|
|
||||||
|
#[derive(Debug, Deserialize)]
|
||||||
|
pub struct Config {
|
||||||
|
scene: SceneConfig,
|
||||||
|
camera: CameraConfig,
|
||||||
|
materials: Vec<MaterialConfig>,
|
||||||
|
hitables: Vec<HitableConfig>,
|
||||||
|
envmap: Option<EnvMapConfig>,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Error, Debug)]
|
||||||
|
pub enum ConfigError {
|
||||||
|
#[error("failed to load image")]
|
||||||
|
ImageError(#[from] image::ImageError),
|
||||||
|
#[error("failed to parser STL")]
|
||||||
|
STLError(#[from] stl::ParseError),
|
||||||
|
#[error("I/O error")]
|
||||||
|
IOError(#[from] std::io::Error),
|
||||||
|
#[error("duplication material named '{0}'")]
|
||||||
|
DuplicateMaterial(String),
|
||||||
|
#[error("unkown material named '{0}'")]
|
||||||
|
UnknownMaterial(String),
|
||||||
|
}
|
||||||
|
|
||||||
|
impl TryFrom<Config> for Scene {
|
||||||
|
type Error = ConfigError;
|
||||||
|
|
||||||
|
fn try_from(c: Config) -> Result<Scene, Self::Error> {
|
||||||
|
let mut materials = HashMap::new();
|
||||||
|
for mc in c.materials {
|
||||||
|
let v: Arc<dyn Material> = match mc.material {
|
||||||
|
Materials::Metal { albedo, fuzzy } => Arc::new(Metal::new(albedo, fuzzy)),
|
||||||
|
Materials::Dielectric { ref_idx } => Arc::new(Dielectric::new(ref_idx)),
|
||||||
|
Materials::DiffuseLight { texture } => Arc::new(DiffuseLight::new(texture)),
|
||||||
|
Materials::Isotropic { texture } => Arc::new(Isotropic::new(texture)),
|
||||||
|
Materials::Lambertian { texture } => Arc::new(Lambertian::new(texture)),
|
||||||
|
};
|
||||||
|
if materials.insert(mc.name.clone(), v).is_some() {
|
||||||
|
return Err(ConfigError::DuplicateMaterial(mc.name));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
let hitables: Result<Vec<Box<dyn Hit>>, Self::Error> = c
|
||||||
|
.hitables
|
||||||
|
.into_iter()
|
||||||
|
.map(|hc| -> Result<Box<dyn Hit>, Self::Error> {
|
||||||
|
match hc.hitable {
|
||||||
|
Hitables::Sphere { center, radius } => Ok(Box::new(Sphere::new(
|
||||||
|
center,
|
||||||
|
radius,
|
||||||
|
Arc::clone(
|
||||||
|
materials
|
||||||
|
.get(&hc.material_name)
|
||||||
|
.ok_or(ConfigError::UnknownMaterial(hc.material_name))?,
|
||||||
|
),
|
||||||
|
))),
|
||||||
|
Hitables::Cuboid { min, max } => Ok(Box::new(Cuboid::new(
|
||||||
|
min.into(),
|
||||||
|
max.into(),
|
||||||
|
Arc::clone(
|
||||||
|
materials
|
||||||
|
.get(&hc.material_name)
|
||||||
|
.ok_or(ConfigError::UnknownMaterial(hc.material_name))?,
|
||||||
|
),
|
||||||
|
))),
|
||||||
|
Hitables::STL { path, scale } => {
|
||||||
|
let r = BufReader::new(File::open(path)?);
|
||||||
|
let stl = STL::parse(r, false)?;
|
||||||
|
Ok(Box::new(BVHTriangles::new(
|
||||||
|
&stl,
|
||||||
|
Arc::clone(
|
||||||
|
materials
|
||||||
|
.get(&hc.material_name)
|
||||||
|
.ok_or(ConfigError::UnknownMaterial(hc.material_name))?,
|
||||||
|
),
|
||||||
|
scale.unwrap_or(1.),
|
||||||
|
)))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
})
|
||||||
|
.collect();
|
||||||
|
let hitables = hitables?;
|
||||||
|
|
||||||
|
let world: Box<dyn Hit> = Box::new(HitableList::new(hitables));
|
||||||
|
let mut env_map: Option<EnvMap> = None;
|
||||||
|
|
||||||
|
if let Some(em) = c.envmap {
|
||||||
|
let im = image::open(em.path)?.into_rgb();
|
||||||
|
env_map = Some(EnvMap::new(im));
|
||||||
|
};
|
||||||
|
|
||||||
|
let camera = make_camera(&c.camera, c.scene.width, c.scene.height);
|
||||||
|
|
||||||
|
let scene = Scene {
|
||||||
|
world,
|
||||||
|
camera,
|
||||||
|
env_map,
|
||||||
|
subsamples: c.scene.subsamples.unwrap_or(8),
|
||||||
|
adaptive_subsampling: c.scene.adaptive_subsampling,
|
||||||
|
num_threads: c.scene.num_threads,
|
||||||
|
width: c.scene.width,
|
||||||
|
height: c.scene.height,
|
||||||
|
global_illumination: c.scene.global_illumination.unwrap_or(true),
|
||||||
|
};
|
||||||
|
Ok(scene)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn make_camera(cfg: &CameraConfig, width: usize, height: usize) -> Camera {
|
||||||
|
Camera::new(
|
||||||
|
cfg.lookfrom.into(),
|
||||||
|
cfg.lookat.into(),
|
||||||
|
Vec3::new(0., 1., 0.),
|
||||||
|
cfg.fov,
|
||||||
|
width as f32 / height as f32,
|
||||||
|
cfg.aperture,
|
||||||
|
cfg.focus_dist,
|
||||||
|
cfg.time_min,
|
||||||
|
cfg.time_max,
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug, Deserialize)]
|
||||||
|
struct SceneConfig {
|
||||||
|
subsamples: Option<usize>,
|
||||||
|
adaptive_subsampling: Option<f32>,
|
||||||
|
num_threads: Option<usize>,
|
||||||
|
width: usize,
|
||||||
|
height: usize,
|
||||||
|
global_illumination: Option<bool>,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug, Deserialize)]
|
||||||
|
#[serde(tag = "type")]
|
||||||
|
struct HitableConfig {
|
||||||
|
material_name: String,
|
||||||
|
#[serde(flatten)]
|
||||||
|
hitable: Hitables,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug, Deserialize)]
|
||||||
|
#[serde(tag = "type")]
|
||||||
|
enum Hitables {
|
||||||
|
#[serde(rename = "sphere")]
|
||||||
|
Sphere { center: [f32; 3], radius: f32 },
|
||||||
|
#[serde(rename = "cuboid")]
|
||||||
|
Cuboid { min: [f32; 3], max: [f32; 3] },
|
||||||
|
#[serde(rename = "stl")]
|
||||||
|
STL { path: PathBuf, scale: Option<f32> },
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug, Deserialize)]
|
||||||
|
struct MaterialConfig {
|
||||||
|
name: String,
|
||||||
|
#[serde(flatten)]
|
||||||
|
material: Materials,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug, Deserialize)]
|
||||||
|
#[serde(tag = "type")]
|
||||||
|
enum Materials {
|
||||||
|
#[serde(rename = "metal")]
|
||||||
|
Metal { albedo: [f32; 3], fuzzy: f32 },
|
||||||
|
#[serde(rename = "dielectric")]
|
||||||
|
Dielectric { ref_idx: f32 },
|
||||||
|
// TODO(wathiede): these all take Textures, for now, only support RGB
|
||||||
|
#[serde(rename = "diffuse_light")]
|
||||||
|
DiffuseLight { texture: [f32; 3] },
|
||||||
|
#[serde(rename = "isotropic")]
|
||||||
|
Isotropic { texture: [f32; 3] },
|
||||||
|
#[serde(rename = "lambertian")]
|
||||||
|
Lambertian { texture: [f32; 3] },
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug, Deserialize)]
|
||||||
|
pub struct CameraConfig {
|
||||||
|
lookfrom: [f32; 3],
|
||||||
|
lookat: [f32; 3],
|
||||||
|
fov: f32,
|
||||||
|
aperture: f32,
|
||||||
|
focus_dist: f32,
|
||||||
|
time_min: f32,
|
||||||
|
time_max: f32,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug, Deserialize)]
|
||||||
|
struct EnvMapConfig {
|
||||||
|
path: PathBuf,
|
||||||
|
}
|
||||||
@@ -5,6 +5,9 @@ pub struct Ray {
|
|||||||
pub origin: Vec3,
|
pub origin: Vec3,
|
||||||
pub direction: Vec3,
|
pub direction: Vec3,
|
||||||
pub time: f32,
|
pub time: f32,
|
||||||
|
|
||||||
|
// Precache 1/direction, a single ray intersects multiple AABB's, and divides are more
|
||||||
|
// expensive than multiplies.
|
||||||
pub inv_direction: Vec3,
|
pub inv_direction: Vec3,
|
||||||
pub sign: [usize; 3],
|
pub sign: [usize; 3],
|
||||||
}
|
}
|
||||||
@@ -14,7 +17,7 @@ impl Ray {
|
|||||||
where
|
where
|
||||||
V: Into<Vec3>,
|
V: Into<Vec3>,
|
||||||
{
|
{
|
||||||
let direction = direction.into();
|
let direction: Vec3 = direction.into();
|
||||||
let origin = origin.into();
|
let origin = origin.into();
|
||||||
let inv = 1. / direction;
|
let inv = 1. / direction;
|
||||||
Ray {
|
Ray {
|
||||||
|
|||||||
@@ -1,12 +1,14 @@
|
|||||||
// There are many math functions in this file, so we allow single letter variable names.
|
// There are many math functions in this file, so we allow single letter variable names.
|
||||||
#![allow(clippy::many_single_char_names)]
|
#![allow(clippy::many_single_char_names)]
|
||||||
use crate::aabb::AABB;
|
use crate::{
|
||||||
use crate::hitable::Hit;
|
aabb::AABB,
|
||||||
use crate::hitable::HitRecord;
|
hitable::{Hit, HitRecord},
|
||||||
use crate::material::Material;
|
material::Material,
|
||||||
use crate::ray::Ray;
|
ray::Ray,
|
||||||
use crate::vec3::Vec3;
|
vec3::Vec3,
|
||||||
|
};
|
||||||
|
|
||||||
|
#[derive(Debug)]
|
||||||
pub struct XYRect<M>
|
pub struct XYRect<M>
|
||||||
where
|
where
|
||||||
M: Material,
|
M: Material,
|
||||||
@@ -68,6 +70,7 @@ where
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[derive(Debug)]
|
||||||
pub struct XZRect<M>
|
pub struct XZRect<M>
|
||||||
where
|
where
|
||||||
M: Material,
|
M: Material,
|
||||||
@@ -129,6 +132,7 @@ where
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[derive(Debug)]
|
||||||
pub struct YZRect<M>
|
pub struct YZRect<M>
|
||||||
where
|
where
|
||||||
M: Material,
|
M: Material,
|
||||||
|
|||||||
@@ -1,40 +1,41 @@
|
|||||||
use std::fmt;
|
use std::{
|
||||||
use std::ops::AddAssign;
|
collections::HashMap,
|
||||||
use std::ops::Range;
|
fmt,
|
||||||
use std::path::Path;
|
ops::{AddAssign, Range},
|
||||||
use std::path::PathBuf;
|
path::{Path, PathBuf},
|
||||||
use std::str;
|
str, sync,
|
||||||
use std::sync;
|
sync::{
|
||||||
use std::sync::mpsc::sync_channel;
|
mpsc::{sync_channel, Receiver, SyncSender},
|
||||||
use std::sync::mpsc::Receiver;
|
Arc, Mutex,
|
||||||
use std::sync::mpsc::SyncSender;
|
},
|
||||||
use std::sync::Arc;
|
thread,
|
||||||
use std::sync::Mutex;
|
time::{Duration, Instant},
|
||||||
use std::thread;
|
};
|
||||||
use std::time;
|
|
||||||
|
|
||||||
use core_affinity;
|
use core_affinity;
|
||||||
use log::info;
|
use log::{info, trace};
|
||||||
use log::trace;
|
|
||||||
use num_cpus;
|
use num_cpus;
|
||||||
use rand;
|
use rand::{self, Rng};
|
||||||
use rand::Rng;
|
|
||||||
use serde_derive::Serialize;
|
use serde_derive::Serialize;
|
||||||
use structopt::StructOpt;
|
use structopt::StructOpt;
|
||||||
|
|
||||||
use crate::camera::Camera;
|
use crate::{
|
||||||
use crate::hitable::Hit;
|
camera::Camera,
|
||||||
use crate::human;
|
hitable::Hit,
|
||||||
use crate::material::Lambertian;
|
human,
|
||||||
use crate::output;
|
material::{Lambertian, Material},
|
||||||
use crate::ray::Ray;
|
output,
|
||||||
use crate::scenes;
|
output::OutputManager,
|
||||||
use crate::sphere::Sphere;
|
ray::Ray,
|
||||||
use crate::texture::ConstantTexture;
|
scenes,
|
||||||
use crate::texture::EnvMap;
|
sphere::Sphere,
|
||||||
use crate::vec3::Vec3;
|
texture::{ConstantTexture, EnvMap},
|
||||||
|
vec3::Vec3,
|
||||||
|
};
|
||||||
|
use strum::{EnumString, EnumVariantNames};
|
||||||
|
|
||||||
#[derive(Debug)]
|
#[derive(Debug, EnumString, EnumVariantNames, strum::Display)]
|
||||||
|
#[strum(serialize_all = "snake_case")]
|
||||||
pub enum Model {
|
pub enum Model {
|
||||||
BVH,
|
BVH,
|
||||||
Bench,
|
Bench,
|
||||||
@@ -45,24 +46,30 @@ pub enum Model {
|
|||||||
Mandelbrot,
|
Mandelbrot,
|
||||||
PerlinDebug,
|
PerlinDebug,
|
||||||
Spheramid,
|
Spheramid,
|
||||||
|
Stltest,
|
||||||
Test,
|
Test,
|
||||||
|
Tron,
|
||||||
Tutorial,
|
Tutorial,
|
||||||
|
Dragon,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Model {
|
impl Model {
|
||||||
pub fn scene(&self, opt: &Opt) -> Scene {
|
pub fn scene(&self, opt: &Opt) -> Scene {
|
||||||
match self {
|
match self {
|
||||||
Model::BVH => scenes::bvh::new(&opt),
|
Model::BVH => scenes::bvh::new(opt),
|
||||||
Model::Bench => scenes::bench::new(&opt),
|
Model::Bench => scenes::bench::new(opt),
|
||||||
Model::Book => scenes::book::new(&opt),
|
Model::Book => scenes::book::new(opt),
|
||||||
Model::CornellBox => scenes::cornell_box::new(&opt),
|
Model::CornellBox => scenes::cornell_box::new(opt),
|
||||||
Model::CornellSmoke => scenes::cornell_smoke::new(&opt),
|
Model::CornellSmoke => scenes::cornell_smoke::new(opt),
|
||||||
Model::Final => scenes::final_scene::new(&opt),
|
Model::Dragon => scenes::dragon::new(opt),
|
||||||
Model::Mandelbrot => scenes::mandelbrot::new(&opt),
|
Model::Final => scenes::final_scene::new(opt),
|
||||||
Model::PerlinDebug => scenes::perlin_debug::new(&opt),
|
Model::Mandelbrot => scenes::mandelbrot::new(opt),
|
||||||
Model::Spheramid => scenes::spheramid::new(&opt),
|
Model::PerlinDebug => scenes::perlin_debug::new(opt),
|
||||||
Model::Test => scenes::test::new(&opt),
|
Model::Spheramid => scenes::spheramid::new(opt),
|
||||||
Model::Tutorial => scenes::tutorial::new(&opt),
|
Model::Stltest => scenes::stltest::new(opt),
|
||||||
|
Model::Test => scenes::test::new(opt),
|
||||||
|
Model::Tron => scenes::tron::new(opt),
|
||||||
|
Model::Tutorial => scenes::tutorial::new(opt),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -76,44 +83,6 @@ impl fmt::Display for ModelParseError {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl str::FromStr for Model {
|
|
||||||
type Err = ModelParseError;
|
|
||||||
fn from_str(s: &str) -> std::result::Result<Self, Self::Err> {
|
|
||||||
match s {
|
|
||||||
"bench" => Ok(Model::Bench),
|
|
||||||
"book" => Ok(Model::Book),
|
|
||||||
"bvh" => Ok(Model::BVH),
|
|
||||||
"cornell_box" => Ok(Model::CornellBox),
|
|
||||||
"cornell_smoke" => Ok(Model::CornellSmoke),
|
|
||||||
"final" => Ok(Model::Final),
|
|
||||||
"mandelbrot" => Ok(Model::Mandelbrot),
|
|
||||||
"perlin_debug" => Ok(Model::PerlinDebug),
|
|
||||||
"spheramid" => Ok(Model::Spheramid),
|
|
||||||
"test" => Ok(Model::Test),
|
|
||||||
"tutorial" => Ok(Model::Tutorial),
|
|
||||||
_ => Err(ModelParseError(s.to_owned())),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl std::string::ToString for Model {
|
|
||||||
fn to_string(&self) -> String {
|
|
||||||
match self {
|
|
||||||
Model::BVH => "bvh".to_string(),
|
|
||||||
Model::Bench => "bench".to_string(),
|
|
||||||
Model::Book => "book".to_string(),
|
|
||||||
Model::CornellBox => "cornell_box".to_string(),
|
|
||||||
Model::CornellSmoke => "cornell_smoke".to_string(),
|
|
||||||
Model::Final => "final".to_string(),
|
|
||||||
Model::Mandelbrot => "mandelbrot".to_string(),
|
|
||||||
Model::PerlinDebug => "perlin_debug".to_string(),
|
|
||||||
Model::Spheramid => "spheramid".to_string(),
|
|
||||||
Model::Test => "test".to_string(),
|
|
||||||
Model::Tutorial => "tutorial".to_string(),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Debug, StructOpt)]
|
#[derive(Debug, StructOpt)]
|
||||||
#[structopt(name = "tracer", about = "An experimental ray tracer.")]
|
#[structopt(name = "tracer", about = "An experimental ray tracer.")]
|
||||||
pub struct Opt {
|
pub struct Opt {
|
||||||
@@ -129,19 +98,29 @@ pub struct Opt {
|
|||||||
/// Sub-samples per pixel
|
/// Sub-samples per pixel
|
||||||
#[structopt(short = "s", long = "subsample", default_value = "8")]
|
#[structopt(short = "s", long = "subsample", default_value = "8")]
|
||||||
pub subsamples: usize,
|
pub subsamples: usize,
|
||||||
/// Select scene to render, one of: "bench", "book", "tutorial", "bvh", "test", "cornell_box",
|
/// Select scene to render.
|
||||||
/// "cornell_smoke", "perlin_debug", "final"
|
#[structopt(long = "model")]
|
||||||
#[structopt(long = "model", default_value = "book")]
|
pub model: Option<Model>,
|
||||||
pub model: Model,
|
/// Toml config describing scene.
|
||||||
|
#[structopt(long = "config")]
|
||||||
|
pub config: Option<PathBuf>,
|
||||||
/// Path to store pprof profile data, i.e. /tmp/cpuprofile.pprof
|
/// Path to store pprof profile data, i.e. /tmp/cpuprofile.pprof
|
||||||
#[structopt(long = "pprof", parse(from_os_str))]
|
#[structopt(long = "pprof", parse(from_os_str))]
|
||||||
pub pprof: Option<PathBuf>,
|
pub pprof: Option<PathBuf>,
|
||||||
/// Use acceleration data structure, may be BVH or kd-tree depending on scene.
|
/// Use acceleration data structure, may be BVH or kd-tree depending on scene.
|
||||||
#[structopt(long = "use_accel")]
|
#[structopt(long = "use_accel")]
|
||||||
pub use_accel: bool,
|
pub use_accel: bool,
|
||||||
|
/// Host:port of running tev instance.
|
||||||
|
#[structopt(long = "tev_addr")]
|
||||||
|
pub tev_addr: Option<String>,
|
||||||
|
|
||||||
/// Output directory
|
/// Output directory
|
||||||
#[structopt(parse(from_os_str), default_value = "/tmp/tracer")]
|
#[structopt(
|
||||||
|
short = "o",
|
||||||
|
long = "output",
|
||||||
|
parse(from_os_str),
|
||||||
|
default_value = "/tmp/tracer"
|
||||||
|
)]
|
||||||
pub output: PathBuf,
|
pub output: PathBuf,
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -153,18 +132,20 @@ pub fn opt_hash(opt: &Opt) -> String {
|
|||||||
opt.height,
|
opt.height,
|
||||||
opt.subsamples,
|
opt.subsamples,
|
||||||
opt.pprof.is_some(),
|
opt.pprof.is_some(),
|
||||||
opt.model.to_string(),
|
opt.model.as_ref().unwrap().to_string(),
|
||||||
opt.use_accel,
|
opt.use_accel,
|
||||||
opt.output.display().to_string().replace("/", "_")
|
opt.output.display().to_string().replace('/', "_")
|
||||||
)
|
)
|
||||||
}
|
}
|
||||||
|
|
||||||
// TODO(wathiede): implement the skips and then the renderer could use json as an input file type.
|
// TODO(wathiede): implement the skips and then the renderer could use json as an input file type.
|
||||||
#[derive(Serialize)]
|
#[derive(Debug, Serialize)]
|
||||||
#[serde(rename_all = "camelCase")]
|
#[serde(rename_all = "camelCase")]
|
||||||
pub struct Scene {
|
pub struct Scene {
|
||||||
#[serde(skip)]
|
#[serde(skip)]
|
||||||
pub world: Box<dyn Hit>,
|
pub world: Box<dyn Hit>,
|
||||||
|
//#[serde(skip)]
|
||||||
|
//pub materials: HashMap<String, Box<dyn Material>>,
|
||||||
#[serde(skip)]
|
#[serde(skip)]
|
||||||
pub camera: Camera,
|
pub camera: Camera,
|
||||||
pub subsamples: usize,
|
pub subsamples: usize,
|
||||||
@@ -269,6 +250,7 @@ fn trace_pixel_adaptive(
|
|||||||
x_range: Range<f32>,
|
x_range: Range<f32>,
|
||||||
y_range: Range<f32>,
|
y_range: Range<f32>,
|
||||||
scene: &Scene,
|
scene: &Scene,
|
||||||
|
output: &OutputManager,
|
||||||
) -> (Vec3, usize) {
|
) -> (Vec3, usize) {
|
||||||
let w = scene.width as f32;
|
let w = scene.width as f32;
|
||||||
let h = scene.height as f32;
|
let h = scene.height as f32;
|
||||||
@@ -283,7 +265,7 @@ fn trace_pixel_adaptive(
|
|||||||
&scene.env_map,
|
&scene.env_map,
|
||||||
);
|
);
|
||||||
if depth == 0 {
|
if depth == 0 {
|
||||||
output::set_pixel(output::ADAPTIVE_DEPTH, x, y, [1., 0., 0.].into());
|
output.set_pixel(output::ADAPTIVE_DEPTH, x, y, [1., 0., 0.].into());
|
||||||
return (center, rays);
|
return (center, rays);
|
||||||
}
|
}
|
||||||
// t = top
|
// t = top
|
||||||
@@ -325,6 +307,7 @@ fn trace_pixel_adaptive(
|
|||||||
x_range.start..x_mid,
|
x_range.start..x_mid,
|
||||||
y_range.start..y_mid,
|
y_range.start..y_mid,
|
||||||
scene,
|
scene,
|
||||||
|
output,
|
||||||
);
|
);
|
||||||
let tr = trace_pixel_adaptive(
|
let tr = trace_pixel_adaptive(
|
||||||
depth - 1,
|
depth - 1,
|
||||||
@@ -334,6 +317,7 @@ fn trace_pixel_adaptive(
|
|||||||
x_mid..x_range.end,
|
x_mid..x_range.end,
|
||||||
y_range.start..y_mid,
|
y_range.start..y_mid,
|
||||||
scene,
|
scene,
|
||||||
|
output,
|
||||||
);
|
);
|
||||||
let bl = trace_pixel_adaptive(
|
let bl = trace_pixel_adaptive(
|
||||||
depth - 1,
|
depth - 1,
|
||||||
@@ -343,6 +327,7 @@ fn trace_pixel_adaptive(
|
|||||||
x_range.start..x_mid,
|
x_range.start..x_mid,
|
||||||
y_mid..y_range.end,
|
y_mid..y_range.end,
|
||||||
scene,
|
scene,
|
||||||
|
output,
|
||||||
);
|
);
|
||||||
let br = trace_pixel_adaptive(
|
let br = trace_pixel_adaptive(
|
||||||
depth - 1,
|
depth - 1,
|
||||||
@@ -352,13 +337,14 @@ fn trace_pixel_adaptive(
|
|||||||
x_mid..x_range.end,
|
x_mid..x_range.end,
|
||||||
y_mid..y_range.end,
|
y_mid..y_range.end,
|
||||||
scene,
|
scene,
|
||||||
|
output,
|
||||||
);
|
);
|
||||||
let pixel = (tl.0 + tr.0 + bl.0 + br.0) / 4.;
|
let pixel = (tl.0 + tr.0 + bl.0 + br.0) / 4.;
|
||||||
let rays = tl.1 + tr.1 + bl.1 + br.1;
|
let rays = tl.1 + tr.1 + bl.1 + br.1;
|
||||||
(pixel, rays)
|
(pixel, rays)
|
||||||
} else {
|
} else {
|
||||||
if depth == MAX_ADAPTIVE_DEPTH {
|
if depth == MAX_ADAPTIVE_DEPTH {
|
||||||
output::set_pixel(output::ADAPTIVE_DEPTH, x, y, [0., 1., 0.].into());
|
output.set_pixel(output::ADAPTIVE_DEPTH, x, y, [0., 1., 0.].into());
|
||||||
}
|
}
|
||||||
(corners, rays)
|
(corners, rays)
|
||||||
}
|
}
|
||||||
@@ -378,7 +364,7 @@ fn trace_pixel_random(x: usize, y: usize, scene: &Scene) -> (Vec3, usize) {
|
|||||||
)
|
)
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Clone, Copy)]
|
#[derive(Clone, Copy, Default)]
|
||||||
struct RenderStats {
|
struct RenderStats {
|
||||||
rays: usize,
|
rays: usize,
|
||||||
pixels: usize,
|
pixels: usize,
|
||||||
@@ -394,33 +380,37 @@ impl AddAssign for RenderStats {
|
|||||||
}
|
}
|
||||||
|
|
||||||
fn progress(
|
fn progress(
|
||||||
|
start_time: Instant,
|
||||||
last_stat: &RenderStats,
|
last_stat: &RenderStats,
|
||||||
current_stat: &RenderStats,
|
current_stat: &RenderStats,
|
||||||
time_diff: time::Duration,
|
time_diff: Duration,
|
||||||
pixel_total: usize,
|
pixel_total: usize,
|
||||||
) -> String {
|
) -> String {
|
||||||
let human = human::Formatter::new();
|
let human = human::Formatter::new();
|
||||||
let pixel_diff = current_stat.pixels - last_stat.pixels;
|
let pixel_diff = current_stat.pixels - last_stat.pixels;
|
||||||
let ray_diff = current_stat.rays - last_stat.rays;
|
let ray_diff = current_stat.rays - last_stat.rays;
|
||||||
|
let now = Instant::now();
|
||||||
|
let start_diff = now - start_time;
|
||||||
|
let ratio = current_stat.pixels as f32 / pixel_total as f32;
|
||||||
|
let percent = ratio * 100.;
|
||||||
|
let elapsed = start_diff.as_secs_f32();
|
||||||
|
let total = elapsed * (1. / ratio);
|
||||||
|
let eta = total - elapsed;
|
||||||
format!(
|
format!(
|
||||||
"{:7} / {:7}pixels ({:2}%) {:7}pixels/s {:7}rays/s",
|
"{:7} / {:7}pixels ({:2.0}%) {:7}pixels/s {:7}rays/s eta {:.0}s",
|
||||||
human.format(current_stat.pixels as f64),
|
human.format(current_stat.pixels as f64),
|
||||||
human.format(pixel_total as f64),
|
human.format(pixel_total as f64),
|
||||||
100 * current_stat.pixels / pixel_total,
|
percent,
|
||||||
human.format(pixel_diff as f64 / time_diff.as_secs_f64()),
|
human.format(pixel_diff as f64 / time_diff.as_secs_f64()),
|
||||||
human.format(ray_diff as f64 / time_diff.as_secs_f64())
|
human.format(ray_diff as f64 / time_diff.as_secs_f64()),
|
||||||
|
eta
|
||||||
)
|
)
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Default for RenderStats {
|
|
||||||
fn default() -> Self {
|
|
||||||
RenderStats { rays: 0, pixels: 0 }
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
enum Request {
|
enum Request {
|
||||||
Pixel { x: usize, y: usize },
|
Pixel { x: usize, y: usize },
|
||||||
Line { width: usize, y: usize },
|
Line { width: usize, y: usize },
|
||||||
|
// TODO(wathiede): add Cohort that does 4x4 or 8x8 pixel chunks.
|
||||||
}
|
}
|
||||||
|
|
||||||
enum Response {
|
enum Response {
|
||||||
@@ -437,7 +427,7 @@ enum Response {
|
|||||||
},
|
},
|
||||||
}
|
}
|
||||||
|
|
||||||
fn render_pixel(scene: &Scene, x: usize, y: usize) -> (Vec3, usize) {
|
fn render_pixel(scene: &Scene, x: usize, y: usize, output: &OutputManager) -> (Vec3, usize) {
|
||||||
let (pixel, rays) = if let Some(threshold) = scene.adaptive_subsampling {
|
let (pixel, rays) = if let Some(threshold) = scene.adaptive_subsampling {
|
||||||
trace_pixel_adaptive(
|
trace_pixel_adaptive(
|
||||||
MAX_ADAPTIVE_DEPTH,
|
MAX_ADAPTIVE_DEPTH,
|
||||||
@@ -447,6 +437,7 @@ fn render_pixel(scene: &Scene, x: usize, y: usize) -> (Vec3, usize) {
|
|||||||
0.0..1.0,
|
0.0..1.0,
|
||||||
0.0..1.0,
|
0.0..1.0,
|
||||||
scene,
|
scene,
|
||||||
|
output,
|
||||||
)
|
)
|
||||||
} else {
|
} else {
|
||||||
let (pixel, rays) = (0..scene.subsamples)
|
let (pixel, rays) = (0..scene.subsamples)
|
||||||
@@ -455,7 +446,7 @@ fn render_pixel(scene: &Scene, x: usize, y: usize) -> (Vec3, usize) {
|
|||||||
([0., 0., 0.].into(), 0),
|
([0., 0., 0.].into(), 0),
|
||||||
|(p1, r1): (Vec3, usize), (p2, r2): (Vec3, usize)| ((p1 + p2), (r1 + r2)),
|
|(p1, r1): (Vec3, usize), (p2, r2): (Vec3, usize)| ((p1 + p2), (r1 + r2)),
|
||||||
);
|
);
|
||||||
output::set_pixel_grey(output::RAYS_PER_PIXEL, x, y, rays as f32);
|
output.set_pixel_grey(output::RAYS_PER_PIXEL, x, y, rays as f32);
|
||||||
(pixel / scene.subsamples as f32, rays)
|
(pixel / scene.subsamples as f32, rays)
|
||||||
};
|
};
|
||||||
// Gamma correct, use gamma 2 correction, which is 1/gamma where gamma=2 which is 1/2 or
|
// Gamma correct, use gamma 2 correction, which is 1/gamma where gamma=2 which is 1/2 or
|
||||||
@@ -471,6 +462,7 @@ fn render_worker(
|
|||||||
scene: &Scene,
|
scene: &Scene,
|
||||||
input_chan: Arc<Mutex<Receiver<Request>>>,
|
input_chan: Arc<Mutex<Receiver<Request>>>,
|
||||||
output_chan: &SyncSender<Response>,
|
output_chan: &SyncSender<Response>,
|
||||||
|
output: &OutputManager,
|
||||||
) {
|
) {
|
||||||
loop {
|
loop {
|
||||||
let job = { input_chan.lock().unwrap().recv() };
|
let job = { input_chan.lock().unwrap().recv() };
|
||||||
@@ -485,7 +477,7 @@ fn render_worker(
|
|||||||
let batch = false;
|
let batch = false;
|
||||||
if batch {
|
if batch {
|
||||||
let (pixels, rays): (Vec<Vec3>, Vec<usize>) = (0..width)
|
let (pixels, rays): (Vec<Vec3>, Vec<usize>) = (0..width)
|
||||||
.map(|x| render_pixel(scene, x, y))
|
.map(|x| render_pixel(scene, x, y, output))
|
||||||
.collect::<Vec<(_, _)>>()
|
.collect::<Vec<(_, _)>>()
|
||||||
.into_iter()
|
.into_iter()
|
||||||
.unzip();
|
.unzip();
|
||||||
@@ -502,7 +494,7 @@ fn render_worker(
|
|||||||
.expect("failed to send pixel response");
|
.expect("failed to send pixel response");
|
||||||
} else {
|
} else {
|
||||||
(0..width).for_each(|x| {
|
(0..width).for_each(|x| {
|
||||||
let (pixel, rays) = render_pixel(scene, x, y);
|
let (pixel, rays) = render_pixel(scene, x, y, output);
|
||||||
output_chan
|
output_chan
|
||||||
.send(Response::Pixel {
|
.send(Response::Pixel {
|
||||||
x,
|
x,
|
||||||
@@ -516,7 +508,7 @@ fn render_worker(
|
|||||||
}
|
}
|
||||||
Request::Pixel { x, y } => {
|
Request::Pixel { x, y } => {
|
||||||
trace!("tid {} x {} y {}", tid, x, y);
|
trace!("tid {} x {} y {}", tid, x, y);
|
||||||
let (pixel, rays) = render_pixel(scene, x, y);
|
let (pixel, rays) = render_pixel(scene, x, y, output);
|
||||||
output_chan
|
output_chan
|
||||||
.send(Response::Pixel {
|
.send(Response::Pixel {
|
||||||
x,
|
x,
|
||||||
@@ -531,8 +523,19 @@ fn render_worker(
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn render(scene: Scene, output_dir: &Path) -> std::result::Result<(), std::io::Error> {
|
/*
|
||||||
let num_threads = scene.num_threads.unwrap_or_else(num_cpus::get);
|
lazy_static! {
|
||||||
|
static ref DEBUGGER: Arc<Mutex<OutputManager>> = Arc::new(Mutex::new(OutputManager::new()));
|
||||||
|
}
|
||||||
|
*/
|
||||||
|
|
||||||
|
pub fn render(
|
||||||
|
scene: Scene,
|
||||||
|
output_dir: &Path,
|
||||||
|
tev_addr: &Option<String>,
|
||||||
|
) -> std::result::Result<(), std::io::Error> {
|
||||||
|
// Default to half the cores to disable hyperthreading.
|
||||||
|
let num_threads = scene.num_threads.unwrap_or_else(|| num_cpus::get() / 2);
|
||||||
let (pixel_req_tx, pixel_req_rx) = sync_channel(2 * num_threads);
|
let (pixel_req_tx, pixel_req_rx) = sync_channel(2 * num_threads);
|
||||||
let (pixel_resp_tx, pixel_resp_rx) = sync_channel(2 * num_threads);
|
let (pixel_resp_tx, pixel_resp_rx) = sync_channel(2 * num_threads);
|
||||||
|
|
||||||
@@ -546,20 +549,23 @@ pub fn render(scene: Scene, output_dir: &Path) -> std::result::Result<(), std::i
|
|||||||
} else {
|
} else {
|
||||||
core_ids
|
core_ids
|
||||||
};
|
};
|
||||||
|
let output = output::OutputManager::new(tev_addr)?;
|
||||||
|
let output = Arc::new(output);
|
||||||
|
|
||||||
info!("Creating {} render threads", core_ids.len());
|
info!("Creating {} render threads", core_ids.len());
|
||||||
output::register_image(
|
output.register_image(
|
||||||
output::MAIN_IMAGE.to_string(),
|
output::MAIN_IMAGE.to_string(),
|
||||||
(scene.width, scene.height),
|
(scene.width, scene.height),
|
||||||
output::ImageType::RGB01,
|
output::ImageType::RGB01,
|
||||||
);
|
);
|
||||||
if scene.adaptive_subsampling.is_some() {
|
if scene.adaptive_subsampling.is_some() {
|
||||||
output::register_image(
|
output.register_image(
|
||||||
output::ADAPTIVE_DEPTH.to_string(),
|
output::ADAPTIVE_DEPTH.to_string(),
|
||||||
(scene.width, scene.height),
|
(scene.width, scene.height),
|
||||||
output::ImageType::RGB01,
|
output::ImageType::RGB01,
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
output::register_image(
|
output.register_image(
|
||||||
output::RAYS_PER_PIXEL.to_string(),
|
output::RAYS_PER_PIXEL.to_string(),
|
||||||
(scene.width, scene.height),
|
(scene.width, scene.height),
|
||||||
output::ImageType::GreyNormalized,
|
output::ImageType::GreyNormalized,
|
||||||
@@ -573,19 +579,19 @@ pub fn render(scene: Scene, output_dir: &Path) -> std::result::Result<(), std::i
|
|||||||
let s = sync::Arc::clone(&scene);
|
let s = sync::Arc::clone(&scene);
|
||||||
let pixel_req_rx = pixel_req_rx.clone();
|
let pixel_req_rx = pixel_req_rx.clone();
|
||||||
let pixel_resp_tx = pixel_resp_tx.clone();
|
let pixel_resp_tx = pixel_resp_tx.clone();
|
||||||
|
let output = sync::Arc::clone(&output);
|
||||||
thread::spawn(move || {
|
thread::spawn(move || {
|
||||||
core_affinity::set_for_current(id);
|
core_affinity::set_for_current(id);
|
||||||
render_worker(i, &s, pixel_req_rx, &pixel_resp_tx);
|
render_worker(i, &s, pixel_req_rx, &pixel_resp_tx, &output);
|
||||||
})
|
})
|
||||||
})
|
})
|
||||||
.collect::<Vec<_>>();
|
.collect::<Vec<_>>();
|
||||||
drop(pixel_req_rx);
|
drop(pixel_req_rx);
|
||||||
drop(pixel_resp_tx);
|
drop(pixel_resp_tx);
|
||||||
|
|
||||||
let start_time = time::Instant::now();
|
|
||||||
let (w, h) = (scene.width, scene.height);
|
let (w, h) = (scene.width, scene.height);
|
||||||
handles.push(thread::spawn(move || {
|
handles.push(thread::spawn(move || {
|
||||||
let batch_line_requests = true;
|
let batch_line_requests = false;
|
||||||
if batch_line_requests {
|
if batch_line_requests {
|
||||||
for y in 0..h {
|
for y in 0..h {
|
||||||
pixel_req_tx
|
pixel_req_tx
|
||||||
@@ -607,29 +613,36 @@ pub fn render(scene: Scene, output_dir: &Path) -> std::result::Result<(), std::i
|
|||||||
info!("Rendering with {} subsamples", scene.subsamples);
|
info!("Rendering with {} subsamples", scene.subsamples);
|
||||||
|
|
||||||
let pixel_total = scene.width * scene.height;
|
let pixel_total = scene.width * scene.height;
|
||||||
let mut last_time = time::Instant::now();
|
let mut last_time = Instant::now();
|
||||||
let mut last_stat: RenderStats = Default::default();
|
let mut last_stat: RenderStats = Default::default();
|
||||||
let mut current_stat: RenderStats = Default::default();
|
let mut current_stat: RenderStats = Default::default();
|
||||||
|
let render_start_time = Instant::now();
|
||||||
for resp in pixel_resp_rx {
|
for resp in pixel_resp_rx {
|
||||||
match resp {
|
match resp {
|
||||||
Response::Pixel { x, y, pixel, rs } => {
|
Response::Pixel { x, y, pixel, rs } => {
|
||||||
current_stat += rs;
|
current_stat += rs;
|
||||||
output::set_pixel(output::MAIN_IMAGE, x, y, pixel);
|
output.set_pixel(output::MAIN_IMAGE, x, y, pixel);
|
||||||
}
|
}
|
||||||
Response::Line { y, pixels, rs } => {
|
Response::Line { y, pixels, rs } => {
|
||||||
current_stat += rs;
|
current_stat += rs;
|
||||||
for (x, pixel) in pixels.iter().enumerate() {
|
for (x, pixel) in pixels.iter().enumerate() {
|
||||||
output::set_pixel(output::MAIN_IMAGE, x, y, *pixel);
|
output.set_pixel(output::MAIN_IMAGE, x, y, *pixel);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
let now = time::Instant::now();
|
let now = Instant::now();
|
||||||
let time_diff = now - last_time;
|
let time_diff = now - last_time;
|
||||||
if time_diff > time::Duration::from_secs(1) {
|
if time_diff > Duration::from_secs(5) {
|
||||||
info!(
|
println!(
|
||||||
"{}",
|
"{}",
|
||||||
progress(&last_stat, ¤t_stat, time_diff, pixel_total)
|
progress(
|
||||||
|
render_start_time,
|
||||||
|
&last_stat,
|
||||||
|
¤t_stat,
|
||||||
|
time_diff,
|
||||||
|
pixel_total
|
||||||
|
)
|
||||||
);
|
);
|
||||||
last_stat = current_stat;
|
last_stat = current_stat;
|
||||||
last_time = now;
|
last_time = now;
|
||||||
@@ -638,12 +651,18 @@ pub fn render(scene: Scene, output_dir: &Path) -> std::result::Result<(), std::i
|
|||||||
for thr in handles {
|
for thr in handles {
|
||||||
thr.join().expect("thread join");
|
thr.join().expect("thread join");
|
||||||
}
|
}
|
||||||
let time_diff = time::Instant::now() - start_time;
|
let time_diff = Instant::now() - render_start_time;
|
||||||
info!(
|
println!(
|
||||||
"Runtime {} seconds {}",
|
"Render {} seconds {}",
|
||||||
time_diff.as_secs_f32(),
|
time_diff.as_secs_f32(),
|
||||||
progress(&Default::default(), ¤t_stat, time_diff, pixel_total)
|
progress(
|
||||||
|
render_start_time,
|
||||||
|
&Default::default(),
|
||||||
|
¤t_stat,
|
||||||
|
time_diff,
|
||||||
|
pixel_total
|
||||||
|
)
|
||||||
);
|
);
|
||||||
|
|
||||||
output::write_images(&scene, time_diff, output_dir)
|
output.write_images(&scene, time_diff, output_dir)
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,13 +1,13 @@
|
|||||||
use std::f32::consts::PI;
|
use std::f32::{consts::PI, MAX, MIN};
|
||||||
use std::f32::MAX;
|
|
||||||
use std::f32::MIN;
|
|
||||||
|
|
||||||
use crate::aabb::AABB;
|
use crate::{
|
||||||
use crate::hitable::Hit;
|
aabb::AABB,
|
||||||
use crate::hitable::HitRecord;
|
hitable::{Hit, HitRecord},
|
||||||
use crate::ray::Ray;
|
ray::Ray,
|
||||||
use crate::vec3::Vec3;
|
vec3::Vec3,
|
||||||
|
};
|
||||||
|
|
||||||
|
#[derive(Debug)]
|
||||||
pub struct RotateY<H>
|
pub struct RotateY<H>
|
||||||
where
|
where
|
||||||
H: Hit,
|
H: Hit,
|
||||||
|
|||||||
47
rtiow/renderer/src/scale.rs
Normal file
47
rtiow/renderer/src/scale.rs
Normal file
@@ -0,0 +1,47 @@
|
|||||||
|
use crate::{
|
||||||
|
aabb::AABB,
|
||||||
|
hitable::{Hit, HitRecord},
|
||||||
|
ray::Ray,
|
||||||
|
};
|
||||||
|
|
||||||
|
#[derive(Debug)]
|
||||||
|
pub struct Scale<H>
|
||||||
|
where
|
||||||
|
H: Hit,
|
||||||
|
{
|
||||||
|
hitable: H,
|
||||||
|
scale: f32,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<H> Scale<H>
|
||||||
|
where
|
||||||
|
H: Hit,
|
||||||
|
{
|
||||||
|
pub fn new(hitable: H, scale: f32) -> Scale<H> {
|
||||||
|
Scale { hitable, scale }
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<H> Hit for Scale<H>
|
||||||
|
where
|
||||||
|
H: Hit,
|
||||||
|
{
|
||||||
|
fn hit(&self, r: Ray, t_min: f32, t_max: f32) -> Option<HitRecord> {
|
||||||
|
let moved_r = Ray::new(r.origin / self.scale, r.direction, r.time);
|
||||||
|
if let Some(rec) = self.hitable.hit(moved_r, t_min, t_max) {
|
||||||
|
return Some(HitRecord {
|
||||||
|
p: rec.p * self.scale,
|
||||||
|
t: rec.t * self.scale,
|
||||||
|
..rec
|
||||||
|
});
|
||||||
|
}
|
||||||
|
None
|
||||||
|
}
|
||||||
|
|
||||||
|
fn bounding_box(&self, t_min: f32, t_max: f32) -> Option<AABB> {
|
||||||
|
if let Some(bbox) = self.hitable.bounding_box(t_min, t_max) {
|
||||||
|
return Some(AABB::new(bbox.min() * self.scale, bbox.max() * self.scale));
|
||||||
|
}
|
||||||
|
None
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -1,18 +1,18 @@
|
|||||||
use log::trace;
|
use log::trace;
|
||||||
use rand;
|
use rand::{self, Rng};
|
||||||
use rand::Rng;
|
|
||||||
|
|
||||||
use crate::bvh::BVH;
|
use crate::{
|
||||||
use crate::camera::Camera;
|
bvh::BVH,
|
||||||
use crate::hitable::Hit;
|
camera::Camera,
|
||||||
use crate::hitable_list::HitableList;
|
hitable::Hit,
|
||||||
use crate::kdtree::KDTree;
|
hitable_list::HitableList,
|
||||||
use crate::material::Lambertian;
|
kdtree::KDTree,
|
||||||
use crate::renderer::Opt;
|
material::Lambertian,
|
||||||
use crate::renderer::Scene;
|
renderer::{Opt, Scene},
|
||||||
use crate::sphere::Sphere;
|
sphere::Sphere,
|
||||||
use crate::texture::ConstantTexture;
|
texture::ConstantTexture,
|
||||||
use crate::vec3::Vec3;
|
vec3::Vec3,
|
||||||
|
};
|
||||||
|
|
||||||
pub fn new(opt: &Opt) -> Scene {
|
pub fn new(opt: &Opt) -> Scene {
|
||||||
let lookfrom = Vec3::new(20., 20., 20.);
|
let lookfrom = Vec3::new(20., 20., 20.);
|
||||||
|
|||||||
@@ -1,21 +1,16 @@
|
|||||||
use rand;
|
use rand::{self, Rng};
|
||||||
use rand::Rng;
|
|
||||||
|
|
||||||
use crate::camera::Camera;
|
use crate::{
|
||||||
use crate::hitable::Hit;
|
camera::Camera,
|
||||||
use crate::hitable_list::HitableList;
|
hitable::Hit,
|
||||||
use crate::kdtree::KDTree;
|
hitable_list::HitableList,
|
||||||
use crate::material::Dielectric;
|
kdtree::KDTree,
|
||||||
use crate::material::Lambertian;
|
material::{Dielectric, Lambertian, Material, Metal},
|
||||||
use crate::material::Material;
|
renderer::{Opt, Scene},
|
||||||
use crate::material::Metal;
|
sphere::Sphere,
|
||||||
use crate::renderer::Opt;
|
texture::{CheckerTexture, ConstantTexture, EnvMap},
|
||||||
use crate::renderer::Scene;
|
vec3::Vec3,
|
||||||
use crate::sphere::Sphere;
|
};
|
||||||
use crate::texture::CheckerTexture;
|
|
||||||
use crate::texture::ConstantTexture;
|
|
||||||
use crate::texture::EnvMap;
|
|
||||||
use crate::vec3::Vec3;
|
|
||||||
|
|
||||||
pub fn new(opt: &Opt) -> Scene {
|
pub fn new(opt: &Opt) -> Scene {
|
||||||
let lookfrom = Vec3::new(13., 2., 3.);
|
let lookfrom = Vec3::new(13., 2., 3.);
|
||||||
|
|||||||
@@ -1,16 +1,16 @@
|
|||||||
use log::trace;
|
use log::trace;
|
||||||
|
|
||||||
use crate::bvh::BVH;
|
use crate::{
|
||||||
use crate::camera::Camera;
|
bvh::BVH,
|
||||||
use crate::hitable::Hit;
|
camera::Camera,
|
||||||
use crate::material::Lambertian;
|
hitable::Hit,
|
||||||
use crate::material::Metal;
|
material::{Lambertian, Metal},
|
||||||
use crate::moving_sphere::MovingSphere;
|
moving_sphere::MovingSphere,
|
||||||
use crate::renderer::Opt;
|
renderer::{Opt, Scene},
|
||||||
use crate::renderer::Scene;
|
sphere::Sphere,
|
||||||
use crate::sphere::Sphere;
|
texture::ConstantTexture,
|
||||||
use crate::texture::ConstantTexture;
|
vec3::Vec3,
|
||||||
use crate::vec3::Vec3;
|
};
|
||||||
|
|
||||||
pub fn new(opt: &Opt) -> Scene {
|
pub fn new(opt: &Opt) -> Scene {
|
||||||
let lookfrom = Vec3::new(3., 3., 2.);
|
let lookfrom = Vec3::new(3., 3., 2.);
|
||||||
|
|||||||
@@ -1,22 +1,20 @@
|
|||||||
use std::sync::Arc;
|
use std::sync::Arc;
|
||||||
|
|
||||||
use crate::camera::Camera;
|
use crate::{
|
||||||
use crate::cuboid::Cuboid;
|
camera::Camera,
|
||||||
use crate::flip_normals::FlipNormals;
|
cuboid::Cuboid,
|
||||||
use crate::hitable::Hit;
|
flip_normals::FlipNormals,
|
||||||
use crate::hitable_list::HitableList;
|
hitable::Hit,
|
||||||
use crate::kdtree::KDTree;
|
hitable_list::HitableList,
|
||||||
use crate::material::DiffuseLight;
|
kdtree::KDTree,
|
||||||
use crate::material::Lambertian;
|
material::{DiffuseLight, Lambertian},
|
||||||
use crate::rect::XYRect;
|
rect::{XYRect, XZRect, YZRect},
|
||||||
use crate::rect::XZRect;
|
renderer::{Opt, Scene},
|
||||||
use crate::rect::YZRect;
|
rotate::RotateY,
|
||||||
use crate::renderer::Opt;
|
texture::ConstantTexture,
|
||||||
use crate::renderer::Scene;
|
translate::Translate,
|
||||||
use crate::rotate::RotateY;
|
vec3::Vec3,
|
||||||
use crate::texture::ConstantTexture;
|
};
|
||||||
use crate::translate::Translate;
|
|
||||||
use crate::vec3::Vec3;
|
|
||||||
|
|
||||||
pub fn new(opt: &Opt) -> Scene {
|
pub fn new(opt: &Opt) -> Scene {
|
||||||
let lookfrom = Vec3::new(278., 278., -800.);
|
let lookfrom = Vec3::new(278., 278., -800.);
|
||||||
|
|||||||
@@ -1,24 +1,21 @@
|
|||||||
use std::sync::Arc;
|
use std::sync::Arc;
|
||||||
|
|
||||||
use crate::camera::Camera;
|
use crate::{
|
||||||
use crate::constant_medium::ConstantMedium;
|
camera::Camera,
|
||||||
use crate::cuboid::Cuboid;
|
constant_medium::ConstantMedium,
|
||||||
use crate::flip_normals::FlipNormals;
|
cuboid::Cuboid,
|
||||||
use crate::hitable::Hit;
|
flip_normals::FlipNormals,
|
||||||
use crate::hitable_list::HitableList;
|
hitable::Hit,
|
||||||
use crate::kdtree::KDTree;
|
hitable_list::HitableList,
|
||||||
use crate::material::DiffuseLight;
|
kdtree::KDTree,
|
||||||
use crate::material::Lambertian;
|
material::{DiffuseLight, Lambertian, Material},
|
||||||
use crate::material::Material;
|
rect::{XYRect, XZRect, YZRect},
|
||||||
use crate::rect::XYRect;
|
renderer::{Opt, Scene},
|
||||||
use crate::rect::XZRect;
|
rotate::RotateY,
|
||||||
use crate::rect::YZRect;
|
texture::ConstantTexture,
|
||||||
use crate::renderer::Opt;
|
translate::Translate,
|
||||||
use crate::renderer::Scene;
|
vec3::Vec3,
|
||||||
use crate::rotate::RotateY;
|
};
|
||||||
use crate::texture::ConstantTexture;
|
|
||||||
use crate::translate::Translate;
|
|
||||||
use crate::vec3::Vec3;
|
|
||||||
|
|
||||||
pub fn new(opt: &Opt) -> Scene {
|
pub fn new(opt: &Opt) -> Scene {
|
||||||
let lookfrom = Vec3::new(278., 278., -800.);
|
let lookfrom = Vec3::new(278., 278., -800.);
|
||||||
|
|||||||
116
rtiow/renderer/src/scenes/dragon.rs
Normal file
116
rtiow/renderer/src/scenes/dragon.rs
Normal file
@@ -0,0 +1,116 @@
|
|||||||
|
use std::{
|
||||||
|
f32::consts::PI,
|
||||||
|
io::{BufReader, Cursor},
|
||||||
|
};
|
||||||
|
|
||||||
|
use stl::STL;
|
||||||
|
|
||||||
|
use crate::{
|
||||||
|
bvh_triangles::BVHTriangles,
|
||||||
|
camera::Camera,
|
||||||
|
colors::generate_rainbow,
|
||||||
|
hitable::Hit,
|
||||||
|
hitable_list::HitableList,
|
||||||
|
kdtree::KDTree,
|
||||||
|
material::{Lambertian, Metal},
|
||||||
|
renderer::{Opt, Scene},
|
||||||
|
rotate::RotateY,
|
||||||
|
scale::Scale,
|
||||||
|
sphere::Sphere,
|
||||||
|
texture::{ConstantTexture, EnvMap},
|
||||||
|
translate::Translate,
|
||||||
|
vec3::Vec3,
|
||||||
|
};
|
||||||
|
|
||||||
|
pub fn new(opt: &Opt) -> Scene {
|
||||||
|
let lookfrom = Vec3::new(0., 80., 80.);
|
||||||
|
let lookat = Vec3::new(0., 0., 0.);
|
||||||
|
let dist_to_focus = 10.0;
|
||||||
|
let aperture = 0.0;
|
||||||
|
let time_min = 0.;
|
||||||
|
let time_max = 1.;
|
||||||
|
let camera = Camera::new(
|
||||||
|
lookfrom,
|
||||||
|
lookat,
|
||||||
|
Vec3::new(0., 1., 0.),
|
||||||
|
45.,
|
||||||
|
opt.width as f32 / opt.height as f32,
|
||||||
|
aperture,
|
||||||
|
dist_to_focus,
|
||||||
|
time_min,
|
||||||
|
time_max,
|
||||||
|
);
|
||||||
|
//let dragon_material = Dielectric::new(1.5);
|
||||||
|
let dragon_material = Metal::new(Vec3::new(0.6, 0.6, 0.6), 0.0);
|
||||||
|
//let dragon_material = Lambertian::new(ConstantTexture::new(Vec3::new(1.0, 1.0, 0.2)));
|
||||||
|
|
||||||
|
let ground_color = if opt.use_accel {
|
||||||
|
ConstantTexture::new(Vec3::new(1.0, 0.4, 0.4))
|
||||||
|
} else {
|
||||||
|
ConstantTexture::new(Vec3::new(0.4, 0.4, 0.4))
|
||||||
|
};
|
||||||
|
|
||||||
|
let stl_cube = STL::parse(
|
||||||
|
BufReader::new(Cursor::new(include_bytes!("../../stls/dragon.stl"))),
|
||||||
|
false,
|
||||||
|
)
|
||||||
|
.expect("failed to parse cube");
|
||||||
|
let _light_size = 50.;
|
||||||
|
let _light_height = 200.;
|
||||||
|
let sphere_radius = 5.;
|
||||||
|
let circle_radius = 40.;
|
||||||
|
let num_spheres = 16;
|
||||||
|
let palette = generate_rainbow(num_spheres);
|
||||||
|
let spheres: Vec<Box<dyn Hit>> = (0..num_spheres)
|
||||||
|
.map(|i| (i, i as f32, num_spheres as f32))
|
||||||
|
.map(|(idx, idx_f, n)| (idx, idx_f * 2. * PI / n))
|
||||||
|
.map(|(idx, rad)| -> Box<dyn Hit> {
|
||||||
|
let x = circle_radius * rad.cos();
|
||||||
|
let y = 4. * sphere_radius;
|
||||||
|
let z = circle_radius * rad.sin();
|
||||||
|
let c = palette[idx];
|
||||||
|
Box::new(Sphere::new(
|
||||||
|
[x, y, z],
|
||||||
|
sphere_radius,
|
||||||
|
Lambertian::new(ConstantTexture::new(c)),
|
||||||
|
))
|
||||||
|
})
|
||||||
|
.collect();
|
||||||
|
let mut objects: Vec<Box<dyn Hit>> = vec![
|
||||||
|
Box::new(Sphere::new(
|
||||||
|
Vec3::new(0., 0.1, -0.5),
|
||||||
|
0.1,
|
||||||
|
Lambertian::new([0., 1., 1.]),
|
||||||
|
)),
|
||||||
|
// Earth sized sphere
|
||||||
|
//Box::new(Sphere::new( Vec3::new(0., -10000., 0.), 10000., Lambertian::new(ground_color),)),
|
||||||
|
// STL Mesh
|
||||||
|
Box::new(crate::debug_hit::DebugHit::new(RotateY::new(
|
||||||
|
Translate::new(
|
||||||
|
BVHTriangles::new(&stl_cube, dragon_material, 250.),
|
||||||
|
[0., -10., 0.],
|
||||||
|
),
|
||||||
|
180.,
|
||||||
|
))),
|
||||||
|
];
|
||||||
|
objects.extend(spheres);
|
||||||
|
|
||||||
|
let world: Box<dyn Hit> = if opt.use_accel {
|
||||||
|
Box::new(KDTree::new(objects, time_min, time_max))
|
||||||
|
} else {
|
||||||
|
Box::new(HitableList::new(objects))
|
||||||
|
};
|
||||||
|
let skybox_bytes = include_bytes!("../../images/envmap.jpg");
|
||||||
|
let skybox = image::load_from_memory(skybox_bytes).unwrap().to_rgb();
|
||||||
|
Scene {
|
||||||
|
camera,
|
||||||
|
world,
|
||||||
|
subsamples: opt.subsamples,
|
||||||
|
num_threads: opt.num_threads,
|
||||||
|
width: opt.width,
|
||||||
|
height: opt.height,
|
||||||
|
global_illumination: true,
|
||||||
|
env_map: Some(EnvMap::new(skybox)),
|
||||||
|
..Default::default()
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -1,33 +1,26 @@
|
|||||||
use std::sync::Arc;
|
use std::sync::Arc;
|
||||||
|
|
||||||
use image;
|
use image;
|
||||||
use rand;
|
use rand::{self, Rng};
|
||||||
use rand::Rng;
|
|
||||||
|
|
||||||
use crate::camera::Camera;
|
use crate::{
|
||||||
use crate::constant_medium::ConstantMedium;
|
camera::Camera,
|
||||||
use crate::cuboid::Cuboid;
|
constant_medium::ConstantMedium,
|
||||||
use crate::hitable::Hit;
|
cuboid::Cuboid,
|
||||||
use crate::hitable_list::HitableList;
|
hitable::Hit,
|
||||||
use crate::kdtree::KDTree;
|
hitable_list::HitableList,
|
||||||
use crate::material::Dielectric;
|
kdtree::KDTree,
|
||||||
use crate::material::DiffuseLight;
|
material::{Dielectric, DiffuseLight, Lambertian, Material, Metal},
|
||||||
use crate::material::Lambertian;
|
moving_sphere::MovingSphere,
|
||||||
use crate::material::Material;
|
noise::{perlin::Perlin, NoiseType},
|
||||||
use crate::material::Metal;
|
rect::XZRect,
|
||||||
use crate::moving_sphere::MovingSphere;
|
renderer::{Opt, Scene},
|
||||||
use crate::noise::perlin::Perlin;
|
rotate::RotateY,
|
||||||
use crate::noise::NoiseType;
|
sphere::Sphere,
|
||||||
use crate::rect::XZRect;
|
texture::{ConstantTexture, ImageTexture, NoiseTexture},
|
||||||
use crate::renderer::Opt;
|
translate::Translate,
|
||||||
use crate::renderer::Scene;
|
vec3::Vec3,
|
||||||
use crate::rotate::RotateY;
|
};
|
||||||
use crate::sphere::Sphere;
|
|
||||||
use crate::texture::ConstantTexture;
|
|
||||||
use crate::texture::ImageTexture;
|
|
||||||
use crate::texture::NoiseTexture;
|
|
||||||
use crate::translate::Translate;
|
|
||||||
use crate::vec3::Vec3;
|
|
||||||
|
|
||||||
pub fn new(opt: &Opt) -> Scene {
|
pub fn new(opt: &Opt) -> Scene {
|
||||||
let lookfrom = Vec3::new(478., 278., -600.);
|
let lookfrom = Vec3::new(478., 278., -600.);
|
||||||
|
|||||||
@@ -1,24 +1,18 @@
|
|||||||
use rand;
|
use rand;
|
||||||
|
|
||||||
use crate::camera::Camera;
|
use crate::{
|
||||||
use crate::hitable::Hit;
|
camera::Camera,
|
||||||
use crate::hitable_list::HitableList;
|
hitable::Hit,
|
||||||
use crate::kdtree::KDTree;
|
hitable_list::HitableList,
|
||||||
use crate::material::DiffuseLight;
|
kdtree::KDTree,
|
||||||
use crate::material::Lambertian;
|
material::{DiffuseLight, Lambertian},
|
||||||
use crate::noise::perlin::Perlin;
|
noise::{perlin::Perlin, NoiseType},
|
||||||
use crate::noise::NoiseType;
|
rect::{XYRect, XZRect, YZRect},
|
||||||
use crate::rect::XYRect;
|
renderer::{Opt, Scene},
|
||||||
use crate::rect::XZRect;
|
sphere::Sphere,
|
||||||
use crate::rect::YZRect;
|
texture::{ConstantTexture, ImageTexture, Mandelbrot, NoiseTexture},
|
||||||
use crate::renderer::Opt;
|
vec3::Vec3,
|
||||||
use crate::renderer::Scene;
|
};
|
||||||
use crate::sphere::Sphere;
|
|
||||||
use crate::texture::ConstantTexture;
|
|
||||||
use crate::texture::ImageTexture;
|
|
||||||
use crate::texture::Mandelbrot;
|
|
||||||
use crate::texture::NoiseTexture;
|
|
||||||
use crate::vec3::Vec3;
|
|
||||||
|
|
||||||
pub fn new(opt: &Opt) -> Scene {
|
pub fn new(opt: &Opt) -> Scene {
|
||||||
let lookfrom = Vec3::new(20., 20., 20.);
|
let lookfrom = Vec3::new(20., 20., 20.);
|
||||||
|
|||||||
@@ -3,9 +3,12 @@ pub mod book;
|
|||||||
pub mod bvh;
|
pub mod bvh;
|
||||||
pub mod cornell_box;
|
pub mod cornell_box;
|
||||||
pub mod cornell_smoke;
|
pub mod cornell_smoke;
|
||||||
|
pub mod dragon;
|
||||||
pub mod final_scene;
|
pub mod final_scene;
|
||||||
pub mod mandelbrot;
|
pub mod mandelbrot;
|
||||||
pub mod perlin_debug;
|
pub mod perlin_debug;
|
||||||
pub mod spheramid;
|
pub mod spheramid;
|
||||||
|
pub mod stltest;
|
||||||
pub mod test;
|
pub mod test;
|
||||||
|
pub mod tron;
|
||||||
pub mod tutorial;
|
pub mod tutorial;
|
||||||
|
|||||||
@@ -2,19 +2,18 @@ use std::sync::Arc;
|
|||||||
|
|
||||||
use rand;
|
use rand;
|
||||||
|
|
||||||
use crate::camera::Camera;
|
use crate::{
|
||||||
use crate::hitable::Hit;
|
camera::Camera,
|
||||||
use crate::hitable_list::HitableList;
|
hitable::Hit,
|
||||||
use crate::kdtree::KDTree;
|
hitable_list::HitableList,
|
||||||
use crate::material::Lambertian;
|
kdtree::KDTree,
|
||||||
use crate::noise::perlin::Perlin;
|
material::Lambertian,
|
||||||
use crate::noise::NoiseType;
|
noise::{perlin::Perlin, NoiseType},
|
||||||
use crate::renderer::Opt;
|
renderer::{Opt, Scene},
|
||||||
use crate::renderer::Scene;
|
sphere::Sphere,
|
||||||
use crate::sphere::Sphere;
|
texture::{NoiseTexture, Texture},
|
||||||
use crate::texture::NoiseTexture;
|
vec3::Vec3,
|
||||||
use crate::texture::Texture;
|
};
|
||||||
use crate::vec3::Vec3;
|
|
||||||
|
|
||||||
pub fn new(opt: &Opt) -> Scene {
|
pub fn new(opt: &Opt) -> Scene {
|
||||||
let lookfrom = Vec3::new(13., 2., 3.);
|
let lookfrom = Vec3::new(13., 2., 3.);
|
||||||
|
|||||||
@@ -1,18 +1,15 @@
|
|||||||
use crate::camera::Camera;
|
use crate::{
|
||||||
use crate::hitable::Hit;
|
camera::Camera,
|
||||||
use crate::hitable_list::HitableList;
|
hitable::Hit,
|
||||||
use crate::material::Dielectric;
|
hitable_list::HitableList,
|
||||||
use crate::material::Lambertian;
|
material::{Dielectric, Lambertian, Metal},
|
||||||
use crate::material::Metal;
|
moving_sphere::MovingSphere,
|
||||||
use crate::moving_sphere::MovingSphere;
|
rect::XYRect,
|
||||||
use crate::rect::XYRect;
|
renderer::{Opt, Scene},
|
||||||
use crate::renderer::Opt;
|
sphere::Sphere,
|
||||||
use crate::renderer::Scene;
|
texture::{CheckerTexture, ConstantTexture, EnvMap},
|
||||||
use crate::sphere::Sphere;
|
vec3::Vec3,
|
||||||
use crate::texture::CheckerTexture;
|
};
|
||||||
use crate::texture::ConstantTexture;
|
|
||||||
use crate::texture::EnvMap;
|
|
||||||
use crate::vec3::Vec3;
|
|
||||||
|
|
||||||
// Draws many spheres along each positive axis.
|
// Draws many spheres along each positive axis.
|
||||||
// Blue X-positive
|
// Blue X-positive
|
||||||
|
|||||||
122
rtiow/renderer/src/scenes/stltest.rs
Normal file
122
rtiow/renderer/src/scenes/stltest.rs
Normal file
@@ -0,0 +1,122 @@
|
|||||||
|
use std::{
|
||||||
|
io::{BufReader, Cursor},
|
||||||
|
sync::Arc,
|
||||||
|
};
|
||||||
|
|
||||||
|
use stl::STL;
|
||||||
|
|
||||||
|
use crate::{
|
||||||
|
bvh_triangles::BVHTriangles,
|
||||||
|
camera::Camera,
|
||||||
|
cuboid::Cuboid,
|
||||||
|
hitable::Hit,
|
||||||
|
hitable_list::HitableList,
|
||||||
|
kdtree::KDTree,
|
||||||
|
material::{Dielectric, Lambertian, Metal},
|
||||||
|
renderer::{Opt, Scene},
|
||||||
|
sphere::Sphere,
|
||||||
|
texture::{ConstantTexture, EnvMap},
|
||||||
|
translate::Translate,
|
||||||
|
vec3::Vec3,
|
||||||
|
};
|
||||||
|
|
||||||
|
pub fn new(opt: &Opt) -> Scene {
|
||||||
|
let lookfrom = Vec3::new(0., 40., -100.);
|
||||||
|
let lookat = Vec3::new(0., 10., 0.);
|
||||||
|
let dist_to_focus = 10.0;
|
||||||
|
let aperture = 0.0;
|
||||||
|
let time_min = 0.;
|
||||||
|
let time_max = 1.;
|
||||||
|
let camera = Camera::new(
|
||||||
|
lookfrom,
|
||||||
|
lookat,
|
||||||
|
Vec3::new(0., 1., 0.),
|
||||||
|
45.,
|
||||||
|
opt.width as f32 / opt.height as f32,
|
||||||
|
aperture,
|
||||||
|
dist_to_focus,
|
||||||
|
time_min,
|
||||||
|
time_max,
|
||||||
|
);
|
||||||
|
let ground_color = if opt.use_accel {
|
||||||
|
ConstantTexture::new(Vec3::new(1.0, 0.4, 0.4))
|
||||||
|
} else {
|
||||||
|
ConstantTexture::new(Vec3::new(0.4, 0.4, 0.4))
|
||||||
|
};
|
||||||
|
|
||||||
|
let glass = Dielectric::new(1.5);
|
||||||
|
let metal = Metal::new(Vec3::new(0.8, 0.8, 0.8), 0.2);
|
||||||
|
let red = Lambertian::new(ConstantTexture::new(Vec3::new(1.0, 0.2, 0.2)));
|
||||||
|
|
||||||
|
//let box_material = glass;
|
||||||
|
let _ = glass;
|
||||||
|
let _ = metal;
|
||||||
|
let _ = red;
|
||||||
|
|
||||||
|
let stl_cube = STL::parse(
|
||||||
|
BufReader::new(Cursor::new(include_bytes!("../../stls/cube.stl"))),
|
||||||
|
false,
|
||||||
|
)
|
||||||
|
.expect("failed to parse cube");
|
||||||
|
let objects: Vec<Box<dyn Hit>> = vec![
|
||||||
|
// Light from above - white
|
||||||
|
// Earth sized sphere
|
||||||
|
Box::new(Sphere::new(
|
||||||
|
Vec3::new(0., -10000., 0.),
|
||||||
|
10000.,
|
||||||
|
Lambertian::new(ground_color),
|
||||||
|
)),
|
||||||
|
Box::new(Sphere::new(
|
||||||
|
Vec3::new(0., 20., -40.),
|
||||||
|
10.,
|
||||||
|
Lambertian::new(ConstantTexture::new(Vec3::new(1., 0.2, 1.))),
|
||||||
|
)),
|
||||||
|
// Blue sphere
|
||||||
|
Box::new(Sphere::new(
|
||||||
|
Vec3::new(0., 20., 40.),
|
||||||
|
20.,
|
||||||
|
Lambertian::new(ConstantTexture::new(Vec3::new(0.2, 0.2, 1.))),
|
||||||
|
)),
|
||||||
|
Box::new(Sphere::new(
|
||||||
|
Vec3::new(40., 20., 40.),
|
||||||
|
20.,
|
||||||
|
//Metal::new(Vec3::new(0.8, 0.8, 0.8), 0.2),
|
||||||
|
Lambertian::new(ConstantTexture::new(Vec3::new(0.2, 1.0, 0.2))),
|
||||||
|
)),
|
||||||
|
Box::new(Sphere::new(
|
||||||
|
Vec3::new(-40., 20., 40.),
|
||||||
|
20.,
|
||||||
|
//Metal::new(Vec3::new(0.8, 0.8, 0.8), 0.2),
|
||||||
|
Lambertian::new(ConstantTexture::new(Vec3::new(1.0, 0.2, 0.2))),
|
||||||
|
)),
|
||||||
|
// STL Mesh
|
||||||
|
Box::new(Translate::new(
|
||||||
|
BVHTriangles::new(&stl_cube, glass, 1.),
|
||||||
|
[0., 10., 0.],
|
||||||
|
)),
|
||||||
|
//Box::new(BVHTriangles::new(&stl_cube, box_material.clone())),
|
||||||
|
Box::new(Cuboid::new(
|
||||||
|
[-20., 0., 0.].into(),
|
||||||
|
[0., 20., 20.].into(),
|
||||||
|
Arc::new(red),
|
||||||
|
)),
|
||||||
|
];
|
||||||
|
let world: Box<dyn Hit> = if opt.use_accel {
|
||||||
|
Box::new(KDTree::new(objects, time_min, time_max))
|
||||||
|
} else {
|
||||||
|
Box::new(HitableList::new(objects))
|
||||||
|
};
|
||||||
|
let skybox_bytes = include_bytes!("../../images/envmap.jpg");
|
||||||
|
let skybox = image::load_from_memory(skybox_bytes).unwrap().to_rgb();
|
||||||
|
Scene {
|
||||||
|
camera,
|
||||||
|
world,
|
||||||
|
subsamples: opt.subsamples,
|
||||||
|
num_threads: opt.num_threads,
|
||||||
|
width: opt.width,
|
||||||
|
height: opt.height,
|
||||||
|
global_illumination: true,
|
||||||
|
env_map: Some(EnvMap::new(skybox)),
|
||||||
|
..Default::default()
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -1,24 +1,19 @@
|
|||||||
use image;
|
use image;
|
||||||
use rand;
|
use rand;
|
||||||
|
|
||||||
use crate::camera::Camera;
|
use crate::{
|
||||||
use crate::hitable::Hit;
|
camera::Camera,
|
||||||
use crate::hitable_list::HitableList;
|
hitable::Hit,
|
||||||
use crate::kdtree::KDTree;
|
hitable_list::HitableList,
|
||||||
use crate::material::DiffuseLight;
|
kdtree::KDTree,
|
||||||
use crate::material::Lambertian;
|
material::{DiffuseLight, Lambertian},
|
||||||
use crate::noise::perlin::Perlin;
|
noise::{perlin::Perlin, NoiseType},
|
||||||
use crate::noise::NoiseType;
|
rect::{XYRect, XZRect, YZRect},
|
||||||
use crate::rect::XYRect;
|
renderer::{Opt, Scene},
|
||||||
use crate::rect::XZRect;
|
sphere::Sphere,
|
||||||
use crate::rect::YZRect;
|
texture::{ConstantTexture, ImageTexture, NoiseTexture},
|
||||||
use crate::renderer::Opt;
|
vec3::Vec3,
|
||||||
use crate::renderer::Scene;
|
};
|
||||||
use crate::sphere::Sphere;
|
|
||||||
use crate::texture::ConstantTexture;
|
|
||||||
use crate::texture::ImageTexture;
|
|
||||||
use crate::texture::NoiseTexture;
|
|
||||||
use crate::vec3::Vec3;
|
|
||||||
|
|
||||||
pub fn new(opt: &Opt) -> Scene {
|
pub fn new(opt: &Opt) -> Scene {
|
||||||
let lookfrom = Vec3::new(20., 20., 20.);
|
let lookfrom = Vec3::new(20., 20., 20.);
|
||||||
|
|||||||
124
rtiow/renderer/src/scenes/tron.rs
Normal file
124
rtiow/renderer/src/scenes/tron.rs
Normal file
@@ -0,0 +1,124 @@
|
|||||||
|
use std::sync::Arc;
|
||||||
|
|
||||||
|
use log::info;
|
||||||
|
|
||||||
|
use crate::{
|
||||||
|
camera::Camera,
|
||||||
|
glowybox::Glowybox,
|
||||||
|
hitable::Hit,
|
||||||
|
hitable_list::HitableList,
|
||||||
|
kdtree::KDTree,
|
||||||
|
material::{DiffuseLight, Lambertian},
|
||||||
|
rect::{XYRect, XZRect, YZRect},
|
||||||
|
renderer::{Opt, Scene},
|
||||||
|
sphere::Sphere,
|
||||||
|
texture::ConstantTexture,
|
||||||
|
vec3::Vec3,
|
||||||
|
};
|
||||||
|
|
||||||
|
pub fn new(opt: &Opt) -> Scene {
|
||||||
|
let lookfrom = Vec3::new(20., 20., 20.);
|
||||||
|
let lookat = Vec3::new(0., 1., 0.);
|
||||||
|
let dist_to_focus = 10.0;
|
||||||
|
let aperture = 0.0;
|
||||||
|
let time_min = 0.;
|
||||||
|
let time_max = 1.;
|
||||||
|
let camera = Camera::new(
|
||||||
|
lookfrom,
|
||||||
|
lookat,
|
||||||
|
Vec3::new(0., 1., 0.),
|
||||||
|
20.,
|
||||||
|
opt.width as f32 / opt.height as f32,
|
||||||
|
aperture,
|
||||||
|
dist_to_focus,
|
||||||
|
time_min,
|
||||||
|
time_max,
|
||||||
|
);
|
||||||
|
let ground_color = if opt.use_accel {
|
||||||
|
ConstantTexture::new(Vec3::new(1.0, 0.4, 0.4))
|
||||||
|
} else {
|
||||||
|
ConstantTexture::new(Vec3::new(0.4, 1.0, 0.4))
|
||||||
|
};
|
||||||
|
|
||||||
|
let lights: Vec<Box<dyn Hit>> = vec![
|
||||||
|
Box::new(XZRect::new(
|
||||||
|
-100.,
|
||||||
|
100.,
|
||||||
|
-100.,
|
||||||
|
1000.,
|
||||||
|
60.,
|
||||||
|
DiffuseLight::new(ConstantTexture::new(Vec3::new(1., 1., 1.))),
|
||||||
|
)),
|
||||||
|
Box::new(YZRect::new(
|
||||||
|
1.,
|
||||||
|
3.,
|
||||||
|
-1.,
|
||||||
|
1.,
|
||||||
|
4.,
|
||||||
|
DiffuseLight::new(ConstantTexture::new(Vec3::new(4., 0., 4.))),
|
||||||
|
)),
|
||||||
|
Box::new(YZRect::new(
|
||||||
|
1.,
|
||||||
|
3.,
|
||||||
|
-1.,
|
||||||
|
1.,
|
||||||
|
-4.,
|
||||||
|
DiffuseLight::new(ConstantTexture::new(Vec3::new(0., 4., 0.))),
|
||||||
|
)),
|
||||||
|
Box::new(XZRect::new(
|
||||||
|
-1.,
|
||||||
|
1.,
|
||||||
|
-1.,
|
||||||
|
1.,
|
||||||
|
6.,
|
||||||
|
DiffuseLight::new(ConstantTexture::new(Vec3::new(4., 4., 0.))),
|
||||||
|
)),
|
||||||
|
Box::new(XYRect::new(
|
||||||
|
-1.,
|
||||||
|
1.,
|
||||||
|
1.,
|
||||||
|
3.,
|
||||||
|
-4.,
|
||||||
|
DiffuseLight::new(ConstantTexture::new(Vec3::new(0., 0., 4.))),
|
||||||
|
)),
|
||||||
|
Box::new(XYRect::new(
|
||||||
|
-1.,
|
||||||
|
1.,
|
||||||
|
1.,
|
||||||
|
3.,
|
||||||
|
4.,
|
||||||
|
DiffuseLight::new(ConstantTexture::new(Vec3::new(0., 4., 4.))),
|
||||||
|
)),
|
||||||
|
];
|
||||||
|
let mut objects: Vec<Box<dyn Hit>> = vec![
|
||||||
|
// Earth sized sphere
|
||||||
|
Box::new(Sphere::new(
|
||||||
|
Vec3::new(0., -1010., 0.),
|
||||||
|
1000.,
|
||||||
|
Lambertian::new(ground_color),
|
||||||
|
)),
|
||||||
|
Box::new(Glowybox::new(
|
||||||
|
[-4., -4., -4.].into(),
|
||||||
|
[4., 4., 4.].into(),
|
||||||
|
0.01,
|
||||||
|
Arc::new(Lambertian::new(ConstantTexture::new([0., 0., 0.]))),
|
||||||
|
Arc::new(DiffuseLight::new(ConstantTexture::new([100., 0., 0.]))),
|
||||||
|
)),
|
||||||
|
];
|
||||||
|
objects.extend(lights);
|
||||||
|
info!("objects {:?}", objects);
|
||||||
|
let world: Box<dyn Hit> = if opt.use_accel {
|
||||||
|
Box::new(KDTree::new(objects, time_min, time_max))
|
||||||
|
} else {
|
||||||
|
Box::new(HitableList::new(objects))
|
||||||
|
};
|
||||||
|
Scene {
|
||||||
|
camera,
|
||||||
|
world,
|
||||||
|
subsamples: opt.subsamples,
|
||||||
|
num_threads: opt.num_threads,
|
||||||
|
width: opt.width,
|
||||||
|
height: opt.height,
|
||||||
|
..Default::default()
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -1,18 +1,15 @@
|
|||||||
use crate::camera::Camera;
|
use crate::{
|
||||||
use crate::hitable::Hit;
|
camera::Camera,
|
||||||
use crate::hitable_list::HitableList;
|
hitable::Hit,
|
||||||
use crate::kdtree::KDTree;
|
hitable_list::HitableList,
|
||||||
use crate::material::Dielectric;
|
kdtree::KDTree,
|
||||||
use crate::material::Lambertian;
|
material::{Dielectric, Lambertian, Metal},
|
||||||
use crate::material::Metal;
|
moving_sphere::MovingSphere,
|
||||||
use crate::moving_sphere::MovingSphere;
|
renderer::{Opt, Scene},
|
||||||
use crate::renderer::Opt;
|
sphere::Sphere,
|
||||||
use crate::renderer::Scene;
|
texture::{CheckerTexture, ConstantTexture, Texture},
|
||||||
use crate::sphere::Sphere;
|
vec3::Vec3,
|
||||||
use crate::texture::CheckerTexture;
|
};
|
||||||
use crate::texture::ConstantTexture;
|
|
||||||
use crate::texture::Texture;
|
|
||||||
use crate::vec3::Vec3;
|
|
||||||
|
|
||||||
pub fn new(opt: &Opt) -> Scene {
|
pub fn new(opt: &Opt) -> Scene {
|
||||||
let lookfrom = Vec3::new(3., 4., 2.);
|
let lookfrom = Vec3::new(3., 4., 2.);
|
||||||
|
|||||||
@@ -1,13 +1,14 @@
|
|||||||
use std::f32::consts::PI;
|
use std::f32::consts::PI;
|
||||||
|
|
||||||
use crate::aabb::AABB;
|
use crate::{
|
||||||
use crate::hitable::Hit;
|
aabb::AABB,
|
||||||
use crate::hitable::HitRecord;
|
hitable::{Hit, HitRecord},
|
||||||
use crate::material::Material;
|
material::Material,
|
||||||
use crate::ray::Ray;
|
ray::Ray,
|
||||||
use crate::vec3::dot;
|
vec3::{dot, Vec3},
|
||||||
use crate::vec3::Vec3;
|
};
|
||||||
|
|
||||||
|
#[derive(Debug)]
|
||||||
pub struct Sphere<M>
|
pub struct Sphere<M>
|
||||||
where
|
where
|
||||||
M: Material,
|
M: Material,
|
||||||
|
|||||||
@@ -1,6 +1,6 @@
|
|||||||
use crate::texture::Texture;
|
use crate::{texture::Texture, vec3::Vec3};
|
||||||
use crate::vec3::Vec3;
|
|
||||||
|
|
||||||
|
#[derive(Debug)]
|
||||||
pub struct CheckerTexture<T>
|
pub struct CheckerTexture<T>
|
||||||
where
|
where
|
||||||
T: Texture,
|
T: Texture,
|
||||||
|
|||||||
@@ -1,7 +1,6 @@
|
|||||||
use crate::texture::Texture;
|
use crate::{texture::Texture, vec3::Vec3};
|
||||||
use crate::vec3::Vec3;
|
|
||||||
|
|
||||||
#[derive(Debug, PartialEq)]
|
#[derive(Clone, Debug, PartialEq)]
|
||||||
pub struct ConstantTexture {
|
pub struct ConstantTexture {
|
||||||
color: Vec3,
|
color: Vec3,
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -2,9 +2,10 @@ use std::f32;
|
|||||||
|
|
||||||
use image::RgbImage;
|
use image::RgbImage;
|
||||||
|
|
||||||
use crate::texture::ImageTexture;
|
use crate::{
|
||||||
use crate::texture::Texture;
|
texture::{ImageTexture, Texture},
|
||||||
use crate::vec3::Vec3;
|
vec3::Vec3,
|
||||||
|
};
|
||||||
|
|
||||||
#[derive(Debug)]
|
#[derive(Debug)]
|
||||||
pub struct EnvMap {
|
pub struct EnvMap {
|
||||||
|
|||||||
@@ -1,7 +1,6 @@
|
|||||||
use image::RgbImage;
|
use image::RgbImage;
|
||||||
|
|
||||||
use crate::texture::Texture;
|
use crate::{texture::Texture, vec3::Vec3};
|
||||||
use crate::vec3::Vec3;
|
|
||||||
|
|
||||||
#[derive(Debug)]
|
#[derive(Debug)]
|
||||||
pub struct ImageTexture {
|
pub struct ImageTexture {
|
||||||
|
|||||||
@@ -1,55 +1,15 @@
|
|||||||
#![allow(clippy::many_single_char_names)]
|
#![allow(clippy::many_single_char_names)]
|
||||||
use rand;
|
|
||||||
use rand::Rng;
|
|
||||||
|
|
||||||
use crate::texture::Texture;
|
use crate::{colors::generate_palette, texture::Texture, vec3::Vec3};
|
||||||
use crate::vec3::Vec3;
|
|
||||||
|
|
||||||
#[derive(Debug)]
|
#[derive(Debug)]
|
||||||
pub struct Mandelbrot {
|
pub struct Mandelbrot {
|
||||||
scale: f32,
|
|
||||||
palette: Vec<Vec3>,
|
palette: Vec<Vec3>,
|
||||||
}
|
}
|
||||||
|
|
||||||
// HSV values in [0..1]
|
|
||||||
// returns [r, g, b] values from 0 to 255
|
|
||||||
//From https://martin.ankerl.com/2009/12/09/how-to-create-random-colors-programmatically/
|
|
||||||
fn hsv_to_rgb(h: f32, s: f32, v: f32) -> Vec3 {
|
|
||||||
let h_i = (h * 6.) as i32;
|
|
||||||
let f = h * 6. - h_i as f32;
|
|
||||||
let p = v * (1. - s);
|
|
||||||
let q = v * (1. - f * s);
|
|
||||||
let t = v * (1. - (1. - f) * s);
|
|
||||||
match h_i {
|
|
||||||
0 => Vec3::new(v, t, p),
|
|
||||||
1 => Vec3::new(q, v, p),
|
|
||||||
2 => Vec3::new(p, v, t),
|
|
||||||
3 => Vec3::new(p, q, v),
|
|
||||||
4 => Vec3::new(t, p, v),
|
|
||||||
5 => Vec3::new(v, p, q),
|
|
||||||
_ => panic!("Unknown H value {}", h_i),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn generate_palette(num: usize) -> Vec<Vec3> {
|
|
||||||
let mut rng = rand::thread_rng();
|
|
||||||
let mut random = || rng.gen();
|
|
||||||
// use golden ratio
|
|
||||||
let golden_ratio_conjugate = 0.618_034;
|
|
||||||
let mut h = random();
|
|
||||||
(0..num)
|
|
||||||
.map(|_| {
|
|
||||||
h += golden_ratio_conjugate;
|
|
||||||
h %= 1.0;
|
|
||||||
hsv_to_rgb(h, 0.99, 0.99)
|
|
||||||
})
|
|
||||||
.collect()
|
|
||||||
}
|
|
||||||
|
|
||||||
impl Default for Mandelbrot {
|
impl Default for Mandelbrot {
|
||||||
fn default() -> Self {
|
fn default() -> Self {
|
||||||
Mandelbrot {
|
Mandelbrot {
|
||||||
scale: 2.0,
|
|
||||||
palette: generate_palette(10),
|
palette: generate_palette(10),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -4,18 +4,16 @@ mod envmap;
|
|||||||
mod image;
|
mod image;
|
||||||
mod mandelbrot;
|
mod mandelbrot;
|
||||||
mod noise;
|
mod noise;
|
||||||
pub use crate::texture::checker::CheckerTexture;
|
pub use crate::texture::{
|
||||||
pub use crate::texture::constant::ConstantTexture;
|
checker::CheckerTexture, constant::ConstantTexture, envmap::EnvMap, image::ImageTexture,
|
||||||
pub use crate::texture::envmap::EnvMap;
|
mandelbrot::Mandelbrot, noise::NoiseTexture,
|
||||||
pub use crate::texture::image::ImageTexture;
|
};
|
||||||
pub use crate::texture::mandelbrot::Mandelbrot;
|
|
||||||
pub use crate::texture::noise::NoiseTexture;
|
|
||||||
|
|
||||||
use std::sync::Arc;
|
use std::{fmt::Debug, sync::Arc};
|
||||||
|
|
||||||
use crate::vec3::Vec3;
|
use crate::vec3::Vec3;
|
||||||
|
|
||||||
pub trait Texture: Send + Sync {
|
pub trait Texture: Send + Sync + Debug {
|
||||||
fn value(&self, u: f32, v: f32, p: Vec3) -> Vec3;
|
fn value(&self, u: f32, v: f32, p: Vec3) -> Vec3;
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -31,6 +29,12 @@ impl Texture for Box<dyn Texture> {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
impl Texture for [f32; 3] {
|
||||||
|
fn value(&self, u: f32, v: f32, p: Vec3) -> Vec3 {
|
||||||
|
(*self).into()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
mod test {
|
mod test {
|
||||||
use super::*;
|
use super::*;
|
||||||
|
|||||||
@@ -1,8 +1,10 @@
|
|||||||
use crate::noise::NoiseSource;
|
use crate::{
|
||||||
use crate::noise::NoiseType;
|
noise::{NoiseSource, NoiseType},
|
||||||
use crate::texture::Texture;
|
texture::Texture,
|
||||||
use crate::vec3::Vec3;
|
vec3::Vec3,
|
||||||
|
};
|
||||||
|
|
||||||
|
#[derive(Debug)]
|
||||||
pub struct NoiseTexture<N>
|
pub struct NoiseTexture<N>
|
||||||
where
|
where
|
||||||
N: NoiseSource,
|
N: NoiseSource,
|
||||||
|
|||||||
@@ -1,9 +1,11 @@
|
|||||||
use crate::aabb::AABB;
|
use crate::{
|
||||||
use crate::hitable::Hit;
|
aabb::AABB,
|
||||||
use crate::hitable::HitRecord;
|
hitable::{Hit, HitRecord},
|
||||||
use crate::ray::Ray;
|
ray::Ray,
|
||||||
use crate::vec3::Vec3;
|
vec3::Vec3,
|
||||||
|
};
|
||||||
|
|
||||||
|
#[derive(Debug)]
|
||||||
pub struct Translate<H>
|
pub struct Translate<H>
|
||||||
where
|
where
|
||||||
H: Hit,
|
H: Hit,
|
||||||
|
|||||||
168
rtiow/renderer/src/triangles.rs
Normal file
168
rtiow/renderer/src/triangles.rs
Normal file
@@ -0,0 +1,168 @@
|
|||||||
|
use std::f32::EPSILON;
|
||||||
|
|
||||||
|
use stl::STL;
|
||||||
|
|
||||||
|
use crate::{
|
||||||
|
aabb::AABB,
|
||||||
|
hitable::{Hit, HitRecord},
|
||||||
|
material::Material,
|
||||||
|
ray::Ray,
|
||||||
|
vec3::{cross, dot, Vec3},
|
||||||
|
};
|
||||||
|
|
||||||
|
#[derive(Debug)]
|
||||||
|
pub struct Triangle {
|
||||||
|
normal: Vec3,
|
||||||
|
verts: [Vec3; 3],
|
||||||
|
// Precomputed data
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug)]
|
||||||
|
pub struct Triangles<M>
|
||||||
|
where
|
||||||
|
M: Material,
|
||||||
|
{
|
||||||
|
pub triangles: Vec<Triangle>,
|
||||||
|
pub bbox: AABB,
|
||||||
|
material: M,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<M> Triangles<M>
|
||||||
|
where
|
||||||
|
M: Material,
|
||||||
|
{
|
||||||
|
pub fn new(stl: &STL, material: M, scale_factor: f32) -> Triangles<M> {
|
||||||
|
let triangles: Vec<_> = stl
|
||||||
|
.triangles
|
||||||
|
.iter()
|
||||||
|
.map(|t| {
|
||||||
|
let v0 = t.verts[0] * scale_factor;
|
||||||
|
let v1 = t.verts[1] * scale_factor;
|
||||||
|
let v2 = t.verts[2] * scale_factor;
|
||||||
|
Triangle {
|
||||||
|
normal: t.normal,
|
||||||
|
verts: [v0, v1, v2],
|
||||||
|
}
|
||||||
|
})
|
||||||
|
.collect();
|
||||||
|
let (min, max) = triangles.iter().fold(
|
||||||
|
(Vec3::from(f32::MAX), Vec3::from(f32::MIN)),
|
||||||
|
|(min, max), t| {
|
||||||
|
let t_min_x = t.verts[0].x.min(t.verts[1].x.min(t.verts[2].x));
|
||||||
|
let t_min_y = t.verts[0].y.min(t.verts[1].y.min(t.verts[2].y));
|
||||||
|
let t_min_z = t.verts[0].z.min(t.verts[1].z.min(t.verts[2].z));
|
||||||
|
let t_max_x = t.verts[0].x.max(t.verts[1].x.max(t.verts[2].x));
|
||||||
|
let t_max_y = t.verts[0].y.max(t.verts[1].y.max(t.verts[2].y));
|
||||||
|
let t_max_z = t.verts[0].z.max(t.verts[1].z.max(t.verts[2].z));
|
||||||
|
|
||||||
|
(
|
||||||
|
Vec3::from([min.x.min(t_min_x), min.y.min(t_min_y), min.z.min(t_min_z)]),
|
||||||
|
Vec3::from([max.x.max(t_max_x), max.y.max(t_max_y), max.z.max(t_max_z)]),
|
||||||
|
)
|
||||||
|
},
|
||||||
|
);
|
||||||
|
let bbox = AABB::new(min, max);
|
||||||
|
Triangles {
|
||||||
|
triangles,
|
||||||
|
bbox,
|
||||||
|
material,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Based on https://www.scratchapixel.com/lessons/3d-basic-rendering/ray-tracing-rendering-a-triangle/moller-trumbore-ray-triangle-intersection.html
|
||||||
|
fn ray_triangle_intersect_moller_trumbore(r: Ray, tri: &Triangle) -> Option<RayTriangleResult> {
|
||||||
|
// #ifdef MOLLER_TRUMBORE
|
||||||
|
// Vec3f v0v1 = v1 - v0;
|
||||||
|
// Vec3f v0v2 = v2 - v0;
|
||||||
|
// Vec3f pvec = dir.crossProduct(v0v2);
|
||||||
|
// float det = v0v1.dotProduct(pvec);
|
||||||
|
// #ifdef CULLING
|
||||||
|
// // if the determinant is negative, the triangle is 'back facing'
|
||||||
|
// // if the determinant is close to 0, the ray misses the triangle
|
||||||
|
// if (det < kEpsilon) return false;
|
||||||
|
// #else
|
||||||
|
// // ray and triangle are parallel if det is close to 0
|
||||||
|
// if (fabs(det) < kEpsilon) return false;
|
||||||
|
// #endif
|
||||||
|
// float invDet = 1 / det;
|
||||||
|
//
|
||||||
|
// Vec3f tvec = orig - v0;
|
||||||
|
// u = tvec.dotProduct(pvec) * invDet;
|
||||||
|
// if (u < 0 || u > 1) return false;
|
||||||
|
//
|
||||||
|
// Vec3f qvec = tvec.crossProduct(v0v1);
|
||||||
|
// v = dir.dotProduct(qvec) * invDet;
|
||||||
|
// if (v < 0 || u + v > 1) return false;
|
||||||
|
//
|
||||||
|
// t = v0v2.dotProduct(qvec) * invDet;
|
||||||
|
//
|
||||||
|
let v0 = tri.verts[0];
|
||||||
|
let v1 = tri.verts[1];
|
||||||
|
let v2 = tri.verts[2];
|
||||||
|
|
||||||
|
let v0v1 = v1 - v0;
|
||||||
|
let v0v2 = v2 - v0;
|
||||||
|
let p = cross(r.direction, v0v2);
|
||||||
|
let det = dot(v0v1, p);
|
||||||
|
if det < EPSILON {
|
||||||
|
return None;
|
||||||
|
}
|
||||||
|
|
||||||
|
let inv_det = 1. / det;
|
||||||
|
|
||||||
|
let t = r.origin - v0;
|
||||||
|
let u = dot(t, p) * inv_det;
|
||||||
|
if u < 0. || u > 1. {
|
||||||
|
return None;
|
||||||
|
}
|
||||||
|
|
||||||
|
let q = cross(t, v0v1);
|
||||||
|
let v = dot(r.direction, q) * inv_det;
|
||||||
|
if v < 0. || u + v > 1. {
|
||||||
|
return None;
|
||||||
|
}
|
||||||
|
let t = dot(v0v2, q) * inv_det;
|
||||||
|
|
||||||
|
if t > EPSILON {
|
||||||
|
return Some(RayTriangleResult {
|
||||||
|
t,
|
||||||
|
p: r.origin + r.direction * t,
|
||||||
|
});
|
||||||
|
}
|
||||||
|
None
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<M> Hit for Triangles<M>
|
||||||
|
where
|
||||||
|
M: Material,
|
||||||
|
{
|
||||||
|
fn hit(&self, r: Ray, _t_min: f32, _t_max: f32) -> Option<HitRecord> {
|
||||||
|
// TODO(wathiede): add an acceleration structure to more cheaply skip some triangles.
|
||||||
|
for tri in &self.triangles {
|
||||||
|
if let Some(RayTriangleResult { t, p }) = ray_triangle_intersect_moller_trumbore(r, tri)
|
||||||
|
{
|
||||||
|
//if let Some(RayTriangleResult { t, p }) = ray_triangle_intersect_geometric(r, tri) {
|
||||||
|
// We don't support UV (yet?).
|
||||||
|
let uv = (0.5, 0.5);
|
||||||
|
return Some(HitRecord {
|
||||||
|
t,
|
||||||
|
uv,
|
||||||
|
p,
|
||||||
|
normal: tri.normal,
|
||||||
|
material: &self.material,
|
||||||
|
});
|
||||||
|
}
|
||||||
|
}
|
||||||
|
None
|
||||||
|
}
|
||||||
|
|
||||||
|
fn bounding_box(&self, _t_min: f32, _t_max: f32) -> Option<AABB> {
|
||||||
|
Some(self.bbox)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
struct RayTriangleResult {
|
||||||
|
t: f32,
|
||||||
|
p: Vec3,
|
||||||
|
}
|
||||||
@@ -1,279 +1 @@
|
|||||||
use std::convert::From;
|
pub use vec3::*;
|
||||||
use std::fmt;
|
|
||||||
use std::num::ParseFloatError;
|
|
||||||
use std::ops::Add;
|
|
||||||
use std::ops::Div;
|
|
||||||
use std::ops::Index;
|
|
||||||
use std::ops::Mul;
|
|
||||||
use std::ops::Neg;
|
|
||||||
use std::ops::Sub;
|
|
||||||
use std::str;
|
|
||||||
|
|
||||||
use serde_derive::Deserialize;
|
|
||||||
use serde_derive::Serialize;
|
|
||||||
|
|
||||||
#[derive(Default, Debug, Deserialize, Serialize, PartialEq, Copy, Clone)]
|
|
||||||
pub struct Vec3 {
|
|
||||||
pub x: f32,
|
|
||||||
pub y: f32,
|
|
||||||
pub z: f32,
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn cross(v1: Vec3, v2: Vec3) -> Vec3 {
|
|
||||||
Vec3 {
|
|
||||||
x: v1.y * v2.z - v1.z * v2.y,
|
|
||||||
y: v1.z * v2.x - v1.x * v2.z,
|
|
||||||
z: v1.x * v2.y - v1.y * v2.x,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn dot(v1: Vec3, v2: Vec3) -> f32 {
|
|
||||||
v1.x * v2.x + v1.y * v2.y + v1.z * v2.z
|
|
||||||
}
|
|
||||||
|
|
||||||
impl Vec3 {
|
|
||||||
pub fn new(x: f32, y: f32, z: f32) -> Vec3 {
|
|
||||||
Vec3 { x, y, z }
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn min(self) -> f32 {
|
|
||||||
self.x.min(self.y).min(self.z)
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn max(self) -> f32 {
|
|
||||||
self.x.max(self.y).max(self.z)
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn length(self) -> f32 {
|
|
||||||
self.squared_length().sqrt()
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn squared_length(self) -> f32 {
|
|
||||||
self.x * self.x + self.y * self.y + self.z * self.z
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn unit_vector(self) -> Vec3 {
|
|
||||||
self / self.length()
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn make_unit_vector(&mut self) {
|
|
||||||
*self = self.unit_vector();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl From<[f32; 3]> for Vec3 {
|
|
||||||
fn from(v: [f32; 3]) -> Self {
|
|
||||||
Vec3 {
|
|
||||||
x: v[0],
|
|
||||||
y: v[1],
|
|
||||||
z: v[2],
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl fmt::Display for Vec3 {
|
|
||||||
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
|
|
||||||
write!(f, "{} {} {}", self.x, self.y, self.z)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl str::FromStr for Vec3 {
|
|
||||||
type Err = ParseFloatError;
|
|
||||||
fn from_str(s: &str) -> Result<Self, Self::Err> {
|
|
||||||
let coords: Vec<&str> = s.split(' ').collect();
|
|
||||||
Ok(Vec3 {
|
|
||||||
x: coords[0].parse::<f32>()?,
|
|
||||||
y: coords[1].parse::<f32>()?,
|
|
||||||
z: coords[2].parse::<f32>()?,
|
|
||||||
})
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl Add<f32> for Vec3 {
|
|
||||||
type Output = Vec3;
|
|
||||||
|
|
||||||
fn add(self, r: f32) -> Vec3 {
|
|
||||||
Vec3 {
|
|
||||||
x: self.x + r,
|
|
||||||
y: self.y + r,
|
|
||||||
z: self.z + r,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl Add for Vec3 {
|
|
||||||
type Output = Vec3;
|
|
||||||
|
|
||||||
fn add(self, r: Vec3) -> Vec3 {
|
|
||||||
Vec3 {
|
|
||||||
x: self.x + r.x,
|
|
||||||
y: self.y + r.y,
|
|
||||||
z: self.z + r.z,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl Div<Vec3> for f32 {
|
|
||||||
type Output = Vec3;
|
|
||||||
|
|
||||||
fn div(self, r: Vec3) -> Vec3 {
|
|
||||||
Vec3 {
|
|
||||||
x: self / r.x,
|
|
||||||
y: self / r.y,
|
|
||||||
z: self / r.z,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl Div<f32> for Vec3 {
|
|
||||||
type Output = Vec3;
|
|
||||||
|
|
||||||
fn div(self, r: f32) -> Vec3 {
|
|
||||||
Vec3 {
|
|
||||||
x: self.x / r,
|
|
||||||
y: self.y / r,
|
|
||||||
z: self.z / r,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl Index<usize> for Vec3 {
|
|
||||||
type Output = f32;
|
|
||||||
fn index(&self, idx: usize) -> &f32 {
|
|
||||||
match idx {
|
|
||||||
0 => &self.x,
|
|
||||||
1 => &self.y,
|
|
||||||
2 => &self.z,
|
|
||||||
_ => panic!("idx {} out of range for vec3", idx),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl Mul for Vec3 {
|
|
||||||
type Output = Vec3;
|
|
||||||
|
|
||||||
fn mul(self, r: Vec3) -> Vec3 {
|
|
||||||
Vec3 {
|
|
||||||
x: self.x * r.x,
|
|
||||||
y: self.y * r.y,
|
|
||||||
z: self.z * r.z,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl Mul<Vec3> for f32 {
|
|
||||||
type Output = Vec3;
|
|
||||||
|
|
||||||
fn mul(self, v: Vec3) -> Vec3 {
|
|
||||||
Vec3 {
|
|
||||||
x: v.x * self,
|
|
||||||
y: v.y * self,
|
|
||||||
z: v.z * self,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl Mul<f32> for Vec3 {
|
|
||||||
type Output = Vec3;
|
|
||||||
|
|
||||||
fn mul(self, r: f32) -> Vec3 {
|
|
||||||
Vec3 {
|
|
||||||
x: self.x * r,
|
|
||||||
y: self.y * r,
|
|
||||||
z: self.z * r,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl Neg for Vec3 {
|
|
||||||
type Output = Vec3;
|
|
||||||
|
|
||||||
fn neg(self) -> Vec3 {
|
|
||||||
-1. * self
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl Sub for Vec3 {
|
|
||||||
type Output = Vec3;
|
|
||||||
|
|
||||||
fn sub(self, r: Vec3) -> Vec3 {
|
|
||||||
Vec3 {
|
|
||||||
x: self.x - r.x,
|
|
||||||
y: self.y - r.y,
|
|
||||||
z: self.z - r.z,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[cfg(test)]
|
|
||||||
mod tests {
|
|
||||||
use std::f32::consts::PI;
|
|
||||||
use std::str::FromStr;
|
|
||||||
|
|
||||||
use super::*;
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn vec_add() {
|
|
||||||
let v0 = Vec3::new(1., 2., 3.);
|
|
||||||
let v1 = Vec3::new(1., 1., 1.);
|
|
||||||
assert_eq!(v0 + v1, Vec3::new(2., 3., 4.));
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn vec_div() {
|
|
||||||
let v0 = Vec3::new(1., 2., 4.);
|
|
||||||
assert_eq!(v0 / 2., Vec3::new(0.5, 1., 2.));
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn vec_mul() {
|
|
||||||
let v0 = Vec3::new(1., 2., 4.);
|
|
||||||
assert_eq!(v0 * 0.5, Vec3::new(0.5, 1., 2.));
|
|
||||||
assert_eq!(v0 * v0, Vec3::new(1., 4., 16.));
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn vec_sub() {
|
|
||||||
let v0 = Vec3::new(1., 2., 3.);
|
|
||||||
let v1 = Vec3::new(1., 1., 1.);
|
|
||||||
assert_eq!(v0 - v1, Vec3::new(0., 1., 2.));
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn vec_idx() {
|
|
||||||
let v0 = Vec3::new(1., 2., 3.);
|
|
||||||
assert_eq!(v0[2], 3.);
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn vec_display() {
|
|
||||||
let v0 = Vec3::new(1., 2., 3.);
|
|
||||||
assert_eq!(format!("{}", v0), "1 2 3".to_owned());
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn vec_from_str() {
|
|
||||||
assert_eq!(Vec3::from_str("1. 2. 3.").unwrap(), Vec3::new(1., 2., 3.));
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn vec_str_roundtrip() {
|
|
||||||
let v = Vec3::from_str("1 2 3").unwrap();
|
|
||||||
let s = format!("{}", v);
|
|
||||||
assert_eq!(v, Vec3::from_str(&s).unwrap());
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn vec_dot() {
|
|
||||||
let v0 = Vec3::new(1., 0., 0.);
|
|
||||||
let v1 = Vec3::new(-1., 0., 0.);
|
|
||||||
assert_eq!(dot(v0, v1), v0.length() * v1.length() * PI.cos());
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn vec_cross() {
|
|
||||||
let v0 = Vec3::new(1., 0., 0.);
|
|
||||||
let v1 = Vec3::new(0., 1., 0.);
|
|
||||||
assert_eq!(cross(v0, v1), Vec3::new(0., 0., 1.));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|||||||
BIN
rtiow/renderer/stls/cube.stl
Normal file
BIN
rtiow/renderer/stls/cube.stl
Normal file
Binary file not shown.
Some files were not shown because too many files have changed in this diff Show More
Reference in New Issue
Block a user