- brute force solver need own back propagation

[libfirm] / pbqp_node.c

diff --git a/pbqp_node.c b/pbqp_node.c
index 51a2248..b324e49 100644 (file)
--- a/pbqp_node.c
+++ b/pbqp_node.c
@@ -1,12 +1,43 @@
+/*
+ * 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 "vector.h"
 
-pbqp_node *alloc_node(pbqp *pbqp, vector *costs)
+pbqp_node *alloc_node(pbqp *pbqp, unsigned node_index, vector *costs)
 {
        pbqp_node *node = obstack_alloc(&pbqp->obstack, sizeof(*node));
        assert(node);
@@ -15,10 +46,35 @@ pbqp_node *alloc_node(pbqp *pbqp, vector *costs)
        node->costs = vector_copy(pbqp, costs);
        node->bucket_index = UINT_MAX;
        node->solution = UINT_MAX;
+       node->index = node_index;
 
        return node;
 }
 
+int is_connected(pbqp_node *node, pbqp_edge *edge)
+{
+       pbqp_edge **edges;
+       unsigned    edge_index;
+       unsigned    edge_len;
+
+       assert(node);
+       assert(edge);
+
+       if (edge->src != node && edge->tgt != node) return 0;
+
+       edges = node->edges;
+       edge_len = ARR_LEN(edges);
+
+       for (edge_index = 0; edge_index < edge_len; ++edge_index) {
+               pbqp_edge *edge_candidate = edges[edge_index];
+               if (edge_candidate == edge) {
+                       return 1;
+               }
+       }
+
+       return 0;
+}
+
 void disconnect_edge(pbqp_node *node, pbqp_edge *edge)
 {
        pbqp_edge **edges;
@@ -37,3 +93,71 @@ void disconnect_edge(pbqp_node *node, pbqp_edge *edge)
                }
        }
 }
+
+unsigned pbqp_node_get_degree(pbqp_node *node)
+{
+       assert(node);
+       return ARR_LEN(node->edges);
+}
+
+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) {
+               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->costs        = vector_copy(pbqp, node->costs);
+       copy->bucket_index = node->bucket_index;
+       copy->solution     = node->solution;
+       copy->index   = node->index;
+
+       return copy;
+}