98 lines
2.3 KiB
C
98 lines
2.3 KiB
C
/*
|
|
handytree.h
|
|
Created By:
|
|
Andrew Lalis (andrewlalisofficial@gmail.com)
|
|
02 October 2017
|
|
This file is free to use, as long as the above comment remains in the file.
|
|
Please contact the author regarding bugs and/or feature requests.
|
|
*/
|
|
|
|
#ifdef TREE_TYPE
|
|
|
|
#include "templates.h"
|
|
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <assert.h>
|
|
#include <stdarg.h>
|
|
|
|
/*
|
|
Tree:
|
|
A data structure in which each node has some child nodes, and data about itself. A tree has only one root node, can have a great number of leaf nodes. Each node contains some data, and pointers to child
|
|
nodes.
|
|
*/
|
|
|
|
/*
|
|
TREE_NODE: the tree type for a tree node.
|
|
*/
|
|
#define TREE_NODE TEMPLATE(TREE_TYPE,tree_node)
|
|
|
|
/*
|
|
Tree structure:
|
|
Each tree node has some data, and a list of pointers to its child nodes, if any.
|
|
*/
|
|
typedef struct {
|
|
TREE_NODE** children;
|
|
int childCount;
|
|
TREE_TYPE data;
|
|
} TREE_NODE;
|
|
|
|
/*
|
|
Tree node creation:
|
|
Creates a new tree node, with some amount of child nodes, and data.
|
|
*/
|
|
TREE_NODE* TEMPLATE(new,TREE_NODE)(TREE_TYPE data, int childNum, ...){
|
|
TREE_NODE* node = malloc(sizeof(TREE_NODE));
|
|
assert(node != NULL);
|
|
node->data = data;
|
|
node->childCount = childNum;
|
|
//If there are children to add, add as many as given.
|
|
if (childNum > 0){
|
|
va_list args;
|
|
va_start(args, childNum);
|
|
node->children = malloc(childNum*sizeof(TREE_NODE*));
|
|
assert(node->children != NULL);
|
|
for (int i = 0; i < childNum; i++){
|
|
node->children[i] = va_arg(args, TREE_NODE*);
|
|
}
|
|
va_end(args);
|
|
}
|
|
return node;
|
|
}
|
|
|
|
/*
|
|
Tree node creation: Empty tree node.
|
|
Creates a new tree node with no children, just data.
|
|
*/
|
|
TREE_NODE* TEMPLATE(newEmpty,TREE_NODE)(TREE_TYPE data){
|
|
TREE_NODE* node = malloc(sizeof(TREE_NODE));
|
|
assert(node != NULL);
|
|
node->data = data;
|
|
node->childCount = 0;
|
|
return node;
|
|
}
|
|
|
|
/*
|
|
Free tree node:
|
|
Frees a node, and all children of this node.
|
|
*/
|
|
void TEMPLATE(free,TREE_NODE)(TREE_NODE* node){
|
|
for (int i = 0; i < node->childCount; i++){
|
|
TEMPLATE(free,TREE_NODE)(node->children[i]);
|
|
}
|
|
free(node->children);
|
|
}
|
|
|
|
/*
|
|
Add node:
|
|
Adds a node to a given parent node.
|
|
*/
|
|
void TEMPLATE(addNode,TREE_NODE)(TREE_NODE* node, TREE_NODE newChild){
|
|
node->childCount = node->childCount + 1;
|
|
TREE_NODE** oldChildren = node->children;
|
|
node->children = malloc(node->childCount * sizeof(TREE_NODE*));
|
|
assert(node->children != NULL);
|
|
memcpy(node->children, oldChildren, node->)
|
|
}
|
|
|
|
#endif |