Added support for out edges.

[libfirm] / ir / ana / irdom.c

diff --git a/ir/ana/irdom.c b/ir/ana/irdom.c
index 0ed62d8..930078e 100644 (file)
--- a/ir/ana/irdom.c
+++ b/ir/ana/irdom.c
@@ -118,23 +118,20 @@ ir_node *get_Block_dominated_next(const ir_node *bl)
        return get_dom_info(bl)->next;
 }
 
-void dom_tree_walk(ir_node *n, irg_walk_func *pre,
+void dom_tree_walk(ir_node *bl, irg_walk_func *pre,
                irg_walk_func *post, void *env)
 {
        ir_node *p;
 
-  assert(get_irn_irg(n)->dom_state == dom_consistent
-                       && "The dominators of the irg must be consistent");
-
-       for(p = get_dom_info(p)->first; p; p = get_dom_info(p)->next) {
-               if(pre)
-                       pre(p, env);
+       if(pre)
+               pre(bl, env);
 
+       dominates_for_each(bl, p) {
                dom_tree_walk(p, pre, post, env);
-
-               if(post)
-                       post(p, env);
        }
+
+       if(post)
+               post(bl, env);
 }
 
 void dom_tree_walk_irg(ir_graph *irg, irg_walk_func *pre,
@@ -180,6 +177,7 @@ static void count_and_init_blocks(ir_node *bl, void *env) {
   int *n_blocks = (int *) env;
   (*n_blocks) ++;
 
+       memset(get_dom_info(bl), 0, sizeof(dom_info));
   set_Block_idom(bl, NULL);
   set_Block_pre_num(bl, -1);
   set_Block_dom_depth(bl, -1);