src/rcrt/gi/PKDNode.h

Go to the documentation of this file.

#ifndef PKDNODE_H_
#define PKDNODE_H_

#include "../primitives/AABB.h"
#include "Photon.h"
#include <list>
#include <vector>

namespace rcrt{

struct PKDSplitPlane{
        PKDSplitPlane(Photon* phot, Axis ax):p(phot),axis(ax)
        {}
        Photon* p;
        const Axis axis;
};

struct PKDSearchP{
        PKDSearchP(Photon* phot, const float& d):p(phot),distSqr(d)
        {}
        Photon* p;
        float distSqr;
        
        bool operator<(const PKDSearchP& pks) const
        {
                return distSqr < pks.distSqr;
        }
};

class PKDNode
{
private:
        PKDNode* children[2];
        Photon* photon;
        Axis splitAxis;
        
public:
        PKDNode(std::list<PKDSplitPlane*>* splits, const float& maxRadius, const AABB& bbox);
        ~PKDNode();
        
        bool isLeaf() const;
        
        void getKNearest(std::vector<PKDSearchP>* result, const unsigned int& k,
                                const Point3D& pos, const Vec3D& normal,
                                const float& maxRadius,const float& flat) const;
        
};

}//namespace

#endif /*PKDNODE_H_*/

---