#include "common.h"
#include "code.h"
#include "sym.h"
#include "mem.h"

struct _mem node_mem;

void
TreeFree(Node *root)
{
	if(!root)
		return;
	TreeFree(root->children);
	TreeFree(root->siblings);
	free(root);
}

void
TreePrint(Node *tree, int flag)
{
	if(!tree)
		return;
	print("%s(", (tree->sym)->name);
	TreePrint(tree->children, 0);
	print(")");
	if(tree->siblings != NULL){
		print(",");
		TreePrint(tree->siblings, 0);
	}
	if(flag)
		print("\n");
}

Node *
rev(Node *sl, Node *nl, int *nlleaves)
{
	Node *sl2;

	if(!sl)
		return nl;
	sl2 = sl->siblings;
	sl->siblings = nl;
	*nlleaves += sl->nlleaves;
	return rev(sl2, sl, nlleaves);
}

Node *
TreeBuild(SymbolEntry *symp, Node *children)
{
	Node *np = (struct node*)mem_get(&node_mem);

	np->sym = symp;
	np->nlleaves = symp->attr==A_LABEL ? 1 : 0;
	np->children = rev(children, NULL, &np->nlleaves);
	return np;
}

void
TreeInit(void)
{
	mem_init(&node_mem, sizeof(struct node));
}

int
cntnodes(Node *np)
{
	if(!np)
		return 0;
	return 1 + cntnodes(np->children) + cntnodes(np->siblings);
}