aboutsummaryrefslogtreecommitdiffstats
path: root/src/tree.h
diff options
context:
space:
mode:
Diffstat (limited to 'src/tree.h')
-rw-r--r--src/tree.h51
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