Guido van Rossum | f70e43a | 1991-02-19 12:39:46 +0000 | [diff] [blame] | 1 | |
Guido van Rossum | 85a5fbb | 1990-10-14 12:07:46 +0000 | [diff] [blame] | 2 | /* List a node on a file */ |
| 3 | |
Pablo Galindo | f2cf1e3 | 2019-04-13 17:05:14 +0100 | [diff] [blame] | 4 | #include "Python.h" |
Victor Stinner | e5014be | 2020-04-14 17:52:15 +0200 | [diff] [blame] | 5 | #include "pycore_interp.h" // PyInterpreterState.parser |
| 6 | #include "pycore_pystate.h" // _PyInterpreterState_GET() |
Guido van Rossum | 85a5fbb | 1990-10-14 12:07:46 +0000 | [diff] [blame] | 7 | #include "token.h" |
| 8 | #include "node.h" |
| 9 | |
Guido van Rossum | 3f5da24 | 1990-12-20 15:06:42 +0000 | [diff] [blame] | 10 | /* Forward */ |
Tim Peters | dbd9ba6 | 2000-07-09 03:09:57 +0000 | [diff] [blame] | 11 | static void list1node(FILE *, node *); |
| 12 | static void listnode(FILE *, node *); |
Guido van Rossum | 3f5da24 | 1990-12-20 15:06:42 +0000 | [diff] [blame] | 13 | |
| 14 | void |
Thomas Wouters | 23c9e00 | 2000-07-22 19:20:54 +0000 | [diff] [blame] | 15 | PyNode_ListTree(node *n) |
Guido van Rossum | 3f5da24 | 1990-12-20 15:06:42 +0000 | [diff] [blame] | 16 | { |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 17 | listnode(stdout, n); |
Guido van Rossum | 3f5da24 | 1990-12-20 15:06:42 +0000 | [diff] [blame] | 18 | } |
| 19 | |
Guido van Rossum | 9e90a67 | 1993-06-24 11:10:19 +0000 | [diff] [blame] | 20 | static void |
Thomas Wouters | 23c9e00 | 2000-07-22 19:20:54 +0000 | [diff] [blame] | 21 | listnode(FILE *fp, node *n) |
Guido van Rossum | 3f5da24 | 1990-12-20 15:06:42 +0000 | [diff] [blame] | 22 | { |
Victor Stinner | 81a7be3 | 2020-04-14 15:14:01 +0200 | [diff] [blame] | 23 | PyInterpreterState *interp = _PyInterpreterState_GET(); |
Vinay Sajip | 9def81a | 2019-11-07 10:08:58 +0000 | [diff] [blame] | 24 | |
| 25 | interp->parser.listnode.level = 0; |
| 26 | interp->parser.listnode.atbol = 1; |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 27 | list1node(fp, n); |
Guido van Rossum | 3f5da24 | 1990-12-20 15:06:42 +0000 | [diff] [blame] | 28 | } |
| 29 | |
Guido van Rossum | 85a5fbb | 1990-10-14 12:07:46 +0000 | [diff] [blame] | 30 | static void |
Thomas Wouters | 23c9e00 | 2000-07-22 19:20:54 +0000 | [diff] [blame] | 31 | list1node(FILE *fp, node *n) |
Guido van Rossum | 85a5fbb | 1990-10-14 12:07:46 +0000 | [diff] [blame] | 32 | { |
Vinay Sajip | 9def81a | 2019-11-07 10:08:58 +0000 | [diff] [blame] | 33 | PyInterpreterState *interp; |
| 34 | |
Serhiy Storchaka | 0b3ec19 | 2017-03-23 17:53:47 +0200 | [diff] [blame] | 35 | if (n == NULL) |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 36 | return; |
| 37 | if (ISNONTERMINAL(TYPE(n))) { |
| 38 | int i; |
| 39 | for (i = 0; i < NCH(n); i++) |
| 40 | list1node(fp, CHILD(n, i)); |
| 41 | } |
| 42 | else if (ISTERMINAL(TYPE(n))) { |
Victor Stinner | 81a7be3 | 2020-04-14 15:14:01 +0200 | [diff] [blame] | 43 | interp = _PyInterpreterState_GET(); |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 44 | switch (TYPE(n)) { |
| 45 | case INDENT: |
Vinay Sajip | 9def81a | 2019-11-07 10:08:58 +0000 | [diff] [blame] | 46 | interp->parser.listnode.level++; |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 47 | break; |
| 48 | case DEDENT: |
Vinay Sajip | 9def81a | 2019-11-07 10:08:58 +0000 | [diff] [blame] | 49 | interp->parser.listnode.level--; |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 50 | break; |
| 51 | default: |
Vinay Sajip | 9def81a | 2019-11-07 10:08:58 +0000 | [diff] [blame] | 52 | if (interp->parser.listnode.atbol) { |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 53 | int i; |
Vinay Sajip | 9def81a | 2019-11-07 10:08:58 +0000 | [diff] [blame] | 54 | for (i = 0; i < interp->parser.listnode.level; ++i) |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 55 | fprintf(fp, "\t"); |
Vinay Sajip | 9def81a | 2019-11-07 10:08:58 +0000 | [diff] [blame] | 56 | interp->parser.listnode.atbol = 0; |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 57 | } |
| 58 | if (TYPE(n) == NEWLINE) { |
| 59 | if (STR(n) != NULL) |
| 60 | fprintf(fp, "%s", STR(n)); |
| 61 | fprintf(fp, "\n"); |
Vinay Sajip | 9def81a | 2019-11-07 10:08:58 +0000 | [diff] [blame] | 62 | interp->parser.listnode.atbol = 1; |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 63 | } |
| 64 | else |
| 65 | fprintf(fp, "%s ", STR(n)); |
| 66 | break; |
| 67 | } |
| 68 | } |
| 69 | else |
| 70 | fprintf(fp, "? "); |
Guido van Rossum | 85a5fbb | 1990-10-14 12:07:46 +0000 | [diff] [blame] | 71 | } |