5 #include "heuristical.h"
6 #include "html_dumper.h"
10 #include "pbqp_edge_t.h"
11 #include "pbqp_node.h"
12 #include "pbqp_node_t.h"
15 static pbqp_edge **edge_bucket;
16 static pbqp_node **node_buckets[4];
17 static pbqp_node **reduced_bucket = NULL;
18 static int buckets_filled = 0;
20 static void insert_into_edge_bucket(pbqp_edge *edge)
22 unsigned bucket_len = ARR_LEN(edge_bucket);
24 if (edge->bucket_index < bucket_len && edge_bucket[edge->bucket_index]
26 /* Edge is already inserted. */
29 edge->bucket_index = bucket_len;
30 ARR_APP1(pbqp_edge *, edge_bucket, edge);
33 static void init_buckets(void)
37 edge_bucket = NEW_ARR_F(pbqp_edge *, 0);
38 reduced_bucket = NEW_ARR_F(pbqp_node *, 0);
40 for (i = 0; i < 4; ++i) {
41 node_buckets[i] = NEW_ARR_F(pbqp_node *, 0);
45 static void free_buckets(void)
49 for (i = 0; i < 4; ++i) {
50 DEL_ARR_F(node_buckets[i]);
51 node_buckets[i] = NULL;
54 DEL_ARR_F(edge_bucket);
57 DEL_ARR_F(reduced_bucket);
58 reduced_bucket = NULL;
63 static void fill_node_buckets(pbqp *pbqp)
69 node_len = pbqp->num_nodes;
71 for (node_index = 0; node_index < node_len; ++node_index) {
73 pbqp_node *node = get_node(pbqp, node_index);
77 arity = ARR_LEN(node->edges);
79 /* We have only one bucket for nodes with arity >= 3. */
84 node->bucket_index = ARR_LEN(node_buckets[arity]);
86 ARR_APP1(pbqp_node *, node_buckets[arity], node);
92 static void normalize_towards_source(pbqp *pbqp, pbqp_edge *edge)
106 src_node = edge->src;
107 tgt_node = edge->tgt;
111 src_vec = src_node->costs;
112 tgt_vec = tgt_node->costs;
116 src_len = src_vec->len;
117 tgt_len = tgt_vec->len;
124 /* Normalize towards source node. */
125 for (src_index = 0; src_index < src_len; ++src_index) {
126 num min = pbqp_matrix_get_row_min(mat, src_index, tgt_vec);
129 if (src_vec->entries[src_index].data == INF_COSTS) {
130 pbqp_matrix_set_row_value(mat, src_index, 0);
132 pbqp_matrix_sub_row_value(mat, src_index, tgt_vec, min);
134 src_vec->entries[src_index].data = pbqp_add(
135 src_vec->entries[src_index].data, min);
137 if (min == INF_COSTS) {
138 insert_into_edge_bucket(edge);
144 static void normalize_towards_target(pbqp *pbqp, pbqp_edge *edge)
158 src_node = edge->src;
159 tgt_node = edge->tgt;
163 src_vec = src_node->costs;
164 tgt_vec = tgt_node->costs;
168 src_len = src_vec->len;
169 tgt_len = tgt_vec->len;
176 for (tgt_index = 0; tgt_index < tgt_len; ++tgt_index) {
177 num min = pbqp_matrix_get_col_min(mat, tgt_index, src_vec);
180 if (tgt_vec->entries[tgt_index].data == INF_COSTS) {
181 pbqp_matrix_set_col_value(mat, tgt_index, 0);
183 pbqp_matrix_sub_col_value(mat, tgt_index, src_vec, min);
185 tgt_vec->entries[tgt_index].data = pbqp_add(
186 tgt_vec->entries[tgt_index].data, min);
188 if (min == INF_COSTS) {
189 insert_into_edge_bucket(edge);
195 static void reorder_node(pbqp_node *node)
199 unsigned old_bucket_len;
200 unsigned old_bucket_index;
201 pbqp_node **old_bucket;
204 if (!buckets_filled) return;
208 arity = ARR_LEN(node->edges);
210 /* Same bucket as before */
211 if (arity > 2) return;
213 /* Assume node lost one incident edge. */
214 old_arity = arity + 1;
215 old_bucket = node_buckets[old_arity];
216 old_bucket_len = ARR_LEN(old_bucket);
217 old_bucket_index = node->bucket_index;
219 if (old_bucket_len <= old_bucket_index || old_bucket[old_bucket_index]
221 unsigned bucket_len = ARR_LEN(node_buckets[arity]);
223 /* Old arity is new arity, so we have nothing to do. */
224 assert(old_bucket_index < bucket_len);
225 assert(node_buckets[arity][old_bucket_index] == node);
229 assert(old_bucket[old_bucket_index] == node);
231 /* Delete node from old bucket... */
232 other = old_bucket[old_bucket_len - 1];
233 other->bucket_index = old_bucket_index;
234 old_bucket[old_bucket_index] = other;
235 ARR_SHRINKLEN(node_buckets[old_arity], old_bucket_len - 1);
237 /* ..and add to new one. */
238 node->bucket_index = ARR_LEN(node_buckets[arity]);
239 ARR_APP1(pbqp_node*, node_buckets[arity], node);
242 static void simplify_edge(pbqp *pbqp, pbqp_edge *edge)
255 src_node = edge->src;
256 tgt_node = edge->tgt;
260 /* If edge are already deleted, we have nothing to do. */
261 if (!is_connected(src_node, edge) || !is_connected(tgt_node, edge))
264 if (pbqp->dump_file) {
266 sprintf(txt, "Simplification of Edge n%d-n%d", src_node->index, tgt_node->index);
267 dump_section(pbqp->dump_file, 3, txt);
270 src_vec = src_node->costs;
271 tgt_vec = tgt_node->costs;
275 src_len = src_vec->len;
276 tgt_len = tgt_vec->len;
283 if (pbqp->dump_file) {
284 fputs("Input:<br>\n", pbqp->dump_file);
285 dump_simplifyedge(pbqp, edge);
288 normalize_towards_source(pbqp, edge);
289 normalize_towards_target(pbqp, edge);
291 if (pbqp->dump_file) {
292 fputs("<br>\nOutput:<br>\n", pbqp->dump_file);
293 dump_simplifyedge(pbqp, edge);
296 if (pbqp_matrix_is_zero(mat, src_vec, tgt_vec)) {
297 if (pbqp->dump_file) {
298 fputs("edge has been eliminated<br>\n", pbqp->dump_file);
302 reorder_node(src_node);
303 reorder_node(tgt_node);
307 void solve_pbqp_heuristical(pbqp *pbqp)
314 if (pbqp->dump_file) {
315 pbqp_dump_input(pbqp);
316 dump_section(pbqp->dump_file, 1, "2. Simplification of Cost Matrices");
319 node_len = pbqp->num_nodes;
323 /* First simplify all edges. */
324 for (node_index = 0; node_index < node_len; ++node_index) {
326 pbqp_node *node = get_node(pbqp, node_index);
333 edge_len = ARR_LEN(edges);
335 for (edge_index = 0; edge_index < edge_len; ++edge_index) {
336 pbqp_edge *edge = edges[edge_index];
338 /* Simplify only once per edge. */
339 if (node != edge->src) continue;
341 simplify_edge(pbqp, edge);
345 /* Put node into bucket representing their arity. */
346 fill_node_buckets(pbqp);
349 if (ARR_LEN(edge_bucket) > 0) {
351 } else if (ARR_LEN(node_buckets[1]) > 0) {
353 } else if (ARR_LEN(node_buckets[2]) > 0) {
355 } else if (ARR_LEN(node_buckets[3]) > 0) {
362 if (pbqp->dump_file) {
363 dump_section(pbqp->dump_file, 1, "4. Determine Solution/Minimum");
364 dump_section(pbqp->dump_file, 2, "4.1. Trivial Solution");
367 /* Solve trivial nodes and calculate solution. */
368 node_len = ARR_LEN(node_buckets[0]);
369 for (node_index = 0; node_index < node_len; ++node_index) {
370 pbqp_node *node = node_buckets[0][node_index];
373 node->solution = vector_get_min_index(node->costs);
374 pbqp->solution = pbqp_add(pbqp->solution,
375 node->costs->entries[node->solution].data);
376 if (pbqp->dump_file) {
377 fprintf(pbqp->dump_file, "node n%d is set to %d<br>\n", node->index, node->solution);
378 dump_node(pbqp, node);
382 if (pbqp->dump_file) {
383 dump_section(pbqp->dump_file, 2, "Minimum");
384 fprintf(pbqp->dump_file, "Minimum is equal to %d.", pbqp->solution);
385 dump_section(pbqp->dump_file, 2, "Back Propagation");
388 /* Solve reduced nodes. */
389 node_len = ARR_LEN(reduced_bucket);
390 for (node_index = node_len; node_index > 0; --node_index) {
391 pbqp_node *node = reduced_bucket[node_index - 1];
394 switch (ARR_LEN(node->edges)) {
396 back_propagate_RI(pbqp, node);
399 back_propagate_RII(pbqp, node);
402 panic("Only nodes with degree one or two should be in this bucket");
410 void apply_edge(pbqp *pbqp)
412 unsigned bucket_len = ARR_LEN(edge_bucket);
413 pbqp_edge *edge = edge_bucket[bucket_len - 1];
415 ARR_SHRINKLEN(edge_bucket, (int)bucket_len - 1);
417 simplify_edge(pbqp, edge);
420 void apply_RI(pbqp *pbqp)
422 pbqp_node **bucket = node_buckets[1];
423 unsigned bucket_len = ARR_LEN(bucket);
424 pbqp_node *node = bucket[bucket_len - 1];
425 pbqp_edge *edge = node->edges[0];
426 pbqp_matrix *mat = edge->costs;
427 int is_src = edge->src == node;
428 pbqp_node *other_node;
431 other_node = edge->tgt;
433 other_node = edge->src;
436 if (pbqp->dump_file) {
438 sprintf(txt, "RI-Reduction of Node n%d", node->index);
439 dump_section(pbqp->dump_file, 2, txt);
440 pbqp_dump_graph(pbqp);
441 fputs("<br>\nBefore reduction:<br>\n", pbqp->dump_file);
442 dump_node(pbqp, node);
443 dump_node(pbqp, other_node);
444 dump_edge(pbqp, edge);
448 pbqp_matrix_add_to_all_cols(mat, node->costs);
449 normalize_towards_target(pbqp, edge);
451 pbqp_matrix_add_to_all_rows(mat, node->costs);
452 normalize_towards_source(pbqp, edge);
454 disconnect_edge(other_node, edge);
456 if (pbqp->dump_file) {
457 fputs("<br>\nAfter reduction:<br>\n", pbqp->dump_file);
458 dump_node(pbqp, other_node);
461 /* Remove node from bucket... */
462 ARR_SHRINKLEN(bucket, (int)bucket_len - 1);
463 reorder_node(other_node);
465 /* ...and add it to back propagation list. */
466 node->bucket_index = ARR_LEN(reduced_bucket);
467 ARR_APP1(pbqp_node *, reduced_bucket, node);
470 void apply_RII(pbqp *pbqp)
472 pbqp_node **bucket = node_buckets[2];
473 unsigned bucket_len = ARR_LEN(bucket);
474 pbqp_node *node = bucket[bucket_len - 1];
475 pbqp_edge *src_edge = node->edges[0];
476 pbqp_edge *tgt_edge = node->edges[1];
477 int src_is_src = src_edge->src == node;
478 int tgt_is_src = tgt_edge->src == node;
479 pbqp_matrix *src_mat;
480 pbqp_matrix *tgt_mat;
497 src_node = src_edge->tgt;
499 src_node = src_edge->src;
503 tgt_node = tgt_edge->tgt;
505 tgt_node = tgt_edge->src;
508 /* Swap nodes if necessary. */
509 if (tgt_node->index < src_node->index) {
521 src_is_src = src_edge->src == node;
522 tgt_is_src = tgt_edge->src == node;
525 if (pbqp->dump_file) {
527 sprintf(txt, "RII-Reduction of Node n%d", node->index);
528 dump_section(pbqp->dump_file, 2, txt);
529 pbqp_dump_graph(pbqp);
530 fputs("<br>\nBefore reduction:<br>\n", pbqp->dump_file);
531 dump_node(pbqp, src_node);
532 dump_edge(pbqp, src_edge);
533 dump_node(pbqp, node);
534 dump_edge(pbqp, tgt_edge);
535 dump_node(pbqp, tgt_node);
538 src_mat = src_edge->costs;
539 tgt_mat = tgt_edge->costs;
541 src_vec = src_node->costs;
542 tgt_vec = tgt_node->costs;
543 node_vec = node->costs;
545 row_len = src_vec->len;
546 col_len = tgt_vec->len;
547 node_len = node_vec->len;
549 mat = pbqp_matrix_alloc(pbqp, row_len, col_len);
551 for (row_index = 0; row_index < row_len; ++row_index) {
552 for (col_index = 0; col_index < col_len; ++col_index) {
553 vec = vector_copy(pbqp, node_vec);
556 vector_add_matrix_col(vec, src_mat, row_index);
558 vector_add_matrix_row(vec, src_mat, row_index);
562 vector_add_matrix_col(vec, tgt_mat, col_index);
564 vector_add_matrix_row(vec, tgt_mat, col_index);
567 mat->entries[row_index * col_len + col_index] = vector_get_min(vec);
569 obstack_free(&pbqp->obstack, vec);
573 pbqp_edge *edge = get_edge(pbqp, src_node->index, tgt_node->index);
575 /* Disconnect node. */
576 disconnect_edge(src_node, src_edge);
577 disconnect_edge(tgt_node, tgt_edge);
579 /* Remove node from bucket... */
580 ARR_SHRINKLEN(bucket, (int)bucket_len - 1);
582 /* ...and add it to back propagation list. */
583 node->bucket_index = ARR_LEN(reduced_bucket);
584 ARR_APP1(pbqp_node *, reduced_bucket, node);
587 edge = alloc_edge(pbqp, src_node->index, tgt_node->index, mat);
589 pbqp_matrix_add(edge->costs, mat);
591 /* Free local matrix. */
592 obstack_free(&pbqp->obstack, mat);
594 reorder_node(src_node);
595 reorder_node(tgt_node);
598 if (pbqp->dump_file) {
599 fputs("<br>\nAfter reduction:<br>\n", pbqp->dump_file);
600 dump_edge(pbqp, edge);
603 /* Edge has changed so we simplify it. */
604 simplify_edge(pbqp, edge);
607 void apply_RN(pbqp *pbqp)
609 pbqp_node **bucket = node_buckets[3];
610 unsigned bucket_len = ARR_LEN(bucket);
611 pbqp_node *node = bucket[bucket_len - 1];
613 vector *node_vec = node->costs;
617 unsigned edge_len = ARR_LEN(node->edges);
619 unsigned node_len = node_vec->len;
620 unsigned min_index = 0;
626 if (pbqp->dump_file) {
628 sprintf(txt, "RN-Reduction of Node n%d", node->index);
629 dump_section(pbqp->dump_file, 2, txt);
630 pbqp_dump_graph(pbqp);
633 for (node_index = 0; node_index < node_len; ++node_index) {
636 for (edge_index = 0; edge_index < edge_len; ++edge_index) {
637 edge = node->edges[edge_index];
639 is_src = edge->src == node;
642 vec = vector_copy(pbqp, edge->tgt->costs);
643 vector_add_matrix_row(vec, mat, node_index);
645 vec = vector_copy(pbqp, edge->src->costs);
646 vector_add_matrix_col(vec, mat, node_index);
649 value = pbqp_add(value, vector_get_min(vec));
651 obstack_free(&pbqp->obstack, vec);
656 min_index = node_index;
660 if (pbqp->dump_file) {
661 fprintf(pbqp->dump_file, "node n%d is set to %d<br><br>\n",
662 node->index, min_index);
663 fprintf(pbqp->dump_file, "Minimal cost of RN reduction: %d<br>\n",
667 node->solution = min_index;
669 /* Now that we found the local minimum set all other costs to infinity. */
670 for (node_index = 0; node_index < node_len; ++node_index) {
671 if (node_index != min_index) {
672 node_vec->entries[node_index].data = INF_COSTS;
676 /* Add all incident edges to edge bucket, since they are now independent. */
677 for (edge_index = 0; edge_index < edge_len; ++edge_index) {
678 insert_into_edge_bucket(node->edges[edge_index]);
682 void back_propagate_RI(pbqp *pbqp, pbqp_node *node)
693 edge = node->edges[0];
695 is_src = edge->src == node;
701 vector_add_matrix_col(vec, mat, other->solution);
705 vector_add_matrix_row(vec, mat, other->solution);
708 node->solution = vector_get_min_index(vec);
709 if (pbqp->dump_file) {
710 fprintf(pbqp->dump_file, "node n%d is set to %d<br>\n", node->index, node->solution);
714 void back_propagate_RII(pbqp *pbqp, pbqp_node *node)
716 pbqp_edge *src_edge = node->edges[0];
717 pbqp_edge *tgt_edge = node->edges[1];
718 int src_is_src = src_edge->src == node;
719 int tgt_is_src = tgt_edge->src == node;
720 pbqp_matrix *src_mat;
721 pbqp_matrix *tgt_mat;
732 src_node = src_edge->tgt;
734 src_node = src_edge->src;
738 tgt_node = tgt_edge->tgt;
740 tgt_node = tgt_edge->src;
743 /* Swap nodes if necessary. */
744 if (tgt_node->index < src_node->index) {
756 src_is_src = src_edge->src == node;
757 tgt_is_src = tgt_edge->src == node;
760 src_mat = src_edge->costs;
761 tgt_mat = tgt_edge->costs;
763 node_vec = node->costs;
765 row_index = src_node->solution;
766 col_index = tgt_node->solution;
768 vec = vector_copy(pbqp, node_vec);
771 vector_add_matrix_col(vec, src_mat, row_index);
773 vector_add_matrix_row(vec, src_mat, row_index);
777 vector_add_matrix_col(vec, tgt_mat, col_index);
779 vector_add_matrix_row(vec, tgt_mat, col_index);
782 node->solution = vector_get_min_index(vec);
783 if (pbqp->dump_file) {
784 fprintf(pbqp->dump_file, "node n%d is set to %d<br>\n", node->index, node->solution);
787 obstack_free(&pbqp->obstack, vec);
790 int node_is_reduced(pbqp_node *node)
792 if (!reduced_bucket) return 0;
795 if (ARR_LEN(node->edges) == 0) return 1;
797 unsigned bucket_length = ARR_LEN(reduced_bucket);
798 unsigned bucket_index = node->bucket_index;
800 return bucket_index < bucket_length && reduced_bucket[bucket_index] == node;