COIN-OR::LEMON - Graph Library

source: lemon/test/maps_test.cc @ 741:71939d63ae77

Last change on this file since 741:71939d63ae77 was 741:71939d63ae77, checked in by Peter Kovacs <kpeter@…>, 10 years ago

Improvements for iterable maps (#73)

File size: 16.9 KB
Line 
1/* -*- mode: C++; indent-tabs-mode: nil; -*-
2 *
3 * This file is a part of LEMON, a generic C++ optimization library.
4 *
5 * Copyright (C) 2003-2009
6 * Egervary Jeno Kombinatorikus Optimalizalasi Kutatocsoport
7 * (Egervary Research Group on Combinatorial Optimization, EGRES).
8 *
9 * Permission to use, modify and distribute this software is granted
10 * provided that this copyright notice appears in all copies. For
11 * precise terms see the accompanying LICENSE file.
12 *
13 * This software is provided "AS IS" with no warranty of any kind,
14 * express or implied, and with no claim as to its suitability for any
15 * purpose.
16 *
17 */
18
19#include <deque>
20#include <set>
21
22#include <lemon/concept_check.h>
23#include <lemon/concepts/maps.h>
24#include <lemon/maps.h>
25#include <lemon/smart_graph.h>
26
27#include "test_tools.h"
28
29using namespace lemon;
30using namespace lemon::concepts;
31
32struct A {};
33inline bool operator<(A, A) { return true; }
34struct B {};
35
36class C {
37  int x;
38public:
39  C(int _x) : x(_x) {}
40};
41
42class F {
43public:
44  typedef A argument_type;
45  typedef B result_type;
46
47  B operator()(const A&) const { return B(); }
48private:
49  F& operator=(const F&);
50};
51
52int func(A) { return 3; }
53
54int binc(int a, B) { return a+1; }
55
56typedef ReadMap<A, double> DoubleMap;
57typedef ReadWriteMap<A, double> DoubleWriteMap;
58typedef ReferenceMap<A, double, double&, const double&> DoubleRefMap;
59
60typedef ReadMap<A, bool> BoolMap;
61typedef ReadWriteMap<A, bool> BoolWriteMap;
62typedef ReferenceMap<A, bool, bool&, const bool&> BoolRefMap;
63
64int main()
65{
66  // Map concepts
67  checkConcept<ReadMap<A,B>, ReadMap<A,B> >();
68  checkConcept<ReadMap<A,C>, ReadMap<A,C> >();
69  checkConcept<WriteMap<A,B>, WriteMap<A,B> >();
70  checkConcept<WriteMap<A,C>, WriteMap<A,C> >();
71  checkConcept<ReadWriteMap<A,B>, ReadWriteMap<A,B> >();
72  checkConcept<ReadWriteMap<A,C>, ReadWriteMap<A,C> >();
73  checkConcept<ReferenceMap<A,B,B&,const B&>, ReferenceMap<A,B,B&,const B&> >();
74  checkConcept<ReferenceMap<A,C,C&,const C&>, ReferenceMap<A,C,C&,const C&> >();
75
76  // NullMap
77  {
78    checkConcept<ReadWriteMap<A,B>, NullMap<A,B> >();
79    NullMap<A,B> map1;
80    NullMap<A,B> map2 = map1;
81    map1 = nullMap<A,B>();
82  }
83
84  // ConstMap
85  {
86    checkConcept<ReadWriteMap<A,B>, ConstMap<A,B> >();
87    checkConcept<ReadWriteMap<A,C>, ConstMap<A,C> >();
88    ConstMap<A,B> map1;
89    ConstMap<A,B> map2 = B();
90    ConstMap<A,B> map3 = map1;
91    map1 = constMap<A>(B());
92    map1 = constMap<A,B>();
93    map1.setAll(B());
94    ConstMap<A,C> map4(C(1));
95    ConstMap<A,C> map5 = map4;
96    map4 = constMap<A>(C(2));
97    map4.setAll(C(3));
98
99    checkConcept<ReadWriteMap<A,int>, ConstMap<A,int> >();
100    check(constMap<A>(10)[A()] == 10, "Something is wrong with ConstMap");
101
102    checkConcept<ReadWriteMap<A,int>, ConstMap<A,Const<int,10> > >();
103    ConstMap<A,Const<int,10> > map6;
104    ConstMap<A,Const<int,10> > map7 = map6;
105    map6 = constMap<A,int,10>();
106    map7 = constMap<A,Const<int,10> >();
107    check(map6[A()] == 10 && map7[A()] == 10,
108          "Something is wrong with ConstMap");
109  }
110
111  // IdentityMap
112  {
113    checkConcept<ReadMap<A,A>, IdentityMap<A> >();
114    IdentityMap<A> map1;
115    IdentityMap<A> map2 = map1;
116    map1 = identityMap<A>();
117
118    checkConcept<ReadMap<double,double>, IdentityMap<double> >();
119    check(identityMap<double>()[1.0] == 1.0 &&
120          identityMap<double>()[3.14] == 3.14,
121          "Something is wrong with IdentityMap");
122  }
123
124  // RangeMap
125  {
126    checkConcept<ReferenceMap<int,B,B&,const B&>, RangeMap<B> >();
127    RangeMap<B> map1;
128    RangeMap<B> map2(10);
129    RangeMap<B> map3(10,B());
130    RangeMap<B> map4 = map1;
131    RangeMap<B> map5 = rangeMap<B>();
132    RangeMap<B> map6 = rangeMap<B>(10);
133    RangeMap<B> map7 = rangeMap(10,B());
134
135    checkConcept< ReferenceMap<int, double, double&, const double&>,
136                  RangeMap<double> >();
137    std::vector<double> v(10, 0);
138    v[5] = 100;
139    RangeMap<double> map8(v);
140    RangeMap<double> map9 = rangeMap(v);
141    check(map9.size() == 10 && map9[2] == 0 && map9[5] == 100,
142          "Something is wrong with RangeMap");
143  }
144
145  // SparseMap
146  {
147    checkConcept<ReferenceMap<A,B,B&,const B&>, SparseMap<A,B> >();
148    SparseMap<A,B> map1;
149    SparseMap<A,B> map2 = B();
150    SparseMap<A,B> map3 = sparseMap<A,B>();
151    SparseMap<A,B> map4 = sparseMap<A>(B());
152
153    checkConcept< ReferenceMap<double, int, int&, const int&>,
154                  SparseMap<double, int> >();
155    std::map<double, int> m;
156    SparseMap<double, int> map5(m);
157    SparseMap<double, int> map6(m,10);
158    SparseMap<double, int> map7 = sparseMap(m);
159    SparseMap<double, int> map8 = sparseMap(m,10);
160
161    check(map5[1.0] == 0 && map5[3.14] == 0 &&
162          map6[1.0] == 10 && map6[3.14] == 10,
163          "Something is wrong with SparseMap");
164    map5[1.0] = map6[3.14] = 100;
165    check(map5[1.0] == 100 && map5[3.14] == 0 &&
166          map6[1.0] == 10 && map6[3.14] == 100,
167          "Something is wrong with SparseMap");
168  }
169
170  // ComposeMap
171  {
172    typedef ComposeMap<DoubleMap, ReadMap<B,A> > CompMap;
173    checkConcept<ReadMap<B,double>, CompMap>();
174    CompMap map1 = CompMap(DoubleMap(),ReadMap<B,A>());
175    CompMap map2 = composeMap(DoubleMap(), ReadMap<B,A>());
176
177    SparseMap<double, bool> m1(false); m1[3.14] = true;
178    RangeMap<double> m2(2); m2[0] = 3.0; m2[1] = 3.14;
179    check(!composeMap(m1,m2)[0] && composeMap(m1,m2)[1],
180          "Something is wrong with ComposeMap")
181  }
182
183  // CombineMap
184  {
185    typedef CombineMap<DoubleMap, DoubleMap, std::plus<double> > CombMap;
186    checkConcept<ReadMap<A,double>, CombMap>();
187    CombMap map1 = CombMap(DoubleMap(), DoubleMap());
188    CombMap map2 = combineMap(DoubleMap(), DoubleMap(), std::plus<double>());
189
190    check(combineMap(constMap<B,int,2>(), identityMap<B>(), &binc)[B()] == 3,
191          "Something is wrong with CombineMap");
192  }
193
194  // FunctorToMap, MapToFunctor
195  {
196    checkConcept<ReadMap<A,B>, FunctorToMap<F,A,B> >();
197    checkConcept<ReadMap<A,B>, FunctorToMap<F> >();
198    FunctorToMap<F> map1;
199    FunctorToMap<F> map2 = FunctorToMap<F>(F());
200    B b = functorToMap(F())[A()];
201
202    checkConcept<ReadMap<A,B>, MapToFunctor<ReadMap<A,B> > >();
203    MapToFunctor<ReadMap<A,B> > map = MapToFunctor<ReadMap<A,B> >(ReadMap<A,B>());
204
205    check(functorToMap(&func)[A()] == 3,
206          "Something is wrong with FunctorToMap");
207    check(mapToFunctor(constMap<A,int>(2))(A()) == 2,
208          "Something is wrong with MapToFunctor");
209    check(mapToFunctor(functorToMap(&func))(A()) == 3 &&
210          mapToFunctor(functorToMap(&func))[A()] == 3,
211          "Something is wrong with FunctorToMap or MapToFunctor");
212    check(functorToMap(mapToFunctor(constMap<A,int>(2)))[A()] == 2,
213          "Something is wrong with FunctorToMap or MapToFunctor");
214  }
215
216  // ConvertMap
217  {
218    checkConcept<ReadMap<double,double>,
219      ConvertMap<ReadMap<double, int>, double> >();
220    ConvertMap<RangeMap<bool>, int> map1(rangeMap(1, true));
221    ConvertMap<RangeMap<bool>, int> map2 = convertMap<int>(rangeMap(2, false));
222  }
223
224  // ForkMap
225  {
226    checkConcept<DoubleWriteMap, ForkMap<DoubleWriteMap, DoubleWriteMap> >();
227
228    typedef RangeMap<double> RM;
229    typedef SparseMap<int, double> SM;
230    RM m1(10, -1);
231    SM m2(-1);
232    checkConcept<ReadWriteMap<int, double>, ForkMap<RM, SM> >();
233    checkConcept<ReadWriteMap<int, double>, ForkMap<SM, RM> >();
234    ForkMap<RM, SM> map1(m1,m2);
235    ForkMap<SM, RM> map2 = forkMap(m2,m1);
236    map2.set(5, 10);
237    check(m1[1] == -1 && m1[5] == 10 && m2[1] == -1 &&
238          m2[5] == 10 && map2[1] == -1 && map2[5] == 10,
239          "Something is wrong with ForkMap");
240  }
241
242  // Arithmetic maps:
243  // - AddMap, SubMap, MulMap, DivMap
244  // - ShiftMap, ShiftWriteMap, ScaleMap, ScaleWriteMap
245  // - NegMap, NegWriteMap, AbsMap
246  {
247    checkConcept<DoubleMap, AddMap<DoubleMap,DoubleMap> >();
248    checkConcept<DoubleMap, SubMap<DoubleMap,DoubleMap> >();
249    checkConcept<DoubleMap, MulMap<DoubleMap,DoubleMap> >();
250    checkConcept<DoubleMap, DivMap<DoubleMap,DoubleMap> >();
251
252    ConstMap<int, double> c1(1.0), c2(3.14);
253    IdentityMap<int> im;
254    ConvertMap<IdentityMap<int>, double> id(im);
255    check(addMap(c1,id)[0] == 1.0  && addMap(c1,id)[10] == 11.0,
256          "Something is wrong with AddMap");
257    check(subMap(id,c1)[0] == -1.0 && subMap(id,c1)[10] == 9.0,
258          "Something is wrong with SubMap");
259    check(mulMap(id,c2)[0] == 0    && mulMap(id,c2)[2]  == 6.28,
260          "Something is wrong with MulMap");
261    check(divMap(c2,id)[1] == 3.14 && divMap(c2,id)[2]  == 1.57,
262          "Something is wrong with DivMap");
263
264    checkConcept<DoubleMap, ShiftMap<DoubleMap> >();
265    checkConcept<DoubleWriteMap, ShiftWriteMap<DoubleWriteMap> >();
266    checkConcept<DoubleMap, ScaleMap<DoubleMap> >();
267    checkConcept<DoubleWriteMap, ScaleWriteMap<DoubleWriteMap> >();
268    checkConcept<DoubleMap, NegMap<DoubleMap> >();
269    checkConcept<DoubleWriteMap, NegWriteMap<DoubleWriteMap> >();
270    checkConcept<DoubleMap, AbsMap<DoubleMap> >();
271
272    check(shiftMap(id, 2.0)[1] == 3.0 && shiftMap(id, 2.0)[10] == 12.0,
273          "Something is wrong with ShiftMap");
274    check(shiftWriteMap(id, 2.0)[1] == 3.0 &&
275          shiftWriteMap(id, 2.0)[10] == 12.0,
276          "Something is wrong with ShiftWriteMap");
277    check(scaleMap(id, 2.0)[1] == 2.0 && scaleMap(id, 2.0)[10] == 20.0,
278          "Something is wrong with ScaleMap");
279    check(scaleWriteMap(id, 2.0)[1] == 2.0 &&
280          scaleWriteMap(id, 2.0)[10] == 20.0,
281          "Something is wrong with ScaleWriteMap");
282    check(negMap(id)[1] == -1.0 && negMap(id)[-10] == 10.0,
283          "Something is wrong with NegMap");
284    check(negWriteMap(id)[1] == -1.0 && negWriteMap(id)[-10] == 10.0,
285          "Something is wrong with NegWriteMap");
286    check(absMap(id)[1] == 1.0 && absMap(id)[-10] == 10.0,
287          "Something is wrong with AbsMap");
288  }
289
290  // Logical maps:
291  // - TrueMap, FalseMap
292  // - AndMap, OrMap
293  // - NotMap, NotWriteMap
294  // - EqualMap, LessMap
295  {
296    checkConcept<BoolMap, TrueMap<A> >();
297    checkConcept<BoolMap, FalseMap<A> >();
298    checkConcept<BoolMap, AndMap<BoolMap,BoolMap> >();
299    checkConcept<BoolMap, OrMap<BoolMap,BoolMap> >();
300    checkConcept<BoolMap, NotMap<BoolMap> >();
301    checkConcept<BoolWriteMap, NotWriteMap<BoolWriteMap> >();
302    checkConcept<BoolMap, EqualMap<DoubleMap,DoubleMap> >();
303    checkConcept<BoolMap, LessMap<DoubleMap,DoubleMap> >();
304
305    TrueMap<int> tm;
306    FalseMap<int> fm;
307    RangeMap<bool> rm(2);
308    rm[0] = true; rm[1] = false;
309    check(andMap(tm,rm)[0] && !andMap(tm,rm)[1] &&
310          !andMap(fm,rm)[0] && !andMap(fm,rm)[1],
311          "Something is wrong with AndMap");
312    check(orMap(tm,rm)[0] && orMap(tm,rm)[1] &&
313          orMap(fm,rm)[0] && !orMap(fm,rm)[1],
314          "Something is wrong with OrMap");
315    check(!notMap(rm)[0] && notMap(rm)[1],
316          "Something is wrong with NotMap");
317    check(!notWriteMap(rm)[0] && notWriteMap(rm)[1],
318          "Something is wrong with NotWriteMap");
319
320    ConstMap<int, double> cm(2.0);
321    IdentityMap<int> im;
322    ConvertMap<IdentityMap<int>, double> id(im);
323    check(lessMap(id,cm)[1] && !lessMap(id,cm)[2] && !lessMap(id,cm)[3],
324          "Something is wrong with LessMap");
325    check(!equalMap(id,cm)[1] && equalMap(id,cm)[2] && !equalMap(id,cm)[3],
326          "Something is wrong with EqualMap");
327  }
328
329  // LoggerBoolMap
330  {
331    typedef std::vector<int> vec;
332    vec v1;
333    vec v2(10);
334    LoggerBoolMap<std::back_insert_iterator<vec> >
335      map1(std::back_inserter(v1));
336    LoggerBoolMap<vec::iterator> map2(v2.begin());
337    map1.set(10, false);
338    map1.set(20, true);   map2.set(20, true);
339    map1.set(30, false);  map2.set(40, false);
340    map1.set(50, true);   map2.set(50, true);
341    map1.set(60, true);   map2.set(60, true);
342    check(v1.size() == 3 && v2.size() == 10 &&
343          v1[0]==20 && v1[1]==50 && v1[2]==60 &&
344          v2[0]==20 && v2[1]==50 && v2[2]==60,
345          "Something is wrong with LoggerBoolMap");
346
347    int i = 0;
348    for ( LoggerBoolMap<vec::iterator>::Iterator it = map2.begin();
349          it != map2.end(); ++it )
350      check(v1[i++] == *it, "Something is wrong with LoggerBoolMap");
351  }
352
353  // Iterable bool map
354  {
355    typedef SmartGraph Graph;
356    typedef SmartGraph::Node Item;
357
358    typedef IterableBoolMap<SmartGraph, SmartGraph::Node> Ibm;
359    checkConcept<ReferenceMap<Item, bool, bool&, const bool&>, Ibm>();
360
361    const int num = 10;
362    Graph g;
363    std::vector<Item> items;
364    for (int i = 0; i < num; ++i) {
365      items.push_back(g.addNode());
366    }
367
368    Ibm map1(g, true);
369    int n = 0;
370    for (Ibm::TrueIt it(map1); it != INVALID; ++it) {
371      check(map1[static_cast<Item>(it)], "Wrong TrueIt");
372      ++n;
373    }
374    check(n == num, "Wrong number");
375
376    n = 0;
377    for (Ibm::ItemIt it(map1, true); it != INVALID; ++it) {
378        check(map1[static_cast<Item>(it)], "Wrong ItemIt for true");
379        ++n;
380    }
381    check(n == num, "Wrong number");
382    check(Ibm::FalseIt(map1) == INVALID, "Wrong FalseIt");
383    check(Ibm::ItemIt(map1, false) == INVALID, "Wrong ItemIt for false");
384
385    map1[items[5]] = true;
386
387    n = 0;
388    for (Ibm::ItemIt it(map1, true); it != INVALID; ++it) {
389        check(map1[static_cast<Item>(it)], "Wrong ItemIt for true");
390        ++n;
391    }
392    check(n == num, "Wrong number");
393
394    map1[items[num / 2]] = false;
395    check(map1[items[num / 2]] == false, "Wrong map value");
396
397    n = 0;
398    for (Ibm::TrueIt it(map1); it != INVALID; ++it) {
399        check(map1[static_cast<Item>(it)], "Wrong TrueIt for true");
400        ++n;
401    }
402    check(n == num - 1, "Wrong number");
403
404    n = 0;
405    for (Ibm::FalseIt it(map1); it != INVALID; ++it) {
406        check(!map1[static_cast<Item>(it)], "Wrong FalseIt for true");
407        ++n;
408    }
409    check(n == 1, "Wrong number");
410
411    map1[items[0]] = false;
412    check(map1[items[0]] == false, "Wrong map value");
413
414    map1[items[num - 1]] = false;
415    check(map1[items[num - 1]] == false, "Wrong map value");
416
417    n = 0;
418    for (Ibm::TrueIt it(map1); it != INVALID; ++it) {
419        check(map1[static_cast<Item>(it)], "Wrong TrueIt for true");
420        ++n;
421    }
422    check(n == num - 3, "Wrong number");
423    check(map1.trueNum() == num - 3, "Wrong number");
424
425    n = 0;
426    for (Ibm::FalseIt it(map1); it != INVALID; ++it) {
427        check(!map1[static_cast<Item>(it)], "Wrong FalseIt for true");
428        ++n;
429    }
430    check(n == 3, "Wrong number");
431    check(map1.falseNum() == 3, "Wrong number");
432  }
433
434  // Iterable int map
435  {
436    typedef SmartGraph Graph;
437    typedef SmartGraph::Node Item;
438    typedef IterableIntMap<SmartGraph, SmartGraph::Node> Iim;
439
440    checkConcept<ReferenceMap<Item, int, int&, const int&>, Iim>();
441
442    const int num = 10;
443    Graph g;
444    std::vector<Item> items;
445    for (int i = 0; i < num; ++i) {
446      items.push_back(g.addNode());
447    }
448
449    Iim map1(g);
450    check(map1.size() == 0, "Wrong size");
451
452    for (int i = 0; i < num; ++i) {
453      map1[items[i]] = i;
454    }
455    check(map1.size() == num, "Wrong size");
456
457    for (int i = 0; i < num; ++i) {
458      Iim::ItemIt it(map1, i);
459      check(static_cast<Item>(it) == items[i], "Wrong value");
460      ++it;
461      check(static_cast<Item>(it) == INVALID, "Wrong value");
462    }
463
464    for (int i = 0; i < num; ++i) {
465      map1[items[i]] = i % 2;
466    }
467    check(map1.size() == 2, "Wrong size");
468
469    int n = 0;
470    for (Iim::ItemIt it(map1, 0); it != INVALID; ++it) {
471      check(map1[static_cast<Item>(it)] == 0, "Wrong value");
472      ++n;
473    }
474    check(n == (num + 1) / 2, "Wrong number");
475
476    for (Iim::ItemIt it(map1, 1); it != INVALID; ++it) {
477      check(map1[static_cast<Item>(it)] == 1, "Wrong value");
478      ++n;
479    }
480    check(n == num, "Wrong number");
481
482  }
483
484  // Iterable value map
485  {
486    typedef SmartGraph Graph;
487    typedef SmartGraph::Node Item;
488    typedef IterableValueMap<SmartGraph, SmartGraph::Node, double> Ivm;
489
490    checkConcept<ReadWriteMap<Item, double>, Ivm>();
491
492    const int num = 10;
493    Graph g;
494    std::vector<Item> items;
495    for (int i = 0; i < num; ++i) {
496      items.push_back(g.addNode());
497    }
498
499    Ivm map1(g, 0.0);
500    check(distance(map1.beginValue(), map1.endValue()) == 1, "Wrong size");
501    check(*map1.beginValue() == 0.0, "Wrong value");
502
503    for (int i = 0; i < num; ++i) {
504      map1.set(items[i], static_cast<double>(i));
505    }
506    check(distance(map1.beginValue(), map1.endValue()) == num, "Wrong size");
507
508    for (int i = 0; i < num; ++i) {
509      Ivm::ItemIt it(map1, static_cast<double>(i));
510      check(static_cast<Item>(it) == items[i], "Wrong value");
511      ++it;
512      check(static_cast<Item>(it) == INVALID, "Wrong value");
513    }
514
515    for (Ivm::ValueIterator vit = map1.beginValue();
516         vit != map1.endValue(); ++vit) {
517      check(map1[static_cast<Item>(Ivm::ItemIt(map1, *vit))] == *vit,
518            "Wrong ValueIterator");
519    }
520
521    for (int i = 0; i < num; ++i) {
522      map1.set(items[i], static_cast<double>(i % 2));
523    }
524    check(distance(map1.beginValue(), map1.endValue()) == 2, "Wrong size");
525
526    int n = 0;
527    for (Ivm::ItemIt it(map1, 0.0); it != INVALID; ++it) {
528      check(map1[static_cast<Item>(it)] == 0.0, "Wrong value");
529      ++n;
530    }
531    check(n == (num + 1) / 2, "Wrong number");
532
533    for (Ivm::ItemIt it(map1, 1.0); it != INVALID; ++it) {
534      check(map1[static_cast<Item>(it)] == 1.0, "Wrong value");
535      ++n;
536    }
537    check(n == num, "Wrong number");
538
539  }
540  return 0;
541}
Note: See TracBrowser for help on using the repository browser.