codea/tree.c

   1 #include <stdio.h>
   2 #include <stdlib.h>
   3 #include <string.h>
   4 #include "tree.h"
   5
   6 #if 0
   7 #define DDTREE
   8 #endif
   9
  10 static struct treenode *_new_plain(int op)
  11 {
  12         struct treenode *new = (TREECAST) malloc(TREESIZE);
  13         new->op = op;
  14         new->kids[0] = new->kids[1] = TREENULL;
  15         new->parent = TREENULL;
  16         new->label = NULL;
  17         new->name = new->reg = (char *)NULL;
  18         new->val = new->skip = new->exprcount = 0;
  19         new->param_index = -1;
  20         return new;
  21 }
  22
  23 struct treenode *new_node(int op, struct treenode *l, struct treenode *r, int exprcount)
  24 {
  25         struct treenode *new = TREENULL;
  26
  27 #ifdef DDTREE
  28         fprintf(stderr, "new_node: %i (%s)\n", op, o_names[op]);
  29 #endif
  30         if(op == O_SUB && l != TREENULL && r != TREENULL && l->op == O_NUM && r->op == O_NUM) {
  31                 new = new_number(l->val - r->val, exprcount);
  32                 free(l); free(r);
  33         } else  {
  34                 new = _new_plain(op);
  35                 new->kids[0] = l;
  36                 new->kids[1] = r;
  37                 new->exprcount = exprcount;
  38         }
  39         new->name = (char *)NULL;
  40         return new;
  41 }
  42
  43 struct treenode *new_param(int op, char *name, struct treenode *l, struct treenode *r, int param_index, int exprcount)
  44 {
  45         struct treenode *new = new_node(op, l, r, exprcount);
  46
  47 #ifdef DDTREE
  48         fprintf(stderr, "new_param: %i (index)\n", param_index);
  49 #endif
  50         new->param_index = param_index;
  51         new->name = name;
  52         return new;
  53 }
  54
  55 struct treenode *new_field(char *name, struct treenode *l, struct treenode *r, int soffset, int exprcount)
  56 {
  57         struct treenode *new = new_node(O_FIELD, l, r, exprcount);
  58
  59 #ifdef DDTREE
  60         fprintf(stderr, "new_field: %i (soffset)\n", soffset);
  61 #endif
  62         new->soffset = soffset;
  63         new->name = name;
  64         return new;
  65 }
  66
  67 struct treenode *new_number(long val, int exprcount)
  68 {
  69         struct treenode *new;
  70         /* TODO: maximal groesse? */
  71 #define BUFMAX 40
  72         char *t = (char*) malloc(BUFMAX);
  73
  74         if (val == 0) {
  75                 new = _new_plain(O_NULL);
  76         } else if(val == -1) {
  77                 new = _new_plain(O_MONE);
  78         } else if(val == -2) {
  79                 new = _new_plain(O_MTWO);
  80         } else if(val == -4) {
  81                 new = _new_plain(O_MFOUR);
  82         } else if(val == -8) {
  83                 new = _new_plain(O_MEIGHT);
  84         } else {
  85                 new = _new_plain(O_NUM);
  86         }
  87
  88 #ifdef DDTREE
  89         fprintf(stderr, "new_number: %i\n", val);
  90 #endif
  91         new->val = val;
  92         new->exprcount = exprcount;
  93         memset(t, 0, BUFMAX);
  94         sprintf(t, "%li", val);
  95         new->name = t;
  96
  97         return new;
  98 }
  99
 100 static void write_indent(int i)
 101 {
 102         int a;
 103         for(a = 0; a < i; a++) {
 104                 fprintf(stderr, "| ");
 105         }
 106 }
 107
 108 void write_tree(struct treenode *node, int indent)
 109 {
 110         if(node == TREENULL) return;
 111         write_indent(indent);
 112         fprintf(stderr, "%s @ %%%s (%i). \"%s\"\n", o_names[node->op], node->reg, node->exprcount, node->name == (char*) NULL ? "" : node->name);
 113         if(node->kids[0] != TREENULL) {
 114                 write_tree(node->kids[0], indent+1);
 115         }
 116         if(node->kids[1] != TREENULL) {
 117                 write_tree(node->kids[1], indent+1);
 118         }
 119 }
 120