Added support for out edges.

[libfirm] / ir / ana / irtypeinfo.c

diff --git a/ir/ana/irtypeinfo.c b/ir/ana/irtypeinfo.c
index ab63ada..e14cedb 100644 (file)
--- a/ir/ana/irtypeinfo.c
+++ b/ir/ana/irtypeinfo.c
@@ -34,7 +34,7 @@
 #include <stddef.h>
 
 #include "irgraph_t.h"   /* for setting the state flag. */
-#include "irprog.h"
+#include "irprog_t.h"
 #include "pmap.h"
 
 /* ------------ The map. ---------------------------------------------- */
@@ -71,7 +71,6 @@ void init_irtypeinfo(void) {
 
   for (i = 0; i < get_irp_n_irgs(); ++i)
     set_irg_typeinfo_state(get_irp_irg(i), irg_typeinfo_none);
-
 }
 
 void free_irtypeinfo(void) {
@@ -83,7 +82,6 @@ void free_irtypeinfo(void) {
   } else
     assert(0 && "call init_type_info before freeing");
 
-
   if (type_node_map) {
     pmap_destroy(type_node_map);
     type_node_map = NULL;
@@ -113,7 +111,7 @@ irg_typeinfo_state get_irg_typeinfo_state(ir_graph *irg) {
  * irg_typeinfo_consistent or irg_typeinfo_inconsistent.  They
  * assume current_ir_graph set properly.
  */
-type *get_irn_type(ir_node *n) {
+type *get_irn_typeinfo_type(ir_node *n) {
   type *res = initial_type;
   assert(get_irg_typeinfo_state(current_ir_graph) == irg_typeinfo_consistent  ||
         get_irg_typeinfo_state(current_ir_graph) == irg_typeinfo_inconsistent  );
@@ -124,9 +122,49 @@ type *get_irn_type(ir_node *n) {
   return res;
 }
 
-void set_irn_type(ir_node *n, type *tp) {
+void set_irn_typeinfo_type(ir_node *n, type *tp) {
   assert(get_irg_typeinfo_state(current_ir_graph) == irg_typeinfo_consistent  ||
         get_irg_typeinfo_state(current_ir_graph) == irg_typeinfo_inconsistent  );
 
   pmap_insert(type_node_map, (void *)n, (void *)tp);
 }
+
+type *get_irn_type(ir_node *n) {
+  type *tp = NULL;
+  switch(get_irn_opcode(n)) {
+  case iro_Const:     tp = get_Const_type(n); break;
+  case iro_SymConst:  tp = get_SymConst_value_type(n); break;
+  case iro_Cast:      tp = get_Cast_type(n);  break;
+  case iro_Proj: {
+    ir_node *pred = get_Proj_pred(n);
+    switch (get_irn_opcode(pred)) {
+    case iro_Proj: {
+      ir_node *pred_pred;
+      /* Deal with Start / Call here: we need to know the Proj Nr. */
+      assert(get_irn_mode(pred) == mode_T);
+      pred_pred = get_Proj_pred(pred);
+      if (get_irn_op(pred_pred) == op_Start)  {
+       type *mtp = get_entity_type(get_irg_entity(get_irn_irg(pred_pred)));
+       tp = get_method_param_type(mtp, get_Proj_proj(n));
+      } else if (get_irn_op(pred_pred) == op_Call) {
+       type *mtp = get_Call_type(pred_pred);
+       tp = get_method_res_type(mtp, get_Proj_proj(n));
+      }
+    } break;
+    case iro_Start: break;
+    case iro_Call: break;
+    case iro_Load: {
+      ir_node *a = get_Load_ptr(pred);
+      if (get_irn_op(a) == op_Sel)
+       tp = get_entity_type(get_Sel_entity(a));
+    } break;
+    default:
+      break;
+    }
+  } break; /* iro_Proj */
+  default:
+    tp = NULL;
+  }
+
+  return tp;
+}