COIN-OR::LEMON - Graph Library

Ignore:
File:
1 edited

Legend:

Unmodified
Added
Removed
  • test/maps_test.cc

    r731 r773  
    2424#include <lemon/maps.h>
    2525#include <lemon/list_graph.h>
     26#include <lemon/smart_graph.h>
     27#include <lemon/adaptors.h>
     28#include <lemon/dfs.h>
    2629
    2730#include "test_tools.h"
     
    6164typedef ReadWriteMap<A, bool> BoolWriteMap;
    6265typedef ReferenceMap<A, bool, bool&, const bool&> BoolRefMap;
     66
     67template<typename Map1, typename Map2, typename ItemIt>
     68void compareMap(const Map1& map1, const Map2& map2, ItemIt it) {
     69  for (; it != INVALID; ++it)
     70    check(map1[it] == map2[it], "The maps are not equal");
     71}
    6372
    6473int main()
     
    330339  {
    331340    typedef std::vector<int> vec;
     341    checkConcept<WriteMap<int, bool>, LoggerBoolMap<vec::iterator> >();
     342    checkConcept<WriteMap<int, bool>,
     343                 LoggerBoolMap<std::back_insert_iterator<vec> > >();
     344
    332345    vec v1;
    333346    vec v2(10);
     
    349362          it != map2.end(); ++it )
    350363      check(v1[i++] == *it, "Something is wrong with LoggerBoolMap");
     364   
     365    typedef ListDigraph Graph;
     366    DIGRAPH_TYPEDEFS(Graph);
     367    Graph gr;
     368
     369    Node n0 = gr.addNode();
     370    Node n1 = gr.addNode();
     371    Node n2 = gr.addNode();
     372    Node n3 = gr.addNode();
     373   
     374    gr.addArc(n3, n0);
     375    gr.addArc(n3, n2);
     376    gr.addArc(n0, n2);
     377    gr.addArc(n2, n1);
     378    gr.addArc(n0, n1);
     379   
     380    {
     381      std::vector<Node> v;
     382      dfs(gr).processedMap(loggerBoolMap(std::back_inserter(v))).run();
     383
     384      check(v.size()==4 && v[0]==n1 && v[1]==n2 && v[2]==n0 && v[3]==n3,
     385            "Something is wrong with LoggerBoolMap");
     386    }
     387    {
     388      std::vector<Node> v(countNodes(gr));
     389      dfs(gr).processedMap(loggerBoolMap(v.begin())).run();
     390     
     391      check(v.size()==4 && v[0]==n1 && v[1]==n2 && v[2]==n0 && v[3]==n3,
     392            "Something is wrong with LoggerBoolMap");
     393    }
     394  }
     395 
     396  // IdMap, RangeIdMap
     397  {
     398    typedef ListDigraph Graph;
     399    DIGRAPH_TYPEDEFS(Graph);
     400
     401    checkConcept<ReadMap<Node, int>, IdMap<Graph, Node> >();
     402    checkConcept<ReadMap<Arc, int>, IdMap<Graph, Arc> >();
     403    checkConcept<ReadMap<Node, int>, RangeIdMap<Graph, Node> >();
     404    checkConcept<ReadMap<Arc, int>, RangeIdMap<Graph, Arc> >();
     405   
     406    Graph gr;
     407    IdMap<Graph, Node> nmap(gr);
     408    IdMap<Graph, Arc> amap(gr);
     409    RangeIdMap<Graph, Node> nrmap(gr);
     410    RangeIdMap<Graph, Arc> armap(gr);
     411   
     412    Node n0 = gr.addNode();
     413    Node n1 = gr.addNode();
     414    Node n2 = gr.addNode();
     415   
     416    Arc a0 = gr.addArc(n0, n1);
     417    Arc a1 = gr.addArc(n0, n2);
     418    Arc a2 = gr.addArc(n2, n1);
     419    Arc a3 = gr.addArc(n2, n0);
     420   
     421    check(nmap[n0] == gr.id(n0) && nmap(gr.id(n0)) == n0, "Wrong IdMap");
     422    check(nmap[n1] == gr.id(n1) && nmap(gr.id(n1)) == n1, "Wrong IdMap");
     423    check(nmap[n2] == gr.id(n2) && nmap(gr.id(n2)) == n2, "Wrong IdMap");
     424
     425    check(amap[a0] == gr.id(a0) && amap(gr.id(a0)) == a0, "Wrong IdMap");
     426    check(amap[a1] == gr.id(a1) && amap(gr.id(a1)) == a1, "Wrong IdMap");
     427    check(amap[a2] == gr.id(a2) && amap(gr.id(a2)) == a2, "Wrong IdMap");
     428    check(amap[a3] == gr.id(a3) && amap(gr.id(a3)) == a3, "Wrong IdMap");
     429
     430    check(nmap.inverse()[gr.id(n0)] == n0, "Wrong IdMap::InverseMap");
     431    check(amap.inverse()[gr.id(a0)] == a0, "Wrong IdMap::InverseMap");
     432   
     433    check(nrmap.size() == 3 && armap.size() == 4,
     434          "Wrong RangeIdMap::size()");
     435
     436    check(nrmap[n0] == 0 && nrmap(0) == n0, "Wrong RangeIdMap");
     437    check(nrmap[n1] == 1 && nrmap(1) == n1, "Wrong RangeIdMap");
     438    check(nrmap[n2] == 2 && nrmap(2) == n2, "Wrong RangeIdMap");
     439   
     440    check(armap[a0] == 0 && armap(0) == a0, "Wrong RangeIdMap");
     441    check(armap[a1] == 1 && armap(1) == a1, "Wrong RangeIdMap");
     442    check(armap[a2] == 2 && armap(2) == a2, "Wrong RangeIdMap");
     443    check(armap[a3] == 3 && armap(3) == a3, "Wrong RangeIdMap");
     444
     445    check(nrmap.inverse()[0] == n0, "Wrong RangeIdMap::InverseMap");
     446    check(armap.inverse()[0] == a0, "Wrong RangeIdMap::InverseMap");
     447   
     448    gr.erase(n1);
     449   
     450    if (nrmap[n0] == 1) nrmap.swap(n0, n2);
     451    nrmap.swap(n2, n0);
     452    if (armap[a1] == 1) armap.swap(a1, a3);
     453    armap.swap(a3, a1);
     454   
     455    check(nrmap.size() == 2 && armap.size() == 2,
     456          "Wrong RangeIdMap::size()");
     457
     458    check(nrmap[n0] == 1 && nrmap(1) == n0, "Wrong RangeIdMap");
     459    check(nrmap[n2] == 0 && nrmap(0) == n2, "Wrong RangeIdMap");
     460   
     461    check(armap[a1] == 1 && armap(1) == a1, "Wrong RangeIdMap");
     462    check(armap[a3] == 0 && armap(0) == a3, "Wrong RangeIdMap");
     463
     464    check(nrmap.inverse()[0] == n2, "Wrong RangeIdMap::InverseMap");
     465    check(armap.inverse()[0] == a3, "Wrong RangeIdMap::InverseMap");
     466  }
     467 
     468  // SourceMap, TargetMap, ForwardMap, BackwardMap, InDegMap, OutDegMap
     469  {
     470    typedef ListGraph Graph;
     471    GRAPH_TYPEDEFS(Graph);
     472   
     473    checkConcept<ReadMap<Arc, Node>, SourceMap<Graph> >();
     474    checkConcept<ReadMap<Arc, Node>, TargetMap<Graph> >();
     475    checkConcept<ReadMap<Edge, Arc>, ForwardMap<Graph> >();
     476    checkConcept<ReadMap<Edge, Arc>, BackwardMap<Graph> >();
     477    checkConcept<ReadMap<Node, int>, InDegMap<Graph> >();
     478    checkConcept<ReadMap<Node, int>, OutDegMap<Graph> >();
     479
     480    Graph gr;
     481    Node n0 = gr.addNode();
     482    Node n1 = gr.addNode();
     483    Node n2 = gr.addNode();
     484   
     485    gr.addEdge(n0,n1);
     486    gr.addEdge(n1,n2);
     487    gr.addEdge(n0,n2);
     488    gr.addEdge(n2,n1);
     489    gr.addEdge(n1,n2);
     490    gr.addEdge(n0,n1);
     491   
     492    for (EdgeIt e(gr); e != INVALID; ++e) {
     493      check(forwardMap(gr)[e] == gr.direct(e, true), "Wrong ForwardMap");
     494      check(backwardMap(gr)[e] == gr.direct(e, false), "Wrong BackwardMap");
     495    }
     496   
     497    compareMap(sourceMap(orienter(gr, constMap<Edge, bool>(true))),
     498               targetMap(orienter(gr, constMap<Edge, bool>(false))),
     499               EdgeIt(gr));
     500
     501    typedef Orienter<Graph, const ConstMap<Edge, bool> > Digraph;
     502    Digraph dgr(gr, constMap<Edge, bool>(true));
     503    OutDegMap<Digraph> odm(dgr);
     504    InDegMap<Digraph> idm(dgr);
     505   
     506    check(odm[n0] == 3 && odm[n1] == 2 && odm[n2] == 1, "Wrong OutDegMap");
     507    check(idm[n0] == 0 && idm[n1] == 3 && idm[n2] == 3, "Wrong InDegMap");
     508   
     509    gr.addEdge(n2, n0);
     510
     511    check(odm[n0] == 3 && odm[n1] == 2 && odm[n2] == 2, "Wrong OutDegMap");
     512    check(idm[n0] == 1 && idm[n1] == 3 && idm[n2] == 3, "Wrong InDegMap");
    351513  }
    352514 
     
    354516  {
    355517    typedef ListDigraph Graph;
     518    DIGRAPH_TYPEDEFS(Graph);
     519
     520    checkConcept<ReadWriteMap<Node, int>,
     521                 CrossRefMap<Graph, Node, int> >();
     522    checkConcept<ReadWriteMap<Node, bool>,
     523                 CrossRefMap<Graph, Node, bool> >();
     524    checkConcept<ReadWriteMap<Node, double>,
     525                 CrossRefMap<Graph, Node, double> >();
     526   
     527    Graph gr;
     528    typedef CrossRefMap<Graph, Node, char> CRMap;
     529    CRMap map(gr);
     530   
     531    Node n0 = gr.addNode();
     532    Node n1 = gr.addNode();
     533    Node n2 = gr.addNode();
     534   
     535    map.set(n0, 'A');
     536    map.set(n1, 'B');
     537    map.set(n2, 'C');
     538   
     539    check(map[n0] == 'A' && map('A') == n0 && map.inverse()['A'] == n0,
     540          "Wrong CrossRefMap");
     541    check(map[n1] == 'B' && map('B') == n1 && map.inverse()['B'] == n1,
     542          "Wrong CrossRefMap");
     543    check(map[n2] == 'C' && map('C') == n2 && map.inverse()['C'] == n2,
     544          "Wrong CrossRefMap");
     545    check(map.count('A') == 1 && map.count('B') == 1 && map.count('C') == 1,
     546          "Wrong CrossRefMap::count()");
     547   
     548    CRMap::ValueIt it = map.beginValue();
     549    check(*it++ == 'A' && *it++ == 'B' && *it++ == 'C' &&
     550          it == map.endValue(), "Wrong value iterator");
     551   
     552    map.set(n2, 'A');
     553
     554    check(map[n0] == 'A' && map[n1] == 'B' && map[n2] == 'A',
     555          "Wrong CrossRefMap");
     556    check(map('A') == n0 && map.inverse()['A'] == n0, "Wrong CrossRefMap");
     557    check(map('B') == n1 && map.inverse()['B'] == n1, "Wrong CrossRefMap");
     558    check(map('C') == INVALID && map.inverse()['C'] == INVALID,
     559          "Wrong CrossRefMap");
     560    check(map.count('A') == 2 && map.count('B') == 1 && map.count('C') == 0,
     561          "Wrong CrossRefMap::count()");
     562
     563    it = map.beginValue();
     564    check(*it++ == 'A' && *it++ == 'A' && *it++ == 'B' &&
     565          it == map.endValue(), "Wrong value iterator");
     566
     567    map.set(n0, 'C');
     568
     569    check(map[n0] == 'C' && map[n1] == 'B' && map[n2] == 'A',
     570          "Wrong CrossRefMap");
     571    check(map('A') == n2 && map.inverse()['A'] == n2, "Wrong CrossRefMap");
     572    check(map('B') == n1 && map.inverse()['B'] == n1, "Wrong CrossRefMap");
     573    check(map('C') == n0 && map.inverse()['C'] == n0, "Wrong CrossRefMap");
     574    check(map.count('A') == 1 && map.count('B') == 1 && map.count('C') == 1,
     575          "Wrong CrossRefMap::count()");
     576
     577    it = map.beginValue();
     578    check(*it++ == 'A' && *it++ == 'B' && *it++ == 'C' &&
     579          it == map.endValue(), "Wrong value iterator");
     580  }
     581
     582  // CrossRefMap
     583  {
     584    typedef SmartDigraph Graph;
    356585    DIGRAPH_TYPEDEFS(Graph);
    357586
     
    384613          it == map.endValue(), "Wrong value iterator");
    385614  }
    386 
     615 
     616  // Iterable bool map
     617  {
     618    typedef SmartGraph Graph;
     619    typedef SmartGraph::Node Item;
     620
     621    typedef IterableBoolMap<SmartGraph, SmartGraph::Node> Ibm;
     622    checkConcept<ReferenceMap<Item, bool, bool&, const bool&>, Ibm>();
     623
     624    const int num = 10;
     625    Graph g;
     626    std::vector<Item> items;
     627    for (int i = 0; i < num; ++i) {
     628      items.push_back(g.addNode());
     629    }
     630
     631    Ibm map1(g, true);
     632    int n = 0;
     633    for (Ibm::TrueIt it(map1); it != INVALID; ++it) {
     634      check(map1[static_cast<Item>(it)], "Wrong TrueIt");
     635      ++n;
     636    }
     637    check(n == num, "Wrong number");
     638
     639    n = 0;
     640    for (Ibm::ItemIt it(map1, true); it != INVALID; ++it) {
     641        check(map1[static_cast<Item>(it)], "Wrong ItemIt for true");
     642        ++n;
     643    }
     644    check(n == num, "Wrong number");
     645    check(Ibm::FalseIt(map1) == INVALID, "Wrong FalseIt");
     646    check(Ibm::ItemIt(map1, false) == INVALID, "Wrong ItemIt for false");
     647
     648    map1[items[5]] = true;
     649
     650    n = 0;
     651    for (Ibm::ItemIt it(map1, true); it != INVALID; ++it) {
     652        check(map1[static_cast<Item>(it)], "Wrong ItemIt for true");
     653        ++n;
     654    }
     655    check(n == num, "Wrong number");
     656
     657    map1[items[num / 2]] = false;
     658    check(map1[items[num / 2]] == false, "Wrong map value");
     659
     660    n = 0;
     661    for (Ibm::TrueIt it(map1); it != INVALID; ++it) {
     662        check(map1[static_cast<Item>(it)], "Wrong TrueIt for true");
     663        ++n;
     664    }
     665    check(n == num - 1, "Wrong number");
     666
     667    n = 0;
     668    for (Ibm::FalseIt it(map1); it != INVALID; ++it) {
     669        check(!map1[static_cast<Item>(it)], "Wrong FalseIt for true");
     670        ++n;
     671    }
     672    check(n == 1, "Wrong number");
     673
     674    map1[items[0]] = false;
     675    check(map1[items[0]] == false, "Wrong map value");
     676
     677    map1[items[num - 1]] = false;
     678    check(map1[items[num - 1]] == false, "Wrong map value");
     679
     680    n = 0;
     681    for (Ibm::TrueIt it(map1); it != INVALID; ++it) {
     682        check(map1[static_cast<Item>(it)], "Wrong TrueIt for true");
     683        ++n;
     684    }
     685    check(n == num - 3, "Wrong number");
     686    check(map1.trueNum() == num - 3, "Wrong number");
     687
     688    n = 0;
     689    for (Ibm::FalseIt it(map1); it != INVALID; ++it) {
     690        check(!map1[static_cast<Item>(it)], "Wrong FalseIt for true");
     691        ++n;
     692    }
     693    check(n == 3, "Wrong number");
     694    check(map1.falseNum() == 3, "Wrong number");
     695  }
     696
     697  // Iterable int map
     698  {
     699    typedef SmartGraph Graph;
     700    typedef SmartGraph::Node Item;
     701    typedef IterableIntMap<SmartGraph, SmartGraph::Node> Iim;
     702
     703    checkConcept<ReferenceMap<Item, int, int&, const int&>, Iim>();
     704
     705    const int num = 10;
     706    Graph g;
     707    std::vector<Item> items;
     708    for (int i = 0; i < num; ++i) {
     709      items.push_back(g.addNode());
     710    }
     711
     712    Iim map1(g);
     713    check(map1.size() == 0, "Wrong size");
     714
     715    for (int i = 0; i < num; ++i) {
     716      map1[items[i]] = i;
     717    }
     718    check(map1.size() == num, "Wrong size");
     719
     720    for (int i = 0; i < num; ++i) {
     721      Iim::ItemIt it(map1, i);
     722      check(static_cast<Item>(it) == items[i], "Wrong value");
     723      ++it;
     724      check(static_cast<Item>(it) == INVALID, "Wrong value");
     725    }
     726
     727    for (int i = 0; i < num; ++i) {
     728      map1[items[i]] = i % 2;
     729    }
     730    check(map1.size() == 2, "Wrong size");
     731
     732    int n = 0;
     733    for (Iim::ItemIt it(map1, 0); it != INVALID; ++it) {
     734      check(map1[static_cast<Item>(it)] == 0, "Wrong value");
     735      ++n;
     736    }
     737    check(n == (num + 1) / 2, "Wrong number");
     738
     739    for (Iim::ItemIt it(map1, 1); it != INVALID; ++it) {
     740      check(map1[static_cast<Item>(it)] == 1, "Wrong value");
     741      ++n;
     742    }
     743    check(n == num, "Wrong number");
     744
     745  }
     746
     747  // Iterable value map
     748  {
     749    typedef SmartGraph Graph;
     750    typedef SmartGraph::Node Item;
     751    typedef IterableValueMap<SmartGraph, SmartGraph::Node, double> Ivm;
     752
     753    checkConcept<ReadWriteMap<Item, double>, Ivm>();
     754
     755    const int num = 10;
     756    Graph g;
     757    std::vector<Item> items;
     758    for (int i = 0; i < num; ++i) {
     759      items.push_back(g.addNode());
     760    }
     761
     762    Ivm map1(g, 0.0);
     763    check(distance(map1.beginValue(), map1.endValue()) == 1, "Wrong size");
     764    check(*map1.beginValue() == 0.0, "Wrong value");
     765
     766    for (int i = 0; i < num; ++i) {
     767      map1.set(items[i], static_cast<double>(i));
     768    }
     769    check(distance(map1.beginValue(), map1.endValue()) == num, "Wrong size");
     770
     771    for (int i = 0; i < num; ++i) {
     772      Ivm::ItemIt it(map1, static_cast<double>(i));
     773      check(static_cast<Item>(it) == items[i], "Wrong value");
     774      ++it;
     775      check(static_cast<Item>(it) == INVALID, "Wrong value");
     776    }
     777
     778    for (Ivm::ValueIt vit = map1.beginValue();
     779         vit != map1.endValue(); ++vit) {
     780      check(map1[static_cast<Item>(Ivm::ItemIt(map1, *vit))] == *vit,
     781            "Wrong ValueIt");
     782    }
     783
     784    for (int i = 0; i < num; ++i) {
     785      map1.set(items[i], static_cast<double>(i % 2));
     786    }
     787    check(distance(map1.beginValue(), map1.endValue()) == 2, "Wrong size");
     788
     789    int n = 0;
     790    for (Ivm::ItemIt it(map1, 0.0); it != INVALID; ++it) {
     791      check(map1[static_cast<Item>(it)] == 0.0, "Wrong value");
     792      ++n;
     793    }
     794    check(n == (num + 1) / 2, "Wrong number");
     795
     796    for (Ivm::ItemIt it(map1, 1.0); it != INVALID; ++it) {
     797      check(map1[static_cast<Item>(it)] == 1.0, "Wrong value");
     798      ++n;
     799    }
     800    check(n == num, "Wrong number");
     801
     802  }
    387803  return 0;
    388804}
Note: See TracChangeset for help on using the changeset viewer.