44a310b88f802c3c51109d61f4a2a706321bcae3
[libfirm] / ir / be / belower.c
1 /*
2  * Copyright (C) 1995-2008 University of Karlsruhe.  All right reserved.
3  *
4  * This file is part of libFirm.
5  *
6  * This file may be distributed and/or modified under the terms of the
7  * GNU General Public License version 2 as published by the Free Software
8  * Foundation and appearing in the file LICENSE.GPL included in the
9  * packaging of this file.
10  *
11  * Licensees holding valid libFirm Professional Edition licenses may use
12  * this file in accordance with the libFirm Commercial License.
13  * Agreement provided with the Software.
14  *
15  * This file is provided AS IS with NO WARRANTY OF ANY KIND, INCLUDING THE
16  * WARRANTY OF DESIGN, MERCHANTABILITY AND FITNESS FOR A PARTICULAR
17  * PURPOSE.
18  */
19
20 /**
21  * @file
22  * @brief       Performs lowering of perm nodes. Inserts copies to assure register constraints.
23  * @author      Christian Wuerdig
24  * @date        14.12.2005
25  * @version     $Id$
26  */
27 #ifdef HAVE_CONFIG_H
28 #include "config.h"
29 #endif
30
31 #include <stdlib.h>
32
33 #include "ircons.h"
34 #include "debug.h"
35 #include "irhooks.h"
36 #include "xmalloc.h"
37 #include "irnodeset.h"
38 #include "irgmod.h"
39 #include "iredges_t.h"
40 #include "irgwalk.h"
41 #include "array_t.h"
42
43 #include "bearch_t.h"
44 #include "belower.h"
45 #include "benode_t.h"
46 #include "besched_t.h"
47 #include "bestat.h"
48 #include "bessaconstr.h"
49 #include "beintlive_t.h"
50
51 #undef KEEP_ALIVE_COPYKEEP_HACK
52
53 /** Associates an op with it's copy and CopyKeep. */
54 typedef struct {
55         ir_node *op;         /**< an irn which must be different */
56         ir_nodeset_t copies; /**< all non-spillable copies of this irn */
57         const arch_register_class_t *cls;
58 } op_copy_assoc_t;
59
60 /** Environment for constraints. */
61 typedef struct {
62         be_irg_t       *birg;
63         pset           *op_set;
64         struct obstack obst;
65         DEBUG_ONLY(firm_dbg_module_t *dbg;)
66 } constraint_env_t;
67
68 /** Lowering walker environment. */
69 typedef struct _lower_env_t {
70         be_irg_t         *birg;
71         const arch_env_t *arch_env;
72         unsigned          do_copy : 1;
73         DEBUG_ONLY(firm_dbg_module_t *dbg_module;)
74 } lower_env_t;
75
76 /** Holds a Perm register pair. */
77 typedef struct _reg_pair_t {
78         const arch_register_t *in_reg;    /**< a perm IN register */
79         ir_node               *in_node;   /**< the in node to which the register belongs */
80
81         const arch_register_t *out_reg;   /**< a perm OUT register */
82         ir_node               *out_node;  /**< the out node to which the register belongs */
83
84         int                    checked;   /**< indicates whether the pair was check for cycle or not */
85 } reg_pair_t;
86
87 typedef enum _perm_type_t {
88         PERM_CYCLE,
89         PERM_CHAIN,
90         PERM_SWAP,
91         PERM_COPY
92 } perm_type_t;
93
94 /** Structure to represent cycles or chains in a Perm. */
95 typedef struct _perm_cycle_t {
96         const arch_register_t **elems;       /**< the registers in the cycle */
97         int                     n_elems;     /**< number of elements in the cycle */
98         perm_type_t             type;        /**< type (CHAIN or CYCLE) */
99 } perm_cycle_t;
100
101 /** Compare the two operands. */
102 static int cmp_op_copy_assoc(const void *a, const void *b) {
103         const op_copy_assoc_t *op1 = a;
104         const op_copy_assoc_t *op2 = b;
105
106         return op1->op != op2->op;
107 }
108
109 /** Compare the in registers of two register pairs. */
110 static int compare_reg_pair(const void *a, const void *b) {
111         const reg_pair_t *pair_a = a;
112         const reg_pair_t *pair_b = b;
113
114         if (pair_a->in_reg->index > pair_b->in_reg->index)
115                 return 1;
116         else
117                 return -1;
118 }
119
120 /** returns the number register pairs marked as checked. */
121 static int get_n_checked_pairs(reg_pair_t *pairs, int n) {
122         int i, n_checked = 0;
123
124         for (i = 0; i < n; i++) {
125                 if (pairs[i].checked)
126                         n_checked++;
127         }
128
129         return n_checked;
130 }
131
132 /**
133  * Gets the node corresponding to a register from an array of register pairs.
134  * NOTE: The given registers pairs and the register to look for must belong
135  *       to the same register class.
136  *
137  * @param pairs  The array of register pairs
138  * @param n      The number of pairs
139  * @param reg    The register to look for
140  * @param in_out 0 == look for IN register, 1 == look for OUT register
141  * @return The corresponding node or NULL if not found
142  */
143 static ir_node *get_node_for_register(reg_pair_t *pairs, int n, const arch_register_t *reg, int in_out) {
144         int i;
145
146         if (in_out) {
147                 for (i = 0; i < n; i++) {
148                         /* out register matches */
149                         if (pairs[i].out_reg->index == reg->index)
150                                 return pairs[i].out_node;
151                 }
152         }
153         else {
154                 for (i = 0; i < n; i++) {
155                         /* in register matches */
156                         if (pairs[i].in_reg->index == reg->index)
157                                 return pairs[i].in_node;
158                 }
159         }
160
161         return NULL;
162 }
163
164 /**
165  * Gets the index in the register pair array where the in/out register
166  * corresponds to reg_idx.
167  *
168  * @param pairs  The array of register pairs
169  * @param n      The number of pairs
170  * @param reg    The register index to look for
171  * @param in_out 0 == look for IN register, 1 == look for OUT register
172  * @return The corresponding index in pairs or -1 if not found
173  */
174 static int get_pairidx_for_regidx(reg_pair_t *pairs, int n, int reg_idx, int in_out) {
175         int i;
176
177         if (in_out) {
178                 for (i = 0; i < n; i++) {
179                         /* out register matches */
180                         if ((int) pairs[i].out_reg->index == reg_idx)
181                                 return i;
182                 }
183         }
184         else {
185                 for (i = 0; i < n; i++) {
186                         /* in register matches */
187                         if ((int) pairs[i].in_reg->index == reg_idx)
188                                 return i;
189                 }
190         }
191
192         return -1;
193 }
194
195 /**
196  * Gets an array of register pairs and tries to identify a cycle or chain starting
197  * at position start.
198  *
199  * @param cycle Variable to hold the cycle
200  * @param pairs Array of register pairs
201  * @param start Index to start
202  * @return The cycle or chain
203  */
204 static perm_cycle_t *get_perm_cycle(perm_cycle_t *cycle, reg_pair_t *pairs, int n, int start) {
205         int head         = pairs[start].in_reg->index;
206         int cur_idx      = pairs[start].out_reg->index;
207         int cur_pair_idx = start;
208         int n_pairs_done = get_n_checked_pairs(pairs, n);
209         int idx;
210         perm_type_t cycle_tp = PERM_CYCLE;
211
212         /* We could be right in the middle of a chain, so we need to find the start */
213         while (head != cur_idx) {
214                 /* goto previous register in cycle or chain */
215                 cur_pair_idx = get_pairidx_for_regidx(pairs, n, head, 1);
216
217                 if (cur_pair_idx < 0) {
218                         cycle_tp = PERM_CHAIN;
219                         break;
220                 }
221                 else {
222                         head  = pairs[cur_pair_idx].in_reg->index;
223                         start = cur_pair_idx;
224                 }
225         }
226
227         /* assume worst case: all remaining pairs build a cycle or chain */
228         cycle->elems    = xcalloc((n - n_pairs_done) * 2, sizeof(cycle->elems[0]));
229         cycle->n_elems  = 2;  /* initial number of elements is 2 */
230         cycle->elems[0] = pairs[start].in_reg;
231         cycle->elems[1] = pairs[start].out_reg;
232         cycle->type     = cycle_tp;
233         cur_idx         = pairs[start].out_reg->index;
234
235         idx = 2;
236         /* check for cycle or end of a chain */
237         while (cur_idx != head) {
238                 /* goto next register in cycle or chain */
239                 cur_pair_idx = get_pairidx_for_regidx(pairs, n, cur_idx, 0);
240
241                 if (cur_pair_idx < 0)
242                         break;
243
244                 cur_idx = pairs[cur_pair_idx].out_reg->index;
245
246                 /* it's not the first element: insert it */
247                 if (cur_idx != head) {
248                         cycle->elems[idx++] = pairs[cur_pair_idx].out_reg;
249                         cycle->n_elems++;
250                 }
251                 else {
252                         /* we are there where we started -> CYCLE */
253                         cycle->type = PERM_CYCLE;
254                 }
255         }
256
257         /* mark all pairs having one in/out register with cycle in common as checked */
258         for (idx = 0; idx < cycle->n_elems; idx++) {
259                 cur_pair_idx = get_pairidx_for_regidx(pairs, n, cycle->elems[idx]->index, 0);
260
261                 if (cur_pair_idx >= 0)
262                         pairs[cur_pair_idx].checked = 1;
263
264                 cur_pair_idx = get_pairidx_for_regidx(pairs, n, cycle->elems[idx]->index, 1);
265
266                 if (cur_pair_idx >= 0)
267                         pairs[cur_pair_idx].checked = 1;
268         }
269
270         return cycle;
271 }
272
273 /**
274  * Lowers a perm node.  Resolves cycles and creates a bunch of
275  * copy and swap operations to permute registers.
276  * Note: The caller of this function has to make sure, that irn
277  *       is a Perm node.
278  *
279  * @param irn      The perm node
280  * @param block    The block the perm node belongs to
281  * @param walk_env The environment
282  */
283 static void lower_perm_node(ir_node *irn, void *walk_env) {
284         ir_graph        *irg = get_irn_irg(irn);
285         const arch_register_class_t *reg_class;
286         const arch_env_t            *arch_env;
287         lower_env_t     *env         = walk_env;
288         int             real_size    = 0;
289         int             keep_perm    = 0;
290         int             n, i, pn, do_copy, j, n_ops;
291         reg_pair_t      *pairs;
292         const ir_edge_t *edge;
293         perm_cycle_t    *cycle;
294         ir_node         *sched_point, *block, *in[2];
295         ir_node         *arg1, *arg2, *res1, *res2;
296         ir_node         *cpyxchg = NULL;
297         DEBUG_ONLY(firm_dbg_module_t *mod;)
298
299         arch_env = env->arch_env;
300         do_copy  = env->do_copy;
301         DEBUG_ONLY(mod = env->dbg_module;)
302         block    = get_nodes_block(irn);
303
304         /*
305                 Get the schedule predecessor node to the perm
306                 NOTE: This works with auto-magic. If we insert the
307                         new copy/exchange nodes after this node, everything
308                         should be ok.
309         */
310         sched_point = sched_prev(irn);
311         DBG((mod, LEVEL_1, "perm: %+F\n", irn));
312         DBG((mod, LEVEL_1, "sched point is %+F\n", sched_point));
313         assert(sched_point && "Perm is not scheduled or has no predecessor");
314
315         n = get_irn_arity(irn);
316         assert(n == get_irn_n_edges(irn) && "perm's in and out numbers different");
317
318         reg_class = arch_get_irn_register(arch_env, get_irn_n(irn, 0))->reg_class;
319         pairs     = alloca(n * sizeof(pairs[0]));
320
321         /* build the list of register pairs (in, out) */
322         i = 0;
323         foreach_out_edge(irn, edge) {
324                 pairs[i].out_node = get_edge_src_irn(edge);
325                 pn                = get_Proj_proj(pairs[i].out_node);
326                 pairs[i].in_node  = get_irn_n(irn, pn);
327
328                 pairs[i].in_reg  = arch_get_irn_register(arch_env, pairs[i].in_node);
329                 pairs[i].out_reg = arch_get_irn_register(arch_env, pairs[i].out_node);
330
331                 pairs[i].checked = 0;
332                 i++;
333         }
334
335         /* sort the register pairs by the indices of the in registers */
336         qsort(pairs, n, sizeof(pairs[0]), compare_reg_pair);
337
338         /* Mark all equal pairs as checked, and exchange the OUT proj with
339                 the IN node. */
340         for (i = 0; i < n; i++) {
341                 if (pairs[i].in_reg->index == pairs[i].out_reg->index) {
342                         DBG((mod, LEVEL_1, "%+F removing equal perm register pair (%+F, %+F, %s)\n",
343                                 irn, pairs[i].in_node, pairs[i].out_node, pairs[i].out_reg->name));
344
345                         /* We have to check for a special case:
346                                 The in-node could be a Proj from a Perm. In this case,
347                                 we need to correct the projnum */
348                         if (be_is_Perm(pairs[i].in_node) && is_Proj(pairs[i].in_node)) {
349                                 set_Proj_proj(pairs[i].out_node, get_Proj_proj(pairs[i].in_node));
350                         }
351
352                         /* reroute the edges from the proj to the argument */
353                         exchange(pairs[i].out_node, pairs[i].in_node);
354                         //edges_reroute(pairs[i].out_node, pairs[i].in_node, env->birg->irg);
355                         //set_irn_n(pairs[i].out_node, 0, new_Bad());
356
357                         pairs[i].checked = 1;
358                 }
359         }
360
361         /* Set do_copy to 0 if it's on but we have no free register */
362         if (do_copy) {
363                 do_copy = 0;
364         }
365
366         real_size = n - get_n_checked_pairs(pairs, n);
367
368         /* check for cycles and chains */
369         while (get_n_checked_pairs(pairs, n) < n) {
370                 i = n_ops = 0;
371
372                 /* go to the first not-checked pair */
373                 while (pairs[i].checked) i++;
374                 cycle = xcalloc(1, sizeof(*cycle));
375                 cycle = get_perm_cycle(cycle, pairs, n, i);
376
377                 DB((mod, LEVEL_1, "%+F: following %s created:\n  ", irn, cycle->type == PERM_CHAIN ? "chain" : "cycle"));
378                 for (j = 0; j < cycle->n_elems; j++) {
379                         DB((mod, LEVEL_1, " %s", cycle->elems[j]->name));
380                 }
381                 DB((mod, LEVEL_1, "\n"));
382
383                 /*
384                         We don't need to do anything if we have a Perm with two
385                         elements which represents a cycle, because those nodes
386                         already represent exchange nodes
387                 */
388                 if (n == 2 && cycle->type == PERM_CYCLE) {
389                         free(cycle);
390                         keep_perm = 1;
391                         continue;
392                 }
393
394 //TODO: - iff PERM_CYCLE && do_copy -> determine free temp reg and insert copy to/from it before/after
395 //        the copy cascade (this reduces the cycle into a chain)
396
397                 /* build copy/swap nodes from back to front */
398                 for (i = cycle->n_elems - 2; i >= 0; i--) {
399                         arg1 = get_node_for_register(pairs, n, cycle->elems[i], 0);
400                         arg2 = get_node_for_register(pairs, n, cycle->elems[i + 1], 0);
401
402                         res1 = get_node_for_register(pairs, n, cycle->elems[i], 1);
403                         res2 = get_node_for_register(pairs, n, cycle->elems[i + 1], 1);
404                         /*
405                                 If we have a cycle and don't copy: we need to create exchange nodes
406                                 NOTE: An exchange node is a perm node with 2 INs and 2 OUTs
407                                 IN_1  = in node with register i
408                                 IN_2  = in node with register i + 1
409                                 OUT_1 = out node with register i + 1
410                                 OUT_2 = out node with register i
411                         */
412                         if (cycle->type == PERM_CYCLE && !do_copy) {
413                                 in[0] = arg1;
414                                 in[1] = arg2;
415
416                                 /* At this point we have to handle the following problem:     */
417                                 /*                                                            */
418                                 /* If we have a cycle with more than two elements, then       */
419                                 /* this could correspond to the following Perm node:          */
420                                 /*                                                            */
421                                 /*   +----+   +----+   +----+                                 */
422                                 /*   | r1 |   | r2 |   | r3 |                                 */
423                                 /*   +-+--+   +-+--+   +--+-+                                 */
424                                 /*     |        |         |                                   */
425                                 /*     |        |         |                                   */
426                                 /*   +-+--------+---------+-+                                 */
427                                 /*   |         Perm         |                                 */
428                                 /*   +-+--------+---------+-+                                 */
429                                 /*     |        |         |                                   */
430                                 /*     |        |         |                                   */
431                                 /*   +-+--+   +-+--+   +--+-+                                 */
432                                 /*   |Proj|   |Proj|   |Proj|                                 */
433                                 /*   | r2 |   | r3 |   | r1 |                                 */
434                                 /*   +----+   +----+   +----+                                 */
435                                 /*                                                            */
436                                 /* This node is about to be split up into two 2x Perm's       */
437                                 /* for which we need 4 Proj's and the one additional Proj     */
438                                 /* of the first Perm has to be one IN of the second. So in    */
439                                 /* general we need to create one additional Proj for each     */
440                                 /* "middle" Perm and set this to one in node of the successor */
441                                 /* Perm.                                                      */
442
443                                 DBG((mod, LEVEL_1, "%+F creating exchange node (%+F, %s) and (%+F, %s) with\n",
444                                         irn, arg1, cycle->elems[i]->name, arg2, cycle->elems[i + 1]->name));
445                                 DBG((mod, LEVEL_1, "%+F                        (%+F, %s) and (%+F, %s)\n",
446                                         irn, res1, cycle->elems[i]->name, res2, cycle->elems[i + 1]->name));
447
448                                 cpyxchg = be_new_Perm(reg_class, irg, block, 2, in);
449                                 n_ops++;
450
451                                 if (i > 0) {
452                                         /* cycle is not done yet */
453                                         int pidx = get_pairidx_for_regidx(pairs, n, cycle->elems[i]->index, 0);
454
455                                         /* create intermediate proj */
456                                         res1 = new_r_Proj(irg, block, cpyxchg, get_irn_mode(res1), 0);
457
458                                         /* set as in for next Perm */
459                                         pairs[pidx].in_node = res1;
460                                 }
461
462                                 set_Proj_pred(res2, cpyxchg);
463                                 set_Proj_proj(res2, 0);
464                                 set_Proj_pred(res1, cpyxchg);
465                                 set_Proj_proj(res1, 1);
466
467                                 arch_set_irn_register(arch_env, res2, cycle->elems[i + 1]);
468                                 arch_set_irn_register(arch_env, res1, cycle->elems[i]);
469
470                                 /* insert the copy/exchange node in schedule after the magic schedule node (see above) */
471                                 sched_add_after(sched_point, cpyxchg);
472
473                                 DBG((mod, LEVEL_1, "replacing %+F with %+F, placed new node after %+F\n", irn, cpyxchg, sched_point));
474
475                                 /* set the new scheduling point */
476                                 sched_point = res1;
477                         }
478                         else {
479                                 DBG((mod, LEVEL_1, "%+F creating copy node (%+F, %s) -> (%+F, %s)\n",
480                                         irn, arg1, cycle->elems[i]->name, res2, cycle->elems[i + 1]->name));
481
482                                 cpyxchg = be_new_Copy(reg_class, irg, block, arg1);
483                                 arch_set_irn_register(arch_env, cpyxchg, cycle->elems[i + 1]);
484                                 n_ops++;
485
486                                 /* exchange copy node and proj */
487                                 exchange(res2, cpyxchg);
488
489                                 /* insert the copy/exchange node in schedule after the magic schedule node (see above) */
490                                 sched_add_after(sched_point, cpyxchg);
491
492                                 /* set the new scheduling point */
493                                 sched_point = cpyxchg;
494                         }
495                 }
496
497                 free((void *) cycle->elems);
498                 free(cycle);
499         }
500
501         /* remove the perm from schedule */
502         if (! keep_perm) {
503                 sched_remove(irn);
504                 kill_node(irn);
505         }
506 }
507
508
509
510 static int has_irn_users(const ir_node *irn) {
511         return get_irn_out_edge_first_kind(irn, EDGE_KIND_NORMAL) != 0;
512 }
513
514 /**
515  * Skip all Proj nodes.
516  */
517 static INLINE ir_node *belower_skip_proj(ir_node *irn) {
518         while(is_Proj(irn))
519                 irn = get_Proj_pred(irn);
520         return irn;
521 }
522
523 static ir_node *find_copy(constraint_env_t *env, ir_node *irn, ir_node *op) {
524         const arch_env_t *arch_env = be_get_birg_arch_env(env->birg);
525         ir_node          *block    = get_nodes_block(irn);
526         ir_node          *cur_node;
527
528         for (cur_node = sched_prev(irn);
529                 ! is_Block(cur_node) && be_is_Copy(cur_node) && get_nodes_block(cur_node) == block;
530                 cur_node = sched_prev(cur_node))
531         {
532                 if (be_get_Copy_op(cur_node) == op && arch_irn_is(arch_env, cur_node, dont_spill))
533                         return cur_node;
534         }
535
536         return NULL;
537 }
538
539 static void gen_assure_different_pattern(ir_node *irn, ir_node *other_different, constraint_env_t *env) {
540         be_irg_t                    *birg     = env->birg;
541         ir_graph                    *irg      = be_get_birg_irg(birg);
542         pset                        *op_set   = env->op_set;
543         const arch_env_t            *arch_env = be_get_birg_arch_env(birg);
544         ir_node                     *block    = get_nodes_block(irn);
545         const arch_register_class_t *cls      = arch_get_irn_reg_class(arch_env, other_different, -1);
546         ir_node                     *in[2], *keep, *cpy;
547         op_copy_assoc_t             key, *entry;
548         DEBUG_ONLY(firm_dbg_module_t *mod     = env->dbg;)
549
550         if (arch_irn_is(arch_env, other_different, ignore) || ! mode_is_datab(get_irn_mode(other_different))) {
551                 DBG((mod, LEVEL_1, "ignore constraint for %+F because other_irn is ignore or not a datab node\n", irn));
552                 return;
553         }
554
555         /* Make a not spillable copy of the different node   */
556         /* this is needed because the different irn could be */
557         /* in block far far away                             */
558         /* The copy is optimized later if not needed         */
559
560         /* check if already exists such a copy in the schedule immediately before */
561         cpy = find_copy(env, belower_skip_proj(irn), other_different);
562         if (! cpy) {
563                 cpy = be_new_Copy(cls, irg, block, other_different);
564                 be_node_set_flags(cpy, BE_OUT_POS(0), arch_irn_flags_dont_spill);
565                 DBG((mod, LEVEL_1, "created non-spillable %+F for value %+F\n", cpy, other_different));
566         }
567         else {
568                 DBG((mod, LEVEL_1, "using already existing %+F for value %+F\n", cpy, other_different));
569         }
570
571         in[0] = irn;
572         in[1] = cpy;
573
574         /* Add the Keep resp. CopyKeep and reroute the users */
575         /* of the other_different irn in case of CopyKeep.   */
576         if (has_irn_users(other_different)) {
577                 keep = be_new_CopyKeep_single(cls, irg, block, cpy, irn, get_irn_mode(other_different));
578                 be_node_set_reg_class(keep, 1, cls);
579         }
580         else {
581                 keep = be_new_Keep(cls, irg, block, 2, in);
582         }
583
584         DBG((mod, LEVEL_1, "created %+F(%+F, %+F)\n\n", keep, irn, cpy));
585
586         /* insert copy and keep into schedule */
587         assert(sched_is_scheduled(irn) && "need schedule to assure constraints");
588         if (! sched_is_scheduled(cpy))
589                 sched_add_before(belower_skip_proj(irn), cpy);
590         sched_add_after(irn, keep);
591
592         /* insert the other different and it's copies into the set */
593         key.op = other_different;
594         entry  = pset_find(op_set, &key, hash_irn(other_different));
595
596         if (! entry) {
597                 entry         = obstack_alloc(&env->obst, sizeof(*entry));
598                 ir_nodeset_init(&entry->copies);
599                 entry->op     = other_different;
600                 entry->cls    = cls;
601         }
602
603         /* insert copy */
604         ir_nodeset_insert(&entry->copies, cpy);
605
606         /* insert keep in case of CopyKeep */
607         if (be_is_CopyKeep(keep)) {
608                 ir_nodeset_insert(&entry->copies, keep);
609         }
610
611         pset_insert(op_set, entry, hash_irn(other_different));
612 }
613
614 /**
615  * Checks if node has a must_be_different constraint in output and adds a Keep
616  * then to assure the constraint.
617  */
618 static void assure_different_constraints(ir_node *irn, constraint_env_t *env) {
619         const arch_register_req_t *req;
620         const arch_env_t          *arch_env    = be_get_birg_arch_env(env->birg);
621         ir_node                   *skipped_irn = belower_skip_proj(irn);
622
623         req = arch_get_register_req(arch_env, irn, -1);
624
625         if (arch_register_req_is(req, must_be_different)) {
626                 const unsigned other = req->other_different;
627                 int i;
628
629                 if (arch_register_req_is(req, should_be_same)) {
630                         const unsigned same = req->other_same;
631
632                         if (is_po2(other) && is_po2(same)) {
633                                 int idx_other = ntz(other);
634                                 int idx_same  = ntz(same);
635
636                                 /*
637                                  * We can safely ignore a should_be_same x must_be_different y
638                                  * IFF both inputs are equal!
639                                  */
640                                 if (get_irn_n(skipped_irn, idx_other) == get_irn_n(skipped_irn, idx_same)) {
641                                         return;
642                                 }
643                         }
644                 }
645                 for (i = 0; 1U << i <= other; ++i) {
646                         if (other & (1U << i)) {
647                                 ir_node *different_from = get_irn_n(skipped_irn, i);
648                                 gen_assure_different_pattern(irn, different_from, env);
649                         }
650                 }
651         }
652 }
653
654 /**
655  * Calls the functions to assure register constraints.
656  *
657  * @param irn      The node to be checked for lowering
658  * @param walk_env The walker environment
659  */
660 static void assure_constraints_walker(ir_node *irn, void *walk_env) {
661         if (is_Block(irn))
662                 return;
663
664         if (sched_is_scheduled(irn) && mode_is_datab(get_irn_mode(irn)))
665                 assure_different_constraints(irn, walk_env);
666 }
667
668 /**
669  * Melt all copykeeps pointing to the same node
670  * (or Projs of the same node), copying the same operand.
671  */
672 static void melt_copykeeps(constraint_env_t *cenv) {
673         be_irg_t *birg = cenv->birg;
674         ir_graph *irg  = be_get_birg_irg(birg);
675         op_copy_assoc_t *entry;
676
677         /* for all */
678         foreach_pset(cenv->op_set, entry) {
679                 int     idx, num_ck;
680                 ir_node *cp;
681                 struct obstack obst;
682                 ir_nodeset_iterator_t iter;
683                 ir_node **ck_arr, **melt_arr;
684
685                 obstack_init(&obst);
686
687                 /* collect all copykeeps */
688                 num_ck = idx = 0;
689                 foreach_ir_nodeset(&entry->copies, cp, iter) {
690                         if (be_is_CopyKeep(cp)) {
691                                 obstack_grow(&obst, &cp, sizeof(cp));
692                                 ++num_ck;
693                         }
694 #ifdef KEEP_ALIVE_COPYKEEP_HACK
695                         else {
696                                 set_irn_mode(cp, mode_ANY);
697                                 keep_alive(cp);
698                         }
699 #endif /* KEEP_ALIVE_COPYKEEP_HACK */
700                 }
701
702                 /* compare each copykeep with all other copykeeps */
703                 ck_arr = (ir_node **)obstack_finish(&obst);
704                 for (idx = 0; idx < num_ck; ++idx) {
705                         ir_node *ref, *ref_mode_T;
706
707                         if (ck_arr[idx]) {
708                                 int j, n_melt;
709                                 ir_node **new_ck_in;
710                                 ir_node *new_ck;
711                                 ir_node *sched_pt = NULL;
712
713                                 n_melt     = 1;
714                                 ref        = ck_arr[idx];
715                                 ref_mode_T = skip_Proj(get_irn_n(ref, 1));
716                                 obstack_grow(&obst, &ref, sizeof(ref));
717
718                                 DBG((cenv->dbg, LEVEL_1, "Trying to melt %+F:\n", ref));
719
720                                 /* check for copykeeps pointing to the same mode_T node as the reference copykeep */
721                                 for (j = 0; j < num_ck; ++j) {
722                                         ir_node *cur_ck = ck_arr[j];
723
724                                         if (j != idx && cur_ck && skip_Proj(get_irn_n(cur_ck, 1)) == ref_mode_T) {
725                                                 obstack_grow(&obst, &cur_ck, sizeof(cur_ck));
726                                                 ir_nodeset_remove(&entry->copies, cur_ck);
727                                                 DBG((cenv->dbg, LEVEL_1, "\t%+F\n", cur_ck));
728                                                 ck_arr[j] = NULL;
729                                                 ++n_melt;
730                                                 sched_remove(cur_ck);
731                                         }
732                                 }
733                                 ck_arr[idx] = NULL;
734
735                                 /* check, if we found some candidates for melting */
736                                 if (n_melt == 1) {
737                                         DBG((cenv->dbg, LEVEL_1, "\tno candidate found\n"));
738                                         continue;
739                                 }
740
741                                 ir_nodeset_remove(&entry->copies, ref);
742                                 sched_remove(ref);
743
744                                 melt_arr = (ir_node **)obstack_finish(&obst);
745                                 /* melt all found copykeeps */
746                                 NEW_ARR_A(ir_node *, new_ck_in, n_melt);
747                                 for (j = 0; j < n_melt; ++j) {
748                                         new_ck_in[j] = get_irn_n(melt_arr[j], 1);
749
750                                         /* now, we can kill the melted keep, except the */
751                                         /* ref one, we still need some information      */
752                                         if (melt_arr[j] != ref)
753                                                 kill_node(melt_arr[j]);
754                                 }
755
756 #ifdef KEEP_ALIVE_COPYKEEP_HACK
757                                 new_ck = be_new_CopyKeep(entry->cls, irg, get_nodes_block(ref), be_get_CopyKeep_op(ref), n_melt, new_ck_in, mode_ANY);
758                                 keep_alive(new_ck);
759 #else
760                                 new_ck = be_new_CopyKeep(entry->cls, irg, get_nodes_block(ref), be_get_CopyKeep_op(ref), n_melt, new_ck_in, get_irn_mode(ref));
761 #endif /* KEEP_ALIVE_COPYKEEP_HACK */
762
763                                 /* set register class for all kept inputs */
764                                 for (j = 1; j <= n_melt; ++j)
765                                         be_node_set_reg_class(new_ck, j, entry->cls);
766
767                                 ir_nodeset_insert(&entry->copies, new_ck);
768
769                                 /* find scheduling point */
770                                 sched_pt = ref_mode_T;
771                                 do {
772                                         /* just walk along the schedule until a non-Keep/CopyKeep node is found */
773                                         sched_pt = sched_next(sched_pt);
774                                 } while (be_is_Keep(sched_pt) || be_is_CopyKeep(sched_pt));
775
776                                 sched_add_before(sched_pt, new_ck);
777                                 DBG((cenv->dbg, LEVEL_1, "created %+F, scheduled before %+F\n", new_ck, sched_pt));
778
779                                 /* finally: kill the reference copykeep */
780                                 kill_node(ref);
781                         }
782                 }
783
784                 obstack_free(&obst, NULL);
785         }
786 }
787
788 /**
789  * Walks over all nodes to assure register constraints.
790  *
791  * @param birg  The birg structure containing the irg
792  */
793 void assure_constraints(be_irg_t *birg) {
794         ir_graph         *irg      = be_get_birg_irg(birg);
795         const arch_env_t *arch_env = be_get_birg_arch_env(birg);
796         constraint_env_t cenv;
797         op_copy_assoc_t  *entry;
798         ir_node          **nodes;
799         FIRM_DBG_REGISTER(firm_dbg_module_t *mod, "firm.be.lower.constr");
800
801         DEBUG_ONLY(cenv.dbg = mod;)
802         cenv.birg   = birg;
803         cenv.op_set = new_pset(cmp_op_copy_assoc, 16);
804         obstack_init(&cenv.obst);
805
806         irg_walk_blkwise_graph(irg, NULL, assure_constraints_walker, &cenv);
807
808         /* melt copykeeps, pointing to projs of */
809         /* the same mode_T node and keeping the */
810         /* same operand                         */
811         melt_copykeeps(&cenv);
812
813         /* for all */
814         foreach_pset(cenv.op_set, entry) {
815                 int     n;
816                 ir_node *cp;
817                 ir_nodeset_iterator_t iter;
818                 be_ssa_construction_env_t senv;
819
820                 n     = ir_nodeset_size(&entry->copies);
821                 nodes = alloca(n * sizeof(nodes[0]));
822
823                 /* put the node in an array */
824                 DBG((mod, LEVEL_1, "introduce copies for %+F ", entry->op));
825
826                 /* collect all copies */
827                 n = 0;
828                 foreach_ir_nodeset(&entry->copies, cp, iter) {
829                         nodes[n++] = cp;
830                         DB((mod, LEVEL_1, ", %+F ", cp));
831                 }
832
833                 DB((mod, LEVEL_1, "\n"));
834
835                 /* introduce the copies for the operand and it's copies */
836                 be_ssa_construction_init(&senv, birg);
837                 be_ssa_construction_add_copy(&senv, entry->op);
838                 be_ssa_construction_add_copies(&senv, nodes, n);
839                 be_ssa_construction_fix_users(&senv, entry->op);
840                 be_ssa_construction_destroy(&senv);
841
842                 /* Could be that not all CopyKeeps are really needed, */
843                 /* so we transform unnecessary ones into Keeps.       */
844                 foreach_ir_nodeset(&entry->copies, cp, iter) {
845                         if (be_is_CopyKeep(cp) && get_irn_n_edges(cp) < 1) {
846                                 ir_node *keep;
847                                 int     n = get_irn_arity(cp);
848
849                                 keep = be_new_Keep(arch_get_irn_reg_class(arch_env, cp, -1),
850                                         irg, get_nodes_block(cp), n, get_irn_in(cp) + 1);
851                                 sched_add_before(cp, keep);
852
853                                 /* Set all ins (including the block) of the CopyKeep BAD to keep the verifier happy. */
854                                 sched_remove(cp);
855                                 kill_node(cp);
856                         }
857                 }
858
859                 ir_nodeset_destroy(&entry->copies);
860         }
861
862         del_pset(cenv.op_set);
863         obstack_free(&cenv.obst, NULL);
864         be_liveness_invalidate(be_get_birg_liveness(birg));
865 }
866
867
868 /**
869  * Push nodes that do not need to be permed through the Perm.
870  * This is commonly a reload cascade at block ends.
871  * @note This routine needs interference.
872  * @note Probably, we can implement it a little more efficient.
873  *       Especially searching the frontier lazily might be better.
874  * @param perm The perm.
875  * @param data The walker data (lower_env_t).
876  * @return     1, if there is something left to perm over.
877  *             0, if removed the complete perm.
878  */
879 static int push_through_perm(ir_node *perm, void *data)
880 {
881         lower_env_t *env       = data;
882         const arch_env_t *aenv = env->arch_env;
883
884         ir_graph *irg     = get_irn_irg(perm);
885         ir_node *bl       = get_nodes_block(perm);
886         ir_node *node;
887         int  arity        = get_irn_arity(perm);
888         int *map;
889         int *proj_map;
890         bitset_t *moved   = bitset_alloca(arity);
891         int n_moved;
892         int new_size;
893         ir_node *frontier = bl;
894         FIRM_DBG_REGISTER(firm_dbg_module_t *mod, "firm.be.lower.permmove");
895
896         int i, n;
897         const ir_edge_t *edge;
898         ir_node *one_proj = NULL, *irn;
899         const arch_register_class_t *cls = NULL;
900
901         DBG((mod, LEVEL_1, "perm move %+F irg %+F\n", perm, irg));
902
903         /* get some Proj and find out the register class of that Proj. */
904         edge     = get_irn_out_edge_first_kind(perm, EDGE_KIND_NORMAL);
905         one_proj = get_edge_src_irn(edge);
906         assert(is_Proj(one_proj));
907         cls      = arch_get_irn_reg_class(aenv, one_proj, -1);
908
909         /* Find the point in the schedule after which the
910          * potentially movable nodes must be defined.
911          * A Perm will only be pushed up to first instruction
912          * which lets an operand of itself die.
913          * If we would allow to move the Perm above this instruction,
914          * the former dead operand would be live now at the point of
915          * the Perm, increasing the register pressure by one.
916          */
917         sched_foreach_reverse_from (sched_prev(perm), irn) {
918                 for (i = get_irn_arity(irn) - 1; i >= 0; --i) {
919                         ir_node *op = get_irn_n(irn, i);
920                         if (arch_irn_consider_in_reg_alloc(aenv, cls, op) &&
921                             !values_interfere(env->birg, op, one_proj)) {
922                                 frontier = irn;
923                                 goto found_front;
924                         }
925                 }
926         }
927 found_front:
928
929         DBG((mod, LEVEL_2, "\tfrontier: %+F\n", frontier));
930
931         node = sched_prev(perm);
932         n_moved = 0;
933         while(!sched_is_begin(node)) {
934                 const arch_register_req_t *req;
935                 int                        input = -1;
936                 ir_node                   *proj;
937
938                 /* search if node is a INPUT of Perm */
939                 foreach_out_edge(perm, edge) {
940                         ir_node *out = get_edge_src_irn(edge);
941                         int      pn  = get_Proj_proj(out);
942                         ir_node *in  = get_irn_n(perm, pn);
943                         if (node == in) {
944                                 proj  = out;
945                                 input = pn;
946                                 break;
947                         }
948                 }
949                 /* it wasn't an input to the perm, we can't do anything more */
950                 if(input < 0)
951                         break;
952                 if(!sched_comes_after(frontier, node))
953                         break;
954                 if(arch_irn_is(aenv, node, modify_flags))
955                         break;
956                 if(is_Proj(node)) {
957                         req = arch_get_register_req(aenv, get_Proj_pred(node),
958                                                     -1 - get_Proj_proj(node));
959                 } else {
960                         req = arch_get_register_req(aenv, node, -1);
961                 }
962                 if(req->type != arch_register_req_type_normal)
963                         break;
964                 for(i = get_irn_arity(node) - 1; i >= 0; --i) {
965                         ir_node *opop = get_irn_n(node, i);
966                         if (arch_irn_consider_in_reg_alloc(aenv, cls, opop)) {
967                                 break;
968                         }
969                 }
970                 if(i >= 0)
971                         break;
972
973                 DBG((mod, LEVEL_2, "\tmoving %+F after %+F, killing %+F\n", node, perm, proj));
974
975                 /* move the movable node in front of the Perm */
976                 sched_remove(node);
977                 sched_add_after(perm, node);
978
979                 /* give it the proj's register */
980                 arch_set_irn_register(aenv, node, arch_get_irn_register(aenv, proj));
981
982                 /* reroute all users of the proj to the moved node. */
983                 edges_reroute(proj, node, irg);
984
985                 /* and kill it */
986                 set_Proj_pred(proj, new_Bad());
987                 kill_node(proj);
988
989                 bitset_set(moved, input);
990                 n_moved++;
991
992                 node = sched_prev(node);
993         }
994
995         /* well, we could not push anything through the perm */
996         if(n_moved == 0)
997                 return 1;
998
999         new_size = arity - n_moved;
1000         if(new_size == 0) {
1001                 return 0;
1002         }
1003
1004         map      = alloca(new_size * sizeof(map[0]));
1005         proj_map = alloca(arity * sizeof(proj_map[0]));
1006         memset(proj_map, -1, sizeof(proj_map[0]));
1007         n   = 0;
1008         for(i = 0; i < arity; ++i) {
1009                 if(bitset_is_set(moved, i))
1010                         continue;
1011                 map[n]      = i;
1012                 proj_map[i] = n;
1013                 n++;
1014         }
1015         assert(n == new_size);
1016         foreach_out_edge(perm, edge) {
1017                 ir_node *proj = get_edge_src_irn(edge);
1018                 int      pn   = get_Proj_proj(proj);
1019                 pn = proj_map[pn];
1020                 assert(pn >= 0);
1021                 set_Proj_proj(proj, pn);
1022         }
1023
1024         be_Perm_reduce(perm, new_size, map);
1025         return 1;
1026 }
1027
1028 /**
1029  * Calls the corresponding lowering function for the node.
1030  *
1031  * @param irn      The node to be checked for lowering
1032  * @param walk_env The walker environment
1033  */
1034 static void lower_nodes_after_ra_walker(ir_node *irn, void *walk_env) {
1035         int perm_stayed;
1036
1037         if (is_Block(irn) || is_Proj(irn))
1038                 return;
1039         if (!be_is_Perm(irn))
1040                 return;
1041
1042         perm_stayed = push_through_perm(irn, walk_env);
1043         if (!perm_stayed)
1044                 return;
1045
1046         lower_perm_node(irn, walk_env);
1047 }
1048
1049 /**
1050  * Walks over all blocks in an irg and performs lowering need to be
1051  * done after register allocation (e.g. perm lowering).
1052  *
1053  * @param birg      The birg object
1054  * @param do_copy   1 == resolve cycles with a free reg if available
1055  */
1056 void lower_nodes_after_ra(be_irg_t *birg, int do_copy) {
1057         lower_env_t env;
1058         ir_graph    *irg = be_get_birg_irg(birg);
1059
1060         env.birg     = birg;
1061         env.arch_env = be_get_birg_arch_env(birg);
1062         env.do_copy  = do_copy;
1063         FIRM_DBG_REGISTER(env.dbg_module, "firm.be.lower");
1064
1065         /* we will need interference */
1066         be_liveness_assure_chk(be_get_birg_liveness(birg));
1067
1068         irg_walk_blkwise_graph(irg, NULL, lower_nodes_after_ra_walker, &env);
1069 }