mirror of
https://github.com/vonhyou/ray-tracer-comp371.git
synced 2025-06-07 23:12:00 +00:00
77 lines
2.4 KiB
C++
77 lines
2.4 KiB
C++
#include "Light.h"
|
|
#include "Scene.h"
|
|
|
|
#include <algorithm>
|
|
#include <cmath>
|
|
|
|
void Light::setTransform(const Matrix4f &transform) {
|
|
this->transform = transform;
|
|
}
|
|
|
|
void Light::setGridSize(unsigned int gridSize) { this->gridSize = gridSize; }
|
|
|
|
void Light::setUseCenter(bool useCenter) { this->useCenter = useCenter; }
|
|
|
|
void Light::setIsUse(bool isUse) { this->use = isUse; }
|
|
|
|
Vector3f Light::id() const { return id_; }
|
|
|
|
Vector3f Light::is() const { return is_; }
|
|
|
|
bool Light::isUse() const { return use; }
|
|
|
|
Vector3f PointLight::getCenter() const { return center; }
|
|
|
|
bool lightOnSurface(Vector3f center, Geometry *g) {
|
|
return (g->sample() - center).dot(g->normal(center)) < 1e-5;
|
|
}
|
|
|
|
Vector3f PointLight::illumination(const HitRecord &hit,
|
|
const vector<Geometry *> &geometries) const {
|
|
Vector3f shadingPoint = hit.point();
|
|
Vector3f rayDirection = (center - shadingPoint).normalized();
|
|
Geometry *geometry = hit.geometry();
|
|
Ray shadowRay(shadingPoint, rayDirection);
|
|
|
|
for (auto g : geometries)
|
|
if (g != geometry && g->intersect(shadowRay).hasValue())
|
|
if (g->type() == Geometry::Type::SPHERE || !lightOnSurface(center, g))
|
|
return Vector3f::Zero();
|
|
|
|
Vector3f ambient_ =
|
|
geometry->ka() * geometry->ca().array() * Scene::current->ai().array();
|
|
|
|
Vector3f diffuse_ = geometry->kd() * geometry->cd().array() * id_.array() *
|
|
std::max(0.0f, hit.normal().dot(rayDirection));
|
|
|
|
Vector3f halfWay = (hit.viewDirection() + rayDirection).normalized();
|
|
Vector3f specular_ =
|
|
geometry->ks() * geometry->cs().array() * is_.array() *
|
|
pow(std::max(0.0f, hit.normal().dot(halfWay)), geometry->phong());
|
|
|
|
return specular_ + ambient_ + diffuse_;
|
|
}
|
|
|
|
Vector3f AreaLight::getCenter() const {
|
|
return p1 + (p4 - p1) / 2 + (p2 - p1) / 2;
|
|
}
|
|
|
|
Vector3f AreaLight::illumination(const HitRecord &hit,
|
|
const vector<Geometry *> &geometries) const {
|
|
Vector3f u = p4 - p1;
|
|
Vector3f v = p2 - p1;
|
|
|
|
if (useCenter) {
|
|
return PointLight(*this, getCenter()).illumination(hit, geometries);
|
|
} else {
|
|
Vector3f color = Vector3f::Zero();
|
|
for (int y = 0; y < gridSize; ++y)
|
|
for (int x = 0; x < gridSize; ++x) {
|
|
Vector3f contribution =
|
|
PointLight(*this, p1 + (u * x + v * y) / gridSize)
|
|
.illumination(hit, geometries);
|
|
color += contribution;
|
|
}
|
|
return color / gridSize / gridSize;
|
|
}
|
|
}
|