#ifndef _MOOF_OCTREE_HH_
#define _MOOF_OCTREE_HH_
+#include <algorithm>
#include <list>
#include <boost/shared_ptr.hpp>
#include <Moof/Aabb.hh>
#include <Moof/Drawable.hh>
#include <Moof/Entity.hh>
+#include <Moof/Log.hh>
#include <Moof/Math.hh>
#include <Moof/Sphere.hh>
namespace Mf {
+class Camera;
+
+
+struct OctreeNode;
+typedef stlplus::ntree<OctreeNode>::iterator OctreeNodeP;
+
+class Octree;
+typedef boost::shared_ptr<Octree> OctreeP;
+
+
struct OctreeNode : public Entity
{
- std::list<EntityPtr> objects;
+ std::list<EntityP> objects;
OctreeNode()
{
void draw(Scalar alpha) const
{
- std::list<EntityPtr>::const_iterator it;
+ std::list<EntityP>::const_iterator it;
for (it = objects.begin(); it != objects.end(); ++it)
{
(*it)->draw(alpha);
}
+
if (!objects.empty())
aabb_.draw(); // temporary
}
void drawIfVisible(Scalar alpha, const Camera& cam) const
{
- std::list<EntityPtr>::const_iterator it;
+ std::list<EntityP>::const_iterator it;
for (it = objects.begin(); it != objects.end(); ++it)
{
(*it)->drawIfVisible(alpha, cam);
}
+
if (!objects.empty())
aabb_.draw();
}
return false;
}
-};
-class Octree;
-typedef boost::shared_ptr<Octree> OctreePtr;
-
-class Octree
-{
-
- stlplus::ntree<OctreeNode> root_;
-
-public:
-
- explicit Octree(const OctreeNode& rootNode)
+ static bool compareZOrder(EntityP a, EntityP b)
{
- root_.insert(rootNode);
+ return a->getSphere().point[2] < b->getSphere().point[2];
}
- void insert(EntityPtr entity)
+ void sort()
{
- insert(root_.root(), entity);
+ //std::sort(objects.begin(), objects.end(), compareZOrder);
+ objects.sort(compareZOrder);
}
+};
- void insert(stlplus::ntree<OctreeNode>::iterator node, EntityPtr entity)
- {
- Plane::Halfspace halfspace;
- int octantNum = -1;
- if (!node.valid())
- {
- std::cerr << "cannot insert into invalid node" << std::endl;
- return;
- }
+class Octree
+{
+ OctreeNodeP insert(EntityP entity, OctreeNodeP node);
+
+ void addChild(OctreeNodeP node, int index);
- Plane xy = node->getAabb().getPlaneXY();
- halfspace = xy.intersectsSphere(entity->getSphere());
+ void draw(Scalar alpha, OctreeNodeP node);
+ void drawIfVisible(Scalar alpha, const Camera& cam, OctreeNodeP node);
- if (halfspace == Plane::POSITIVE)
- {
- Plane xz = node->getAabb().getPlaneXZ();
- halfspace = xz.intersectsSphere(entity->getSphere());
-
- if (halfspace == Plane::POSITIVE)
- {
- Plane yz = 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->getAabb().getPlaneYZ();
- halfspace = yz.intersectsSphere(entity->getSphere());
-
- if (halfspace == Plane::POSITIVE)
- {
- octantNum = 1;
- }
- else if (halfspace == Plane::NEGATIVE)
- {
- octantNum = 0;
- }
- }
- }
- else if (halfspace == Plane::NEGATIVE)
- {
- Plane xz = node->getAabb().getPlaneXZ();
- halfspace = xz.intersectsSphere(entity->getSphere());
-
- if (halfspace == Plane::POSITIVE)
- {
- Plane yz = 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->getAabb().getPlaneYZ();
- halfspace = yz.intersectsSphere(entity->getSphere());
-
- if (halfspace == Plane::POSITIVE)
- {
- octantNum = 5;
- }
- else if (halfspace == Plane::NEGATIVE)
- {
- octantNum = 4;
- }
- }
- }
+ stlplus::ntree<OctreeNode> tree_;
- if (octantNum == -1)
- {
- node->objects.push_front(entity);
- //return node;
- }
- else
- {
- if (root_.children(node) == 0)
- {
- addChildren(node);
- }
-
- stlplus::ntree<OctreeNode>::iterator child = root_.child(node, octantNum);
-
- if (child.valid())
- {
- return insert(child, entity);
- }
- else
- {
- std::cerr << "expected but found no child at index " << octantNum << std::endl;
- //return stlplus::ntree<OctreeNode>::iterator();
- }
- //return WeakPtr();
- }
- }
+public:
- void addChildren(stlplus::ntree<OctreeNode>::iterator node)
+ inline void print(OctreeNodeP node)
{
- Aabb octant;
-
- if (!node.valid())
- {
- std::cerr << "cannot add children to invalid node" << std::endl;
- return;
- }
-
- for (int i = 0; i < 8; ++i)
- {
- node->getAabb().getOctant(octant, i);
- //OctreeNode octantNode(octant);
-
- root_.append(node, octant);
- }
+ logDebug("-----");
+ logDebug("depth to node: %d", tree_.depth(node));
+ logDebug("size of node: %d", tree_.size(node));
}
- void draw(stlplus::ntree<OctreeNode>::iterator node, Scalar alpha)
+ inline static OctreeP alloc(const OctreeNode& rootNode)
{
- if (!node.valid())
- {
- std::cerr << "cannot draw null child node :-(" << std::endl;
- return;
- }
-
- node->draw(alpha);
-
- for (unsigned i = 0; i < root_.children(node); ++i)
- {
- stlplus::ntree<OctreeNode>::iterator child = root_.child(node, i);
-
- if (child.valid())
- {
- draw(child, alpha);
- }
- else
- {
- std::cerr << "node is not a leaf, but has an invalid child" << std::endl;
- }
-
- }
+ return OctreeP(new Octree(rootNode));
}
- void drawIfVisible(stlplus::ntree<OctreeNode>::iterator node,
- Scalar alpha, const Camera& cam)
+ explicit Octree(const OctreeNode& rootNode)
{
- //node.drawIfVisible(alpha, cam);
-
- if (!node.valid())
- {
- std::cerr << "invalid child while drawing :-(" << std::endl;
- return;
- }
-
- Frustum::Collision collision =
- cam.getFrustum().containsSphere(node->getSphere());
- if (collision == Frustum::OUTSIDE) return;
-
- collision = cam.getFrustum().containsAabb(node->getAabb());
- if (collision == Frustum::OUTSIDE) return;
-
-
- if (collision == Frustum::INSIDE)
- {
- node->draw(alpha);
- }
- else // collision == Frustum::INTERSECT
- {
- node->drawIfVisible(alpha, cam);
- }
+ tree_.insert(rootNode);
+ }
- if (root_.children(node) > 0)
- {
- if (collision == Frustum::INSIDE)
- {
- for (unsigned i = 0; i < root_.children(node); ++i)
- {
- stlplus::ntree<OctreeNode>::iterator child = root_.child(node, i);
-
- if (child.valid())
- {
- draw(child, alpha);
- }
- else
- {
- std::cerr << "node is not a leaf, but has an invalid child" << std::endl;
- }
-
- }
- }
- else // collision == Frustum::INTERSECT
- {
- for (unsigned i = 0; i < root_.children(node); ++i)
- {
- stlplus::ntree<OctreeNode>::iterator child = root_.child(node, i);
-
- if (child.valid())
- {
- drawIfVisible(child, alpha, cam);
- }
- else
- {
- std::cerr << "node is not a leaf, but has an invalid child" << std::endl;
- }
- }
- }
- }
+ OctreeNodeP insert(EntityP entity)
+ {
+ return insert(entity, tree_.root());
}
+ OctreeNodeP reinsert(EntityP entity, OctreeNodeP node);
+
void drawIfVisible(Scalar alpha, const Camera& cam)
{
- drawIfVisible(root_.root(), alpha, cam);
+ drawIfVisible(alpha, cam, tree_.root());
}
+ void sort();
};
-
} // namespace Mf
#endif // _MOOF_OCTREE_HH_