print parse tree

[umpf.git] / cond.y

diff --git a/cond.y b/cond.y
index 34af26373f77b43e4ff1553811ff5997e183fc45..e69668b30b7fc2a94264b545b21c3e4bf1cfe781 100644 (file)
--- a/cond.y
+++ b/cond.y
@@ -13,6 +13,7 @@ void yyerror (char const *);
 
 int regex_cmp(char* s, char* r);
 struct tree* tree_malloc(int type);
+void print_tree(struct tree* t, int ind);
 
 struct tree {
        enum {
@@ -22,19 +23,19 @@ struct tree {
                ST_ASS,
                ST_LEAF,
                ST_EMPTY
-       } st;
+       } st;   /* subtree type */
        union {
                struct {
-                       struct tree* c;
-                       struct tree* i;
-                       struct tree* e; 
+                       struct tree* c; /* condition */
+                       struct tree* i; /* if */
+                       struct tree* e; /* else */
                } tif;
 
                struct {
-                       int op; 
+                       int op;
                        struct tree* left;
                        struct tree* right;
-               } cond;
+               } cond; /* binary operator */
 
                struct {
                        struct tree* head;
@@ -61,6 +62,8 @@ struct tree {
        } pt;
 };
 
+struct tree* input_tree;
+
 %}
 
 %union {
@@ -74,17 +77,18 @@ struct tree {
 %token <str> VAR
 %token KW_PIPE KW_MAIL KW_COPY
 %token '(' ')' '{' '}' ';'
-%left KW_ELSE
-%left KW_IF
+%nonassoc KW_IF
+%nonassoc KW_ELSE
 %left ARROW
 %left <n> EQ NEQ GE LE '<' '>' RE NRE
+%left '='
+%left '.'
 %left '+' '-' 
 %left '*' '/'
 %left <n> '|'
 %left <n> '^'
 %left <n> '&'
 %left <n> '!'
-%left '='
 %type <tr> input 
 %type <tr> command 
 %type <tr> next 
@@ -95,10 +99,12 @@ struct tree {
 %type <n> rop 
 
 %%
-input: /* empty */     { $$ = NULL }
+input: /* empty */     { $$ = input_tree = NULL; }
        | command input {       $$ = tree_malloc(ST_BLOCK); 
                                $$->pt.block.head = $1;
-                               $$->pt.block.tail = $2; 
+                               $$->pt.block.tail = $2;
+
+                               input_tree = $$;
                        } 
 ;
 
@@ -108,8 +114,8 @@ command:     ';' { $$ = tree_malloc(ST_EMPTY); }
                                                $$->pt.block.head = $2;
                                                $$->pt.block.tail = $3; 
                                        }
-               | '{' '}' { $$ = NULL }
-               | cif { $$ = $1 }
+               | '{' '}' { $$ = tree_malloc(ST_EMPTY); }
+               | cif
                | ass ';' { $$ = $1}
                
        
@@ -248,9 +254,83 @@ yyerror (char const *s)
        fprintf (stderr, "%s\n", s);
 }
 
+void
+print_ind(int num, char c)
+{
+       int i;
+
+       for (i = 0; i < num; i++){
+               putchar(c);
+       }
+}
+
+void
+print_tree(struct tree* t, int ind)
+{
+       if (!t)
+               return; 
+
+       switch (t->st){
+               case ST_IF:
+                       print_ind(ind,' ');
+                       puts("if");
+                       print_tree(t->pt.tif.c,ind+1);
+                       print_ind(ind,' ');
+                       puts("then");
+                       print_tree(t->pt.tif.i,ind+1);
+                       print_ind(ind,' ');
+                       puts("else");
+                       print_tree(t->pt.tif.e,ind+1);
+                       break;
+               case ST_COND:
+#define UPPER(a) ((a) >> 8)
+#define LOWER(a) ((a) & 0xFF)
+                       print_tree(t->pt.cond.left, ind+1);
+                       print_ind(ind,' ');
+
+                       if (UPPER(t->pt.cond.op) > 0)
+                               putchar(UPPER(t->pt.cond.op));
+                       putchar(LOWER(t->pt.cond.op));
+                       putchar('\n'); 
+                       print_tree(t->pt.cond.right, ind+1);    
+                       break;
+               case ST_BLOCK:
+                       print_tree(t->pt.block.head,ind);
+                       print_tree(t->pt.block.tail,ind);
+                       break;
+               case ST_ASS:
+                       print_tree(t->pt.ass.left, ind+1);
+                       print_ind(ind,' ');     
+                       puts("=");
+                       print_tree(t->pt.ass.right, ind+1);
+                       break;
+               case ST_LEAF:
+                       print_ind(ind, ' ');
+                       switch (t->pt.leaf.type){
+                               case L_VAR:
+                                       putchar('$');
+                               case L_CONST:
+                                       puts(t->pt.leaf.value.s);
+                                       break;
+                               case L_NUM:
+                                       printf("%d\n",t->pt.leaf.value.n);
+                                       break;
+                       }
+                       break;
+               case ST_EMPTY:
+                       break;  
+
+
+       }
+}
+
+
 int
 main(void)
 {
-//     yydebug=1;
-       return yyparse ();
+       yydebug=1;
+       yyparse ();
+
+       print_tree(input_tree,0);
+       return 0;
 }