/******************************************************************************* * * TRIQS: a Toolbox for Research in Interacting Quantum Systems * * Copyright (C) 2011 by O. Parcollet * * TRIQS 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. * * TRIQS 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 * TRIQS. If not, see <http://www.gnu.org/licenses/>. * ******************************************************************************/ #ifndef TRIQS_ARRAYS_EXPRESSION_MAP_H #define TRIQS_ARRAYS_EXPRESSION_MAP_H #include "../impl/common.hpp" #include <functional> namespace triqs { namespace arrays { template<class F, int arity=F::arity> class map_impl; /** * Given a function f : arg_type -> result_type, map(f) is the function promoted to arrays * map(f) : array<arg_type, N, Opt> --> array<result_type, N, Opt> */ template<class F> map_impl<F,1> map (F const & f) { return map_impl<F,1>(f,true); } template<class F> map_impl<F,2> map2 (F const & f) { return map_impl<F,2>(f,true); } // ----------- implementation ------------------------------------- // NB The bool is to make constructor not ambiguous // clang on os X with lib++ has a pb otherwise (not clear what the pb is) template<class F> class map_impl<F,1> { F f; public : // map_impl(F const & f_):f(f_) {} map_impl(F const & f_, bool):f(f_) {} map_impl(F && f_, bool):f(f_) {} map_impl(map_impl const &) = default; //map_impl(map_impl &&) = default; template<typename A, typename Enable = void> class m_result; template<typename A> class m_result<A,typename boost::enable_if<ImmutableCuboidArray<A> >::type> : TRIQS_MODEL_CONCEPT(ImmutableCuboidArray) { public: typedef typename std::result_of<F(typename A::value_type)>::type value_type; typedef typename A::domain_type domain_type; A const & a; F f; m_result(F const & f_, A const & a_):a(a_),f(f_) {} domain_type domain() const { return a.domain(); } template<typename ... Args> value_type operator() (Args const & ... args) const { return f(a(args...)); } //value_type operator[] ( typename domain_type::index_value_type const & key) const { return f(a[key]); } friend std::ostream & operator<<(std::ostream & out, m_result const & x){ return out<<"lazy matrix resulting of a mapping";} }; template<typename A> class m_result<A,typename boost::enable_if<ImmutableMatrix<A> >::type> : TRIQS_MODEL_CONCEPT(ImmutableMatrix) { public: typedef typename std::result_of<F(typename A::value_type)>::type value_type; typedef typename A::domain_type domain_type; A const & a; F f; m_result(F const & f_, A const & a_):a(a_),f(f_) {} domain_type domain() const { return a.domain(); } size_t dim0() const { return a.dim0();} size_t dim1() const { return a.dim1();} template<typename ... Args> value_type operator() (Args const & ... args) const { return f(a(args...)); } //value_type operator[] ( typename domain_type::index_value_type const & key) const { return f(a[key]); } friend std::ostream & operator<<(std::ostream & out, m_result const & x){ return out<<"lazy matrix resulting of a mapping";} }; template<typename A> class m_result<A,typename boost::enable_if<ImmutableVector<A> >::type> : TRIQS_MODEL_CONCEPT(ImmutableVector) { public: typedef typename std::result_of<F(typename A::value_type)>::type value_type; typedef typename A::domain_type domain_type; A const & a; F f; m_result(F const & f_, A const & a_):a(a_),f(f_) {} domain_type domain() const { return a.domain(); } size_t size() const { return a.size();} template<typename ... Args> value_type operator() (Args const & ... args) const { return f(a(args...)); } //value_type operator[] ( typename domain_type::index_value_type const & key) const { return f(a[key]); } value_type operator() ( size_t i) const { return f(a(i)); } friend std::ostream & operator<<(std::ostream & out, m_result const & x){ return out<<"lazy matrix resulting of a mapping";} }; template<typename Sig> struct result; template<typename This, typename A> struct result<This(A)> { typedef m_result<A> type;}; template< class A > m_result<A> operator()(A const & a) const { //static_assert( (ImmutableArray<A>::value), "map : A does not model ImmutableArray"); return m_result<A>(f,a); } friend std::ostream & operator<<(std::ostream & out, map_impl const & x){ return out<<"map("<<"F"<<")";} }; // ----------- // TO DO : make this with preprocessor .... template<class F> class map_impl<F,2> { F f; public : map_impl(F const & f_, bool):f(f_) {} map_impl(F && f_, bool):f(f_) {} map_impl(map_impl const &) = default; //map_impl(map_impl &&) = default; template<class A, class B> class m_result : TRIQS_MODEL_CONCEPT(ImmutableArray) { static_assert( (std::is_same<typename A::domain_type, typename B::domain_type>::value), "type mismatch"); public: typedef typename std::result_of<F(typename A::value_type,typename B::value_type)>::type value_type; typedef typename A::domain_type domain_type; A const & a; B const & b; F f; m_result(F const & f_, A const & a_, B const & b_):a(a_),b(b_),f(f_) { if (a.domain() != b.domain()) TRIQS_RUNTIME_ERROR<<"map2 : domain mismatch"; } domain_type domain() const { return a.domain(); } template<typename ... Args> value_type operator() (Args const & ... args) const { return f(a(args...),b(args...)); } //value_type operator[] ( typename domain_type::index_value_type const & key) const { return f(a[key],b[key]); } friend std::ostream & operator<<(std::ostream & out, m_result const & x){ return out<<"lazy matrix resulting of a mapping";} }; template<typename Sig> struct result; template<typename This, typename A,typename B> struct result<This(A,B)> { typedef m_result<A,B> type;}; template< class A, class B > m_result<A,B> operator()(A const & a, B const & b) { static_assert( (ImmutableArray<A>::value), "map1 : A does not model ImmutableArray"); static_assert( (ImmutableArray<B>::value), "map1 : B does not model ImmutableArray"); return m_result<A,B>(f,a,b); } friend std::ostream & operator<<(std::ostream & out, map_impl const & x){ return out<<"map("<<"F"<<")";} }; }}//namespace triqs::arrays #endif