- brute force solver need own back propagation

[libfirm] / pbqp_node.c

diff --git a/pbqp_node.c b/pbqp_node.c
index b4aaae6..b324e49 100644 (file)
--- a/pbqp_node.c
+++ b/pbqp_node.c
@@ -1,10 +1,40 @@
+/*
+ * Copyright (C) 1995-2008 University of Karlsruhe.  All right reserved.
+ *
+ * This file is part of libFirm.
+ *
+ * This file may be distributed and/or modified under the terms of the
+ * GNU General Public License version 2 as published by the Free Software
+ * Foundation and appearing in the file LICENSE.GPL included in the
+ * packaging of this file.
+ *
+ * Licensees holding valid libFirm Professional Edition licenses may use
+ * this file in accordance with the libFirm Commercial License.
+ * Agreement provided with the Software.
+ *
+ * This file is provided AS IS with NO WARRANTY OF ANY KIND, INCLUDING THE
+ * WARRANTY OF DESIGN, MERCHANTABILITY AND FITNESS FOR A PARTICULAR
+ * PURPOSE.
+ */
+
+/**
+ * @file
+ * @brief   PBQP nodes.
+ * @date    02.10.2008
+ * @author  Sebastian Buchwald
+ * @version $Id$
+ */
+#include "config.h"
+
 #include "adt/array.h"
 
 #include "assert.h"
 
+#include "bucket.h"
+#include "pbqp_edge.h"
+#include "pbqp_edge_t.h"
 #include "pbqp_node.h"
 #include "pbqp_node_t.h"
-#include "pbqp_edge_t.h"
 #include "vector.h"
 
 pbqp_node *alloc_node(pbqp *pbqp, unsigned node_index, vector *costs)
@@ -70,21 +100,64 @@ unsigned pbqp_node_get_degree(pbqp_node *node)
        return ARR_LEN(node->edges);
 }
 
-pbqp_node *pbqp_node_deep_copy(pbqp *pbqp, pbqp_node *node)
+pbqp_node *pbqp_node_deep_copy(pbqp *pbqp, pbqp_node_bucket new_bucket, pbqp_node *node)
 {
        unsigned   edge_index;
        unsigned   edge_length = pbqp_node_get_degree(node);
        pbqp_node *copy        = obstack_alloc(&pbqp->obstack, sizeof(*node));
        assert(copy);
 
+       copy->edges        = NEW_ARR_F(pbqp_edge *, 0);
        for (edge_index = 0; edge_index < edge_length; ++edge_index) {
-               copy->edges[edge_index] = pbqp_edge_deep_copy(node->edges[edge_index]);
+               pbqp_edge *edge_copy;
+               pbqp_edge *edge        = node->edges[edge_index];
+               int        is_src      = edge->src == node;
+
+               if (is_src) {
+                       unsigned other_index = edge->tgt->bucket_index;
+                       unsigned is_copied   = other_index < node->bucket_index;
+
+                       if (is_copied) {
+                               pbqp_node *other_copy = new_bucket[other_index];
+                               unsigned degree = pbqp_node_get_degree(other_copy);
+                               unsigned index;
+
+                               for (index = 0; index < degree; ++index) {
+                                       if (other_copy->edges[index]->src == node) {
+                                               edge_copy      = other_copy->edges[index];
+                                               edge_copy->src = copy;
+                                               break;
+                                       }
+                               }
+                       } else {
+                               edge_copy = pbqp_edge_deep_copy(pbqp, edge, copy, edge->tgt);
+                       }
+               } else {
+                       unsigned other_index = edge->src->bucket_index;
+                       unsigned is_copied   = other_index < node->bucket_index;
+
+                       if (is_copied) {
+                               pbqp_node *other_copy = new_bucket[other_index];
+                               unsigned degree = pbqp_node_get_degree(other_copy);
+                               unsigned index;
+
+                               for (index = 0; index < degree; ++index) {
+                                       if (other_copy->edges[index]->tgt == node) {
+                                               edge_copy      = other_copy->edges[index];
+                                               edge_copy->tgt = copy;
+                                               break;
+                                       }
+                               }
+                       } else {
+                               edge_copy = pbqp_edge_deep_copy(pbqp, edge, edge->src, copy);
+                       }
+               }
+               ARR_APP1(pbqp_edge *, copy->edges, edge_copy);
        }
-       copy->edges        = NEW_ARR_F(pbqp_edge *, 0);
        copy->costs        = vector_copy(pbqp, node->costs);
        copy->bucket_index = node->bucket_index;
        copy->solution     = node->solution;
        copy->index   = node->index;
 
-       return node;
+       return copy;
 }