Bugfix
[libfirm] / ir / be / benode.c
index 5adbad8..b7a53f0 100644 (file)
@@ -18,6 +18,7 @@
 #include "set.h"
 #include "pmap.h"
 #include "util.h"
+#include "debug.h"
 
 #include "irop_t.h"
 #include "irmode_t.h"
@@ -31,6 +32,8 @@
 
 #include "beirgmod.h"
 
+#define DBG_LEVEL 0
+
 typedef enum _node_kind_t {
   node_kind_spill,
   node_kind_reload,
@@ -96,17 +99,15 @@ ir_node *new_Spill(const be_node_factory_t *factory,
 }
 
 ir_node *new_Reload(const be_node_factory_t *factory,
-    const arch_register_class_t *cls,
-    ir_graph *irg, ir_node *bl, ir_node *spill_node)
+    const arch_register_class_t *cls, ir_graph *irg,
+    ir_node *bl, ir_mode *mode, ir_node *spill_node)
 {
-  ir_mode *mode;
   ir_node *in[1];
   ir_op *op = get_op(factory, cls, node_kind_reload)->op;
 
   assert(op && "Reload opcode must be present for this register class");
-  assert(is_Spill(factory, spill_node) && "Operand of Reload must be a Spill");
+  // assert(is_Spill(factory, spill_node) && "Operand of Reload must be a Spill");
   in[0] = spill_node;
-  mode = get_irn_mode(get_irn_n(spill_node, 0));
 
   return new_ir_node(NULL, irg, bl, op, mode, 1, in);
 }
@@ -125,13 +126,60 @@ ir_node *new_Copy(const be_node_factory_t *factory,
     ir_graph *irg, ir_node *bl, ir_node *in)
 {
   ir_node *ins[1];
-  ir_op *op = get_op(factory, cls, node_kind_perm)->op;
+  ir_op *op = get_op(factory, cls, node_kind_copy)->op;
 
   ins[0] = in;
 
   return new_ir_node(NULL, irg, bl, op, get_irn_mode(in), 1, ins);
 }
 
+ir_node *be_spill(
+               const be_node_factory_t *factory,
+               const arch_env_t *arch_env,
+               ir_node *irn)
+{
+  const arch_register_class_t *cls
+               = arch_get_irn_reg_class(arch_env, irn, arch_pos_make_out(0));
+  ir_node *bl    = get_nodes_block(irn);
+  ir_graph *irg  = get_irn_irg(bl);
+  ir_node *spill = new_Spill(factory, cls, irg, bl, irn);
+
+       ir_node *insert;
+
+       /*
+        * search the right insertion point. a spill of a phi cannot be put
+        * directly after the phi, if there are some phis behind the one which
+        * is spilled.
+        */
+       insert = sched_next(irn);
+       while(is_Phi(insert) && !sched_is_end(insert))
+               insert = sched_next(insert);
+
+       sched_add_before(insert, spill);
+  return spill;
+}
+
+ir_node *be_reload(
+               const be_node_factory_t *factory,
+               const arch_env_t *arch_env,
+               const arch_register_class_t *cls,
+               ir_node *irn, int pos, ir_mode *mode, ir_node *spill)
+{
+       ir_node *reload;
+
+  ir_node *bl   = get_nodes_block(irn);
+  ir_graph *irg = get_irn_irg(bl);
+
+       assert(is_Spill(factory, spill)
+                       || (is_Phi(spill) && get_irn_mode(spill) == mode_M));
+
+  reload = new_Reload(factory, cls, irg, bl, mode, spill);
+
+       set_irn_n(irn, pos, reload);
+  sched_add_before(irn, reload);
+  return reload;
+}
+
 /**
  * If the node is a proj, reset the node to the proj's target and return
  * the proj number.
@@ -158,10 +206,12 @@ be_node_get_irn_reg_req(const arch_irn_ops_t *_self,
     arch_register_req_t *req, const ir_node *irn, int pos)
 {
   be_op_t *bo;
+       int index = arch_pos_get_index(pos);
   const be_node_factory_t *factory =
     container_of(_self, const be_node_factory_t, irn_ops);
 
-  pos = arch_pos_make_out(redir_proj(&irn, arch_pos_get_index(pos)));
+       if(arch_pos_is_out(pos))
+               pos = arch_pos_make_out(redir_proj(&irn, index));
 
   bo = pmap_get(factory->irn_op_map, get_irn_op(irn));
 
@@ -169,9 +219,9 @@ be_node_get_irn_reg_req(const arch_irn_ops_t *_self,
     int i;
 
     req->type = arch_register_req_type_normal;
-    req->cls = bo->cls;
+    req->cls  = bo->cls;
 
-    for(i = 0; i < bo->n_pos; ++pos)
+    for(i = 0; i < bo->n_pos; ++i)
       if(pos == bo->pos[i])
         return req;
   }
@@ -228,9 +278,23 @@ arch_irn_class_t be_node_classify(const arch_irn_ops_t *_self, const ir_node *ir
   idx = redir_proj(&irn, 0);
   bo = pmap_get(factory->irn_op_map, get_irn_op(irn));
 
+       switch(bo->kind) {
+#define XXX(a) case node_kind_ ## a: return arch_irn_class_ ## a;
+               XXX(spill)
+               XXX(reload)
+               XXX(perm)
+               XXX(copy)
+#undef XXX
+       }
+
   return 0;
 }
 
+arch_irn_class_t be_node_get_flags(const arch_irn_ops_t *_self, const ir_node *irn)
+{
+       return 0;
+}
+
 static const arch_irn_ops_t *
 be_node_get_irn_ops(const arch_irn_handler_t *_self, const ir_node *irn)
 {
@@ -253,7 +317,7 @@ int is_Spill(const be_node_factory_t *f, const ir_node *irn)
 {
   be_op_t *bo;
   bo = pmap_get(f->irn_op_map, get_irn_op(irn));
-  return bo->kind == node_kind_spill;
+  return bo != NULL && bo->kind == node_kind_spill;
 }
 
 be_node_factory_t *be_node_factory_init(be_node_factory_t *factory,
@@ -273,6 +337,7 @@ be_node_factory_t *be_node_factory_init(be_node_factory_t *factory,
   factory->irn_ops.set_irn_reg     = be_node_set_irn_reg;
   factory->irn_ops.get_irn_reg     = be_node_get_irn_reg;
   factory->irn_ops.classify        = be_node_classify;
+  factory->irn_ops.get_flags       = be_node_get_flags;
 
   for(i = 0, n = isa->get_n_reg_class(); i < n; ++i) {
     const arch_register_class_t *cls = isa->get_reg_class(i);
@@ -281,31 +346,31 @@ be_node_factory_t *be_node_factory_init(be_node_factory_t *factory,
     ent = get_op(factory, cls, node_kind_spill);
     snprintf(buf, sizeof(buf), "Spill_%s", cls->name);
     ent->op = new_ir_op(get_next_ir_opcode(), buf, op_pin_state_pinned,
-        0, 0, oparity_unary, 0);
+        0, oparity_unary, 0, 0);
     ent->n_pos = ARRSIZE(templ_pos_Spill);
     ent->pos = templ_pos_Spill;
     pmap_insert(factory->irn_op_map, ent->op, ent);
 
     ent = get_op(factory, cls, node_kind_reload);
     snprintf(buf, sizeof(buf), "Reload_%s", cls->name);
-    ent->op = new_ir_op(get_next_ir_opcode(), buf, op_pin_state_pinned, 0, 0,
-        oparity_unary, sizeof(const arch_register_t *));
+    ent->op = new_ir_op(get_next_ir_opcode(), buf, op_pin_state_pinned, 0,
+        oparity_unary, 0, sizeof(const arch_register_t *));
     ent->n_pos = ARRSIZE(templ_pos_Reload);
     ent->pos = templ_pos_Reload;
     pmap_insert(factory->irn_op_map, ent->op, ent);
 
     ent = get_op(factory, cls, node_kind_copy);
     snprintf(buf, sizeof(buf), "Copy_%s", cls->name);
-    ent->op = new_ir_op(get_next_ir_opcode(), buf, op_pin_state_pinned, 0, 0,
-        oparity_unary, sizeof(const arch_register_t *));
+    ent->op = new_ir_op(get_next_ir_opcode(), buf, op_pin_state_pinned, 0,
+        oparity_unary, 0, sizeof(const arch_register_t *));
     ent->n_pos = ARRSIZE(templ_pos_Copy);
     ent->pos = templ_pos_Copy;
     pmap_insert(factory->irn_op_map, ent->op, ent);
 
     ent = get_op(factory, cls, node_kind_perm);
     snprintf(buf, sizeof(buf), "Perm_%s", cls->name);
-    ent->op = new_ir_op(get_next_ir_opcode(), buf, op_pin_state_pinned, 0, 0,
-        oparity_variable, sizeof(const arch_register_t) * cls->n_regs);
+    ent->op = new_ir_op(get_next_ir_opcode(), buf, op_pin_state_pinned, 0,
+        oparity_variable, 0, sizeof(const arch_register_t) * cls->n_regs);
     ent->n_pos = 2 * cls->n_regs;
     ent->pos = obstack_alloc(&factory->obst, sizeof(ent->pos[0]) * ent->n_pos);
     for(j = 0; j < ent->n_pos; j += 2) {
@@ -325,16 +390,27 @@ ir_node *insert_Perm_after(const be_main_session_env_t *env,
   const arch_env_t *arch_env = env->main_env->arch_env;
   ir_node *bl = is_Block(pos) ? pos : get_nodes_block(pos);
   ir_graph *irg = get_irn_irg(bl);
-  pset *live_end = get_live_end(bl);
-  pset *live = pset_new_ptr_default();
+  pset *live = put_live_end(bl, pset_new_ptr_default());
   ir_node *curr, *irn, *perm, **nodes;
+  firm_dbg_module_t *dbg = firm_dbg_register("firm.be.node");
   int i, n;
 
-  /* put all live ends in the live set. */
-  for(irn = pset_first(live_end); irn; irn = pset_next(live_end))
-    pset_insert_ptr(live, irn);
+  firm_dbg_set_mask(dbg, DBG_LEVEL);
+  DBG((dbg, LEVEL_1, "Insert Perm after: %+F\n", pos));
 
   sched_foreach_reverse(bl, irn) {
+    ir_node *x;
+
+       /*
+        * If we encounter the node we want to insert the Perm after,
+        * exit immediately, so that this node is still live
+        */
+    if(irn == pos)
+      break;
+
+    DBG((dbg, LEVEL_1, "%+F\n", irn));
+    for(x = pset_first(live); x; x = pset_next(live))
+      DBG((dbg, LEVEL_1, "\tlive: %+F\n", x));
 
     if(arch_irn_has_reg_class(arch_env, irn, arch_pos_make_out(0), cls))
       pset_remove_ptr(live, irn);
@@ -345,27 +421,31 @@ ir_node *insert_Perm_after(const be_main_session_env_t *env,
       if(arch_irn_has_reg_class(arch_env, op, arch_pos_make_out(0), cls))
         pset_insert_ptr(live, op);
     }
-
-    if(sched_prev(irn) == pos)
-      break;
   }
 
   n = pset_count(live);
   nodes = malloc(n * sizeof(nodes[0]));
 
-  for(irn = pset_first(live), i = 0; irn; irn = pset_next(live), i++)
+  DBG((dbg, LEVEL_1, "live:\n"));
+  for(irn = pset_first(live), i = 0; irn; irn = pset_next(live), i++) {
+       DBG((dbg, LEVEL_1, "\t%+F\n", irn));
     nodes[i] = irn;
+  }
 
-  curr = perm = new_Perm(env->main_env->node_factory, cls, irg, bl, n, nodes);
+  perm = new_Perm(env->main_env->node_factory, cls, irg, bl, n, nodes);
   sched_add_after(pos, perm);
   free(nodes);
 
+  curr = perm;
   for(i = 0; i < n; ++i) {
     ir_node *copies[1];
     ir_node *perm_op = get_irn_n(perm, i);
+       const arch_register_t *reg = arch_get_irn_register(arch_env, perm_op, 0);
 
     ir_mode *mode = get_irn_mode(perm_op);
     ir_node *proj = new_r_Proj(irg, bl, perm, mode, i);
+    arch_set_irn_register(arch_env, proj, 0, reg);
+
     sched_add_after(curr, proj);
     curr = proj;