Octree.h

/*
 * This file is part of Bubba-3D.
 *
 * Bubba-3D is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * Bubba-3D is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with Bubba-3D. If not, see http://www.gnu.org/licenses/.
 */
#pragma once

#include <functional>
#include <linmath/float3.h>
#include <vector>
#include <set>
#include <ctime>
#include <math.h>
#include "AABB2.h"

class Triangle;

const int MAX_DEPTH = 8;
const int MAX_CHILDREN = 16;

class Octree {
public:
    Octree();

    Octree(chag::float3 origin, chag::float3 halfVector);

    ~Octree();

    void insertTriangle(Triangle* t);
    void insertAll(std::vector<Triangle*> &triangles);

    bool hasChildren();
    void getChildren(std::vector<Octree*>* octs);
    void clearChildren();

    int getOctantContainingPoint(const chag::float3& point);
    int getTriangleCount();
    int getTriangleCountRecursively();
    std::vector<Triangle*> *getTriangles();

    int getNumberOfSubTrees();


    void getTrianglesInsersectedByRayCast(chag::float3 rayOrigin, chag::float3 rayVector,
                                          std::vector<Triangle*> *triangleList);

    AABB* getAABB();

private:
    Octree(chag::float3 origin, chag::float3 halfVector, int depth);

    void setupAABB(chag::float3 origin, chag::float3 halfVector);

    void createChildren();

    chag::float3 combineTwoPointsByComparator(chag::float3 p1, chag::float3 p2,
                                              std::function<bool(float, float)> comparator);

    bool rayCastIntersectsAABB(chag::float3 rayOrigin, chag::float3 rayVector);
    void putTrianglesToList(std::vector<Triangle *> *triangleList);

    chag::float3 origin;
    chag::float3 halfVector;

    Octree *children[8];
    std::vector<Triangle*> ts;
    int depth;


    AABB aabb;
};