Added walker to walk entities within a type
[libfirm] / ir / tr / typewalk.c
index ee71534..ba91689 100644 (file)
 #include <stdlib.h>
 #include <stdio.h>
 #include "irgwalk.h"
-#include "irgraph.h"
+#include "irgraph_t.h"
 #include "irnode.h"
 #include "irprog.h"
 #include "type_or_entity.h"
+#include "typegmod.h"
 
 /* Make types visible to allow most efficient access */
 #include "entity_t.h"
@@ -79,17 +80,17 @@ void type_walk_2(type_or_ent *tore,
       switch (get_type_tpop_code(tp)) {
       case tpo_class:
        {
-         for (i=0; i<get_class_n_supertype(tp); i++)
+         for (i=0; i<get_class_n_supertypes(tp); i++)
            type_walk_2((type_or_ent *)get_class_supertype(tp, i), pre, post, env);
-         for (i=0; i<get_class_n_member(tp); i++)
+         for (i=0; i<get_class_n_members(tp); i++)
            type_walk_2((type_or_ent *)get_class_member(tp, i), pre, post, env);
-         for (i=0; i<get_class_n_subtype(tp); i++)
+         for (i=0; i<get_class_n_subtypes(tp); i++)
            type_walk_2((type_or_ent *)get_class_subtype(tp, i), pre, post, env);
        }
        break;
       case tpo_struct:
        {
-         for (i=0; i<get_struct_n_member(tp); i++)
+         for (i=0; i<get_struct_n_members(tp); i++)
            type_walk_2((type_or_ent *)get_struct_member(tp, i), pre, post, env);
        }
        break;
@@ -97,7 +98,7 @@ void type_walk_2(type_or_ent *tore,
        {
          for (i = 0; i < get_method_n_params(tp); i++)
            type_walk_2((type_or_ent *)get_method_param_type(tp, i), pre, post, env);
-         for (i = 0; i < get_method_n_res(tp); i++)
+         for (i = 0; i < get_method_n_ress(tp); i++)
            type_walk_2((type_or_ent *)get_method_res_type(tp, i), pre, post, env);
        }
        break;
@@ -110,6 +111,8 @@ void type_walk_2(type_or_ent *tore,
       case tpo_array:
        type_walk_2((type_or_ent *)get_array_element_type(tp),
                    pre, post, env);
+       type_walk_2((type_or_ent *)get_array_element_entity(tp),
+                   pre, post, env);
        break;
       case tpo_enumeration:
        /* a leave */
@@ -162,7 +165,6 @@ void start_type_walk(ir_node *node, void *env) {
     type_walk_2((type_or_ent *)get_Alloc_type(node), pre, post, envi);
     break;
   case iro_Free:
-    printf("here in typewalk\n");
     type_walk_2((type_or_ent *)get_Free_type(node), pre, post, envi);
     break;
   default:
@@ -175,10 +177,12 @@ void type_walk(void (pre)(type_or_ent*, void*),
               void *env) {
   int i;
   ++type_visited;
-  type_walk_2((type_or_ent *)get_glob_type(), pre, post, env);
+  /*type_walk_2((type_or_ent *)get_glob_type(), pre, post, env);
+   global type is on the list visited below, too. */
   for (i = 0; i < get_irp_n_types(); i++) {
     type_walk_2((type_or_ent *)get_irp_type(i), pre, post, env);
   }
+  type_walk_2((type_or_ent *)get_glob_type(), pre, post, env);
 }
 
 void type_walk_irg (ir_graph *irg,
@@ -235,16 +239,19 @@ void type_walk_s2s_2(type_or_ent *tore,
       switch (get_type_tpop_code(tp)) {
       case tpo_class:
        {
-         for (i=0; i<get_class_n_supertype(tp); i++)
-           type_walk_s2s_2((type_or_ent *)get_class_supertype(tp, i), pre, post, env);
-
+         for (i = 0; i < get_class_n_supertypes(tp); i++) {
+           type_walk_s2s_2((type_or_ent *)get_class_supertype(tp, i), pre,
+                           post, env);
+         }
          /* execute pre method */
          if(pre)
            pre(tore, env);
          tp = skip_tid((type*)tore);
 
-         for (i=0; i<get_class_n_subtype(tp); i++)
-           type_walk_s2s_2((type_or_ent *)get_class_subtype(tp, i), pre, post, env);
+         for (i = 0; i < get_class_n_subtypes(tp); i++) {
+           type_walk_s2s_2((type_or_ent *)get_class_subtype(tp, i), pre,
+                           post, env);
+         }
 
          /* execute post method */
          if(post)
@@ -280,9 +287,207 @@ void type_walk_super2sub(void (pre)(type_or_ent*, void*),
                         void (post)(type_or_ent*, void*),
                         void *env) {
   int i;
+  type *tp;
   ++type_visited;
   type_walk_s2s_2((type_or_ent *)get_glob_type(), pre, post, env);
   for (i = 0; i < get_irp_n_types(); i++) {
-    type_walk_s2s_2((type_or_ent *)get_irp_type(i), pre, post, env);
+    tp = get_irp_type(i);
+    type_walk_s2s_2((type_or_ent *)tp, pre, post, env);
+  }
+}
+
+/*****************************************************************************/
+
+static
+void type_walk_super_2(type_or_ent *tore,
+                      void (pre)(type_or_ent*, void*),
+                      void (post)(type_or_ent*, void*),
+                      void *env)
+{
+  int i;
+
+  /* marked? */
+  switch (get_kind(tore)) {
+  case k_entity:
+    if (((entity *)tore)->visit >= type_visited) return;
+    break;
+  case k_type:
+    if(type_id == get_type_tpop((type*)tore)) {
+      type_walk_super_2((type_or_ent *)skip_tid((type *)tore), pre, post, env);
+      return;
+    }
+    if (((type *)tore)->visit >= type_visited) return;
+    break;
+  default:
+    break;
+  }
+
+  /* iterate */
+  switch (get_kind(tore)) {
+  case k_type:
+    {
+      type *tp = (type *)tore;
+      mark_type_visited(tp);
+      switch (get_type_tpop_code(tp)) {
+      case tpo_class:
+       {
+         /* execute pre method */
+         if(pre)
+           pre(tore, env);
+         tp = skip_tid((type*)tore);
+
+         for (i = 0; i < get_class_n_supertypes(tp); i++) {
+           type_walk_super_2((type_or_ent *)get_class_supertype(tp, i), pre,
+                             post, env);
+         }
+
+         /* execute post method */
+         if(post)
+           post(tore, env);
+       }
+       break;
+      case tpo_struct:
+      case tpo_method:
+      case tpo_union:
+      case tpo_array:
+      case tpo_enumeration:
+      case tpo_pointer:
+      case tpo_primitive:
+      case tpo_id:
+       /* dont care */
+       break;
+      default:
+       printf(" *** Faulty type! \n");
+       break;
+      }
+    } break; /* end case k_type */
+  case k_entity:
+    /* dont care */
+    break;
+  default:
+    printf(" *** Faulty type or entity! \n");
+    break;
+  }
+  return;
+}
+
+void type_walk_super(void (pre)(type_or_ent*, void*),
+                    void (post)(type_or_ent*, void*),
+                    void *env) {
+  int i;
+  type *tp;
+  ++type_visited;
+  type_walk_super_2((type_or_ent *)get_glob_type(), pre, post, env);
+  for (i = 0; i < get_irp_n_types(); i++) {
+    tp = get_irp_type(i);
+    type_walk_super_2((type_or_ent *)tp, pre, post, env);
+  }
+}
+
+/*****************************************************************************/
+
+
+void class_walk_s2s_2(type *tp,
+                    void (pre)(type*, void*),
+                    void (post)(type*, void*),
+                    void *env)
+{
+  int i;
+
+  /* marked? */
+  if (tp->visit >= type_visited) return;
+
+  assert(is_class_type(tp));
+  /* Assure all supertypes are visited before */
+  for (i=0; i < get_class_n_supertypes(tp); i++) {
+    if (get_type_visited(get_class_supertype(tp, i)) < type_visited)
+      return;
+  }
+
+  mark_type_visited(tp);
+
+  /* execute pre method */
+  if(pre)
+    pre(tp, env);
+
+  tp = skip_tid((type*)tp);
+  for (i=0; i<get_class_n_subtypes(tp); i++) {
+    class_walk_s2s_2(get_class_subtype(tp, i), pre, post, env);
+  }
+  /* execute post method */
+  if(post)
+    post(tp, env);
+
+  return;
+}
+
+#if 0
+void class_walk_super2sub(void (pre)(type*, void*),
+                         void (post)(type*, void*),
+                         void *env) {
+  int i;
+  type *tp;
+
+  ++type_visited;
+  for (i = 0; i < get_irp_n_types(); i++) {
+    tp = get_irp_type(i);
+    if (is_class_type(tp) &&
+       (get_class_n_supertypes(tp) == 0) &&
+       (tp->visit < type_visited) &&
+       (!is_frame_type(tp)) &&
+       (tp != get_glob_type())) {
+      class_walk_s2s_2(tp, pre, post, env);
+    }
+  }
+}
+#endif
+void class_walk_super2sub(void (pre)(type*, void*),
+                         void (post)(type*, void*),
+                         void *env) {
+  int i;
+  type *tp;
+
+  ++type_visited;
+  for (i = 0; i < get_irp_n_types(); i++) {
+    tp = get_irp_type(i);
+    if (is_class_type(tp) &&
+       (get_class_n_supertypes(tp) == 0) &&
+       (tp->visit < type_visited))  {
+      assert(!is_frame_type(tp));
+      assert(tp != get_glob_type());
+      class_walk_s2s_2(tp, pre, post, env);
+    }
+  }
+}
+
+
+/* Walks over all entities in the type */
+void walk_types_entities(type *tp,
+                        void (doit)(entity*, void*),
+                        void *env) {
+  int i;
+  switch(get_type_tpop_code(tp)) {
+  case tpo_class: {
+    for (i=0; i<get_class_n_members(tp); i++)
+      doit(get_class_member(tp, i), env);
+  } break;
+  case tpo_struct: {
+    for (i=0; i<get_struct_n_members(tp); i++)
+      doit(get_struct_member(tp, i), env);
+  } break;
+  case tpo_union: {
+    for (i = 0; i < get_union_n_members(tp); i++)
+      doit(get_union_member(tp, i), env);
+  } break;
+  case tpo_array: {
+      doit(get_array_element_entity(tp), env);
+  } break;
+  case tpo_method:
+  case tpo_enumeration:
+  case tpo_pointer:
+  case tpo_primitive:
+  case tpo_id:
+  default:
+    break;
   }
 }