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 pbqp_matrix_sub_row_value(mat, src_index, tgt_vec, min);
130 src_vec->entries[src_index].data = pbqp_add(
131 src_vec->entries[src_index].data, min);
133 if (min == INF_COSTS) {
134 insert_into_edge_bucket(edge);
140 static void normalize_towards_target(pbqp *pbqp, pbqp_edge *edge)
154 src_node = edge->src;
155 tgt_node = edge->tgt;
159 src_vec = src_node->costs;
160 tgt_vec = tgt_node->costs;
164 src_len = src_vec->len;
165 tgt_len = tgt_vec->len;
172 for (tgt_index = 0; tgt_index < tgt_len; ++tgt_index) {
173 num min = pbqp_matrix_get_col_min(mat, tgt_index, src_vec);
176 pbqp_matrix_sub_col_value(mat, tgt_index, src_vec, min);
177 tgt_vec->entries[tgt_index].data = pbqp_add(
178 tgt_vec->entries[tgt_index].data, min);
180 if (min == INF_COSTS) {
181 insert_into_edge_bucket(edge);
187 static void reorder_node(pbqp_node *node)
191 unsigned old_bucket_len;
192 unsigned old_bucket_index;
193 pbqp_node **old_bucket;
196 if (!buckets_filled) return;
200 arity = ARR_LEN(node->edges);
202 /* Same bucket as before */
203 if (arity > 2) return;
205 /* Assume node lost one incident edge. */
206 old_arity = arity + 1;
207 old_bucket = node_buckets[old_arity];
208 old_bucket_len = ARR_LEN(old_bucket);
209 old_bucket_index = node->bucket_index;
211 if (old_bucket_len <= old_bucket_index ||
212 old_bucket[old_bucket_index] != node) {
213 /* Old arity is new arity, so we have nothing to do. */
214 assert(old_bucket_index < ARR_LEN(node_buckets[arity]) &&
215 node_buckets[arity][old_bucket_index] == node);
219 assert(old_bucket[old_bucket_index] == node);
221 /* Delete node from old bucket... */
222 other = old_bucket[old_bucket_len - 1];
223 other->bucket_index = old_bucket_index;
224 old_bucket[old_bucket_index] = other;
225 ARR_SHRINKLEN(node_buckets[old_arity], old_bucket_len - 1);
227 /* ..and add to new one. */
228 node->bucket_index = ARR_LEN(node_buckets[arity]);
229 ARR_APP1(pbqp_node*, node_buckets[arity], node);
232 static void simplify_edge(pbqp *pbqp, pbqp_edge *edge)
245 src_node = edge->src;
246 tgt_node = edge->tgt;
250 /* If edge are already deleted, we have nothing to do. */
251 if (!is_connected(src_node, edge) || !is_connected(tgt_node, edge))
254 if (pbqp->dump_file) {
256 sprintf(txt, "Simplification of Edge n%d-n%d", src_node->index, tgt_node->index);
257 dump_section(pbqp->dump_file, 3, txt);
260 src_vec = src_node->costs;
261 tgt_vec = tgt_node->costs;
265 src_len = src_vec->len;
266 tgt_len = tgt_vec->len;
273 if (pbqp->dump_file) {
274 fputs("Input:<br>\n", pbqp->dump_file);
275 dump_simplifyedge(pbqp, edge);
278 normalize_towards_source(pbqp, edge);
279 normalize_towards_target(pbqp, edge);
281 if (pbqp->dump_file) {
282 fputs("<br>\nOutput:<br>\n", pbqp->dump_file);
283 dump_simplifyedge(pbqp, edge);
286 if (pbqp_matrix_is_zero(mat, src_vec, tgt_vec)) {
287 if (pbqp->dump_file) {
288 fputs("edge has been eliminated<br>\n", pbqp->dump_file);
292 reorder_node(src_node);
293 reorder_node(tgt_node);
297 void solve_pbqp_heuristical(pbqp *pbqp)
304 if (pbqp->dump_file) {
305 pbqp_dump_input(pbqp);
306 dump_section(pbqp->dump_file, 1, "2. Simplification of Cost Matrices");
309 node_len = pbqp->num_nodes;
313 /* First simplify all edges. */
314 for (node_index = 0; node_index < node_len; ++node_index) {
316 pbqp_node *node = get_node(pbqp, node_index);
323 edge_len = ARR_LEN(edges);
325 for (edge_index = 0; edge_index < edge_len; ++edge_index) {
326 pbqp_edge *edge = edges[edge_index];
328 /* Simplify only once per edge. */
329 if (node != edge->src) continue;
331 simplify_edge(pbqp, edge);
335 /* Put node into bucket representing their arity. */
336 fill_node_buckets(pbqp);
339 if (ARR_LEN(edge_bucket) > 0) {
341 } else if (ARR_LEN(node_buckets[1]) > 0) {
343 } else if (ARR_LEN(node_buckets[2]) > 0) {
345 } else if (ARR_LEN(node_buckets[3]) > 0) {
346 panic("Please implement RN simplification");
352 if (pbqp->dump_file) {
353 dump_section(pbqp->dump_file, 1, "4. Determine Solution/Minimum");
354 dump_section(pbqp->dump_file, 2, "4.1. Trivial Solution");
357 /* Solve trivial nodes and calculate solution. */
358 node_len = ARR_LEN(node_buckets[0]);
359 for (node_index = 0; node_index < node_len; ++node_index) {
360 pbqp_node *node = node_buckets[0][node_index];
363 node->solution = vector_get_min_index(node->costs);
364 pbqp->solution = pbqp_add(pbqp->solution,
365 node->costs->entries[node->solution].data);
366 if (pbqp->dump_file) {
367 fprintf(pbqp->dump_file, "node n%d is set to %d<br>\n", node->index, node->solution);
368 dump_node(pbqp, node);
372 if (pbqp->dump_file) {
373 dump_section(pbqp->dump_file, 2, "Minimum");
374 fprintf(pbqp->dump_file, "Minimum is equal to %d.", pbqp->solution);
375 dump_section(pbqp->dump_file, 2, "Back Propagation");
378 /* Solve reduced nodes. */
379 node_len = ARR_LEN(reduced_bucket);
380 for (node_index = node_len; node_index > 0; --node_index) {
381 pbqp_node *node = reduced_bucket[node_index - 1];
384 switch (ARR_LEN(node->edges)) {
386 back_propagate_RI(pbqp, node);
389 back_propagate_RII(pbqp, node);
392 panic("Only nodes with degree one or two should be in this bucket");
400 void apply_edge(pbqp *pbqp)
402 unsigned bucket_len = ARR_LEN(edge_bucket);
403 pbqp_edge *edge = edge_bucket[bucket_len - 1];
405 ARR_SHRINKLEN(edge_bucket, (int)bucket_len - 1);
407 simplify_edge(pbqp, edge);
410 void apply_RI(pbqp *pbqp)
412 pbqp_node **bucket = node_buckets[1];
413 unsigned bucket_len = ARR_LEN(bucket);
414 pbqp_node *node = bucket[bucket_len - 1];
415 pbqp_edge *edge = node->edges[0];
416 pbqp_matrix *mat = edge->costs;
417 int is_src = edge->src == node;
418 pbqp_node *other_node;
421 other_node = edge->tgt;
423 other_node = edge->src;
426 if (pbqp->dump_file) {
428 sprintf(txt, "RI-Reduktion of Node n%d", node->index);
429 dump_section(pbqp->dump_file, 2, txt);
430 pbqp_dump_graph(pbqp);
431 fputs("<br>\nBefore reduction:<br>\n", pbqp->dump_file);
432 dump_node(pbqp, node);
433 dump_node(pbqp, other_node);
434 dump_edge(pbqp, edge);
438 pbqp_matrix_add_to_all_cols(mat, node->costs);
439 normalize_towards_target(pbqp, edge);
441 pbqp_matrix_add_to_all_rows(mat, node->costs);
442 normalize_towards_source(pbqp, edge);
444 disconnect_edge(other_node, edge);
446 if (pbqp->dump_file) {
447 fputs("<br>\nAfter reduction:<br>\n", pbqp->dump_file);
448 dump_node(pbqp, other_node);
451 /* Remove node from bucket... */
452 ARR_SHRINKLEN(bucket, (int)bucket_len - 1);
453 reorder_node(other_node);
455 /* ...and add it to back propagation list. */
456 node->bucket_index = ARR_LEN(reduced_bucket);
457 ARR_APP1(pbqp_node *, reduced_bucket, node);
460 void apply_RII(pbqp *pbqp)
462 pbqp_node **bucket = node_buckets[2];
463 unsigned bucket_len = ARR_LEN(bucket);
464 pbqp_node *node = bucket[bucket_len - 1];
465 pbqp_edge *src_edge = node->edges[0];
466 pbqp_edge *tgt_edge = node->edges[1];
467 int src_is_src = src_edge->src == node;
468 int tgt_is_src = tgt_edge->src == node;
469 pbqp_matrix *src_mat;
470 pbqp_matrix *tgt_mat;
487 src_node = src_edge->tgt;
489 src_node = src_edge->src;
493 tgt_node = tgt_edge->tgt;
495 tgt_node = tgt_edge->src;
498 /* Swap nodes if necessary. */
499 if (tgt_node->index < src_node->index) {
511 src_is_src = src_edge->src == node;
512 tgt_is_src = tgt_edge->src == node;
515 if (pbqp->dump_file) {
517 sprintf(txt, "RII-Reduktion of Node n%d", node->index);
518 dump_section(pbqp->dump_file, 2, txt);
519 pbqp_dump_graph(pbqp);
520 fputs("<br>\nBefore reduction:<br>\n", pbqp->dump_file);
521 dump_node(pbqp, src_node);
522 dump_edge(pbqp, src_edge);
523 dump_node(pbqp, node);
524 dump_edge(pbqp, tgt_edge);
525 dump_node(pbqp, tgt_node);
528 src_mat = src_edge->costs;
529 tgt_mat = tgt_edge->costs;
531 src_vec = src_node->costs;
532 tgt_vec = tgt_node->costs;
533 node_vec = node->costs;
535 row_len = src_vec->len;
536 col_len = tgt_vec->len;
537 node_len = node_vec->len;
539 mat = pbqp_matrix_alloc(pbqp, row_len, col_len);
541 for (row_index = 0; row_index < row_len; ++row_index) {
542 for (col_index = 0; col_index < col_len; ++col_index) {
543 vec = vector_copy(pbqp, node_vec);
546 vector_add_matrix_col(vec, src_mat, row_index);
548 vector_add_matrix_row(vec, src_mat, row_index);
552 vector_add_matrix_col(vec, tgt_mat, col_index);
554 vector_add_matrix_row(vec, tgt_mat, col_index);
557 mat->entries[row_index * col_len + col_index] = vector_get_min(vec);
559 obstack_free(&pbqp->obstack, vec);
563 pbqp_edge *edge = get_edge(pbqp, src_node->index, tgt_node->index);
565 /* Disconnect node. */
566 disconnect_edge(src_node, src_edge);
567 disconnect_edge(tgt_node, tgt_edge);
569 /* Remove node from bucket... */
570 ARR_SHRINKLEN(bucket, (int)bucket_len - 1);
572 /* ...and add it to back propagation list. */
573 node->bucket_index = ARR_LEN(reduced_bucket);
574 ARR_APP1(pbqp_node *, reduced_bucket, node);
577 edge = alloc_edge(pbqp, src_node->index, tgt_node->index, mat);
579 pbqp_matrix_add(edge->costs, mat);
581 /* Free local matrix. */
582 obstack_free(&pbqp->obstack, mat);
584 reorder_node(src_node);
585 reorder_node(tgt_node);
588 if (pbqp->dump_file) {
589 fputs("<br>\nAfter reduction:<br>\n", pbqp->dump_file);
590 dump_edge(pbqp, edge);
593 /* Edge has changed so we simplify it. */
594 simplify_edge(pbqp, edge);
597 void apply_RN(pbqp *pbqp)
599 pbqp_node **bucket = node_buckets[3];
600 unsigned bucket_len = ARR_LEN(bucket);
601 pbqp_node *node = bucket[bucket_len - 1];
603 vector *node_vec = node->costs;
607 unsigned edge_len = ARR_LEN(node->edges);
609 unsigned node_len = ARR_LEN(node_vec);
610 unsigned min_index = 0;
616 for (node_index = 0; node_index < node_len; ++node_index) {
619 for (edge_index = 0; edge_index < edge_len; ++edge_index) {
620 edge = node->edges[edge_index];
622 is_src = edge->src == node;
625 vec = vector_copy(pbqp, edge->tgt->costs);
626 vector_add_matrix_row(vec, mat, node_index);
628 vec = vector_copy(pbqp, edge->src->costs);
629 vector_add_matrix_col(vec, mat, node_index);
632 value = pbqp_add(value, vector_get_min(vec));
634 obstack_free(&pbqp->obstack, vec);
639 min_index = node_index;
643 node->solution = min_index;
645 /* Now that we found the local minimum set all other costs to infinity. */
646 for (node_index = 0; node_index < node_len; ++node_index) {
647 if (node_index != min_index) {
648 node_vec->entries[node_index].data = INF_COSTS;
652 /* Add all incident edges to edge bucket, since they are now independent. */
653 for (edge_index = 0; edge_index < edge_len; ++edge_index) {
654 insert_into_edge_bucket(node->edges[node_index]);
658 void back_propagate_RI(pbqp *pbqp, pbqp_node *node)
669 edge = node->edges[0];
671 is_src = edge->src == node;
677 vector_add_matrix_col(vec, mat, other->solution);
681 vector_add_matrix_row(vec, mat, other->solution);
684 node->solution = vector_get_min_index(vec);
685 if (pbqp->dump_file) {
686 fprintf(pbqp->dump_file, "node n%d is set to %d<br>\n", node->index, node->solution);
690 void back_propagate_RII(pbqp *pbqp, pbqp_node *node)
692 pbqp_edge *src_edge = node->edges[0];
693 pbqp_edge *tgt_edge = node->edges[1];
694 int src_is_src = src_edge->src == node;
695 int tgt_is_src = tgt_edge->src == node;
696 pbqp_matrix *src_mat;
697 pbqp_matrix *tgt_mat;
708 src_node = src_edge->tgt;
710 src_node = src_edge->src;
714 tgt_node = tgt_edge->tgt;
716 tgt_node = tgt_edge->src;
719 /* Swap nodes if necessary. */
720 if (tgt_node->index < src_node->index) {
732 src_is_src = src_edge->src == node;
733 tgt_is_src = tgt_edge->src == node;
736 src_mat = src_edge->costs;
737 tgt_mat = tgt_edge->costs;
739 node_vec = node->costs;
741 row_index = src_node->solution;
742 col_index = tgt_node->solution;
744 vec = vector_copy(pbqp, node_vec);
747 vector_add_matrix_col(vec, src_mat, row_index);
749 vector_add_matrix_row(vec, src_mat, row_index);
753 vector_add_matrix_col(vec, tgt_mat, col_index);
755 vector_add_matrix_row(vec, tgt_mat, col_index);
758 node->solution = vector_get_min_index(vec);
759 if (pbqp->dump_file) {
760 fprintf(pbqp->dump_file, "node n%d is set to %d<br>\n", node->index, node->solution);
763 obstack_free(&pbqp->obstack, vec);
766 int node_is_reduced(pbqp_node *node)
768 if (!reduced_bucket) return 0;
771 if (ARR_LEN(node->edges) == 0) return 1;
773 unsigned bucket_length = ARR_LEN(reduced_bucket);
774 unsigned bucket_index = node->bucket_index;
776 return bucket_index < bucket_length && reduced_bucket[bucket_index] == node;