replace and kill survive_dce stuff
[libfirm] / ir / be / beflags.c
index 1e6e330..6ea8dd1 100644 (file)
 #include "beflags.h"
 #include "bearch.h"
 #include "beirg.h"
+#include "beirgmod.h"
 #include "besched.h"
 #include "benode.h"
 #include "belive.h"
 
-static const arch_register_class_t *flag_class = NULL;
-static const arch_register_t       *flags_reg  = NULL;
-static func_rematerialize           remat      = NULL;
+static const arch_register_class_t *flag_class;
+static const arch_register_t       *flags_reg;
+static func_rematerialize           remat;
+static check_modifies_flags         check_modify;
 static int                          changed;
 
 static ir_node *default_remat(ir_node *node, ir_node *after)
@@ -74,6 +76,11 @@ static ir_node *default_remat(ir_node *node, ir_node *after)
        return copy;
 }
 
+static bool default_check_modifies(const ir_node *node)
+{
+       return arch_irn_is(node, modify_flags);
+}
+
 /**
  * tests whether we can legally move node node after node after
  * (only works for nodes in same block)
@@ -81,7 +88,8 @@ static ir_node *default_remat(ir_node *node, ir_node *after)
 static bool can_move(ir_node *node, ir_node *after)
 {
        const ir_edge_t *edge;
-       assert(get_nodes_block(node) == get_nodes_block(after));
+       ir_node *node_block = get_nodes_block(node);
+       assert(node_block == get_nodes_block(after));
 
        /* TODO respect dep edges */
        assert(get_irn_n_edges_kind(node, EDGE_KIND_DEP) == 0);
@@ -94,6 +102,8 @@ static bool can_move(ir_node *node, ir_node *after)
                        assert(get_irn_n_edges_kind(out, EDGE_KIND_DEP) == 0);
                        foreach_out_edge(out, edge2) {
                                ir_node *out2 = get_edge_src_irn(edge2);
+                               if (get_nodes_block(out2) != node_block)
+                                       continue;
                                /* Phi or End represents a usage at block end. */
                                if (is_Phi(out2) || is_End(out2))
                                        continue;
@@ -114,6 +124,8 @@ static bool can_move(ir_node *node, ir_node *after)
                                }
                        }
                } else {
+                       if (get_nodes_block(out) != node_block)
+                               continue;
                        /* phi represents a usage at block end */
                        if (is_Phi(out))
                                continue;
@@ -165,7 +177,7 @@ static void rematerialize_or_move(ir_node *flags_needed, ir_node *node,
                                break;
                        }
                }
-               n = get_irn_link(n);
+               n = (ir_node*)get_irn_link(n);
        } while (n != NULL);
 
        /* No need to introduce the copy, because it only lives in this block, but
@@ -184,11 +196,6 @@ static void rematerialize_or_move(ir_node *flags_needed, ir_node *node,
        }
 }
 
-static bool is_modify_flags(ir_node *node)
-{
-       return arch_irn_is(node, modify_flags);
-}
-
 /**
  * walks up the schedule and makes sure there are no flag-destroying nodes
  * between a flag-consumer -> flag-producer chain. Fixes problematic situations
@@ -223,7 +230,7 @@ static void fix_flags_walker(ir_node *block, void *env)
                if (be_is_Keep(test))
                        test = sched_prev(test);
 
-               if (flags_needed != NULL && is_modify_flags(test)) {
+               if (flags_needed != NULL && check_modify(test)) {
                        /* rematerialize */
                        rematerialize_or_move(flags_needed, node, flag_consumers, pn);
                        flags_needed   = NULL;
@@ -282,14 +289,18 @@ static void fix_flags_walker(ir_node *block, void *env)
 }
 
 void be_sched_fix_flags(ir_graph *irg, const arch_register_class_t *flag_cls,
-                        func_rematerialize remat_func)
+                        func_rematerialize remat_func,
+                        check_modifies_flags check_modifies_flags_func)
 {
-       flag_class = flag_cls;
-       flags_reg  = & flag_class->regs[0];
-       remat      = remat_func;
-       changed    = 0;
+       flag_class   = flag_cls;
+       flags_reg    = & flag_class->regs[0];
+       remat        = remat_func;
+       check_modify = check_modifies_flags_func;
+       changed      = 0;
        if (remat == NULL)
                remat = &default_remat;
+       if (check_modify == NULL)
+               check_modify = &default_check_modifies;
 
        ir_reserve_resources(irg, IR_RESOURCE_IRN_LINK);
        irg_block_walk_graph(irg, fix_flags_walker, NULL, NULL);