alpar@25: /* -*- C++ -*-
alpar@25:  *
alpar@25:  * This file is a part of LEMON, a generic C++ optimization library
alpar@25:  *
alpar@39:  * Copyright (C) 2003-2008
alpar@25:  * Egervary Jeno Kombinatorikus Optimalizalasi Kutatocsoport
alpar@25:  * (Egervary Research Group on Combinatorial Optimization, EGRES).
alpar@25:  *
alpar@25:  * Permission to use, modify and distribute this software is granted
alpar@25:  * provided that this copyright notice appears in all copies. For
alpar@25:  * precise terms see the accompanying LICENSE file.
alpar@25:  *
alpar@25:  * This software is provided "AS IS" with no warranty of any kind,
alpar@25:  * express or implied, and with no claim as to its suitability for any
alpar@25:  * purpose.
alpar@25:  *
alpar@25:  */
alpar@25: 
alpar@25: #include <deque>
alpar@25: #include <set>
alpar@25: 
alpar@25: #include <lemon/concept_check.h>
alpar@25: #include <lemon/concepts/maps.h>
alpar@25: #include <lemon/maps.h>
alpar@25: 
alpar@25: #include "test_tools.h"
alpar@25: 
alpar@25: using namespace lemon;
alpar@25: using namespace lemon::concepts;
alpar@25: 
alpar@25: struct A {};
alpar@25: inline bool operator<(A, A) { return true; }
alpar@25: struct B {};
alpar@25: 
kpeter@94: class C {
kpeter@94:   int x;
kpeter@94: public:
kpeter@94:   C(int _x) : x(_x) {}
kpeter@94: };
kpeter@94: 
alpar@25: class F {
alpar@25: public:
alpar@25:   typedef A argument_type;
alpar@25:   typedef B result_type;
alpar@25: 
kpeter@80:   B operator()(const A&) const { return B(); }
kpeter@80: private:
kpeter@80:   F& operator=(const F&);
alpar@25: };
alpar@25: 
kpeter@80: int func(A) { return 3; }
alpar@25: 
kpeter@80: int binc(int a, B) { return a+1; }
alpar@25: 
kpeter@80: typedef ReadMap<A, double> DoubleMap;
kpeter@80: typedef ReadWriteMap<A, double> DoubleWriteMap;
kpeter@80: typedef ReferenceMap<A, double, double&, const double&> DoubleRefMap;
alpar@25: 
kpeter@80: typedef ReadMap<A, bool> BoolMap;
alpar@25: typedef ReadWriteMap<A, bool> BoolWriteMap;
kpeter@80: typedef ReferenceMap<A, bool, bool&, const bool&> BoolRefMap;
alpar@25: 
alpar@25: int main()
kpeter@80: {
kpeter@80:   // Map concepts
alpar@25:   checkConcept<ReadMap<A,B>, ReadMap<A,B> >();
kpeter@94:   checkConcept<ReadMap<A,C>, ReadMap<A,C> >();
alpar@25:   checkConcept<WriteMap<A,B>, WriteMap<A,B> >();
kpeter@94:   checkConcept<WriteMap<A,C>, WriteMap<A,C> >();
alpar@25:   checkConcept<ReadWriteMap<A,B>, ReadWriteMap<A,B> >();
kpeter@94:   checkConcept<ReadWriteMap<A,C>, ReadWriteMap<A,C> >();
alpar@25:   checkConcept<ReferenceMap<A,B,B&,const B&>, ReferenceMap<A,B,B&,const B&> >();
kpeter@94:   checkConcept<ReferenceMap<A,C,C&,const C&>, ReferenceMap<A,C,C&,const C&> >();
alpar@25: 
kpeter@80:   // NullMap
kpeter@80:   {
kpeter@80:     checkConcept<ReadWriteMap<A,B>, NullMap<A,B> >();
kpeter@80:     NullMap<A,B> map1;
kpeter@80:     NullMap<A,B> map2 = map1;
kpeter@80:     map1 = nullMap<A,B>();
kpeter@80:   }
kpeter@80: 
kpeter@80:   // ConstMap
kpeter@80:   {
kpeter@80:     checkConcept<ReadWriteMap<A,B>, ConstMap<A,B> >();
kpeter@123:     checkConcept<ReadWriteMap<A,C>, ConstMap<A,C> >();
kpeter@80:     ConstMap<A,B> map1;
deba@136:     ConstMap<A,B> map2 = B();
kpeter@80:     ConstMap<A,B> map3 = map1;
kpeter@80:     map1 = constMap<A>(B());
kpeter@123:     map1 = constMap<A,B>();
kpeter@80:     map1.setAll(B());
kpeter@123:     ConstMap<A,C> map4(C(1));
kpeter@123:     ConstMap<A,C> map5 = map4;
kpeter@123:     map4 = constMap<A>(C(2));
kpeter@123:     map4.setAll(C(3));
kpeter@82: 
kpeter@80:     checkConcept<ReadWriteMap<A,int>, ConstMap<A,int> >();
kpeter@80:     check(constMap<A>(10)[A()] == 10, "Something is wrong with ConstMap");
kpeter@80: 
kpeter@80:     checkConcept<ReadWriteMap<A,int>, ConstMap<A,Const<int,10> > >();
kpeter@123:     ConstMap<A,Const<int,10> > map6;
kpeter@123:     ConstMap<A,Const<int,10> > map7 = map6;
kpeter@123:     map6 = constMap<A,int,10>();
kpeter@123:     map7 = constMap<A,Const<int,10> >();
kpeter@123:     check(map6[A()] == 10 && map7[A()] == 10, "Something is wrong with ConstMap");
kpeter@80:   }
kpeter@80: 
kpeter@80:   // IdentityMap
kpeter@80:   {
kpeter@80:     checkConcept<ReadMap<A,A>, IdentityMap<A> >();
kpeter@80:     IdentityMap<A> map1;
kpeter@80:     IdentityMap<A> map2 = map1;
kpeter@80:     map1 = identityMap<A>();
kpeter@82: 
kpeter@80:     checkConcept<ReadMap<double,double>, IdentityMap<double> >();
kpeter@80:     check(identityMap<double>()[1.0] == 1.0 && identityMap<double>()[3.14] == 3.14,
kpeter@80:           "Something is wrong with IdentityMap");
kpeter@80:   }
kpeter@80: 
kpeter@80:   // RangeMap
kpeter@80:   {
kpeter@80:     checkConcept<ReferenceMap<int,B,B&,const B&>, RangeMap<B> >();
kpeter@80:     RangeMap<B> map1;
kpeter@80:     RangeMap<B> map2(10);
kpeter@80:     RangeMap<B> map3(10,B());
kpeter@80:     RangeMap<B> map4 = map1;
kpeter@80:     RangeMap<B> map5 = rangeMap<B>();
kpeter@80:     RangeMap<B> map6 = rangeMap<B>(10);
kpeter@80:     RangeMap<B> map7 = rangeMap(10,B());
kpeter@80: 
kpeter@80:     checkConcept< ReferenceMap<int, double, double&, const double&>,
kpeter@80:                   RangeMap<double> >();
kpeter@80:     std::vector<double> v(10, 0);
kpeter@80:     v[5] = 100;
kpeter@80:     RangeMap<double> map8(v);
kpeter@80:     RangeMap<double> map9 = rangeMap(v);
kpeter@80:     check(map9.size() == 10 && map9[2] == 0 && map9[5] == 100,
kpeter@80:           "Something is wrong with RangeMap");
kpeter@80:   }
kpeter@80: 
kpeter@80:   // SparseMap
kpeter@80:   {
kpeter@80:     checkConcept<ReferenceMap<A,B,B&,const B&>, SparseMap<A,B> >();
kpeter@80:     SparseMap<A,B> map1;
deba@136:     SparseMap<A,B> map2 = B();
kpeter@80:     SparseMap<A,B> map3 = sparseMap<A,B>();
kpeter@80:     SparseMap<A,B> map4 = sparseMap<A>(B());
kpeter@80: 
kpeter@80:     checkConcept< ReferenceMap<double, int, int&, const int&>,
kpeter@80:                   SparseMap<double, int> >();
kpeter@80:     std::map<double, int> m;
kpeter@80:     SparseMap<double, int> map5(m);
kpeter@80:     SparseMap<double, int> map6(m,10);
kpeter@80:     SparseMap<double, int> map7 = sparseMap(m);
kpeter@80:     SparseMap<double, int> map8 = sparseMap(m,10);
kpeter@80: 
kpeter@80:     check(map5[1.0] == 0 && map5[3.14] == 0 && map6[1.0] == 10 && map6[3.14] == 10,
kpeter@80:           "Something is wrong with SparseMap");
kpeter@80:     map5[1.0] = map6[3.14] = 100;
kpeter@80:     check(map5[1.0] == 100 && map5[3.14] == 0 && map6[1.0] == 10 && map6[3.14] == 100,
kpeter@80:           "Something is wrong with SparseMap");
kpeter@80:   }
kpeter@80: 
kpeter@80:   // ComposeMap
kpeter@80:   {
kpeter@80:     typedef ComposeMap<DoubleMap, ReadMap<B,A> > CompMap;
kpeter@80:     checkConcept<ReadMap<B,double>, CompMap>();
kpeter@80:     CompMap map1(DoubleMap(),ReadMap<B,A>());
kpeter@80:     CompMap map2 = composeMap(DoubleMap(), ReadMap<B,A>());
kpeter@82: 
kpeter@80:     SparseMap<double, bool> m1(false); m1[3.14] = true;
kpeter@80:     RangeMap<double> m2(2); m2[0] = 3.0; m2[1] = 3.14;
kpeter@80:     check(!composeMap(m1,m2)[0] && composeMap(m1,m2)[1], "Something is wrong with ComposeMap")
kpeter@80:   }
kpeter@80: 
kpeter@80:   // CombineMap
kpeter@80:   {
kpeter@80:     typedef CombineMap<DoubleMap, DoubleMap, std::plus<double> > CombMap;
kpeter@80:     checkConcept<ReadMap<A,double>, CombMap>();
kpeter@80:     CombMap map1(DoubleMap(), DoubleMap());
kpeter@80:     CombMap map2 = combineMap(DoubleMap(), DoubleMap(), std::plus<double>());
kpeter@80: 
kpeter@80:     check(combineMap(constMap<B,int,2>(), identityMap<B>(), &binc)[B()] == 3,
kpeter@80:           "Something is wrong with CombineMap");
kpeter@80:   }
kpeter@80: 
kpeter@80:   // FunctorToMap, MapToFunctor
kpeter@80:   {
kpeter@80:     checkConcept<ReadMap<A,B>, FunctorToMap<F,A,B> >();
kpeter@80:     checkConcept<ReadMap<A,B>, FunctorToMap<F> >();
kpeter@80:     FunctorToMap<F> map1;
kpeter@80:     FunctorToMap<F> map2(F());
kpeter@80:     B b = functorToMap(F())[A()];
kpeter@80: 
kpeter@80:     checkConcept<ReadMap<A,B>, MapToFunctor<ReadMap<A,B> > >();
kpeter@80:     MapToFunctor<ReadMap<A,B> > map(ReadMap<A,B>());
kpeter@80: 
kpeter@80:     check(functorToMap(&func)[A()] == 3, "Something is wrong with FunctorToMap");
kpeter@80:     check(mapToFunctor(constMap<A,int>(2))(A()) == 2, "Something is wrong with MapToFunctor");
kpeter@80:     check(mapToFunctor(functorToMap(&func))(A()) == 3 && mapToFunctor(functorToMap(&func))[A()] == 3,
kpeter@80:           "Something is wrong with FunctorToMap or MapToFunctor");
kpeter@80:     check(functorToMap(mapToFunctor(constMap<A,int>(2)))[A()] == 2,
kpeter@80:           "Something is wrong with FunctorToMap or MapToFunctor");
kpeter@80:   }
kpeter@80: 
kpeter@80:   // ConvertMap
kpeter@80:   {
kpeter@80:     checkConcept<ReadMap<double,double>, ConvertMap<ReadMap<double, int>, double> >();
kpeter@80:     ConvertMap<RangeMap<bool>, int> map1(rangeMap(1, true));
kpeter@80:     ConvertMap<RangeMap<bool>, int> map2 = convertMap<int>(rangeMap(2, false));
kpeter@80:   }
kpeter@80: 
kpeter@80:   // ForkMap
kpeter@80:   {
kpeter@80:     checkConcept<DoubleWriteMap, ForkMap<DoubleWriteMap, DoubleWriteMap> >();
kpeter@82: 
kpeter@80:     typedef RangeMap<double> RM;
kpeter@80:     typedef SparseMap<int, double> SM;
kpeter@80:     RM m1(10, -1);
kpeter@80:     SM m2(-1);
kpeter@80:     checkConcept<ReadWriteMap<int, double>, ForkMap<RM, SM> >();
kpeter@80:     checkConcept<ReadWriteMap<int, double>, ForkMap<SM, RM> >();
kpeter@80:     ForkMap<RM, SM> map1(m1,m2);
kpeter@80:     ForkMap<SM, RM> map2 = forkMap(m2,m1);
kpeter@80:     map2.set(5, 10);
kpeter@80:     check(m1[1] == -1 && m1[5] == 10 && m2[1] == -1 && m2[5] == 10 && map2[1] == -1 && map2[5] == 10,
kpeter@80:           "Something is wrong with ForkMap");
kpeter@80:   }
kpeter@82: 
kpeter@80:   // Arithmetic maps:
kpeter@80:   // - AddMap, SubMap, MulMap, DivMap
kpeter@80:   // - ShiftMap, ShiftWriteMap, ScaleMap, ScaleWriteMap
kpeter@80:   // - NegMap, NegWriteMap, AbsMap
kpeter@80:   {
kpeter@80:     checkConcept<DoubleMap, AddMap<DoubleMap,DoubleMap> >();
kpeter@80:     checkConcept<DoubleMap, SubMap<DoubleMap,DoubleMap> >();
kpeter@80:     checkConcept<DoubleMap, MulMap<DoubleMap,DoubleMap> >();
kpeter@80:     checkConcept<DoubleMap, DivMap<DoubleMap,DoubleMap> >();
kpeter@82: 
kpeter@80:     ConstMap<int, double> c1(1.0), c2(3.14);
kpeter@80:     IdentityMap<int> im;
kpeter@80:     ConvertMap<IdentityMap<int>, double> id(im);
kpeter@80:     check(addMap(c1,id)[0] == 1.0  && addMap(c1,id)[10] == 11.0, "Something is wrong with AddMap");
kpeter@80:     check(subMap(id,c1)[0] == -1.0 && subMap(id,c1)[10] == 9.0,  "Something is wrong with SubMap");
kpeter@80:     check(mulMap(id,c2)[0] == 0    && mulMap(id,c2)[2]  == 6.28, "Something is wrong with MulMap");
kpeter@80:     check(divMap(c2,id)[1] == 3.14 && divMap(c2,id)[2]  == 1.57, "Something is wrong with DivMap");
kpeter@82: 
kpeter@80:     checkConcept<DoubleMap, ShiftMap<DoubleMap> >();
kpeter@80:     checkConcept<DoubleWriteMap, ShiftWriteMap<DoubleWriteMap> >();
kpeter@80:     checkConcept<DoubleMap, ScaleMap<DoubleMap> >();
kpeter@80:     checkConcept<DoubleWriteMap, ScaleWriteMap<DoubleWriteMap> >();
kpeter@80:     checkConcept<DoubleMap, NegMap<DoubleMap> >();
kpeter@80:     checkConcept<DoubleWriteMap, NegWriteMap<DoubleWriteMap> >();
kpeter@80:     checkConcept<DoubleMap, AbsMap<DoubleMap> >();
alpar@25: 
kpeter@80:     check(shiftMap(id, 2.0)[1] == 3.0 && shiftMap(id, 2.0)[10] == 12.0,
kpeter@80:           "Something is wrong with ShiftMap");
kpeter@80:     check(shiftWriteMap(id, 2.0)[1] == 3.0 && shiftWriteMap(id, 2.0)[10] == 12.0,
kpeter@80:           "Something is wrong with ShiftWriteMap");
kpeter@80:     check(scaleMap(id, 2.0)[1] == 2.0 && scaleMap(id, 2.0)[10] == 20.0,
kpeter@80:           "Something is wrong with ScaleMap");
kpeter@80:     check(scaleWriteMap(id, 2.0)[1] == 2.0 && scaleWriteMap(id, 2.0)[10] == 20.0,
kpeter@80:           "Something is wrong with ScaleWriteMap");
kpeter@80:     check(negMap(id)[1] == -1.0 && negMap(id)[-10] == 10.0,
kpeter@80:           "Something is wrong with NegMap");
kpeter@80:     check(negWriteMap(id)[1] == -1.0 && negWriteMap(id)[-10] == 10.0,
kpeter@80:           "Something is wrong with NegWriteMap");
kpeter@80:     check(absMap(id)[1] == 1.0 && absMap(id)[-10] == 10.0,
kpeter@80:           "Something is wrong with AbsMap");
kpeter@80:   }
kpeter@82: 
kpeter@82:   // Logical maps:
kpeter@82:   // - TrueMap, FalseMap
kpeter@82:   // - AndMap, OrMap
kpeter@82:   // - NotMap, NotWriteMap
kpeter@82:   // - EqualMap, LessMap
kpeter@80:   {
kpeter@82:     checkConcept<BoolMap, TrueMap<A> >();
kpeter@82:     checkConcept<BoolMap, FalseMap<A> >();
kpeter@82:     checkConcept<BoolMap, AndMap<BoolMap,BoolMap> >();
kpeter@82:     checkConcept<BoolMap, OrMap<BoolMap,BoolMap> >();
kpeter@80:     checkConcept<BoolMap, NotMap<BoolMap> >();
kpeter@80:     checkConcept<BoolWriteMap, NotWriteMap<BoolWriteMap> >();
kpeter@82:     checkConcept<BoolMap, EqualMap<DoubleMap,DoubleMap> >();
kpeter@82:     checkConcept<BoolMap, LessMap<DoubleMap,DoubleMap> >();
kpeter@82: 
kpeter@82:     TrueMap<int> tm;
kpeter@82:     FalseMap<int> fm;
kpeter@80:     RangeMap<bool> rm(2);
kpeter@80:     rm[0] = true; rm[1] = false;
kpeter@82:     check(andMap(tm,rm)[0] && !andMap(tm,rm)[1] && !andMap(fm,rm)[0] && !andMap(fm,rm)[1],
kpeter@82:           "Something is wrong with AndMap");
kpeter@82:     check(orMap(tm,rm)[0] && orMap(tm,rm)[1] && orMap(fm,rm)[0] && !orMap(fm,rm)[1],
kpeter@82:           "Something is wrong with OrMap");
kpeter@82:     check(!notMap(rm)[0] && notMap(rm)[1], "Something is wrong with NotMap");
kpeter@82:     check(!notWriteMap(rm)[0] && notWriteMap(rm)[1], "Something is wrong with NotWriteMap");
kpeter@82: 
kpeter@82:     ConstMap<int, double> cm(2.0);
kpeter@82:     IdentityMap<int> im;
kpeter@82:     ConvertMap<IdentityMap<int>, double> id(im);
kpeter@82:     check(lessMap(id,cm)[1] && !lessMap(id,cm)[2] && !lessMap(id,cm)[3],
kpeter@82:           "Something is wrong with LessMap");
kpeter@82:     check(!equalMap(id,cm)[1] && equalMap(id,cm)[2] && !equalMap(id,cm)[3],
kpeter@82:           "Something is wrong with EqualMap");
kpeter@80:   }
kpeter@159:   
kpeter@159:   // StoreBoolMap
kpeter@159:   {
kpeter@159:     typedef std::vector<int> vec;
kpeter@159:     vec v1;
kpeter@159:     vec v2(10);
kpeter@159:     StoreBoolMap<std::back_insert_iterator<vec> > map1(std::back_inserter(v1));
kpeter@159:     StoreBoolMap<vec::iterator> map2(v2.begin());
kpeter@159:     map1.set(10, false);
kpeter@159:     map1.set(20, true);   map2.set(20, true);
kpeter@159:     map1.set(30, false);  map2.set(40, false);
kpeter@159:     map1.set(50, true);   map2.set(50, true);
kpeter@159:     map1.set(60, true);   map2.set(60, true);
kpeter@159:     check(v1.size() == 3 && v2.size() == 10 &&
kpeter@159:           v1[0]==20 && v1[1]==50 && v1[2]==60 && v2[0]==20 && v2[1]==50 && v2[2]==60,
kpeter@159:           "Something is wrong with StoreBoolMap");
kpeter@159:           
kpeter@159:     int i = 0;
kpeter@159:     for ( StoreBoolMap<vec::iterator>::Iterator it = map2.begin();
kpeter@159:           it != map2.end(); ++it )
kpeter@159:       check(v1[i++] == *it, "Something is wrong with StoreBoolMap");
kpeter@159:   }
alpar@25: 
alpar@25:   return 0;
alpar@25: }