6 #include "heuristical.h"
7 #include "html_dumper.h"
10 #include "pbqp_edge.h"
11 #include "pbqp_edge_t.h"
12 #include "pbqp_node.h"
13 #include "pbqp_node_t.h"
16 static pbqp_edge **edge_bucket;
17 static pbqp_node **node_buckets[4];
18 static pbqp_node **reduced_bucket = NULL;
19 static int buckets_filled = 0;
21 static void insert_into_edge_bucket(pbqp_edge *edge)
23 if (edge_bucket_contains(edge_bucket, edge)) {
24 /* Edge is already inserted. */
28 edge_bucket_insert(&edge_bucket, edge);
31 static void init_buckets(void)
35 edge_bucket_init(&edge_bucket);
36 node_bucket_init(&reduced_bucket);
38 for (i = 0; i < 4; ++i) {
39 node_bucket_init(&node_buckets[i]);
43 static void free_buckets(void)
47 for (i = 0; i < 4; ++i) {
48 node_bucket_free(&node_buckets[i]);
51 edge_bucket_free(&edge_bucket);
52 node_bucket_free(&reduced_bucket);
57 static void fill_node_buckets(pbqp *pbqp)
63 node_len = pbqp->num_nodes;
65 for (node_index = 0; node_index < node_len; ++node_index) {
67 pbqp_node *node = get_node(pbqp, node_index);
71 arity = ARR_LEN(node->edges);
73 /* We have only one bucket for nodes with arity >= 3. */
78 node_bucket_insert(&node_buckets[arity], node);
84 static void normalize_towards_source(pbqp *pbqp, pbqp_edge *edge)
103 src_vec = src_node->costs;
104 tgt_vec = tgt_node->costs;
108 src_len = src_vec->len;
109 tgt_len = tgt_vec->len;
116 /* Normalize towards source node. */
117 for (src_index = 0; src_index < src_len; ++src_index) {
118 num min = pbqp_matrix_get_row_min(mat, src_index, tgt_vec);
121 if (src_vec->entries[src_index].data == INF_COSTS) {
122 pbqp_matrix_set_row_value(mat, src_index, 0);
124 pbqp_matrix_sub_row_value(mat, src_index, tgt_vec, min);
126 src_vec->entries[src_index].data = pbqp_add(
127 src_vec->entries[src_index].data, min);
129 if (min == INF_COSTS) {
131 unsigned edge_len = ARR_LEN(src_node->edges);
133 for (edge_index = 0; edge_index < edge_len; ++edge_index) {
134 pbqp_edge *edge_candidate = src_node->edges[edge_index];
135 if (edge_candidate != edge) {
136 insert_into_edge_bucket(edge_candidate);
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) {
190 unsigned edge_len = ARR_LEN(tgt_node->edges);
192 for (edge_index = 0; edge_index < edge_len; ++edge_index) {
193 pbqp_edge *edge_candidate = tgt_node->edges[edge_index];
194 if (edge_candidate != edge) {
195 insert_into_edge_bucket(edge_candidate);
203 static void reorder_node(pbqp_node *node)
207 unsigned old_bucket_len;
208 unsigned old_bucket_index;
209 pbqp_node **old_bucket;
212 if (!buckets_filled) return;
216 arity = ARR_LEN(node->edges);
218 /* Same bucket as before */
219 if (arity > 2) return;
221 /* Assume node lost one incident edge. */
222 old_arity = arity + 1;
223 old_bucket = node_buckets[old_arity];
224 old_bucket_len = node_bucket_get_length(old_bucket);
225 old_bucket_index = node->bucket_index;
227 if (old_bucket_len <= old_bucket_index || old_bucket[old_bucket_index]
229 unsigned bucket_len = node_bucket_get_length(node_buckets[arity]);
231 /* Old arity is new arity, so we have nothing to do. */
232 assert(old_bucket_index < bucket_len);
233 assert(node_buckets[arity][old_bucket_index] == node);
237 assert(old_bucket[old_bucket_index] == node);
239 /* Delete node from old bucket... */
240 other = old_bucket[old_bucket_len - 1];
241 other->bucket_index = old_bucket_index;
242 old_bucket[old_bucket_index] = other;
243 ARR_SHRINKLEN(node_buckets[old_arity], old_bucket_len - 1);
245 /* ..and add to new one. */
246 node->bucket_index = node_bucket_get_length(node_buckets[arity]);
247 ARR_APP1(pbqp_node*, node_buckets[arity], node);
250 static void check_melting_possibility(pbqp *pbqp, pbqp_edge *edge)
265 src_node = edge->src;
266 tgt_node = edge->tgt;
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 (src_len == 1 && tgt_len == 1) {
284 //panic("Something is wrong");
288 for (src_index = 0; src_index < src_len; ++src_index) {
290 if (src_vec->entries[src_index].data == INF_COSTS) {
293 for (tgt_index = 0; tgt_index < tgt_len; ++tgt_index) {
294 if (tgt_vec->entries[tgt_index].data == INF_COSTS) {
297 if (mat->entries[src_index * tgt_len + tgt_index] == 0) {
306 if (mat->entries[src_index * tgt_len + tgt_index] == INF_COSTS) {
312 allRowsOk &= onlyOneZero;
316 for (tgt_index = 0; tgt_index < tgt_len; ++tgt_index) {
318 if (tgt_vec->entries[tgt_index].data == INF_COSTS) {
321 for (src_index = 0; src_index < src_len; ++src_index) {
322 if (src_vec->entries[src_index].data == INF_COSTS) {
325 if (mat->entries[src_index * tgt_len + tgt_index] == 0) {
334 if (mat->entries[src_index * tgt_len + tgt_index] == INF_COSTS) {
340 allColsOk &= onlyOneZero;
343 if (allRowsOk && allColsOk) {
348 static void simplify_edge(pbqp *pbqp, pbqp_edge *edge)
361 src_node = edge->src;
362 tgt_node = edge->tgt;
366 /* If edge are already deleted, we have nothing to do. */
367 if (!is_connected(src_node, edge) || !is_connected(tgt_node, edge))
370 if (pbqp->dump_file) {
372 sprintf(txt, "Simplification of Edge n%d-n%d", src_node->index, tgt_node->index);
373 dump_section(pbqp->dump_file, 3, txt);
376 src_vec = src_node->costs;
377 tgt_vec = tgt_node->costs;
381 src_len = src_vec->len;
382 tgt_len = tgt_vec->len;
389 if (pbqp->dump_file) {
390 fputs("Input:<br>\n", pbqp->dump_file);
391 dump_simplifyedge(pbqp, edge);
394 normalize_towards_source(pbqp, edge);
395 normalize_towards_target(pbqp, edge);
397 if (pbqp->dump_file) {
398 fputs("<br>\nOutput:<br>\n", pbqp->dump_file);
399 dump_simplifyedge(pbqp, edge);
402 if (pbqp_matrix_is_zero(mat, src_vec, tgt_vec)) {
403 if (pbqp->dump_file) {
404 fputs("edge has been eliminated<br>\n", pbqp->dump_file);
408 reorder_node(src_node);
409 reorder_node(tgt_node);
411 //check_melting_possibility(pbqp, edge);
415 void solve_pbqp_heuristical(pbqp *pbqp)
422 if (pbqp->dump_file) {
423 pbqp_dump_input(pbqp);
424 dump_section(pbqp->dump_file, 1, "2. Simplification of Cost Matrices");
427 node_len = pbqp->num_nodes;
431 /* First simplify all edges. */
432 for (node_index = 0; node_index < node_len; ++node_index) {
434 pbqp_node *node = get_node(pbqp, node_index);
441 edge_len = ARR_LEN(edges);
443 for (edge_index = 0; edge_index < edge_len; ++edge_index) {
444 pbqp_edge *edge = edges[edge_index];
446 /* Simplify only once per edge. */
447 if (node != edge->src) continue;
449 simplify_edge(pbqp, edge);
453 /* Put node into bucket representing their arity. */
454 fill_node_buckets(pbqp);
457 if (edge_bucket_get_length(edge_bucket) > 0) {
459 } else if (node_bucket_get_length(node_buckets[1]) > 0) {
461 } else if (node_bucket_get_length(node_buckets[2]) > 0) {
463 } else if (node_bucket_get_length(node_buckets[3]) > 0) {
470 if (pbqp->dump_file) {
471 dump_section(pbqp->dump_file, 1, "4. Determine Solution/Minimum");
472 dump_section(pbqp->dump_file, 2, "4.1. Trivial Solution");
475 /* Solve trivial nodes and calculate solution. */
476 node_len = node_bucket_get_length(node_buckets[0]);
477 for (node_index = 0; node_index < node_len; ++node_index) {
478 pbqp_node *node = node_buckets[0][node_index];
481 node->solution = vector_get_min_index(node->costs);
482 pbqp->solution = pbqp_add(pbqp->solution,
483 node->costs->entries[node->solution].data);
484 if (pbqp->dump_file) {
485 fprintf(pbqp->dump_file, "node n%d is set to %d<br>\n", node->index, node->solution);
486 dump_node(pbqp, node);
490 if (pbqp->dump_file) {
491 dump_section(pbqp->dump_file, 2, "Minimum");
492 fprintf(pbqp->dump_file, "Minimum is equal to %lld.", pbqp->solution);
493 dump_section(pbqp->dump_file, 2, "Back Propagation");
496 /* Solve reduced nodes. */
497 node_len = node_bucket_get_length(reduced_bucket);
498 for (node_index = node_len; node_index > 0; --node_index) {
499 pbqp_node *node = reduced_bucket[node_index - 1];
502 switch (ARR_LEN(node->edges)) {
504 back_propagate_RI(pbqp, node);
507 back_propagate_RII(pbqp, node);
510 panic("Only nodes with degree one or two should be in this bucket");
518 void apply_edge(pbqp *pbqp)
520 pbqp_edge *edge = edge_bucket_pop(&edge_bucket);
522 simplify_edge(pbqp, edge);
525 void apply_RI(pbqp *pbqp)
527 pbqp_node *node = node_bucket_pop(&node_buckets[1]);
528 pbqp_edge *edge = node->edges[0];
529 pbqp_matrix *mat = edge->costs;
530 int is_src = edge->src == node;
531 pbqp_node *other_node;
534 other_node = edge->tgt;
536 other_node = edge->src;
539 if (pbqp->dump_file) {
541 sprintf(txt, "RI-Reduction of Node n%d", node->index);
542 dump_section(pbqp->dump_file, 2, txt);
543 pbqp_dump_graph(pbqp);
544 fputs("<br>\nBefore reduction:<br>\n", pbqp->dump_file);
545 dump_node(pbqp, node);
546 dump_node(pbqp, other_node);
547 dump_edge(pbqp, edge);
551 pbqp_matrix_add_to_all_cols(mat, node->costs);
552 normalize_towards_target(pbqp, edge);
554 pbqp_matrix_add_to_all_rows(mat, node->costs);
555 normalize_towards_source(pbqp, edge);
557 disconnect_edge(other_node, edge);
559 if (pbqp->dump_file) {
560 fputs("<br>\nAfter reduction:<br>\n", pbqp->dump_file);
561 dump_node(pbqp, other_node);
564 reorder_node(other_node);
566 /* Add node to back propagation list. */
567 node_bucket_insert(&reduced_bucket, node);
570 void apply_RII(pbqp *pbqp)
572 pbqp_node *node = node_bucket_pop(&node_buckets[2]);
573 pbqp_edge *src_edge = node->edges[0];
574 pbqp_edge *tgt_edge = node->edges[1];
575 int src_is_src = src_edge->src == node;
576 int tgt_is_src = tgt_edge->src == node;
577 pbqp_matrix *src_mat;
578 pbqp_matrix *tgt_mat;
595 src_node = src_edge->tgt;
597 src_node = src_edge->src;
601 tgt_node = tgt_edge->tgt;
603 tgt_node = tgt_edge->src;
606 /* Swap nodes if necessary. */
607 if (tgt_node->index < src_node->index) {
619 src_is_src = src_edge->src == node;
620 tgt_is_src = tgt_edge->src == node;
623 if (pbqp->dump_file) {
625 sprintf(txt, "RII-Reduction of Node n%d", node->index);
626 dump_section(pbqp->dump_file, 2, txt);
627 pbqp_dump_graph(pbqp);
628 fputs("<br>\nBefore reduction:<br>\n", pbqp->dump_file);
629 dump_node(pbqp, src_node);
630 dump_edge(pbqp, src_edge);
631 dump_node(pbqp, node);
632 dump_edge(pbqp, tgt_edge);
633 dump_node(pbqp, tgt_node);
636 src_mat = src_edge->costs;
637 tgt_mat = tgt_edge->costs;
639 src_vec = src_node->costs;
640 tgt_vec = tgt_node->costs;
641 node_vec = node->costs;
643 row_len = src_vec->len;
644 col_len = tgt_vec->len;
645 node_len = node_vec->len;
647 mat = pbqp_matrix_alloc(pbqp, row_len, col_len);
649 for (row_index = 0; row_index < row_len; ++row_index) {
650 for (col_index = 0; col_index < col_len; ++col_index) {
651 vec = vector_copy(pbqp, node_vec);
654 vector_add_matrix_col(vec, src_mat, row_index);
656 vector_add_matrix_row(vec, src_mat, row_index);
660 vector_add_matrix_col(vec, tgt_mat, col_index);
662 vector_add_matrix_row(vec, tgt_mat, col_index);
665 mat->entries[row_index * col_len + col_index] = vector_get_min(vec);
667 obstack_free(&pbqp->obstack, vec);
671 pbqp_edge *edge = get_edge(pbqp, src_node->index, tgt_node->index);
673 /* Disconnect node. */
674 disconnect_edge(src_node, src_edge);
675 disconnect_edge(tgt_node, tgt_edge);
677 /* Add node to back propagation list. */
678 node_bucket_insert(&reduced_bucket, node);
681 edge = alloc_edge(pbqp, src_node->index, tgt_node->index, mat);
683 pbqp_matrix_add(edge->costs, mat);
685 /* Free local matrix. */
686 obstack_free(&pbqp->obstack, mat);
688 reorder_node(src_node);
689 reorder_node(tgt_node);
692 if (pbqp->dump_file) {
693 fputs("<br>\nAfter reduction:<br>\n", pbqp->dump_file);
694 dump_edge(pbqp, edge);
697 /* Edge has changed so we simplify it. */
698 simplify_edge(pbqp, edge);
701 void apply_RN(pbqp *pbqp)
703 pbqp_node **bucket = node_buckets[3];
704 unsigned bucket_len = node_bucket_get_length(bucket);
705 unsigned bucket_index;
706 pbqp_node *node = NULL;
712 unsigned max_degree = 0;
715 unsigned min_index = 0;
721 /* Search for node with maximum degree. */
722 for (bucket_index = 0; bucket_index < bucket_len; ++bucket_index) {
723 pbqp_node *candidate = bucket[bucket_index];
724 unsigned degree = ARR_LEN(candidate->edges);
726 if (degree > max_degree) {
732 node_vec = node->costs;
733 node_len = node_vec->len;
735 if (pbqp->dump_file) {
737 sprintf(txt, "RN-Reduction of Node n%d", node->index);
738 dump_section(pbqp->dump_file, 2, txt);
739 pbqp_dump_graph(pbqp);
742 for (node_index = 0; node_index < node_len; ++node_index) {
743 num value = node_vec->entries[node_index].data;
745 for (edge_index = 0; edge_index < max_degree; ++edge_index) {
746 edge = node->edges[edge_index];
748 is_src = edge->src == node;
751 vec = vector_copy(pbqp, edge->tgt->costs);
752 vector_add_matrix_row(vec, mat, node_index);
754 vec = vector_copy(pbqp, edge->src->costs);
755 vector_add_matrix_col(vec, mat, node_index);
758 value = pbqp_add(value, vector_get_min(vec));
760 obstack_free(&pbqp->obstack, vec);
765 min_index = node_index;
769 if (pbqp->dump_file) {
770 fprintf(pbqp->dump_file, "node n%d is set to %d<br><br>\n",
771 node->index, min_index);
772 fprintf(pbqp->dump_file, "Minimal cost of RN reduction: %lld<br>\n",
776 node->solution = min_index;
778 /* Now that we found the local minimum set all other costs to infinity. */
779 for (node_index = 0; node_index < node_len; ++node_index) {
780 if (node_index != min_index) {
781 node_vec->entries[node_index].data = INF_COSTS;
785 /* Add all incident edges to edge bucket, since they are now independent. */
786 for (edge_index = 0; edge_index < max_degree; ++edge_index) {
787 insert_into_edge_bucket(node->edges[edge_index]);
791 void back_propagate_RI(pbqp *pbqp, pbqp_node *node)
802 edge = node->edges[0];
804 is_src = edge->src == node;
810 vector_add_matrix_col(vec, mat, other->solution);
814 vector_add_matrix_row(vec, mat, other->solution);
817 node->solution = vector_get_min_index(vec);
818 if (pbqp->dump_file) {
819 fprintf(pbqp->dump_file, "node n%d is set to %d<br>\n", node->index, node->solution);
823 void back_propagate_RII(pbqp *pbqp, pbqp_node *node)
825 pbqp_edge *src_edge = node->edges[0];
826 pbqp_edge *tgt_edge = node->edges[1];
827 int src_is_src = src_edge->src == node;
828 int tgt_is_src = tgt_edge->src == node;
829 pbqp_matrix *src_mat;
830 pbqp_matrix *tgt_mat;
841 src_node = src_edge->tgt;
843 src_node = src_edge->src;
847 tgt_node = tgt_edge->tgt;
849 tgt_node = tgt_edge->src;
852 /* Swap nodes if necessary. */
853 if (tgt_node->index < src_node->index) {
865 src_is_src = src_edge->src == node;
866 tgt_is_src = tgt_edge->src == node;
869 src_mat = src_edge->costs;
870 tgt_mat = tgt_edge->costs;
872 node_vec = node->costs;
874 row_index = src_node->solution;
875 col_index = tgt_node->solution;
877 vec = vector_copy(pbqp, node_vec);
880 vector_add_matrix_col(vec, src_mat, row_index);
882 vector_add_matrix_row(vec, src_mat, row_index);
886 vector_add_matrix_col(vec, tgt_mat, col_index);
888 vector_add_matrix_row(vec, tgt_mat, col_index);
891 node->solution = vector_get_min_index(vec);
892 if (pbqp->dump_file) {
893 fprintf(pbqp->dump_file, "node n%d is set to %d<br>\n", node->index, node->solution);
896 obstack_free(&pbqp->obstack, vec);
899 int node_is_reduced(pbqp_node *node)
901 if (!reduced_bucket) return 0;
904 if (ARR_LEN(node->edges) == 0) return 1;
906 return node_bucket_contains(reduced_bucket, node);