mirror of
https://github.com/triqs/dft_tools
synced 2024-12-26 22:33:48 +01:00
141 lines
6.7 KiB
C++
141 lines
6.7 KiB
C++
/*******************************************************************************
|
|
*
|
|
* 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
|
|
|