#pragma once

#include "../lib/mathlib.h"
#include "../platform/gl.h"

#include "bvh.h"
#include "trace.h"

namespace PT {

struct Tri_Mesh_Vert {
    Vec3 position;
    Vec3 normal;
};

class Triangle {
public:
    BBox bbox() const;
    Trace hit(const Ray &ray) const;

    size_t visualize(GL::Lines &, GL::Lines &, size_t, const Mat4 &) const { return size_t(0); }

private:
    Triangle(Tri_Mesh_Vert *verts, unsigned int v0, unsigned int v1, unsigned int v2);

    unsigned int v0, v1, v2;
    Tri_Mesh_Vert *vertex_list;
    friend class Tri_Mesh;
};

class Tri_Mesh {
public:
    Tri_Mesh() = default;
    Tri_Mesh(const GL::Mesh &mesh);

    BBox bbox() const;
    Trace hit(const Ray &ray) const;

    size_t visualize(GL::Lines &lines, GL::Lines &active, size_t level, const Mat4 &trans) const;

    void build(const GL::Mesh &mesh);

private:
    std::vector<Tri_Mesh_Vert> verts;
    BVH<Triangle> triangles;
};

} // namespace PT