0
13
0
1 | 1 |
/* -*- mode: C++; indent-tabs-mode: nil; -*- |
2 | 2 |
* |
3 | 3 |
* This file is a part of LEMON, a generic C++ optimization library. |
4 | 4 |
* |
5 |
* Copyright (C) 2003- |
|
5 |
* Copyright (C) 2003-2011 |
|
6 | 6 |
* Egervary Jeno Kombinatorikus Optimalizalasi Kutatocsoport |
7 | 7 |
* (Egervary Research Group on Combinatorial Optimization, EGRES). |
8 | 8 |
* |
9 | 9 |
* Permission to use, modify and distribute this software is granted |
10 | 10 |
* provided that this copyright notice appears in all copies. For |
11 | 11 |
* precise terms see the accompanying LICENSE file. |
12 | 12 |
* |
13 | 13 |
* This software is provided "AS IS" with no warranty of any kind, |
14 | 14 |
* express or implied, and with no claim as to its suitability for any |
15 | 15 |
* purpose. |
16 | 16 |
* |
17 | 17 |
*/ |
18 | 18 |
|
19 | 19 |
namespace lemon { |
20 | 20 |
/*! |
21 | 21 |
|
22 | 22 |
|
23 | 23 |
|
24 | 24 |
\page lgf-format LEMON Graph Format (LGF) |
25 | 25 |
|
26 | 26 |
The \e LGF is a <em>column oriented</em> |
27 | 27 |
file format for storing graphs and associated data like |
28 | 28 |
node and edge maps. |
29 | 29 |
|
30 | 30 |
Each line with \c '#' first non-whitespace |
31 | 31 |
character is considered as a comment line. |
32 | 32 |
|
33 | 33 |
Otherwise the file consists of sections starting with |
34 | 34 |
a header line. The header lines starts with an \c '@' character followed by the |
35 | 35 |
type of section. The standard section types are \c \@nodes, \c |
36 | 36 |
\@arcs and \c \@edges |
37 | 37 |
and \@attributes. Each header line may also have an optional |
38 | 38 |
\e name, which can be use to distinguish the sections of the same |
39 | 39 |
type. |
40 | 40 |
|
41 | 41 |
The standard sections are column oriented, each line consists of |
42 | 42 |
<em>token</em>s separated by whitespaces. A token can be \e plain or |
43 | 43 |
\e quoted. A plain token is just a sequence of non-whitespace characters, |
44 | 44 |
while a quoted token is a |
45 | 45 |
character sequence surrounded by double quotes, and it can also |
46 | 46 |
contain whitespaces and escape sequences. |
47 | 47 |
|
48 | 48 |
The \c \@nodes section describes a set of nodes and associated |
49 | 49 |
maps. The first is a header line, its columns are the names of the |
50 | 50 |
maps appearing in the following lines. |
51 | 51 |
One of the maps must be called \c |
52 | 52 |
"label", which plays special role in the file. |
53 | 53 |
The following |
54 | 54 |
non-empty lines until the next section describes nodes of the |
55 | 55 |
graph. Each line contains the values of the node maps |
56 | 56 |
associated to the current node. |
57 | 57 |
|
58 | 58 |
\code |
59 | 59 |
@nodes |
60 | 60 |
label coordinates size title |
61 | 61 |
1 (10,20) 10 "First node" |
62 | 62 |
2 (80,80) 8 "Second node" |
63 | 63 |
3 (40,10) 10 "Third node" |
64 | 64 |
\endcode |
65 | 65 |
|
66 | 66 |
The \c \@arcs section is very similar to the \c \@nodes section, it |
67 | 67 |
again starts with a header line describing the names of the maps, but |
68 | 68 |
the \c "label" map is not obligatory here. The following lines |
69 | 69 |
describe the arcs. The first two tokens of each line are the source |
70 | 70 |
and the target node of the arc, respectively, then come the map |
71 | 71 |
values. The source and target tokens must be node labels. |
72 | 72 |
|
73 | 73 |
\code |
74 | 74 |
@arcs |
75 | 75 |
capacity |
76 | 76 |
1 2 16 |
77 | 77 |
1 3 12 |
78 | 78 |
2 3 18 |
79 | 79 |
\endcode |
80 | 80 |
|
81 | 81 |
If there is no map in the \c \@arcs section at all, then it must be |
82 | 82 |
indicated by a sole '-' sign in the first line. |
83 | 83 |
|
84 | 84 |
\code |
85 | 85 |
@arcs |
86 | 86 |
- |
87 | 87 |
1 2 |
88 | 88 |
1 3 |
89 | 89 |
2 3 |
90 | 90 |
\endcode |
91 | 91 |
|
92 | 92 |
The \c \@edges is just a synonym of \c \@arcs. The \@arcs section can |
93 | 93 |
also store the edge set of an undirected graph. In such case there is |
94 | 94 |
a conventional method for store arc maps in the file, if two columns |
95 | 95 |
have the same caption with \c '+' and \c '-' prefix, then these columns |
96 | 96 |
can be regarded as the values of an arc map. |
97 | 97 |
|
98 | 98 |
The \c \@attributes section contains key-value pairs, each line |
99 | 99 |
consists of two tokens, an attribute name, and then an attribute |
100 | 100 |
value. The value of the attribute could be also a label value of a |
101 | 101 |
node or an edge, or even an edge label prefixed with \c '+' or \c '-', |
102 | 102 |
which regards to the forward or backward directed arc of the |
103 | 103 |
corresponding edge. |
104 | 104 |
|
105 | 105 |
\code |
106 | 106 |
@attributes |
107 | 107 |
source 1 |
108 | 108 |
target 3 |
109 | 109 |
caption "LEMON test digraph" |
110 | 110 |
\endcode |
111 | 111 |
|
112 | 112 |
The \e LGF can contain extra sections, but there is no restriction on |
113 | 113 |
the format of such sections. |
114 | 114 |
|
115 | 115 |
*/ |
116 | 116 |
} |
117 | 117 |
|
118 | 118 |
// LocalWords: whitespace whitespaces |
1 | 1 |
/* -*- mode: C++; indent-tabs-mode: nil; -*- |
2 | 2 |
* |
3 | 3 |
* This file is a part of LEMON, a generic C++ optimization library. |
4 | 4 |
* |
5 |
* Copyright (C) 2003- |
|
5 |
* Copyright (C) 2003-2011 |
|
6 | 6 |
* Egervary Jeno Kombinatorikus Optimalizalasi Kutatocsoport |
7 | 7 |
* (Egervary Research Group on Combinatorial Optimization, EGRES). |
8 | 8 |
* |
9 | 9 |
* Permission to use, modify and distribute this software is granted |
10 | 10 |
* provided that this copyright notice appears in all copies. For |
11 | 11 |
* precise terms see the accompanying LICENSE file. |
12 | 12 |
* |
13 | 13 |
* This software is provided "AS IS" with no warranty of any kind, |
14 | 14 |
* express or implied, and with no claim as to its suitability for any |
15 | 15 |
* purpose. |
16 | 16 |
* |
17 | 17 |
*/ |
18 | 18 |
|
19 | 19 |
#ifndef LEMON_BITS_GRAPH_ADAPTOR_EXTENDER_H |
20 | 20 |
#define LEMON_BITS_GRAPH_ADAPTOR_EXTENDER_H |
21 | 21 |
|
22 | 22 |
#include <lemon/core.h> |
23 | 23 |
#include <lemon/error.h> |
24 | 24 |
|
25 | 25 |
namespace lemon { |
26 | 26 |
|
27 | 27 |
template <typename _Digraph> |
28 | 28 |
class DigraphAdaptorExtender : public _Digraph { |
29 | 29 |
typedef _Digraph Parent; |
30 | 30 |
|
31 | 31 |
public: |
32 | 32 |
|
33 | 33 |
typedef _Digraph Digraph; |
34 | 34 |
typedef DigraphAdaptorExtender Adaptor; |
35 | 35 |
|
36 | 36 |
// Base extensions |
37 | 37 |
|
38 | 38 |
typedef typename Parent::Node Node; |
39 | 39 |
typedef typename Parent::Arc Arc; |
40 | 40 |
|
41 | 41 |
int maxId(Node) const { |
42 | 42 |
return Parent::maxNodeId(); |
43 | 43 |
} |
44 | 44 |
|
45 | 45 |
int maxId(Arc) const { |
46 | 46 |
return Parent::maxArcId(); |
47 | 47 |
} |
48 | 48 |
|
49 | 49 |
Node fromId(int id, Node) const { |
50 | 50 |
return Parent::nodeFromId(id); |
51 | 51 |
} |
52 | 52 |
|
53 | 53 |
Arc fromId(int id, Arc) const { |
54 | 54 |
return Parent::arcFromId(id); |
55 | 55 |
} |
56 | 56 |
|
57 | 57 |
Node oppositeNode(const Node &n, const Arc &e) const { |
58 | 58 |
if (n == Parent::source(e)) |
59 | 59 |
return Parent::target(e); |
60 | 60 |
else if(n==Parent::target(e)) |
61 | 61 |
return Parent::source(e); |
62 | 62 |
else |
63 | 63 |
return INVALID; |
64 | 64 |
} |
65 | 65 |
|
66 | 66 |
class NodeIt : public Node { |
67 | 67 |
const Adaptor* _adaptor; |
68 | 68 |
public: |
69 | 69 |
|
70 | 70 |
NodeIt() {} |
71 | 71 |
|
72 | 72 |
NodeIt(Invalid i) : Node(i) { } |
73 | 73 |
|
74 | 74 |
explicit NodeIt(const Adaptor& adaptor) : _adaptor(&adaptor) { |
75 | 75 |
_adaptor->first(static_cast<Node&>(*this)); |
76 | 76 |
} |
77 | 77 |
|
78 | 78 |
NodeIt(const Adaptor& adaptor, const Node& node) |
79 | 79 |
: Node(node), _adaptor(&adaptor) {} |
80 | 80 |
|
81 | 81 |
NodeIt& operator++() { |
82 | 82 |
_adaptor->next(*this); |
83 | 83 |
return *this; |
84 | 84 |
} |
85 | 85 |
|
86 | 86 |
}; |
87 | 87 |
|
88 | 88 |
|
89 | 89 |
class ArcIt : public Arc { |
90 | 90 |
const Adaptor* _adaptor; |
91 | 91 |
public: |
92 | 92 |
|
93 | 93 |
ArcIt() { } |
94 | 94 |
|
95 | 95 |
ArcIt(Invalid i) : Arc(i) { } |
96 | 96 |
|
97 | 97 |
explicit ArcIt(const Adaptor& adaptor) : _adaptor(&adaptor) { |
98 | 98 |
_adaptor->first(static_cast<Arc&>(*this)); |
99 | 99 |
} |
100 | 100 |
|
101 | 101 |
ArcIt(const Adaptor& adaptor, const Arc& e) : |
102 | 102 |
Arc(e), _adaptor(&adaptor) { } |
103 | 103 |
|
104 | 104 |
ArcIt& operator++() { |
105 | 105 |
_adaptor->next(*this); |
106 | 106 |
return *this; |
107 | 107 |
} |
108 | 108 |
|
109 | 109 |
}; |
110 | 110 |
|
111 | 111 |
|
112 | 112 |
class OutArcIt : public Arc { |
113 | 113 |
const Adaptor* _adaptor; |
114 | 114 |
public: |
115 | 115 |
|
116 | 116 |
OutArcIt() { } |
117 | 117 |
|
118 | 118 |
OutArcIt(Invalid i) : Arc(i) { } |
119 | 119 |
|
120 | 120 |
OutArcIt(const Adaptor& adaptor, const Node& node) |
121 | 121 |
: _adaptor(&adaptor) { |
122 | 122 |
_adaptor->firstOut(*this, node); |
123 | 123 |
} |
124 | 124 |
|
125 | 125 |
OutArcIt(const Adaptor& adaptor, const Arc& arc) |
126 | 126 |
: Arc(arc), _adaptor(&adaptor) {} |
127 | 127 |
|
128 | 128 |
OutArcIt& operator++() { |
129 | 129 |
_adaptor->nextOut(*this); |
130 | 130 |
return *this; |
131 | 131 |
} |
132 | 132 |
|
133 | 133 |
}; |
134 | 134 |
|
135 | 135 |
|
136 | 136 |
class InArcIt : public Arc { |
137 | 137 |
const Adaptor* _adaptor; |
138 | 138 |
public: |
139 | 139 |
|
140 | 140 |
InArcIt() { } |
141 | 141 |
|
142 | 142 |
InArcIt(Invalid i) : Arc(i) { } |
143 | 143 |
|
144 | 144 |
InArcIt(const Adaptor& adaptor, const Node& node) |
145 | 145 |
: _adaptor(&adaptor) { |
146 | 146 |
_adaptor->firstIn(*this, node); |
147 | 147 |
} |
148 | 148 |
|
149 | 149 |
InArcIt(const Adaptor& adaptor, const Arc& arc) : |
150 | 150 |
Arc(arc), _adaptor(&adaptor) {} |
151 | 151 |
|
152 | 152 |
InArcIt& operator++() { |
153 | 153 |
_adaptor->nextIn(*this); |
154 | 154 |
return *this; |
155 | 155 |
} |
156 | 156 |
|
157 | 157 |
}; |
158 | 158 |
|
159 | 159 |
Node baseNode(const OutArcIt &e) const { |
160 | 160 |
return Parent::source(e); |
161 | 161 |
} |
162 | 162 |
Node runningNode(const OutArcIt &e) const { |
163 | 163 |
return Parent::target(e); |
164 | 164 |
} |
165 | 165 |
|
166 | 166 |
Node baseNode(const InArcIt &e) const { |
167 | 167 |
return Parent::target(e); |
168 | 168 |
} |
169 | 169 |
Node runningNode(const InArcIt &e) const { |
170 | 170 |
return Parent::source(e); |
171 | 171 |
} |
172 | 172 |
|
173 | 173 |
}; |
174 | 174 |
|
175 | 175 |
template <typename _Graph> |
176 | 176 |
class GraphAdaptorExtender : public _Graph { |
177 | 177 |
typedef _Graph Parent; |
178 | 178 |
|
179 | 179 |
public: |
180 | 180 |
|
181 | 181 |
typedef _Graph Graph; |
182 | 182 |
typedef GraphAdaptorExtender Adaptor; |
183 | 183 |
|
184 | 184 |
typedef True UndirectedTag; |
185 | 185 |
|
186 | 186 |
typedef typename Parent::Node Node; |
187 | 187 |
typedef typename Parent::Arc Arc; |
188 | 188 |
typedef typename Parent::Edge Edge; |
189 | 189 |
|
190 | 190 |
// Graph extension |
191 | 191 |
|
192 | 192 |
int maxId(Node) const { |
193 | 193 |
return Parent::maxNodeId(); |
194 | 194 |
} |
195 | 195 |
|
196 | 196 |
int maxId(Arc) const { |
197 | 197 |
return Parent::maxArcId(); |
198 | 198 |
} |
199 | 199 |
|
200 | 200 |
int maxId(Edge) const { |
201 | 201 |
return Parent::maxEdgeId(); |
202 | 202 |
} |
203 | 203 |
|
204 | 204 |
Node fromId(int id, Node) const { |
205 | 205 |
return Parent::nodeFromId(id); |
206 | 206 |
} |
207 | 207 |
|
208 | 208 |
Arc fromId(int id, Arc) const { |
209 | 209 |
return Parent::arcFromId(id); |
210 | 210 |
} |
211 | 211 |
|
212 | 212 |
Edge fromId(int id, Edge) const { |
213 | 213 |
return Parent::edgeFromId(id); |
214 | 214 |
} |
215 | 215 |
|
216 | 216 |
Node oppositeNode(const Node &n, const Edge &e) const { |
217 | 217 |
if( n == Parent::u(e)) |
218 | 218 |
return Parent::v(e); |
219 | 219 |
else if( n == Parent::v(e)) |
220 | 220 |
return Parent::u(e); |
221 | 221 |
else |
222 | 222 |
return INVALID; |
223 | 223 |
} |
224 | 224 |
|
225 | 225 |
Arc oppositeArc(const Arc &a) const { |
226 | 226 |
return Parent::direct(a, !Parent::direction(a)); |
227 | 227 |
} |
228 | 228 |
|
229 | 229 |
using Parent::direct; |
230 | 230 |
Arc direct(const Edge &e, const Node &s) const { |
231 | 231 |
return Parent::direct(e, Parent::u(e) == s); |
232 | 232 |
} |
233 | 233 |
|
234 | 234 |
|
235 | 235 |
class NodeIt : public Node { |
236 | 236 |
const Adaptor* _adaptor; |
237 | 237 |
public: |
238 | 238 |
|
239 | 239 |
NodeIt() {} |
240 | 240 |
|
241 | 241 |
NodeIt(Invalid i) : Node(i) { } |
242 | 242 |
|
243 | 243 |
explicit NodeIt(const Adaptor& adaptor) : _adaptor(&adaptor) { |
244 | 244 |
_adaptor->first(static_cast<Node&>(*this)); |
245 | 245 |
} |
246 | 246 |
|
247 | 247 |
NodeIt(const Adaptor& adaptor, const Node& node) |
248 | 248 |
: Node(node), _adaptor(&adaptor) {} |
249 | 249 |
|
250 | 250 |
NodeIt& operator++() { |
251 | 251 |
_adaptor->next(*this); |
252 | 252 |
return *this; |
253 | 253 |
} |
254 | 254 |
|
255 | 255 |
}; |
256 | 256 |
|
257 | 257 |
|
258 | 258 |
class ArcIt : public Arc { |
259 | 259 |
const Adaptor* _adaptor; |
260 | 260 |
public: |
261 | 261 |
|
262 | 262 |
ArcIt() { } |
263 | 263 |
|
264 | 264 |
ArcIt(Invalid i) : Arc(i) { } |
265 | 265 |
|
266 | 266 |
explicit ArcIt(const Adaptor& adaptor) : _adaptor(&adaptor) { |
267 | 267 |
_adaptor->first(static_cast<Arc&>(*this)); |
268 | 268 |
} |
269 | 269 |
|
270 | 270 |
ArcIt(const Adaptor& adaptor, const Arc& e) : |
271 | 271 |
Arc(e), _adaptor(&adaptor) { } |
272 | 272 |
|
273 | 273 |
ArcIt& operator++() { |
274 | 274 |
_adaptor->next(*this); |
275 | 275 |
return *this; |
276 | 276 |
} |
277 | 277 |
|
278 | 278 |
}; |
279 | 279 |
|
280 | 280 |
|
281 | 281 |
class OutArcIt : public Arc { |
282 | 282 |
const Adaptor* _adaptor; |
283 | 283 |
public: |
284 | 284 |
|
285 | 285 |
OutArcIt() { } |
286 | 286 |
|
287 | 287 |
OutArcIt(Invalid i) : Arc(i) { } |
288 | 288 |
|
289 | 289 |
OutArcIt(const Adaptor& adaptor, const Node& node) |
290 | 290 |
: _adaptor(&adaptor) { |
291 | 291 |
_adaptor->firstOut(*this, node); |
292 | 292 |
} |
293 | 293 |
|
294 | 294 |
OutArcIt(const Adaptor& adaptor, const Arc& arc) |
295 | 295 |
: Arc(arc), _adaptor(&adaptor) {} |
296 | 296 |
|
297 | 297 |
OutArcIt& operator++() { |
298 | 298 |
_adaptor->nextOut(*this); |
299 | 299 |
return *this; |
300 | 300 |
} |
301 | 301 |
|
302 | 302 |
}; |
303 | 303 |
|
304 | 304 |
|
305 | 305 |
class InArcIt : public Arc { |
306 | 306 |
const Adaptor* _adaptor; |
307 | 307 |
public: |
308 | 308 |
|
309 | 309 |
InArcIt() { } |
310 | 310 |
|
311 | 311 |
InArcIt(Invalid i) : Arc(i) { } |
312 | 312 |
|
313 | 313 |
InArcIt(const Adaptor& adaptor, const Node& node) |
314 | 314 |
: _adaptor(&adaptor) { |
315 | 315 |
_adaptor->firstIn(*this, node); |
316 | 316 |
} |
317 | 317 |
|
318 | 318 |
InArcIt(const Adaptor& adaptor, const Arc& arc) : |
319 | 319 |
Arc(arc), _adaptor(&adaptor) {} |
320 | 320 |
|
321 | 321 |
InArcIt& operator++() { |
322 | 322 |
_adaptor->nextIn(*this); |
323 | 323 |
return *this; |
324 | 324 |
} |
325 | 325 |
|
326 | 326 |
}; |
327 | 327 |
|
328 | 328 |
class EdgeIt : public Parent::Edge { |
329 | 329 |
const Adaptor* _adaptor; |
330 | 330 |
public: |
331 | 331 |
|
332 | 332 |
EdgeIt() { } |
333 | 333 |
|
334 | 334 |
EdgeIt(Invalid i) : Edge(i) { } |
335 | 335 |
|
336 | 336 |
explicit EdgeIt(const Adaptor& adaptor) : _adaptor(&adaptor) { |
337 | 337 |
_adaptor->first(static_cast<Edge&>(*this)); |
338 | 338 |
} |
339 | 339 |
|
340 | 340 |
EdgeIt(const Adaptor& adaptor, const Edge& e) : |
341 | 341 |
Edge(e), _adaptor(&adaptor) { } |
342 | 342 |
|
343 | 343 |
EdgeIt& operator++() { |
344 | 344 |
_adaptor->next(*this); |
345 | 345 |
return *this; |
346 | 346 |
} |
347 | 347 |
|
348 | 348 |
}; |
349 | 349 |
|
350 | 350 |
class IncEdgeIt : public Edge { |
351 | 351 |
friend class GraphAdaptorExtender; |
352 | 352 |
const Adaptor* _adaptor; |
353 | 353 |
bool direction; |
354 | 354 |
public: |
355 | 355 |
|
356 | 356 |
IncEdgeIt() { } |
357 | 357 |
|
358 | 358 |
IncEdgeIt(Invalid i) : Edge(i), direction(false) { } |
359 | 359 |
|
360 | 360 |
IncEdgeIt(const Adaptor& adaptor, const Node &n) : _adaptor(&adaptor) { |
361 | 361 |
_adaptor->firstInc(static_cast<Edge&>(*this), direction, n); |
362 | 362 |
} |
363 | 363 |
|
364 | 364 |
IncEdgeIt(const Adaptor& adaptor, const Edge &e, const Node &n) |
365 | 365 |
: _adaptor(&adaptor), Edge(e) { |
366 | 366 |
direction = (_adaptor->u(e) == n); |
367 | 367 |
} |
368 | 368 |
|
369 | 369 |
IncEdgeIt& operator++() { |
370 | 370 |
_adaptor->nextInc(*this, direction); |
371 | 371 |
return *this; |
372 | 372 |
} |
373 | 373 |
}; |
374 | 374 |
|
375 | 375 |
Node baseNode(const OutArcIt &a) const { |
376 | 376 |
return Parent::source(a); |
377 | 377 |
} |
378 | 378 |
Node runningNode(const OutArcIt &a) const { |
379 | 379 |
return Parent::target(a); |
380 | 380 |
} |
381 | 381 |
|
382 | 382 |
Node baseNode(const InArcIt &a) const { |
383 | 383 |
return Parent::target(a); |
384 | 384 |
} |
385 | 385 |
Node runningNode(const InArcIt &a) const { |
386 | 386 |
return Parent::source(a); |
387 | 387 |
} |
388 | 388 |
|
389 | 389 |
Node baseNode(const IncEdgeIt &e) const { |
390 | 390 |
return e.direction ? Parent::u(e) : Parent::v(e); |
391 | 391 |
} |
392 | 392 |
Node runningNode(const IncEdgeIt &e) const { |
393 | 393 |
return e.direction ? Parent::v(e) : Parent::u(e); |
394 | 394 |
} |
395 | 395 |
|
396 | 396 |
}; |
397 | 397 |
|
398 | 398 |
} |
399 | 399 |
|
400 | 400 |
|
401 | 401 |
#endif |
1 | 1 |
/* -*- mode: C++; indent-tabs-mode: nil; -*- |
2 | 2 |
* |
3 | 3 |
* This file is a part of LEMON, a generic C++ optimization library. |
4 | 4 |
* |
5 |
* Copyright (C) 2003- |
|
5 |
* Copyright (C) 2003-2011 |
|
6 | 6 |
* Egervary Jeno Kombinatorikus Optimalizalasi Kutatocsoport |
7 | 7 |
* (Egervary Research Group on Combinatorial Optimization, EGRES). |
8 | 8 |
* |
9 | 9 |
* Permission to use, modify and distribute this software is granted |
10 | 10 |
* provided that this copyright notice appears in all copies. For |
11 | 11 |
* precise terms see the accompanying LICENSE file. |
12 | 12 |
* |
13 | 13 |
* This software is provided "AS IS" with no warranty of any kind, |
14 | 14 |
* express or implied, and with no claim as to its suitability for any |
15 | 15 |
* purpose. |
16 | 16 |
* |
17 | 17 |
*/ |
18 | 18 |
|
19 | 19 |
#ifndef LEMON_BITS_PATH_DUMP_H |
20 | 20 |
#define LEMON_BITS_PATH_DUMP_H |
21 | 21 |
|
22 | 22 |
#include <lemon/core.h> |
23 | 23 |
#include <lemon/concept_check.h> |
24 | 24 |
|
25 | 25 |
namespace lemon { |
26 | 26 |
|
27 | 27 |
template <typename _Digraph, typename _PredMap> |
28 | 28 |
class PredMapPath { |
29 | 29 |
public: |
30 | 30 |
typedef True RevPathTag; |
31 | 31 |
|
32 | 32 |
typedef _Digraph Digraph; |
33 | 33 |
typedef typename Digraph::Arc Arc; |
34 | 34 |
typedef _PredMap PredMap; |
35 | 35 |
|
36 | 36 |
PredMapPath(const Digraph& _digraph, const PredMap& _predMap, |
37 | 37 |
typename Digraph::Node _target) |
38 | 38 |
: digraph(_digraph), predMap(_predMap), target(_target) {} |
39 | 39 |
|
40 | 40 |
int length() const { |
41 | 41 |
int len = 0; |
42 | 42 |
typename Digraph::Node node = target; |
43 | 43 |
typename Digraph::Arc arc; |
44 | 44 |
while ((arc = predMap[node]) != INVALID) { |
45 | 45 |
node = digraph.source(arc); |
46 | 46 |
++len; |
47 | 47 |
} |
48 | 48 |
return len; |
49 | 49 |
} |
50 | 50 |
|
51 | 51 |
bool empty() const { |
52 | 52 |
return predMap[target] == INVALID; |
53 | 53 |
} |
54 | 54 |
|
55 | 55 |
class RevArcIt { |
56 | 56 |
public: |
57 | 57 |
RevArcIt() {} |
58 | 58 |
RevArcIt(Invalid) : path(0), current(INVALID) {} |
59 | 59 |
RevArcIt(const PredMapPath& _path) |
60 | 60 |
: path(&_path), current(_path.target) { |
61 | 61 |
if (path->predMap[current] == INVALID) current = INVALID; |
62 | 62 |
} |
63 | 63 |
|
64 | 64 |
operator const typename Digraph::Arc() const { |
65 | 65 |
return path->predMap[current]; |
66 | 66 |
} |
67 | 67 |
|
68 | 68 |
RevArcIt& operator++() { |
69 | 69 |
current = path->digraph.source(path->predMap[current]); |
70 | 70 |
if (path->predMap[current] == INVALID) current = INVALID; |
71 | 71 |
return *this; |
72 | 72 |
} |
73 | 73 |
|
74 | 74 |
bool operator==(const RevArcIt& e) const { |
75 | 75 |
return current == e.current; |
76 | 76 |
} |
77 | 77 |
|
78 | 78 |
bool operator!=(const RevArcIt& e) const { |
79 | 79 |
return current != e.current; |
80 | 80 |
} |
81 | 81 |
|
82 | 82 |
bool operator<(const RevArcIt& e) const { |
83 | 83 |
return current < e.current; |
84 | 84 |
} |
85 | 85 |
|
86 | 86 |
private: |
87 | 87 |
const PredMapPath* path; |
88 | 88 |
typename Digraph::Node current; |
89 | 89 |
}; |
90 | 90 |
|
91 | 91 |
private: |
92 | 92 |
const Digraph& digraph; |
93 | 93 |
const PredMap& predMap; |
94 | 94 |
typename Digraph::Node target; |
95 | 95 |
}; |
96 | 96 |
|
97 | 97 |
|
98 | 98 |
template <typename _Digraph, typename _PredMatrixMap> |
99 | 99 |
class PredMatrixMapPath { |
100 | 100 |
public: |
101 | 101 |
typedef True RevPathTag; |
102 | 102 |
|
103 | 103 |
typedef _Digraph Digraph; |
104 | 104 |
typedef typename Digraph::Arc Arc; |
105 | 105 |
typedef _PredMatrixMap PredMatrixMap; |
106 | 106 |
|
107 | 107 |
PredMatrixMapPath(const Digraph& _digraph, |
108 | 108 |
const PredMatrixMap& _predMatrixMap, |
109 | 109 |
typename Digraph::Node _source, |
110 | 110 |
typename Digraph::Node _target) |
111 | 111 |
: digraph(_digraph), predMatrixMap(_predMatrixMap), |
112 | 112 |
source(_source), target(_target) {} |
113 | 113 |
|
114 | 114 |
int length() const { |
115 | 115 |
int len = 0; |
116 | 116 |
typename Digraph::Node node = target; |
117 | 117 |
typename Digraph::Arc arc; |
118 | 118 |
while ((arc = predMatrixMap(source, node)) != INVALID) { |
119 | 119 |
node = digraph.source(arc); |
120 | 120 |
++len; |
121 | 121 |
} |
122 | 122 |
return len; |
123 | 123 |
} |
124 | 124 |
|
125 | 125 |
bool empty() const { |
126 | 126 |
return predMatrixMap(source, target) == INVALID; |
127 | 127 |
} |
128 | 128 |
|
129 | 129 |
class RevArcIt { |
130 | 130 |
public: |
131 | 131 |
RevArcIt() {} |
132 | 132 |
RevArcIt(Invalid) : path(0), current(INVALID) {} |
133 | 133 |
RevArcIt(const PredMatrixMapPath& _path) |
134 | 134 |
: path(&_path), current(_path.target) { |
135 | 135 |
if (path->predMatrixMap(path->source, current) == INVALID) |
136 | 136 |
current = INVALID; |
137 | 137 |
} |
138 | 138 |
|
139 | 139 |
operator const typename Digraph::Arc() const { |
140 | 140 |
return path->predMatrixMap(path->source, current); |
141 | 141 |
} |
142 | 142 |
|
143 | 143 |
RevArcIt& operator++() { |
144 | 144 |
current = |
145 | 145 |
path->digraph.source(path->predMatrixMap(path->source, current)); |
146 | 146 |
if (path->predMatrixMap(path->source, current) == INVALID) |
147 | 147 |
current = INVALID; |
148 | 148 |
return *this; |
149 | 149 |
} |
150 | 150 |
|
151 | 151 |
bool operator==(const RevArcIt& e) const { |
152 | 152 |
return current == e.current; |
153 | 153 |
} |
154 | 154 |
|
155 | 155 |
bool operator!=(const RevArcIt& e) const { |
156 | 156 |
return current != e.current; |
157 | 157 |
} |
158 | 158 |
|
159 | 159 |
bool operator<(const RevArcIt& e) const { |
160 | 160 |
return current < e.current; |
161 | 161 |
} |
162 | 162 |
|
163 | 163 |
private: |
164 | 164 |
const PredMatrixMapPath* path; |
165 | 165 |
typename Digraph::Node current; |
166 | 166 |
}; |
167 | 167 |
|
168 | 168 |
private: |
169 | 169 |
const Digraph& digraph; |
170 | 170 |
const PredMatrixMap& predMatrixMap; |
171 | 171 |
typename Digraph::Node source; |
172 | 172 |
typename Digraph::Node target; |
173 | 173 |
}; |
174 | 174 |
|
175 | 175 |
} |
176 | 176 |
|
177 | 177 |
#endif |
1 | 1 |
/* -*- mode: C++; indent-tabs-mode: nil; -*- |
2 | 2 |
* |
3 | 3 |
* This file is a part of LEMON, a generic C++ optimization library. |
4 | 4 |
* |
5 |
* Copyright (C) 2003- |
|
5 |
* Copyright (C) 2003-2011 |
|
6 | 6 |
* Egervary Jeno Kombinatorikus Optimalizalasi Kutatocsoport |
7 | 7 |
* (Egervary Research Group on Combinatorial Optimization, EGRES). |
8 | 8 |
* |
9 | 9 |
* Permission to use, modify and distribute this software is granted |
10 | 10 |
* provided that this copyright notice appears in all copies. For |
11 | 11 |
* precise terms see the accompanying LICENSE file. |
12 | 12 |
* |
13 | 13 |
* This software is provided "AS IS" with no warranty of any kind, |
14 | 14 |
* express or implied, and with no claim as to its suitability for any |
15 | 15 |
* purpose. |
16 | 16 |
* |
17 | 17 |
*/ |
18 | 18 |
|
19 | 19 |
///\file |
20 | 20 |
///\brief Some basic non-inline functions and static global data. |
21 | 21 |
|
22 | 22 |
#include<lemon/bits/windows.h> |
23 | 23 |
|
24 | 24 |
#ifdef WIN32 |
25 | 25 |
#ifndef WIN32_LEAN_AND_MEAN |
26 | 26 |
#define WIN32_LEAN_AND_MEAN |
27 | 27 |
#endif |
28 | 28 |
#ifndef NOMINMAX |
29 | 29 |
#define NOMINMAX |
30 | 30 |
#endif |
31 | 31 |
#ifdef UNICODE |
32 | 32 |
#undef UNICODE |
33 | 33 |
#endif |
34 | 34 |
#include <windows.h> |
35 | 35 |
#ifdef LOCALE_INVARIANT |
36 | 36 |
#define MY_LOCALE LOCALE_INVARIANT |
37 | 37 |
#else |
38 | 38 |
#define MY_LOCALE LOCALE_NEUTRAL |
39 | 39 |
#endif |
40 | 40 |
#else |
41 | 41 |
#include <unistd.h> |
42 | 42 |
#include <ctime> |
43 | 43 |
#ifndef WIN32 |
44 | 44 |
#include <sys/times.h> |
45 | 45 |
#endif |
46 | 46 |
#include <sys/time.h> |
47 | 47 |
#endif |
48 | 48 |
|
49 | 49 |
#include <cmath> |
50 | 50 |
#include <sstream> |
51 | 51 |
|
52 | 52 |
namespace lemon { |
53 | 53 |
namespace bits { |
54 | 54 |
void getWinProcTimes(double &rtime, |
55 | 55 |
double &utime, double &stime, |
56 | 56 |
double &cutime, double &cstime) |
57 | 57 |
{ |
58 | 58 |
#ifdef WIN32 |
59 | 59 |
static const double ch = 4294967296.0e-7; |
60 | 60 |
static const double cl = 1.0e-7; |
61 | 61 |
|
62 | 62 |
FILETIME system; |
63 | 63 |
GetSystemTimeAsFileTime(&system); |
64 | 64 |
rtime = ch * system.dwHighDateTime + cl * system.dwLowDateTime; |
65 | 65 |
|
66 | 66 |
FILETIME create, exit, kernel, user; |
67 | 67 |
if (GetProcessTimes(GetCurrentProcess(),&create, &exit, &kernel, &user)) { |
68 | 68 |
utime = ch * user.dwHighDateTime + cl * user.dwLowDateTime; |
69 | 69 |
stime = ch * kernel.dwHighDateTime + cl * kernel.dwLowDateTime; |
70 | 70 |
cutime = 0; |
71 | 71 |
cstime = 0; |
72 | 72 |
} else { |
73 | 73 |
rtime = 0; |
74 | 74 |
utime = 0; |
75 | 75 |
stime = 0; |
76 | 76 |
cutime = 0; |
77 | 77 |
cstime = 0; |
78 | 78 |
} |
79 | 79 |
#else |
80 | 80 |
timeval tv; |
81 | 81 |
gettimeofday(&tv, 0); |
82 | 82 |
rtime=tv.tv_sec+double(tv.tv_usec)/1e6; |
83 | 83 |
|
84 | 84 |
tms ts; |
85 | 85 |
double tck=sysconf(_SC_CLK_TCK); |
86 | 86 |
times(&ts); |
87 | 87 |
utime=ts.tms_utime/tck; |
88 | 88 |
stime=ts.tms_stime/tck; |
89 | 89 |
cutime=ts.tms_cutime/tck; |
90 | 90 |
cstime=ts.tms_cstime/tck; |
91 | 91 |
#endif |
92 | 92 |
} |
93 | 93 |
|
94 | 94 |
std::string getWinFormattedDate() |
95 | 95 |
{ |
96 | 96 |
std::ostringstream os; |
97 | 97 |
#ifdef WIN32 |
98 | 98 |
SYSTEMTIME time; |
99 | 99 |
GetSystemTime(&time); |
100 | 100 |
char buf1[11], buf2[9], buf3[5]; |
101 | 101 |
if (GetDateFormat(MY_LOCALE, 0, &time, |
102 | 102 |
("ddd MMM dd"), buf1, 11) && |
103 | 103 |
GetTimeFormat(MY_LOCALE, 0, &time, |
104 | 104 |
("HH':'mm':'ss"), buf2, 9) && |
105 | 105 |
GetDateFormat(MY_LOCALE, 0, &time, |
106 | 106 |
("yyyy"), buf3, 5)) { |
107 | 107 |
os << buf1 << ' ' << buf2 << ' ' << buf3; |
108 | 108 |
} |
109 | 109 |
else os << "unknown"; |
110 | 110 |
#else |
111 | 111 |
timeval tv; |
112 | 112 |
gettimeofday(&tv, 0); |
113 | 113 |
|
114 | 114 |
char cbuf[26]; |
115 | 115 |
ctime_r(&tv.tv_sec,cbuf); |
116 | 116 |
os << cbuf; |
117 | 117 |
#endif |
118 | 118 |
return os.str(); |
119 | 119 |
} |
120 | 120 |
|
121 | 121 |
int getWinRndSeed() |
122 | 122 |
{ |
123 | 123 |
#ifdef WIN32 |
124 | 124 |
FILETIME time; |
125 | 125 |
GetSystemTimeAsFileTime(&time); |
126 | 126 |
return GetCurrentProcessId() + time.dwHighDateTime + time.dwLowDateTime; |
127 | 127 |
#else |
128 | 128 |
timeval tv; |
129 | 129 |
gettimeofday(&tv, 0); |
130 | 130 |
return getpid() + tv.tv_sec + tv.tv_usec; |
131 | 131 |
#endif |
132 | 132 |
} |
133 | 133 |
} |
134 | 134 |
} |
1 | 1 |
/* -*- mode: C++; indent-tabs-mode: nil; -*- |
2 | 2 |
* |
3 | 3 |
* This file is a part of LEMON, a generic C++ optimization library. |
4 | 4 |
* |
5 |
* Copyright (C) 2003- |
|
5 |
* Copyright (C) 2003-2011 |
|
6 | 6 |
* Egervary Jeno Kombinatorikus Optimalizalasi Kutatocsoport |
7 | 7 |
* (Egervary Research Group on Combinatorial Optimization, EGRES). |
8 | 8 |
* |
9 | 9 |
* Permission to use, modify and distribute this software is granted |
10 | 10 |
* provided that this copyright notice appears in all copies. For |
11 | 11 |
* precise terms see the accompanying LICENSE file. |
12 | 12 |
* |
13 | 13 |
* This software is provided "AS IS" with no warranty of any kind, |
14 | 14 |
* express or implied, and with no claim as to its suitability for any |
15 | 15 |
* purpose. |
16 | 16 |
* |
17 | 17 |
*/ |
18 | 18 |
|
19 | 19 |
#ifndef LEMON_COST_SCALING_H |
20 | 20 |
#define LEMON_COST_SCALING_H |
21 | 21 |
|
22 | 22 |
/// \ingroup min_cost_flow_algs |
23 | 23 |
/// \file |
24 | 24 |
/// \brief Cost scaling algorithm for finding a minimum cost flow. |
25 | 25 |
|
26 | 26 |
#include <vector> |
27 | 27 |
#include <deque> |
28 | 28 |
#include <limits> |
29 | 29 |
|
30 | 30 |
#include <lemon/core.h> |
31 | 31 |
#include <lemon/maps.h> |
32 | 32 |
#include <lemon/math.h> |
33 | 33 |
#include <lemon/static_graph.h> |
34 | 34 |
#include <lemon/circulation.h> |
35 | 35 |
#include <lemon/bellman_ford.h> |
36 | 36 |
|
37 | 37 |
namespace lemon { |
38 | 38 |
|
39 | 39 |
/// \brief Default traits class of CostScaling algorithm. |
40 | 40 |
/// |
41 | 41 |
/// Default traits class of CostScaling algorithm. |
42 | 42 |
/// \tparam GR Digraph type. |
43 | 43 |
/// \tparam V The number type used for flow amounts, capacity bounds |
44 | 44 |
/// and supply values. By default it is \c int. |
45 | 45 |
/// \tparam C The number type used for costs and potentials. |
46 | 46 |
/// By default it is the same as \c V. |
47 | 47 |
#ifdef DOXYGEN |
48 | 48 |
template <typename GR, typename V = int, typename C = V> |
49 | 49 |
#else |
50 | 50 |
template < typename GR, typename V = int, typename C = V, |
51 | 51 |
bool integer = std::numeric_limits<C>::is_integer > |
52 | 52 |
#endif |
53 | 53 |
struct CostScalingDefaultTraits |
54 | 54 |
{ |
55 | 55 |
/// The type of the digraph |
56 | 56 |
typedef GR Digraph; |
57 | 57 |
/// The type of the flow amounts, capacity bounds and supply values |
58 | 58 |
typedef V Value; |
59 | 59 |
/// The type of the arc costs |
60 | 60 |
typedef C Cost; |
61 | 61 |
|
62 | 62 |
/// \brief The large cost type used for internal computations |
63 | 63 |
/// |
64 | 64 |
/// The large cost type used for internal computations. |
65 | 65 |
/// It is \c long \c long if the \c Cost type is integer, |
66 | 66 |
/// otherwise it is \c double. |
67 | 67 |
/// \c Cost must be convertible to \c LargeCost. |
68 | 68 |
typedef double LargeCost; |
69 | 69 |
}; |
70 | 70 |
|
71 | 71 |
// Default traits class for integer cost types |
72 | 72 |
template <typename GR, typename V, typename C> |
73 | 73 |
struct CostScalingDefaultTraits<GR, V, C, true> |
74 | 74 |
{ |
75 | 75 |
typedef GR Digraph; |
76 | 76 |
typedef V Value; |
77 | 77 |
typedef C Cost; |
78 | 78 |
#ifdef LEMON_HAVE_LONG_LONG |
79 | 79 |
typedef long long LargeCost; |
80 | 80 |
#else |
81 | 81 |
typedef long LargeCost; |
82 | 82 |
#endif |
83 | 83 |
}; |
84 | 84 |
|
85 | 85 |
|
86 | 86 |
/// \addtogroup min_cost_flow_algs |
87 | 87 |
/// @{ |
88 | 88 |
|
89 | 89 |
/// \brief Implementation of the Cost Scaling algorithm for |
90 | 90 |
/// finding a \ref min_cost_flow "minimum cost flow". |
91 | 91 |
/// |
92 | 92 |
/// \ref CostScaling implements a cost scaling algorithm that performs |
93 | 93 |
/// push/augment and relabel operations for finding a \ref min_cost_flow |
94 | 94 |
/// "minimum cost flow" \ref amo93networkflows, \ref goldberg90approximation, |
95 | 95 |
/// \ref goldberg97efficient, \ref bunnagel98efficient. |
96 | 96 |
/// It is a highly efficient primal-dual solution method, which |
97 | 97 |
/// can be viewed as the generalization of the \ref Preflow |
98 | 98 |
/// "preflow push-relabel" algorithm for the maximum flow problem. |
99 | 99 |
/// |
100 | 100 |
/// Most of the parameters of the problem (except for the digraph) |
101 | 101 |
/// can be given using separate functions, and the algorithm can be |
102 | 102 |
/// executed using the \ref run() function. If some parameters are not |
103 | 103 |
/// specified, then default values will be used. |
104 | 104 |
/// |
105 | 105 |
/// \tparam GR The digraph type the algorithm runs on. |
106 | 106 |
/// \tparam V The number type used for flow amounts, capacity bounds |
107 | 107 |
/// and supply values in the algorithm. By default, it is \c int. |
108 | 108 |
/// \tparam C The number type used for costs and potentials in the |
109 | 109 |
/// algorithm. By default, it is the same as \c V. |
110 | 110 |
/// \tparam TR The traits class that defines various types used by the |
111 | 111 |
/// algorithm. By default, it is \ref CostScalingDefaultTraits |
112 | 112 |
/// "CostScalingDefaultTraits<GR, V, C>". |
113 | 113 |
/// In most cases, this parameter should not be set directly, |
114 | 114 |
/// consider to use the named template parameters instead. |
115 | 115 |
/// |
116 | 116 |
/// \warning Both number types must be signed and all input data must |
117 | 117 |
/// be integer. |
118 | 118 |
/// \warning This algorithm does not support negative costs for such |
119 | 119 |
/// arcs that have infinite upper bound. |
120 | 120 |
/// |
121 | 121 |
/// \note %CostScaling provides three different internal methods, |
122 | 122 |
/// from which the most efficient one is used by default. |
123 | 123 |
/// For more information, see \ref Method. |
124 | 124 |
#ifdef DOXYGEN |
125 | 125 |
template <typename GR, typename V, typename C, typename TR> |
126 | 126 |
#else |
127 | 127 |
template < typename GR, typename V = int, typename C = V, |
128 | 128 |
typename TR = CostScalingDefaultTraits<GR, V, C> > |
129 | 129 |
#endif |
130 | 130 |
class CostScaling |
131 | 131 |
{ |
132 | 132 |
public: |
133 | 133 |
|
134 | 134 |
/// The type of the digraph |
135 | 135 |
typedef typename TR::Digraph Digraph; |
136 | 136 |
/// The type of the flow amounts, capacity bounds and supply values |
137 | 137 |
typedef typename TR::Value Value; |
138 | 138 |
/// The type of the arc costs |
139 | 139 |
typedef typename TR::Cost Cost; |
140 | 140 |
|
141 | 141 |
/// \brief The large cost type |
142 | 142 |
/// |
143 | 143 |
/// The large cost type used for internal computations. |
144 | 144 |
/// By default, it is \c long \c long if the \c Cost type is integer, |
145 | 145 |
/// otherwise it is \c double. |
146 | 146 |
typedef typename TR::LargeCost LargeCost; |
147 | 147 |
|
148 | 148 |
/// The \ref CostScalingDefaultTraits "traits class" of the algorithm |
149 | 149 |
typedef TR Traits; |
150 | 150 |
|
151 | 151 |
public: |
152 | 152 |
|
153 | 153 |
/// \brief Problem type constants for the \c run() function. |
154 | 154 |
/// |
155 | 155 |
/// Enum type containing the problem type constants that can be |
156 | 156 |
/// returned by the \ref run() function of the algorithm. |
157 | 157 |
enum ProblemType { |
158 | 158 |
/// The problem has no feasible solution (flow). |
159 | 159 |
INFEASIBLE, |
160 | 160 |
/// The problem has optimal solution (i.e. it is feasible and |
161 | 161 |
/// bounded), and the algorithm has found optimal flow and node |
162 | 162 |
/// potentials (primal and dual solutions). |
163 | 163 |
OPTIMAL, |
164 | 164 |
/// The digraph contains an arc of negative cost and infinite |
165 | 165 |
/// upper bound. It means that the objective function is unbounded |
166 | 166 |
/// on that arc, however, note that it could actually be bounded |
167 | 167 |
/// over the feasible flows, but this algroithm cannot handle |
168 | 168 |
/// these cases. |
169 | 169 |
UNBOUNDED |
170 | 170 |
}; |
171 | 171 |
|
172 | 172 |
/// \brief Constants for selecting the internal method. |
173 | 173 |
/// |
174 | 174 |
/// Enum type containing constants for selecting the internal method |
175 | 175 |
/// for the \ref run() function. |
176 | 176 |
/// |
177 | 177 |
/// \ref CostScaling provides three internal methods that differ mainly |
178 | 178 |
/// in their base operations, which are used in conjunction with the |
179 | 179 |
/// relabel operation. |
180 | 180 |
/// By default, the so called \ref PARTIAL_AUGMENT |
181 | 181 |
/// "Partial Augment-Relabel" method is used, which proved to be |
182 | 182 |
/// the most efficient and the most robust on various test inputs. |
183 | 183 |
/// However, the other methods can be selected using the \ref run() |
184 | 184 |
/// function with the proper parameter. |
185 | 185 |
enum Method { |
186 | 186 |
/// Local push operations are used, i.e. flow is moved only on one |
187 | 187 |
/// admissible arc at once. |
188 | 188 |
PUSH, |
189 | 189 |
/// Augment operations are used, i.e. flow is moved on admissible |
190 | 190 |
/// paths from a node with excess to a node with deficit. |
191 | 191 |
AUGMENT, |
192 | 192 |
/// Partial augment operations are used, i.e. flow is moved on |
193 | 193 |
/// admissible paths started from a node with excess, but the |
194 | 194 |
/// lengths of these paths are limited. This method can be viewed |
195 | 195 |
/// as a combined version of the previous two operations. |
196 | 196 |
PARTIAL_AUGMENT |
197 | 197 |
}; |
198 | 198 |
|
199 | 199 |
private: |
200 | 200 |
|
201 | 201 |
TEMPLATE_DIGRAPH_TYPEDEFS(GR); |
202 | 202 |
|
203 | 203 |
typedef std::vector<int> IntVector; |
204 | 204 |
typedef std::vector<Value> ValueVector; |
205 | 205 |
typedef std::vector<Cost> CostVector; |
206 | 206 |
typedef std::vector<LargeCost> LargeCostVector; |
207 | 207 |
typedef std::vector<char> BoolVector; |
208 | 208 |
// Note: vector<char> is used instead of vector<bool> for efficiency reasons |
209 | 209 |
|
210 | 210 |
private: |
211 | 211 |
|
212 | 212 |
template <typename KT, typename VT> |
213 | 213 |
class StaticVectorMap { |
214 | 214 |
public: |
215 | 215 |
typedef KT Key; |
216 | 216 |
typedef VT Value; |
217 | 217 |
|
218 | 218 |
StaticVectorMap(std::vector<Value>& v) : _v(v) {} |
219 | 219 |
|
220 | 220 |
const Value& operator[](const Key& key) const { |
221 | 221 |
return _v[StaticDigraph::id(key)]; |
222 | 222 |
} |
223 | 223 |
|
224 | 224 |
Value& operator[](const Key& key) { |
225 | 225 |
return _v[StaticDigraph::id(key)]; |
226 | 226 |
} |
227 | 227 |
|
228 | 228 |
void set(const Key& key, const Value& val) { |
229 | 229 |
_v[StaticDigraph::id(key)] = val; |
230 | 230 |
} |
231 | 231 |
|
232 | 232 |
private: |
233 | 233 |
std::vector<Value>& _v; |
234 | 234 |
}; |
235 | 235 |
|
236 | 236 |
typedef StaticVectorMap<StaticDigraph::Node, LargeCost> LargeCostNodeMap; |
237 | 237 |
typedef StaticVectorMap<StaticDigraph::Arc, LargeCost> LargeCostArcMap; |
238 | 238 |
|
239 | 239 |
private: |
240 | 240 |
|
241 | 241 |
// Data related to the underlying digraph |
242 | 242 |
const GR &_graph; |
243 | 243 |
int _node_num; |
244 | 244 |
int _arc_num; |
245 | 245 |
int _res_node_num; |
246 | 246 |
int _res_arc_num; |
247 | 247 |
int _root; |
248 | 248 |
|
249 | 249 |
// Parameters of the problem |
250 | 250 |
bool _have_lower; |
251 | 251 |
Value _sum_supply; |
252 | 252 |
int _sup_node_num; |
253 | 253 |
|
254 | 254 |
// Data structures for storing the digraph |
255 | 255 |
IntNodeMap _node_id; |
256 | 256 |
IntArcMap _arc_idf; |
257 | 257 |
IntArcMap _arc_idb; |
258 | 258 |
IntVector _first_out; |
259 | 259 |
BoolVector _forward; |
260 | 260 |
IntVector _source; |
261 | 261 |
IntVector _target; |
262 | 262 |
IntVector _reverse; |
263 | 263 |
|
264 | 264 |
// Node and arc data |
265 | 265 |
ValueVector _lower; |
266 | 266 |
ValueVector _upper; |
267 | 267 |
CostVector _scost; |
268 | 268 |
ValueVector _supply; |
269 | 269 |
|
270 | 270 |
ValueVector _res_cap; |
271 | 271 |
LargeCostVector _cost; |
272 | 272 |
LargeCostVector _pi; |
273 | 273 |
ValueVector _excess; |
274 | 274 |
IntVector _next_out; |
275 | 275 |
std::deque<int> _active_nodes; |
276 | 276 |
|
277 | 277 |
// Data for scaling |
278 | 278 |
LargeCost _epsilon; |
279 | 279 |
int _alpha; |
280 | 280 |
|
281 | 281 |
IntVector _buckets; |
282 | 282 |
IntVector _bucket_next; |
283 | 283 |
IntVector _bucket_prev; |
284 | 284 |
IntVector _rank; |
285 | 285 |
int _max_rank; |
286 | 286 |
|
287 | 287 |
// Data for a StaticDigraph structure |
288 | 288 |
typedef std::pair<int, int> IntPair; |
289 | 289 |
StaticDigraph _sgr; |
290 | 290 |
std::vector<IntPair> _arc_vec; |
291 | 291 |
std::vector<LargeCost> _cost_vec; |
292 | 292 |
LargeCostArcMap _cost_map; |
293 | 293 |
LargeCostNodeMap _pi_map; |
294 | 294 |
|
295 | 295 |
public: |
296 | 296 |
|
297 | 297 |
/// \brief Constant for infinite upper bounds (capacities). |
298 | 298 |
/// |
299 | 299 |
/// Constant for infinite upper bounds (capacities). |
300 | 300 |
/// It is \c std::numeric_limits<Value>::infinity() if available, |
301 | 301 |
/// \c std::numeric_limits<Value>::max() otherwise. |
302 | 302 |
const Value INF; |
303 | 303 |
|
304 | 304 |
public: |
305 | 305 |
|
306 | 306 |
/// \name Named Template Parameters |
307 | 307 |
/// @{ |
308 | 308 |
|
309 | 309 |
template <typename T> |
310 | 310 |
struct SetLargeCostTraits : public Traits { |
311 | 311 |
typedef T LargeCost; |
312 | 312 |
}; |
313 | 313 |
|
314 | 314 |
/// \brief \ref named-templ-param "Named parameter" for setting |
315 | 315 |
/// \c LargeCost type. |
316 | 316 |
/// |
317 | 317 |
/// \ref named-templ-param "Named parameter" for setting \c LargeCost |
318 | 318 |
/// type, which is used for internal computations in the algorithm. |
319 | 319 |
/// \c Cost must be convertible to \c LargeCost. |
320 | 320 |
template <typename T> |
321 | 321 |
struct SetLargeCost |
322 | 322 |
: public CostScaling<GR, V, C, SetLargeCostTraits<T> > { |
323 | 323 |
typedef CostScaling<GR, V, C, SetLargeCostTraits<T> > Create; |
324 | 324 |
}; |
325 | 325 |
|
326 | 326 |
/// @} |
327 | 327 |
|
328 | 328 |
protected: |
329 | 329 |
|
330 | 330 |
CostScaling() {} |
331 | 331 |
|
332 | 332 |
public: |
333 | 333 |
|
334 | 334 |
/// \brief Constructor. |
335 | 335 |
/// |
336 | 336 |
/// The constructor of the class. |
337 | 337 |
/// |
338 | 338 |
/// \param graph The digraph the algorithm runs on. |
339 | 339 |
CostScaling(const GR& graph) : |
340 | 340 |
_graph(graph), _node_id(graph), _arc_idf(graph), _arc_idb(graph), |
341 | 341 |
_cost_map(_cost_vec), _pi_map(_pi), |
342 | 342 |
INF(std::numeric_limits<Value>::has_infinity ? |
343 | 343 |
std::numeric_limits<Value>::infinity() : |
344 | 344 |
std::numeric_limits<Value>::max()) |
345 | 345 |
{ |
346 | 346 |
// Check the number types |
347 | 347 |
LEMON_ASSERT(std::numeric_limits<Value>::is_signed, |
348 | 348 |
"The flow type of CostScaling must be signed"); |
349 | 349 |
LEMON_ASSERT(std::numeric_limits<Cost>::is_signed, |
350 | 350 |
"The cost type of CostScaling must be signed"); |
351 | 351 |
|
352 | 352 |
// Reset data structures |
353 | 353 |
reset(); |
354 | 354 |
} |
355 | 355 |
|
356 | 356 |
/// \name Parameters |
357 | 357 |
/// The parameters of the algorithm can be specified using these |
358 | 358 |
/// functions. |
359 | 359 |
|
360 | 360 |
/// @{ |
361 | 361 |
|
362 | 362 |
/// \brief Set the lower bounds on the arcs. |
363 | 363 |
/// |
364 | 364 |
/// This function sets the lower bounds on the arcs. |
365 | 365 |
/// If it is not used before calling \ref run(), the lower bounds |
366 | 366 |
/// will be set to zero on all arcs. |
367 | 367 |
/// |
368 | 368 |
/// \param map An arc map storing the lower bounds. |
369 | 369 |
/// Its \c Value type must be convertible to the \c Value type |
370 | 370 |
/// of the algorithm. |
371 | 371 |
/// |
372 | 372 |
/// \return <tt>(*this)</tt> |
373 | 373 |
template <typename LowerMap> |
374 | 374 |
CostScaling& lowerMap(const LowerMap& map) { |
375 | 375 |
_have_lower = true; |
376 | 376 |
for (ArcIt a(_graph); a != INVALID; ++a) { |
377 | 377 |
_lower[_arc_idf[a]] = map[a]; |
378 | 378 |
_lower[_arc_idb[a]] = map[a]; |
379 | 379 |
} |
380 | 380 |
return *this; |
381 | 381 |
} |
382 | 382 |
|
383 | 383 |
/// \brief Set the upper bounds (capacities) on the arcs. |
384 | 384 |
/// |
385 | 385 |
/// This function sets the upper bounds (capacities) on the arcs. |
386 | 386 |
/// If it is not used before calling \ref run(), the upper bounds |
387 | 387 |
/// will be set to \ref INF on all arcs (i.e. the flow value will be |
388 | 388 |
/// unbounded from above). |
389 | 389 |
/// |
390 | 390 |
/// \param map An arc map storing the upper bounds. |
391 | 391 |
/// Its \c Value type must be convertible to the \c Value type |
392 | 392 |
/// of the algorithm. |
393 | 393 |
/// |
394 | 394 |
/// \return <tt>(*this)</tt> |
395 | 395 |
template<typename UpperMap> |
396 | 396 |
CostScaling& upperMap(const UpperMap& map) { |
397 | 397 |
for (ArcIt a(_graph); a != INVALID; ++a) { |
398 | 398 |
_upper[_arc_idf[a]] = map[a]; |
399 | 399 |
} |
400 | 400 |
return *this; |
401 | 401 |
} |
402 | 402 |
|
403 | 403 |
/// \brief Set the costs of the arcs. |
404 | 404 |
/// |
405 | 405 |
/// This function sets the costs of the arcs. |
406 | 406 |
/// If it is not used before calling \ref run(), the costs |
407 | 407 |
/// will be set to \c 1 on all arcs. |
408 | 408 |
/// |
409 | 409 |
/// \param map An arc map storing the costs. |
410 | 410 |
/// Its \c Value type must be convertible to the \c Cost type |
411 | 411 |
/// of the algorithm. |
412 | 412 |
/// |
413 | 413 |
/// \return <tt>(*this)</tt> |
414 | 414 |
template<typename CostMap> |
415 | 415 |
CostScaling& costMap(const CostMap& map) { |
416 | 416 |
for (ArcIt a(_graph); a != INVALID; ++a) { |
417 | 417 |
_scost[_arc_idf[a]] = map[a]; |
418 | 418 |
_scost[_arc_idb[a]] = -map[a]; |
419 | 419 |
} |
420 | 420 |
return *this; |
421 | 421 |
} |
422 | 422 |
|
423 | 423 |
/// \brief Set the supply values of the nodes. |
424 | 424 |
/// |
425 | 425 |
/// This function sets the supply values of the nodes. |
426 | 426 |
/// If neither this function nor \ref stSupply() is used before |
427 | 427 |
/// calling \ref run(), the supply of each node will be set to zero. |
428 | 428 |
/// |
429 | 429 |
/// \param map A node map storing the supply values. |
430 | 430 |
/// Its \c Value type must be convertible to the \c Value type |
431 | 431 |
/// of the algorithm. |
432 | 432 |
/// |
433 | 433 |
/// \return <tt>(*this)</tt> |
434 | 434 |
template<typename SupplyMap> |
435 | 435 |
CostScaling& supplyMap(const SupplyMap& map) { |
436 | 436 |
for (NodeIt n(_graph); n != INVALID; ++n) { |
437 | 437 |
_supply[_node_id[n]] = map[n]; |
438 | 438 |
} |
439 | 439 |
return *this; |
440 | 440 |
} |
441 | 441 |
|
442 | 442 |
/// \brief Set single source and target nodes and a supply value. |
443 | 443 |
/// |
444 | 444 |
/// This function sets a single source node and a single target node |
445 | 445 |
/// and the required flow value. |
446 | 446 |
/// If neither this function nor \ref supplyMap() is used before |
447 | 447 |
/// calling \ref run(), the supply of each node will be set to zero. |
448 | 448 |
/// |
449 | 449 |
/// Using this function has the same effect as using \ref supplyMap() |
450 | 450 |
/// with such a map in which \c k is assigned to \c s, \c -k is |
451 | 451 |
/// assigned to \c t and all other nodes have zero supply value. |
452 | 452 |
/// |
453 | 453 |
/// \param s The source node. |
454 | 454 |
/// \param t The target node. |
455 | 455 |
/// \param k The required amount of flow from node \c s to node \c t |
456 | 456 |
/// (i.e. the supply of \c s and the demand of \c t). |
457 | 457 |
/// |
458 | 458 |
/// \return <tt>(*this)</tt> |
459 | 459 |
CostScaling& stSupply(const Node& s, const Node& t, Value k) { |
460 | 460 |
for (int i = 0; i != _res_node_num; ++i) { |
461 | 461 |
_supply[i] = 0; |
462 | 462 |
} |
463 | 463 |
_supply[_node_id[s]] = k; |
464 | 464 |
_supply[_node_id[t]] = -k; |
465 | 465 |
return *this; |
466 | 466 |
} |
467 | 467 |
|
468 | 468 |
/// @} |
469 | 469 |
|
470 | 470 |
/// \name Execution control |
471 | 471 |
/// The algorithm can be executed using \ref run(). |
472 | 472 |
|
473 | 473 |
/// @{ |
474 | 474 |
|
475 | 475 |
/// \brief Run the algorithm. |
476 | 476 |
/// |
477 | 477 |
/// This function runs the algorithm. |
478 | 478 |
/// The paramters can be specified using functions \ref lowerMap(), |
479 | 479 |
/// \ref upperMap(), \ref costMap(), \ref supplyMap(), \ref stSupply(). |
480 | 480 |
/// For example, |
481 | 481 |
/// \code |
482 | 482 |
/// CostScaling<ListDigraph> cs(graph); |
483 | 483 |
/// cs.lowerMap(lower).upperMap(upper).costMap(cost) |
484 | 484 |
/// .supplyMap(sup).run(); |
485 | 485 |
/// \endcode |
486 | 486 |
/// |
487 | 487 |
/// This function can be called more than once. All the given parameters |
488 | 488 |
/// are kept for the next call, unless \ref resetParams() or \ref reset() |
489 | 489 |
/// is used, thus only the modified parameters have to be set again. |
490 | 490 |
/// If the underlying digraph was also modified after the construction |
491 | 491 |
/// of the class (or the last \ref reset() call), then the \ref reset() |
492 | 492 |
/// function must be called. |
493 | 493 |
/// |
494 | 494 |
/// \param method The internal method that will be used in the |
495 | 495 |
/// algorithm. For more information, see \ref Method. |
496 | 496 |
/// \param factor The cost scaling factor. It must be larger than one. |
497 | 497 |
/// |
498 | 498 |
/// \return \c INFEASIBLE if no feasible flow exists, |
499 | 499 |
/// \n \c OPTIMAL if the problem has optimal solution |
500 | 500 |
/// (i.e. it is feasible and bounded), and the algorithm has found |
501 | 501 |
/// optimal flow and node potentials (primal and dual solutions), |
502 | 502 |
/// \n \c UNBOUNDED if the digraph contains an arc of negative cost |
503 | 503 |
/// and infinite upper bound. It means that the objective function |
504 | 504 |
/// is unbounded on that arc, however, note that it could actually be |
505 | 505 |
/// bounded over the feasible flows, but this algroithm cannot handle |
506 | 506 |
/// these cases. |
507 | 507 |
/// |
508 | 508 |
/// \see ProblemType, Method |
509 | 509 |
/// \see resetParams(), reset() |
510 | 510 |
ProblemType run(Method method = PARTIAL_AUGMENT, int factor = 8) { |
511 | 511 |
_alpha = factor; |
512 | 512 |
ProblemType pt = init(); |
513 | 513 |
if (pt != OPTIMAL) return pt; |
514 | 514 |
start(method); |
515 | 515 |
return OPTIMAL; |
516 | 516 |
} |
517 | 517 |
1 | 1 |
/* -*- mode: C++; indent-tabs-mode: nil; -*- |
2 | 2 |
* |
3 | 3 |
* This file is a part of LEMON, a generic C++ optimization library. |
4 | 4 |
* |
5 |
* Copyright (C) 2003- |
|
5 |
* Copyright (C) 2003-2011 |
|
6 | 6 |
* Egervary Jeno Kombinatorikus Optimalizalasi Kutatocsoport |
7 | 7 |
* (Egervary Research Group on Combinatorial Optimization, EGRES). |
8 | 8 |
* |
9 | 9 |
* Permission to use, modify and distribute this software is granted |
10 | 10 |
* provided that this copyright notice appears in all copies. For |
11 | 11 |
* precise terms see the accompanying LICENSE file. |
12 | 12 |
* |
13 | 13 |
* This software is provided "AS IS" with no warranty of any kind, |
14 | 14 |
* express or implied, and with no claim as to its suitability for any |
15 | 15 |
* purpose. |
16 | 16 |
* |
17 | 17 |
*/ |
18 | 18 |
|
19 | 19 |
#ifndef LEMON_MAPS_H |
20 | 20 |
#define LEMON_MAPS_H |
21 | 21 |
|
22 | 22 |
#include <iterator> |
23 | 23 |
#include <functional> |
24 | 24 |
#include <vector> |
25 | 25 |
#include <map> |
26 | 26 |
|
27 | 27 |
#include <lemon/core.h> |
28 | 28 |
|
29 | 29 |
///\file |
30 | 30 |
///\ingroup maps |
31 | 31 |
///\brief Miscellaneous property maps |
32 | 32 |
|
33 | 33 |
namespace lemon { |
34 | 34 |
|
35 | 35 |
/// \addtogroup maps |
36 | 36 |
/// @{ |
37 | 37 |
|
38 | 38 |
/// Base class of maps. |
39 | 39 |
|
40 | 40 |
/// Base class of maps. It provides the necessary type definitions |
41 | 41 |
/// required by the map %concepts. |
42 | 42 |
template<typename K, typename V> |
43 | 43 |
class MapBase { |
44 | 44 |
public: |
45 | 45 |
/// \brief The key type of the map. |
46 | 46 |
typedef K Key; |
47 | 47 |
/// \brief The value type of the map. |
48 | 48 |
/// (The type of objects associated with the keys). |
49 | 49 |
typedef V Value; |
50 | 50 |
}; |
51 | 51 |
|
52 | 52 |
|
53 | 53 |
/// Null map. (a.k.a. DoNothingMap) |
54 | 54 |
|
55 | 55 |
/// This map can be used if you have to provide a map only for |
56 | 56 |
/// its type definitions, or if you have to provide a writable map, |
57 | 57 |
/// but data written to it is not required (i.e. it will be sent to |
58 | 58 |
/// <tt>/dev/null</tt>). |
59 | 59 |
/// It conforms to the \ref concepts::ReadWriteMap "ReadWriteMap" concept. |
60 | 60 |
/// |
61 | 61 |
/// \sa ConstMap |
62 | 62 |
template<typename K, typename V> |
63 | 63 |
class NullMap : public MapBase<K, V> { |
64 | 64 |
public: |
65 | 65 |
///\e |
66 | 66 |
typedef K Key; |
67 | 67 |
///\e |
68 | 68 |
typedef V Value; |
69 | 69 |
|
70 | 70 |
/// Gives back a default constructed element. |
71 | 71 |
Value operator[](const Key&) const { return Value(); } |
72 | 72 |
/// Absorbs the value. |
73 | 73 |
void set(const Key&, const Value&) {} |
74 | 74 |
}; |
75 | 75 |
|
76 | 76 |
/// Returns a \c NullMap class |
77 | 77 |
|
78 | 78 |
/// This function just returns a \c NullMap class. |
79 | 79 |
/// \relates NullMap |
80 | 80 |
template <typename K, typename V> |
81 | 81 |
NullMap<K, V> nullMap() { |
82 | 82 |
return NullMap<K, V>(); |
83 | 83 |
} |
84 | 84 |
|
85 | 85 |
|
86 | 86 |
/// Constant map. |
87 | 87 |
|
88 | 88 |
/// This \ref concepts::ReadMap "readable map" assigns a specified |
89 | 89 |
/// value to each key. |
90 | 90 |
/// |
91 | 91 |
/// In other aspects it is equivalent to \c NullMap. |
92 | 92 |
/// So it conforms to the \ref concepts::ReadWriteMap "ReadWriteMap" |
93 | 93 |
/// concept, but it absorbs the data written to it. |
94 | 94 |
/// |
95 | 95 |
/// The simplest way of using this map is through the constMap() |
96 | 96 |
/// function. |
97 | 97 |
/// |
98 | 98 |
/// \sa NullMap |
99 | 99 |
/// \sa IdentityMap |
100 | 100 |
template<typename K, typename V> |
101 | 101 |
class ConstMap : public MapBase<K, V> { |
102 | 102 |
private: |
103 | 103 |
V _value; |
104 | 104 |
public: |
105 | 105 |
///\e |
106 | 106 |
typedef K Key; |
107 | 107 |
///\e |
108 | 108 |
typedef V Value; |
109 | 109 |
|
110 | 110 |
/// Default constructor |
111 | 111 |
|
112 | 112 |
/// Default constructor. |
113 | 113 |
/// The value of the map will be default constructed. |
114 | 114 |
ConstMap() {} |
115 | 115 |
|
116 | 116 |
/// Constructor with specified initial value |
117 | 117 |
|
118 | 118 |
/// Constructor with specified initial value. |
119 | 119 |
/// \param v The initial value of the map. |
120 | 120 |
ConstMap(const Value &v) : _value(v) {} |
121 | 121 |
|
122 | 122 |
/// Gives back the specified value. |
123 | 123 |
Value operator[](const Key&) const { return _value; } |
124 | 124 |
|
125 | 125 |
/// Absorbs the value. |
126 | 126 |
void set(const Key&, const Value&) {} |
127 | 127 |
|
128 | 128 |
/// Sets the value that is assigned to each key. |
129 | 129 |
void setAll(const Value &v) { |
130 | 130 |
_value = v; |
131 | 131 |
} |
132 | 132 |
|
133 | 133 |
template<typename V1> |
134 | 134 |
ConstMap(const ConstMap<K, V1> &, const Value &v) : _value(v) {} |
135 | 135 |
}; |
136 | 136 |
|
137 | 137 |
/// Returns a \c ConstMap class |
138 | 138 |
|
139 | 139 |
/// This function just returns a \c ConstMap class. |
140 | 140 |
/// \relates ConstMap |
141 | 141 |
template<typename K, typename V> |
142 | 142 |
inline ConstMap<K, V> constMap(const V &v) { |
143 | 143 |
return ConstMap<K, V>(v); |
144 | 144 |
} |
145 | 145 |
|
146 | 146 |
template<typename K, typename V> |
147 | 147 |
inline ConstMap<K, V> constMap() { |
148 | 148 |
return ConstMap<K, V>(); |
149 | 149 |
} |
150 | 150 |
|
151 | 151 |
|
152 | 152 |
template<typename T, T v> |
153 | 153 |
struct Const {}; |
154 | 154 |
|
155 | 155 |
/// Constant map with inlined constant value. |
156 | 156 |
|
157 | 157 |
/// This \ref concepts::ReadMap "readable map" assigns a specified |
158 | 158 |
/// value to each key. |
159 | 159 |
/// |
160 | 160 |
/// In other aspects it is equivalent to \c NullMap. |
161 | 161 |
/// So it conforms to the \ref concepts::ReadWriteMap "ReadWriteMap" |
162 | 162 |
/// concept, but it absorbs the data written to it. |
163 | 163 |
/// |
164 | 164 |
/// The simplest way of using this map is through the constMap() |
165 | 165 |
/// function. |
166 | 166 |
/// |
167 | 167 |
/// \sa NullMap |
168 | 168 |
/// \sa IdentityMap |
169 | 169 |
template<typename K, typename V, V v> |
170 | 170 |
class ConstMap<K, Const<V, v> > : public MapBase<K, V> { |
171 | 171 |
public: |
172 | 172 |
///\e |
173 | 173 |
typedef K Key; |
174 | 174 |
///\e |
175 | 175 |
typedef V Value; |
176 | 176 |
|
177 | 177 |
/// Constructor. |
178 | 178 |
ConstMap() {} |
179 | 179 |
|
180 | 180 |
/// Gives back the specified value. |
181 | 181 |
Value operator[](const Key&) const { return v; } |
182 | 182 |
|
183 | 183 |
/// Absorbs the value. |
184 | 184 |
void set(const Key&, const Value&) {} |
185 | 185 |
}; |
186 | 186 |
|
187 | 187 |
/// Returns a \c ConstMap class with inlined constant value |
188 | 188 |
|
189 | 189 |
/// This function just returns a \c ConstMap class with inlined |
190 | 190 |
/// constant value. |
191 | 191 |
/// \relates ConstMap |
192 | 192 |
template<typename K, typename V, V v> |
193 | 193 |
inline ConstMap<K, Const<V, v> > constMap() { |
194 | 194 |
return ConstMap<K, Const<V, v> >(); |
195 | 195 |
} |
196 | 196 |
|
197 | 197 |
|
198 | 198 |
/// Identity map. |
199 | 199 |
|
200 | 200 |
/// This \ref concepts::ReadMap "read-only map" gives back the given |
201 | 201 |
/// key as value without any modification. |
202 | 202 |
/// |
203 | 203 |
/// \sa ConstMap |
204 | 204 |
template <typename T> |
205 | 205 |
class IdentityMap : public MapBase<T, T> { |
206 | 206 |
public: |
207 | 207 |
///\e |
208 | 208 |
typedef T Key; |
209 | 209 |
///\e |
210 | 210 |
typedef T Value; |
211 | 211 |
|
212 | 212 |
/// Gives back the given value without any modification. |
213 | 213 |
Value operator[](const Key &k) const { |
214 | 214 |
return k; |
215 | 215 |
} |
216 | 216 |
}; |
217 | 217 |
|
218 | 218 |
/// Returns an \c IdentityMap class |
219 | 219 |
|
220 | 220 |
/// This function just returns an \c IdentityMap class. |
221 | 221 |
/// \relates IdentityMap |
222 | 222 |
template<typename T> |
223 | 223 |
inline IdentityMap<T> identityMap() { |
224 | 224 |
return IdentityMap<T>(); |
225 | 225 |
} |
226 | 226 |
|
227 | 227 |
|
228 | 228 |
/// \brief Map for storing values for integer keys from the range |
229 | 229 |
/// <tt>[0..size-1]</tt>. |
230 | 230 |
/// |
231 | 231 |
/// This map is essentially a wrapper for \c std::vector. It assigns |
232 | 232 |
/// values to integer keys from the range <tt>[0..size-1]</tt>. |
233 | 233 |
/// It can be used together with some data structures, e.g. |
234 | 234 |
/// heap types and \c UnionFind, when the used items are small |
235 | 235 |
/// integers. This map conforms to the \ref concepts::ReferenceMap |
236 | 236 |
/// "ReferenceMap" concept. |
237 | 237 |
/// |
238 | 238 |
/// The simplest way of using this map is through the rangeMap() |
239 | 239 |
/// function. |
240 | 240 |
template <typename V> |
241 | 241 |
class RangeMap : public MapBase<int, V> { |
242 | 242 |
template <typename V1> |
243 | 243 |
friend class RangeMap; |
244 | 244 |
private: |
245 | 245 |
|
246 | 246 |
typedef std::vector<V> Vector; |
247 | 247 |
Vector _vector; |
248 | 248 |
|
249 | 249 |
public: |
250 | 250 |
|
251 | 251 |
/// Key type |
252 | 252 |
typedef int Key; |
253 | 253 |
/// Value type |
254 | 254 |
typedef V Value; |
255 | 255 |
/// Reference type |
256 | 256 |
typedef typename Vector::reference Reference; |
257 | 257 |
/// Const reference type |
258 | 258 |
typedef typename Vector::const_reference ConstReference; |
259 | 259 |
|
260 | 260 |
typedef True ReferenceMapTag; |
261 | 261 |
|
262 | 262 |
public: |
263 | 263 |
|
264 | 264 |
/// Constructor with specified default value. |
265 | 265 |
RangeMap(int size = 0, const Value &value = Value()) |
266 | 266 |
: _vector(size, value) {} |
267 | 267 |
|
268 | 268 |
/// Constructs the map from an appropriate \c std::vector. |
269 | 269 |
template <typename V1> |
270 | 270 |
RangeMap(const std::vector<V1>& vector) |
271 | 271 |
: _vector(vector.begin(), vector.end()) {} |
272 | 272 |
|
273 | 273 |
/// Constructs the map from another \c RangeMap. |
274 | 274 |
template <typename V1> |
275 | 275 |
RangeMap(const RangeMap<V1> &c) |
276 | 276 |
: _vector(c._vector.begin(), c._vector.end()) {} |
277 | 277 |
|
278 | 278 |
/// Returns the size of the map. |
279 | 279 |
int size() { |
280 | 280 |
return _vector.size(); |
281 | 281 |
} |
282 | 282 |
|
283 | 283 |
/// Resizes the map. |
284 | 284 |
|
285 | 285 |
/// Resizes the underlying \c std::vector container, so changes the |
286 | 286 |
/// keyset of the map. |
287 | 287 |
/// \param size The new size of the map. The new keyset will be the |
288 | 288 |
/// range <tt>[0..size-1]</tt>. |
289 | 289 |
/// \param value The default value to assign to the new keys. |
290 | 290 |
void resize(int size, const Value &value = Value()) { |
291 | 291 |
_vector.resize(size, value); |
292 | 292 |
} |
293 | 293 |
|
294 | 294 |
private: |
295 | 295 |
|
296 | 296 |
RangeMap& operator=(const RangeMap&); |
297 | 297 |
|
298 | 298 |
public: |
299 | 299 |
|
300 | 300 |
///\e |
301 | 301 |
Reference operator[](const Key &k) { |
302 | 302 |
return _vector[k]; |
303 | 303 |
} |
304 | 304 |
|
305 | 305 |
///\e |
306 | 306 |
ConstReference operator[](const Key &k) const { |
307 | 307 |
return _vector[k]; |
308 | 308 |
} |
309 | 309 |
|
310 | 310 |
///\e |
311 | 311 |
void set(const Key &k, const Value &v) { |
312 | 312 |
_vector[k] = v; |
313 | 313 |
} |
314 | 314 |
}; |
315 | 315 |
|
316 | 316 |
/// Returns a \c RangeMap class |
317 | 317 |
|
318 | 318 |
/// This function just returns a \c RangeMap class. |
319 | 319 |
/// \relates RangeMap |
320 | 320 |
template<typename V> |
321 | 321 |
inline RangeMap<V> rangeMap(int size = 0, const V &value = V()) { |
322 | 322 |
return RangeMap<V>(size, value); |
323 | 323 |
} |
324 | 324 |
|
325 | 325 |
/// \brief Returns a \c RangeMap class created from an appropriate |
326 | 326 |
/// \c std::vector |
327 | 327 |
|
328 | 328 |
/// This function just returns a \c RangeMap class created from an |
329 | 329 |
/// appropriate \c std::vector. |
330 | 330 |
/// \relates RangeMap |
331 | 331 |
template<typename V> |
332 | 332 |
inline RangeMap<V> rangeMap(const std::vector<V> &vector) { |
333 | 333 |
return RangeMap<V>(vector); |
334 | 334 |
} |
335 | 335 |
|
336 | 336 |
|
337 | 337 |
/// Map type based on \c std::map |
338 | 338 |
|
339 | 339 |
/// This map is essentially a wrapper for \c std::map with addition |
340 | 340 |
/// that you can specify a default value for the keys that are not |
341 | 341 |
/// stored actually. This value can be different from the default |
342 | 342 |
/// contructed value (i.e. \c %Value()). |
343 | 343 |
/// This type conforms to the \ref concepts::ReferenceMap "ReferenceMap" |
344 | 344 |
/// concept. |
345 | 345 |
/// |
346 | 346 |
/// This map is useful if a default value should be assigned to most of |
347 | 347 |
/// the keys and different values should be assigned only to a few |
348 | 348 |
/// keys (i.e. the map is "sparse"). |
349 | 349 |
/// The name of this type also refers to this important usage. |
350 | 350 |
/// |
351 | 351 |
/// Apart form that, this map can be used in many other cases since it |
352 | 352 |
/// is based on \c std::map, which is a general associative container. |
353 | 353 |
/// However, keep in mind that it is usually not as efficient as other |
354 | 354 |
/// maps. |
355 | 355 |
/// |
356 | 356 |
/// The simplest way of using this map is through the sparseMap() |
357 | 357 |
/// function. |
358 | 358 |
template <typename K, typename V, typename Comp = std::less<K> > |
359 | 359 |
class SparseMap : public MapBase<K, V> { |
360 | 360 |
template <typename K1, typename V1, typename C1> |
361 | 361 |
friend class SparseMap; |
362 | 362 |
public: |
363 | 363 |
|
364 | 364 |
/// Key type |
365 | 365 |
typedef K Key; |
366 | 366 |
/// Value type |
367 | 367 |
typedef V Value; |
368 | 368 |
/// Reference type |
369 | 369 |
typedef Value& Reference; |
370 | 370 |
/// Const reference type |
371 | 371 |
typedef const Value& ConstReference; |
372 | 372 |
|
373 | 373 |
typedef True ReferenceMapTag; |
374 | 374 |
|
375 | 375 |
private: |
376 | 376 |
|
377 | 377 |
typedef std::map<K, V, Comp> Map; |
378 | 378 |
Map _map; |
379 | 379 |
Value _value; |
380 | 380 |
|
381 | 381 |
public: |
382 | 382 |
|
383 | 383 |
/// \brief Constructor with specified default value. |
384 | 384 |
SparseMap(const Value &value = Value()) : _value(value) {} |
385 | 385 |
/// \brief Constructs the map from an appropriate \c std::map, and |
386 | 386 |
/// explicitly specifies a default value. |
387 | 387 |
template <typename V1, typename Comp1> |
388 | 388 |
SparseMap(const std::map<Key, V1, Comp1> &map, |
389 | 389 |
const Value &value = Value()) |
390 | 390 |
: _map(map.begin(), map.end()), _value(value) {} |
391 | 391 |
|
392 | 392 |
/// \brief Constructs the map from another \c SparseMap. |
393 | 393 |
template<typename V1, typename Comp1> |
394 | 394 |
SparseMap(const SparseMap<Key, V1, Comp1> &c) |
395 | 395 |
: _map(c._map.begin(), c._map.end()), _value(c._value) {} |
396 | 396 |
|
397 | 397 |
private: |
398 | 398 |
|
399 | 399 |
SparseMap& operator=(const SparseMap&); |
400 | 400 |
|
401 | 401 |
public: |
402 | 402 |
|
403 | 403 |
///\e |
404 | 404 |
Reference operator[](const Key &k) { |
405 | 405 |
typename Map::iterator it = _map.lower_bound(k); |
406 | 406 |
if (it != _map.end() && !_map.key_comp()(k, it->first)) |
407 | 407 |
return it->second; |
408 | 408 |
else |
409 | 409 |
return _map.insert(it, std::make_pair(k, _value))->second; |
410 | 410 |
} |
411 | 411 |
|
412 | 412 |
///\e |
413 | 413 |
ConstReference operator[](const Key &k) const { |
414 | 414 |
typename Map::const_iterator it = _map.find(k); |
415 | 415 |
if (it != _map.end()) |
416 | 416 |
return it->second; |
417 | 417 |
else |
418 | 418 |
return _value; |
419 | 419 |
} |
420 | 420 |
|
421 | 421 |
///\e |
422 | 422 |
void set(const Key &k, const Value &v) { |
423 | 423 |
typename Map::iterator it = _map.lower_bound(k); |
424 | 424 |
if (it != _map.end() && !_map.key_comp()(k, it->first)) |
425 | 425 |
it->second = v; |
426 | 426 |
else |
427 | 427 |
_map.insert(it, std::make_pair(k, v)); |
428 | 428 |
} |
429 | 429 |
|
430 | 430 |
///\e |
431 | 431 |
void setAll(const Value &v) { |
432 | 432 |
_value = v; |
433 | 433 |
_map.clear(); |
434 | 434 |
} |
435 | 435 |
}; |
436 | 436 |
|
437 | 437 |
/// Returns a \c SparseMap class |
438 | 438 |
|
439 | 439 |
/// This function just returns a \c SparseMap class with specified |
440 | 440 |
/// default value. |
441 | 441 |
/// \relates SparseMap |
442 | 442 |
template<typename K, typename V, typename Compare> |
443 | 443 |
inline SparseMap<K, V, Compare> sparseMap(const V& value = V()) { |
444 | 444 |
return SparseMap<K, V, Compare>(value); |
445 | 445 |
} |
446 | 446 |
|
447 | 447 |
template<typename K, typename V> |
448 | 448 |
inline SparseMap<K, V, std::less<K> > sparseMap(const V& value = V()) { |
449 | 449 |
return SparseMap<K, V, std::less<K> >(value); |
450 | 450 |
} |
451 | 451 |
|
452 | 452 |
/// \brief Returns a \c SparseMap class created from an appropriate |
453 | 453 |
/// \c std::map |
454 | 454 |
|
455 | 455 |
/// This function just returns a \c SparseMap class created from an |
456 | 456 |
/// appropriate \c std::map. |
457 | 457 |
/// \relates SparseMap |
458 | 458 |
template<typename K, typename V, typename Compare> |
459 | 459 |
inline SparseMap<K, V, Compare> |
460 | 460 |
sparseMap(const std::map<K, V, Compare> &map, const V& value = V()) |
461 | 461 |
{ |
462 | 462 |
return SparseMap<K, V, Compare>(map, value); |
463 | 463 |
} |
464 | 464 |
|
465 | 465 |
/// @} |
466 | 466 |
|
467 | 467 |
/// \addtogroup map_adaptors |
468 | 468 |
/// @{ |
469 | 469 |
|
470 | 470 |
/// Composition of two maps |
471 | 471 |
|
472 | 472 |
/// This \ref concepts::ReadMap "read-only map" returns the |
473 | 473 |
/// composition of two given maps. That is to say, if \c m1 is of |
474 | 474 |
/// type \c M1 and \c m2 is of \c M2, then for |
475 | 475 |
/// \code |
476 | 476 |
/// ComposeMap<M1, M2> cm(m1,m2); |
477 | 477 |
/// \endcode |
478 | 478 |
/// <tt>cm[x]</tt> will be equal to <tt>m1[m2[x]]</tt>. |
479 | 479 |
/// |
480 | 480 |
/// The \c Key type of the map is inherited from \c M2 and the |
481 | 481 |
/// \c Value type is from \c M1. |
482 | 482 |
/// \c M2::Value must be convertible to \c M1::Key. |
483 | 483 |
/// |
484 | 484 |
/// The simplest way of using this map is through the composeMap() |
485 | 485 |
/// function. |
486 | 486 |
/// |
487 | 487 |
/// \sa CombineMap |
488 | 488 |
template <typename M1, typename M2> |
489 | 489 |
class ComposeMap : public MapBase<typename M2::Key, typename M1::Value> { |
490 | 490 |
const M1 &_m1; |
491 | 491 |
const M2 &_m2; |
492 | 492 |
public: |
493 | 493 |
///\e |
494 | 494 |
typedef typename M2::Key Key; |
495 | 495 |
///\e |
496 | 496 |
typedef typename M1::Value Value; |
497 | 497 |
|
498 | 498 |
/// Constructor |
499 | 499 |
ComposeMap(const M1 &m1, const M2 &m2) : _m1(m1), _m2(m2) {} |
500 | 500 |
|
501 | 501 |
///\e |
502 | 502 |
typename MapTraits<M1>::ConstReturnValue |
503 | 503 |
operator[](const Key &k) const { return _m1[_m2[k]]; } |
504 | 504 |
}; |
505 | 505 |
|
506 | 506 |
/// Returns a \c ComposeMap class |
507 | 507 |
|
508 | 508 |
/// This function just returns a \c ComposeMap class. |
509 | 509 |
/// |
510 | 510 |
/// If \c m1 and \c m2 are maps and the \c Value type of \c m2 is |
511 | 511 |
/// convertible to the \c Key of \c m1, then <tt>composeMap(m1,m2)[x]</tt> |
512 | 512 |
/// will be equal to <tt>m1[m2[x]]</tt>. |
513 | 513 |
/// |
514 | 514 |
/// \relates ComposeMap |
515 | 515 |
template <typename M1, typename M2> |
516 | 516 |
inline ComposeMap<M1, M2> composeMap(const M1 &m1, const M2 &m2) { |
517 | 517 |
return ComposeMap<M1, M2>(m1, m2); |
1 | 1 |
/* -*- mode: C++; indent-tabs-mode: nil; -*- |
2 | 2 |
* |
3 | 3 |
* This file is a part of LEMON, a generic C++ optimization library. |
4 | 4 |
* |
5 |
* Copyright (C) 2003- |
|
5 |
* Copyright (C) 2003-2011 |
|
6 | 6 |
* Egervary Jeno Kombinatorikus Optimalizalasi Kutatocsoport |
7 | 7 |
* (Egervary Research Group on Combinatorial Optimization, EGRES). |
8 | 8 |
* |
9 | 9 |
* Permission to use, modify and distribute this software is granted |
10 | 10 |
* provided that this copyright notice appears in all copies. For |
11 | 11 |
* precise terms see the accompanying LICENSE file. |
12 | 12 |
* |
13 | 13 |
* This software is provided "AS IS" with no warranty of any kind, |
14 | 14 |
* express or implied, and with no claim as to its suitability for any |
15 | 15 |
* purpose. |
16 | 16 |
* |
17 | 17 |
*/ |
18 | 18 |
|
19 | 19 |
#ifndef LEMON_PREFLOW_H |
20 | 20 |
#define LEMON_PREFLOW_H |
21 | 21 |
|
22 | 22 |
#include <lemon/tolerance.h> |
23 | 23 |
#include <lemon/elevator.h> |
24 | 24 |
|
25 | 25 |
/// \file |
26 | 26 |
/// \ingroup max_flow |
27 | 27 |
/// \brief Implementation of the preflow algorithm. |
28 | 28 |
|
29 | 29 |
namespace lemon { |
30 | 30 |
|
31 | 31 |
/// \brief Default traits class of Preflow class. |
32 | 32 |
/// |
33 | 33 |
/// Default traits class of Preflow class. |
34 | 34 |
/// \tparam GR Digraph type. |
35 | 35 |
/// \tparam CAP Capacity map type. |
36 | 36 |
template <typename GR, typename CAP> |
37 | 37 |
struct PreflowDefaultTraits { |
38 | 38 |
|
39 | 39 |
/// \brief The type of the digraph the algorithm runs on. |
40 | 40 |
typedef GR Digraph; |
41 | 41 |
|
42 | 42 |
/// \brief The type of the map that stores the arc capacities. |
43 | 43 |
/// |
44 | 44 |
/// The type of the map that stores the arc capacities. |
45 | 45 |
/// It must meet the \ref concepts::ReadMap "ReadMap" concept. |
46 | 46 |
typedef CAP CapacityMap; |
47 | 47 |
|
48 | 48 |
/// \brief The type of the flow values. |
49 | 49 |
typedef typename CapacityMap::Value Value; |
50 | 50 |
|
51 | 51 |
/// \brief The type of the map that stores the flow values. |
52 | 52 |
/// |
53 | 53 |
/// The type of the map that stores the flow values. |
54 | 54 |
/// It must meet the \ref concepts::ReadWriteMap "ReadWriteMap" concept. |
55 | 55 |
#ifdef DOXYGEN |
56 | 56 |
typedef GR::ArcMap<Value> FlowMap; |
57 | 57 |
#else |
58 | 58 |
typedef typename Digraph::template ArcMap<Value> FlowMap; |
59 | 59 |
#endif |
60 | 60 |
|
61 | 61 |
/// \brief Instantiates a FlowMap. |
62 | 62 |
/// |
63 | 63 |
/// This function instantiates a \ref FlowMap. |
64 | 64 |
/// \param digraph The digraph for which we would like to define |
65 | 65 |
/// the flow map. |
66 | 66 |
static FlowMap* createFlowMap(const Digraph& digraph) { |
67 | 67 |
return new FlowMap(digraph); |
68 | 68 |
} |
69 | 69 |
|
70 | 70 |
/// \brief The elevator type used by Preflow algorithm. |
71 | 71 |
/// |
72 | 72 |
/// The elevator type used by Preflow algorithm. |
73 | 73 |
/// |
74 | 74 |
/// \sa Elevator, LinkedElevator |
75 | 75 |
#ifdef DOXYGEN |
76 | 76 |
typedef lemon::Elevator<GR, GR::Node> Elevator; |
77 | 77 |
#else |
78 | 78 |
typedef lemon::Elevator<Digraph, typename Digraph::Node> Elevator; |
79 | 79 |
#endif |
80 | 80 |
|
81 | 81 |
/// \brief Instantiates an Elevator. |
82 | 82 |
/// |
83 | 83 |
/// This function instantiates an \ref Elevator. |
84 | 84 |
/// \param digraph The digraph for which we would like to define |
85 | 85 |
/// the elevator. |
86 | 86 |
/// \param max_level The maximum level of the elevator. |
87 | 87 |
static Elevator* createElevator(const Digraph& digraph, int max_level) { |
88 | 88 |
return new Elevator(digraph, max_level); |
89 | 89 |
} |
90 | 90 |
|
91 | 91 |
/// \brief The tolerance used by the algorithm |
92 | 92 |
/// |
93 | 93 |
/// The tolerance used by the algorithm to handle inexact computation. |
94 | 94 |
typedef lemon::Tolerance<Value> Tolerance; |
95 | 95 |
|
96 | 96 |
}; |
97 | 97 |
|
98 | 98 |
|
99 | 99 |
/// \ingroup max_flow |
100 | 100 |
/// |
101 | 101 |
/// \brief %Preflow algorithm class. |
102 | 102 |
/// |
103 | 103 |
/// This class provides an implementation of Goldberg-Tarjan's \e preflow |
104 | 104 |
/// \e push-relabel algorithm producing a \ref max_flow |
105 | 105 |
/// "flow of maximum value" in a digraph \ref clrs01algorithms, |
106 | 106 |
/// \ref amo93networkflows, \ref goldberg88newapproach. |
107 | 107 |
/// The preflow algorithms are the fastest known maximum |
108 | 108 |
/// flow algorithms. The current implementation uses a mixture of the |
109 | 109 |
/// \e "highest label" and the \e "bound decrease" heuristics. |
110 | 110 |
/// The worst case time complexity of the algorithm is \f$O(n^2\sqrt{e})\f$. |
111 | 111 |
/// |
112 | 112 |
/// The algorithm consists of two phases. After the first phase |
113 | 113 |
/// the maximum flow value and the minimum cut is obtained. The |
114 | 114 |
/// second phase constructs a feasible maximum flow on each arc. |
115 | 115 |
/// |
116 | 116 |
/// \warning This implementation cannot handle infinite or very large |
117 | 117 |
/// capacities (e.g. the maximum value of \c CAP::Value). |
118 | 118 |
/// |
119 | 119 |
/// \tparam GR The type of the digraph the algorithm runs on. |
120 | 120 |
/// \tparam CAP The type of the capacity map. The default map |
121 | 121 |
/// type is \ref concepts::Digraph::ArcMap "GR::ArcMap<int>". |
122 | 122 |
/// \tparam TR The traits class that defines various types used by the |
123 | 123 |
/// algorithm. By default, it is \ref PreflowDefaultTraits |
124 | 124 |
/// "PreflowDefaultTraits<GR, CAP>". |
125 | 125 |
/// In most cases, this parameter should not be set directly, |
126 | 126 |
/// consider to use the named template parameters instead. |
127 | 127 |
#ifdef DOXYGEN |
128 | 128 |
template <typename GR, typename CAP, typename TR> |
129 | 129 |
#else |
130 | 130 |
template <typename GR, |
131 | 131 |
typename CAP = typename GR::template ArcMap<int>, |
132 | 132 |
typename TR = PreflowDefaultTraits<GR, CAP> > |
133 | 133 |
#endif |
134 | 134 |
class Preflow { |
135 | 135 |
public: |
136 | 136 |
|
137 | 137 |
///The \ref PreflowDefaultTraits "traits class" of the algorithm. |
138 | 138 |
typedef TR Traits; |
139 | 139 |
///The type of the digraph the algorithm runs on. |
140 | 140 |
typedef typename Traits::Digraph Digraph; |
141 | 141 |
///The type of the capacity map. |
142 | 142 |
typedef typename Traits::CapacityMap CapacityMap; |
143 | 143 |
///The type of the flow values. |
144 | 144 |
typedef typename Traits::Value Value; |
145 | 145 |
|
146 | 146 |
///The type of the flow map. |
147 | 147 |
typedef typename Traits::FlowMap FlowMap; |
148 | 148 |
///The type of the elevator. |
149 | 149 |
typedef typename Traits::Elevator Elevator; |
150 | 150 |
///The type of the tolerance. |
151 | 151 |
typedef typename Traits::Tolerance Tolerance; |
152 | 152 |
|
153 | 153 |
private: |
154 | 154 |
|
155 | 155 |
TEMPLATE_DIGRAPH_TYPEDEFS(Digraph); |
156 | 156 |
|
157 | 157 |
const Digraph& _graph; |
158 | 158 |
const CapacityMap* _capacity; |
159 | 159 |
|
160 | 160 |
int _node_num; |
161 | 161 |
|
162 | 162 |
Node _source, _target; |
163 | 163 |
|
164 | 164 |
FlowMap* _flow; |
165 | 165 |
bool _local_flow; |
166 | 166 |
|
167 | 167 |
Elevator* _level; |
168 | 168 |
bool _local_level; |
169 | 169 |
|
170 | 170 |
typedef typename Digraph::template NodeMap<Value> ExcessMap; |
171 | 171 |
ExcessMap* _excess; |
172 | 172 |
|
173 | 173 |
Tolerance _tolerance; |
174 | 174 |
|
175 | 175 |
bool _phase; |
176 | 176 |
|
177 | 177 |
|
178 | 178 |
void createStructures() { |
179 | 179 |
_node_num = countNodes(_graph); |
180 | 180 |
|
181 | 181 |
if (!_flow) { |
182 | 182 |
_flow = Traits::createFlowMap(_graph); |
183 | 183 |
_local_flow = true; |
184 | 184 |
} |
185 | 185 |
if (!_level) { |
186 | 186 |
_level = Traits::createElevator(_graph, _node_num); |
187 | 187 |
_local_level = true; |
188 | 188 |
} |
189 | 189 |
if (!_excess) { |
190 | 190 |
_excess = new ExcessMap(_graph); |
191 | 191 |
} |
192 | 192 |
} |
193 | 193 |
|
194 | 194 |
void destroyStructures() { |
195 | 195 |
if (_local_flow) { |
196 | 196 |
delete _flow; |
197 | 197 |
} |
198 | 198 |
if (_local_level) { |
199 | 199 |
delete _level; |
200 | 200 |
} |
201 | 201 |
if (_excess) { |
202 | 202 |
delete _excess; |
203 | 203 |
} |
204 | 204 |
} |
205 | 205 |
|
206 | 206 |
public: |
207 | 207 |
|
208 | 208 |
typedef Preflow Create; |
209 | 209 |
|
210 | 210 |
///\name Named Template Parameters |
211 | 211 |
|
212 | 212 |
///@{ |
213 | 213 |
|
214 | 214 |
template <typename T> |
215 | 215 |
struct SetFlowMapTraits : public Traits { |
216 | 216 |
typedef T FlowMap; |
217 | 217 |
static FlowMap *createFlowMap(const Digraph&) { |
218 | 218 |
LEMON_ASSERT(false, "FlowMap is not initialized"); |
219 | 219 |
return 0; // ignore warnings |
220 | 220 |
} |
221 | 221 |
}; |
222 | 222 |
|
223 | 223 |
/// \brief \ref named-templ-param "Named parameter" for setting |
224 | 224 |
/// FlowMap type |
225 | 225 |
/// |
226 | 226 |
/// \ref named-templ-param "Named parameter" for setting FlowMap |
227 | 227 |
/// type. |
228 | 228 |
template <typename T> |
229 | 229 |
struct SetFlowMap |
230 | 230 |
: public Preflow<Digraph, CapacityMap, SetFlowMapTraits<T> > { |
231 | 231 |
typedef Preflow<Digraph, CapacityMap, |
232 | 232 |
SetFlowMapTraits<T> > Create; |
233 | 233 |
}; |
234 | 234 |
|
235 | 235 |
template <typename T> |
236 | 236 |
struct SetElevatorTraits : public Traits { |
237 | 237 |
typedef T Elevator; |
238 | 238 |
static Elevator *createElevator(const Digraph&, int) { |
239 | 239 |
LEMON_ASSERT(false, "Elevator is not initialized"); |
240 | 240 |
return 0; // ignore warnings |
241 | 241 |
} |
242 | 242 |
}; |
243 | 243 |
|
244 | 244 |
/// \brief \ref named-templ-param "Named parameter" for setting |
245 | 245 |
/// Elevator type |
246 | 246 |
/// |
247 | 247 |
/// \ref named-templ-param "Named parameter" for setting Elevator |
248 | 248 |
/// type. If this named parameter is used, then an external |
249 | 249 |
/// elevator object must be passed to the algorithm using the |
250 | 250 |
/// \ref elevator(Elevator&) "elevator()" function before calling |
251 | 251 |
/// \ref run() or \ref init(). |
252 | 252 |
/// \sa SetStandardElevator |
253 | 253 |
template <typename T> |
254 | 254 |
struct SetElevator |
255 | 255 |
: public Preflow<Digraph, CapacityMap, SetElevatorTraits<T> > { |
256 | 256 |
typedef Preflow<Digraph, CapacityMap, |
257 | 257 |
SetElevatorTraits<T> > Create; |
258 | 258 |
}; |
259 | 259 |
|
260 | 260 |
template <typename T> |
261 | 261 |
struct SetStandardElevatorTraits : public Traits { |
262 | 262 |
typedef T Elevator; |
263 | 263 |
static Elevator *createElevator(const Digraph& digraph, int max_level) { |
264 | 264 |
return new Elevator(digraph, max_level); |
265 | 265 |
} |
266 | 266 |
}; |
267 | 267 |
|
268 | 268 |
/// \brief \ref named-templ-param "Named parameter" for setting |
269 | 269 |
/// Elevator type with automatic allocation |
270 | 270 |
/// |
271 | 271 |
/// \ref named-templ-param "Named parameter" for setting Elevator |
272 | 272 |
/// type with automatic allocation. |
273 | 273 |
/// The Elevator should have standard constructor interface to be |
274 | 274 |
/// able to automatically created by the algorithm (i.e. the |
275 | 275 |
/// digraph and the maximum level should be passed to it). |
276 | 276 |
/// However, an external elevator object could also be passed to the |
277 | 277 |
/// algorithm with the \ref elevator(Elevator&) "elevator()" function |
278 | 278 |
/// before calling \ref run() or \ref init(). |
279 | 279 |
/// \sa SetElevator |
280 | 280 |
template <typename T> |
281 | 281 |
struct SetStandardElevator |
282 | 282 |
: public Preflow<Digraph, CapacityMap, |
283 | 283 |
SetStandardElevatorTraits<T> > { |
284 | 284 |
typedef Preflow<Digraph, CapacityMap, |
285 | 285 |
SetStandardElevatorTraits<T> > Create; |
286 | 286 |
}; |
287 | 287 |
|
288 | 288 |
/// @} |
289 | 289 |
|
290 | 290 |
protected: |
291 | 291 |
|
292 | 292 |
Preflow() {} |
293 | 293 |
|
294 | 294 |
public: |
295 | 295 |
|
296 | 296 |
|
297 | 297 |
/// \brief The constructor of the class. |
298 | 298 |
/// |
299 | 299 |
/// The constructor of the class. |
300 | 300 |
/// \param digraph The digraph the algorithm runs on. |
301 | 301 |
/// \param capacity The capacity of the arcs. |
302 | 302 |
/// \param source The source node. |
303 | 303 |
/// \param target The target node. |
304 | 304 |
Preflow(const Digraph& digraph, const CapacityMap& capacity, |
305 | 305 |
Node source, Node target) |
306 | 306 |
: _graph(digraph), _capacity(&capacity), |
307 | 307 |
_node_num(0), _source(source), _target(target), |
308 | 308 |
_flow(0), _local_flow(false), |
309 | 309 |
_level(0), _local_level(false), |
310 | 310 |
_excess(0), _tolerance(), _phase() {} |
311 | 311 |
|
312 | 312 |
/// \brief Destructor. |
313 | 313 |
/// |
314 | 314 |
/// Destructor. |
315 | 315 |
~Preflow() { |
316 | 316 |
destroyStructures(); |
317 | 317 |
} |
318 | 318 |
|
319 | 319 |
/// \brief Sets the capacity map. |
320 | 320 |
/// |
321 | 321 |
/// Sets the capacity map. |
322 | 322 |
/// \return <tt>(*this)</tt> |
323 | 323 |
Preflow& capacityMap(const CapacityMap& map) { |
324 | 324 |
_capacity = ↦ |
325 | 325 |
return *this; |
326 | 326 |
} |
327 | 327 |
|
328 | 328 |
/// \brief Sets the flow map. |
329 | 329 |
/// |
330 | 330 |
/// Sets the flow map. |
331 | 331 |
/// If you don't use this function before calling \ref run() or |
332 | 332 |
/// \ref init(), an instance will be allocated automatically. |
333 | 333 |
/// The destructor deallocates this automatically allocated map, |
334 | 334 |
/// of course. |
335 | 335 |
/// \return <tt>(*this)</tt> |
336 | 336 |
Preflow& flowMap(FlowMap& map) { |
337 | 337 |
if (_local_flow) { |
338 | 338 |
delete _flow; |
339 | 339 |
_local_flow = false; |
340 | 340 |
} |
341 | 341 |
_flow = ↦ |
342 | 342 |
return *this; |
343 | 343 |
} |
344 | 344 |
|
345 | 345 |
/// \brief Sets the source node. |
346 | 346 |
/// |
347 | 347 |
/// Sets the source node. |
348 | 348 |
/// \return <tt>(*this)</tt> |
349 | 349 |
Preflow& source(const Node& node) { |
350 | 350 |
_source = node; |
351 | 351 |
return *this; |
352 | 352 |
} |
353 | 353 |
|
354 | 354 |
/// \brief Sets the target node. |
355 | 355 |
/// |
356 | 356 |
/// Sets the target node. |
357 | 357 |
/// \return <tt>(*this)</tt> |
358 | 358 |
Preflow& target(const Node& node) { |
359 | 359 |
_target = node; |
360 | 360 |
return *this; |
361 | 361 |
} |
362 | 362 |
|
363 | 363 |
/// \brief Sets the elevator used by algorithm. |
364 | 364 |
/// |
365 | 365 |
/// Sets the elevator used by algorithm. |
366 | 366 |
/// If you don't use this function before calling \ref run() or |
367 | 367 |
/// \ref init(), an instance will be allocated automatically. |
368 | 368 |
/// The destructor deallocates this automatically allocated elevator, |
369 | 369 |
/// of course. |
370 | 370 |
/// \return <tt>(*this)</tt> |
371 | 371 |
Preflow& elevator(Elevator& elevator) { |
372 | 372 |
if (_local_level) { |
373 | 373 |
delete _level; |
374 | 374 |
_local_level = false; |
375 | 375 |
} |
376 | 376 |
_level = &elevator; |
377 | 377 |
return *this; |
378 | 378 |
} |
379 | 379 |
|
380 | 380 |
/// \brief Returns a const reference to the elevator. |
381 | 381 |
/// |
382 | 382 |
/// Returns a const reference to the elevator. |
383 | 383 |
/// |
384 | 384 |
/// \pre Either \ref run() or \ref init() must be called before |
385 | 385 |
/// using this function. |
386 | 386 |
const Elevator& elevator() const { |
387 | 387 |
return *_level; |
388 | 388 |
} |
389 | 389 |
|
390 | 390 |
/// \brief Sets the tolerance used by the algorithm. |
391 | 391 |
/// |
392 | 392 |
/// Sets the tolerance object used by the algorithm. |
393 | 393 |
/// \return <tt>(*this)</tt> |
394 | 394 |
Preflow& tolerance(const Tolerance& tolerance) { |
395 | 395 |
_tolerance = tolerance; |
396 | 396 |
return *this; |
397 | 397 |
} |
398 | 398 |
|
399 | 399 |
/// \brief Returns a const reference to the tolerance. |
400 | 400 |
/// |
401 | 401 |
/// Returns a const reference to the tolerance object used by |
402 | 402 |
/// the algorithm. |
403 | 403 |
const Tolerance& tolerance() const { |
404 | 404 |
return _tolerance; |
405 | 405 |
} |
406 | 406 |
|
407 | 407 |
/// \name Execution Control |
408 | 408 |
/// The simplest way to execute the preflow algorithm is to use |
409 | 409 |
/// \ref run() or \ref runMinCut().\n |
410 | 410 |
/// If you need better control on the initial solution or the execution, |
411 | 411 |
/// you have to call one of the \ref init() functions first, then |
412 | 412 |
/// \ref startFirstPhase() and if you need it \ref startSecondPhase(). |
413 | 413 |
|
414 | 414 |
///@{ |
415 | 415 |
|
416 | 416 |
/// \brief Initializes the internal data structures. |
417 | 417 |
/// |
418 | 418 |
/// Initializes the internal data structures and sets the initial |
419 | 419 |
/// flow to zero on each arc. |
420 | 420 |
void init() { |
421 | 421 |
createStructures(); |
422 | 422 |
|
423 | 423 |
_phase = true; |
424 | 424 |
for (NodeIt n(_graph); n != INVALID; ++n) { |
425 | 425 |
(*_excess)[n] = 0; |
426 | 426 |
} |
427 | 427 |
|
428 | 428 |
for (ArcIt e(_graph); e != INVALID; ++e) { |
429 | 429 |
_flow->set(e, 0); |
430 | 430 |
} |
431 | 431 |
|
432 | 432 |
typename Digraph::template NodeMap<bool> reached(_graph, false); |
433 | 433 |
|
434 | 434 |
_level->initStart(); |
435 | 435 |
_level->initAddItem(_target); |
436 | 436 |
|
437 | 437 |
std::vector<Node> queue; |
438 | 438 |
reached[_source] = true; |
439 | 439 |
|
440 | 440 |
queue.push_back(_target); |
441 | 441 |
reached[_target] = true; |
442 | 442 |
while (!queue.empty()) { |
443 | 443 |
_level->initNewLevel(); |
444 | 444 |
std::vector<Node> nqueue; |
445 | 445 |
for (int i = 0; i < int(queue.size()); ++i) { |
446 | 446 |
Node n = queue[i]; |
447 | 447 |
for (InArcIt e(_graph, n); e != INVALID; ++e) { |
448 | 448 |
Node u = _graph.source(e); |
449 | 449 |
if (!reached[u] && _tolerance.positive((*_capacity)[e])) { |
450 | 450 |
reached[u] = true; |
451 | 451 |
_level->initAddItem(u); |
452 | 452 |
nqueue.push_back(u); |
453 | 453 |
} |
454 | 454 |
} |
455 | 455 |
} |
456 | 456 |
queue.swap(nqueue); |
457 | 457 |
} |
458 | 458 |
_level->initFinish(); |
459 | 459 |
|
460 | 460 |
for (OutArcIt e(_graph, _source); e != INVALID; ++e) { |
461 | 461 |
if (_tolerance.positive((*_capacity)[e])) { |
462 | 462 |
Node u = _graph.target(e); |
463 | 463 |
if ((*_level)[u] == _level->maxLevel()) continue; |
464 | 464 |
_flow->set(e, (*_capacity)[e]); |
465 | 465 |
(*_excess)[u] += (*_capacity)[e]; |
466 | 466 |
if (u != _target && !_level->active(u)) { |
467 | 467 |
_level->activate(u); |
468 | 468 |
} |
469 | 469 |
} |
470 | 470 |
} |
471 | 471 |
} |
472 | 472 |
|
473 | 473 |
/// \brief Initializes the internal data structures using the |
474 | 474 |
/// given flow map. |
475 | 475 |
/// |
476 | 476 |
/// Initializes the internal data structures and sets the initial |
477 | 477 |
/// flow to the given \c flowMap. The \c flowMap should contain a |
478 | 478 |
/// flow or at least a preflow, i.e. at each node excluding the |
479 | 479 |
/// source node the incoming flow should greater or equal to the |
480 | 480 |
/// outgoing flow. |
481 | 481 |
/// \return \c false if the given \c flowMap is not a preflow. |
482 | 482 |
template <typename FlowMap> |
483 | 483 |
bool init(const FlowMap& flowMap) { |
484 | 484 |
createStructures(); |
485 | 485 |
|
486 | 486 |
for (ArcIt e(_graph); e != INVALID; ++e) { |
487 | 487 |
_flow->set(e, flowMap[e]); |
488 | 488 |
} |
489 | 489 |
|
490 | 490 |
for (NodeIt n(_graph); n != INVALID; ++n) { |
491 | 491 |
Value excess = 0; |
492 | 492 |
for (InArcIt e(_graph, n); e != INVALID; ++e) { |
493 | 493 |
excess += (*_flow)[e]; |
494 | 494 |
} |
495 | 495 |
for (OutArcIt e(_graph, n); e != INVALID; ++e) { |
496 | 496 |
excess -= (*_flow)[e]; |
497 | 497 |
} |
498 | 498 |
if (excess < 0 && n != _source) return false; |
499 | 499 |
(*_excess)[n] = excess; |
500 | 500 |
} |
501 | 501 |
|
502 | 502 |
typename Digraph::template NodeMap<bool> reached(_graph, false); |
503 | 503 |
|
504 | 504 |
_level->initStart(); |
505 | 505 |
_level->initAddItem(_target); |
506 | 506 |
|
507 | 507 |
std::vector<Node> queue; |
508 | 508 |
reached[_source] = true; |
509 | 509 |
|
510 | 510 |
queue.push_back(_target); |
511 | 511 |
reached[_target] = true; |
512 | 512 |
while (!queue.empty()) { |
513 | 513 |
_level->initNewLevel(); |
514 | 514 |
std::vector<Node> nqueue; |
515 | 515 |
for (int i = 0; i < int(queue.size()); ++i) { |
516 | 516 |
Node n = queue[i]; |
517 | 517 |
for (InArcIt e(_graph, n); e != INVALID; ++e) { |
1 | 1 |
/* -*- mode: C++; indent-tabs-mode: nil; -*- |
2 | 2 |
* |
3 | 3 |
* This file is a part of LEMON, a generic C++ optimization library. |
4 | 4 |
* |
5 |
* Copyright (C) 2003- |
|
5 |
* Copyright (C) 2003-2011 |
|
6 | 6 |
* Egervary Jeno Kombinatorikus Optimalizalasi Kutatocsoport |
7 | 7 |
* (Egervary Research Group on Combinatorial Optimization, EGRES). |
8 | 8 |
* |
9 | 9 |
* Permission to use, modify and distribute this software is granted |
10 | 10 |
* provided that this copyright notice appears in all copies. For |
11 | 11 |
* precise terms see the accompanying LICENSE file. |
12 | 12 |
* |
13 | 13 |
* This software is provided "AS IS" with no warranty of any kind, |
14 | 14 |
* express or implied, and with no claim as to its suitability for any |
15 | 15 |
* purpose. |
16 | 16 |
* |
17 | 17 |
*/ |
18 | 18 |
|
19 | 19 |
#include <lemon/concepts/digraph.h> |
20 | 20 |
#include <lemon/smart_graph.h> |
21 | 21 |
#include <lemon/list_graph.h> |
22 | 22 |
#include <lemon/lgf_reader.h> |
23 | 23 |
#include <lemon/dfs.h> |
24 | 24 |
#include <lemon/path.h> |
25 | 25 |
|
26 | 26 |
#include "graph_test.h" |
27 | 27 |
#include "test_tools.h" |
28 | 28 |
|
29 | 29 |
using namespace lemon; |
30 | 30 |
|
31 | 31 |
char test_lgf[] = |
32 | 32 |
"@nodes\n" |
33 | 33 |
"label\n" |
34 | 34 |
"0\n" |
35 | 35 |
"1\n" |
36 | 36 |
"2\n" |
37 | 37 |
"3\n" |
38 | 38 |
"4\n" |
39 | 39 |
"5\n" |
40 | 40 |
"6\n" |
41 | 41 |
"@arcs\n" |
42 | 42 |
" label\n" |
43 | 43 |
"0 1 0\n" |
44 | 44 |
"1 2 1\n" |
45 | 45 |
"2 3 2\n" |
46 | 46 |
"1 4 3\n" |
47 | 47 |
"4 2 4\n" |
48 | 48 |
"4 5 5\n" |
49 | 49 |
"5 0 6\n" |
50 | 50 |
"6 3 7\n" |
51 | 51 |
"@attributes\n" |
52 | 52 |
"source 0\n" |
53 | 53 |
"target 5\n" |
54 | 54 |
"source1 6\n" |
55 | 55 |
"target1 3\n"; |
56 | 56 |
|
57 | 57 |
|
58 | 58 |
void checkDfsCompile() |
59 | 59 |
{ |
60 | 60 |
typedef concepts::Digraph Digraph; |
61 | 61 |
typedef Dfs<Digraph> DType; |
62 | 62 |
typedef Digraph::Node Node; |
63 | 63 |
typedef Digraph::Arc Arc; |
64 | 64 |
|
65 | 65 |
Digraph G; |
66 | 66 |
Node s, t; |
67 | 67 |
Arc e; |
68 | 68 |
int l, i; |
69 | 69 |
bool b; |
70 | 70 |
DType::DistMap d(G); |
71 | 71 |
DType::PredMap p(G); |
72 | 72 |
Path<Digraph> pp; |
73 | 73 |
concepts::ReadMap<Arc,bool> am; |
74 | 74 |
|
75 | 75 |
{ |
76 | 76 |
DType dfs_test(G); |
77 | 77 |
const DType& const_dfs_test = dfs_test; |
78 | 78 |
|
79 | 79 |
dfs_test.run(s); |
80 | 80 |
dfs_test.run(s,t); |
81 | 81 |
dfs_test.run(); |
82 | 82 |
|
83 | 83 |
dfs_test.init(); |
84 | 84 |
dfs_test.addSource(s); |
85 | 85 |
e = dfs_test.processNextArc(); |
86 | 86 |
e = const_dfs_test.nextArc(); |
87 | 87 |
b = const_dfs_test.emptyQueue(); |
88 | 88 |
i = const_dfs_test.queueSize(); |
89 | 89 |
|
90 | 90 |
dfs_test.start(); |
91 | 91 |
dfs_test.start(t); |
92 | 92 |
dfs_test.start(am); |
93 | 93 |
|
94 | 94 |
l = const_dfs_test.dist(t); |
95 | 95 |
e = const_dfs_test.predArc(t); |
96 | 96 |
s = const_dfs_test.predNode(t); |
97 | 97 |
b = const_dfs_test.reached(t); |
98 | 98 |
d = const_dfs_test.distMap(); |
99 | 99 |
p = const_dfs_test.predMap(); |
100 | 100 |
pp = const_dfs_test.path(t); |
101 | 101 |
} |
102 | 102 |
{ |
103 | 103 |
DType |
104 | 104 |
::SetPredMap<concepts::ReadWriteMap<Node,Arc> > |
105 | 105 |
::SetDistMap<concepts::ReadWriteMap<Node,int> > |
106 | 106 |
::SetReachedMap<concepts::ReadWriteMap<Node,bool> > |
107 | 107 |
::SetStandardProcessedMap |
108 | 108 |
::SetProcessedMap<concepts::WriteMap<Node,bool> > |
109 | 109 |
::Create dfs_test(G); |
110 | 110 |
|
111 | 111 |
concepts::ReadWriteMap<Node,Arc> pred_map; |
112 | 112 |
concepts::ReadWriteMap<Node,int> dist_map; |
113 | 113 |
concepts::ReadWriteMap<Node,bool> reached_map; |
114 | 114 |
concepts::WriteMap<Node,bool> processed_map; |
115 | 115 |
|
116 | 116 |
dfs_test |
117 | 117 |
.predMap(pred_map) |
118 | 118 |
.distMap(dist_map) |
119 | 119 |
.reachedMap(reached_map) |
120 | 120 |
.processedMap(processed_map); |
121 | 121 |
|
122 | 122 |
dfs_test.run(s); |
123 | 123 |
dfs_test.run(s,t); |
124 | 124 |
dfs_test.run(); |
125 | 125 |
dfs_test.init(); |
126 | 126 |
|
127 | 127 |
dfs_test.addSource(s); |
128 | 128 |
e = dfs_test.processNextArc(); |
129 | 129 |
e = dfs_test.nextArc(); |
130 | 130 |
b = dfs_test.emptyQueue(); |
131 | 131 |
i = dfs_test.queueSize(); |
132 | 132 |
|
133 | 133 |
dfs_test.start(); |
134 | 134 |
dfs_test.start(t); |
135 | 135 |
dfs_test.start(am); |
136 | 136 |
|
137 | 137 |
l = dfs_test.dist(t); |
138 | 138 |
e = dfs_test.predArc(t); |
139 | 139 |
s = dfs_test.predNode(t); |
140 | 140 |
b = dfs_test.reached(t); |
141 | 141 |
pp = dfs_test.path(t); |
142 | 142 |
} |
143 | 143 |
} |
144 | 144 |
|
145 | 145 |
void checkDfsFunctionCompile() |
146 | 146 |
{ |
147 | 147 |
typedef int VType; |
148 | 148 |
typedef concepts::Digraph Digraph; |
149 | 149 |
typedef Digraph::Arc Arc; |
150 | 150 |
typedef Digraph::Node Node; |
151 | 151 |
|
152 | 152 |
Digraph g; |
153 | 153 |
bool b; |
154 | 154 |
dfs(g).run(Node()); |
155 | 155 |
b=dfs(g).run(Node(),Node()); |
156 | 156 |
dfs(g).run(); |
157 | 157 |
dfs(g) |
158 | 158 |
.predMap(concepts::ReadWriteMap<Node,Arc>()) |
159 | 159 |
.distMap(concepts::ReadWriteMap<Node,VType>()) |
160 | 160 |
.reachedMap(concepts::ReadWriteMap<Node,bool>()) |
161 | 161 |
.processedMap(concepts::WriteMap<Node,bool>()) |
162 | 162 |
.run(Node()); |
163 | 163 |
b=dfs(g) |
164 | 164 |
.predMap(concepts::ReadWriteMap<Node,Arc>()) |
165 | 165 |
.distMap(concepts::ReadWriteMap<Node,VType>()) |
166 | 166 |
.reachedMap(concepts::ReadWriteMap<Node,bool>()) |
167 | 167 |
.processedMap(concepts::WriteMap<Node,bool>()) |
168 | 168 |
.path(concepts::Path<Digraph>()) |
169 | 169 |
.dist(VType()) |
170 | 170 |
.run(Node(),Node()); |
171 | 171 |
dfs(g) |
172 | 172 |
.predMap(concepts::ReadWriteMap<Node,Arc>()) |
173 | 173 |
.distMap(concepts::ReadWriteMap<Node,VType>()) |
174 | 174 |
.reachedMap(concepts::ReadWriteMap<Node,bool>()) |
175 | 175 |
.processedMap(concepts::WriteMap<Node,bool>()) |
176 | 176 |
.run(); |
177 | 177 |
} |
178 | 178 |
|
179 | 179 |
template <class Digraph> |
180 | 180 |
void checkDfs() { |
181 | 181 |
TEMPLATE_DIGRAPH_TYPEDEFS(Digraph); |
182 | 182 |
|
183 | 183 |
Digraph G; |
184 | 184 |
Node s, t; |
185 | 185 |
Node s1, t1; |
186 | 186 |
|
187 | 187 |
std::istringstream input(test_lgf); |
188 | 188 |
digraphReader(G, input). |
189 | 189 |
node("source", s). |
190 | 190 |
node("target", t). |
191 | 191 |
node("source1", s1). |
192 | 192 |
node("target1", t1). |
193 | 193 |
run(); |
194 | 194 |
|
195 | 195 |
Dfs<Digraph> dfs_test(G); |
196 | 196 |
dfs_test.run(s); |
197 | 197 |
|
198 | 198 |
Path<Digraph> p = dfs_test.path(t); |
199 | 199 |
check(p.length() == dfs_test.dist(t),"path() found a wrong path."); |
200 | 200 |
check(checkPath(G, p),"path() found a wrong path."); |
201 | 201 |
check(pathSource(G, p) == s,"path() found a wrong path."); |
202 | 202 |
check(pathTarget(G, p) == t,"path() found a wrong path."); |
203 | 203 |
|
204 | 204 |
for(NodeIt v(G); v!=INVALID; ++v) { |
205 | 205 |
if (dfs_test.reached(v)) { |
206 | 206 |
check(v==s || dfs_test.predArc(v)!=INVALID, "Wrong tree."); |
207 | 207 |
if (dfs_test.predArc(v)!=INVALID ) { |
208 | 208 |
Arc e=dfs_test.predArc(v); |
209 | 209 |
Node u=G.source(e); |
210 | 210 |
check(u==dfs_test.predNode(v),"Wrong tree."); |
211 | 211 |
check(dfs_test.dist(v) - dfs_test.dist(u) == 1, |
212 | 212 |
"Wrong distance. (" << dfs_test.dist(u) << "->" |
213 | 213 |
<< dfs_test.dist(v) << ")"); |
214 | 214 |
} |
215 | 215 |
} |
216 | 216 |
} |
217 | 217 |
|
218 | 218 |
{ |
219 | 219 |
Dfs<Digraph> dfs(G); |
220 | 220 |
check(dfs.run(s1,t1) && dfs.reached(t1),"Node 3 is reachable from Node 6."); |
221 | 221 |
} |
222 | 222 |
|
223 | 223 |
{ |
224 | 224 |
NullMap<Node,Arc> myPredMap; |
225 | 225 |
dfs(G).predMap(myPredMap).run(s); |
226 | 226 |
} |
227 | 227 |
} |
228 | 228 |
|
229 | 229 |
int main() |
230 | 230 |
{ |
231 | 231 |
checkDfs<ListDigraph>(); |
232 | 232 |
checkDfs<SmartDigraph>(); |
233 | 233 |
return 0; |
234 | 234 |
} |
1 | 1 |
/* -*- mode: C++; indent-tabs-mode: nil; -*- |
2 | 2 |
* |
3 | 3 |
* This file is a part of LEMON, a generic C++ optimization library. |
4 | 4 |
* |
5 |
* Copyright (C) 2003- |
|
5 |
* Copyright (C) 2003-2011 |
|
6 | 6 |
* Egervary Jeno Kombinatorikus Optimalizalasi Kutatocsoport |
7 | 7 |
* (Egervary Research Group on Combinatorial Optimization, EGRES). |
8 | 8 |
* |
9 | 9 |
* Permission to use, modify and distribute this software is granted |
10 | 10 |
* provided that this copyright notice appears in all copies. For |
11 | 11 |
* precise terms see the accompanying LICENSE file. |
12 | 12 |
* |
13 | 13 |
* This software is provided "AS IS" with no warranty of any kind, |
14 | 14 |
* express or implied, and with no claim as to its suitability for any |
15 | 15 |
* purpose. |
16 | 16 |
* |
17 | 17 |
*/ |
18 | 18 |
|
19 | 19 |
#include <lemon/smart_graph.h> |
20 | 20 |
#include <lemon/list_graph.h> |
21 | 21 |
#include <lemon/lgf_reader.h> |
22 | 22 |
#include <lemon/error.h> |
23 | 23 |
|
24 | 24 |
#include "test_tools.h" |
25 | 25 |
|
26 | 26 |
using namespace std; |
27 | 27 |
using namespace lemon; |
28 | 28 |
|
29 | 29 |
void digraph_copy_test() { |
30 | 30 |
const int nn = 10; |
31 | 31 |
|
32 | 32 |
// Build a digraph |
33 | 33 |
SmartDigraph from; |
34 | 34 |
SmartDigraph::NodeMap<int> fnm(from); |
35 | 35 |
SmartDigraph::ArcMap<int> fam(from); |
36 | 36 |
SmartDigraph::Node fn = INVALID; |
37 | 37 |
SmartDigraph::Arc fa = INVALID; |
38 | 38 |
|
39 | 39 |
std::vector<SmartDigraph::Node> fnv; |
40 | 40 |
for (int i = 0; i < nn; ++i) { |
41 | 41 |
SmartDigraph::Node node = from.addNode(); |
42 | 42 |
fnv.push_back(node); |
43 | 43 |
fnm[node] = i * i; |
44 | 44 |
if (i == 0) fn = node; |
45 | 45 |
} |
46 | 46 |
|
47 | 47 |
for (int i = 0; i < nn; ++i) { |
48 | 48 |
for (int j = 0; j < nn; ++j) { |
49 | 49 |
SmartDigraph::Arc arc = from.addArc(fnv[i], fnv[j]); |
50 | 50 |
fam[arc] = i + j * j; |
51 | 51 |
if (i == 0 && j == 0) fa = arc; |
52 | 52 |
} |
53 | 53 |
} |
54 | 54 |
|
55 | 55 |
// Test digraph copy |
56 | 56 |
ListDigraph to; |
57 | 57 |
ListDigraph::NodeMap<int> tnm(to); |
58 | 58 |
ListDigraph::ArcMap<int> tam(to); |
59 | 59 |
ListDigraph::Node tn; |
60 | 60 |
ListDigraph::Arc ta; |
61 | 61 |
|
62 | 62 |
SmartDigraph::NodeMap<ListDigraph::Node> nr(from); |
63 | 63 |
SmartDigraph::ArcMap<ListDigraph::Arc> er(from); |
64 | 64 |
|
65 | 65 |
ListDigraph::NodeMap<SmartDigraph::Node> ncr(to); |
66 | 66 |
ListDigraph::ArcMap<SmartDigraph::Arc> ecr(to); |
67 | 67 |
|
68 | 68 |
digraphCopy(from, to). |
69 | 69 |
nodeMap(fnm, tnm).arcMap(fam, tam). |
70 | 70 |
nodeRef(nr).arcRef(er). |
71 | 71 |
nodeCrossRef(ncr).arcCrossRef(ecr). |
72 | 72 |
node(fn, tn).arc(fa, ta).run(); |
73 | 73 |
|
74 | 74 |
check(countNodes(from) == countNodes(to), "Wrong copy."); |
75 | 75 |
check(countArcs(from) == countArcs(to), "Wrong copy."); |
76 | 76 |
|
77 | 77 |
for (SmartDigraph::NodeIt it(from); it != INVALID; ++it) { |
78 | 78 |
check(ncr[nr[it]] == it, "Wrong copy."); |
79 | 79 |
check(fnm[it] == tnm[nr[it]], "Wrong copy."); |
80 | 80 |
} |
81 | 81 |
|
82 | 82 |
for (SmartDigraph::ArcIt it(from); it != INVALID; ++it) { |
83 | 83 |
check(ecr[er[it]] == it, "Wrong copy."); |
84 | 84 |
check(fam[it] == tam[er[it]], "Wrong copy."); |
85 | 85 |
check(nr[from.source(it)] == to.source(er[it]), "Wrong copy."); |
86 | 86 |
check(nr[from.target(it)] == to.target(er[it]), "Wrong copy."); |
87 | 87 |
} |
88 | 88 |
|
89 | 89 |
for (ListDigraph::NodeIt it(to); it != INVALID; ++it) { |
90 | 90 |
check(nr[ncr[it]] == it, "Wrong copy."); |
91 | 91 |
} |
92 | 92 |
|
93 | 93 |
for (ListDigraph::ArcIt it(to); it != INVALID; ++it) { |
94 | 94 |
check(er[ecr[it]] == it, "Wrong copy."); |
95 | 95 |
} |
96 | 96 |
check(tn == nr[fn], "Wrong copy."); |
97 | 97 |
check(ta == er[fa], "Wrong copy."); |
98 | 98 |
|
99 | 99 |
// Test repeated copy |
100 | 100 |
digraphCopy(from, to).run(); |
101 | 101 |
|
102 | 102 |
check(countNodes(from) == countNodes(to), "Wrong copy."); |
103 | 103 |
check(countArcs(from) == countArcs(to), "Wrong copy."); |
104 | 104 |
} |
105 | 105 |
|
106 | 106 |
void graph_copy_test() { |
107 | 107 |
const int nn = 10; |
108 | 108 |
|
109 | 109 |
// Build a graph |
110 | 110 |
SmartGraph from; |
111 | 111 |
SmartGraph::NodeMap<int> fnm(from); |
112 | 112 |
SmartGraph::ArcMap<int> fam(from); |
113 | 113 |
SmartGraph::EdgeMap<int> fem(from); |
114 | 114 |
SmartGraph::Node fn = INVALID; |
115 | 115 |
SmartGraph::Arc fa = INVALID; |
116 | 116 |
SmartGraph::Edge fe = INVALID; |
117 | 117 |
|
118 | 118 |
std::vector<SmartGraph::Node> fnv; |
119 | 119 |
for (int i = 0; i < nn; ++i) { |
120 | 120 |
SmartGraph::Node node = from.addNode(); |
121 | 121 |
fnv.push_back(node); |
122 | 122 |
fnm[node] = i * i; |
123 | 123 |
if (i == 0) fn = node; |
124 | 124 |
} |
125 | 125 |
|
126 | 126 |
for (int i = 0; i < nn; ++i) { |
127 | 127 |
for (int j = 0; j < nn; ++j) { |
128 | 128 |
SmartGraph::Edge edge = from.addEdge(fnv[i], fnv[j]); |
129 | 129 |
fem[edge] = i * i + j * j; |
130 | 130 |
fam[from.direct(edge, true)] = i + j * j; |
131 | 131 |
fam[from.direct(edge, false)] = i * i + j; |
132 | 132 |
if (i == 0 && j == 0) fa = from.direct(edge, true); |
133 | 133 |
if (i == 0 && j == 0) fe = edge; |
134 | 134 |
} |
135 | 135 |
} |
136 | 136 |
|
137 | 137 |
// Test graph copy |
138 | 138 |
ListGraph to; |
139 | 139 |
ListGraph::NodeMap<int> tnm(to); |
140 | 140 |
ListGraph::ArcMap<int> tam(to); |
141 | 141 |
ListGraph::EdgeMap<int> tem(to); |
142 | 142 |
ListGraph::Node tn; |
143 | 143 |
ListGraph::Arc ta; |
144 | 144 |
ListGraph::Edge te; |
145 | 145 |
|
146 | 146 |
SmartGraph::NodeMap<ListGraph::Node> nr(from); |
147 | 147 |
SmartGraph::ArcMap<ListGraph::Arc> ar(from); |
148 | 148 |
SmartGraph::EdgeMap<ListGraph::Edge> er(from); |
149 | 149 |
|
150 | 150 |
ListGraph::NodeMap<SmartGraph::Node> ncr(to); |
151 | 151 |
ListGraph::ArcMap<SmartGraph::Arc> acr(to); |
152 | 152 |
ListGraph::EdgeMap<SmartGraph::Edge> ecr(to); |
153 | 153 |
|
154 | 154 |
graphCopy(from, to). |
155 | 155 |
nodeMap(fnm, tnm).arcMap(fam, tam).edgeMap(fem, tem). |
156 | 156 |
nodeRef(nr).arcRef(ar).edgeRef(er). |
157 | 157 |
nodeCrossRef(ncr).arcCrossRef(acr).edgeCrossRef(ecr). |
158 | 158 |
node(fn, tn).arc(fa, ta).edge(fe, te).run(); |
159 | 159 |
|
160 | 160 |
check(countNodes(from) == countNodes(to), "Wrong copy."); |
161 | 161 |
check(countEdges(from) == countEdges(to), "Wrong copy."); |
162 | 162 |
check(countArcs(from) == countArcs(to), "Wrong copy."); |
163 | 163 |
|
164 | 164 |
for (SmartGraph::NodeIt it(from); it != INVALID; ++it) { |
165 | 165 |
check(ncr[nr[it]] == it, "Wrong copy."); |
166 | 166 |
check(fnm[it] == tnm[nr[it]], "Wrong copy."); |
167 | 167 |
} |
168 | 168 |
|
169 | 169 |
for (SmartGraph::ArcIt it(from); it != INVALID; ++it) { |
170 | 170 |
check(acr[ar[it]] == it, "Wrong copy."); |
171 | 171 |
check(fam[it] == tam[ar[it]], "Wrong copy."); |
172 | 172 |
check(nr[from.source(it)] == to.source(ar[it]), "Wrong copy."); |
173 | 173 |
check(nr[from.target(it)] == to.target(ar[it]), "Wrong copy."); |
174 | 174 |
} |
175 | 175 |
|
176 | 176 |
for (SmartGraph::EdgeIt it(from); it != INVALID; ++it) { |
177 | 177 |
check(ecr[er[it]] == it, "Wrong copy."); |
178 | 178 |
check(fem[it] == tem[er[it]], "Wrong copy."); |
179 | 179 |
check(nr[from.u(it)] == to.u(er[it]) || nr[from.u(it)] == to.v(er[it]), |
180 | 180 |
"Wrong copy."); |
181 | 181 |
check(nr[from.v(it)] == to.u(er[it]) || nr[from.v(it)] == to.v(er[it]), |
182 | 182 |
"Wrong copy."); |
183 | 183 |
check((from.u(it) != from.v(it)) == (to.u(er[it]) != to.v(er[it])), |
184 | 184 |
"Wrong copy."); |
185 | 185 |
} |
186 | 186 |
|
187 | 187 |
for (ListGraph::NodeIt it(to); it != INVALID; ++it) { |
188 | 188 |
check(nr[ncr[it]] == it, "Wrong copy."); |
189 | 189 |
} |
190 | 190 |
|
191 | 191 |
for (ListGraph::ArcIt it(to); it != INVALID; ++it) { |
192 | 192 |
check(ar[acr[it]] == it, "Wrong copy."); |
193 | 193 |
} |
194 | 194 |
for (ListGraph::EdgeIt it(to); it != INVALID; ++it) { |
195 | 195 |
check(er[ecr[it]] == it, "Wrong copy."); |
196 | 196 |
} |
197 | 197 |
check(tn == nr[fn], "Wrong copy."); |
198 | 198 |
check(ta == ar[fa], "Wrong copy."); |
199 | 199 |
check(te == er[fe], "Wrong copy."); |
200 | 200 |
|
201 | 201 |
// Test repeated copy |
202 | 202 |
graphCopy(from, to).run(); |
203 | 203 |
|
204 | 204 |
check(countNodes(from) == countNodes(to), "Wrong copy."); |
205 | 205 |
check(countEdges(from) == countEdges(to), "Wrong copy."); |
206 | 206 |
check(countArcs(from) == countArcs(to), "Wrong copy."); |
207 | 207 |
} |
208 | 208 |
|
209 | 209 |
|
210 | 210 |
int main() { |
211 | 211 |
digraph_copy_test(); |
212 | 212 |
graph_copy_test(); |
213 | 213 |
|
214 | 214 |
return 0; |
215 | 215 |
} |
1 | 1 |
/* -*- mode: C++; indent-tabs-mode: nil; -*- |
2 | 2 |
* |
3 | 3 |
* This file is a part of LEMON, a generic C++ optimization library. |
4 | 4 |
* |
5 |
* Copyright (C) 2003- |
|
5 |
* Copyright (C) 2003-2011 |
|
6 | 6 |
* Egervary Jeno Kombinatorikus Optimalizalasi Kutatocsoport |
7 | 7 |
* (Egervary Research Group on Combinatorial Optimization, EGRES). |
8 | 8 |
* |
9 | 9 |
* Permission to use, modify and distribute this software is granted |
10 | 10 |
* provided that this copyright notice appears in all copies. For |
11 | 11 |
* precise terms see the accompanying LICENSE file. |
12 | 12 |
* |
13 | 13 |
* This software is provided "AS IS" with no warranty of any kind, |
14 | 14 |
* express or implied, and with no claim as to its suitability for any |
15 | 15 |
* purpose. |
16 | 16 |
* |
17 | 17 |
*/ |
18 | 18 |
|
19 | 19 |
#include <iostream> |
20 | 20 |
#include <fstream> |
21 | 21 |
#include <string> |
22 | 22 |
#include <vector> |
23 | 23 |
|
24 | 24 |
#include <lemon/concept_check.h> |
25 | 25 |
#include <lemon/concepts/heap.h> |
26 | 26 |
|
27 | 27 |
#include <lemon/smart_graph.h> |
28 | 28 |
#include <lemon/lgf_reader.h> |
29 | 29 |
#include <lemon/dijkstra.h> |
30 | 30 |
#include <lemon/maps.h> |
31 | 31 |
|
32 | 32 |
#include <lemon/bin_heap.h> |
33 | 33 |
#include <lemon/quad_heap.h> |
34 | 34 |
#include <lemon/dheap.h> |
35 | 35 |
#include <lemon/fib_heap.h> |
36 | 36 |
#include <lemon/pairing_heap.h> |
37 | 37 |
#include <lemon/radix_heap.h> |
38 | 38 |
#include <lemon/binomial_heap.h> |
39 | 39 |
#include <lemon/bucket_heap.h> |
40 | 40 |
|
41 | 41 |
#include "test_tools.h" |
42 | 42 |
|
43 | 43 |
using namespace lemon; |
44 | 44 |
using namespace lemon::concepts; |
45 | 45 |
|
46 | 46 |
typedef ListDigraph Digraph; |
47 | 47 |
DIGRAPH_TYPEDEFS(Digraph); |
48 | 48 |
|
49 | 49 |
char test_lgf[] = |
50 | 50 |
"@nodes\n" |
51 | 51 |
"label\n" |
52 | 52 |
"0\n" |
53 | 53 |
"1\n" |
54 | 54 |
"2\n" |
55 | 55 |
"3\n" |
56 | 56 |
"4\n" |
57 | 57 |
"5\n" |
58 | 58 |
"6\n" |
59 | 59 |
"7\n" |
60 | 60 |
"8\n" |
61 | 61 |
"9\n" |
62 | 62 |
"@arcs\n" |
63 | 63 |
" label capacity\n" |
64 | 64 |
"0 5 0 94\n" |
65 | 65 |
"3 9 1 11\n" |
66 | 66 |
"8 7 2 83\n" |
67 | 67 |
"1 2 3 94\n" |
68 | 68 |
"5 7 4 35\n" |
69 | 69 |
"7 4 5 84\n" |
70 | 70 |
"9 5 6 38\n" |
71 | 71 |
"0 4 7 96\n" |
72 | 72 |
"6 7 8 6\n" |
73 | 73 |
"3 1 9 27\n" |
74 | 74 |
"5 2 10 77\n" |
75 | 75 |
"5 6 11 69\n" |
76 | 76 |
"6 5 12 41\n" |
77 | 77 |
"4 6 13 70\n" |
78 | 78 |
"3 2 14 45\n" |
79 | 79 |
"7 9 15 93\n" |
80 | 80 |
"5 9 16 50\n" |
81 | 81 |
"9 0 17 94\n" |
82 | 82 |
"9 6 18 67\n" |
83 | 83 |
"0 9 19 86\n" |
84 | 84 |
"@attributes\n" |
85 | 85 |
"source 3\n"; |
86 | 86 |
|
87 | 87 |
int test_seq[] = { 2, 28, 19, 27, 33, 25, 13, 41, 10, 26, 1, 9, 4, 34}; |
88 | 88 |
int test_inc[] = {20, 28, 34, 16, 0, 46, 44, 0, 42, 32, 14, 8, 6, 37}; |
89 | 89 |
|
90 | 90 |
int test_len = sizeof(test_seq) / sizeof(test_seq[0]); |
91 | 91 |
|
92 | 92 |
template <typename Heap> |
93 | 93 |
void heapSortTest() { |
94 | 94 |
RangeMap<int> map(test_len, -1); |
95 | 95 |
Heap heap(map); |
96 | 96 |
|
97 | 97 |
std::vector<int> v(test_len); |
98 | 98 |
for (int i = 0; i < test_len; ++i) { |
99 | 99 |
v[i] = test_seq[i]; |
100 | 100 |
heap.push(i, v[i]); |
101 | 101 |
} |
102 | 102 |
std::sort(v.begin(), v.end()); |
103 | 103 |
for (int i = 0; i < test_len; ++i) { |
104 | 104 |
check(v[i] == heap.prio(), "Wrong order in heap sort."); |
105 | 105 |
heap.pop(); |
106 | 106 |
} |
107 | 107 |
} |
108 | 108 |
|
109 | 109 |
template <typename Heap> |
110 | 110 |
void heapIncreaseTest() { |
111 | 111 |
RangeMap<int> map(test_len, -1); |
112 | 112 |
|
113 | 113 |
Heap heap(map); |
114 | 114 |
|
115 | 115 |
std::vector<int> v(test_len); |
116 | 116 |
for (int i = 0; i < test_len; ++i) { |
117 | 117 |
v[i] = test_seq[i]; |
118 | 118 |
heap.push(i, v[i]); |
119 | 119 |
} |
120 | 120 |
for (int i = 0; i < test_len; ++i) { |
121 | 121 |
v[i] += test_inc[i]; |
122 | 122 |
heap.increase(i, v[i]); |
123 | 123 |
} |
124 | 124 |
std::sort(v.begin(), v.end()); |
125 | 125 |
for (int i = 0; i < test_len; ++i) { |
126 | 126 |
check(v[i] == heap.prio(), "Wrong order in heap increase test."); |
127 | 127 |
heap.pop(); |
128 | 128 |
} |
129 | 129 |
} |
130 | 130 |
|
131 | 131 |
template <typename Heap> |
132 | 132 |
void dijkstraHeapTest(const Digraph& digraph, const IntArcMap& length, |
133 | 133 |
Node source) { |
134 | 134 |
|
135 | 135 |
typename Dijkstra<Digraph, IntArcMap>::template SetStandardHeap<Heap>:: |
136 | 136 |
Create dijkstra(digraph, length); |
137 | 137 |
|
138 | 138 |
dijkstra.run(source); |
139 | 139 |
|
140 | 140 |
for(ArcIt a(digraph); a != INVALID; ++a) { |
141 | 141 |
Node s = digraph.source(a); |
142 | 142 |
Node t = digraph.target(a); |
143 | 143 |
if (dijkstra.reached(s)) { |
144 | 144 |
check( dijkstra.dist(t) - dijkstra.dist(s) <= length[a], |
145 | 145 |
"Error in shortest path tree."); |
146 | 146 |
} |
147 | 147 |
} |
148 | 148 |
|
149 | 149 |
for(NodeIt n(digraph); n != INVALID; ++n) { |
150 | 150 |
if ( dijkstra.reached(n) && dijkstra.predArc(n) != INVALID ) { |
151 | 151 |
Arc a = dijkstra.predArc(n); |
152 | 152 |
Node s = digraph.source(a); |
153 | 153 |
check( dijkstra.dist(n) - dijkstra.dist(s) == length[a], |
154 | 154 |
"Error in shortest path tree."); |
155 | 155 |
} |
156 | 156 |
} |
157 | 157 |
|
158 | 158 |
} |
159 | 159 |
|
160 | 160 |
int main() { |
161 | 161 |
|
162 | 162 |
typedef int Item; |
163 | 163 |
typedef int Prio; |
164 | 164 |
typedef RangeMap<int> ItemIntMap; |
165 | 165 |
|
166 | 166 |
Digraph digraph; |
167 | 167 |
IntArcMap length(digraph); |
168 | 168 |
Node source; |
169 | 169 |
|
170 | 170 |
std::istringstream input(test_lgf); |
171 | 171 |
digraphReader(digraph, input). |
172 | 172 |
arcMap("capacity", length). |
173 | 173 |
node("source", source). |
174 | 174 |
run(); |
175 | 175 |
|
176 | 176 |
// BinHeap |
177 | 177 |
{ |
178 | 178 |
typedef BinHeap<Prio, ItemIntMap> IntHeap; |
179 | 179 |
checkConcept<Heap<Prio, ItemIntMap>, IntHeap>(); |
180 | 180 |
heapSortTest<IntHeap>(); |
181 | 181 |
heapIncreaseTest<IntHeap>(); |
182 | 182 |
|
183 | 183 |
typedef BinHeap<Prio, IntNodeMap > NodeHeap; |
184 | 184 |
checkConcept<Heap<Prio, IntNodeMap >, NodeHeap>(); |
185 | 185 |
dijkstraHeapTest<NodeHeap>(digraph, length, source); |
186 | 186 |
} |
187 | 187 |
|
188 | 188 |
// QuadHeap |
189 | 189 |
{ |
190 | 190 |
typedef QuadHeap<Prio, ItemIntMap> IntHeap; |
191 | 191 |
checkConcept<Heap<Prio, ItemIntMap>, IntHeap>(); |
192 | 192 |
heapSortTest<IntHeap>(); |
193 | 193 |
heapIncreaseTest<IntHeap>(); |
194 | 194 |
|
195 | 195 |
typedef QuadHeap<Prio, IntNodeMap > NodeHeap; |
196 | 196 |
checkConcept<Heap<Prio, IntNodeMap >, NodeHeap>(); |
197 | 197 |
dijkstraHeapTest<NodeHeap>(digraph, length, source); |
198 | 198 |
} |
199 | 199 |
|
200 | 200 |
// DHeap |
201 | 201 |
{ |
202 | 202 |
typedef DHeap<Prio, ItemIntMap> IntHeap; |
203 | 203 |
checkConcept<Heap<Prio, ItemIntMap>, IntHeap>(); |
204 | 204 |
heapSortTest<IntHeap>(); |
205 | 205 |
heapIncreaseTest<IntHeap>(); |
206 | 206 |
|
207 | 207 |
typedef DHeap<Prio, IntNodeMap > NodeHeap; |
208 | 208 |
checkConcept<Heap<Prio, IntNodeMap >, NodeHeap>(); |
209 | 209 |
dijkstraHeapTest<NodeHeap>(digraph, length, source); |
210 | 210 |
} |
211 | 211 |
|
212 | 212 |
// FibHeap |
213 | 213 |
{ |
214 | 214 |
typedef FibHeap<Prio, ItemIntMap> IntHeap; |
215 | 215 |
checkConcept<Heap<Prio, ItemIntMap>, IntHeap>(); |
216 | 216 |
heapSortTest<IntHeap>(); |
217 | 217 |
heapIncreaseTest<IntHeap>(); |
218 | 218 |
|
219 | 219 |
typedef FibHeap<Prio, IntNodeMap > NodeHeap; |
220 | 220 |
checkConcept<Heap<Prio, IntNodeMap >, NodeHeap>(); |
221 | 221 |
dijkstraHeapTest<NodeHeap>(digraph, length, source); |
222 | 222 |
} |
223 | 223 |
|
224 | 224 |
// PairingHeap |
225 | 225 |
{ |
226 | 226 |
typedef PairingHeap<Prio, ItemIntMap> IntHeap; |
227 | 227 |
checkConcept<Heap<Prio, ItemIntMap>, IntHeap>(); |
228 | 228 |
heapSortTest<IntHeap>(); |
229 | 229 |
heapIncreaseTest<IntHeap>(); |
230 | 230 |
|
231 | 231 |
typedef PairingHeap<Prio, IntNodeMap > NodeHeap; |
232 | 232 |
checkConcept<Heap<Prio, IntNodeMap >, NodeHeap>(); |
233 | 233 |
dijkstraHeapTest<NodeHeap>(digraph, length, source); |
234 | 234 |
} |
235 | 235 |
|
236 | 236 |
// RadixHeap |
237 | 237 |
{ |
238 | 238 |
typedef RadixHeap<ItemIntMap> IntHeap; |
239 | 239 |
checkConcept<Heap<Prio, ItemIntMap>, IntHeap>(); |
240 | 240 |
heapSortTest<IntHeap>(); |
241 | 241 |
heapIncreaseTest<IntHeap>(); |
242 | 242 |
|
243 | 243 |
typedef RadixHeap<IntNodeMap > NodeHeap; |
244 | 244 |
checkConcept<Heap<Prio, IntNodeMap >, NodeHeap>(); |
245 | 245 |
dijkstraHeapTest<NodeHeap>(digraph, length, source); |
246 | 246 |
} |
247 | 247 |
|
248 | 248 |
// BinomialHeap |
249 | 249 |
{ |
250 | 250 |
typedef BinomialHeap<Prio, ItemIntMap> IntHeap; |
251 | 251 |
checkConcept<Heap<Prio, ItemIntMap>, IntHeap>(); |
252 | 252 |
heapSortTest<IntHeap>(); |
253 | 253 |
heapIncreaseTest<IntHeap>(); |
254 | 254 |
|
255 | 255 |
typedef BinomialHeap<Prio, IntNodeMap > NodeHeap; |
256 | 256 |
checkConcept<Heap<Prio, IntNodeMap >, NodeHeap>(); |
257 | 257 |
dijkstraHeapTest<NodeHeap>(digraph, length, source); |
258 | 258 |
} |
259 | 259 |
|
260 | 260 |
// BucketHeap, SimpleBucketHeap |
261 | 261 |
{ |
262 | 262 |
typedef BucketHeap<ItemIntMap> IntHeap; |
263 | 263 |
checkConcept<Heap<Prio, ItemIntMap>, IntHeap>(); |
264 | 264 |
heapSortTest<IntHeap>(); |
265 | 265 |
heapIncreaseTest<IntHeap>(); |
266 | 266 |
|
267 | 267 |
typedef BucketHeap<IntNodeMap > NodeHeap; |
268 | 268 |
checkConcept<Heap<Prio, IntNodeMap >, NodeHeap>(); |
269 | 269 |
dijkstraHeapTest<NodeHeap>(digraph, length, source); |
270 | 270 |
|
271 | 271 |
typedef SimpleBucketHeap<ItemIntMap> SimpleIntHeap; |
272 | 272 |
heapSortTest<SimpleIntHeap>(); |
273 | 273 |
} |
274 | 274 |
|
275 | 275 |
{ |
276 | 276 |
typedef FibHeap<Prio, ItemIntMap> IntHeap; |
277 | 277 |
checkConcept<Heap<Prio, ItemIntMap>, IntHeap>(); |
278 | 278 |
heapSortTest<IntHeap>(); |
279 | 279 |
heapIncreaseTest<IntHeap>(); |
280 | 280 |
|
281 | 281 |
typedef FibHeap<Prio, IntNodeMap > NodeHeap; |
282 | 282 |
checkConcept<Heap<Prio, IntNodeMap >, NodeHeap>(); |
283 | 283 |
dijkstraHeapTest<NodeHeap>(digraph, length, source); |
284 | 284 |
} |
285 | 285 |
|
286 | 286 |
{ |
287 | 287 |
typedef RadixHeap<ItemIntMap> IntHeap; |
288 | 288 |
checkConcept<Heap<Prio, ItemIntMap>, IntHeap>(); |
289 | 289 |
heapSortTest<IntHeap>(); |
290 | 290 |
heapIncreaseTest<IntHeap>(); |
291 | 291 |
|
292 | 292 |
typedef RadixHeap<IntNodeMap > NodeHeap; |
293 | 293 |
checkConcept<Heap<Prio, IntNodeMap >, NodeHeap>(); |
294 | 294 |
dijkstraHeapTest<NodeHeap>(digraph, length, source); |
295 | 295 |
} |
296 | 296 |
|
297 | 297 |
{ |
298 | 298 |
typedef BucketHeap<ItemIntMap> IntHeap; |
299 | 299 |
checkConcept<Heap<Prio, ItemIntMap>, IntHeap>(); |
300 | 300 |
heapSortTest<IntHeap>(); |
301 | 301 |
heapIncreaseTest<IntHeap>(); |
302 | 302 |
|
303 | 303 |
typedef BucketHeap<IntNodeMap > NodeHeap; |
304 | 304 |
checkConcept<Heap<Prio, IntNodeMap >, NodeHeap>(); |
305 | 305 |
dijkstraHeapTest<NodeHeap>(digraph, length, source); |
306 | 306 |
} |
307 | 307 |
|
308 | 308 |
|
309 | 309 |
return 0; |
310 | 310 |
} |
1 | 1 |
/* -*- mode: C++; indent-tabs-mode: nil; -*- |
2 | 2 |
* |
3 | 3 |
* This file is a part of LEMON, a generic C++ optimization library. |
4 | 4 |
* |
5 |
* Copyright (C) 2003- |
|
5 |
* Copyright (C) 2003-2011 |
|
6 | 6 |
* Egervary Jeno Kombinatorikus Optimalizalasi Kutatocsoport |
7 | 7 |
* (Egervary Research Group on Combinatorial Optimization, EGRES). |
8 | 8 |
* |
9 | 9 |
* Permission to use, modify and distribute this software is granted |
10 | 10 |
* provided that this copyright notice appears in all copies. For |
11 | 11 |
* precise terms see the accompanying LICENSE file. |
12 | 12 |
* |
13 | 13 |
* This software is provided "AS IS" with no warranty of any kind, |
14 | 14 |
* express or implied, and with no claim as to its suitability for any |
15 | 15 |
* purpose. |
16 | 16 |
* |
17 | 17 |
*/ |
18 | 18 |
|
19 | 19 |
#include <deque> |
20 | 20 |
#include <set> |
21 | 21 |
|
22 | 22 |
#include <lemon/concept_check.h> |
23 | 23 |
#include <lemon/concepts/maps.h> |
24 | 24 |
#include <lemon/maps.h> |
25 | 25 |
#include <lemon/list_graph.h> |
26 | 26 |
#include <lemon/smart_graph.h> |
27 | 27 |
#include <lemon/adaptors.h> |
28 | 28 |
#include <lemon/dfs.h> |
29 | 29 |
#include <algorithm> |
30 | 30 |
|
31 | 31 |
#include "test_tools.h" |
32 | 32 |
|
33 | 33 |
using namespace lemon; |
34 | 34 |
using namespace lemon::concepts; |
35 | 35 |
|
36 | 36 |
struct A {}; |
37 | 37 |
inline bool operator<(A, A) { return true; } |
38 | 38 |
struct B {}; |
39 | 39 |
|
40 | 40 |
class C { |
41 | 41 |
int _x; |
42 | 42 |
public: |
43 | 43 |
C(int x) : _x(x) {} |
44 | 44 |
int get() const { return _x; } |
45 | 45 |
}; |
46 | 46 |
inline bool operator<(C c1, C c2) { return c1.get() < c2.get(); } |
47 | 47 |
inline bool operator==(C c1, C c2) { return c1.get() == c2.get(); } |
48 | 48 |
|
49 | 49 |
C createC(int x) { return C(x); } |
50 | 50 |
|
51 | 51 |
template <typename T> |
52 | 52 |
class Less { |
53 | 53 |
T _t; |
54 | 54 |
public: |
55 | 55 |
Less(T t): _t(t) {} |
56 | 56 |
bool operator()(const T& t) const { return t < _t; } |
57 | 57 |
}; |
58 | 58 |
|
59 | 59 |
class F { |
60 | 60 |
public: |
61 | 61 |
typedef A argument_type; |
62 | 62 |
typedef B result_type; |
63 | 63 |
|
64 | 64 |
B operator()(const A&) const { return B(); } |
65 | 65 |
private: |
66 | 66 |
F& operator=(const F&); |
67 | 67 |
}; |
68 | 68 |
|
69 | 69 |
int func(A) { return 3; } |
70 | 70 |
|
71 | 71 |
int binc(int a, B) { return a+1; } |
72 | 72 |
|
73 | 73 |
template <typename T> |
74 | 74 |
class Sum { |
75 | 75 |
T& _sum; |
76 | 76 |
public: |
77 | 77 |
Sum(T& sum) : _sum(sum) {} |
78 | 78 |
void operator()(const T& t) { _sum += t; } |
79 | 79 |
}; |
80 | 80 |
|
81 | 81 |
typedef ReadMap<A, double> DoubleMap; |
82 | 82 |
typedef ReadWriteMap<A, double> DoubleWriteMap; |
83 | 83 |
typedef ReferenceMap<A, double, double&, const double&> DoubleRefMap; |
84 | 84 |
|
85 | 85 |
typedef ReadMap<A, bool> BoolMap; |
86 | 86 |
typedef ReadWriteMap<A, bool> BoolWriteMap; |
87 | 87 |
typedef ReferenceMap<A, bool, bool&, const bool&> BoolRefMap; |
88 | 88 |
|
89 | 89 |
int main() |
90 | 90 |
{ |
91 | 91 |
// Map concepts |
92 | 92 |
checkConcept<ReadMap<A,B>, ReadMap<A,B> >(); |
93 | 93 |
checkConcept<ReadMap<A,C>, ReadMap<A,C> >(); |
94 | 94 |
checkConcept<WriteMap<A,B>, WriteMap<A,B> >(); |
95 | 95 |
checkConcept<WriteMap<A,C>, WriteMap<A,C> >(); |
96 | 96 |
checkConcept<ReadWriteMap<A,B>, ReadWriteMap<A,B> >(); |
97 | 97 |
checkConcept<ReadWriteMap<A,C>, ReadWriteMap<A,C> >(); |
98 | 98 |
checkConcept<ReferenceMap<A,B,B&,const B&>, ReferenceMap<A,B,B&,const B&> >(); |
99 | 99 |
checkConcept<ReferenceMap<A,C,C&,const C&>, ReferenceMap<A,C,C&,const C&> >(); |
100 | 100 |
|
101 | 101 |
// NullMap |
102 | 102 |
{ |
103 | 103 |
checkConcept<ReadWriteMap<A,B>, NullMap<A,B> >(); |
104 | 104 |
NullMap<A,B> map1; |
105 | 105 |
NullMap<A,B> map2 = map1; |
106 | 106 |
map1 = nullMap<A,B>(); |
107 | 107 |
} |
108 | 108 |
|
109 | 109 |
// ConstMap |
110 | 110 |
{ |
111 | 111 |
checkConcept<ReadWriteMap<A,B>, ConstMap<A,B> >(); |
112 | 112 |
checkConcept<ReadWriteMap<A,C>, ConstMap<A,C> >(); |
113 | 113 |
ConstMap<A,B> map1; |
114 | 114 |
ConstMap<A,B> map2 = B(); |
115 | 115 |
ConstMap<A,B> map3 = map1; |
116 | 116 |
map1 = constMap<A>(B()); |
117 | 117 |
map1 = constMap<A,B>(); |
118 | 118 |
map1.setAll(B()); |
119 | 119 |
ConstMap<A,C> map4(C(1)); |
120 | 120 |
ConstMap<A,C> map5 = map4; |
121 | 121 |
map4 = constMap<A>(C(2)); |
122 | 122 |
map4.setAll(C(3)); |
123 | 123 |
|
124 | 124 |
checkConcept<ReadWriteMap<A,int>, ConstMap<A,int> >(); |
125 | 125 |
check(constMap<A>(10)[A()] == 10, "Something is wrong with ConstMap"); |
126 | 126 |
|
127 | 127 |
checkConcept<ReadWriteMap<A,int>, ConstMap<A,Const<int,10> > >(); |
128 | 128 |
ConstMap<A,Const<int,10> > map6; |
129 | 129 |
ConstMap<A,Const<int,10> > map7 = map6; |
130 | 130 |
map6 = constMap<A,int,10>(); |
131 | 131 |
map7 = constMap<A,Const<int,10> >(); |
132 | 132 |
check(map6[A()] == 10 && map7[A()] == 10, |
133 | 133 |
"Something is wrong with ConstMap"); |
134 | 134 |
} |
135 | 135 |
|
136 | 136 |
// IdentityMap |
137 | 137 |
{ |
138 | 138 |
checkConcept<ReadMap<A,A>, IdentityMap<A> >(); |
139 | 139 |
IdentityMap<A> map1; |
140 | 140 |
IdentityMap<A> map2 = map1; |
141 | 141 |
map1 = identityMap<A>(); |
142 | 142 |
|
143 | 143 |
checkConcept<ReadMap<double,double>, IdentityMap<double> >(); |
144 | 144 |
check(identityMap<double>()[1.0] == 1.0 && |
145 | 145 |
identityMap<double>()[3.14] == 3.14, |
146 | 146 |
"Something is wrong with IdentityMap"); |
147 | 147 |
} |
148 | 148 |
|
149 | 149 |
// RangeMap |
150 | 150 |
{ |
151 | 151 |
checkConcept<ReferenceMap<int,B,B&,const B&>, RangeMap<B> >(); |
152 | 152 |
RangeMap<B> map1; |
153 | 153 |
RangeMap<B> map2(10); |
154 | 154 |
RangeMap<B> map3(10,B()); |
155 | 155 |
RangeMap<B> map4 = map1; |
156 | 156 |
RangeMap<B> map5 = rangeMap<B>(); |
157 | 157 |
RangeMap<B> map6 = rangeMap<B>(10); |
158 | 158 |
RangeMap<B> map7 = rangeMap(10,B()); |
159 | 159 |
|
160 | 160 |
checkConcept< ReferenceMap<int, double, double&, const double&>, |
161 | 161 |
RangeMap<double> >(); |
162 | 162 |
std::vector<double> v(10, 0); |
163 | 163 |
v[5] = 100; |
164 | 164 |
RangeMap<double> map8(v); |
165 | 165 |
RangeMap<double> map9 = rangeMap(v); |
166 | 166 |
check(map9.size() == 10 && map9[2] == 0 && map9[5] == 100, |
167 | 167 |
"Something is wrong with RangeMap"); |
168 | 168 |
} |
169 | 169 |
|
170 | 170 |
// SparseMap |
171 | 171 |
{ |
172 | 172 |
checkConcept<ReferenceMap<A,B,B&,const B&>, SparseMap<A,B> >(); |
173 | 173 |
SparseMap<A,B> map1; |
174 | 174 |
SparseMap<A,B> map2 = B(); |
175 | 175 |
SparseMap<A,B> map3 = sparseMap<A,B>(); |
176 | 176 |
SparseMap<A,B> map4 = sparseMap<A>(B()); |
177 | 177 |
|
178 | 178 |
checkConcept< ReferenceMap<double, int, int&, const int&>, |
179 | 179 |
SparseMap<double, int> >(); |
180 | 180 |
std::map<double, int> m; |
181 | 181 |
SparseMap<double, int> map5(m); |
182 | 182 |
SparseMap<double, int> map6(m,10); |
183 | 183 |
SparseMap<double, int> map7 = sparseMap(m); |
184 | 184 |
SparseMap<double, int> map8 = sparseMap(m,10); |
185 | 185 |
|
186 | 186 |
check(map5[1.0] == 0 && map5[3.14] == 0 && |
187 | 187 |
map6[1.0] == 10 && map6[3.14] == 10, |
188 | 188 |
"Something is wrong with SparseMap"); |
189 | 189 |
map5[1.0] = map6[3.14] = 100; |
190 | 190 |
check(map5[1.0] == 100 && map5[3.14] == 0 && |
191 | 191 |
map6[1.0] == 10 && map6[3.14] == 100, |
192 | 192 |
"Something is wrong with SparseMap"); |
193 | 193 |
} |
194 | 194 |
|
195 | 195 |
// ComposeMap |
196 | 196 |
{ |
197 | 197 |
typedef ComposeMap<DoubleMap, ReadMap<B,A> > CompMap; |
198 | 198 |
checkConcept<ReadMap<B,double>, CompMap>(); |
199 | 199 |
CompMap map1 = CompMap(DoubleMap(),ReadMap<B,A>()); |
200 | 200 |
CompMap map2 = composeMap(DoubleMap(), ReadMap<B,A>()); |
201 | 201 |
|
202 | 202 |
SparseMap<double, bool> m1(false); m1[3.14] = true; |
203 | 203 |
RangeMap<double> m2(2); m2[0] = 3.0; m2[1] = 3.14; |
204 | 204 |
check(!composeMap(m1,m2)[0] && composeMap(m1,m2)[1], |
205 | 205 |
"Something is wrong with ComposeMap") |
206 | 206 |
} |
207 | 207 |
|
208 | 208 |
// CombineMap |
209 | 209 |
{ |
210 | 210 |
typedef CombineMap<DoubleMap, DoubleMap, std::plus<double> > CombMap; |
211 | 211 |
checkConcept<ReadMap<A,double>, CombMap>(); |
212 | 212 |
CombMap map1 = CombMap(DoubleMap(), DoubleMap()); |
213 | 213 |
CombMap map2 = combineMap(DoubleMap(), DoubleMap(), std::plus<double>()); |
214 | 214 |
|
215 | 215 |
check(combineMap(constMap<B,int,2>(), identityMap<B>(), &binc)[B()] == 3, |
216 | 216 |
"Something is wrong with CombineMap"); |
217 | 217 |
} |
218 | 218 |
|
219 | 219 |
// FunctorToMap, MapToFunctor |
220 | 220 |
{ |
221 | 221 |
checkConcept<ReadMap<A,B>, FunctorToMap<F,A,B> >(); |
222 | 222 |
checkConcept<ReadMap<A,B>, FunctorToMap<F> >(); |
223 | 223 |
FunctorToMap<F> map1; |
224 | 224 |
FunctorToMap<F> map2 = FunctorToMap<F>(F()); |
225 | 225 |
B b = functorToMap(F())[A()]; |
226 | 226 |
|
227 | 227 |
checkConcept<ReadMap<A,B>, MapToFunctor<ReadMap<A,B> > >(); |
228 | 228 |
MapToFunctor<ReadMap<A,B> > map = |
229 | 229 |
MapToFunctor<ReadMap<A,B> >(ReadMap<A,B>()); |
230 | 230 |
|
231 | 231 |
check(functorToMap(&func)[A()] == 3, |
232 | 232 |
"Something is wrong with FunctorToMap"); |
233 | 233 |
check(mapToFunctor(constMap<A,int>(2))(A()) == 2, |
234 | 234 |
"Something is wrong with MapToFunctor"); |
235 | 235 |
check(mapToFunctor(functorToMap(&func))(A()) == 3 && |
236 | 236 |
mapToFunctor(functorToMap(&func))[A()] == 3, |
237 | 237 |
"Something is wrong with FunctorToMap or MapToFunctor"); |
238 | 238 |
check(functorToMap(mapToFunctor(constMap<A,int>(2)))[A()] == 2, |
239 | 239 |
"Something is wrong with FunctorToMap or MapToFunctor"); |
240 | 240 |
} |
241 | 241 |
|
242 | 242 |
// ConvertMap |
243 | 243 |
{ |
244 | 244 |
checkConcept<ReadMap<double,double>, |
245 | 245 |
ConvertMap<ReadMap<double, int>, double> >(); |
246 | 246 |
ConvertMap<RangeMap<bool>, int> map1(rangeMap(1, true)); |
247 | 247 |
ConvertMap<RangeMap<bool>, int> map2 = convertMap<int>(rangeMap(2, false)); |
248 | 248 |
} |
249 | 249 |
|
250 | 250 |
// ForkMap |
251 | 251 |
{ |
252 | 252 |
checkConcept<DoubleWriteMap, ForkMap<DoubleWriteMap, DoubleWriteMap> >(); |
253 | 253 |
|
254 | 254 |
typedef RangeMap<double> RM; |
255 | 255 |
typedef SparseMap<int, double> SM; |
256 | 256 |
RM m1(10, -1); |
257 | 257 |
SM m2(-1); |
258 | 258 |
checkConcept<ReadWriteMap<int, double>, ForkMap<RM, SM> >(); |
259 | 259 |
checkConcept<ReadWriteMap<int, double>, ForkMap<SM, RM> >(); |
260 | 260 |
ForkMap<RM, SM> map1(m1,m2); |
261 | 261 |
ForkMap<SM, RM> map2 = forkMap(m2,m1); |
262 | 262 |
map2.set(5, 10); |
263 | 263 |
check(m1[1] == -1 && m1[5] == 10 && m2[1] == -1 && |
264 | 264 |
m2[5] == 10 && map2[1] == -1 && map2[5] == 10, |
265 | 265 |
"Something is wrong with ForkMap"); |
266 | 266 |
} |
267 | 267 |
|
268 | 268 |
// Arithmetic maps: |
269 | 269 |
// - AddMap, SubMap, MulMap, DivMap |
270 | 270 |
// - ShiftMap, ShiftWriteMap, ScaleMap, ScaleWriteMap |
271 | 271 |
// - NegMap, NegWriteMap, AbsMap |
272 | 272 |
{ |
273 | 273 |
checkConcept<DoubleMap, AddMap<DoubleMap,DoubleMap> >(); |
274 | 274 |
checkConcept<DoubleMap, SubMap<DoubleMap,DoubleMap> >(); |
275 | 275 |
checkConcept<DoubleMap, MulMap<DoubleMap,DoubleMap> >(); |
276 | 276 |
checkConcept<DoubleMap, DivMap<DoubleMap,DoubleMap> >(); |
277 | 277 |
|
278 | 278 |
ConstMap<int, double> c1(1.0), c2(3.14); |
279 | 279 |
IdentityMap<int> im; |
280 | 280 |
ConvertMap<IdentityMap<int>, double> id(im); |
281 | 281 |
check(addMap(c1,id)[0] == 1.0 && addMap(c1,id)[10] == 11.0, |
282 | 282 |
"Something is wrong with AddMap"); |
283 | 283 |
check(subMap(id,c1)[0] == -1.0 && subMap(id,c1)[10] == 9.0, |
284 | 284 |
"Something is wrong with SubMap"); |
285 | 285 |
check(mulMap(id,c2)[0] == 0 && mulMap(id,c2)[2] == 6.28, |
286 | 286 |
"Something is wrong with MulMap"); |
287 | 287 |
check(divMap(c2,id)[1] == 3.14 && divMap(c2,id)[2] == 1.57, |
288 | 288 |
"Something is wrong with DivMap"); |
289 | 289 |
|
290 | 290 |
checkConcept<DoubleMap, ShiftMap<DoubleMap> >(); |
291 | 291 |
checkConcept<DoubleWriteMap, ShiftWriteMap<DoubleWriteMap> >(); |
292 | 292 |
checkConcept<DoubleMap, ScaleMap<DoubleMap> >(); |
293 | 293 |
checkConcept<DoubleWriteMap, ScaleWriteMap<DoubleWriteMap> >(); |
294 | 294 |
checkConcept<DoubleMap, NegMap<DoubleMap> >(); |
295 | 295 |
checkConcept<DoubleWriteMap, NegWriteMap<DoubleWriteMap> >(); |
296 | 296 |
checkConcept<DoubleMap, AbsMap<DoubleMap> >(); |
297 | 297 |
|
298 | 298 |
check(shiftMap(id, 2.0)[1] == 3.0 && shiftMap(id, 2.0)[10] == 12.0, |
299 | 299 |
"Something is wrong with ShiftMap"); |
300 | 300 |
check(shiftWriteMap(id, 2.0)[1] == 3.0 && |
301 | 301 |
shiftWriteMap(id, 2.0)[10] == 12.0, |
302 | 302 |
"Something is wrong with ShiftWriteMap"); |
303 | 303 |
check(scaleMap(id, 2.0)[1] == 2.0 && scaleMap(id, 2.0)[10] == 20.0, |
304 | 304 |
"Something is wrong with ScaleMap"); |
305 | 305 |
check(scaleWriteMap(id, 2.0)[1] == 2.0 && |
306 | 306 |
scaleWriteMap(id, 2.0)[10] == 20.0, |
307 | 307 |
"Something is wrong with ScaleWriteMap"); |
308 | 308 |
check(negMap(id)[1] == -1.0 && negMap(id)[-10] == 10.0, |
309 | 309 |
"Something is wrong with NegMap"); |
310 | 310 |
check(negWriteMap(id)[1] == -1.0 && negWriteMap(id)[-10] == 10.0, |
311 | 311 |
"Something is wrong with NegWriteMap"); |
312 | 312 |
check(absMap(id)[1] == 1.0 && absMap(id)[-10] == 10.0, |
313 | 313 |
"Something is wrong with AbsMap"); |
314 | 314 |
} |
315 | 315 |
|
316 | 316 |
// Logical maps: |
317 | 317 |
// - TrueMap, FalseMap |
318 | 318 |
// - AndMap, OrMap |
319 | 319 |
// - NotMap, NotWriteMap |
320 | 320 |
// - EqualMap, LessMap |
321 | 321 |
{ |
322 | 322 |
checkConcept<BoolMap, TrueMap<A> >(); |
323 | 323 |
checkConcept<BoolMap, FalseMap<A> >(); |
324 | 324 |
checkConcept<BoolMap, AndMap<BoolMap,BoolMap> >(); |
325 | 325 |
checkConcept<BoolMap, OrMap<BoolMap,BoolMap> >(); |
326 | 326 |
checkConcept<BoolMap, NotMap<BoolMap> >(); |
327 | 327 |
checkConcept<BoolWriteMap, NotWriteMap<BoolWriteMap> >(); |
328 | 328 |
checkConcept<BoolMap, EqualMap<DoubleMap,DoubleMap> >(); |
329 | 329 |
checkConcept<BoolMap, LessMap<DoubleMap,DoubleMap> >(); |
330 | 330 |
|
331 | 331 |
TrueMap<int> tm; |
332 | 332 |
FalseMap<int> fm; |
333 | 333 |
RangeMap<bool> rm(2); |
334 | 334 |
rm[0] = true; rm[1] = false; |
335 | 335 |
check(andMap(tm,rm)[0] && !andMap(tm,rm)[1] && |
336 | 336 |
!andMap(fm,rm)[0] && !andMap(fm,rm)[1], |
337 | 337 |
"Something is wrong with AndMap"); |
338 | 338 |
check(orMap(tm,rm)[0] && orMap(tm,rm)[1] && |
339 | 339 |
orMap(fm,rm)[0] && !orMap(fm,rm)[1], |
340 | 340 |
"Something is wrong with OrMap"); |
341 | 341 |
check(!notMap(rm)[0] && notMap(rm)[1], |
342 | 342 |
"Something is wrong with NotMap"); |
343 | 343 |
check(!notWriteMap(rm)[0] && notWriteMap(rm)[1], |
344 | 344 |
"Something is wrong with NotWriteMap"); |
345 | 345 |
|
346 | 346 |
ConstMap<int, double> cm(2.0); |
347 | 347 |
IdentityMap<int> im; |
348 | 348 |
ConvertMap<IdentityMap<int>, double> id(im); |
349 | 349 |
check(lessMap(id,cm)[1] && !lessMap(id,cm)[2] && !lessMap(id,cm)[3], |
350 | 350 |
"Something is wrong with LessMap"); |
351 | 351 |
check(!equalMap(id,cm)[1] && equalMap(id,cm)[2] && !equalMap(id,cm)[3], |
352 | 352 |
"Something is wrong with EqualMap"); |
353 | 353 |
} |
354 | 354 |
|
355 | 355 |
// LoggerBoolMap |
356 | 356 |
{ |
357 | 357 |
typedef std::vector<int> vec; |
358 | 358 |
checkConcept<WriteMap<int, bool>, LoggerBoolMap<vec::iterator> >(); |
359 | 359 |
checkConcept<WriteMap<int, bool>, |
360 | 360 |
LoggerBoolMap<std::back_insert_iterator<vec> > >(); |
361 | 361 |
|
362 | 362 |
vec v1; |
363 | 363 |
vec v2(10); |
364 | 364 |
LoggerBoolMap<std::back_insert_iterator<vec> > |
365 | 365 |
map1(std::back_inserter(v1)); |
366 | 366 |
LoggerBoolMap<vec::iterator> map2(v2.begin()); |
367 | 367 |
map1.set(10, false); |
368 | 368 |
map1.set(20, true); map2.set(20, true); |
369 | 369 |
map1.set(30, false); map2.set(40, false); |
370 | 370 |
map1.set(50, true); map2.set(50, true); |
371 | 371 |
map1.set(60, true); map2.set(60, true); |
372 | 372 |
check(v1.size() == 3 && v2.size() == 10 && |
373 | 373 |
v1[0]==20 && v1[1]==50 && v1[2]==60 && |
374 | 374 |
v2[0]==20 && v2[1]==50 && v2[2]==60, |
375 | 375 |
"Something is wrong with LoggerBoolMap"); |
376 | 376 |
|
377 | 377 |
int i = 0; |
378 | 378 |
for ( LoggerBoolMap<vec::iterator>::Iterator it = map2.begin(); |
379 | 379 |
it != map2.end(); ++it ) |
380 | 380 |
check(v1[i++] == *it, "Something is wrong with LoggerBoolMap"); |
381 | 381 |
|
382 | 382 |
typedef ListDigraph Graph; |
383 | 383 |
DIGRAPH_TYPEDEFS(Graph); |
384 | 384 |
Graph gr; |
385 | 385 |
|
386 | 386 |
Node n0 = gr.addNode(); |
387 | 387 |
Node n1 = gr.addNode(); |
388 | 388 |
Node n2 = gr.addNode(); |
389 | 389 |
Node n3 = gr.addNode(); |
390 | 390 |
|
391 | 391 |
gr.addArc(n3, n0); |
392 | 392 |
gr.addArc(n3, n2); |
393 | 393 |
gr.addArc(n0, n2); |
394 | 394 |
gr.addArc(n2, n1); |
395 | 395 |
gr.addArc(n0, n1); |
396 | 396 |
|
397 | 397 |
{ |
398 | 398 |
std::vector<Node> v; |
399 | 399 |
dfs(gr).processedMap(loggerBoolMap(std::back_inserter(v))).run(); |
400 | 400 |
|
401 | 401 |
check(v.size()==4 && v[0]==n1 && v[1]==n2 && v[2]==n0 && v[3]==n3, |
402 | 402 |
"Something is wrong with LoggerBoolMap"); |
403 | 403 |
} |
404 | 404 |
{ |
405 | 405 |
std::vector<Node> v(countNodes(gr)); |
406 | 406 |
dfs(gr).processedMap(loggerBoolMap(v.begin())).run(); |
407 | 407 |
|
408 | 408 |
check(v.size()==4 && v[0]==n1 && v[1]==n2 && v[2]==n0 && v[3]==n3, |
409 | 409 |
"Something is wrong with LoggerBoolMap"); |
410 | 410 |
} |
411 | 411 |
} |
412 | 412 |
|
413 | 413 |
// IdMap, RangeIdMap |
414 | 414 |
{ |
415 | 415 |
typedef ListDigraph Graph; |
416 | 416 |
DIGRAPH_TYPEDEFS(Graph); |
417 | 417 |
|
418 | 418 |
checkConcept<ReadMap<Node, int>, IdMap<Graph, Node> >(); |
419 | 419 |
checkConcept<ReadMap<Arc, int>, IdMap<Graph, Arc> >(); |
420 | 420 |
checkConcept<ReadMap<Node, int>, RangeIdMap<Graph, Node> >(); |
421 | 421 |
checkConcept<ReadMap<Arc, int>, RangeIdMap<Graph, Arc> >(); |
422 | 422 |
|
423 | 423 |
Graph gr; |
424 | 424 |
IdMap<Graph, Node> nmap(gr); |
425 | 425 |
IdMap<Graph, Arc> amap(gr); |
426 | 426 |
RangeIdMap<Graph, Node> nrmap(gr); |
427 | 427 |
RangeIdMap<Graph, Arc> armap(gr); |
428 | 428 |
|
429 | 429 |
Node n0 = gr.addNode(); |
430 | 430 |
Node n1 = gr.addNode(); |
431 | 431 |
Node n2 = gr.addNode(); |
432 | 432 |
|
433 | 433 |
Arc a0 = gr.addArc(n0, n1); |
434 | 434 |
Arc a1 = gr.addArc(n0, n2); |
435 | 435 |
Arc a2 = gr.addArc(n2, n1); |
436 | 436 |
Arc a3 = gr.addArc(n2, n0); |
437 | 437 |
|
438 | 438 |
check(nmap[n0] == gr.id(n0) && nmap(gr.id(n0)) == n0, "Wrong IdMap"); |
439 | 439 |
check(nmap[n1] == gr.id(n1) && nmap(gr.id(n1)) == n1, "Wrong IdMap"); |
440 | 440 |
check(nmap[n2] == gr.id(n2) && nmap(gr.id(n2)) == n2, "Wrong IdMap"); |
441 | 441 |
|
442 | 442 |
check(amap[a0] == gr.id(a0) && amap(gr.id(a0)) == a0, "Wrong IdMap"); |
443 | 443 |
check(amap[a1] == gr.id(a1) && amap(gr.id(a1)) == a1, "Wrong IdMap"); |
444 | 444 |
check(amap[a2] == gr.id(a2) && amap(gr.id(a2)) == a2, "Wrong IdMap"); |
445 | 445 |
check(amap[a3] == gr.id(a3) && amap(gr.id(a3)) == a3, "Wrong IdMap"); |
446 | 446 |
|
447 | 447 |
check(nmap.inverse()[gr.id(n0)] == n0, "Wrong IdMap::InverseMap"); |
448 | 448 |
check(amap.inverse()[gr.id(a0)] == a0, "Wrong IdMap::InverseMap"); |
449 | 449 |
|
450 | 450 |
check(nrmap.size() == 3 && armap.size() == 4, |
451 | 451 |
"Wrong RangeIdMap::size()"); |
452 | 452 |
|
453 | 453 |
check(nrmap[n0] == 0 && nrmap(0) == n0, "Wrong RangeIdMap"); |
454 | 454 |
check(nrmap[n1] == 1 && nrmap(1) == n1, "Wrong RangeIdMap"); |
455 | 455 |
check(nrmap[n2] == 2 && nrmap(2) == n2, "Wrong RangeIdMap"); |
456 | 456 |
|
457 | 457 |
check(armap[a0] == 0 && armap(0) == a0, "Wrong RangeIdMap"); |
458 | 458 |
check(armap[a1] == 1 && armap(1) == a1, "Wrong RangeIdMap"); |
459 | 459 |
check(armap[a2] == 2 && armap(2) == a2, "Wrong RangeIdMap"); |
460 | 460 |
check(armap[a3] == 3 && armap(3) == a3, "Wrong RangeIdMap"); |
461 | 461 |
|
462 | 462 |
check(nrmap.inverse()[0] == n0, "Wrong RangeIdMap::InverseMap"); |
463 | 463 |
check(armap.inverse()[0] == a0, "Wrong RangeIdMap::InverseMap"); |
464 | 464 |
|
465 | 465 |
gr.erase(n1); |
466 | 466 |
|
467 | 467 |
if (nrmap[n0] == 1) nrmap.swap(n0, n2); |
468 | 468 |
nrmap.swap(n2, n0); |
469 | 469 |
if (armap[a1] == 1) armap.swap(a1, a3); |
470 | 470 |
armap.swap(a3, a1); |
471 | 471 |
|
472 | 472 |
check(nrmap.size() == 2 && armap.size() == 2, |
473 | 473 |
"Wrong RangeIdMap::size()"); |
474 | 474 |
|
475 | 475 |
check(nrmap[n0] == 1 && nrmap(1) == n0, "Wrong RangeIdMap"); |
476 | 476 |
check(nrmap[n2] == 0 && nrmap(0) == n2, "Wrong RangeIdMap"); |
477 | 477 |
|
478 | 478 |
check(armap[a1] == 1 && armap(1) == a1, "Wrong RangeIdMap"); |
479 | 479 |
check(armap[a3] == 0 && armap(0) == a3, "Wrong RangeIdMap"); |
480 | 480 |
|
481 | 481 |
check(nrmap.inverse()[0] == n2, "Wrong RangeIdMap::InverseMap"); |
482 | 482 |
check(armap.inverse()[0] == a3, "Wrong RangeIdMap::InverseMap"); |
483 | 483 |
} |
484 | 484 |
|
485 | 485 |
// SourceMap, TargetMap, ForwardMap, BackwardMap, InDegMap, OutDegMap |
486 | 486 |
{ |
487 | 487 |
typedef ListGraph Graph; |
488 | 488 |
GRAPH_TYPEDEFS(Graph); |
489 | 489 |
|
490 | 490 |
checkConcept<ReadMap<Arc, Node>, SourceMap<Graph> >(); |
491 | 491 |
checkConcept<ReadMap<Arc, Node>, TargetMap<Graph> >(); |
492 | 492 |
checkConcept<ReadMap<Edge, Arc>, ForwardMap<Graph> >(); |
493 | 493 |
checkConcept<ReadMap<Edge, Arc>, BackwardMap<Graph> >(); |
494 | 494 |
checkConcept<ReadMap<Node, int>, InDegMap<Graph> >(); |
495 | 495 |
checkConcept<ReadMap<Node, int>, OutDegMap<Graph> >(); |
496 | 496 |
|
497 | 497 |
Graph gr; |
498 | 498 |
Node n0 = gr.addNode(); |
499 | 499 |
Node n1 = gr.addNode(); |
500 | 500 |
Node n2 = gr.addNode(); |
501 | 501 |
|
502 | 502 |
gr.addEdge(n0,n1); |
503 | 503 |
gr.addEdge(n1,n2); |
504 | 504 |
gr.addEdge(n0,n2); |
505 | 505 |
gr.addEdge(n2,n1); |
506 | 506 |
gr.addEdge(n1,n2); |
507 | 507 |
gr.addEdge(n0,n1); |
508 | 508 |
|
509 | 509 |
for (EdgeIt e(gr); e != INVALID; ++e) { |
510 | 510 |
check(forwardMap(gr)[e] == gr.direct(e, true), "Wrong ForwardMap"); |
511 | 511 |
check(backwardMap(gr)[e] == gr.direct(e, false), "Wrong BackwardMap"); |
512 | 512 |
} |
513 | 513 |
|
514 | 514 |
check(mapCompare(gr, |
515 | 515 |
sourceMap(orienter(gr, constMap<Edge, bool>(true))), |
516 | 516 |
targetMap(orienter(gr, constMap<Edge, bool>(false)))), |
517 | 517 |
"Wrong SourceMap or TargetMap"); |
1 | 1 |
/* -*- mode: C++; indent-tabs-mode: nil; -*- |
2 | 2 |
* |
3 | 3 |
* This file is a part of LEMON, a generic C++ optimization library. |
4 | 4 |
* |
5 |
* Copyright (C) 2003- |
|
5 |
* Copyright (C) 2003-2011 |
|
6 | 6 |
* Egervary Jeno Kombinatorikus Optimalizalasi Kutatocsoport |
7 | 7 |
* (Egervary Research Group on Combinatorial Optimization, EGRES). |
8 | 8 |
* |
9 | 9 |
* Permission to use, modify and distribute this software is granted |
10 | 10 |
* provided that this copyright notice appears in all copies. For |
11 | 11 |
* precise terms see the accompanying LICENSE file. |
12 | 12 |
* |
13 | 13 |
* This software is provided "AS IS" with no warranty of any kind, |
14 | 14 |
* express or implied, and with no claim as to its suitability for any |
15 | 15 |
* purpose. |
16 | 16 |
* |
17 | 17 |
*/ |
18 | 18 |
|
19 | 19 |
#include <iostream> |
20 | 20 |
|
21 | 21 |
#include "test_tools.h" |
22 | 22 |
#include <lemon/smart_graph.h> |
23 | 23 |
#include <lemon/preflow.h> |
24 | 24 |
#include <lemon/concepts/digraph.h> |
25 | 25 |
#include <lemon/concepts/maps.h> |
26 | 26 |
#include <lemon/lgf_reader.h> |
27 | 27 |
#include <lemon/elevator.h> |
28 | 28 |
|
29 | 29 |
using namespace lemon; |
30 | 30 |
|
31 | 31 |
char test_lgf[] = |
32 | 32 |
"@nodes\n" |
33 | 33 |
"label\n" |
34 | 34 |
"0\n" |
35 | 35 |
"1\n" |
36 | 36 |
"2\n" |
37 | 37 |
"3\n" |
38 | 38 |
"4\n" |
39 | 39 |
"5\n" |
40 | 40 |
"6\n" |
41 | 41 |
"7\n" |
42 | 42 |
"8\n" |
43 | 43 |
"9\n" |
44 | 44 |
"@arcs\n" |
45 | 45 |
" label capacity\n" |
46 | 46 |
"0 1 0 20\n" |
47 | 47 |
"0 2 1 0\n" |
48 | 48 |
"1 1 2 3\n" |
49 | 49 |
"1 2 3 8\n" |
50 | 50 |
"1 3 4 8\n" |
51 | 51 |
"2 5 5 5\n" |
52 | 52 |
"3 2 6 5\n" |
53 | 53 |
"3 5 7 5\n" |
54 | 54 |
"3 6 8 5\n" |
55 | 55 |
"4 3 9 3\n" |
56 | 56 |
"5 7 10 3\n" |
57 | 57 |
"5 6 11 10\n" |
58 | 58 |
"5 8 12 10\n" |
59 | 59 |
"6 8 13 8\n" |
60 | 60 |
"8 9 14 20\n" |
61 | 61 |
"8 1 15 5\n" |
62 | 62 |
"9 5 16 5\n" |
63 | 63 |
"@attributes\n" |
64 | 64 |
"source 1\n" |
65 | 65 |
"target 8\n"; |
66 | 66 |
|
67 | 67 |
void checkPreflowCompile() |
68 | 68 |
{ |
69 | 69 |
typedef int VType; |
70 | 70 |
typedef concepts::Digraph Digraph; |
71 | 71 |
|
72 | 72 |
typedef Digraph::Node Node; |
73 | 73 |
typedef Digraph::Arc Arc; |
74 | 74 |
typedef concepts::ReadMap<Arc,VType> CapMap; |
75 | 75 |
typedef concepts::ReadWriteMap<Arc,VType> FlowMap; |
76 | 76 |
typedef concepts::WriteMap<Node,bool> CutMap; |
77 | 77 |
|
78 | 78 |
typedef Elevator<Digraph, Digraph::Node> Elev; |
79 | 79 |
typedef LinkedElevator<Digraph, Digraph::Node> LinkedElev; |
80 | 80 |
|
81 | 81 |
Digraph g; |
82 | 82 |
Node n; |
83 | 83 |
Arc e; |
84 | 84 |
CapMap cap; |
85 | 85 |
FlowMap flow; |
86 | 86 |
CutMap cut; |
87 | 87 |
VType v; |
88 | 88 |
bool b; |
89 | 89 |
|
90 | 90 |
typedef Preflow<Digraph, CapMap> |
91 | 91 |
::SetFlowMap<FlowMap> |
92 | 92 |
::SetElevator<Elev> |
93 | 93 |
::SetStandardElevator<LinkedElev> |
94 | 94 |
::Create PreflowType; |
95 | 95 |
PreflowType preflow_test(g, cap, n, n); |
96 | 96 |
const PreflowType& const_preflow_test = preflow_test; |
97 | 97 |
|
98 | 98 |
const PreflowType::Elevator& elev = const_preflow_test.elevator(); |
99 | 99 |
preflow_test.elevator(const_cast<PreflowType::Elevator&>(elev)); |
100 | 100 |
PreflowType::Tolerance tol = const_preflow_test.tolerance(); |
101 | 101 |
preflow_test.tolerance(tol); |
102 | 102 |
|
103 | 103 |
preflow_test |
104 | 104 |
.capacityMap(cap) |
105 | 105 |
.flowMap(flow) |
106 | 106 |
.source(n) |
107 | 107 |
.target(n); |
108 | 108 |
|
109 | 109 |
preflow_test.init(); |
110 | 110 |
preflow_test.init(cap); |
111 | 111 |
preflow_test.startFirstPhase(); |
112 | 112 |
preflow_test.startSecondPhase(); |
113 | 113 |
preflow_test.run(); |
114 | 114 |
preflow_test.runMinCut(); |
115 | 115 |
|
116 | 116 |
v = const_preflow_test.flowValue(); |
117 | 117 |
v = const_preflow_test.flow(e); |
118 | 118 |
const FlowMap& fm = const_preflow_test.flowMap(); |
119 | 119 |
b = const_preflow_test.minCut(n); |
120 | 120 |
const_preflow_test.minCutMap(cut); |
121 | 121 |
|
122 | 122 |
ignore_unused_variable_warning(fm); |
123 | 123 |
} |
124 | 124 |
|
125 | 125 |
int cutValue (const SmartDigraph& g, |
126 | 126 |
const SmartDigraph::NodeMap<bool>& cut, |
127 | 127 |
const SmartDigraph::ArcMap<int>& cap) { |
128 | 128 |
|
129 | 129 |
int c=0; |
130 | 130 |
for(SmartDigraph::ArcIt e(g); e!=INVALID; ++e) { |
131 | 131 |
if (cut[g.source(e)] && !cut[g.target(e)]) c+=cap[e]; |
132 | 132 |
} |
133 | 133 |
return c; |
134 | 134 |
} |
135 | 135 |
|
136 | 136 |
bool checkFlow(const SmartDigraph& g, |
137 | 137 |
const SmartDigraph::ArcMap<int>& flow, |
138 | 138 |
const SmartDigraph::ArcMap<int>& cap, |
139 | 139 |
SmartDigraph::Node s, SmartDigraph::Node t) { |
140 | 140 |
|
141 | 141 |
for (SmartDigraph::ArcIt e(g); e != INVALID; ++e) { |
142 | 142 |
if (flow[e] < 0 || flow[e] > cap[e]) return false; |
143 | 143 |
} |
144 | 144 |
|
145 | 145 |
for (SmartDigraph::NodeIt n(g); n != INVALID; ++n) { |
146 | 146 |
if (n == s || n == t) continue; |
147 | 147 |
int sum = 0; |
148 | 148 |
for (SmartDigraph::OutArcIt e(g, n); e != INVALID; ++e) { |
149 | 149 |
sum += flow[e]; |
150 | 150 |
} |
151 | 151 |
for (SmartDigraph::InArcIt e(g, n); e != INVALID; ++e) { |
152 | 152 |
sum -= flow[e]; |
153 | 153 |
} |
154 | 154 |
if (sum != 0) return false; |
155 | 155 |
} |
156 | 156 |
return true; |
157 | 157 |
} |
158 | 158 |
|
159 | 159 |
void initFlowTest() |
160 | 160 |
{ |
161 | 161 |
DIGRAPH_TYPEDEFS(SmartDigraph); |
162 | 162 |
|
163 | 163 |
SmartDigraph g; |
164 | 164 |
SmartDigraph::ArcMap<int> cap(g),iflow(g); |
165 | 165 |
Node s=g.addNode(); Node t=g.addNode(); |
166 | 166 |
Node n1=g.addNode(); Node n2=g.addNode(); |
167 | 167 |
Arc a; |
168 | 168 |
a=g.addArc(s,n1); cap[a]=20; iflow[a]=20; |
169 | 169 |
a=g.addArc(n1,n2); cap[a]=10; iflow[a]=0; |
170 | 170 |
a=g.addArc(n2,t); cap[a]=20; iflow[a]=0; |
171 | 171 |
|
172 | 172 |
Preflow<SmartDigraph> pre(g,cap,s,t); |
173 | 173 |
pre.init(iflow); |
174 | 174 |
pre.startFirstPhase(); |
175 | 175 |
check(pre.flowValue() == 10, "The incorrect max flow value."); |
176 | 176 |
check(pre.minCut(s), "Wrong min cut (Node s)."); |
177 | 177 |
check(pre.minCut(n1), "Wrong min cut (Node n1)."); |
178 | 178 |
check(!pre.minCut(n2), "Wrong min cut (Node n2)."); |
179 | 179 |
check(!pre.minCut(t), "Wrong min cut (Node t)."); |
180 | 180 |
} |
181 | 181 |
|
182 | 182 |
|
183 | 183 |
int main() { |
184 | 184 |
|
185 | 185 |
typedef SmartDigraph Digraph; |
186 | 186 |
|
187 | 187 |
typedef Digraph::Node Node; |
188 | 188 |
typedef Digraph::NodeIt NodeIt; |
189 | 189 |
typedef Digraph::ArcIt ArcIt; |
190 | 190 |
typedef Digraph::ArcMap<int> CapMap; |
191 | 191 |
typedef Digraph::ArcMap<int> FlowMap; |
192 | 192 |
typedef Digraph::NodeMap<bool> CutMap; |
193 | 193 |
|
194 | 194 |
typedef Preflow<Digraph, CapMap> PType; |
195 | 195 |
|
196 | 196 |
Digraph g; |
197 | 197 |
Node s, t; |
198 | 198 |
CapMap cap(g); |
199 | 199 |
std::istringstream input(test_lgf); |
200 | 200 |
DigraphReader<Digraph>(g,input). |
201 | 201 |
arcMap("capacity", cap). |
202 | 202 |
node("source",s). |
203 | 203 |
node("target",t). |
204 | 204 |
run(); |
205 | 205 |
|
206 | 206 |
PType preflow_test(g, cap, s, t); |
207 | 207 |
preflow_test.run(); |
208 | 208 |
|
209 | 209 |
check(checkFlow(g, preflow_test.flowMap(), cap, s, t), |
210 | 210 |
"The flow is not feasible."); |
211 | 211 |
|
212 | 212 |
CutMap min_cut(g); |
213 | 213 |
preflow_test.minCutMap(min_cut); |
214 | 214 |
int min_cut_value=cutValue(g,min_cut,cap); |
215 | 215 |
|
216 | 216 |
check(preflow_test.flowValue() == min_cut_value, |
217 | 217 |
"The max flow value is not equal to the three min cut values."); |
218 | 218 |
|
219 | 219 |
FlowMap flow(g); |
220 | 220 |
for(ArcIt e(g); e!=INVALID; ++e) flow[e] = preflow_test.flowMap()[e]; |
221 | 221 |
|
222 | 222 |
int flow_value=preflow_test.flowValue(); |
223 | 223 |
|
224 | 224 |
for(ArcIt e(g); e!=INVALID; ++e) cap[e]=2*cap[e]; |
225 | 225 |
preflow_test.init(flow); |
226 | 226 |
preflow_test.startFirstPhase(); |
227 | 227 |
|
228 | 228 |
CutMap min_cut1(g); |
229 | 229 |
preflow_test.minCutMap(min_cut1); |
230 | 230 |
min_cut_value=cutValue(g,min_cut1,cap); |
231 | 231 |
|
232 | 232 |
check(preflow_test.flowValue() == min_cut_value && |
233 | 233 |
min_cut_value == 2*flow_value, |
234 | 234 |
"The max flow value or the min cut value is wrong."); |
235 | 235 |
|
236 | 236 |
preflow_test.startSecondPhase(); |
237 | 237 |
|
238 | 238 |
check(checkFlow(g, preflow_test.flowMap(), cap, s, t), |
239 | 239 |
"The flow is not feasible."); |
240 | 240 |
|
241 | 241 |
CutMap min_cut2(g); |
242 | 242 |
preflow_test.minCutMap(min_cut2); |
243 | 243 |
min_cut_value=cutValue(g,min_cut2,cap); |
244 | 244 |
|
245 | 245 |
check(preflow_test.flowValue() == min_cut_value && |
246 | 246 |
min_cut_value == 2*flow_value, |
247 | 247 |
"The max flow value or the three min cut values were not doubled"); |
248 | 248 |
|
249 | 249 |
|
250 | 250 |
preflow_test.flowMap(flow); |
251 | 251 |
|
252 | 252 |
NodeIt tmp1(g,s); |
253 | 253 |
++tmp1; |
254 | 254 |
if ( tmp1 != INVALID ) s=tmp1; |
255 | 255 |
|
256 | 256 |
NodeIt tmp2(g,t); |
257 | 257 |
++tmp2; |
258 | 258 |
if ( tmp2 != INVALID ) t=tmp2; |
259 | 259 |
|
260 | 260 |
preflow_test.source(s); |
261 | 261 |
preflow_test.target(t); |
262 | 262 |
|
263 | 263 |
preflow_test.run(); |
264 | 264 |
|
265 | 265 |
CutMap min_cut3(g); |
266 | 266 |
preflow_test.minCutMap(min_cut3); |
267 | 267 |
min_cut_value=cutValue(g,min_cut3,cap); |
268 | 268 |
|
269 | 269 |
|
270 | 270 |
check(preflow_test.flowValue() == min_cut_value, |
271 | 271 |
"The max flow value or the three min cut values are incorrect."); |
272 | 272 |
|
273 | 273 |
initFlowTest(); |
274 | 274 |
|
275 | 275 |
return 0; |
276 | 276 |
} |
0 comments (0 inline)