X-Git-Url: https://git.ucc.asn.au/?p=ipdf%2Fcode.git;a=blobdiff_plain;f=src%2Fdocument.h;h=0f2fea11dc836eefcf8e616a8a7ae14d59301286;hp=54ccdd08a48e961480a02a442caa750c90609bc6;hb=8793e48b5e8f28f59d9db8325ce05ec7c0431d40;hpb=f8ef964f021d1d6da6ea46bbb1fe8f0250a5be8c

diff --git a/src/document.h b/src/document.h
index 54ccdd0..0f2fea1 100644
--- a/src/document.h
+++ b/src/document.h
@@ -2,6 +2,7 @@
 #define _DOCUMENT_H
 
 #include "ipdf.h"
+#include "quadtree.h"
 
 namespace IPDF
 {
@@ -13,7 +14,6 @@ namespace IPDF
 
 			void Load(const std::string & filename = "");
 			void Save(const std::string & filename);
-			void Add(ObjectType type, const Rect & bounds);
 			void DebugDumpObjects();
 
 			unsigned ObjectCount() const {return m_count;}
@@ -22,10 +22,23 @@ namespace IPDF
 			bool operator==(const Document & equ) const;
 			bool operator!=(const Document & equ) const {return !(this->operator==(equ));}
 
+			void Add(ObjectType type, const Rect & bounds, unsigned data_index = 0);
+			unsigned AddBezierData(const Bezier & bezier);
+
+#ifndef QUADTREE_DISABLED
+			inline const QuadTree& GetQuadTree() { if (m_quadtree.root_id == QUADTREE_EMPTY) { GenBaseQuadtree(); } return m_quadtree; }
+#endif
+
 		private:
 			friend class View;
 			Objects m_objects;
+#ifndef QUADTREE_DISABLED
+			QuadTree m_quadtree;
+			void GenBaseQuadtree();
+#endif
 			unsigned m_count;
+			
+
 	};
 }