nsz Git - libfirm/blob - ir/ir/irgwalk.c

   1 /* Copyright (C) 1998 - 2000 by Universitaet Karlsruhe
   2 ** All rights reserved.
   3 **
   4 ** Author: Boris Boesler
   5 **
   6 ** traverse an ir graph
   7 ** - execute the pre function before recursion
   8 ** - execute the post function after recursion
   9 */
  10
  11 # include "irnode.h"
  12 # include "irgraph.h" /* visited flag */
  13
  14
  15 void irg_walk_2(ir_node *node,
  16               void (pre)(ir_node*, void*), void (post)(ir_node*, void*),
  17               void *env)
  18 {
  19   int i;
  20
  21   assert(node);
  22   if(get_irn_visited(node) < get_irg_visited(current_ir_graph)) {
  23     set_irn_visited(node, get_irg_visited(current_ir_graph));
  24     if(pre) {
  25       pre(node, env);
  26     }
  27
  28     if (is_no_Block(node))
  29       irg_walk_2(get_nodes_Block(node), pre, post, env);
  30     for(i = get_irn_arity(node) - 1; i >= 0; --i) {
  31       irg_walk_2(get_irn_n(node, i), pre, post, env);
  32     }
  33
  34     if(post)
  35       post(node, env);
  36   }
  37   return;
  38 }
  39
  40
  41 void irg_walk(ir_node *node,
  42               void (pre)(ir_node*, void*), void (post)(ir_node*, void*),
  43               void *env)
  44 {
  45   assert(node);
  46   inc_irg_visited (current_ir_graph);
  47   irg_walk_2(node, pre, post, env);
  48   return;
  49 }
  50
  51 /***************************************************************************/
  52 void irg_block_walk_2(ir_node *node, void (pre)(ir_node*, void*),
  53                       void (post)(ir_node*, void*), void *env)
  54 {
  55   int i;
  56
  57   assert(get_irn_opcode(node) == iro_Block);
  58
  59   if(get_Block_block_visited(node) < get_irg_block_visited(current_ir_graph)) {
  60     set_Block_block_visited(node, get_irg_block_visited(current_ir_graph));
  61
  62     if(pre)
  63       pre(node, env);
  64
  65     for(i = get_Block_n_cfgpreds(node) - 1; i >= 0; --i) {
  66
  67       /* find the corresponding predecessor block. */
  68       ir_node *pred = skip_Proj(get_Block_cfgpred(node, i));
  69       /* GL: I'm not sure whether this assert must go through.  There
  70          could be Id chains?? */
  71       assert(is_cfop(pred) || is_fragile_op(pred));
  72       pred = get_nodes_Block(pred);
  73       assert(get_irn_opcode(pred) == iro_Block);
  74
  75       /* recursion */
  76       irg_block_walk_2(pred, pre, post, env);
  77     }
  78
  79     if(post)
  80       post(node, env);
  81   }
  82   return;
  83 }
  84
  85
  86 /* walks only over Block nodes in the graph.  Has it's own visited
  87    flag, so that it can be interleaved with the other walker.         */
  88 void irg_block_walk(ir_node *node,
  89                     void (pre)(ir_node*, void*), void (post)(ir_node*, void*),
  90                     void *env)
  91 {
  92   assert(node);
  93   inc_irg_block_visited(current_ir_graph);
  94   if (is_no_Block(node)) node = get_nodes_Block(node);
  95   assert(get_irn_opcode(node)  == iro_Block);
  96   irg_block_walk_2(node, pre, post, env);
  97   return;
  98 }