[libfirm] / ir / opt / code_placement.c

/*
 * Copyright (C) 1995-2008 University of Karlsruhe.  All right reserved.
 *
 * This file is part of libFirm.
 *
 * This file may be distributed and/or modified under the terms of the
 * GNU General Public License version 2 as published by the Free Software
 * Foundation and appearing in the file LICENSE.GPL included in the
 * packaging of this file.
 *
 * Licensees holding valid libFirm Professional Edition licenses may use
 * this file in accordance with the libFirm Commercial License.
 * Agreement provided with the Software.
 *
 * This file is provided AS IS with NO WARRANTY OF ANY KIND, INCLUDING THE
 * WARRANTY OF DESIGN, MERCHANTABILITY AND FITNESS FOR A PARTICULAR
 * PURPOSE.
 */

/**
 * @file
 * @brief    Move nodes to a block where they will be executed the least
 *           often.
 * @author   Christian Schaefer, Goetz Lindenmaier, Sebastian Felis,
 *           Michael Beck
 * @version  $Id$
 *
 * The idea here is to push nodes as deep into the dominance tree as their
 * dependencies allow. After pushing them back up out of as many loops as
 * possible.
 */
#include "config.h"

#include <stdbool.h>

#include "iroptimize.h"
#include "adt/pdeq.h"
#include "irnode_t.h"
#include "irouts.h"
#include "irgopt.h"
#include "irpass.h"

static bool is_block_reachable(ir_node *block)
{
        return get_Block_dom_depth(block) >= 0;
}

/* mark node as not-visited */
static void clear_irn_visited(ir_node *node)
{
        ir_graph    *irg     = get_irn_irg(node);
        ir_visited_t visited = get_irg_visited(irg);
        set_irn_visited(node, visited-1);
}

/**
 * Find the earliest correct block for node n.  --- Place n into the
 * same Block as its dominance-deepest Input.
 *
 * move_out_of_loops() expects that place_floats_early() have placed
 * all "living" nodes into a living block. That's why we must
 * move nodes in dead block with "live" successors into a valid
 * block.
 * We move them just into the same block as its successor (or
 * in case of a Phi into the effective use block). For Phi successors,
 * this may still be a dead block, but then there is no real use, as
 * the control flow will be dead later.
 */
static void place_floats_early(ir_node *n, waitq *worklist)
{
        int       i;
        int       arity;
        ir_node  *block;
        int       new_depth;
        ir_graph *irg;
        ir_node  *start_block;
        ir_node  *new_block;

        /* we must not run into an infinite loop */
        if (irn_visited_else_mark(n))
                return;

        /* The algorithm relies on the fact that all predecessors of a block are
         * moved up after a call to place_float_early of the predecessors
         * (see the loop below).
         * However we have to break cycles somewhere. Relying on the visited flag
         * will result in nodes not being moved up despite their place_floats_early
         * call.
         * Instead we break cycles at pinned nodes which won't move anyway:
         * This works because in firm each cycle contains a Phi or Block node
         * (which are pinned)
         */
        if (get_irn_pinned(n) != op_pin_state_floats) {
                /* we can't move pinned nodes */
                arity = get_irn_arity(n);
                for (i = 0; i < arity; ++i) {
                        ir_node *pred = get_irn_n(n, i);
                        pdeq_putr(worklist, pred);
                }
                if (!is_Block(n))
                        pdeq_putr(worklist, get_nodes_block(n));
                return;
        }

        block = get_nodes_block(n);
        /* do not move unreachable code (or its predecessors around) since dominance
         * is inalid there */
        if (!is_block_reachable(block))
                return;

        /* first move predecessors up */
        arity = get_irn_arity(n);
        place_floats_early(block, worklist);
        for (i = 0; i < arity; ++i) {
                ir_node *pred       = get_irn_n(n, i);
                ir_node *pred_block = get_nodes_block(pred);

                /* gcse can lead to predecessors of reachable code being unreachable.
                 * Move them into the current block in this case */
                if (!is_block_reachable(pred_block)) {
                        ir_node *new_pred_block = block;
                        assert(get_irn_pinned(pred) == op_pin_state_floats);
                        if (is_Phi(n)) {
                                new_pred_block = get_Block_cfgpred_block(block, i);
                        }
                        set_nodes_block(pred, new_pred_block);
                        clear_irn_visited(pred);
                }
                place_floats_early(pred, worklist);
        }

        /* determine earliest point */
        new_block = NULL;
        new_depth = 0;
        for (i = 0; i < arity; ++i) {
                ir_node *pred       = get_irn_n(n, i);
                ir_node *pred_block = get_nodes_block(pred);
                int      pred_depth = get_Block_dom_depth(pred_block);
                if (pred_depth > new_depth) {
                        new_depth = pred_depth;
                        new_block = pred_block;
                }
        }

        /* avoid moving nodes into the start block if we are not in the backend */
        irg         = get_irn_irg(n);
        start_block = get_irg_start_block(irg);
        if (new_block == start_block && block != start_block &&
            get_irg_phase_state(irg) != phase_backend) {
                assert(get_Block_n_cfg_outs(start_block) == 1);
                new_block = get_Block_cfg_out(start_block, 0);
        }

        /* Set the new block */
        if (new_block != NULL)
                set_nodes_block(n, new_block);
}

/**
 * Floating nodes form subgraphs that begin at nodes as Const, Load,
 * Start, Call and that end at op_pin_state_pinned nodes as Store, Call.
 * Place_early places all floating nodes reachable from its argument through
 * floating nodes and adds all beginnings at op_pin_state_pinned nodes to the
 * worklist.
 *
 * @param worklist   a worklist, used for the algorithm, empty on in/output
 */
static void place_early(ir_graph *irg, waitq *worklist)
{
        assert(worklist);
        inc_irg_visited(irg);

        /* this inits the worklist */
        place_floats_early(get_irg_end(irg), worklist);

        /* Work the content of the worklist. */
        while (!waitq_empty(worklist)) {
                ir_node *n = (ir_node*)waitq_get(worklist);
                if (!irn_visited(n))
                        place_floats_early(n, worklist);
        }
        set_irg_pinned(irg, op_pin_state_pinned);
}

/**
 * Compute the deepest common dominator tree ancestor of block and dca.
 *
 * @param dca    the deepest common dominator tree ancestor so far,
 *               might be NULL
 * @param block  a block
 *
 * @return  the deepest common dominator tree ancestor of block and dca
 */
static ir_node *calc_dom_dca(ir_node *dca, ir_node *block)
{
        assert(block != NULL);

        /* We found a first legal placement. */
        if (!dca)
                return block;

        /* Find a placement that is dominates both, dca and block. */
        while (get_Block_dom_depth(block) > get_Block_dom_depth(dca))
                block = get_Block_idom(block);

        while (get_Block_dom_depth(dca) > get_Block_dom_depth(block)) {
                dca = get_Block_idom(dca);
        }

        while (block != dca) {
                block = get_Block_idom(block); dca = get_Block_idom(dca);
        }
        return dca;
}

/**
 * Deepest common dominance ancestor of DCA and CONSUMER of PRODUCER.
 * I.e., DCA is the block where we might place PRODUCER.
 * A data flow edge points from producer to consumer.
 */
static ir_node *consumer_dom_dca(ir_node *dca, ir_node *consumer,
                                 ir_node *producer)
{
        /* Compute the last block into which we can place a node so that it is
           before consumer. */
        if (is_Phi(consumer)) {
                /* our consumer is a Phi-node, the effective use is in all those
                   blocks through which the Phi-node reaches producer */
                ir_node *phi_block = get_nodes_block(consumer);
                int      arity     = get_irn_arity(consumer);
                int      i;

                for (i = 0;  i < arity; i++) {
                        if (get_Phi_pred(consumer, i) == producer) {
                                ir_node *new_block = get_Block_cfgpred_block(phi_block, i);
                                if (is_Bad(new_block))
                                        continue;

                                if (is_block_reachable(new_block))
                                        dca = calc_dom_dca(dca, new_block);
                        }
                }
        } else {
                dca = calc_dom_dca(dca, get_nodes_block(consumer));
        }
        return dca;
}

static inline int get_block_loop_depth(ir_node *block)
{
        return get_loop_depth(get_irn_loop(block));
}

/**
 * Move n to a block with less loop depth than its current block. The
 * new block must be dominated by early.
 *
 * @param n      the node that should be moved
 * @param early  the earliest block we can n move to
 */
static void move_out_of_loops(ir_node *n, ir_node *early)
{
        ir_node *block      = get_nodes_block(n);
        ir_node *best       = block;
        int      best_depth = get_block_loop_depth(best);

        /* Find the region deepest in the dominator tree dominating
           dca with the least loop nesting depth, but still dominated
           by our early placement. */
        while (block != early) {
                ir_node *idom       = get_Block_idom(block);
                int      idom_depth = get_block_loop_depth(idom);
                if (idom_depth < best_depth) {
                        best       = idom;
                        best_depth = idom_depth;
                }
                block = idom;
        }
        if (best != get_nodes_block(n))
                set_nodes_block(n, best);
}

/**
 * Calculate the deepest common ancestor in the dominator tree of all nodes'
 * blocks depending on node; our final placement has to dominate DCA.
 *
 * @param node  the definition node
 * @param dca   the deepest common ancestor block so far, initially
 *              NULL
 *
 * @return the deepest common dominator ancestor of all blocks of node's users
 */
static ir_node *get_deepest_common_dom_ancestor(ir_node *node, ir_node *dca)
{
        int i;

        for (i = get_irn_n_outs(node) - 1; i >= 0; --i) {
                ir_node *succ = get_irn_out(node, i);

                /* keepalive edges are special and don't respect the dominance */
                if (is_End(succ))
                        continue;

                if (is_Proj(succ)) {
                        /* Proj nodes are in the same block as node, so
                         * the users of Proj are our users. */
                        dca = get_deepest_common_dom_ancestor(succ, dca);
                } else {
                        /* ignore successors in unreachable code */
                        ir_node *succ_blk = get_nodes_block(succ);
                        if (!is_block_reachable(succ_blk))
                                continue;
                        dca = consumer_dom_dca(dca, succ, node);
                }
        }
        return dca;
}

/**
 * Put all the Proj nodes of a node into a given block.
 *
 * @param node   the mode_T node
 * @param block  the block to put the Proj nodes to
 */
static void set_projs_block(ir_node *node, ir_node *block)
{
        int i;

        for (i = get_irn_n_outs(node) - 1; i >= 0; --i) {
                ir_node *succ = get_irn_out(node, i);

                assert(is_Proj(succ));

                if (get_irn_mode(succ) == mode_T) {
                        set_projs_block(succ, block);
                }
                set_nodes_block(succ, block);
        }
}

/**
 * Find the latest legal block for N and place N into the
 * `optimal' Block between the latest and earliest legal block.
 * The `optimal' block is the dominance-deepest block of those
 * with the least loop-nesting-depth.  This places N out of as many
 * loops as possible and then makes it as control dependent as
 * possible.
 */
static void place_floats_late(ir_node *n, pdeq *worklist)
{
        int      n_outs;
        int      i;
        ir_node *block;
        ir_node *dca;

        if (irn_visited_else_mark(n))
                return;

        n_outs = get_irn_n_outs(n);
        /* break cycles at pinned nodes (see place place_floats_early) as to why */
        if (get_irn_pinned(n) != op_pin_state_floats) {
                for (i = 0; i < n_outs; ++i) {
                        ir_node *succ = get_irn_out(n, i);
                        pdeq_putr(worklist, succ);
                }
                return;
        }

        /* place our users */
        for (i = 0; i < n_outs; ++i) {
                ir_node *succ = get_irn_out(n, i);
                place_floats_late(succ, worklist);
        }

        /* no point in moving Projs around, they are moved with their predecessor */
        if (is_Proj(n))
                return;
        /* some nodes should simply stay in the startblock */
        if (is_irn_start_block_placed(n)) {
#ifndef NDEBUG
                ir_graph *irg         = get_irn_irg(n);
                ir_node  *start_block = get_irg_start_block(irg);
                assert(get_nodes_block(n) == start_block);
#endif
                return;
        }

        /* don't move unreachable code around */
        block = get_nodes_block(n);
        if (!is_block_reachable(block))
                return;

        /* deepest common ancestor in the dominator tree of all nodes'
           blocks depending on us; our final placement has to dominate
           DCA. */
        dca = get_deepest_common_dom_ancestor(n, NULL);
        if (dca != NULL) {
                set_nodes_block(n, dca);
                move_out_of_loops(n, block);
                if (get_irn_mode(n) == mode_T) {
                        set_projs_block(n, get_nodes_block(n));
                }
        }
}

/**
 * Place floating nodes on the given worklist as late as possible using
 * the dominance tree.
 *
 * @param worklist   the worklist containing the nodes to place
 */
static void place_late(ir_graph *irg, waitq *worklist)
{
        assert(worklist);
        inc_irg_visited(irg);

        /* This fills the worklist initially. */
        place_floats_late(get_irg_start_block(irg), worklist);

        /* And now empty the worklist again... */
        while (!waitq_empty(worklist)) {
                ir_node *n = (ir_node*)waitq_get(worklist);
                if (!irn_visited(n))
                        place_floats_late(n, worklist);
        }
}

/* Code Placement. */
void place_code(ir_graph *irg)
{
        waitq *worklist;

        remove_critical_cf_edges(irg);

        /* Handle graph state */
        assert(get_irg_phase_state(irg) != phase_building);
        assure_irg_outs(irg);
        assure_doms(irg);
        assure_cf_loop(irg);

        /* Place all floating nodes as early as possible. This guarantees
         a legal code placement. */
        worklist = new_waitq();
        place_early(irg, worklist);

        /* Note: place_early changes only blocks, no data edges. So, the
         * data out edges are still valid, no need to recalculate them here. */

        /* Now move the nodes down in the dominator tree. This reduces the
           unnecessary executions of the node. */
        place_late(irg, worklist);

        set_irg_outs_inconsistent(irg);
        set_irg_loopinfo_inconsistent(irg);
        del_waitq(worklist);
}

/**
 * Wrapper for place_code() inside the place_code pass.
 */
static void place_code_wrapper(ir_graph *irg)
{
        set_opt_global_cse(1);
        optimize_graph_df(irg);
        place_code(irg);
        set_opt_global_cse(0);
}

ir_graph_pass_t *place_code_pass(const char *name)
{
        return def_graph_pass(name ? name : "place", place_code_wrapper);
}