F:/My Projects/LSEngine/Modules/LSPhysicsLib/Src/SpatialPartitions/LSPQuadTree.h

#ifndef __LSP_QUADTREE_H__
#define __LSP_QUADTREE_H__

#include "../LSPPhysicsLib.h"
#include "LSPQuadTreeNode.h"
#include "Vector/LSTLVector.h"

namespace lsp {

        class CQuadTree {
        public :
                // == Various constructors.
                LSE_CALLCTOR                                                    CQuadTree();
                LSE_CALLCTOR                                                    ~CQuadTree();


                // == Functions.
                LSBOOL LSE_CALL                                                 CreateQuadTree( LSREAL _fSize );

                LSVOID LSE_CALL                                                 Reset();

                const CQuadTreeNode * LSE_CALL                  Root() const;

                LSBOOL LSE_CALL                                                 InsertObject( CQuadTreeObject * _pqtoObject );

                LSBOOL LSE_CALL                                                 FindObjects( const CAabb2d &_a2Box, CVector<CQuadTreeObject *, LSUINT32> &_vReturn );


        protected :
                // == Members.
                LSUINT8 *                                                               m_pui8Nodes;

                CQuadTreeNode *                                                 m_pqtnLevels[8];

                LSREAL                                                                  m_fRadius;

                LSREAL                                                                  m_fInvRadius;


                // == Functions.
                LSE_INLINE CQuadTreeNode * LSE_CALL             Root();

                static LSE_INLINE LSBOOL LSE_CALL               NodeIsInBounds( const CQuadTreeNode * _pqtnNode, const CAabb2d &_a2Box );

                static LSBOOL LSE_CALL                                  GatherObjectsInNode( const CQuadTreeNode * _pqtnNode, const CAabb2d &_a2Box, CVector<CQuadTreeObject *, LSUINT32> &_vReturn );
        };


        // * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *
        // DEFINITIONS
        // * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *
        // == Functions.
        LSE_INLINE const CQuadTreeNode * LSE_CALL CQuadTree::Root() const { return reinterpret_cast<CQuadTreeNode *>(m_pui8Nodes); }

        LSE_INLINE CQuadTreeNode * LSE_CALL CQuadTree::Root() { return reinterpret_cast<CQuadTreeNode *>(m_pui8Nodes); }

        LSE_INLINE LSBOOL LSE_CALL CQuadTree::NodeIsInBounds( const CQuadTreeNode * _pqtnNode, const CAabb2d &_a2Box ) {
                LSREAL fRad = _pqtnNode->GetRadius();
                // TODO: Try to vectorize this.
                return (_pqtnNode->GetCenter().x - fRad <= _a2Box.m_vMax.x) && (_pqtnNode->GetCenter().y - fRad <= _a2Box.m_vMax.y) &&
                        (_pqtnNode->GetCenter().x + fRad >= _a2Box.m_vMin.x) && (_pqtnNode->GetCenter().y + fRad >= _a2Box.m_vMin.y);
        }

}       // namespace lsp

#endif  // __LSP_QUADTREE_H__