CudaKDTree.hpp

Go to the documentation of this file.

#pragma once
#include "gpu/Buffer.hpp"
#include "io/Stream.hpp"
#include "kdtree/KDTree.hpp"
#include "CudaAS.hpp"
#include "kernels/CudaTracerKernels.hpp"

namespace FW
{
class CudaKDTree : public CudaAS
{
public:
    BVHLayout   getLayout           (void) const            { return BVHLayout::BVHLayout_Compact; }

    explicit    CudaKDTree          (const KDTree& kdtree);

                CudaKDTree          (CudaKDTree& other)         { operator=(other); }

    explicit    CudaKDTree          (InputStream& in);

                ~CudaKDTree         (void);

    Buffer&     getNodeBuffer       (void)                      { return m_nodes; }

    Buffer&     getTriIndexBuffer   (void)                      { return m_triIndex; }

    Buffer&     getTriWoopBuffer    (void)                      { return m_triWoop; }

    void        serialize           (OutputStream& out);

    const AABB& getBBox             (void) const                    { return m_bbox; }

private:
    struct StackEntry
    {
        const KDTreeNode*   node; 
        S32                 idx; 

        StackEntry(const KDTreeNode* n = NULL, int i = 0) : node(n), idx(i) {}

        int encodeIdx(void) const { return (node->isLeaf()) ? ~idx : idx; }
    };
    
private:

    void                createNodeTriIdx    (const KDTree& kdtree);

    void                createWoopTri       (const KDTree& kdtree);

    Buffer              m_nodes;        
    Buffer              m_triIndex;     
    Buffer              m_triWoop;      

    AABB                m_bbox;         
};

}