mirror of
https://github.com/vonhyou/ray-tracer-comp371.git
synced 2025-06-08 07:22:01 +00:00
86 lines
2.5 KiB
C++
86 lines
2.5 KiB
C++
#include "RayTracer.h"
|
|
#include "HitRecord.h"
|
|
#include "Output.h"
|
|
#include "Parser.h"
|
|
#include "Ray.h"
|
|
|
|
#include <Eigen/Core>
|
|
#include <cmath>
|
|
#include <queue>
|
|
|
|
using std::priority_queue;
|
|
|
|
void RayTracer::parse() {
|
|
for (auto i = json["output"].begin(); i != json["output"].end(); ++i)
|
|
scenes.push_back(Parser::getScene(*i));
|
|
|
|
for (auto i = json["geometry"].begin(); i != json["geometry"].end(); ++i)
|
|
geometries.push_back(Parser::getGeometry(*i));
|
|
|
|
for (auto i = json["light"].begin(); i != json["light"].end(); ++i)
|
|
lights.push_back(Parser::getLight(*i));
|
|
}
|
|
|
|
Ray getRay(int x, int y, const Vector3f &camPos, const Vector3f &pxUpperLeft,
|
|
const Vector3f &du, const Vector3f &dv) {
|
|
return Ray(camPos, pxUpperLeft + x * du + y * dv - camPos);
|
|
}
|
|
|
|
void RayTracer::calculateColor(const HitRecord &hit, int i) {
|
|
Vector3f result(0, 0, 0);
|
|
for (auto light : lights)
|
|
result += light->isUse() ? light->illumination(hit, geometries)
|
|
: Vector3f::Zero();
|
|
|
|
result = result.cwiseMax(0.0f).cwiseMin(1.0f);
|
|
Output::current->r(i, result.x());
|
|
Output::current->g(i, result.y());
|
|
Output::current->b(i, result.z());
|
|
}
|
|
|
|
void RayTracer::render() {
|
|
int width = Scene::current->width();
|
|
int height = Scene::current->height();
|
|
Vector3f cameraPos = Scene::current->center();
|
|
Vector3f lookAt = Scene::current->lookAt();
|
|
float vpHeight =
|
|
2 * tan(Scene::current->fov() / 180 * M_PI / 2) * lookAt.norm();
|
|
float vpWidth = vpHeight * width / height;
|
|
Vector3f u = Vector3f(vpWidth, 0, 0);
|
|
Vector3f v = Vector3f(0, -vpHeight, 0);
|
|
Vector3f du = u / width;
|
|
Vector3f dv = v / height;
|
|
|
|
Vector3f vpUpperLeft = cameraPos + lookAt - u / 2.0 - v / 2.0;
|
|
Vector3f pxUpperLeft = vpUpperLeft + (du + dv) / 2.0;
|
|
|
|
Output::current = new Output(Scene::current->backgroundColor(),
|
|
Scene::current->name(), width, height);
|
|
|
|
for (int y = 0; y < height; ++y)
|
|
for (int x = 0; x < width; ++x) {
|
|
Ray ray = getRay(x, y, cameraPos, pxUpperLeft, du, dv);
|
|
priority_queue<HitRecord> records;
|
|
for (auto g : geometries) {
|
|
Optional<float> t = g->intersect(ray);
|
|
if (t.hasValue())
|
|
records.push(HitRecord(t.value(), ray, g));
|
|
}
|
|
|
|
if (!records.empty()) {
|
|
HitRecord hit = records.top();
|
|
hit.calcNormal();
|
|
calculateColor(hit, y * width + x);
|
|
}
|
|
}
|
|
}
|
|
|
|
void RayTracer::run() {
|
|
parse();
|
|
|
|
for (auto scene : scenes) {
|
|
Scene::current = scene;
|
|
render();
|
|
Output::current->write();
|
|
}
|
|
}
|