MinSteinerTreeMehlhorn.h

Go to the documentation of this file.

 
#pragma once
 
#include <ogdf/basic/List.h>
#include <ogdf/graphalg/MinSteinerTreeModule.h>
#include <ogdf/graphalg/Voronoi.h>
#include <ogdf/graphalg/steiner_tree/EdgeWeightedGraphCopy.h>
 
namespace ogdf {
 
template<typename T>
class MinSteinerTreeMehlhorn : public MinSteinerTreeModule<T> {
public:
    MinSteinerTreeMehlhorn() { }
 
    virtual ~MinSteinerTreeMehlhorn() { }
 
    static void calculateCompleteGraph(const EdgeWeightedGraph<T>& wG, const List<node>& terminals,
            const Voronoi<T>& voronoi, EdgeArray<edge>& bridges,
            EdgeWeightedGraphCopy<T>& completeTerminalGraph);
 
protected:
    virtual T computeSteinerTree(const EdgeWeightedGraph<T>& G, const List<node>& terminals,
            const NodeArray<bool>& isTerminal, EdgeWeightedGraphCopy<T>*& finalSteinerTree) override;
 
    void reinsertShortestPaths(EdgeWeightedGraphCopy<T>& completeTerminalGraph,
            const Voronoi<T>& voronoi, const NodeArray<edge>& mstPred, const EdgeArray<edge>& bridges,
            EdgeWeightedGraphCopy<T>& finalSteinerTree, const EdgeWeightedGraph<T>& wG);
 
    void insertPath(node u, const Voronoi<T>& voronoi, EdgeWeightedGraphCopy<T>& finalSteinerTree,
            const EdgeWeightedGraph<T>& wG);
 
    struct MehlhornTriple {
        node u;
        node v;
        T value;
        edge bridge;
    };
 
    class MehlhornTripleBucketMaxFunc : public BucketFunc<MehlhornTriple> {
    public:
        MehlhornTripleBucketMaxFunc() { }
 
        int getBucket(const MehlhornTriple& MT) {
            int source_index = MT.u->index();
            int target_index = MT.v->index();
            OGDF_ASSERT(source_index != target_index);
            if (source_index < target_index) {
                return target_index;
            } else {
                return source_index;
            }
        }
    };
 
    class MehlhornTripleBucketMinFunc : public BucketFunc<MehlhornTriple> {
    public:
        MehlhornTripleBucketMinFunc() { }
 
        int getBucket(const MehlhornTriple& MT) {
            int source_index = MT.u->index();
            int target_index = MT.v->index();
            OGDF_ASSERT(source_index != target_index);
            if (source_index < target_index) {
                return source_index;
            } else {
                return target_index;
            }
        }
    };
};
 
template<typename T>
T MinSteinerTreeMehlhorn<T>::computeSteinerTree(const EdgeWeightedGraph<T>& G,
        const List<node>& terminals, const NodeArray<bool>& isTerminal,
        EdgeWeightedGraphCopy<T>*& finalSteinerTree) {
    EdgeWeightedGraphCopy<T> completeTerminalGraph;
    EdgeArray<edge> bridges;
    Voronoi<T> voronoi(G, G.edgeWeights(), terminals);
 
    calculateCompleteGraph(G, terminals, voronoi, bridges, completeTerminalGraph);
 
    NodeArray<edge> mstPred(completeTerminalGraph);
    computeMinST(completeTerminalGraph, completeTerminalGraph.edgeWeights(), mstPred);
 
    finalSteinerTree = new EdgeWeightedGraphCopy<T>;
    finalSteinerTree->setOriginalGraph(G);
 
    reinsertShortestPaths(completeTerminalGraph, voronoi, mstPred, bridges, *finalSteinerTree, G);
 
    T mstWeight = makeMinimumSpanningTree(*finalSteinerTree, finalSteinerTree->edgeWeights());
    mstWeight -= MinSteinerTreeModule<T>::pruneAllDanglingSteinerPaths(*finalSteinerTree, isTerminal);
 
    return mstWeight;
}
 
template<typename T>
void MinSteinerTreeMehlhorn<T>::calculateCompleteGraph(const EdgeWeightedGraph<T>& wG,
        const List<node>& terminals, const Voronoi<T>& voronoi, EdgeArray<edge>& bridges,
        EdgeWeightedGraphCopy<T>& completeTerminalGraph) {
    completeTerminalGraph.setOriginalGraph(wG);
 
    for (node v : terminals) {
        completeTerminalGraph.newNode(v);
    }
    if (completeTerminalGraph.numberOfNodes() <= 1) {
        bridges.init(completeTerminalGraph);
        return;
    }
 
    // extract complete graph edges
    List<MehlhornTriple> triples;
    for (edge e : wG.edges) {
        MehlhornTriple triple;
        triple.u = voronoi.seed(e->source());
        triple.v = voronoi.seed(e->target());
        if (triple.u != triple.v) {
            triple.value =
                    voronoi.distance(e->source()) + voronoi.distance(e->target()) + wG.weight(e);
            triple.bridge = e;
            triples.pushBack(triple);
        }
    }
 
    MehlhornTripleBucketMaxFunc mtbMax;
    MehlhornTripleBucketMinFunc mtbMin;
 
    triples.bucketSort(0, wG.maxNodeIndex(), mtbMax);
    triples.bucketSort(0, wG.maxNodeIndex(), mtbMin);
 
    int currentSource = triples.front().u->index();
    int currentTarget = triples.front().v->index();
    ListConstIterator<MehlhornTriple> minTriple = triples.begin();
 
    bridges.init(completeTerminalGraph);
 
    for (ListConstIterator<MehlhornTriple> mtIt = triples.begin().succ(); mtIt.valid(); ++mtIt) {
        if (((*mtIt).u->index() == currentSource && (*mtIt).v->index() == currentTarget)
                || ((*mtIt).u->index() == currentTarget && (*mtIt).v->index() == currentSource)) {
            if ((*mtIt).value < (*minTriple).value) {
                minTriple = mtIt;
            }
        } else {
            // add new direct edge
            edge tmp = completeTerminalGraph.newEdge(completeTerminalGraph.copy((*minTriple).u),
                    completeTerminalGraph.copy((*minTriple).v), (*minTriple).value);
            bridges[tmp] = (*minTriple).bridge;
 
            currentSource = (*mtIt).u->index();
            currentTarget = (*mtIt).v->index();
            minTriple = mtIt;
        }
    }
    // insert last triple
    if (minTriple.valid()) {
        edge tmp = completeTerminalGraph.newEdge(completeTerminalGraph.copy((*minTriple).u),
                completeTerminalGraph.copy((*minTriple).v), (*minTriple).value);
        bridges[tmp] = (*minTriple).bridge;
    }
}
 
template<typename T>
void MinSteinerTreeMehlhorn<T>::reinsertShortestPaths(EdgeWeightedGraphCopy<T>& completeTerminalGraph,
        const Voronoi<T>& voronoi, const NodeArray<edge>& mstPred, const EdgeArray<edge>& bridges,
        EdgeWeightedGraphCopy<T>& finalSteinerTree, const EdgeWeightedGraph<T>& wG) {
    for (node u : completeTerminalGraph.nodes) {
        if (mstPred[u] != nullptr) {
            edge bridge = bridges[mstPred[u]];
            node v = bridge->source();
            node w = bridge->target();
            insertPath(v, voronoi, finalSteinerTree, wG);
            insertPath(w, voronoi, finalSteinerTree, wG);
            edge e = finalSteinerTree.newEdge(finalSteinerTree.copy(v), finalSteinerTree.copy(w),
                    wG.weight(bridge));
            finalSteinerTree.setEdge(bridge, e);
        }
    }
}
 
template<typename T>
void MinSteinerTreeMehlhorn<T>::insertPath(node u, const Voronoi<T>& voronoi,
        EdgeWeightedGraphCopy<T>& finalSteinerTree, const EdgeWeightedGraph<T>& wG) {
    node currentSource;
    node currentTarget = finalSteinerTree.copy(u);
    if (!currentTarget) {
        currentTarget = finalSteinerTree.newNode(u);
    }
    edge e = voronoi.predecessorEdge(u);
    edge newE;
    while (e && finalSteinerTree.chain(e).empty()) { // e is not in ST yet
        if ((currentSource = finalSteinerTree.copy(
                     e->opposite(finalSteinerTree.original(currentTarget))))
                == nullptr) {
            currentSource =
                    finalSteinerTree.newNode(e->opposite(finalSteinerTree.original(currentTarget)));
        }
        if (finalSteinerTree.original(currentSource) == e->source()) {
            newE = finalSteinerTree.newEdge(currentSource, currentTarget, wG.weight(e));
        } else {
            newE = finalSteinerTree.newEdge(currentTarget, currentSource, wG.weight(e));
        }
        finalSteinerTree.setEdge(e, newE);
        currentTarget = currentSource;
        e = voronoi.predecessorEdge(finalSteinerTree.original(currentTarget));
    }
}
 
namespace steiner_tree {
 
template<typename T>
T constructTerminalSpanningTreeUsingVoronoiRegions(EdgeWeightedGraphCopy<T>& terminalSpanningTree,
        const EdgeWeightedGraph<T>& graph, const List<node>& terminals) {
    EdgeArray<edge> bridges;
    Voronoi<T> voronoi(graph, graph.edgeWeights(), terminals);
 
    MinSteinerTreeMehlhorn<T>::calculateCompleteGraph(graph, terminals, voronoi, bridges,
            terminalSpanningTree);
 
    return makeMinimumSpanningTree(terminalSpanningTree, terminalSpanningTree.edgeWeights());
}
}
}