RhodeCode

#include <functional>

#include <vector>

#include <lemon/bits/utility.h>

#include <lemon/bits/traits.h>

///\file

///\ingroup maps

///\brief Miscellaneous property maps

#include <map>

namespace lemon {

  /// \addtogroup maps

  /// @{

  /// Base class of maps.

  /// Base class of maps. It provides the necessary type definitions

  /// required by the map %concepts.

  template<typename K, typename V>

  class MapBase {

  public:

    /// \biref The key type of the map.

    typedef K Key;

    /// \brief The value type of the map.

    /// (The type of objects associated with the keys).

    typedef V Value;

  };

  /// Null map. (a.k.a. DoNothingMap)

  /// This map can be used if you have to provide a map only for

  /// its type definitions, or if you have to provide a writable map,

  /// but data written to it is not required (i.e. it will be sent to

  /// <tt>/dev/null</tt>).

  /// It conforms the \ref concepts::ReadWriteMap "ReadWriteMap" concept.

  ///

  /// \sa ConstMap

  template<typename K, typename V>

  class NullMap : public MapBase<K, V> {

  public:

    typedef MapBase<K, V> Parent;

    typedef typename Parent::Key Key;

    typedef typename Parent::Value Value;

    /// Gives back a default constructed element.

    Value operator[](const Key&) const { return Value(); }

    /// Absorbs the value.

    void set(const Key&, const Value&) {}

  };

  /// Returns a \ref NullMap class

  /// This function just returns a \ref NullMap class.

  /// \relates NullMap

  template <typename K, typename V>

  NullMap<K, V> nullMap() {

    return NullMap<K, V>();

  }

  /// Constant map.

  /// This \ref concepts::ReadMap "readable map" assigns a specified

  /// value to each key.

  ///

  /// In other aspects it is equivalent to \ref NullMap.

  /// So it conforms the \ref concepts::ReadWriteMap "ReadWriteMap"

  /// concept, but it absorbs the data written to it.

  ///

  /// The simplest way of using this map is through the constMap()

  /// function.

  ///

  /// \sa NullMap

  /// \sa IdentityMap

  template<typename K, typename V>

  class ConstMap : public MapBase<K, V> {

  private:

    V _value;

  public:

    typedef MapBase<K, V> Parent;

    typedef typename Parent::Key Key;

    typedef typename Parent::Value Value;

    /// Default constructor

    /// Default constructor.

    /// The value of the map will be default constructed.

    ConstMap() {}

    /// Constructor with specified initial value

    /// Constructor with specified initial value.

    ConstMap(const Value &v) : _value(v) {}

    /// Gives back the specified value.

    Value operator[](const Key&) const { return _value; }

    /// Absorbs the value.

    void set(const Key&, const Value&) {}

    /// Sets the value that is assigned to each key.

    void setAll(const Value &v) {

      _value = v;

    }

    template<typename V1>

    ConstMap(const ConstMap<K, V1> &, const Value &v) : _value(v) {}

  };

  /// Returns a \ref ConstMap class

  /// This function just returns a \ref ConstMap class.

  /// \relates ConstMap

  template<typename K, typename V>

  inline ConstMap<K, V> constMap(const V &v) {

    return ConstMap<K, V>(v);

  }

  template<typename T, T v>

  struct Const {};

  /// Constant map with inlined constant value.

  /// This \ref concepts::ReadMap "readable map" assigns a specified

  /// value to each key.

  ///

  /// In other aspects it is equivalent to \ref NullMap.

  /// So it conforms the \ref concepts::ReadWriteMap "ReadWriteMap"

  /// concept, but it absorbs the data written to it.

  ///

  /// The simplest way of using this map is through the constMap()

  /// function.

  ///

  /// \sa NullMap

  /// \sa IdentityMap

  template<typename K, typename V, V v>

  class ConstMap<K, Const<V, v> > : public MapBase<K, V> {

  public:

    typedef MapBase<K, V> Parent;

    typedef typename Parent::Key Key;

    typedef typename Parent::Value Value;

    /// Constructor.

    ConstMap() {}

    /// Gives back the specified value.

    Value operator[](const Key&) const { return v; }

    /// Absorbs the value.

    void set(const Key&, const Value&) {}

  };

  /// Returns a \ref ConstMap class with inlined constant value

  /// This function just returns a \ref ConstMap class with inlined

  /// constant value.

  /// \relates ConstMap

  template<typename K, typename V, V v>

  inline ConstMap<K, Const<V, v> > constMap() {

    return ConstMap<K, Const<V, v> >();

  }

  /// Identity map.

  /// This \ref concepts::ReadMap "read-only map" gives back the given

  /// key as value without any modification.

  ///

  /// \sa ConstMap

  template <typename T>

  class IdentityMap : public MapBase<T, T> {

  public:

    typedef MapBase<T, T> Parent;

    typedef typename Parent::Key Key;

    typedef typename Parent::Value Value;

    /// Gives back the given value without any modification.

    Value operator[](const Key &k) const {

      return k;

    }

  };

  /// Returns an \ref IdentityMap class

  /// This function just returns an \ref IdentityMap class.

  /// \relates IdentityMap

  template<typename T>

  inline IdentityMap<T> identityMap() {

    return IdentityMap<T>();

  }

  /// \brief Map for storing values for integer keys from the range

  /// <tt>[0..size-1]</tt>.

  ///

  /// This map is essentially a wrapper for \c std::vector. It assigns

  /// values to integer keys from the range <tt>[0..size-1]</tt>.

  /// It can be used with some data structures, for example

  /// \ref UnionFind, \ref BinHeap, when the used items are small

  /// integers. This map conforms the \ref concepts::ReferenceMap

  /// "ReferenceMap" concept.

  ///

  /// The simplest way of using this map is through the rangeMap()

  /// function.

  template <typename V>

  class RangeMap : public MapBase<int, V> {

    template <typename V1>

    friend class RangeMap;

  private:

    typedef std::vector<V> Vector;

    Vector _vector;

  /// binary functor and returns the combination of the two given maps

  /// using the functor.

  /// That is to say, if \c m1 is of type \c M1 and \c m2 is of \c M2

  /// and \c f is of \c F, then for

  /// \code

  ///   CombineMap<M1,M2,F,V> cm(m1,m2,f);

  /// \endcode

  /// <tt>cm[x]</tt> will be equal to <tt>f(m1[x],m2[x])</tt>.

  ///

  /// The \c Key type of the map is inherited from \c M1 (\c M1::Key

  /// must be convertible to \c M2::Key) and the \c Value type is \c V.

  /// \c M2::Value and \c M1::Value must be convertible to the

  /// corresponding input parameter of \c F and the return type of \c F

  /// must be convertible to \c V.

  ///

  /// The simplest way of using this map is through the combineMap()

  /// function.

  ///

  /// \sa ComposeMap

  ///

  /// \todo Check the requirements.

  template<typename M1, typename M2, typename F,

	   typename V = typename F::result_type>

  class CombineMap : public MapBase<typename M1::Key, V> {

    const M1 &_m1;

    const M2 &_m2;

    F _f;

  public:

    typedef MapBase<typename M1::Key, V> Parent;

    typedef typename Parent::Key Key;

    typedef typename Parent::Value Value;

    /// Constructor

    CombineMap(const M1 &m1, const M2 &m2, const F &f = F())

      : _m1(m1), _m2(m2), _f(f) {}

    /// \e

    Value operator[](const Key &k) const { return _f(_m1[k],_m2[k]); }

  };

  /// Returns a \ref CombineMap class

  /// This function just returns a \ref CombineMap class.

  ///

  /// For example, if \c m1 and \c m2 are both maps with \c double

  /// values, then

  /// \code

  ///   combineMap(m1,m2,std::plus<double>())

  /// \endcode

  /// is equivalent to

  /// \code

  ///   addMap(m1,m2)

  /// \endcode

  ///

  /// This function is specialized for adaptable binary function

  /// classes and C++ functions.

  ///

  /// \relates CombineMap

  template<typename M1, typename M2, typename F, typename V>

  inline CombineMap<M1, M2, F, V>

  combineMap(const M1 &m1, const M2 &m2, const F &f) {

    return CombineMap<M1, M2, F, V>(m1,m2,f);

  }

  template<typename M1, typename M2, typename F>

  inline CombineMap<M1, M2, F, typename F::result_type>

  combineMap(const M1 &m1, const M2 &m2, const F &f) {

    return combineMap<M1, M2, F, typename F::result_type>(m1,m2,f);

  }

  template<typename M1, typename M2, typename K1, typename K2, typename V>

  inline CombineMap<M1, M2, V (*)(K1, K2), V>

  combineMap(const M1 &m1, const M2 &m2, V (*f)(K1, K2)) {

    return combineMap<M1, M2, V (*)(K1, K2), V>(m1,m2,f);

  }

  /// Converts an STL style (unary) functor to a map

  /// This \ref concepts::ReadMap "read-only map" returns the value

  /// of a given functor. Actually, it just wraps the functor and

  /// provides the \c Key and \c Value typedefs.

  ///

  /// Template parameters \c K and \c V will become its \c Key and

  /// \c Value. In most cases they have to be given explicitly because

  /// a functor typically does not provide \c argument_type and

  /// \c result_type typedefs.

  /// Parameter \c F is the type of the used functor.

  ///

  /// The simplest way of using this map is through the functorToMap()

  /// function.

  ///

  /// \sa MapToFunctor

  template<typename F,

	   typename K = typename F::argument_type,

	   typename V = typename F::result_type>

  class FunctorToMap : public MapBase<K, V> {

  public:

    typedef MapBase<K, V> Parent;

    typedef typename Parent::Key Key;

    typedef typename Parent::Value Value;

    /// Constructor

    FunctorToMap(const F &f = F()) : _f(f) {}

    /// \e

    Value operator[](const Key &k) const { return _f(k); }

  };

  /// Returns a \ref FunctorToMap class

  /// This function just returns a \ref FunctorToMap class.

  ///

  /// This function is specialized for adaptable binary function

  /// classes and C++ functions.

  ///

  /// \relates FunctorToMap

  template<typename K, typename V, typename F>

  inline FunctorToMap<F, K, V> functorToMap(const F &f) {

    return FunctorToMap<F, K, V>(f);

  }

  template <typename F>

  inline FunctorToMap<F, typename F::argument_type, typename F::result_type>

    functorToMap(const F &f)

  {

    return FunctorToMap<F, typename F::argument_type,

      typename F::result_type>(f);

  }

  template <typename K, typename V>

  inline FunctorToMap<V (*)(K), K, V> functorToMap(V (*f)(K)) {

    return FunctorToMap<V (*)(K), K, V>(f);

  }

  /// Converts a map to an STL style (unary) functor

  /// This class converts a map to an STL style (unary) functor.

  /// That is it provides an <tt>operator()</tt> to read its values.

  ///

  /// For the sake of convenience it also works as a usual

  /// \ref concepts::ReadMap "readable map", i.e. <tt>operator[]</tt>

  /// and the \c Key and \c Value typedefs also exist.

  ///

  /// The simplest way of using this map is through the mapToFunctor()

  /// function.

  ///

  ///\sa FunctorToMap

  template <typename M>

  class MapToFunctor : public MapBase<typename M::Key, typename M::Value> {

    const M &_m;

  public:

    typedef MapBase<typename M::Key, typename M::Value> Parent;

    typedef typename Parent::Key Key;

    typedef typename Parent::Value Value;

    typedef typename Parent::Key argument_type;

    typedef typename Parent::Value result_type;

    /// Constructor

    MapToFunctor(const M &m) : _m(m) {}

    /// \e

    Value operator()(const Key &k) const { return _m[k]; }

    /// \e

    Value operator[](const Key &k) const { return _m[k]; }

  };

  /// Returns a \ref MapToFunctor class

  /// This function just returns a \ref MapToFunctor class.

  /// \relates MapToFunctor

  template<typename M>

  inline MapToFunctor<M> mapToFunctor(const M &m) {

    return MapToFunctor<M>(m);

  }

  /// \brief Map adaptor to convert the \c Value type of a map to

  /// another type using the default conversion.

  /// Map adaptor to convert the \c Value type of a \ref concepts::ReadMap

  /// "readable map" to another type using the default conversion.

  /// The \c Key type of it is inherited from \c M and the \c Value

  /// type is \c V.

  /// This type conforms the \ref concepts::ReadMap "ReadMap" concept.

  ///

  /// The simplest way of using this map is through the convertMap()

  /// function.

  template <typename M, typename V>

  class ConvertMap : public MapBase<typename M::Key, V> {

    const M &_m;

  public:

    typedef MapBase<typename M::Key, V> Parent;

/* -*- C++ -*-

 *

 * This file is a part of LEMON, a generic C++ optimization library

 *

 * Copyright (C) 2003-2008

 * Egervary Jeno Kombinatorikus Optimalizalasi Kutatocsoport

 * (Egervary Research Group on Combinatorial Optimization, EGRES).

 *

 * Permission to use, modify and distribute this software is granted

 * provided that this copyright notice appears in all copies. For

 * precise terms see the accompanying LICENSE file.

 *

 * This software is provided "AS IS" with no warranty of any kind,

 * express or implied, and with no claim as to its suitability for any

 * purpose.

 *

 */

#include <deque>

#include <set>

#include <lemon/concept_check.h>

#include <lemon/concepts/maps.h>

#include <lemon/maps.h>

#include "test_tools.h"

using namespace lemon;

using namespace lemon::concepts;

struct A {};

inline bool operator<(A, A) { return true; }

struct B {};

class C {

  int x;

public:

  C(int _x) : x(_x) {}

};

class F {

public:

  typedef A argument_type;

  typedef B result_type;

  B operator()(const A&) const { return B(); }

private:

  F& operator=(const F&);

};

int func(A) { return 3; }

int binc(int a, B) { return a+1; }

typedef ReadMap<A, double> DoubleMap;

typedef ReadWriteMap<A, double> DoubleWriteMap;

typedef ReferenceMap<A, double, double&, const double&> DoubleRefMap;

typedef ReadMap<A, bool> BoolMap;

typedef ReadWriteMap<A, bool> BoolWriteMap;

typedef ReferenceMap<A, bool, bool&, const bool&> BoolRefMap;

int main()

{

  // Map concepts

  checkConcept<ReadMap<A,B>, ReadMap<A,B> >();

  checkConcept<ReadMap<A,C>, ReadMap<A,C> >();

  checkConcept<WriteMap<A,B>, WriteMap<A,B> >();

  checkConcept<WriteMap<A,C>, WriteMap<A,C> >();

  checkConcept<ReadWriteMap<A,B>, ReadWriteMap<A,B> >();

  checkConcept<ReadWriteMap<A,C>, ReadWriteMap<A,C> >();

  checkConcept<ReferenceMap<A,B,B&,const B&>, ReferenceMap<A,B,B&,const B&> >();

  checkConcept<ReferenceMap<A,C,C&,const C&>, ReferenceMap<A,C,C&,const C&> >();

  // NullMap

  {

    checkConcept<ReadWriteMap<A,B>, NullMap<A,B> >();

    NullMap<A,B> map1;

    NullMap<A,B> map2 = map1;

    map1 = nullMap<A,B>();

  }

  // ConstMap

  {

    checkConcept<ReadWriteMap<A,B>, ConstMap<A,B> >();

    ConstMap<A,B> map1;

    ConstMap<A,B> map2(B());

    ConstMap<A,B> map3 = map1;

    map1 = constMap<A>(B());

    map1.setAll(B());

    checkConcept<ReadWriteMap<A,int>, ConstMap<A,int> >();

    check(constMap<A>(10)[A()] == 10, "Something is wrong with ConstMap");

    checkConcept<ReadWriteMap<A,int>, ConstMap<A,Const<int,10> > >();

  }

  // IdentityMap

  {

    checkConcept<ReadMap<A,A>, IdentityMap<A> >();

    IdentityMap<A> map1;

    IdentityMap<A> map2 = map1;

    map1 = identityMap<A>();

    checkConcept<ReadMap<double,double>, IdentityMap<double> >();

    check(identityMap<double>()[1.0] == 1.0 && identityMap<double>()[3.14] == 3.14,

          "Something is wrong with IdentityMap");

  }

  // RangeMap

  {

    checkConcept<ReferenceMap<int,B,B&,const B&>, RangeMap<B> >();

    RangeMap<B> map1;

    RangeMap<B> map2(10);

    RangeMap<B> map3(10,B());

    RangeMap<B> map4 = map1;

    RangeMap<B> map5 = rangeMap<B>();

    RangeMap<B> map6 = rangeMap<B>(10);

    RangeMap<B> map7 = rangeMap(10,B());

    checkConcept< ReferenceMap<int, double, double&, const double&>,

                  RangeMap<double> >();

    std::vector<double> v(10, 0);

    v[5] = 100;

    RangeMap<double> map8(v);

    RangeMap<double> map9 = rangeMap(v);

    check(map9.size() == 10 && map9[2] == 0 && map9[5] == 100,

          "Something is wrong with RangeMap");

  }

  // SparseMap

  {

    checkConcept<ReferenceMap<A,B,B&,const B&>, SparseMap<A,B> >();

    SparseMap<A,B> map1;

    SparseMap<A,B> map2(B());

    SparseMap<A,B> map3 = sparseMap<A,B>();

    SparseMap<A,B> map4 = sparseMap<A>(B());

    checkConcept< ReferenceMap<double, int, int&, const int&>,

                  SparseMap<double, int> >();

    std::map<double, int> m;

    SparseMap<double, int> map5(m);

    SparseMap<double, int> map6(m,10);

    SparseMap<double, int> map7 = sparseMap(m);

    SparseMap<double, int> map8 = sparseMap(m,10);

    check(map5[1.0] == 0 && map5[3.14] == 0 && map6[1.0] == 10 && map6[3.14] == 10,

          "Something is wrong with SparseMap");

    map5[1.0] = map6[3.14] = 100;

    check(map5[1.0] == 100 && map5[3.14] == 0 && map6[1.0] == 10 && map6[3.14] == 100,

          "Something is wrong with SparseMap");

  }

  // ComposeMap

  {

    typedef ComposeMap<DoubleMap, ReadMap<B,A> > CompMap;

    checkConcept<ReadMap<B,double>, CompMap>();

    CompMap map1(DoubleMap(),ReadMap<B,A>());

    CompMap map2 = composeMap(DoubleMap(), ReadMap<B,A>());

    SparseMap<double, bool> m1(false); m1[3.14] = true;

    RangeMap<double> m2(2); m2[0] = 3.0; m2[1] = 3.14;

    check(!composeMap(m1,m2)[0] && composeMap(m1,m2)[1], "Something is wrong with ComposeMap")

  }

  // CombineMap

  {

    typedef CombineMap<DoubleMap, DoubleMap, std::plus<double> > CombMap;

    checkConcept<ReadMap<A,double>, CombMap>();

    CombMap map1(DoubleMap(), DoubleMap());

    CombMap map2 = combineMap(DoubleMap(), DoubleMap(), std::plus<double>());

    check(combineMap(constMap<B,int,2>(), identityMap<B>(), &binc)[B()] == 3,

          "Something is wrong with CombineMap");

  }

  // FunctorToMap, MapToFunctor

  {

    checkConcept<ReadMap<A,B>, FunctorToMap<F,A,B> >();

    checkConcept<ReadMap<A,B>, FunctorToMap<F> >();

    FunctorToMap<F> map1;

    FunctorToMap<F> map2(F());

    B b = functorToMap(F())[A()];

    checkConcept<ReadMap<A,B>, MapToFunctor<ReadMap<A,B> > >();

    MapToFunctor<ReadMap<A,B> > map(ReadMap<A,B>());

    check(functorToMap(&func)[A()] == 3, "Something is wrong with FunctorToMap");

    check(mapToFunctor(constMap<A,int>(2))(A()) == 2, "Something is wrong with MapToFunctor");

    check(mapToFunctor(functorToMap(&func))(A()) == 3 && mapToFunctor(functorToMap(&func))[A()] == 3,

          "Something is wrong with FunctorToMap or MapToFunctor");