#ifndef _MOOF_OCTREE_HH_
#define _MOOF_OCTREE_HH_
+#include <algorithm>
#include <list>
#include <boost/shared_ptr.hpp>
+#include <stlplus/ntree.hpp>
+
#include <Moof/Aabb.hh>
#include <Moof/Drawable.hh>
#include <Moof/Entity.hh>
+#include <Moof/Frustum.hh>
+#include <Moof/Log.hh>
#include <Moof/Math.hh>
#include <Moof/Sphere.hh>
-#include <Moof/Tree.hh>
namespace Mf {
-struct OctreeNode : public Entity
+//class Octree;
+//typedef boost::shared_ptr<Octree> OctreeP;
+
+//class Octree::Node;
+//typedef stlplus::ntree<Octree::Node>::iterator OctreeNodeP;
+
+
+struct OctreeInsertable
{
- std::list<EntityPtr> objects;
+ virtual ~OctreeInsertable() {}
- OctreeNode()
- {
- aabb_.min = Vector3(-1.0, -1.0, -1.0);
- aabb_.max = Vector3(1.0, 1.0, 1.0);
- sphere_.init(Vector3(0.0, 0.0, 0.0), 1.41421);
- }
+ virtual bool isInsideAabb(const Aabb& aabb) const = 0;
+ virtual int getOctant(const Aabb& aabb) const = 0;
+};
- OctreeNode(const Aabb& aabb)
- {
- aabb_ = aabb;
- sphere_.point = aabb.getCenter();
- sphere_.radius = (aabb.min - sphere_.point).length();
- }
- void draw(Scalar alpha) const
+template <class T>
+class Octree : public Entity
+{
+ typedef boost::shared_ptr<T> InsertableP;
+
+ struct Node : public Entity
{
- std::list<EntityPtr>::const_iterator it;
+ std::list<InsertableP> objects;
- for (it = objects.begin(); it != objects.end(); ++it)
+ Aabb aabb;
+ Sphere sphere;
+
+ Node(const Aabb& box) :
+ aabb(box)
{
- (*it)->draw(alpha);
+ sphere.point = aabb.getCenter();
+ sphere.radius = (aabb.min - sphere.point).length();
}
- if (!objects.empty())
- aabb_.draw(); // temporary
- }
-
- void drawIfVisible(Scalar alpha, const Camera& cam) const
- {
- std::list<EntityPtr>::const_iterator it;
- for (it = objects.begin(); it != objects.end(); ++it)
+ void draw(Scalar alpha) const
{
- (*it)->drawIfVisible(alpha, cam);
- }
- if (!objects.empty())
- aabb_.draw();
- }
+ typename std::list<InsertableP>::const_iterator it;
+ for (it = objects.begin(); it != objects.end(); ++it)
+ {
+ (*it)->draw(alpha);
+ }
- bool isVisible(const Camera& cam) const
- {
- if (sphere_.isVisible(cam))
- {
- return aabb_.isVisible(cam);
+ if (!objects.empty())
+ aabb.draw(); // temporary
}
- return false;
- }
-};
+ void drawIfVisible(Scalar alpha, const Frustum& frustum) const
+ {
+ typename std::list<InsertableP>::const_iterator it;
+ for (it = objects.begin(); it != objects.end(); ++it)
+ {
+ (*it)->drawIfVisible(alpha, frustum);
+ }
-class Octree;
-typedef boost::shared_ptr<Octree> OctreePtr;
+ if (!objects.empty())
+ {
+ //aabb.draw();
+ //sphere.draw();
+ }
+ }
-class Octree : public Tree<OctreeNode>
-{
- Octree() {}
+ bool isVisible(const Frustum& frustum) const
+ {
+ if (sphere.isVisible(frustum))
+ {
+ return aabb.isVisible(frustum);
+ }
+
+ return false;
+ }
+ };
- explicit Octree(const OctreeNode& initNode) :
- Tree<OctreeNode>(initNode) {}
public:
- inline static OctreePtr createNewNode(const OctreeNode& item)
- {
- OctreePtr newNode = OctreePtr(new Octree(item));
- init(newNode);
- return newNode;
- }
+ typedef boost::shared_ptr<Octree> Ptr;
+ typedef typename stlplus::ntree<Node>::iterator NodeP;
+
+private:
- static Tree<OctreeNode>::WeakPtr add(Ptr node, EntityPtr entity)
+ NodeP insert(InsertableP entity, NodeP node)
{
- Plane::Halfspace halfspace;
- int octantNum = -1;
-
- Plane xy = node->node.getAabb().getPlaneXY();
- halfspace = xy.intersectsSphere(entity->getSphere());
+ ASSERT(node.valid() && "invalid node passed");
+ ASSERT(entity && "null entity passed");
- if (halfspace == Plane::POSITIVE)
+ if (entity->isInsideAabb(node->aabb))
{
- Plane xz = node->node.getAabb().getPlaneXZ();
- halfspace = xz.intersectsSphere(entity->getSphere());
-
- if (halfspace == Plane::POSITIVE)
- {
- Plane yz = node->node.getAabb().getPlaneYZ();
- halfspace = yz.intersectsSphere(entity->getSphere());
-
- if (halfspace == Plane::POSITIVE)
- {
- octantNum = 2;
- }
- else if (halfspace == Plane::NEGATIVE)
- {
- octantNum = 3;
- }
- }
- else if (halfspace == Plane::NEGATIVE)
- {
- Plane yz = node->node.getAabb().getPlaneYZ();
- halfspace = yz.intersectsSphere(entity->getSphere());
-
- if (halfspace == Plane::POSITIVE)
- {
- octantNum = 1;
- }
- else if (halfspace == Plane::NEGATIVE)
- {
- octantNum = 0;
- }
- }
+ return insert_recurse(entity, node);
}
- else if (halfspace == Plane::NEGATIVE)
+ else
{
- Plane xz = node->node.getAabb().getPlaneXZ();
- halfspace = xz.intersectsSphere(entity->getSphere());
-
- if (halfspace == Plane::POSITIVE)
- {
- Plane yz = node->node.getAabb().getPlaneYZ();
- halfspace = yz.intersectsSphere(entity->getSphere());
-
- if (halfspace == Plane::POSITIVE)
- {
- octantNum = 6;
- }
- else if (halfspace == Plane::NEGATIVE)
- {
- octantNum = 7;
- }
- }
- else if (halfspace == Plane::NEGATIVE)
- {
- Plane yz = node->node.getAabb().getPlaneYZ();
- halfspace = yz.intersectsSphere(entity->getSphere());
-
- if (halfspace == Plane::POSITIVE)
- {
- octantNum = 5;
- }
- else if (halfspace == Plane::NEGATIVE)
- {
- octantNum = 4;
- }
- }
+ node->objects.push_back(entity);
+ return node;
}
+ }
+
+ NodeP insert_recurse(InsertableP entity, NodeP node)
+ {
+ ASSERT(node.valid() && "invalid node passed");
+ ASSERT(entity && "null entity passed");
+ int octantNum = entity->getOctant(node->aabb);
if (octantNum == -1)
{
- node->node.objects.push_front(entity);
+ node->objects.push_back(entity);
return node;
}
else
{
- if (node->isLeaf())
+ if ((int)tree_.children(node) <= octantNum)
{
- addChildren(node);
+ addChild(node, octantNum);
}
- Ptr child = node->getChild(octantNum);
- if (child)
- {
- return add(child, entity);
- }
- else
- {
- std::cerr << "no child at index " << octantNum << std::endl;
- return Ptr();
- }
- //return WeakPtr();
+ NodeP child = tree_.child(node, octantNum);
+ ASSERT(child.valid() && "expected valid child node");
+
+ return insert(entity, child);
}
}
- static void addChildren(Ptr node)
+ void addChild(NodeP node, int index)
{
+ ASSERT(node.valid() && "invalid node passed");
+
Aabb octant;
- for (int i = 0; i < 8; ++i)
+ for (int i = tree_.children(node); i <= index; ++i)
{
- node->node.getAabb().getOctant(octant, i);
- //OctreeNode octantNode(octant);
-
- Ptr newChild = createNewNode(octant);
- node->addChild(newChild);
+ node->aabb.getOctant(octant, i);
+ tree_.append(node, octant);
}
}
- void draw(Ptr node, Scalar alpha)
+
+ void draw(Scalar alpha, NodeP node) const
{
- if (!node)
- {
- std::cerr << "null child :-(" << std::endl;
- return;
- }
+ ASSERT(node.valid() && "invalid node passed");
- node->node.draw(alpha);
+ node->draw(alpha);
- if (!node->isLeaf())
+ for (unsigned i = 0; i < tree_.children(node); ++i)
{
- Ptr firstChild = node->getFirstChild();
- Ptr temp = firstChild;
-
- if (!firstChild)
- {
- std::cerr << "node is not a leaf, but has no first child :-(" << std::endl;
- return;
- }
+ NodeP child = tree_.child(node, i);
+ ASSERT(child.valid() && "expected valid child node");
- do
- {
- draw(temp, alpha);
- temp = temp->getNextSibling();
- }
- while (temp && temp != firstChild);
+ draw(alpha, child);
}
}
- void drawIfVisible(Ptr node, Scalar alpha, const Camera& cam)
+ void drawIfVisible(Scalar alpha, const Frustum& frustum, NodeP node) const
{
- //node.drawIfVisible(alpha, cam);
-
- if (!node)
- {
- std::cerr << "null child :-(" << std::endl;
- return;
- }
+ ASSERT(node.valid() && "invalid node passed");
- Frustum::Collision collision =
- cam.getFrustum().containsSphere(node->node.getSphere());
+ // try to cull by sphere
+ Frustum::Collision collision = frustum.contains(node->sphere);
if (collision == Frustum::OUTSIDE) return;
- collision = cam.getFrustum().containsAabb(node->node.getAabb());
+ // try to cull by aabb
+ collision = frustum.contains(node->aabb);
if (collision == Frustum::OUTSIDE) return;
if (collision == Frustum::INSIDE)
{
- node->node.draw(alpha);
+ node->draw(alpha);
}
else // collision == Frustum::INTERSECT
{
- node->node.drawIfVisible(alpha, cam);
+ node->drawIfVisible(alpha, frustum);
}
- if (!node->isLeaf())
+ if (tree_.children(node) > 0)
{
- Ptr firstChild = node->getFirstChild();
- Ptr temp = firstChild;
-
- if (!firstChild)
- {
- std::cerr << "node is not a leaf, but has no first child :-(" << std::endl;
- return;
- }
-
if (collision == Frustum::INSIDE)
{
- do
+ for (unsigned i = 0; i < tree_.children(node); ++i)
{
- draw(temp, alpha);
- temp = temp->getNextSibling();
+ NodeP child = tree_.child(node, i);
+ ASSERT(child.valid() && "expected valid child node");
+
+ draw(alpha, child);
}
- while (temp && temp != firstChild);
}
else // collision == Frustum::INTERSECT
{
- do
+ for (unsigned i = 0; i < tree_.children(node); ++i)
{
- drawIfVisible(temp, alpha, cam);
- temp = temp->getNextSibling();
+ NodeP child = tree_.child(node, i);
+ ASSERT(child.valid() && "expected valid child node");
+
+ drawIfVisible(alpha, frustum, child);
}
- while (temp && temp != firstChild);
}
}
}
- void drawIfVisible(Scalar alpha, const Camera& cam)
+ mutable stlplus::ntree<Node> tree_;
+
+
+public:
+
+ void print(NodeP node)
{
- drawIfVisible(getThis(), alpha, cam);
+ //logDebug("-----");
+ //logDebug("depth to node: %d", tree_.depth(node));
+ //logDebug("size of node: %d", tree_.size(node));
}
-};
+ static Ptr alloc(const Node& rootNode)
+ {
+ return Ptr(new Octree(rootNode));
+ }
+
+ explicit Octree(const Node& rootNode)
+ {
+ tree_.insert(rootNode);
+ }
+
+ NodeP insert(InsertableP entity)
+ {
+ return insert(entity, tree_.root());
+ }
+ NodeP reinsert(InsertableP entity, NodeP node)
+ {
+ ASSERT(entity && "null entity passed");
+ ASSERT(node.valid() && "invalid node passed");
+
+ typename std::list<InsertableP>::iterator it;
+ it = std::find(node->objects.begin(), node->objects.end(), entity);
+
+ if (it != node->objects.end())
+ {
+ node->objects.erase(it);
+ }
+
+ return insert(entity);
+ }
+
+ void draw(Scalar alpha) const
+ {
+ draw(alpha, tree_.root());
+ }
+
+ void drawIfVisible(Scalar alpha, const Frustum& frustum) const
+ {
+ drawIfVisible(alpha, frustum, tree_.root());
+ }
+};
} // namespace Mf