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 degree = pbqp_node_get_degree(node);
73 /* We have only one bucket for nodes with arity >= 3. */
78 node_bucket_insert(&node_buckets[degree], 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 = pbqp_node_get_degree(src_node);
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 = pbqp_node_get_degree(tgt_node);
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)
205 unsigned degree = pbqp_node_get_degree(node);
206 /* Assume node lost one incident edge. */
207 unsigned old_degree = degree + 1;
209 if (!buckets_filled) return;
211 /* Same bucket as before */
212 if (degree > 2) return;
214 if (!node_bucket_contains(node_buckets[old_degree], node)) {
215 /* Old arity is new arity, so we have nothing to do. */
216 assert(node_bucket_contains(node_buckets[degree], node));
220 /* Delete node from old bucket... */
221 node_bucket_remove(&node_buckets[old_degree], node);
223 /* ..and add to new one. */
224 node_bucket_insert(&node_buckets[degree], node);
227 static void check_melting_possibility(pbqp *pbqp, pbqp_edge *edge)
242 src_node = edge->src;
243 tgt_node = edge->tgt;
247 src_vec = src_node->costs;
248 tgt_vec = tgt_node->costs;
252 src_len = src_vec->len;
253 tgt_len = tgt_vec->len;
260 if (src_len == 1 && tgt_len == 1) {
261 //panic("Something is wrong");
265 for (src_index = 0; src_index < src_len; ++src_index) {
267 if (src_vec->entries[src_index].data == INF_COSTS) {
270 for (tgt_index = 0; tgt_index < tgt_len; ++tgt_index) {
271 if (tgt_vec->entries[tgt_index].data == INF_COSTS) {
274 if (mat->entries[src_index * tgt_len + tgt_index] == 0) {
283 if (mat->entries[src_index * tgt_len + tgt_index] == INF_COSTS) {
289 allRowsOk &= onlyOneZero;
293 for (tgt_index = 0; tgt_index < tgt_len; ++tgt_index) {
295 if (tgt_vec->entries[tgt_index].data == INF_COSTS) {
298 for (src_index = 0; src_index < src_len; ++src_index) {
299 if (src_vec->entries[src_index].data == INF_COSTS) {
302 if (mat->entries[src_index * tgt_len + tgt_index] == 0) {
311 if (mat->entries[src_index * tgt_len + tgt_index] == INF_COSTS) {
317 allColsOk &= onlyOneZero;
320 if (allRowsOk && allColsOk) {
325 static void simplify_edge(pbqp *pbqp, pbqp_edge *edge)
338 src_node = edge->src;
339 tgt_node = edge->tgt;
343 /* If edge are already deleted, we have nothing to do. */
344 if (!is_connected(src_node, edge) || !is_connected(tgt_node, edge))
347 if (pbqp->dump_file) {
349 sprintf(txt, "Simplification of Edge n%d-n%d", src_node->index, tgt_node->index);
350 dump_section(pbqp->dump_file, 3, txt);
353 src_vec = src_node->costs;
354 tgt_vec = tgt_node->costs;
358 src_len = src_vec->len;
359 tgt_len = tgt_vec->len;
366 if (pbqp->dump_file) {
367 fputs("Input:<br>\n", pbqp->dump_file);
368 dump_simplifyedge(pbqp, edge);
371 normalize_towards_source(pbqp, edge);
372 normalize_towards_target(pbqp, edge);
374 if (pbqp->dump_file) {
375 fputs("<br>\nOutput:<br>\n", pbqp->dump_file);
376 dump_simplifyedge(pbqp, edge);
379 if (pbqp_matrix_is_zero(mat, src_vec, tgt_vec)) {
380 if (pbqp->dump_file) {
381 fputs("edge has been eliminated<br>\n", pbqp->dump_file);
385 reorder_node(src_node);
386 reorder_node(tgt_node);
388 //check_melting_possibility(pbqp, edge);
392 static void initial_simplify_edges(pbqp *pbqp)
399 if (pbqp->dump_file) {
400 pbqp_dump_input(pbqp);
401 dump_section(pbqp->dump_file, 1, "2. Simplification of Cost Matrices");
404 node_len = pbqp->num_nodes;
408 /* First simplify all edges. */
409 for (node_index = 0; node_index < node_len; ++node_index) {
411 pbqp_node *node = get_node(pbqp, node_index);
418 edge_len = pbqp_node_get_degree(node);
420 for (edge_index = 0; edge_index < edge_len; ++edge_index) {
421 pbqp_edge *edge = edges[edge_index];
423 /* Simplify only once per edge. */
424 if (node != edge->src) continue;
426 simplify_edge(pbqp, edge);
431 num determine_solution(FILE *file)
438 dump_section(file, 1, "4. Determine Solution/Minimum");
439 dump_section(file, 2, "4.1. Trivial Solution");
442 /* Solve trivial nodes and calculate solution. */
443 node_len = node_bucket_get_length(node_buckets[0]);
444 for (node_index = 0; node_index < node_len; ++node_index) {
445 pbqp_node *node = node_buckets[0][node_index];
448 node->solution = vector_get_min_index(node->costs);
449 solution = pbqp_add(solution,
450 node->costs->entries[node->solution].data);
452 fprintf(file, "node n%d is set to %d<br>\n", node->index, node->solution);
453 dump_node(file, node);
458 dump_section(file, 2, "Minimum");
459 fprintf(file, "Minimum is equal to %lld.", solution);
465 static void back_propagate(pbqp *pbqp)
468 unsigned node_len = node_bucket_get_length(reduced_bucket);
471 if (pbqp->dump_file) {
472 dump_section(pbqp->dump_file, 2, "Back Propagation");
475 for (node_index = node_len; node_index > 0; --node_index) {
476 pbqp_node *node = reduced_bucket[node_index - 1];
478 switch (pbqp_node_get_degree(node)) {
480 back_propagate_RI(pbqp, node);
483 back_propagate_RII(pbqp, node);
486 panic("Only nodes with degree one or two should be in this bucket");
492 void solve_pbqp_heuristical(pbqp *pbqp)
494 /* Reduce nodes degree ... */
495 initial_simplify_edges(pbqp);
497 /* ... and put node into bucket representing their degree. */
498 fill_node_buckets(pbqp);
501 if (edge_bucket_get_length(edge_bucket) > 0) {
503 } else if (node_bucket_get_length(node_buckets[1]) > 0) {
505 } else if (node_bucket_get_length(node_buckets[2]) > 0) {
507 } else if (node_bucket_get_length(node_buckets[3]) > 0) {
514 pbqp->solution = determine_solution(pbqp->dump_file);
516 /* Solve reduced nodes. */
517 back_propagate(pbqp);
522 void apply_edge(pbqp *pbqp)
524 pbqp_edge *edge = edge_bucket_pop(&edge_bucket);
526 simplify_edge(pbqp, edge);
529 void apply_RI(pbqp *pbqp)
531 pbqp_node *node = node_bucket_pop(&node_buckets[1]);
532 pbqp_edge *edge = node->edges[0];
533 pbqp_matrix *mat = edge->costs;
534 int is_src = edge->src == node;
535 pbqp_node *other_node;
538 other_node = edge->tgt;
540 other_node = edge->src;
543 if (pbqp->dump_file) {
545 sprintf(txt, "RI-Reduction of Node n%d", node->index);
546 dump_section(pbqp->dump_file, 2, txt);
547 pbqp_dump_graph(pbqp);
548 fputs("<br>\nBefore reduction:<br>\n", pbqp->dump_file);
549 dump_node(pbqp->dump_file, node);
550 dump_node(pbqp->dump_file, other_node);
551 dump_edge(pbqp->dump_file, edge);
555 pbqp_matrix_add_to_all_cols(mat, node->costs);
556 normalize_towards_target(pbqp, edge);
558 pbqp_matrix_add_to_all_rows(mat, node->costs);
559 normalize_towards_source(pbqp, edge);
561 disconnect_edge(other_node, edge);
563 if (pbqp->dump_file) {
564 fputs("<br>\nAfter reduction:<br>\n", pbqp->dump_file);
565 dump_node(pbqp->dump_file, other_node);
568 reorder_node(other_node);
570 /* Add node to back propagation list. */
571 node_bucket_insert(&reduced_bucket, node);
574 void apply_RII(pbqp *pbqp)
576 pbqp_node *node = node_bucket_pop(&node_buckets[2]);
577 pbqp_edge *src_edge = node->edges[0];
578 pbqp_edge *tgt_edge = node->edges[1];
579 int src_is_src = src_edge->src == node;
580 int tgt_is_src = tgt_edge->src == node;
581 pbqp_matrix *src_mat;
582 pbqp_matrix *tgt_mat;
599 src_node = src_edge->tgt;
601 src_node = src_edge->src;
605 tgt_node = tgt_edge->tgt;
607 tgt_node = tgt_edge->src;
610 /* Swap nodes if necessary. */
611 if (tgt_node->index < src_node->index) {
623 src_is_src = src_edge->src == node;
624 tgt_is_src = tgt_edge->src == node;
627 if (pbqp->dump_file) {
629 sprintf(txt, "RII-Reduction of Node n%d", node->index);
630 dump_section(pbqp->dump_file, 2, txt);
631 pbqp_dump_graph(pbqp);
632 fputs("<br>\nBefore reduction:<br>\n", pbqp->dump_file);
633 dump_node(pbqp->dump_file, src_node);
634 dump_edge(pbqp->dump_file, src_edge);
635 dump_node(pbqp->dump_file, node);
636 dump_edge(pbqp->dump_file, tgt_edge);
637 dump_node(pbqp->dump_file, tgt_node);
640 src_mat = src_edge->costs;
641 tgt_mat = tgt_edge->costs;
643 src_vec = src_node->costs;
644 tgt_vec = tgt_node->costs;
645 node_vec = node->costs;
647 row_len = src_vec->len;
648 col_len = tgt_vec->len;
649 node_len = node_vec->len;
651 mat = pbqp_matrix_alloc(pbqp, row_len, col_len);
653 for (row_index = 0; row_index < row_len; ++row_index) {
654 for (col_index = 0; col_index < col_len; ++col_index) {
655 vec = vector_copy(pbqp, node_vec);
658 vector_add_matrix_col(vec, src_mat, row_index);
660 vector_add_matrix_row(vec, src_mat, row_index);
664 vector_add_matrix_col(vec, tgt_mat, col_index);
666 vector_add_matrix_row(vec, tgt_mat, col_index);
669 mat->entries[row_index * col_len + col_index] = vector_get_min(vec);
671 obstack_free(&pbqp->obstack, vec);
675 pbqp_edge *edge = get_edge(pbqp, src_node->index, tgt_node->index);
677 /* Disconnect node. */
678 disconnect_edge(src_node, src_edge);
679 disconnect_edge(tgt_node, tgt_edge);
681 /* Add node to back propagation list. */
682 node_bucket_insert(&reduced_bucket, node);
685 edge = alloc_edge(pbqp, src_node->index, tgt_node->index, mat);
687 pbqp_matrix_add(edge->costs, mat);
689 /* Free local matrix. */
690 obstack_free(&pbqp->obstack, mat);
692 reorder_node(src_node);
693 reorder_node(tgt_node);
696 if (pbqp->dump_file) {
697 fputs("<br>\nAfter reduction:<br>\n", pbqp->dump_file);
698 dump_edge(pbqp->dump_file, edge);
701 /* Edge has changed so we simplify it. */
702 simplify_edge(pbqp, edge);
705 void apply_RN(pbqp *pbqp)
707 pbqp_node **bucket = node_buckets[3];
708 unsigned bucket_len = node_bucket_get_length(bucket);
709 unsigned bucket_index;
710 pbqp_node *node = NULL;
716 unsigned max_degree = 0;
719 unsigned min_index = 0;
725 /* Search for node with maximum degree. */
726 for (bucket_index = 0; bucket_index < bucket_len; ++bucket_index) {
727 pbqp_node *candidate = bucket[bucket_index];
728 unsigned degree = pbqp_node_get_degree(candidate);
730 if (degree > max_degree) {
736 node_vec = node->costs;
737 node_len = node_vec->len;
739 if (pbqp->dump_file) {
741 sprintf(txt, "RN-Reduction of Node n%d", node->index);
742 dump_section(pbqp->dump_file, 2, txt);
743 pbqp_dump_graph(pbqp);
746 for (node_index = 0; node_index < node_len; ++node_index) {
747 num value = node_vec->entries[node_index].data;
749 for (edge_index = 0; edge_index < max_degree; ++edge_index) {
750 edge = node->edges[edge_index];
752 is_src = edge->src == node;
755 vec = vector_copy(pbqp, edge->tgt->costs);
756 vector_add_matrix_row(vec, mat, node_index);
758 vec = vector_copy(pbqp, edge->src->costs);
759 vector_add_matrix_col(vec, mat, node_index);
762 value = pbqp_add(value, vector_get_min(vec));
764 obstack_free(&pbqp->obstack, vec);
769 min_index = node_index;
773 if (pbqp->dump_file) {
774 fprintf(pbqp->dump_file, "node n%d is set to %d<br><br>\n",
775 node->index, min_index);
776 fprintf(pbqp->dump_file, "Minimal cost of RN reduction: %lld<br>\n",
780 node->solution = min_index;
782 /* Now that we found the local minimum set all other costs to infinity. */
783 for (node_index = 0; node_index < node_len; ++node_index) {
784 if (node_index != min_index) {
785 node_vec->entries[node_index].data = INF_COSTS;
789 /* Add all incident edges to edge bucket, since they are now independent. */
790 for (edge_index = 0; edge_index < max_degree; ++edge_index) {
791 insert_into_edge_bucket(node->edges[edge_index]);
795 void apply_Brute_Force(pbqp *pbqp)
797 pbqp_node **bucket = node_buckets[3];
798 unsigned bucket_len = node_bucket_get_length(bucket);
799 unsigned bucket_index;
800 pbqp_node *node = NULL;
806 unsigned max_degree = 0;
809 unsigned min_index = 0;
815 /* Search for node with maximum degree. */
816 for (bucket_index = 0; bucket_index < bucket_len; ++bucket_index) {
817 pbqp_node *candidate = bucket[bucket_index];
818 unsigned degree = pbqp_node_get_degree(candidate);
820 if (degree > max_degree) {
826 node_vec = node->costs;
827 node_len = node_vec->len;
829 if (pbqp->dump_file) {
831 sprintf(txt, "RN-Reduction of Node n%d", node->index);
832 dump_section(pbqp->dump_file, 2, txt);
833 pbqp_dump_graph(pbqp);
836 for (node_index = 0; node_index < node_len; ++node_index) {
837 num value = node_vec->entries[node_index].data;
844 min_index = node_index;
848 if (pbqp->dump_file) {
849 fprintf(pbqp->dump_file, "node n%d is set to %d<br><br>\n",
850 node->index, min_index);
851 fprintf(pbqp->dump_file, "Minimal cost of RN reduction: %lld<br>\n",
855 node->solution = min_index;
857 /* Now that we found the local minimum set all other costs to infinity. */
858 for (node_index = 0; node_index < node_len; ++node_index) {
859 if (node_index != min_index) {
860 node_vec->entries[node_index].data = INF_COSTS;
864 /* Add all incident edges to edge bucket, since they are now independent. */
865 for (edge_index = 0; edge_index < max_degree; ++edge_index) {
866 insert_into_edge_bucket(node->edges[edge_index]);
870 void back_propagate_RI(pbqp *pbqp, pbqp_node *node)
881 edge = node->edges[0];
883 is_src = edge->src == node;
889 vector_add_matrix_col(vec, mat, other->solution);
893 vector_add_matrix_row(vec, mat, other->solution);
896 node->solution = vector_get_min_index(vec);
897 if (pbqp->dump_file) {
898 fprintf(pbqp->dump_file, "node n%d is set to %d<br>\n", node->index, node->solution);
902 void back_propagate_RII(pbqp *pbqp, pbqp_node *node)
904 pbqp_edge *src_edge = node->edges[0];
905 pbqp_edge *tgt_edge = node->edges[1];
906 int src_is_src = src_edge->src == node;
907 int tgt_is_src = tgt_edge->src == node;
908 pbqp_matrix *src_mat;
909 pbqp_matrix *tgt_mat;
920 src_node = src_edge->tgt;
922 src_node = src_edge->src;
926 tgt_node = tgt_edge->tgt;
928 tgt_node = tgt_edge->src;
931 /* Swap nodes if necessary. */
932 if (tgt_node->index < src_node->index) {
944 src_is_src = src_edge->src == node;
945 tgt_is_src = tgt_edge->src == node;
948 src_mat = src_edge->costs;
949 tgt_mat = tgt_edge->costs;
951 node_vec = node->costs;
953 row_index = src_node->solution;
954 col_index = tgt_node->solution;
956 vec = vector_copy(pbqp, node_vec);
959 vector_add_matrix_col(vec, src_mat, row_index);
961 vector_add_matrix_row(vec, src_mat, row_index);
965 vector_add_matrix_col(vec, tgt_mat, col_index);
967 vector_add_matrix_row(vec, tgt_mat, col_index);
970 node->solution = vector_get_min_index(vec);
971 if (pbqp->dump_file) {
972 fprintf(pbqp->dump_file, "node n%d is set to %d<br>\n", node->index, node->solution);
975 obstack_free(&pbqp->obstack, vec);
978 int node_is_reduced(pbqp_node *node)
980 if (!reduced_bucket) return 0;
982 if (pbqp_node_get_degree(node) == 0) return 1;
984 return node_bucket_contains(reduced_bucket, node);