#pragma once

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

#include "trace.h"
#include "bvh.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;
};

}