1 files changed, 51 insertions, 0 deletions
diff --git a/src/tree.h b/src/tree.h
new file mode 100644
index 000000000..deb420bc8
--- /dev/null
+++ b/src/tree.h
@@ -0,0 +1,51 @@
+#ifndef TREE_H
+#define TREE_H
+
+typedef struct _TreeNode {
+	void * data[2];
+	struct _TreeNode * children[3];
+	struct _TreeNode * parent;
+} TreeNode;
+
+typedef struct _Tree {
+	TreeNode headNode;
+	TreeFreeDataFunc * freeFunc;
+	TreeCompareDataFunc * compareFunc;
+} Tree;
+
+typedef enum _TreeIteratorType {
+	PREORDER,
+	INORDER,
+	POSTORDER
+} TreeIteratorType;
+
+typedef struct _TreeIterator {
+	Data * data;
+	/* private data */
+	TreeIteratorType type;
+	TreeNode * currentNode;
+	int pos;
+} TreeIterator;
+
+typedef int TreeCompareDataFunc(void * left, void * right);
+
+typedef int TreeFreeDataFunc(void * data);
+
+Tree * newTree(TreeFreeDataFunc * freeFunc, TreeCompareDataFunc * compareFunc);
+
+void freeTree(Tree * tree);
+
+int insertInTree(Tree * tree, void * data);
+
+int deleteFromTree(Tree * tree, void * needle);
+
+void * findInTree(Tree * tree, void * needle);
+
+TreeIterator * newTreeIterator(Tree * tree, TreeIteratorType type);
+
+/* will return the same pointer passed in on success
+ * if NULL is returned, this indicates the end of tree
+ */
+TreeIterator * nextTreeIterator(TreeIterator * iter);
+
+#endif // TREE_H