bb8a352e8f268dc4677598d820b4b6dc92fa9598
[libfirm] / ir / kaps / pbqp_node.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   PBQP nodes.
23  * @date    02.10.2008
24  * @author  Sebastian Buchwald
25  * @version $Id$
26  */
27 #include "config.h"
28
29 #include "adt/array.h"
30
31 #include "assert.h"
32
33 #include "bucket.h"
34 #include "pbqp_edge.h"
35 #include "pbqp_edge_t.h"
36 #include "pbqp_node.h"
37 #include "pbqp_node_t.h"
38 #include "vector.h"
39
40 pbqp_node_t *alloc_node(pbqp_t *pbqp, unsigned node_index, vector_t *costs)
41 {
42         pbqp_node_t *node = OALLOC(&pbqp->obstack, pbqp_node_t);
43         assert(node);
44
45         node->edges = NEW_ARR_F(pbqp_edge_t *, 0);
46         node->costs = vector_copy(pbqp, costs);
47         node->bucket_index = UINT_MAX;
48         node->solution = UINT_MAX;
49         node->index = node_index;
50
51         return node;
52 }
53
54 int is_connected(pbqp_node_t *node, pbqp_edge_t *edge)
55 {
56         pbqp_edge_t **edges;
57         unsigned      edge_index;
58         unsigned      edge_len;
59
60         assert(node);
61         assert(edge);
62
63         if (edge->src != node && edge->tgt != node) return 0;
64
65         edges = node->edges;
66         edge_len = ARR_LEN(edges);
67
68         for (edge_index = 0; edge_index < edge_len; ++edge_index) {
69                 pbqp_edge_t *edge_candidate = edges[edge_index];
70                 if (edge_candidate == edge) {
71                         return 1;
72                 }
73         }
74
75         return 0;
76 }
77
78 void disconnect_edge(pbqp_node_t *node, pbqp_edge_t *edge)
79 {
80         pbqp_edge_t **edges;
81         unsigned      edge_index;
82         unsigned      edge_len;
83
84         edges = node->edges;
85         edge_len = ARR_LEN(edges);
86
87         for (edge_index = 0; edge_index < edge_len; ++edge_index) {
88                 pbqp_edge_t *edge_candidate = edges[edge_index];
89                 if (edge_candidate == edge) {
90                         edges[edge_index] = edges[edge_len - 1];
91                         ARR_SHRINKLEN(edges, (int)edge_len - 1);
92                         break;
93                 }
94         }
95 }
96
97 unsigned pbqp_node_get_degree(pbqp_node_t *node)
98 {
99         assert(node);
100         return ARR_LEN(node->edges);
101 }
102
103 pbqp_node_t *pbqp_node_deep_copy(pbqp_t *pbqp, pbqp_node_bucket_t new_bucket,
104                                  pbqp_node_t *node)
105 {
106         unsigned     edge_index;
107         unsigned     edge_length = pbqp_node_get_degree(node);
108         pbqp_node_t *copy        = OALLOC(&pbqp->obstack, pbqp_node_t);
109
110         assert(copy);
111
112         copy->edges        = NEW_ARR_F(pbqp_edge_t *, 0);
113         for (edge_index = 0; edge_index < edge_length; ++edge_index) {
114                 pbqp_edge_t *edge_copy = NULL;
115                 pbqp_edge_t *edge      = node->edges[edge_index];
116                 int          is_src    = edge->src == node;
117
118                 if (is_src) {
119                         unsigned other_index = edge->tgt->bucket_index;
120                         unsigned is_copied   = other_index < node->bucket_index;
121
122                         if (is_copied) {
123                                 pbqp_node_t *other_copy = new_bucket[other_index];
124                                 unsigned     degree     = pbqp_node_get_degree(other_copy);
125                                 unsigned     index;
126
127                                 for (index = 0; index < degree; ++index) {
128                                         if (other_copy->edges[index]->src == node) {
129                                                 edge_copy      = other_copy->edges[index];
130                                                 edge_copy->src = copy;
131                                                 break;
132                                         }
133                                 }
134                         } else {
135                                 edge_copy = pbqp_edge_deep_copy(pbqp, edge, copy, edge->tgt);
136                         }
137                 } else {
138                         unsigned other_index = edge->src->bucket_index;
139                         unsigned is_copied   = other_index < node->bucket_index;
140
141                         if (is_copied) {
142                                 pbqp_node_t *other_copy = new_bucket[other_index];
143                                 unsigned     degree     = pbqp_node_get_degree(other_copy);
144                                 unsigned     index;
145
146                                 for (index = 0; index < degree; ++index) {
147                                         if (other_copy->edges[index]->tgt == node) {
148                                                 edge_copy      = other_copy->edges[index];
149                                                 edge_copy->tgt = copy;
150                                                 break;
151                                         }
152                                 }
153                         } else {
154                                 edge_copy = pbqp_edge_deep_copy(pbqp, edge, edge->src, copy);
155                         }
156                 }
157                 ARR_APP1(pbqp_edge_t *, copy->edges, edge_copy);
158         }
159         copy->costs        = vector_copy(pbqp, node->costs);
160         copy->bucket_index = node->bucket_index;
161         copy->solution     = node->solution;
162         copy->index   = node->index;
163
164         return copy;
165 }