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