72 lines
1.7 KiB
C
72 lines
1.7 KiB
C
|
|
/* List a node on a file */
|
|
|
|
#include "Python.h"
|
|
#include "pycore_interp.h" // PyInterpreterState.parser
|
|
#include "pycore_pystate.h" // _PyInterpreterState_GET()
|
|
#include "token.h"
|
|
#include "node.h"
|
|
|
|
/* Forward */
|
|
static void list1node(FILE *, node *);
|
|
static void listnode(FILE *, node *);
|
|
|
|
void
|
|
PyNode_ListTree(node *n)
|
|
{
|
|
listnode(stdout, n);
|
|
}
|
|
|
|
static void
|
|
listnode(FILE *fp, node *n)
|
|
{
|
|
PyInterpreterState *interp = _PyInterpreterState_GET();
|
|
|
|
interp->parser.listnode.level = 0;
|
|
interp->parser.listnode.atbol = 1;
|
|
list1node(fp, n);
|
|
}
|
|
|
|
static void
|
|
list1node(FILE *fp, node *n)
|
|
{
|
|
if (n == NULL)
|
|
return;
|
|
if (ISNONTERMINAL(TYPE(n))) {
|
|
int i;
|
|
for (i = 0; i < NCH(n); i++)
|
|
list1node(fp, CHILD(n, i));
|
|
}
|
|
else if (ISTERMINAL(TYPE(n))) {
|
|
PyInterpreterState *interp = _PyInterpreterState_GET();
|
|
struct _Py_parser_state *parser = &interp->parser;
|
|
switch (TYPE(n)) {
|
|
case INDENT:
|
|
parser->listnode.level++;
|
|
break;
|
|
case DEDENT:
|
|
parser->listnode.level--;
|
|
break;
|
|
default:
|
|
if (parser->listnode.atbol) {
|
|
int i;
|
|
for (i = 0; i < parser->listnode.level; ++i) {
|
|
fprintf(fp, "\t");
|
|
}
|
|
parser->listnode.atbol = 0;
|
|
}
|
|
if (TYPE(n) == NEWLINE) {
|
|
if (STR(n) != NULL)
|
|
fprintf(fp, "%s", STR(n));
|
|
fprintf(fp, "\n");
|
|
parser->listnode.atbol = 1;
|
|
}
|
|
else
|
|
fprintf(fp, "%s ", STR(n));
|
|
break;
|
|
}
|
|
}
|
|
else
|
|
fprintf(fp, "? ");
|
|
}
|