Add Viewport, cleanup Ray / main

Author: karnkaul

tray/CMakeLists.txt

@@ -9,6 +9,7 @@ target_sources(${PROJECT_NAME} PRIVATE
   src/tray/ray.hpp
   src/tray/rgb.hpp
   src/tray/vec.hpp
+  src/tray/viewport.hpp
 
   src/main.cpp
 )

tray/src/main.cpp

@@ -1,26 +1,32 @@
 #include <tray/image.hpp>
 #include <tray/io.hpp>
 #include <tray/ray.hpp>
-#include <tray/vec.hpp>
+#include <tray/viewport.hpp>
 #include <iostream>
 #include <span>
 
 using namespace tray;
 
 int main() {
 	static constexpr auto extent = uvec2{400U, 300U};
-	static constexpr auto origin = fvec3{0.0f, 0.0f, 0.0f};
-	static constexpr auto horizontal = fvec3(extent.x() / 100.0f, 0.0f, 0.0f);
-	static constexpr auto vertical = fvec3(0.0f, extent.y() / 100.0f, 0.0f);
-	static constexpr auto lower_left = origin - horizontal / 2.0f - vertical / 2.0f - fvec3{0.0f, 0.0f, 1.0f};
+	static constexpr auto viewport = Viewport::make(extent, 2.0f, 2.0f);
+	static constexpr auto origin = fvec3{};
+
+	static constexpr auto horizontal = fvec3{viewport.extent.x(), 0.0f, 0.0f};
+	static constexpr auto vertical = fvec3{0.0f, viewport.extent.y(), 0.0f};
+	static constexpr auto top_left = origin + 0.5f * (-horizontal + vertical) + fvec3{0.0f, 0.0f, -viewport.depth};
+
+	static constexpr fvec3 gradient[] = {Rgb::from_hex(0xffffff).to_f32(), Rgb::from_hex(0x002277).to_f32()};
 	auto image = Image{extent};
+
 	for (std::uint32_t row = 0; row < image.extent().y(); ++row) {
-		auto const yt = static_cast<float>(row) / static_cast<float>(image.extent().y());
+		auto const yt = static_cast<float>(row) / static_cast<float>(image.extent().y() - 1);
 		for (std::uint32_t col = 0; col < image.extent().x(); ++col) {
-			auto const xt = static_cast<float>(col) / static_cast<float>(image.extent().x());
-			auto const ray = Ray{origin, lower_left + xt * horizontal + yt * vertical - origin};
-			auto const t = 0.5f * (normalize(ray.direction.vec()).y() + 1.0f);
-			image[{row, col}] = Rgb::from_f32(Rgb::from_hex(0xff0000).to_f32() * t + (1.0f - t) * Rgb::from_hex(0xffffff).to_f32());
+			auto const xt = static_cast<float>(col) / static_cast<float>(image.extent().x() - 1);
+			auto const dir = top_left + xt * horizontal - yt * vertical - origin;
+			auto const ray = Ray{origin, dir};
+			auto const t = 0.5f * (ray.direction.vec().y() + 1.0f);
+			image[{row, col}] = Rgb::from_f32(lerp(gradient[0], gradient[1], t));
 		}
 	}
 

tray/src/tray/ray.cpp

@@ -0,0 +1,39 @@
+#include <tray/ray.hpp>
+#include <span>
+
+namespace tray {
+namespace {
+constexpr float smallet_positive_root(std::span<float const, 2> roots) {
+	if (roots[0] < 0.0f) { return roots[1]; }
+	if (roots[1] < 0.0f) { return roots[0]; }
+	return std::min(roots[0], roots[1]);
+}
+} // namespace
+
+bool Ray::hit(Sphere const& sphere) const {
+	auto const co = origin - sphere.centre;
+	float const b = 2.0f * dot(direction.vec(), co);
+	float const c = dot(co, co) - sphere.radius * sphere.radius;
+	float const discriminant = b * b - 4 * c;
+	return discriminant > 0.0f;
+}
+
+Hit Ray::hit2(Sphere const& sphere) const {
+	auto const co = origin - sphere.centre;
+	float const b = 2.0f * dot(direction.vec(), co);
+	float const c = dot(co, co) - sphere.radius * sphere.radius;
+	float const discriminant = b * b - 4 * c;
+	if (discriminant < 0.0f) { return {}; }
+
+	float const disc_sq = std::sqrt(discriminant);
+	float const roots[] = {0.5f * (-b - disc_sq), 0.5f * (-b + disc_sq)};
+	auto const t = smallet_positive_root(roots);
+	// auto const root = roots[1];
+	if (t < 0.0f) { return {}; }
+
+	auto const position = at(t);
+	auto const normal = position - sphere.centre;
+	// assert(normal.z() > 0.0f);
+	return {position, normal, t};
+}
+} // namespace tray

tray/src/tray/viewport.hpp

@@ -0,0 +1,14 @@
+#pragma once
+#include <tray/vec.hpp>
+
+namespace tray {
+struct Viewport {
+	fvec2 extent{1.0f};
+	float depth{1.0f};
+
+	static constexpr Viewport make(uvec2 extent, float depth, float scale = 1.0f) {
+		auto const ar = static_cast<float>(extent.x()) / static_cast<float>(extent.y());
+		return ar > 1.0f ? Viewport{fvec2{scale, scale / ar}, depth} : Viewport{fvec2{scale * ar, scale}, depth};
+	}
+};
+} // namespace tray