X-Git-Url: https://git.ucc.asn.au/?p=ipdf%2Fcode.git;a=blobdiff_plain;f=src%2Fdocument.cpp;h=80e065150c94377753ec0b4703c74d4f0bce0ae0;hp=e49e52edb70e925e4f08df5a26376d4796d09cf3;hb=e0cd98bdff7f026b92f2c0e5b08639e08b98874e;hpb=35e4687acd16a28fc923a75d254cdb4fd2fbc524

diff --git a/src/document.cpp b/src/document.cpp
index e49e52e..80e0651 100644
--- a/src/document.cpp
+++ b/src/document.cpp
@@ -89,10 +89,47 @@ void Document::Save(const string & filename)
 
 void Document::GenBaseQuadtree()
 {
-	m_quadtree.nodes.push_back(QuadTreeNode{QUADTREE_EMPTY, QUADTREE_EMPTY, QUADTREE_EMPTY, QUADTREE_EMPTY, QUADTREE_EMPTY, QTC_UNKNOWN, 0, ObjectCount()});
+	m_quadtree.nodes.push_back(QuadTreeNode{QUADTREE_EMPTY, 1, QUADTREE_EMPTY, QUADTREE_EMPTY, QUADTREE_EMPTY, QTC_UNKNOWN, 0, 1});
+	m_quadtree.nodes.push_back(QuadTreeNode{QUADTREE_EMPTY, QUADTREE_EMPTY, QUADTREE_EMPTY, QUADTREE_EMPTY, QUADTREE_EMPTY, QTC_UNKNOWN, 1, ObjectCount()});
 	m_quadtree.root_id = 0;
 }
 
+QuadTreeIndex Document::GenQuadNode(QuadTreeIndex parent, QuadTreeNodeChildren type)
+{
+	QuadTreeIndex new_index = m_quadtree.nodes.size();
+	m_quadtree.nodes.push_back(QuadTreeNode{QUADTREE_EMPTY, QUADTREE_EMPTY, QUADTREE_EMPTY, QUADTREE_EMPTY, parent, type, 0, 0});
+
+	m_quadtree.nodes[new_index].object_begin = m_objects.bounds.size();
+	for (unsigned i = m_quadtree.nodes[parent].object_begin; i < m_quadtree.nodes[parent].object_end; ++i)
+	{
+		if (ContainedInQuadChild(m_objects.bounds[i], type))
+		{
+			m_objects.bounds.push_back(TransformToQuadChild(m_objects.bounds[i], type));
+			m_objects.types.push_back(m_objects.types[i]);
+			m_objects.data_indices.push_back(m_objects.data_indices[i]);
+		}
+	}
+	m_quadtree.nodes[new_index].object_end = m_objects.bounds.size();
+	switch (type)
+	{
+		case QTC_TOP_LEFT:
+			m_quadtree.nodes[parent].top_left = new_index;
+			break;
+		case QTC_TOP_RIGHT:
+			m_quadtree.nodes[parent].top_right = new_index;
+			break;
+		case QTC_BOTTOM_LEFT:
+			m_quadtree.nodes[parent].bottom_left = new_index;
+			break;
+		case QTC_BOTTOM_RIGHT:
+			m_quadtree.nodes[parent].bottom_right = new_index;
+			break;
+		default:
+			Fatal("Tried to add a QuadTree child of invalid type!");
+	}
+	return new_index;
+}
+
 #endif
 
 void Document::Load(const string & filename)
@@ -142,6 +179,12 @@ void Document::Load(const string & filename)
 		}
 	}
 	Debug("Successfully loaded %u objects from \"%s\"", ObjectCount(), filename.c_str());
+#ifndef QUADTREE_DISABLED
+	if (m_quadtree.root_id == QUADTREE_EMPTY)
+	{
+		GenBaseQuadtree();
+	}
+#endif
 }
 
 void Document::Add(ObjectType type, const Rect & bounds, unsigned data_index)