Use foreach_ir_nodeset() instead of reimplementing it.
[libfirm] / ir / be / beschedrand.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       Random node selector.
23  * @author      Matthias Braun
24  * @date        29.08.2006
25  */
26 #include "config.h"
27
28 #include <stdlib.h>
29
30 #include "besched.h"
31 #include "belistsched.h"
32 #include "bemodule.h"
33
34 /**
35  * The random selector:
36  * Just assure that branches are executed last, otherwise select a random node
37  */
38 static ir_node *random_select(void *block_env, ir_nodeset_t *ready_set)
39 {
40         ir_nodeset_iterator_t iter;
41         ir_node          *irn      = NULL;
42         int only_branches_left = 1;
43         (void)block_env;
44
45         /* assure that branches and constants are executed last */
46         foreach_ir_nodeset(ready_set, irn, iter) {
47                 if (!is_cfop(irn)) {
48                         only_branches_left = 0;
49                         break;
50                 }
51         }
52
53         if (only_branches_left) {
54                 /* at last: schedule branches */
55                 ir_nodeset_iterator_init(&iter, ready_set);
56                 irn = ir_nodeset_iterator_next(&iter);
57         } else {
58                 do {
59                         /* take 1 random node */
60                         int n = rand() % ir_nodeset_size(ready_set);
61                         int i = 0;
62                         foreach_ir_nodeset(ready_set, irn, iter) {
63                                 if (i == n) {
64                                         break;
65                                 }
66                                 ++i;
67                         }
68                 } while (is_cfop(irn));
69         }
70
71         return irn;
72 }
73
74 static void *random_init_graph(ir_graph *irg)
75 {
76         (void)irg;
77         /* TODO: add commandline option for the seed */
78         srand(0x4711);
79         return NULL;
80 }
81
82 static void *random_init_block(void *graph_env, ir_node *block)
83 {
84         (void)graph_env;
85         (void)block;
86         return NULL;
87 }
88
89 static void sched_random(ir_graph *irg)
90 {
91         static const list_sched_selector_t random_selector = {
92                 random_init_graph,
93                 random_init_block,
94                 random_select,
95                 NULL,                /* node_ready */
96                 NULL,                /* node_selected */
97                 NULL,                /* finish_block */
98                 NULL                 /* finish_graph */
99         };
100         be_list_sched_graph(irg, &random_selector);
101 }
102
103 BE_REGISTER_MODULE_CONSTRUCTOR(be_init_sched_rand)
104 void be_init_sched_rand(void)
105 {
106         be_register_scheduler("random", sched_random);
107 }