node.c 1.26 KB
Newer Older
1 2
/* Parse tree node implementation */

3
#include "Python.h"
4 5 6
#include "node.h"
#include "errcode.h"

Guido van Rossum's avatar
Guido van Rossum committed
7
node *
Thomas Wouters's avatar
Thomas Wouters committed
8
PyNode_New(int type)
Guido van Rossum's avatar
Guido van Rossum committed
9
{
10
	node *n = PyMem_NEW(node, 1);
Guido van Rossum's avatar
Guido van Rossum committed
11 12 13 14
	if (n == NULL)
		return NULL;
	n->n_type = type;
	n->n_str = NULL;
Guido van Rossum's avatar
Guido van Rossum committed
15
	n->n_lineno = 0;
Guido van Rossum's avatar
Guido van Rossum committed
16 17 18 19 20 21 22 23
	n->n_nchildren = 0;
	n->n_child = NULL;
	return n;
}

#define XXX 3 /* Node alignment factor to speed up realloc */
#define XXXROUNDUP(n) ((n) == 1 ? 1 : ((n) + XXX - 1) / XXX * XXX)

24
int
Thomas Wouters's avatar
Thomas Wouters committed
25
PyNode_AddChild(register node *n1, int type, char *str, int lineno)
Guido van Rossum's avatar
Guido van Rossum committed
26 27 28 29
{
	register int nch = n1->n_nchildren;
	register int nch1 = nch+1;
	register node *n;
30
	if (nch == INT_MAX || nch < 0)
31
		return E_OVERFLOW;
Guido van Rossum's avatar
Guido van Rossum committed
32 33 34
	if (XXXROUNDUP(nch) < nch1) {
		n = n1->n_child;
		nch1 = XXXROUNDUP(nch1);
35
		PyMem_RESIZE(n, node, nch1);
Guido van Rossum's avatar
Guido van Rossum committed
36
		if (n == NULL)
37
			return E_NOMEM;
Guido van Rossum's avatar
Guido van Rossum committed
38 39 40 41 42
		n1->n_child = n;
	}
	n = &n1->n_child[n1->n_nchildren++];
	n->n_type = type;
	n->n_str = str;
Guido van Rossum's avatar
Guido van Rossum committed
43
	n->n_lineno = lineno;
Guido van Rossum's avatar
Guido van Rossum committed
44 45
	n->n_nchildren = 0;
	n->n_child = NULL;
46
	return 0;
Guido van Rossum's avatar
Guido van Rossum committed
47
}
48

Guido van Rossum's avatar
Guido van Rossum committed
49
/* Forward */
50
static void freechildren(node *);
Guido van Rossum's avatar
Guido van Rossum committed
51 52 53


void
Thomas Wouters's avatar
Thomas Wouters committed
54
PyNode_Free(node *n)
Guido van Rossum's avatar
Guido van Rossum committed
55 56 57
{
	if (n != NULL) {
		freechildren(n);
58
		PyMem_DEL(n);
Guido van Rossum's avatar
Guido van Rossum committed
59 60 61
	}
}

62
static void
Thomas Wouters's avatar
Thomas Wouters committed
63
freechildren(node *n)
64 65 66 67 68
{
	int i;
	for (i = NCH(n); --i >= 0; )
		freechildren(CHILD(n, i));
	if (n->n_child != NULL)
69
		PyMem_DEL(n->n_child);
70
	if (STR(n) != NULL)
71
		PyMem_DEL(STR(n));
72
}