ntl_bsptree.h

Go to the documentation of this file.

/******************************************************************************
 *
 * El'Beem - Free Surface Fluid Simulation with the Lattice Boltzmann Method
 * Copyright 2003-2006 Nils Thuerey
 *
 * Tree container for fast triangle intersects
 *
 *****************************************************************************/
#ifndef NTL_TREE_H
#define NTL_TREE_H

#include "ntl_vector3dim.h"
#include "ntl_ray.h"


#define AXIS_X 0
#define AXIS_Y 1
#define AXIS_Z 2

#define BSP_STACK_SIZE 50


//  detailed definition unnecesseary here
class BSPNode;
class BSPStackElement;
class BSPStack;
class TriangleBBox;
class ntlScene;
class ntlTriangle;


class ntlTree
{
    public:

        ntlTree();
        ntlTree(int depth, int objnum, ntlScene *scene, int triFlagMask);
        ~ntlTree();

        void subdivide(BSPNode *node, int depth, int axis);

        void intersect(const ntlRay &ray, gfxReal &distance, ntlVec3Gfx &normal, ntlTriangle *&tri, int flags, bool forceNonsmooth) const;
        void intersectX(const ntlRay &ray, gfxReal &distance, ntlVec3Gfx &normal, ntlTriangle *&tri, int flags, bool forceNonsmooth) const;

        int getCurrentNodes( void ) { return mCurrentNodes; }

    protected:

        // check if a triangle is in a node
        bool checkAABBTriangle(ntlVec3Gfx &min, ntlVec3Gfx &max, ntlTriangle *tri);


        // VARs

        gfxReal distanceToPlane(BSPNode *curr, ntlVec3Gfx plane, ntlRay ray) const;

        void getChildren(BSPNode *curr, ntlVec3Gfx origin, BSPNode *&node_near, BSPNode *&node_far) const;

        void deleteNode(BSPNode *curr);

        //inline bool isLeaf(BSPNode *node) const { return (node->child[0] == NULL); }


        ntlVec3Gfx mStart,mEnd;

        int mMaxDepth;

        int mMaxListLength;

        BSPNode *mpRoot;
        int mNumNodes;
        int mAbortSubdiv;

        BSPStack *mpNodeStack;
        //stack<BSPNode *> nodestack;

        vector<ntlVec3Gfx> *mpVertices;

        vector<ntlVec3Gfx> *mpVertNormals;

        vector<ntlTriangle> *mpTriangles;
        vector<ntlTriangle *> *mppTriangles;

        char *mpTriDist;

        TriangleBBox *mpTBB;

        int mTriangleMask;

        int mCurrentDepth, mCurrentNodes;

        int mTriDoubles; 

};


#endif