bugfixes
[libfirm] / ir / be / bespill.c
index 00d0dc3..139b488 100644 (file)
@@ -1,5 +1,5 @@
 /**
- * Author:      Daniel Grund
+ * Author:      Daniel Grund, Sebastian Hack
  * Date:               29.09.2005
  * Copyright:   (c) Universitaet Karlsruhe
  * Licence:     This file protected by GPL -  GNU GENERAL PUBLIC LICENSE.
@@ -8,23 +8,77 @@
 #include "pset.h"
 #include "irnode_t.h"
 #include "ircons_t.h"
+#include "iredges_t.h"
+#include "debug.h"
 
 #include "besched.h"
 #include "bespill.h"
 #include "benode_t.h"
 
+typedef struct _reloader_t reloader_t;
+typedef struct _spill_info_t spill_info_t;
+
+struct _reloader_t {
+       reloader_t *next;
+       ir_node *reloader;
+};
+
+struct _spill_info_t {
+       ir_node *spilled_node;
+       reloader_t *reloaders;
+};
+
 typedef struct _spill_ctx_t {
        ir_node *spilled;  /**< The spilled node. */
        ir_node *user;     /**< The node this spill is for. */
        ir_node *spill;    /**< The spill itself. */
 } spill_ctx_t;
 
-int be_set_cmp_spillctx(const void *a, const void *b, size_t n) {
+struct _spill_env_t {
+       firm_dbg_module_t *dbg;
+       const arch_register_class_t *cls;
+       const be_main_session_env_t *session;
+       struct obstack obst;
+       set *spill_ctxs;
+       set *spills;                            /**< all spill_info_t's, which must be placed */
+       pset *mem_phis;                         /**< set of all special spilled phis. allocated and freed seperately */
+       decide_irn_t is_mem_phi;        /**< callback func to decide if a phi needs special spilling */
+       void *data;                                     /**< data passed to all callbacks */
+};
+
+static int cmp_spillctx(const void *a, const void *b, size_t n) {
        const spill_ctx_t *p = a;
        const spill_ctx_t *q = b;
        return !(p->user == q->user && p->spilled == q->spilled);
 }
 
+static int cmp_spillinfo(const void *x, const void *y, size_t size) {
+       const spill_info_t *xx = x;
+       const spill_info_t *yy = y;
+       return ! (xx->spilled_node == yy->spilled_node);
+}
+
+spill_env_t *be_new_spill_env(firm_dbg_module_t *dbg, const be_main_session_env_t *session,
+               const arch_register_class_t *cls, decide_irn_t is_mem_phi, void *data) {
+       spill_env_t *env = malloc(sizeof(env[0]));
+       env->spill_ctxs = new_set(cmp_spillctx, 1024);
+       env->spills     = new_set(cmp_spillinfo, 1024);
+       env->session    = session;
+       env->cls        = cls;
+       env->dbg        = dbg;
+       env->is_mem_phi = is_mem_phi;
+       env->data       = data;
+       obstack_init(&env->obst);
+       return env;
+}
+
+void be_delete_spill_env(spill_env_t *senv) {
+       del_set(senv->spill_ctxs);
+       del_set(senv->spills);
+       obstack_free(&senv->obst, NULL);
+       free(senv);
+}
+
 static spill_ctx_t *be_get_spill_ctx(set *sc, ir_node *to_spill, ir_node *ctx_irn) {
        spill_ctx_t templ;
 
@@ -37,7 +91,7 @@ static spill_ctx_t *be_get_spill_ctx(set *sc, ir_node *to_spill, ir_node *ctx_ir
 
 static ir_node *be_spill_irn(spill_env_t *senv, ir_node *irn, ir_node *ctx_irn) {
        spill_ctx_t *ctx;
-//     DBG((dbg, DBG_SPILL, "spill_irn %+F\n", irn));
+       DBG((senv->dbg, LEVEL_1, "%+F\n", irn));
 
        ctx = be_get_spill_ctx(senv->spill_ctxs, irn, ctx_irn);
        if(!ctx->spill) {
@@ -57,11 +111,11 @@ static ir_node *be_spill_irn(spill_env_t *senv, ir_node *irn, ir_node *ctx_irn)
 static ir_node *be_spill_phi(spill_env_t *senv, ir_node *phi, ir_node *ctx_irn) {
        int i, n = get_irn_arity(phi);
        ir_node **ins, *bl = get_nodes_block(phi);
-       ir_graph *irg = get_irn_irg(bl);
+       ir_graph *irg = senv->session->irg;
        spill_ctx_t *ctx;
 
        assert(is_Phi(phi));
-//     DBG((dbg, DBG_SPILL, "spill_phi %+F\n", phi));
+       DBG((senv->dbg, LEVEL_1, "%+F\n", phi));
 
        /* search an existing spill for this context */
        ctx = be_get_spill_ctx(senv->spill_ctxs, phi, ctx_irn);
@@ -91,7 +145,7 @@ static ir_node *be_spill_phi(spill_env_t *senv, ir_node *phi, ir_node *ctx_irn)
        return ctx->spill;
 }
 
-ir_node *be_spill_node(spill_env_t *senv, ir_node *to_spill) {
+static ir_node *be_spill_node(spill_env_t *senv, ir_node *to_spill) {
        ir_node *res;
        if (pset_find_ptr(senv->mem_phis, to_spill))
                res = be_spill_phi(senv, to_spill, to_spill);
@@ -101,13 +155,109 @@ ir_node *be_spill_node(spill_env_t *senv, ir_node *to_spill) {
        return res;
 }
 
-void be_remove_spilled_phis(spill_env_t *senv) {
+static void phi_walker(ir_node *irn, void *env) {
+       spill_env_t *senv = env;
+       const arch_env_t *arch = senv->session->main_env->arch_env;
+
+       if (is_Phi(irn) && arch_irn_has_reg_class(arch, irn, 0, senv->cls)
+                       && senv->is_mem_phi(irn, senv->data)) {
+               DBG((senv->dbg, LEVEL_1, "  %+F\n", irn));
+               pset_insert_ptr(senv->mem_phis, irn);
+       }
+}
+
+void be_insert_spills_reloads(spill_env_t *senv, pset *reload_set) {
+       ir_graph *irg = senv->session->irg;
        ir_node *irn;
-       /* remove all special phis form the irg and the schedule */
+       spill_info_t *si;
+       struct obstack ob;
+
+       obstack_init(&ob);
+
+       /* get all special spilled phis */
+       DBG((senv->dbg, LEVEL_1, "Mem-phis:\n"));
+       senv->mem_phis = pset_new_ptr_default();
+       irg_walk_graph(senv->session->irg, phi_walker, NULL, senv);
+
+       /* Add reloads for mem_phis */
+       DBG((senv->dbg, LEVEL_1, "Reloads for mem-phis:\n"));
+       for(irn = pset_first(senv->mem_phis); irn; irn = pset_next(senv->mem_phis)) {
+               const ir_edge_t *e;
+               DBG((senv->dbg, LEVEL_1, " Mem-phi %+F\n", irn));
+               foreach_out_edge(irn, e) {
+                       ir_node *user = e->src;
+                       if (is_Phi(user) && !pset_find_ptr(senv->mem_phis, user)) {
+                                       DBG((senv->dbg, LEVEL_1, " non-mem-phi user %+F\n", user));
+                                       ir_node *use_bl = get_nodes_block(user);
+                                       be_add_reload_on_edge(senv, irn, use_bl, e->pos);
+                       }
+               }
+       }
+
+       /* process each spilled node */
+       DBG((senv->dbg, LEVEL_1, "Insert spills and reloads:\n"));
+       for(si = set_first(senv->spills); si; si = set_next(senv->spills)) {
+               reloader_t *rld;
+               ir_node **reloads;
+               int n_reloads = 0;
+               ir_mode *mode = get_irn_mode(si->spilled_node);
+
+               /* go through all reloads for this spill */
+               for(rld = si->reloaders; rld; rld = rld->next) {
+                       /* the spill for this reloader */
+                       ir_node *spill   = be_spill_node(senv, si->spilled_node);
+
+                       /* the reload */
+                       ir_node *bl      = is_Block(rld->reloader) ? rld->reloader : get_nodes_block(rld->reloader);
+                       ir_node *reload  = new_Reload(senv->session->main_env->node_factory,
+                                                                                                               senv->cls, irg, bl, mode, spill);
+
+                       DBG((senv->dbg, LEVEL_1, " %+F of %+F before %+F\n", reload, si->spilled_node, rld->reloader));
+                       if(reload_set)
+                               pset_insert_ptr(reload_set, reload);
+
+                       /* remember the reaload */
+                       obstack_ptr_grow(&ob, reload);
+                       sched_add_before(rld->reloader, reload);
+                       n_reloads++;
+               }
+
+               assert(n_reloads > 0);
+               reloads = obstack_finish(&ob);
+               be_introduce_copies_ignore(senv->session->dom_front, si->spilled_node,
+                               n_reloads, reloads, senv->mem_phis);
+               obstack_free(&ob, reloads);
+       }
+
+       obstack_free(&ob, NULL);
+
        for(irn = pset_first(senv->mem_phis); irn; irn = pset_next(senv->mem_phis)) {
                int i, n;
                for(i = 0, n = get_irn_arity(irn); i < n; ++i)
                        set_irn_n(irn, i, new_r_Bad(senv->session->irg));
                sched_remove(irn);
        }
+
+       del_pset(senv->mem_phis);
+}
+
+void be_add_reload(spill_env_t *senv, ir_node *to_spill, ir_node *before) {
+       spill_info_t templ, *res;
+       reloader_t *rel;
+
+       assert(get_irn_opcode(to_spill) != iro_Unknown);
+
+       templ.spilled_node = to_spill;
+       templ.reloaders    = NULL;
+       res = set_insert(senv->spills, &templ, sizeof(templ), HASH_PTR(to_spill));
+
+       rel           = obstack_alloc(&senv->obst, sizeof(rel[0]));
+       rel->reloader = before;
+       rel->next     = res->reloaders;
+       res->reloaders = rel;
+}
+
+void be_add_reload_on_edge(spill_env_t *senv, ir_node *to_spill, ir_node *bl, int pos) {
+       ir_node *insert_bl = get_irn_arity(bl) == 1 ? sched_first(bl) : get_Block_cfgpred_block(bl, pos);
+       be_add_reload(senv, to_spill, insert_bl);
 }