src/rcrt/gi/PhotonKDTree.h

Go to the documentation of this file.

#ifndef PHOTONKDTREE_H_
#define PHOTONKDTREE_H_

#include "../primitives/AABB.h"
#include "PKDNode.h"

namespace rcrt
{

class PhotonKDTree
{
private:
        std::vector<Photon*>* photons;
        float maxSqrRadius;
        PKDNode* root;
        bool empty;
        
        static bool splitPlaneCompare(PKDSplitPlane* a, PKDSplitPlane* b);
        
public:
        PhotonKDTree(const float& maxSqrRadius);
        virtual ~PhotonKDTree();
        
        void buildTree(std::vector<Photon*>* phots);
        
        void getKNearest(std::vector<PKDSearchP>* result, const unsigned int& k,
                        const Point3D& pos, const Vec3D& normal, const float& radiusSqr,
                        const float& flat) const;
};

}

#endif /*PHOTONKDTREE_H_*/

---