/* Grammar subroutines needed by parser */

#include "Python.h"

#include "pgenheaders.h"

#include "grammar.h"

#include "token.h"

/* Return the DFA for the given type */

dfa *

PyGrammar_FindDFA(grammar *g, register int type)

{

	register dfa *d;

#if 1

	/* Massive speed-up */

	d = &g->g_dfa[type - NT_OFFSET];

	assert(d->d_type == type);

	return d;

#else

	/* Old, slow version */

	register int i;

	for (i = g->g_ndfas, d = g->g_dfa; --i >= 0; d++) {

		if (d->d_type == type)

			return d;

	}

	assert(0);

	/* NOTREACHED */

#endif

}

char *

PyGrammar_LabelRepr(label *lb)

{

	static char buf[100];

	if (lb->lb_type == ENDMARKER)

		return "EMPTY";

	else if (ISNONTERMINAL(lb->lb_type)) {

		if (lb->lb_str == NULL) {

			PyOS_snprintf(buf, sizeof(buf), "NT%d", lb->lb_type);

			return buf;

		}

		else

			return lb->lb_str;

	}

	else {

		if (lb->lb_str == NULL)

			return _PyParser_TokenNames[lb->lb_type];

		else {

			PyOS_snprintf(buf, sizeof(buf), "%.32s(%.32s)",

				_PyParser_TokenNames[lb->lb_type], lb->lb_str);

			return buf;

		}

	}

}