X-Git-Url: http://nsz.repo.hu/git/?a=blobdiff_plain;f=ir%2Fbe%2Fbespillslots.c;h=913a67ea9aabc67498fccad091689b0003e10793;hb=f8dfdaaff8e4f364da8494fd35a9bb2f8c267e7f;hp=da9b18c6cc3e9cb08a48724275e77585252b1538;hpb=5cfbace431416dbbf03ec25b6ef16bc3bf5f91d9;p=libfirm

diff --git a/ir/be/bespillslots.c b/ir/be/bespillslots.c
index da9b18c6c..913a67ea9 100644
--- a/ir/be/bespillslots.c
+++ b/ir/be/bespillslots.c
@@ -37,49 +37,52 @@
 #include "unionfind.h"
 #include "irdump_t.h"
 
-#include "benode_t.h"
+#include "benode.h"
 #include "besched.h"
+#include "bespill.h"
 #include "bespillslots.h"
 #include "bechordal_t.h"
 #include "bestatevent.h"
-#include "bespilloptions.h"
 #include "bemodule.h"
 #include "beintlive_t.h"
-#include "beirg_t.h"
-#include "bearch_t.h"
+#include "beirg.h"
+#include "bearch.h"
 
-#define DBG_COALESCING		1
-#define DBG_INTERFERENCES	2
+#define DBG_COALESCING      1
+#define DBG_INTERFERENCES   2
 
 DEBUG_ONLY(static firm_dbg_module_t *dbg = NULL;)
 
-typedef struct _spill_t {
-	ir_node *spill;
-	const ir_mode *mode;     /**< mode of the spilled value */
-	int alignment;           /**< alignment for the spilled value */
-	int spillslot;           /**< index into spillslot_unionfind structure */
+typedef struct spill_t {
+	ir_node       *spill;
+	const ir_mode *mode;      /**< mode of the spilled value */
+	int            alignment; /**< alignment for the spilled value */
+	int            spillslot; /**< index into spillslot_unionfind structure */
 } spill_t;
 
-typedef struct _affinity_edge_t {
+typedef struct affinity_edge_t {
 	double affinity;
-	int slot1, slot2;
+	int    slot1;
+	int    slot2;
 } affinity_edge_t;
 
-struct _be_fec_env_t {
-	struct obstack obst;
-	const arch_env_t *arch_env;
-	be_irg_t *birg;
-	set *spills;
-	ir_node **reloads;
-	affinity_edge_t **affinity_edges;
-	set *memperms;
+struct be_fec_env_t {
+	struct obstack         obst;
+	ir_graph              *irg;
+	set                   *spills;
+	ir_node              **reloads;
+	affinity_edge_t      **affinity_edges;
+	set                   *memperms;
+	set_frame_entity_func  set_frame_entity;
+	bool                   at_begin;  /**< frame entities should be allocate at
+	                                       the beginning of the stackframe */
 };
 
 /** Compare 2 affinity edges (used in quicksort) */
 static int cmp_affinity(const void *d1, const void *d2)
 {
-	const affinity_edge_t * const *e1 = d1;
-	const affinity_edge_t * const *e2 = d2;
+	const affinity_edge_t * const *e1 = (const affinity_edge_t**)d1;
+	const affinity_edge_t * const *e2 = (const affinity_edge_t**)d2;
 
 	/* sort in descending order */
 	return (*e1)->affinity < (*e2)->affinity ? 1 : -1;
@@ -87,8 +90,8 @@ static int cmp_affinity(const void *d1, const void *d2)
 
 static int cmp_spill(const void* d1, const void* d2, size_t size)
 {
-	const spill_t* s1 = d1;
-	const spill_t* s2 = d2;
+	const spill_t* s1 = (const spill_t*)d1;
+	const spill_t* s2 = (const spill_t*)d2;
 	(void) size;
 
 	return s1->spill != s2->spill;
@@ -100,7 +103,7 @@ static spill_t *get_spill(be_fec_env_t *env, ir_node *node)
 	int hash = hash_irn(node);
 
 	spill.spill = node;
-	res = set_find(env->spills, &spill, sizeof(spill), hash);
+	res = (spill_t*)set_find(env->spills, &spill, sizeof(spill), hash);
 
 	return res;
 }
@@ -111,9 +114,9 @@ static inline ir_node *get_memory_edge(const ir_node *node)
 	int i, arity;
 
 	arity = get_irn_arity(node);
-	for(i = arity - 1; i >= 0; --i) {
+	for (i = arity - 1; i >= 0; --i) {
 		ir_node *arg = get_irn_n(node, i);
-		if(get_irn_mode(arg) == mode_M)
+		if (get_irn_mode(arg) == mode_M)
 			return arg;
 	}
 
@@ -128,13 +131,13 @@ static spill_t *collect_spill(be_fec_env_t *env, ir_node *node,
 
 	/* insert into set of spills if not already there */
 	spill.spill = node;
-	res         = set_find(env->spills, &spill, sizeof(spill), hash);
+	res         = (spill_t*)set_find(env->spills, &spill, sizeof(spill), hash);
 
-	if(res == NULL) {
+	if (res == NULL) {
 		spill.spillslot = set_count(env->spills);
 		spill.mode      = mode;
 		spill.alignment = align;
-		res             = set_insert(env->spills, &spill, sizeof(spill), hash);
+		res             = (spill_t*)set_insert(env->spills, &spill, sizeof(spill), hash);
 		DB((dbg, DBG_COALESCING, "Slot %d: %+F\n", spill.spillslot, node));
 	} else {
 		assert(res->mode == mode);
@@ -150,13 +153,13 @@ static spill_t *collect_memphi(be_fec_env_t *env, ir_node *node,
 	int i, arity;
 	spill_t spill, *res;
 	int hash = hash_irn(node);
-	const ir_exec_freq *exec_freq = be_get_birg_exec_freq(env->birg);
+	const ir_exec_freq *exec_freq = be_get_irg_exec_freq(env->irg);
 
 	assert(is_Phi(node));
 
 	spill.spill = node;
-	res = set_find(env->spills, &spill, sizeof(spill), hash);
-	if(res != NULL) {
+	res = (spill_t*)set_find(env->spills, &spill, sizeof(spill), hash);
+	if (res != NULL) {
 		assert(res->mode == mode);
 		assert(res->alignment == align);
 		return res;
@@ -166,23 +169,23 @@ static spill_t *collect_memphi(be_fec_env_t *env, ir_node *node,
 	spill.mode      = mode;
 	spill.alignment = align;
 	DB((dbg, DBG_COALESCING, "Slot %d: %+F\n", spill.spillslot, node));
-	res             = set_insert(env->spills, &spill, sizeof(spill), hash);
+	res             = (spill_t*)set_insert(env->spills, &spill, sizeof(spill), hash);
 
 	/* collect attached spills and mem-phis */
 	arity = get_irn_arity(node);
-	for(i = 0; i < arity; ++i) {
+	for (i = 0; i < arity; ++i) {
 		affinity_edge_t *affinty_edge;
 		ir_node *arg = get_irn_n(node, i);
 		spill_t *arg_spill;
 
-		if(is_Phi(arg)) {
+		if (is_Phi(arg)) {
 			arg_spill = collect_memphi(env, arg, mode, align);
 		} else {
 			arg_spill = collect_spill(env, arg, mode, align);
 		}
 
 		/* add an affinity edge */
-		affinty_edge           = obstack_alloc(&env->obst, sizeof(affinty_edge[0]));
+		affinty_edge           = OALLOC(&env->obst, affinity_edge_t);
 		affinty_edge->affinity = get_block_execfreq(exec_freq, get_nodes_block(arg));
 		affinty_edge->slot1    = res->spillslot;
 		affinty_edge->slot2    = arg_spill->spillslot;
@@ -196,15 +199,14 @@ void be_node_needs_frame_entity(be_fec_env_t *env, ir_node *node,
                                 const ir_mode *mode, int align)
 {
 	ir_node *spillnode = get_memory_edge(node);
-	spill_t *spill;
 
 	assert(spillnode != NULL);
 
 	/* walk upwards and collect all phis and spills on this way */
 	if (is_Phi(spillnode)) {
-		spill = collect_memphi(env, spillnode, mode, align);
+		collect_memphi(env, spillnode, mode, align);
 	} else {
-		spill = collect_spill(env, spillnode, mode, align);
+		collect_spill(env, spillnode, mode, align);
 	}
 
 	ARR_APP1(ir_node *, env->reloads, node);
@@ -222,7 +224,7 @@ static int merge_interferences(be_fec_env_t *env, bitset_t** interferences,
 	/* merge spillslots and interferences */
 	res = uf_union(spillslot_unionfind, s1, s2);
 	/* we assume that we always merge s2 to s1 so swap s1, s2 if necessary */
-	if(res != 0) {
+	if (res != s1) {
 		int t = s1;
 		s1 = s2;
 		s2 = t;
@@ -232,25 +234,25 @@ static int merge_interferences(be_fec_env_t *env, bitset_t** interferences,
 
 	/* update other interferences */
 	spillcount = set_count(env->spills);
-	for(i = 0; i < spillcount; ++i) {
+	for (i = 0; i < spillcount; ++i) {
 		bitset_t *intfs = interferences[i];
-		if(bitset_is_set(intfs, s2))
+		if (bitset_is_set(intfs, s2))
 			bitset_set(intfs, s1);
 	}
 
 	return res;
 }
 
-static int my_values_interfere2(be_irg_t *birg, const ir_node *a,
+static int my_values_interfere2(ir_graph *irg, const ir_node *a,
                                 const ir_node *b)
 {
-	be_lv_t *lv = be_get_birg_liveness(birg);
+	be_lv_t *lv = be_get_irg_liveness(irg);
 
     int a2b = _value_dominates(a, b);
     int b2a = _value_dominates(b, a);
 
     /* If there is no dominance relation, they do not interfere. */
-    if((a2b | b2a) > 0) {
+    if ((a2b | b2a) > 0) {
         const ir_edge_t *edge;
         ir_node *bb;
 
@@ -258,7 +260,7 @@ static int my_values_interfere2(be_irg_t *birg, const ir_node *a,
          * Adjust a and b so, that a dominates b if
          * a dominates b or vice versa.
          */
-        if(b2a) {
+        if (b2a) {
             const ir_node *t = a;
             a = b;
             b = t;
@@ -270,7 +272,7 @@ static int my_values_interfere2(be_irg_t *birg, const ir_node *a,
          * If a is live end in b's block it is
          * live at b's definition (a dominates b)
          */
-        if(be_is_live_end(lv, bb, a))
+        if (be_is_live_end(lv, bb, a))
             return 1;
 
         /*
@@ -286,17 +288,17 @@ static int my_values_interfere2(be_irg_t *birg, const ir_node *a,
          */
         foreach_out_edge(a, edge) {
             const ir_node *user = get_edge_src_irn(edge);
-			if(is_Sync(user)) {
+			if (is_Sync(user)) {
 				const ir_edge_t *edge2;
 				foreach_out_edge(user, edge2) {
 					const ir_node *user2 = get_edge_src_irn(edge2);
 					assert(!is_Sync(user2));
-					if(get_nodes_block(user2) == bb && !is_Phi(user2) &&
+					if (get_nodes_block(user2) == bb && !is_Phi(user2) &&
 					   _value_strictly_dominates(b, user2))
 						return 1;
 				}
 			} else {
-				if(get_nodes_block(user) == bb && !is_Phi(user) &&
+				if (get_nodes_block(user) == bb && !is_Phi(user) &&
 						_value_strictly_dominates(b, user))
                 return 1;
 			}
@@ -309,28 +311,28 @@ static int my_values_interfere2(be_irg_t *birg, const ir_node *a,
 /**
  * same as values_interfere but with special handling for Syncs
  */
-static int my_values_interfere(be_irg_t *birg, ir_node *a, ir_node *b)
+static int my_values_interfere(ir_graph *irg, ir_node *a, ir_node *b)
 {
-	if(is_Sync(a)) {
+	if (is_Sync(a)) {
 		int i, arity = get_irn_arity(a);
-		for(i = 0; i < arity; ++i) {
+		for (i = 0; i < arity; ++i) {
 			ir_node *in = get_irn_n(a, i);
-			if(my_values_interfere(birg, in, b))
+			if (my_values_interfere(irg, in, b))
 				return 1;
 		}
 		return 0;
-	} else if(is_Sync(b)) {
+	} else if (is_Sync(b)) {
 		int i, arity = get_irn_arity(b);
-		for(i = 0; i < arity; ++i) {
+		for (i = 0; i < arity; ++i) {
 			ir_node *in = get_irn_n(b, i);
 			/* a is not a sync, so no need for my_values_interfere */
-			if(my_values_interfere2(birg, a, in))
+			if (my_values_interfere2(irg, a, in))
 				return 1;
 		}
 		return 0;
 	}
 
-	return my_values_interfere2(birg, a, b);
+	return my_values_interfere2(irg, a, b);
 }
 
 /**
@@ -348,31 +350,35 @@ static void do_greedy_coalescing(be_fec_env_t *env)
 	int affinity_edge_count;
 	bitset_t **interferences;
 	int* spillslot_unionfind;
+	struct obstack data;
 
 	spillcount = set_count(env->spills);
-	if(spillcount == 0)
+	if (spillcount == 0)
 		return;
 
+	obstack_init(&data);
+
 	DB((dbg, DBG_COALESCING, "Coalescing %d spillslots\n", spillcount));
 
-	interferences       = ALLOCAN(bitset_t*, spillcount);
-	spillslot_unionfind = ALLOCAN(int,       spillcount);
-	spilllist           = ALLOCAN(spill_t*,  spillcount);
+	interferences       = OALLOCN(&data, bitset_t*, spillcount);
+	spillslot_unionfind = OALLOCN(&data, int,       spillcount);
+	spilllist           = OALLOCN(&data, spill_t*,  spillcount);
 
 	uf_init(spillslot_unionfind, spillcount);
 
 	DEBUG_ONLY(
 		memset(spilllist, 0, spillcount * sizeof(spilllist[0]));
-	);
+	)
 
-	for(spill = set_first(env->spills), i = 0; spill != NULL;
-	    spill = set_next(env->spills), ++i) {
+	i = 0;
+	foreach_set(env->spills, spill_t*, spill) {
 		assert(spill->spillslot < spillcount);
 		spilllist[spill->spillslot] = spill;
+		++i;
 	}
 
-	for(i = 0; i < spillcount; ++i) {
-		interferences[i] = bitset_alloca(spillcount);
+	for (i = 0; i < spillcount; ++i) {
+		interferences[i] = bitset_obstack_alloc(&data, spillcount);
 	}
 
 	/* construct interferences */
@@ -382,13 +388,13 @@ static void do_greedy_coalescing(be_fec_env_t *env)
 		if (is_NoMem(spill1))
 			continue;
 
-		for(i2 = i+1; i2 < spillcount; ++i2) {
+		for (i2 = i+1; i2 < spillcount; ++i2) {
 			ir_node *spill2 = spilllist[i2]->spill;
 
 			if (is_NoMem(spill2))
 				continue;
 
-			if (my_values_interfere(env->birg, spill1, spill2)) {
+			if (my_values_interfere(env->irg, spill1, spill2)) {
 				DB((dbg, DBG_INTERFERENCES,
 				     "Slot %d and %d interfere\n", i, i2));
 
@@ -406,7 +412,7 @@ static void do_greedy_coalescing(be_fec_env_t *env)
 	/*dump_interference_graph(env, interferences, "before"); */
 
 	/* try to merge affine nodes */
-	for(i = 0; i < affinity_edge_count; ++i) {
+	for (i = 0; i < affinity_edge_count; ++i) {
 		const affinity_edge_t *edge = env->affinity_edges[i];
 		int s1 = uf_find(spillslot_unionfind, edge->slot1);
 		int s2 = uf_find(spillslot_unionfind, edge->slot2);
@@ -424,21 +430,21 @@ static void do_greedy_coalescing(be_fec_env_t *env)
 	}
 
 	/* try to merge as much remaining spillslots as possible */
-	for(i = 0; i < spillcount; ++i) {
+	for (i = 0; i < spillcount; ++i) {
 		int s1 = uf_find(spillslot_unionfind, i);
-		if(s1 != i)
+		if (s1 != i)
 			continue;
 
-		for(i2 = i+1; i2 < spillcount; ++i2) {
+		for (i2 = i+1; i2 < spillcount; ++i2) {
 			int s2 = uf_find(spillslot_unionfind, i2);
-			if(s2 != i2)
+			if (s2 != i2)
 				continue;
 
 			/* test if values interfere
 			 * we have to test n1-n2 and n2-n1, because only 1 side gets updated
 			 * when node merging occurs
 			 */
-			if(bitset_is_set(interferences[s1], s2)) {
+			if (bitset_is_set(interferences[s1], s2)) {
 				assert(bitset_is_set(interferences[s2], s1));
 				continue;
 			}
@@ -446,7 +452,7 @@ static void do_greedy_coalescing(be_fec_env_t *env)
 			DB((dbg, DBG_COALESCING,
 			     "Merging %d and %d because it is possible\n", s1, s2));
 
-			if(merge_interferences(env, interferences, spillslot_unionfind, s1, s2) != 0) {
+			if (merge_interferences(env, interferences, spillslot_unionfind, s1, s2) != 0) {
 				/* we can break the loop here, because s2 is the new supernode
 				 * now and we'll test s2 again later anyway */
 				break;
@@ -455,32 +461,33 @@ static void do_greedy_coalescing(be_fec_env_t *env)
 	}
 
 	/* assign spillslots to spills */
-	for(i = 0; i < spillcount; ++i) {
+	for (i = 0; i < spillcount; ++i) {
 		spill_t *spill = spilllist[i];
 
 		spill->spillslot = uf_find(spillslot_unionfind, i);
 	}
 
 	/*dump_interference_graph(env, interferences, "after");*/
+	obstack_free(&data, 0);
 }
 
 
 
-typedef struct _spill_slot_t {
+typedef struct spill_slot_t {
 	int size;
 	int align;
 	ir_entity *entity;
 } spill_slot_t;
 
-typedef struct _memperm_entry_t {
+typedef struct memperm_entry_t {
 	ir_node* node;
 	int pos;
 	ir_entity *in;
 	ir_entity *out;
-	struct _memperm_entry_t *next;
+	struct memperm_entry_t *next;
 } memperm_entry_t;
 
-typedef struct _memperm_t {
+typedef struct memperm_t {
 	ir_node *block;
 	int entrycount;
 	memperm_entry_t *entries;
@@ -488,8 +495,8 @@ typedef struct _memperm_t {
 
 static int cmp_memperm(const void* d1, const void* d2, size_t size)
 {
-	const memperm_t* e1 = d1;
-	const memperm_t* e2 = d2;
+	const memperm_t* e1 = (const memperm_t*)d1;
+	const memperm_t* e2 = (const memperm_t*)d2;
 	(void) size;
 
 	return e1->block != e2->block;
@@ -503,12 +510,12 @@ static memperm_t *get_memperm(be_fec_env_t *env, ir_node *block)
 	entry.block = block;
 	hash        = hash_irn(block);
 
-	res = set_find(env->memperms, &entry, sizeof(entry), hash);
+	res = (memperm_t*)set_find(env->memperms, &entry, sizeof(entry), hash);
 
-	if(res == NULL) {
+	if (res == NULL) {
 		entry.entrycount = 0;
 		entry.entries = NULL;
-		res = set_insert(env->memperms, &entry, sizeof(entry), hash);
+		res = (memperm_t*)set_insert(env->memperms, &entry, sizeof(entry), hash);
 	}
 
 	return res;
@@ -516,17 +523,10 @@ static memperm_t *get_memperm(be_fec_env_t *env, ir_node *block)
 
 static ir_entity* create_stack_entity(be_fec_env_t *env, spill_slot_t *slot)
 {
-	ir_graph *irg = be_get_birg_irg(env->birg);
-	ir_type *frame = get_irg_frame_type(irg);
-	/* TODO: backend should be able to specify wether we want spill slots
-	 * at begin or end of frame */
-	int        at_start = 1;
-	ir_entity *res = frame_alloc_area(frame, slot->size, slot->align, at_start);
-
-	/* adjust size of the entity type... */
-	ir_type *enttype = get_entity_type(res);
-	set_type_size_bytes(enttype, slot->size);
-
+	ir_graph  *irg   = env->irg;
+	ir_type   *frame = get_irg_frame_type(irg);
+	ir_entity *res   = frame_alloc_area(frame, slot->size, slot->align,
+	                                    env->at_begin);
 	slot->entity = res;
 
 	return res;
@@ -538,41 +538,42 @@ static ir_entity* create_stack_entity(be_fec_env_t *env, spill_slot_t *slot)
  */
 static void enlarge_spillslot(spill_slot_t *slot, int otheralign, int othersize)
 {
-	if(othersize > slot->size) {
+	if (othersize > slot->size) {
 		slot->size = othersize;
 	}
-	if(otheralign > slot->align) {
-		if(otheralign % slot->align != 0)
+	if (otheralign > slot->align) {
+		if (otheralign % slot->align != 0)
 			slot->align *= otheralign;
 		else
 			slot->align = otheralign;
-	} else if(slot->align % otheralign != 0) {
+	} else if (slot->align % otheralign != 0) {
 		slot->align *= otheralign;
 	}
 }
 
-
-static void assign_spill_entity(ir_node *node, ir_entity *entity)
+static void assign_spill_entity(be_fec_env_t *env,
+                                ir_node *node, ir_entity *entity)
 {
-	if(is_NoMem(node))
+	if (is_NoMem(node))
 		return;
-	if(is_Sync(node)) {
+	if (is_Sync(node)) {
 		int i, arity;
 
 		arity = get_irn_arity(node);
-		for(i = 0; i < arity; ++i) {
+		for (i = 0; i < arity; ++i) {
 			ir_node *in = get_irn_n(node, i);
 			assert(!is_Phi(in));
 
-			assign_spill_entity(in, entity);
+			assign_spill_entity(env, in, entity);
 		}
 		return;
 	}
 
-	/* beware: we might have Stores with Memory Proj's, ia32 fisttp for instance */
+	/* beware: we might have Stores with Memory Proj's, ia32 fisttp for
+	   instance */
 	node = skip_Proj(node);
 	assert(arch_get_frame_entity(node) == NULL);
-	arch_set_frame_entity(node, entity);
+	env->set_frame_entity(node, entity);
 }
 
 /**
@@ -584,19 +585,17 @@ static void assign_spillslots(be_fec_env_t *env)
 	int           spillcount = set_count(env->spills);
 	spill_slot_t *spillslots = ALLOCANZ(spill_slot_t, spillcount);
 	spill_t      *spill;
-	int           i;
+	size_t        i;
 
 	/* construct spillslots */
-	for(spill = set_first(env->spills); spill != NULL;
-		spill = set_next(env->spills)) {
-
+	foreach_set(env->spills, spill_t*, spill) {
 		int slotid = spill->spillslot;
 		const ir_mode *mode = spill->mode;
 		spill_slot_t *slot = & (spillslots[slotid]);
 		int size = get_mode_size_bytes(mode);
 		int align = spill->alignment;
 
-		if(slot->align == 0 && slot->size == 0) {
+		if (slot->align == 0 && slot->size == 0) {
 			slot->align = align;
 			slot->size = size;
 		} else {
@@ -604,26 +603,24 @@ static void assign_spillslots(be_fec_env_t *env)
 		}
 	}
 
-	for(spill = set_first(env->spills); spill != NULL;
-	    spill = set_next(env->spills)) {
-
+	foreach_set(env->spills, spill_t*, spill) {
 		ir_node      *node   = spill->spill;
 		int           slotid = spill->spillslot;
 		spill_slot_t *slot;
 
 		slot = &spillslots[slotid];
-		if(slot->entity == NULL) {
+		if (slot->entity == NULL) {
 			create_stack_entity(env, slot);
 		}
 
-		if(is_Phi(node)) {
+		if (is_Phi(node)) {
 			int i, arity;
 			ir_node *block = get_nodes_block(node);
 
 			/* should be a PhiM */
-			assert(is_Phi(node));
+			assert(get_irn_mode(node) == mode_M);
 
-			for(i = 0, arity = get_irn_arity(node); i < arity; ++i) {
+			for (i = 0, arity = get_irn_arity(node); i < arity; ++i) {
 				ir_node *arg = get_irn_n(node, i);
 				ir_node *predblock = get_Block_cfgpred_block(block, i);
 				spill_t *argspill;
@@ -633,17 +630,17 @@ static void assign_spillslots(be_fec_env_t *env)
 				assert(argspill != NULL);
 
 				argslotid = argspill->spillslot;
-				if(slotid != argslotid) {
+				if (slotid != argslotid) {
 					memperm_t *memperm;
 					memperm_entry_t *entry;
 					spill_slot_t *argslot = &spillslots[argslotid];
-					if(argslot->entity == NULL) {
+					if (argslot->entity == NULL) {
 						create_stack_entity(env, argslot);
 					}
 
 					memperm = get_memperm(env, predblock);
 
-					entry = obstack_alloc(&env->obst, sizeof(entry[0]));
+					entry = OALLOC(&env->obst, memperm_entry_t);
 					entry->node = node;
 					entry->pos = i;
 					entry->in = argslot->entity;
@@ -654,11 +651,11 @@ static void assign_spillslots(be_fec_env_t *env)
 				}
 			}
 		} else {
-			assign_spill_entity(node, slot->entity);
+			assign_spill_entity(env, node, slot->entity);
 		}
 	}
 
-	for(i = 0; i < ARR_LEN(env->reloads); ++i) {
+	for (i = 0; i < ARR_LEN(env->reloads); ++i) {
 		ir_node            *reload    = env->reloads[i];
 		ir_node            *spillnode = get_memory_edge(reload);
 		spill_t            *spill     = get_spill(env, spillnode);
@@ -666,7 +663,7 @@ static void assign_spillslots(be_fec_env_t *env)
 
 		assert(slot->entity != NULL);
 
-		arch_set_frame_entity(reload, slot->entity);
+		env->set_frame_entity(reload, slot->entity);
 	}
 }
 
@@ -676,15 +673,15 @@ static void assign_spillslots(be_fec_env_t *env)
 static ir_node *get_end_of_block_insertion_point(ir_node* block)
 {
 	ir_node* ins = sched_last(block);
-	while(is_Proj(ins) && get_irn_mode(ins) == mode_X) {
+	while (is_Proj(ins) && get_irn_mode(ins) == mode_X) {
 		ins = sched_prev(ins);
 		assert(ins != NULL);
 	}
 
-	if(is_cfop(ins)) {
-		while(1) {
+	if (is_cfop(ins)) {
+		for (;;) {
 			ir_node *prev = sched_prev(ins);
-			if(!is_cfop(prev))
+			if (!is_cfop(prev))
 				break;
 			ins = prev;
 		}
@@ -695,11 +692,9 @@ static ir_node *get_end_of_block_insertion_point(ir_node* block)
 
 static void create_memperms(be_fec_env_t *env)
 {
-	const arch_env_t *arch_env = env->arch_env;
-	ir_graph *irg = be_get_birg_irg(env->birg);
 	memperm_t *memperm;
 
-	for(memperm = set_first(env->memperms); memperm != NULL; memperm = set_next(env->memperms)) {
+	foreach_set(env->memperms, memperm_t*, memperm) {
 		ir_node         **nodes = ALLOCAN(ir_node*, memperm->entrycount);
 		memperm_entry_t  *entry;
 		ir_node          *blockend;
@@ -708,13 +703,13 @@ static void create_memperms(be_fec_env_t *env)
 
 		assert(memperm->entrycount > 0);
 
-		for(entry = memperm->entries, i = 0; entry != NULL; entry = entry->next, ++i) {
+		for (entry = memperm->entries, i = 0; entry != NULL; entry = entry->next, ++i) {
 			ir_node* arg = get_irn_n(entry->node, entry->pos);
 			nodes[i] = arg;
 		}
 
-		mempermnode = be_new_MemPerm(arch_env, irg, memperm->block,
-		                             memperm->entrycount, nodes);
+		mempermnode = be_new_MemPerm(memperm->block, memperm->entrycount,
+		                             nodes);
 
 		/* insert node into schedule */
 		blockend = get_end_of_block_insertion_point(memperm->block);
@@ -722,14 +717,13 @@ static void create_memperms(be_fec_env_t *env)
 		stat_ev_dbl("mem_perm", memperm->entrycount);
 
 		i = 0;
-		for(entry = memperm->entries; entry != NULL; entry = entry->next, ++i) {
+		for (entry = memperm->entries; entry != NULL; entry = entry->next, ++i) {
 			ir_node *proj;
 			ir_node* arg = get_irn_n(entry->node, entry->pos);
 
 			be_set_MemPerm_in_entity(mempermnode, i, entry->in);
 			be_set_MemPerm_out_entity(mempermnode, i, entry->out);
-			set_irg_current_block(irg, memperm->block);
-			proj = new_Proj(mempermnode, get_irn_mode(arg), i);
+			proj = new_r_Proj(mempermnode, get_irn_mode(arg), i);
 
 			set_irn_n(entry->node, entry->pos, proj);
 		}
@@ -744,10 +738,9 @@ static int count_spillslots(const be_fec_env_t *env)
 	int slotcount;
 
 	slotcount = 0;
-	for(spill = set_first(env->spills); spill != NULL;
-	    spill = set_next(env->spills)) {
+	foreach_set(env->spills, spill_t*, spill) {
 		int spillslot = spill->spillslot;
-		if(!bitset_is_set(counted, spillslot)) {
+		if (!bitset_is_set(counted, spillslot)) {
 			slotcount++;
 			bitset_set(counted, spillslot);
 		}
@@ -756,16 +749,14 @@ static int count_spillslots(const be_fec_env_t *env)
 	return slotcount;
 }
 
-be_fec_env_t *be_new_frame_entity_coalescer(be_irg_t *birg)
+be_fec_env_t *be_new_frame_entity_coalescer(ir_graph *irg)
 {
-	const arch_env_t *arch_env = birg->main_env->arch_env;
-	be_fec_env_t     *env      = XMALLOC(be_fec_env_t);
+	be_fec_env_t *env = XMALLOCZ(be_fec_env_t);
 
-	be_liveness_assure_chk(be_assure_liveness(birg));
+	be_liveness_assure_chk(be_assure_liveness(irg));
 
 	obstack_init(&env->obst);
-	env->arch_env       = arch_env;
-	env->birg           = birg;
+	env->irg            = irg;
 	env->spills         = new_set(cmp_spill, 10);
 	env->reloads        = NEW_ARR_F(ir_node*, 0);
 	env->affinity_edges = NEW_ARR_F(affinity_edge_t*, 0);
@@ -785,11 +776,16 @@ void be_free_frame_entity_coalescer(be_fec_env_t *env)
 	free(env);
 }
 
-void be_assign_entities(be_fec_env_t *env)
+void be_assign_entities(be_fec_env_t *env,
+                        set_frame_entity_func set_frame_entity,
+                        bool alloc_entities_at_begin)
 {
+	env->set_frame_entity = set_frame_entity;
+	env->at_begin         = alloc_entities_at_begin;
+
 	stat_ev_dbl("spillslots", set_count(env->spills));
 
-	if(be_coalesce_spill_slots) {
+	if (be_coalesce_spill_slots) {
 		do_greedy_coalescing(env);
 	}
 
@@ -800,46 +796,8 @@ void be_assign_entities(be_fec_env_t *env)
 	create_memperms(env);
 }
 
-/**
- * This walker function searches for reloads and collects all the spills
- * and memphis attached to them.
- */
-static void collect_spills_walker(ir_node *node, void *data)
-{
-	be_fec_env_t *env = data;
-	const ir_mode *mode;
-	const arch_register_class_t *cls;
-	int align;
-
-	/* classify returns classification of the irn the proj is attached to */
-	if (is_Proj(node))
-		return;
-
-	if (!arch_irn_class_is(node, reload))
-		return;
-
-	mode  = get_irn_mode(node);
-	cls   = arch_get_irn_reg_class_out(node);
-	align = arch_env_get_reg_class_alignment(env->arch_env, cls);
-
-	be_node_needs_frame_entity(env, node, mode, align);
-}
-
-void be_coalesce_spillslots(be_irg_t *birg)
-{
-	be_fec_env_t *env = be_new_frame_entity_coalescer(birg);
-
-	/* collect reloads */
-	irg_walk_graph(birg->irg, NULL, collect_spills_walker, env);
-
-	be_assign_entities(env);
-
-	be_free_frame_entity_coalescer(env);
-}
-
+BE_REGISTER_MODULE_CONSTRUCTOR(be_init_spillslots)
 void be_init_spillslots(void)
 {
 	FIRM_DBG_REGISTER(dbg, "firm.be.spillslots");
 }
-
-BE_REGISTER_MODULE_CONSTRUCTOR(be_init_spillslots);