#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/Log.hh>
#include <Moof/Math.hh>
-#include <Moof/Tree.hh>
+#include <Moof/Sphere.hh>
namespace Mf {
-class Entity;
+class Camera;
-class Octree
+struct OctreeNode;
+typedef stlplus::ntree<OctreeNode>::iterator OctreeNodeP;
+
+class Octree;
+typedef boost::shared_ptr<Octree> OctreeP;
+
+
+struct OctreeNode : public Entity
{
-public:
+ std::list<EntityP> objects;
+
+ 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);
+ }
+
+ OctreeNode(const Aabb& aabb)
+ {
+ aabb_ = aabb;
+ sphere_.point = aabb.getCenter();
+ sphere_.radius = (aabb.min - sphere_.point).length();
+ }
+
+ void draw(Scalar alpha) const
+ {
+ 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<EntityP>::const_iterator it;
+
+ for (it = objects.begin(); it != objects.end(); ++it)
+ {
+ (*it)->drawIfVisible(alpha, cam);
+ }
+
+ if (!objects.empty())
+ aabb_.draw();
+ }
+
- class Node
+ bool isVisible(const Camera& cam) const
{
- Aabb aabb_;
- Vector3 center_;
+ if (sphere_.isVisible(cam))
+ {
+ return aabb_.isVisible(cam);
+ }
- std::list<boost::shared_ptr<Entity> > objects_;
+ return false;
+ }
- public:
- Node() :
- aabb_(-1.0, -1.0, -1.0, 1.0, 1.0, 1.0),
- center_(0.0, 0.0, 0.0) {}
+ static bool compareZOrder(EntityP a, EntityP b)
+ {
+ return a->getSphere().point[2] < b->getSphere().point[2];
+ }
+
+ void sort()
+ {
+ //std::sort(objects.begin(), objects.end(), compareZOrder);
+ objects.sort(compareZOrder);
+ }
+};
+
- Node(const Aabb& aabb) :
- aabb_(aabb),
- center_(aabb.getCenter()) {}
- };
+class Octree
+{
+ OctreeNodeP insert(EntityP entity, OctreeNodeP node);
+
+ void addChild(OctreeNodeP node, int index);
- Octree() :
- root_(new Tree<Node>()) {}
+ void draw(Scalar alpha, OctreeNodeP node);
+ void drawIfVisible(Scalar alpha, const Camera& cam, OctreeNodeP node);
- Octree(const Aabb& aabb) :
- root_(new Tree<Node>(Node(aabb))) {}
+ stlplus::ntree<OctreeNode> tree_;
+public:
- Tree<Node>::WeakPtr add(EntityPtr object);
+ inline void print(OctreeNodeP node)
+ {
+ logDebug("-----");
+ logDebug("depth to node: %d", tree_.depth(node));
+ logDebug("size of node: %d", tree_.size(node));
+ }
+
+ inline static OctreeP alloc(const OctreeNode& rootNode)
+ {
+ return OctreeP(new Octree(rootNode));
+ }
+
+ explicit Octree(const OctreeNode& rootNode)
+ {
+ tree_.insert(rootNode);
+ }
+
+ OctreeNodeP insert(EntityP entity)
+ {
+ return insert(entity, tree_.root());
+ }
+
+ OctreeNodeP reinsert(EntityP entity, OctreeNodeP node);
+
+ void drawIfVisible(Scalar alpha, const Camera& cam)
+ {
+ drawIfVisible(alpha, cam, tree_.root());
+ }
-private:
- Tree<Node>::Ptr root_;
+ void sort();
};