/* -*- C++ -*-

 * src/lemon/maps.h - Part of LEMON, a generic C++ optimization library

 *

 * Copyright (C) 2005 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.

 *

 */

#ifndef LEMON_MAPS_H

#define LEMON_MAPS_H

#include<math.h>

///\file

///\ingroup maps

///\brief Miscellaneous property maps

///

///\todo This file has the same name as the concept file in concept/,

/// and this is not easily detectable in docs...

#include <map>

namespace lemon {

  /// \addtogroup maps

  /// @{

  /// Base class of maps.

  /// Base class of maps.

  /// It provides the necessary <tt>typedef</tt>s required by the map concept.

  template<typename K, typename T>

  class MapBase

  {

  public:

    ///\e

    typedef K Key;

    ///\e

    typedef T Value;

  };

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

  /// 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 will sent to <tt>/dev/null</tt>...

  template<typename K, typename T>

  class NullMap : public MapBase<K,T>

  {

  public:

    /// Gives back a default constructed element.

    T operator[](const K&) const { return T(); }

    /// Absorbs the value.

    void set(const K&, const T&) {}

  };

  /// Constant map.

  /// This is a readable map which assigns a specified value to each key.

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

  /// \todo set could be used to set the value.

  template<typename K, typename T>

  class ConstMap : public MapBase<K,T>

  {

    T v;

  public:

    /// Default constructor

    /// The value of the map will be uninitialized. 

    /// (More exactly it will be default constructed.)

    ConstMap() {}

    ///\e

    /// \param _v The initial value of the map.

    ///

    ConstMap(const T &_v) : v(_v) {}

    T operator[](const K&) const { return v; }

    void set(const K&, const T&) {}

    template<typename T1>

    struct rebind {

      typedef ConstMap<K,T1> other;

    };

    template<typename T1>

    ConstMap(const ConstMap<K,T1> &, const T &_v) : v(_v) {}

  };

  ///Returns a \ref ConstMap class

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

  ///\relates ConstMap

  template<class V,class K> 

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

  {

    return ConstMap<V,K>(k);

  }

  //to document later

  template<typename T, T v>

  struct Const { };

  //to document later

  template<typename K, typename V, V v>

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

  {

  public:

    ConstMap() { }

    V operator[](const K&) const { return v; }

    void set(const K&, const V&) { }

  };

  /// \c std::map wrapper

  /// This is essentially a wrapper for \c std::map. With addition that

  /// you can specify a default value different from \c Value() .

  ///

  /// \todo Provide allocator parameter...

  template <typename K, typename T, typename Compare = std::less<K> >

  class StdMap : public std::map<K,T,Compare> {

    typedef std::map<K,T,Compare> parent;

    T v;

    typedef typename parent::value_type PairType;

  public:

    typedef K Key;

    typedef T Value;

    typedef T& Reference;

    typedef const T& ConstReference;

    StdMap() : v() {}

    /// Constructor with specified default value

    StdMap(const T& _v) : v(_v) {}

    /// \brief Constructs the map from an appropriate std::map.

    ///

    /// \warning Inefficient: copies the content of \c m !

    StdMap(const parent &m) : parent(m) {}

    /// \brief Constructs the map from an appropriate std::map, and explicitly

    /// specifies a default value.

    ///

    /// \warning Inefficient: copies the content of \c m !

    StdMap(const parent &m, const T& _v) : parent(m), v(_v) {}

    template<typename T1, typename Comp1>

    StdMap(const StdMap<Key,T1,Comp1> &m, const T &_v) { 

      //FIXME; 

    }

    Reference operator[](const Key &k) {

      return insert(PairType(k,v)).first -> second;

    }

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

      typename parent::iterator i = lower_bound(k);

      if (i == parent::end() || parent::key_comp()(k, (*i).first))

	return v;

      return (*i).second;

    }

    void set(const Key &k, const T &t) {

      parent::operator[](k) = t;

    }

    /// Changes the default value of the map.

    /// \return Returns the previous default value.

    ///

    /// \warning The value of some keys (which has already been queried, but

    /// the value has been unchanged from the default) may change!

    T setDefault(const T &_v) { T old=v; v=_v; return old; }

    template<typename T1>

    struct rebind {

      typedef StdMap<Key,T1,Compare> other;

    };

  };

  ///Convert the \c Value of a maps to another type.

  ///This \ref concept::ReadMap "read only map"

  ///converts the \c Value of a maps to type \c T.

  ///Its \c Value is inherited from \c M.

  ///

  ///Actually,

  ///\code

  ///  ConvertMap<X> sh(x,v);

  ///\endcode

  ///it is equivalent with

  ///\code

  ///  ConstMap<X::Key, X::Value> c_tmp(v);

  ///  AddMap<X, ConstMap<X::Key, X::Value> > sh(x,v);

  ///\endcode

  ///\bug wrong documentation

  template<class M, class T> 

  class ConvertMap

  {

    const M &m;

  public:

    typedef typename M::Key Key;

    typedef T Value;

    ///Constructor

    ///Constructor

    ///\param _m is the undelying map

    ///\param _v is the convert value

    ConvertMap(const M &_m) : m(_m) {};

    /// \brief The subscript operator.

    ///

    /// The subscript operator.

    /// \param edge The edge 

    /// \return The target of the edge 

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

  };

  ///Returns an \ref ConvertMap class

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

  ///\relates ConvertMap

  ///\todo The order of the template parameters are changed.

  template<class T, class M>

  inline ConvertMap<M,T> convertMap(const M &m) 

  {

    return ConvertMap<M,T>(m);

  }

  /// \brief Returns the source of the given edge.

  ///

  /// The SourceMap gives back the source Node of the given edge. 

  /// \author Balazs Dezso

  template <typename Graph>

  class SourceMap {

  public:

    typedef typename Graph::Node Value;

    typedef typename Graph::Edge Key;

    /// \brief Constructor

    ///

    /// Constructor

    /// \param _graph The graph that the map belongs to.

    SourceMap(const Graph& _graph) : graph(_graph) {}

    /// \brief The subscript operator.

    ///

    /// The subscript operator.

    /// \param edge The edge 

    /// \return The source of the edge 

    Value operator[](const Key& edge) {

      return graph.source(edge);

    }

  private:

    const Graph& graph;

  };

  /// \brief Returns a \ref SourceMap class

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

  /// \relates SourceMap

  template <typename Graph>

  inline SourceMap<Graph> sourceMap(const Graph& graph) {

    return SourceMap<Graph>(graph);

  } 

  /// \brief Returns the target of the given edge.

  ///

  /// The TargetMap gives back the target Node of the given edge. 

  /// \author Balazs Dezso

  template <typename Graph>

  class TargetMap {

  public:

    typedef typename Graph::Node Value;

    typedef typename Graph::Edge Key;

    /// \brief Constructor

    ///

    /// Constructor

    /// \param _graph The graph that the map belongs to.

    TargetMap(const Graph& _graph) : graph(_graph) {}

    /// \brief The subscript operator.

    ///

    /// The subscript operator.

    /// \param edge The edge 

    /// \return The target of the edge 

    Value operator[](const Key& key) {

      return graph.target(key);

    }

  private:

    const Graph& graph;

  };

  /// \brief Returns a \ref TargetMap class

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

  /// \relates TargetMap

  template <typename Graph>

  inline TargetMap<Graph> targetMap(const Graph& graph) {

    return TargetMap<Graph>(graph);

  }

  ///Sum of two maps

  ///This \ref concept::ReadMap "read only map" returns the sum of the two

  ///given maps. Its \c Key and \c Value will be inherited from \c M1.

  ///The \c Key and \c Value of M2 must be convertible to those of \c M1.

  template<class M1,class M2> 

  class AddMap

  {

    const M1 &m1;

    const M2 &m2;

  public:

    typedef typename M1::Key Key;

    typedef typename M1::Value Value;

    ///Constructor

    ///\e

    ///

    AddMap(const M1 &_m1,const M2 &_m2) : m1(_m1), m2(_m2) {};

    Value operator[](Key k) const {return m1[k]+m2[k];}

  };

  ///Returns an \ref AddMap class

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

  ///\todo How to call these type of functions?

  ///

  ///\relates AddMap

  ///\todo Wrong scope in Doxygen when \c \\relates is used

  template<class M1,class M2> 

  inline AddMap<M1,M2> addMap(const M1 &m1,const M2 &m2) 

  {

    return AddMap<M1,M2>(m1,m2);

  }

  ///Shift a maps with a constant.

  ///This \ref concept::ReadMap "read only map" returns the sum of the

  ///given map and a constant value.

  ///Its \c Key and \c Value is inherited from \c M.

  ///

  ///Actually,

  ///\code

  ///  ShiftMap<X> sh(x,v);

  ///\endcode

  ///it is equivalent with

  ///\code

  ///  ConstMap<X::Key, X::Value> c_tmp(v);

  ///  AddMap<X, ConstMap<X::Key, X::Value> > sh(x,v);

  ///\endcode

  template<class M> 

  class ShiftMap

  {

    const M &m;

    typename M::Value v;

  public:

    typedef typename M::Key Key;

    typedef typename M::Value Value;

    ///Constructor

    ///Constructor

    ///\param _m is the undelying map

    ///\param _v is the shift value

    ShiftMap(const M &_m,const Value &_v ) : m(_m), v(_v) {};

    Value operator[](Key k) const {return m[k]+v;}

  };

  ///Returns an \ref ShiftMap class

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

  ///\relates ShiftMap

  ///\todo A better name is required.

  template<class M> 

  inline ShiftMap<M> shiftMap(const M &m,const typename M::Value &v) 

  {

    return ShiftMap<M>(m,v);

  }

  ///Difference of two maps

  ///This \ref concept::ReadMap "read only map" returns the difference

  ///of the values returned by the two

  ///given maps. Its \c Key and \c Value will be inherited from \c M1.

  ///The \c Key and \c Value of \c M2 must be convertible to those of \c M1.

  template<class M1,class M2> 

  class SubMap

  {

    const M1 &m1;

    const M2 &m2;

  public:

    typedef typename M1::Key Key;

    typedef typename M1::Value Value;

    ///Constructor

    ///\e

    ///

    SubMap(const M1 &_m1,const M2 &_m2) : m1(_m1), m2(_m2) {};

    Value operator[](Key k) const {return m1[k]-m2[k];}

  };

  ///Returns a \ref SubMap class

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

  ///

  ///\relates SubMap

  template<class M1,class M2> 

  inline SubMap<M1,M2> subMap(const M1 &m1,const M2 &m2) 

  {

    return SubMap<M1,M2>(m1,m2);

  }

  ///Product of two maps

  ///This \ref concept::ReadMap "read only map" returns the product of the

  ///values returned by the two

  ///given

  ///maps. Its \c Key and \c Value will be inherited from \c M1.

  ///The \c Key and \c Value of \c M2 must be convertible to those of \c M1.

  template<class M1,class M2> 

  class MulMap

  {

    const M1 &m1;

    const M2 &m2;

  public:

    typedef typename M1::Key Key;

    typedef typename M1::Value Value;

    ///Constructor

    ///\e

    ///

    MulMap(const M1 &_m1,const M2 &_m2) : m1(_m1), m2(_m2) {};

    Value operator[](Key k) const {return m1[k]*m2[k];}

  };

  ///Returns a \ref MulMap class

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

  ///\relates MulMap

  template<class M1,class M2> 

  inline MulMap<M1,M2> mulMap(const M1 &m1,const M2 &m2) 

  {

    return MulMap<M1,M2>(m1,m2);

  }

  ///Scale a maps with a constant.

  ///This \ref concept::ReadMap "read only map" returns the value of the

  ///given map multipied with a constant value.

  ///Its \c Key and \c Value is inherited from \c M.

  ///

  ///Actually,

  ///\code

  ///  ScaleMap<X> sc(x,v);

  ///\endcode

  ///it is equivalent with

  ///\code

  ///  ConstMap<X::Key, X::Value> c_tmp(v);

  ///  MulMap<X, ConstMap<X::Key, X::Value> > sc(x,v);

  ///\endcode

  template<class M> 

  class ScaleMap

  {

    const M &m;

    typename M::Value v;

  public:

    typedef typename M::Key Key;

    typedef typename M::Value Value;

    ///Constructor

    ///Constructor

    ///\param _m is the undelying map

    ///\param _v is the scaling value

    ScaleMap(const M &_m,const Value &_v ) : m(_m), v(_v) {};

    Value operator[](Key k) const {return m[k]*v;}

  };

  ///Returns an \ref ScaleMap class

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

  ///\relates ScaleMap

  ///\todo A better name is required.

  template<class M> 

  inline ScaleMap<M> scaleMap(const M &m,const typename M::Value &v) 

  {

    return ScaleMap<M>(m,v);

  }

  ///Quotient of two maps

  ///This \ref concept::ReadMap "read only map" returns the quotient of the

  ///values returned by the two

  ///given maps. Its \c Key and \c Value will be inherited from \c M1.

  ///The \c Key and \c Value of \c M2 must be convertible to those of \c M1.

  template<class M1,class M2> 

  class DivMap

  {

    const M1 &m1;

    const M2 &m2;

  public:

    typedef typename M1::Key Key;

    typedef typename M1::Value Value;

    ///Constructor

    ///\e

    ///

    DivMap(const M1 &_m1,const M2 &_m2) : m1(_m1), m2(_m2) {};

    Value operator[](Key k) const {return m1[k]/m2[k];}

  };

  ///Returns a \ref DivMap class

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

  ///\relates DivMap

  template<class M1,class M2> 

  inline DivMap<M1,M2> divMap(const M1 &m1,const M2 &m2) 

  {

    return DivMap<M1,M2>(m1,m2);

  }

  ///Composition of two maps

  ///This \ref concept::ReadMap "read only map" returns the composition of

  ///two

  ///given maps. That is to say, if \c m1 is of type \c M1 and \c m2 is

  ///of \c M2,

  ///then for

  ///\code

  ///  ComposeMap<M1,M2> cm(m1,m2);

  ///\endcode

  /// <tt>cm[x]</tt> will be equal to <tt>m1[m2[x]]</tt>

  ///

  ///Its \c Key is inherited from \c M2 and its \c Value is from

  ///\c M1.

  ///The \c M2::Value must be convertible to \c M1::Key.

  ///\todo Check the requirements.

  template<class M1,class M2> 

  class ComposeMap

  {

    const M1 &m1;

    const M2 &m2;

  public:

    typedef typename M2::Key Key;

    typedef typename M1::Value Value;

    ///Constructor

    ///\e

    ///

    ComposeMap(const M1 &_m1,const M2 &_m2) : m1(_m1), m2(_m2) {};

    Value operator[](Key k) const {return m1[m2[k]];}

  };

  ///Returns a \ref ComposeMap class

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

  ///

  ///\relates ComposeMap

  template<class M1,class M2> 

  inline ComposeMap<M1,M2> composeMap(const M1 &m1,const M2 &m2) 

  {

    return ComposeMap<M1,M2>(m1,m2);

  }

  ///Combine of two maps using an STL (binary) functor.

  ///Combine of two maps using an STL (binary) functor.

  ///

  ///

  ///This \ref concept::ReadMap "read only map" takes to maps and a

  ///binary functor and returns the composition of

  ///two

  ///given maps unsing the functor. 

  ///That is to say, if \c m1 and \c m2 is of type \c M1 and \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>

  ///

  ///Its \c Key is inherited from \c M1 and its \c Value is \c V.

  ///The \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.

  ///\todo Check the requirements.

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

  class CombineMap

  {

    const M1 &m1;

    const M2 &m2;

    const F &f;

  public:

    typedef typename M1::Key Key;

    typedef V Value;

    ///Constructor

    ///\e

    ///

    CombineMap(const M1 &_m1,const M2 &_m2,const F &_f)

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

    Value operator[](Key k) const {return f(m1[k],m2[k]);}

  };

  ///Returns a \ref CombineMap class

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

  ///

  ///Only the first template parameter (the value type) must be given.

  ///

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

  ///\code

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

  ///\endcode

  ///is equivalent with

  ///\code

  ///addMap(m1,m2)

  ///\endcode

  ///

  ///\relates CombineMap

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

  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);

  }

  ///Negative value of a map

  ///This \ref concept::ReadMap "read only map" returns the negative

  ///value of the

  ///value returned by the

  ///given map. Its \c Key and \c Value will be inherited from \c M.

  ///The unary \c - operator must be defined for \c Value, of course.

  template<class M> 

  class NegMap

  {

    const M &m;

  public:

    typedef typename M::Key Key;

    typedef typename M::Value Value;

    ///Constructor

    ///\e

    ///

    NegMap(const M &_m) : m(_m) {};

    Value operator[](Key k) const {return -m[k];}

  };

  ///Returns a \ref NegMap class

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

  ///\relates NegMap

  template<class M> 

  inline NegMap<M> negMap(const M &m) 

  {

    return NegMap<M>(m);

  }

  ///Absolute value of a map

  ///This \ref concept::ReadMap "read only map" returns the absolute value

  ///of the

  ///value returned by the

  ///given map. Its \c Key and \c Value will be inherited

  ///from <tt>M</tt>. <tt>Value</tt>

  ///must be comparable to <tt>0</tt> and the unary <tt>-</tt>

  ///operator must be defined for it, of course.

  ///

  ///\bug We need a unified way to handle the situation below:

  ///\code

  ///  struct _UnConvertible {};

  ///  template<class A> inline A t_abs(A a) {return _UnConvertible();}

  ///  template<> inline int t_abs<>(int n) {return abs(n);}

  ///  template<> inline long int t_abs<>(long int n) {return labs(n);}

  ///  template<> inline long long int t_abs<>(long long int n) {return ::llabs(n);}

  ///  template<> inline float t_abs<>(float n) {return fabsf(n);}

  ///  template<> inline double t_abs<>(double n) {return fabs(n);}

  ///  template<> inline long double t_abs<>(long double n) {return fabsl(n);}

  ///\endcode

  template<class M> 

  class AbsMap

  {

    const M &m;

  public:

    typedef typename M::Key Key;

    typedef typename M::Value Value;

    ///Constructor

    ///\e

    ///

    AbsMap(const M &_m) : m(_m) {};

    Value operator[](Key k) const {Value tmp=m[k]; return tmp>=0?tmp:-tmp;}

  };

  ///Returns a \ref AbsMap class

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

  ///\relates AbsMap

  template<class M> 

  inline AbsMap<M> absMap(const M &m) 

  {

    return AbsMap<M>(m);

  }

  ///Converts an STL style functor to a a map

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

  ///of a

  ///given map.

  ///

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

  ///\c Key and \c Value. They must be given explicitely

  ///because a functor does not provide such typedefs.

  ///

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

  template<class K,class V,class F> 

  class FunctorMap

  {

    const F &f;

  public:

    typedef K Key;

    typedef V Value;

    ///Constructor

    ///\e

    ///

    FunctorMap(const F &_f) : f(_f) {};

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

  };

  ///Returns a \ref FunctorMap class

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

  ///

  ///The third template parameter isn't necessary to be given.

  ///\relates FunctorMap

  template<class K,class V, class F>

  inline FunctorMap<K,V,F> functorMap(const F &f) 

  {

    return FunctorMap<K,V,F>(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 ususal \ref concept::ReadMap "readable map", i.e

  ///<tt>operator[]</tt> and the \c Key and \c Value typedefs also exist.

  template<class M> 

  class MapFunctor

  {

    const M &m;

  public:

    typedef typename M::Key argument_type;

    typedef typename M::Value result_type;

    typedef typename M::Key Key;

    typedef typename M::Value Value;

    ///Constructor

    ///\e

    ///

    MapFunctor(const M &_m) : m(_m) {};

    ///Returns a value of the map

    ///\e

    ///

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

    ///\e

    ///

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

  };

  ///Returns a \ref MapFunctor class

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

  ///\relates MapFunctor

  template<class M> 

  inline MapFunctor<M> mapFunctor(const M &m) 

  {

    return MapFunctor<M>(m);

  }

  ///Apply all map setting operations to two maps

  ///This map has two \ref concept::WriteMap "writable map"

  ///parameters and each write request will be passed to both of them.

  ///If \c M1 is also \ref concept::ReadMap "readable",

  ///then the read operations will return the

  ///corresponding values of \c M1.

  ///

  ///The \c Key and \c Value will be inherited from \c M1.

  ///The \c Key and \c Value of M2 must be convertible from those of \c M1.

  template<class M1,class M2> 

  class ForkMap

  {

    const M1 &m1;

    const M2 &m2;

  public:

    typedef typename M1::Key Key;

    typedef typename M1::Value Value;

    ///Constructor

    ///\e

    ///

    ForkMap(const M1 &_m1,const M2 &_m2) : m1(_m1), m2(_m2) {};

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

    void set(Key k,const Value &v) {m1.set(k,v); m2.set(k,v);}

  };

  ///Returns an \ref ForkMap class

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

  ///\todo How to call these type of functions?

  ///

  ///\relates ForkMap

  ///\todo Wrong scope in Doxygen when \c \\relates is used

  template<class M1,class M2> 

  inline ForkMap<M1,M2> forkMap(const M1 &m1,const M2 &m2) 

  {

    return ForkMap<M1,M2>(m1,m2);

  }

  /// @}

}

#endif // LEMON_MAPS_H