COSC-3P93-Project/Step 2/include/world.h

104 lines
3.7 KiB
C
Raw Normal View History

2022-10-16 19:24:37 -04:00
/*
* Created by Brett Terpstra 6920201 on 16/10/22.
* Copyright (c) 2022 Brett Terpstra. All Rights Reserved.
*/
#ifndef STEP_2_WORLD_H
#define STEP_2_WORLD_H
#include <util/std.h>
2022-10-17 00:29:34 -04:00
#include <math/vectors.h>
2022-10-17 19:16:10 -04:00
#include <util/models.h>
2022-10-18 00:44:49 -04:00
#include <math/bvh.h>
#include <types.h>
2022-10-17 19:16:10 -04:00
#include <utility>
2022-10-16 19:24:37 -04:00
namespace Raytracing {
2022-10-17 00:29:34 -04:00
class SphereObject : public Object {
private:
PRECISION_TYPE radius;
public:
2022-10-18 00:44:49 -04:00
SphereObject(const vec4& position, PRECISION_TYPE radius, Material* material): radius(radius), Object(material, position) {}
2022-10-17 00:29:34 -04:00
[[nodiscard]] virtual HitData checkIfHit(const Ray& ray, PRECISION_TYPE min, PRECISION_TYPE max) const;
};
2022-10-17 19:16:10 -04:00
class TriangleObject : public Object {
private:
Triangle theTriangle;
public:
2022-10-18 00:44:49 -04:00
TriangleObject(const vec4& position, Triangle theTriangle, Material* material): Object(material, position),
2022-10-17 19:16:10 -04:00
theTriangle(std::move(theTriangle)) {}
[[nodiscard]] virtual HitData checkIfHit(const Ray& ray, PRECISION_TYPE min, PRECISION_TYPE max) const;
};
2022-10-18 00:44:49 -04:00
class ModelObject : public Object {
private:
std::vector<Triangle> triangles;
public:
ModelObject(const vec4& position, ModelData data, Material* material): Object(material, position) {
triangles = data.toTriangles();
this->aabb = data.aabb;
}
[[nodiscard]] virtual HitData checkIfHit(const Ray& ray, PRECISION_TYPE min, PRECISION_TYPE max) const;
};
2022-10-17 19:16:10 -04:00
class DiffuseMaterial : public Material {
private:
public:
explicit DiffuseMaterial(const vec4& scatterColor): Material(scatterColor) {}
[[nodiscard]] virtual ScatterResults scatter(const Ray& ray, const HitData& hitData) const;
};
class MetalMaterial : public Material {
protected:
static inline vec4 reflect(const vec4& incomingVector, const vec4& normal) {
return incomingVector - 2 * vec4::dot(incomingVector, normal) * normal;
}
public:
explicit MetalMaterial(const vec4& metalColor): Material(metalColor) {}
[[nodiscard]] virtual ScatterResults scatter(const Ray& ray, const HitData& hitData) const;
};
class BrushedMetalMaterial : public MetalMaterial {
private:
PRECISION_TYPE fuzzyness;
public:
explicit BrushedMetalMaterial(const vec4& metalColor, PRECISION_TYPE fuzzyness): MetalMaterial(metalColor), fuzzyness(fuzzyness) {}
[[nodiscard]] virtual ScatterResults scatter(const Ray& ray, const HitData& hitData) const;
};
2022-10-17 00:29:34 -04:00
class World {
private:
// store all the objects in the world,
std::vector<Object*> objects;
/*TODO: create a kd-tree or bvh version to store the objects
* this way we can easily tell if a ray is near and object or not
* saving on computation
*/
2022-10-17 19:16:10 -04:00
std::unordered_map<std::string, Material*> materials;
2022-10-17 00:29:34 -04:00
public:
World() = default;
World(const World& world) = delete;
World(const World&& world) = delete;
inline void add(Object* object) { objects.push_back(object); }
2022-10-17 19:16:10 -04:00
inline void addMaterial(const std::string& materialName, Material* mat) { materials.insert({materialName, mat}); }
inline Material* getMaterial(const std::string& materialName) { return materials.at(materialName); }
[[nodiscard]] virtual std::pair<HitData, Object*> checkIfHit(const Ray& ray, PRECISION_TYPE min, PRECISION_TYPE max) const;
2022-10-17 00:29:34 -04:00
~World();
};
2022-10-16 19:24:37 -04:00
}
#endif //STEP_2_WORLD_H