firstsets.c 2.12 KB
Newer Older
1

Guido van Rossum's avatar
Guido van Rossum committed
2 3
/* Computation of FIRST stets */

Guido van Rossum's avatar
Guido van Rossum committed
4
#include "pgenheaders.h"
Guido van Rossum's avatar
Guido van Rossum committed
5 6 7
#include "grammar.h"
#include "token.h"

8
extern int Py_DebugFlag;
Guido van Rossum's avatar
Guido van Rossum committed
9

Guido van Rossum's avatar
Guido van Rossum committed
10
/* Forward */
11
static void calcfirstset(grammar *, dfa *);
Guido van Rossum's avatar
Guido van Rossum committed
12 13

void
Thomas Wouters's avatar
Thomas Wouters committed
14
addfirstsets(grammar *g)
Guido van Rossum's avatar
Guido van Rossum committed
15 16 17 18 19 20 21 22 23 24 25 26
{
	int i;
	dfa *d;
	
	printf("Adding FIRST sets ...\n");
	for (i = 0; i < g->g_ndfas; i++) {
		d = &g->g_dfa[i];
		if (d->d_first == NULL)
			calcfirstset(g, d);
	}
}

Guido van Rossum's avatar
Guido van Rossum committed
27
static void
Thomas Wouters's avatar
Thomas Wouters committed
28
calcfirstset(grammar *g, dfa *d)
Guido van Rossum's avatar
Guido van Rossum committed
29 30 31 32 33 34 35 36 37 38 39 40 41
{
	int i, j;
	state *s;
	arc *a;
	int nsyms;
	int *sym;
	int nbits;
	static bitset dummy;
	bitset result;
	int type;
	dfa *d1;
	label *l0;
	
42
	if (Py_DebugFlag)
Guido van Rossum's avatar
Guido van Rossum committed
43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60
		printf("Calculate FIRST set for '%s'\n", d->d_name);
	
	if (dummy == NULL)
		dummy = newbitset(1);
	if (d->d_first == dummy) {
		fprintf(stderr, "Left-recursion for '%s'\n", d->d_name);
		return;
	}
	if (d->d_first != NULL) {
		fprintf(stderr, "Re-calculating FIRST set for '%s' ???\n",
			d->d_name);
	}
	d->d_first = dummy;
	
	l0 = g->g_ll.ll_label;
	nbits = g->g_ll.ll_nlabels;
	result = newbitset(nbits);
	
61
	sym = PyMem_NEW(int, 1);
Guido van Rossum's avatar
Guido van Rossum committed
62
	if (sym == NULL)
63
		Py_FatalError("no mem for new sym in calcfirstset");
Guido van Rossum's avatar
Guido van Rossum committed
64 65 66 67 68 69 70 71 72 73 74
	nsyms = 1;
	sym[0] = findlabel(&g->g_ll, d->d_type, (char *)NULL);
	
	s = &d->d_state[d->d_initial];
	for (i = 0; i < s->s_narcs; i++) {
		a = &s->s_arc[i];
		for (j = 0; j < nsyms; j++) {
			if (sym[j] == a->a_lbl)
				break;
		}
		if (j >= nsyms) { /* New label */
75
			PyMem_RESIZE(sym, int, nsyms + 1);
Guido van Rossum's avatar
Guido van Rossum committed
76
			if (sym == NULL)
77 78
				Py_FatalError(
				    "no mem to resize sym in calcfirstset");
Guido van Rossum's avatar
Guido van Rossum committed
79 80 81
			sym[nsyms++] = a->a_lbl;
			type = l0[a->a_lbl].lb_type;
			if (ISNONTERMINAL(type)) {
82
				d1 = PyGrammar_FindDFA(g, type);
Guido van Rossum's avatar
Guido van Rossum committed
83 84 85 86 87 88 89 90
				if (d1->d_first == dummy) {
					fprintf(stderr,
						"Left-recursion below '%s'\n",
						d->d_name);
				}
				else {
					if (d1->d_first == NULL)
						calcfirstset(g, d1);
91 92
					mergebitset(result,
						    d1->d_first, nbits);
Guido van Rossum's avatar
Guido van Rossum committed
93 94 95 96 97 98 99 100
				}
			}
			else if (ISTERMINAL(type)) {
				addbit(result, a->a_lbl);
			}
		}
	}
	d->d_first = result;
101
	if (Py_DebugFlag) {
Guido van Rossum's avatar
Guido van Rossum committed
102 103 104
		printf("FIRST set for '%s': {", d->d_name);
		for (i = 0; i < nbits; i++) {
			if (testbit(result, i))
105
				printf(" %s", PyGrammar_LabelRepr(&l0[i]));
Guido van Rossum's avatar
Guido van Rossum committed
106 107 108 109
		}
		printf(" }\n");
	}
}