| | 1 | /* -*- mode: C++; indent-tabs-mode: nil; -*- |
| | 2 | * |
| | 3 | * This file is a part of LEMON, a generic C++ optimization library. |
| | 4 | * |
| | 5 | * Copyright (C) 2003-2009 |
| | 6 | * Egervary Jeno Kombinatorikus Optimalizalasi Kutatocsoport |
| | 7 | * (Egervary Research Group on Combinatorial Optimization, EGRES). |
| | 8 | * |
| | 9 | * Permission to use, modify and distribute this software is granted |
| | 10 | * provided that this copyright notice appears in all copies. For |
| | 11 | * precise terms see the accompanying LICENSE file. |
| | 12 | * |
| | 13 | * This software is provided "AS IS" with no warranty of any kind, |
| | 14 | * express or implied, and with no claim as to its suitability for any |
| | 15 | * purpose. |
| | 16 | * |
| | 17 | */ |
| | 18 | |
| | 19 | #include <lemon/concepts/digraph.h> |
| | 20 | #include <lemon/smart_graph.h> |
| | 21 | #include <lemon/list_graph.h> |
| | 22 | #include <lemon/lgf_reader.h> |
| | 23 | #include <lemon/bellman_ford.h> |
| | 24 | #include <lemon/path.h> |
| | 25 | |
| | 26 | #include "graph_test.h" |
| | 27 | #include "test_tools.h" |
| | 28 | |
| | 29 | using namespace lemon; |
| | 30 | |
| | 31 | char test_lgf[] = |
| | 32 | "@nodes\n" |
| | 33 | "label\n" |
| | 34 | "0\n" |
| | 35 | "1\n" |
| | 36 | "2\n" |
| | 37 | "3\n" |
| | 38 | "4\n" |
| | 39 | "@arcs\n" |
| | 40 | " length\n" |
| | 41 | "0 1 3\n" |
| | 42 | "1 2 -3\n" |
| | 43 | "1 2 -5\n" |
| | 44 | "1 3 -2\n" |
| | 45 | "0 2 -1\n" |
| | 46 | "1 2 -4\n" |
| | 47 | "0 3 2\n" |
| | 48 | "4 2 -5\n" |
| | 49 | "2 3 1\n" |
| | 50 | "@attributes\n" |
| | 51 | "source 0\n" |
| | 52 | "target 3\n"; |
| | 53 | |
| | 54 | |
| | 55 | void checkBellmanFordCompile() |
| | 56 | { |
| | 57 | typedef int Value; |
| | 58 | typedef concepts::Digraph Digraph; |
| | 59 | typedef concepts::ReadMap<Digraph::Arc,Value> LengthMap; |
| | 60 | typedef BellmanFord<Digraph, LengthMap> BF; |
| | 61 | typedef Digraph::Node Node; |
| | 62 | typedef Digraph::Arc Arc; |
| | 63 | |
| | 64 | Digraph gr; |
| | 65 | Node s, t, n; |
| | 66 | Arc e; |
| | 67 | Value l; |
| | 68 | int k; |
| | 69 | bool b; |
| | 70 | BF::DistMap d(gr); |
| | 71 | BF::PredMap p(gr); |
| | 72 | LengthMap length; |
| | 73 | concepts::Path<Digraph> pp; |
| | 74 | |
| | 75 | { |
| | 76 | BF bf_test(gr,length); |
| | 77 | const BF& const_bf_test = bf_test; |
| | 78 | |
| | 79 | bf_test.run(s); |
| | 80 | bf_test.run(s,k); |
| | 81 | |
| | 82 | bf_test.init(); |
| | 83 | bf_test.addSource(s); |
| | 84 | bf_test.addSource(s, 1); |
| | 85 | b = bf_test.processNextRound(); |
| | 86 | b = bf_test.processNextWeakRound(); |
| | 87 | |
| | 88 | bf_test.start(); |
| | 89 | bf_test.checkedStart(); |
| | 90 | bf_test.limitedStart(k); |
| | 91 | |
| | 92 | l = const_bf_test.dist(t); |
| | 93 | e = const_bf_test.predArc(t); |
| | 94 | s = const_bf_test.predNode(t); |
| | 95 | b = const_bf_test.reached(t); |
| | 96 | d = const_bf_test.distMap(); |
| | 97 | p = const_bf_test.predMap(); |
| | 98 | pp = const_bf_test.path(t); |
| | 99 | |
| | 100 | for (BF::ActiveIt it(const_bf_test); it != INVALID; ++it) {} |
| | 101 | } |
| | 102 | { |
| | 103 | BF::SetPredMap<concepts::ReadWriteMap<Node,Arc> > |
| | 104 | ::SetDistMap<concepts::ReadWriteMap<Node,Value> > |
| | 105 | ::SetOperationTraits<BellmanFordDefaultOperationTraits<Value> > |
| | 106 | ::Create bf_test(gr,length); |
| | 107 | |
| | 108 | LengthMap length_map; |
| | 109 | concepts::ReadWriteMap<Node,Arc> pred_map; |
| | 110 | concepts::ReadWriteMap<Node,Value> dist_map; |
| | 111 | |
| | 112 | bf_test |
| | 113 | .lengthMap(length_map) |
| | 114 | .predMap(pred_map) |
| | 115 | .distMap(dist_map); |
| | 116 | |
| | 117 | bf_test.run(s); |
| | 118 | bf_test.run(s,k); |
| | 119 | |
| | 120 | bf_test.init(); |
| | 121 | bf_test.addSource(s); |
| | 122 | bf_test.addSource(s, 1); |
| | 123 | b = bf_test.processNextRound(); |
| | 124 | b = bf_test.processNextWeakRound(); |
| | 125 | |
| | 126 | bf_test.start(); |
| | 127 | bf_test.checkedStart(); |
| | 128 | bf_test.limitedStart(k); |
| | 129 | |
| | 130 | l = bf_test.dist(t); |
| | 131 | e = bf_test.predArc(t); |
| | 132 | s = bf_test.predNode(t); |
| | 133 | b = bf_test.reached(t); |
| | 134 | pp = bf_test.path(t); |
| | 135 | } |
| | 136 | } |
| | 137 | |
| | 138 | void checkBellmanFordFunctionCompile() |
| | 139 | { |
| | 140 | typedef int Value; |
| | 141 | typedef concepts::Digraph Digraph; |
| | 142 | typedef Digraph::Arc Arc; |
| | 143 | typedef Digraph::Node Node; |
| | 144 | typedef concepts::ReadMap<Digraph::Arc,Value> LengthMap; |
| | 145 | |
| | 146 | Digraph g; |
| | 147 | bool b; |
| | 148 | bellmanFord(g,LengthMap()).run(Node()); |
| | 149 | b = bellmanFord(g,LengthMap()).run(Node(),Node()); |
| | 150 | bellmanFord(g,LengthMap()) |
| | 151 | .predMap(concepts::ReadWriteMap<Node,Arc>()) |
| | 152 | .distMap(concepts::ReadWriteMap<Node,Value>()) |
| | 153 | .run(Node()); |
| | 154 | b=bellmanFord(g,LengthMap()) |
| | 155 | .predMap(concepts::ReadWriteMap<Node,Arc>()) |
| | 156 | .distMap(concepts::ReadWriteMap<Node,Value>()) |
| | 157 | .path(concepts::Path<Digraph>()) |
| | 158 | .dist(Value()) |
| | 159 | .run(Node(),Node()); |
| | 160 | } |
| | 161 | |
| | 162 | |
| | 163 | template <typename Digraph, typename Value> |
| | 164 | void checkBellmanFord() { |
| | 165 | TEMPLATE_DIGRAPH_TYPEDEFS(Digraph); |
| | 166 | typedef typename Digraph::template ArcMap<Value> LengthMap; |
| | 167 | |
| | 168 | Digraph gr; |
| | 169 | Node s, t; |
| | 170 | LengthMap length(gr); |
| | 171 | |
| | 172 | std::istringstream input(test_lgf); |
| | 173 | digraphReader(gr, input). |
| | 174 | arcMap("length", length). |
| | 175 | node("source", s). |
| | 176 | node("target", t). |
| | 177 | run(); |
| | 178 | |
| | 179 | BellmanFord<Digraph, LengthMap> |
| | 180 | bf(gr, length); |
| | 181 | bf.run(s); |
| | 182 | Path<Digraph> p = bf.path(t); |
| | 183 | |
| | 184 | check(bf.reached(t) && bf.dist(t) == -1, "Bellman-Ford found a wrong path."); |
| | 185 | check(p.length() == 3, "path() found a wrong path."); |
| | 186 | check(checkPath(gr, p), "path() found a wrong path."); |
| | 187 | check(pathSource(gr, p) == s, "path() found a wrong path."); |
| | 188 | check(pathTarget(gr, p) == t, "path() found a wrong path."); |
| | 189 | |
| | 190 | ListPath<Digraph> path; |
| | 191 | Value dist; |
| | 192 | bool reached = bellmanFord(gr,length).path(path).dist(dist).run(s,t); |
| | 193 | |
| | 194 | check(reached && dist == -1, "Bellman-Ford found a wrong path."); |
| | 195 | check(path.length() == 3, "path() found a wrong path."); |
| | 196 | check(checkPath(gr, path), "path() found a wrong path."); |
| | 197 | check(pathSource(gr, path) == s, "path() found a wrong path."); |
| | 198 | check(pathTarget(gr, path) == t, "path() found a wrong path."); |
| | 199 | |
| | 200 | for(ArcIt e(gr); e!=INVALID; ++e) { |
| | 201 | Node u=gr.source(e); |
| | 202 | Node v=gr.target(e); |
| | 203 | check(!bf.reached(u) || (bf.dist(v) - bf.dist(u) <= length[e]), |
| | 204 | "Wrong output. dist(target)-dist(source)-arc_length=" << |
| | 205 | bf.dist(v) - bf.dist(u) - length[e]); |
| | 206 | } |
| | 207 | |
| | 208 | for(NodeIt v(gr); v!=INVALID; ++v) { |
| | 209 | if (bf.reached(v)) { |
| | 210 | check(v==s || bf.predArc(v)!=INVALID, "Wrong tree."); |
| | 211 | if (bf.predArc(v)!=INVALID ) { |
| | 212 | Arc e=bf.predArc(v); |
| | 213 | Node u=gr.source(e); |
| | 214 | check(u==bf.predNode(v),"Wrong tree."); |
| | 215 | check(bf.dist(v) - bf.dist(u) == length[e], |
| | 216 | "Wrong distance! Difference: " << |
| | 217 | bf.dist(v) - bf.dist(u) - length[e]); |
| | 218 | } |
| | 219 | } |
| | 220 | } |
| | 221 | } |
| | 222 | |
| | 223 | int main() { |
| | 224 | checkBellmanFord<ListDigraph, int>(); |
| | 225 | checkBellmanFord<SmartDigraph, double>(); |
| | 226 | return 0; |
| | 227 | } |