X-Git-Url: https://git.dogcows.com/gitweb?p=chaz%2Fyoink;a=blobdiff_plain;f=src%2FMoof%2FOctree.hh;h=3614606fd911882dc96195cb28cedd3f83282120;hp=2ee617f3744f0baa6155c15df3a767617cb35c56;hb=1dd005530930657fd6216edc1dfcfa4c270a81c9;hpb=29e3d45f7bbbf31eadf793c41ff2b3d9c47b7539

diff --git a/src/Moof/Octree.hh b/src/Moof/Octree.hh
index 2ee617f..3614606 100644
--- a/src/Moof/Octree.hh
+++ b/src/Moof/Octree.hh
@@ -29,53 +29,146 @@
 #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();
 };