Add and fix some comments.

[libfirm] / ir / lower / lower_mux.c

diff --git a/ir/lower/lower_mux.c b/ir/lower/lower_mux.c
index af85428..11de6ae 100644 (file)
--- a/ir/lower/lower_mux.c
+++ b/ir/lower/lower_mux.c
@@ -34,7 +34,7 @@
 #include "irgwalk.h"
 #include "irgmod.h"
 #include "ircons.h"
-#include "irvrfy.h"
+#include "irpass_t.h"
 
 typedef struct walk_env {
        lower_mux_callback *cb_func;
@@ -43,7 +43,7 @@ typedef struct walk_env {
 
 static void find_mux_nodes(ir_node *mux, void *ctx)
 {
-       walk_env_t *env = ctx;
+       walk_env_t *env = (walk_env_t*)ctx;
 
        /* Skip non-mux nodes. */
        if (!is_Mux(mux))
@@ -60,15 +60,18 @@ static void find_mux_nodes(ir_node *mux, void *ctx)
 
 static void lower_mux_node(ir_node* mux)
 {
-       ir_node *upper_block;
-       ir_node *lower_block;
-       ir_node *cond;
-       ir_node *trueProj;
-       ir_node *falseProj;
-       ir_node *falseBlock;
-       ir_node *mux_jmps[2];
-       ir_node *mux_values[2];
-       ir_node *phi;
+       ir_node  *upper_block;
+       ir_node  *lower_block;
+       ir_node  *cond;
+       ir_node  *trueProj;
+       ir_node  *falseProj;
+       ir_node  *falseBlock;
+       ir_node  *mux_jmps[2];
+       ir_node  *mux_values[2];
+       ir_node  *phi;
+       ir_graph *irg;
+
+       irg = get_irn_irg(mux);
 
        /* Split the block in two halfs, with the mux in the upper block. */
        lower_block = get_nodes_block(mux);
@@ -81,9 +84,9 @@ static void lower_mux_node(ir_node* mux)
         * block in-between, so that the phi can be used to select the result
         * value from the old mux node in the lower block. */
        cond        = new_r_Cond(upper_block, get_Mux_sel(mux));
-       trueProj    = new_r_Proj(upper_block, cond, mode_X, pn_Cond_true);
-       falseProj   = new_r_Proj(upper_block, cond, mode_X, pn_Cond_false);
-       falseBlock  = new_r_Block(current_ir_graph, 1, &falseProj);
+       trueProj    = new_r_Proj(cond, mode_X, pn_Cond_true);
+       falseProj   = new_r_Proj(cond, mode_X, pn_Cond_false);
+       falseBlock  = new_r_Block(irg, 1, &falseProj);
        mux_jmps[0] = trueProj;
        mux_jmps[1] = new_r_Jmp(falseBlock);
 
@@ -141,3 +144,28 @@ void lower_mux(ir_graph *irg, lower_mux_callback *cb_func)
        }
        DEL_ARR_F(env.muxes);
 }
+
+typedef struct pass_t {
+       ir_graph_pass_t    pass;
+       lower_mux_callback *cb_func;
+} pass_t;
+
+/**
+ * Wrapper to run ir_lower_mux() as an ir_graph pass
+ */
+static int pass_wrapper(ir_graph *irg, void *context)
+{
+       pass_t *pass = (pass_t*)context;
+
+       lower_mux(irg, pass->cb_func);
+       return 0;
+}
+
+ir_graph_pass_t *lower_mux_pass(const char *name, lower_mux_callback *cb_func)
+{
+       pass_t *pass = XMALLOCZ(pass_t);
+
+       pass->cb_func = cb_func;
+       return def_graph_pass_constructor(
+               &pass->pass, name ? name : "lower_mux", pass_wrapper);
+}