map.hh

// This file is a part of Aurelia.
// Copyright (C) 2010  Valentin David
// Copyright (C) 2010  University of Bergen
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program.  If not, see <http://www.gnu.org/licenses/>.
#ifndef __MAP_HH
# define __MAP_HH

# include "../strategies.hh"
# include <bitset>

namespace aurelia {

  template <typename S>
  struct map_strategy {
    S s;
    map_strategy(const S& s): s(s) {
    }

    template <int N>
    std::bitset<N> operator()(std::bitset<N> t) const {
      typedef typename
        strategy_concept<typename strategy_model<S, typename std::bitset<N>
                                                 ::value_type>::model>
        ::check require;

      for (unsigned i = 0; i < N; ++i) {
        t[i] = s(t[i]);
      }
      return t;
    }

    template <typename T,
              typename M =
              typename strategy_model<S, typename T::value_type>::model>
    T operator()(const T& t) const {
      typedef typename strategy_concept<M>::check require;

      T ret;
      for (typename T::const_iterator i = t.begin(); i != t.end(); ++i)
        ret.insert(ret.end(), s(*i));
      return ret;
    }

    template <typename Pool,
              typename M = typename
              strategy_model<S, untyped_term<Pool> >::model>
    untyped_term<Pool> operator()(const untyped_term<Pool>& t) const {
      typedef typename strategy_concept<M>::check require;

      typedef untyped_term<Pool> T;
      typedef untyped_constructor<Pool> C;

      try {
        if (t.constructor() != C::AS_EMPTY_LIST)
          throw failure();
        return t;
      } catch (failure) {
        if (t.constructor() == C::AS_LIST) {
          T a = s(t[0]);
          T b = (*this)(t[1]);
          return T(C::AS_LIST, (a, b));
        } else
          throw failure();
      }
    }
  };

  template <typename S>
  map_strategy<S> map(const S& s) {
    return map_strategy<S>(s);
  }

}

#endif