aabp_tree.h

Go to the documentation of this file.

/** @file
 *  @author Bram de Greve (bramz@users.sourceforge.net)
 *  @author Tom De Muer (tomdemuer@users.sourceforge.net)
 *
 *  *** BEGIN LICENSE INFORMATION ***
 *  
 *  The contents of this file are subject to the Common Public Attribution License 
 *  Version 1.0 (the "License"); you may not use this file except in compliance with 
 *  the License. You may obtain a copy of the License at 
 *  http://lass.sourceforge.net/cpal-license. The License is based on the 
 *  Mozilla Public License Version 1.1 but Sections 14 and 15 have been added to cover 
 *  use of software over a computer network and provide for limited attribution for 
 *  the Original Developer. In addition, Exhibit A has been modified to be consistent 
 *  with Exhibit B.
 *  
 *  Software distributed under the License is distributed on an "AS IS" basis, WITHOUT 
 *  WARRANTY OF ANY KIND, either express or implied. See the License for the specific 
 *  language governing rights and limitations under the License.
 *  
 *  The Original Code is LASS - Library of Assembled Shared Sources.
 *  
 *  The Initial Developer of the Original Code is Bram de Greve and Tom De Muer.
 *  The Original Developer is the Initial Developer.
 *  
 *  All portions of the code written by the Initial Developer are:
 *  Copyright (C) 2004-2007 the Initial Developer.
 *  All Rights Reserved.
 *  
 *  Contributor(s):
 *
 *  Alternatively, the contents of this file may be used under the terms of the 
 *  GNU General Public License Version 2 or later (the GPL), in which case the 
 *  provisions of GPL are applicable instead of those above.  If you wish to allow use
 *  of your version of this file only under the terms of the GPL and not to allow 
 *  others to use your version of this file under the CPAL, indicate your decision by 
 *  deleting the provisions above and replace them with the notice and other 
 *  provisions required by the GPL License. If you do not delete the provisions above,
 *  a recipient may use your version of this file under either the CPAL or the GPL.
 *  
 *  *** END LICENSE INFORMATION ***
 */



/** @class lass::spat::AabpTree
 *  @brief an AABB bounding volume tree that looks similar to a KdTree
 *  @author Bram de Greve [BdG]
 *
 *  the AabbTree does NOT own the objects.  You must keep them yourself!
 */

#ifndef LASS_GUARDIAN_OF_INCLUSION_SPAT_AABP_TREE_H
#define LASS_GUARDIAN_OF_INCLUSION_SPAT_AABP_TREE_H

//#define LASS_SPAT_AABB_TREE_DIAGNOSTICS

#include "spat_common.h"
#include "default_object_traits.h"
#include "split_heuristics.h"

namespace lass
{
namespace spat
{

template
<
    class ObjectType,
    class ObjectTraits = DefaultObjectTraits<ObjectType>,
    typename SplitHeuristics = DefaultSplitHeuristics<>
>
class AabpTree
{
public:

    typedef AabpTree<ObjectType, ObjectTraits, SplitHeuristics> TSelf;

    typedef ObjectType TObject;
    typedef ObjectTraits TObjectTraits;
    typedef SplitHeuristics TSplitHeuristics;

    typedef typename TObjectTraits::TObjectIterator TObjectIterator;
    typedef typename TObjectTraits::TObjectReference TObjectReference;
    typedef typename TObjectTraits::TAabb TAabb;
    typedef typename TObjectTraits::TRay TRay;
    typedef typename TObjectTraits::TPoint TPoint;
    typedef typename TObjectTraits::TVector TVector;
    typedef typename TObjectTraits::TValue TValue;
    typedef typename TObjectTraits::TParam TParam;
    typedef typename TObjectTraits::TReference TReference;
    typedef typename TObjectTraits::TConstReference TConstReference;
    typedef typename TObjectTraits::TInfo TInfo;

    enum { dimension = TObjectTraits::dimension };

    typedef std::vector<TObjectIterator> TObjectIterators;

    class Neighbour
    {
    public:
        Neighbour() {}
        Neighbour(TObjectIterator object, TValue squaredDistance): 
            object_(object), squaredDistance_(squaredDistance) {}
        TObjectIterator object() const { return object_; }
        TValue squaredDistance() const { return squaredDistance_; }
        TObjectIterator operator->() const { return object_; }
        TObjectReference operator*() const { return TObjectTraits::object(object_); }
        bool operator<(const Neighbour& other) const { return squaredDistance_ < other.squaredDistance_; }
    private:
        TObjectIterator object_;
        TValue squaredDistance_;
    };

    AabpTree();
    AabpTree(TObjectIterator first, TObjectIterator last);

    void reset();
    void reset(TObjectIterator first, TObjectIterator last);

    const TAabb& aabb() const;

    const bool contains(const TPoint& point, const TInfo* info = 0) const;
    template <typename OutputIterator> 
    OutputIterator find(const TPoint& point, OutputIterator result, const TInfo* info = 0) const;
    const TObjectIterator intersect(const TRay& ray, TReference t, TParam minT = 0, 
        const TInfo* info = 0) const;
    const bool intersects(const TRay& ray, TParam minT = 0, 
        TParam maxT = std::numeric_limits<TValue>::infinity(), const TInfo* info = 0) const;
    const Neighbour nearestNeighbour(const TPoint& point, const TInfo* info = 0) const;
    template <typename RandomIterator>
    RandomIterator rangeSearch(const TPoint& center, TParam maxRadius, size_t maxCount,
            RandomIterator first, const TInfo* info = 0) const; 

    void swap(TSelf& other);
    const bool isEmpty() const;
    const TObjectIterator end() const;

private:

    struct Input
    {
        TAabb aabb;
        TObjectIterator object;
        Input(const TAabb& aabb, TObjectIterator object): aabb(aabb), object(object) {}
    };
    typedef std::vector<Input> TInputs;
    typedef typename TInputs::iterator TInputIterator;

    class Node
    {
    public:
        Node(int axis)
        {
            LASS_ASSERT(axis >= 0 && axis < dimension);
            axis_ = axis;
        }
        Node(int first, int last)
        {
            LASS_ASSERT(first >= 0 && last > first);
            first_ = first;
            last_ = -last - 1;
            LASS_ASSERT(last_ < 0);
        }

        const bool isInternal() const { return axis_ >= 0; }
        TParam leftBound() const { LASS_ASSERT(isInternal()); return leftBound_; }
        TReference leftBound() { LASS_ASSERT(isInternal()); return leftBound_; }
        TParam rightBound() const { LASS_ASSERT(isInternal()); return rightBound_; }
        TReference rightBound() { LASS_ASSERT(isInternal()); return rightBound_; }
        const int axis() const { LASS_ASSERT(isInternal()); return axis_; }
        const int right() const { LASS_ASSERT(isInternal()); return right_; }
        int& right() { LASS_ASSERT(isInternal()); return right_; }

        const bool isLeaf() const { return last_ < 0; }
        const int first() const { LASS_ASSERT(isLeaf()); return first_; }
        const int last() const { LASS_ASSERT(isLeaf()); return -last_ - 1; }

    private:
        TValue leftBound_;  // internal
        TValue rightBound_; // internal
        union
        {
            int right_; // internal
            int first_; // leaf
        };
        union
        {
            int axis_; // internal
            int last_; // leaf
        };
    };
    typedef std::vector<Node> TNodes;

    struct BalanceResult
    {
        TAabb aabb;
        int index;
        BalanceResult(const TAabb& aabb, int index): aabb(aabb), index(index) {}
    };

    const BalanceResult balance(TInputIterator iFirst, TInputIterator iLast);
    const int addLeafNode(TInputIterator iFirst, TInputIterator iLast);
    const int addInternalNode(int iAxis);

    const bool doContains(int index, const TPoint& point, const TInfo* info) const;
    template <typename OutputIterator> 
    OutputIterator doFind(int index, const TPoint& point, 
        OutputIterator iResult, const TInfo* info) const;
    const TObjectIterator doIntersect(int index, const TRay& ray, TReference t, TParam tMin,
        const TInfo* info, const TVector& reciprocalDirection, TParam tNear, TParam tFar) const;
    const bool doIntersects(int index, const TRay& ray, TParam tMin, TParam tMax,
        const TInfo* info, const TVector& reciprocalDirection, TParam tNear, TParam tFar) const;
    void doNearestNeighbour(int index, const TPoint& point, const TInfo* info, Neighbour& best) const;
    template <typename RandomIterator>
    RandomIterator doRangeSearch(int index, const TPoint& point, TReference squaredRadius, 
        size_t maxCount, RandomIterator first, RandomIterator last, const TInfo* info) const;

    void getChildren(int index, const TPoint& target, int indices[2], TValue signedDistances[2]) const;

    TAabb aabb_;
    TObjectIterators objects_;
    TNodes nodes_;
    TObjectIterator end_;
};

}

}

#include "aabp_tree.inl"

#endif

// EOF