cnorxz/src/include/multi_array_operation.h

971 lines
27 KiB
C
Raw Normal View History

2017-02-16 11:20:40 +01:00
// -*- C++ -*-
#ifndef __multi_array_operation_h__
#define __multi_array_operation_h__
#include <cstdlib>
#include <tuple>
2017-03-22 21:51:54 +01:00
#include <cmath>
2017-08-30 17:56:38 +02:00
#include <map>
#include <utility>
2017-02-16 11:20:40 +01:00
#include "base_def.h"
2017-12-18 11:19:04 +01:00
#include "mbase_def.h"
2017-12-18 13:13:13 +01:00
#include "ranges/rheader.h"
#include "pack_num.h"
2017-02-16 11:20:40 +01:00
#include "arith.h"
#include "xfor/xfor.h"
2021-01-14 23:57:06 +01:00
#include "type_operations.h"
2017-02-16 11:20:40 +01:00
namespace MultiArrayTools
{
namespace
{
using namespace MultiArrayHelper;
}
2018-05-15 21:18:21 +02:00
template <typename T, class OperationClass>
class OperationBase
2017-08-11 11:30:27 +02:00
{
public:
2017-08-27 17:52:50 +02:00
2017-12-25 13:44:55 +01:00
OperationClass& THIS() { return static_cast<OperationClass&>(*this); }
const OperationClass& THIS() const { return static_cast<OperationClass const&>(*this); }
template <typename U, class Second>
auto operator+(const OperationBase<U,Second>& in) const;
2017-08-26 17:18:42 +02:00
template <typename U, class Second>
auto operator-(const OperationBase<U,Second>& in) const;
2017-08-26 17:18:42 +02:00
template <typename U, class Second>
auto operator*(const OperationBase<U,Second>& in) const;
2017-08-26 17:18:42 +02:00
template <typename U, class Second>
auto operator/(const OperationBase<U,Second>& in) const;
2017-11-05 18:46:38 +01:00
2017-11-02 21:20:31 +01:00
template <class IndexType>
2018-09-11 18:38:30 +02:00
auto c(const std::shared_ptr<IndexType>& ind) const
2017-11-02 21:20:31 +01:00
-> Contraction<T,OperationClass,IndexType>;
2018-09-11 18:38:30 +02:00
template <class... Indices>
auto sl(const std::shared_ptr<Indices>&... inds) const
-> ConstSlice<T,typename Indices::RangeType...>;
2018-09-14 17:50:19 +02:00
template <class... Indices>
2018-09-15 16:33:49 +02:00
auto slc(const std::shared_ptr<Indices>&... inds) const
2018-09-15 01:58:17 +02:00
-> SliceContraction<T,OperationClass,Indices...>;
template <class... Indices>
auto p(const std::shared_ptr<Indices>&... inds) const
-> ConstOperationRoot<T,typename Indices::RangeType...>;
template <class... Indices>
auto to(const std::shared_ptr<Indices>&... inds) const
-> MultiArray<T,typename Indices::RangeType...>;
2019-02-19 16:35:01 +01:00
template <class... Indices>
auto addto(const std::shared_ptr<Indices>&... inds) const
2019-02-19 16:20:38 +01:00
-> MultiArray<T,typename Indices::RangeType...>;
template <class... Indices>
auto pto(const std::shared_ptr<Indices>&... inds) const
-> MultiArray<T,typename Indices::RangeType...>;
template <class... Indices>
auto paddto(const std::shared_ptr<Indices>&... inds) const
-> MultiArray<T,typename Indices::RangeType...>;
template <typename R, class... Args> // Args = Operation Classes
auto a(const std::shared_ptr<function<R,T,typename Args::value_type...>>& ll, const Args&... args) const
-> Operation<R,function<R,T,typename Args::value_type...>,OperationClass, Args...>;
private:
2017-12-25 13:44:55 +01:00
friend OperationClass;
friend OperationTemplate<T,OperationClass>;
OperationBase() = default;
2017-08-11 11:30:27 +02:00
};
template <typename T, class OperationClass>
class OperationTemplate : public OperationBase<T,OperationClass>
{
/* empty per default; specialize if needed */
private:
OperationTemplate() = default;
friend OperationClass;
};
template <class Op>
size_t sumRootNum()
{
return typename Op::rootNum();
}
template <class Op1, class Op2, class... Ops>
size_t sumRootNum()
{
return typename Op1::rootNum() + sumRootNum<Op2,Ops...>();
}
template <size_t N>
struct RootSumN
{
template <class Op1, class... Ops>
struct rs
{
static constexpr size_t SIZE = Op1::SIZE + RootSumN<N-1>::template rs<Ops...>::SIZE;
};
template <class... Exprs>
static inline auto rootSteps(const std::tuple<Exprs...>& etp, std::intptr_t i)
{
return RootSumN<N-1>::rootSteps(etp,i).extend( std::get<N>(etp).rootSteps(i) );
}
template <class ExtType, class... Exprs>
static inline void exec( size_t start, ExtType last, std::tuple<Exprs...>& etp)
{
std::get<sizeof...(Exprs)-N-1>(etp)(start,last);
RootSumN<N-1>::exec(start,last.next(),etp);
}
template <class ExtType, class... Exprs>
static inline size_t get( ExtType last, const std::tuple<Exprs...>& etp)
{
std::get<sizeof...(Exprs)-N-1>(etp).get(last);
return RootSumN<N-1>::get(last.next(),etp);
}
template <class ExtType, class... Exprs>
static inline void set( ExtType last, std::tuple<Exprs...>& etp)
{
std::get<sizeof...(Exprs)-N-1>(etp).set(last);
RootSumN<N-1>::set(last.next(),etp);
}
};
template <>
struct RootSumN<0>
{
template <class Op1>
struct rs
{
static constexpr size_t SIZE = Op1::SIZE;
};
template <class... Exprs>
static inline auto rootSteps(const std::tuple<Exprs...>& etp, std::intptr_t i)
{
return std::get<0>(etp).rootSteps(i);
}
template <class ExtType, class... Exprs>
static inline void exec( size_t start, ExtType last, std::tuple<Exprs...>& etp)
{
std::get<sizeof...(Exprs)-1>(etp)(start,last);
}
template <class ExtType, class... Exprs>
static inline size_t get( ExtType last, const std::tuple<Exprs...>& etp)
{
std::get<sizeof...(Exprs)-1>(etp).get(last);
return 0;
}
template <class ExtType, class... Exprs>
static inline void set( ExtType last, std::tuple<Exprs...>& etp)
{
std::get<sizeof...(Exprs)-1>(etp).set(last);
}
};
template <class... Ops>
struct RootSum
{
static constexpr size_t SIZE = RootSumN<sizeof...(Ops)-1>::template rs<Ops...>::SIZE;
};
template <typename T>
struct SelfIdentity
{
2019-02-26 18:56:57 +01:00
static inline T& sapply(T& a, T b)
{
2019-02-14 14:39:59 +01:00
return a = b;
}
};
2019-01-15 17:41:43 +01:00
enum class OpIndexAff {
EXTERN = 0,
TARGET = 1
};
2021-01-14 14:43:09 +01:00
2021-01-18 19:00:12 +01:00
template <class T>
struct VType
{
typedef T type;
static constexpr size_t MULT = sizeof(type)/sizeof(T);
};
template <>
struct VType<double>
{
typedef v256 type;
static constexpr size_t MULT = sizeof(type)/sizeof(double);
};
template <template <typename...> class F,typename... Ts>
inline auto mkVFuncPtr(const std::shared_ptr<F<Ts...>>& f)
{
return std::shared_ptr<F<typename VType<Ts>::type...>>();
// empty, implement corresponding constructors...!!!
}
template <template <typename...> class F,typename... Ts>
inline auto mkVFunc(const F<Ts...>& f)
{
return F<typename VType<Ts>::type...>();
// empty, implement corresponding constructors...!!!
}
template <class F>
using VFunc = decltype(mkVFunc(std::declval<F>()));
2021-01-23 19:40:15 +01:00
template <typename T, class F>
2021-01-18 19:00:12 +01:00
struct IAccess
{
2021-01-23 19:40:15 +01:00
typedef T value_type;
typedef T in_type;
static constexpr size_t VSIZE = sizeof(value_type) / sizeof(in_type);
template <typename Op, class ExtType>
2021-01-18 19:00:12 +01:00
static inline void f(T*& t, size_t pos, const Op& op, ExtType e)
{
F::selfApply(t[pos],op.get(e));
}
};
2021-01-23 19:40:15 +01:00
template <typename T, class F>
2021-01-18 19:00:12 +01:00
struct IVAccess
{
2021-01-23 19:40:15 +01:00
typedef typename VType<T>::type value_type;
typedef T in_type;
static constexpr size_t VSIZE = sizeof(value_type) / sizeof(in_type);
template <typename Op, class ExtType>
2021-01-18 19:00:12 +01:00
static inline void f(T*& t, size_t pos, const Op& op, ExtType e)
{
2021-01-24 02:10:06 +01:00
//VCHECK(pos);
VFunc<F>::selfApply(*reinterpret_cast<value_type*>(t+pos),op.template vget<value_type>(e));
2021-01-18 19:00:12 +01:00
}
};
template <typename T>
using xxxplus = plus<T>;
2021-01-18 19:00:12 +01:00
template <typename T>
2021-01-23 19:40:15 +01:00
using IAssign = IAccess<T,identity<T>>;
2021-01-18 19:00:12 +01:00
template <typename T>
2021-01-23 19:40:15 +01:00
using IPlus = IAccess<T,plus<T>>;
2021-01-18 19:00:12 +01:00
2021-01-23 19:40:15 +01:00
template <typename T>
using IVAssign = IVAccess<T,identity<T>>;
2021-01-18 19:00:12 +01:00
2021-01-23 19:40:15 +01:00
template <typename T>
using IVPlus = IVAccess<T,plus<T>>;
2021-01-18 19:00:12 +01:00
/*
2021-01-14 16:26:53 +01:00
struct IAssign
{
template <typename T, typename Op, class ExtType>
static inline void f(T*& t, size_t pos, const Op& op, ExtType e)
{
t[pos] = op.get(e);
}
};
template <typename V>
struct IVAssign
{
2021-01-14 23:57:06 +01:00
template <typename T, typename Op, class ExtType>
static inline void f(T*& t, size_t pos, const Op& op, ExtType e)
2021-01-14 16:26:53 +01:00
{
2021-01-14 23:57:06 +01:00
reinterpret_cast<V*>(t)[pos] = op.template vget<V>(e);
2021-01-14 16:26:53 +01:00
}
};
2021-01-18 19:00:12 +01:00
2021-01-14 16:26:53 +01:00
struct IPlus
{
template <typename T, typename Op, class ExtType>
static inline void f(T*& t, size_t pos, const Op& op, ExtType e)
{
t[pos] += op.get(e);
}
};
template <typename V>
struct IVPlus
{
2021-01-14 23:57:06 +01:00
template <typename T, typename Op, class ExtType>
static inline void f(T*& t, size_t pos, const Op& op, ExtType e)
2021-01-14 16:26:53 +01:00
{
2021-01-14 23:57:06 +01:00
reinterpret_cast<V*>(t)[pos] += op.template vget<V>(e);
2021-01-14 16:26:53 +01:00
}
};
2021-01-18 19:00:12 +01:00
*/
2021-01-14 16:26:53 +01:00
2021-01-14 19:20:46 +01:00
template <typename T, class IOp, class Target, class OpClass, OpIndexAff OIA=OpIndexAff::EXTERN>
class AssignmentExpr : public ExpressionBase
{
private:
2021-01-14 19:20:46 +01:00
AssignmentExpr() = default;
Target mTar;
OpClass mSec;
T* mDataPtr;
public:
static constexpr size_t LAYER = 0;
2021-01-24 02:10:06 +01:00
static constexpr size_t NHLAYER = 0;
static constexpr size_t SIZE = Target::SIZE + OpClass::SIZE;
typedef decltype(mTar.rootSteps(0).extend( mSec.rootSteps(0) )) ExtType;
2021-01-14 19:20:46 +01:00
AssignmentExpr(T* dataPtr, const Target& tar, const OpClass& sec);
AssignmentExpr(const AssignmentExpr& in) = default;
AssignmentExpr(AssignmentExpr&& in) = default;
AssignmentExpr& operator=(const AssignmentExpr& in) = default;
AssignmentExpr& operator=(AssignmentExpr&& in) = default;
virtual std::shared_ptr<ExpressionBase> deepCopy() const override final
{
2021-01-14 19:20:46 +01:00
return std::make_shared<AssignmentExpr<T,IOp,Target,OpClass,OIA>>(*this);
}
inline void operator()(size_t start = 0);
inline void operator()(size_t start, ExtType last);
auto rootSteps(std::intptr_t iPtrNum = 0) const -> ExtType;
inline void operator()(size_t mlast, DExt last) override final;
inline DExt dRootSteps(std::intptr_t iPtrNum = 0) const override final;
inline DExt dExtension() const override final;
};
2021-01-14 19:20:46 +01:00
template <typename T, class Target, class OpClass, OpIndexAff OIA=OpIndexAff::EXTERN>
2021-01-18 19:00:12 +01:00
using AssignmentExpr2 = AssignmentExpr<T,IAssign<T>,Target,OpClass,OIA>;
2021-01-14 19:20:46 +01:00
template <typename T, class Target, class OpClass, OpIndexAff OIA=OpIndexAff::EXTERN>
2021-01-18 19:00:12 +01:00
using AddExpr = AssignmentExpr<T,IPlus<T>,Target,OpClass,OIA>;
2021-01-14 19:20:46 +01:00
template <typename T, class... Ops>
class MOp
{
private:
MOp() = default;
std::tuple<Ops...> mOps;
public:
static constexpr size_t LAYER = 0;
2021-01-24 02:10:06 +01:00
static constexpr size_t NHLAYER = 0;
static constexpr size_t SIZE = RootSum<Ops...>::SIZE;
typedef decltype(RootSumN<sizeof...(Ops)-1>::rootSteps(mOps,0) ) ExtType;
MOp(const Ops&... exprs);
MOp(const MOp& in) = default;
MOp(MOp&& in) = default;
MOp& operator=(const MOp& in) = default;
MOp& operator=(MOp&& in) = default;
inline size_t get(ExtType last) const;
2021-01-14 14:43:09 +01:00
template <typename V>
inline size_t vget(ExtType last) const { return get(last); }
inline MOp& set(ExtType last);
auto rootSteps(std::intptr_t iPtrNum = 0) const -> ExtType;
template <class Expr>
auto loop(Expr exp) const
-> decltype(PackNum<sizeof...(Ops)-1>::mkLoop( mOps, exp));
T* data() const { assert(0); return nullptr; }
};
template <class OpClass, class NextExpr>
class GetExpr : public ExpressionBase
{
private:
GetExpr() = default;
OpClass mSec;
NextExpr mNExpr;
public:
static constexpr size_t LAYER = 0;
2021-01-24 02:10:06 +01:00
static constexpr size_t NHLAYER = 0;
static constexpr size_t SIZE = OpClass::SIZE + NextExpr::SIZE;
typedef decltype(mSec.rootSteps(0).extend( mNExpr.rootSteps(0) ) ) ExtType;
GetExpr(const OpClass& sec, const NextExpr& nexpr);
GetExpr(const GetExpr& in) = default;
GetExpr(GetExpr&& in) = default;
GetExpr& operator=(const GetExpr& in) = default;
GetExpr& operator=(GetExpr&& in) = default;
virtual std::shared_ptr<ExpressionBase> deepCopy() const override final
{
return std::make_shared<GetExpr<OpClass,NextExpr>>(*this);
}
inline void operator()(size_t start = 0);
inline void get(ExtType last);
2021-01-14 14:43:09 +01:00
template <typename V>
inline void vget(ExtType last) { get(last); }
inline void operator()(size_t start, ExtType last);
auto rootSteps(std::intptr_t iPtrNum = 0) const -> ExtType;
inline void operator()(size_t mlast, DExt last) override final;
inline DExt dRootSteps(std::intptr_t iPtrNum = 0) const override final;
inline DExt dExtension() const override final;
};
template <class OpClass, class NextExpr>
auto mkGetExpr(const OpClass& op, const NextExpr& nexpr)
{
return GetExpr<OpClass,NextExpr>(op, nexpr);
}
template <typename T, class... Ops>
auto mkMOp(const Ops&... exprs)
{
return MOp<T,Ops...>(exprs...);
}
2017-08-10 15:12:26 +02:00
template <typename T, class... Ranges>
class ConstOperationRoot : public OperationTemplate<T,ConstOperationRoot<T,Ranges...> >
{
public:
2017-08-27 17:52:50 +02:00
typedef T value_type;
typedef OperationBase<T,ConstOperationRoot<T,Ranges...> > OT;
typedef ContainerRange<T,Ranges...> CRange;
typedef ContainerIndex<T,typename Ranges::IndexType...> IndexType;
static constexpr size_t SIZE = 1;
2019-02-27 13:37:53 +01:00
static constexpr bool CONT = true;
2021-01-18 19:00:12 +01:00
static constexpr bool VABLE = true;
ConstOperationRoot(const MultiArrayBase<T,Ranges...>& ma,
2017-08-10 15:12:26 +02:00
const std::shared_ptr<typename Ranges::IndexType>&... indices);
ConstOperationRoot(std::shared_ptr<MultiArrayBase<T,Ranges...> > maptr,
const std::shared_ptr<typename Ranges::IndexType>&... indices);
ConstOperationRoot(const T* data, const IndexType& ind);
2018-02-12 18:26:56 +01:00
template <class ET>
2019-02-27 13:37:53 +01:00
inline const T& get(ET pos) const;
2021-01-14 14:43:09 +01:00
template <typename V, class ET>
inline const V& vget(ET pos) const;
2018-09-15 01:58:17 +02:00
template <class ET>
2019-02-26 18:56:57 +01:00
inline ConstOperationRoot& set(ET pos);
2018-09-15 01:58:17 +02:00
2020-07-09 17:37:28 +02:00
MExt<None> rootSteps(std::intptr_t iPtrNum = 0) const; // nullptr for simple usage with decltype
template <class Expr>
Expr loop(Expr exp) const;
2018-09-11 18:38:30 +02:00
const T* data() const;
private:
2017-12-25 13:44:55 +01:00
2018-02-13 15:38:03 +01:00
const T* mDataPtr;
2019-02-26 18:56:57 +01:00
const T* mOrigDataPtr;
IndexType mIndex;
2018-09-16 15:53:56 +02:00
std::shared_ptr<MultiArrayBase<T,Ranges...> > mMaPtr; // never remove this ptr, otherwise we lose temporary container instances!
};
template <typename T, class Op>
class StaticCast : public OperationTemplate<T,StaticCast<T,Op> >
{
private:
2019-02-26 18:56:57 +01:00
Op mOp;
public:
typedef T value_type;
typedef OperationBase<T,StaticCast<T,Op> > OT;
typedef typename Op::CRange CRange;
typedef typename Op::IndexType IndexType;
static constexpr size_t SIZE = Op::SIZE;
2019-02-27 13:37:53 +01:00
static constexpr bool CONT = false;
2021-01-18 19:00:12 +01:00
static constexpr bool VABLE = false;
StaticCast(const Op& op);
template <class ET>
inline T get(ET pos) const;
2021-01-14 14:43:09 +01:00
template <typename V, class ET>
inline V vget(ET pos) const;
2018-09-15 01:58:17 +02:00
template <class ET>
2019-02-26 18:56:57 +01:00
inline StaticCast& set(ET pos);
2018-09-15 01:58:17 +02:00
auto rootSteps(std::intptr_t iPtrNum = 0) const
-> decltype(mOp.rootSteps(iPtrNum));
template <class Expr>
Expr loop(Expr exp) const;
};
template <typename T, class Op>
StaticCast<T,Op> staticcast(const Op& op)
{
return StaticCast<T,Op>(op);
}
template <class Range>
class MetaOperationRoot : public OperationTemplate<typename Range::MetaType,
MetaOperationRoot<Range> >
{
public:
typedef typename Range::IndexType IndexType;
typedef typename IndexType::MetaType value_type;
typedef OperationBase<value_type,MetaOperationRoot<Range> > OT;
static constexpr size_t SIZE = 1;
2019-02-27 13:37:53 +01:00
static constexpr bool CONT = false;
2021-01-18 19:00:12 +01:00
static constexpr bool VABLE = false;
2019-02-27 13:37:53 +01:00
MetaOperationRoot(const std::shared_ptr<IndexType>& ind);
template <class ET>
inline value_type get(ET pos) const;
2021-01-14 14:43:09 +01:00
template <typename V, class ET>
inline V vget(ET pos) const;
2018-09-15 01:58:17 +02:00
template <class ET>
2019-02-26 18:56:57 +01:00
inline MetaOperationRoot& set(ET pos);
2018-09-15 01:58:17 +02:00
2020-07-09 17:37:28 +02:00
MExt<None> rootSteps(std::intptr_t iPtrNum = 0) const; // nullptr for simple usage with decltype
template <class Expr>
Expr loop(Expr exp) const;
private:
2018-11-26 18:23:38 +01:00
mutable IndexType mWorkIndex;
std::shared_ptr<IndexType> mIndex;
};
2017-08-10 15:12:26 +02:00
template <typename T, class... Ranges>
2018-01-05 13:56:16 +01:00
class OperationRoot : public OperationTemplate<T,OperationRoot<T,Ranges...> >
{
public:
2017-08-27 17:52:50 +02:00
typedef T value_type;
typedef OperationBase<T,OperationRoot<T,Ranges...> > OT;
typedef ContainerRange<T,Ranges...> CRange;
typedef ContainerIndex<T,typename Ranges::IndexType...> IndexType;
static constexpr size_t SIZE = 1;
2019-02-27 13:37:53 +01:00
static constexpr bool CONT = true;
2021-01-18 19:00:12 +01:00
static constexpr bool VABLE = true;
private:
T* mDataPtr;
T* mOrigDataPtr;
IndexType mIndex;
public:
OperationRoot(MutableMultiArrayBase<T,Ranges...>& ma,
2017-08-10 15:12:26 +02:00
const std::shared_ptr<typename Ranges::IndexType>&... indices);
2017-08-11 11:30:27 +02:00
OperationRoot(MutableMultiArrayBase<T,Ranges...>& ma,
const std::tuple<std::shared_ptr<typename Ranges::IndexType>...>& indices);
OperationRoot(T* data, const IndexType& ind);
2021-01-15 01:05:58 +01:00
template <class IOp, class OpClass>
auto asx(const OpClass& in) const
-> decltype(mIndex.ifor(1,in.loop(AssignmentExpr<T,IOp,OperationRoot<T,Ranges...>,OpClass,OpIndexAff::TARGET>
2021-01-23 19:40:15 +01:00
(mOrigDataPtr,*this,in))).template vec<IOp::VSIZE>());
2019-01-15 17:41:43 +01:00
2021-01-15 01:05:58 +01:00
template <class IOp, class OpClass>
auto asxExpr(const OpClass& in) const
-> decltype(in.loop(AssignmentExpr<T,IOp,OperationRoot<T,Ranges...>,OpClass>(mOrigDataPtr,*this,in)));
2021-01-15 01:05:58 +01:00
template <class IOp, class OpClass, class Index>
auto asx(const OpClass& in, const std::shared_ptr<Index>& i) const
-> decltype(i->ifor(1,in.loop(AssignmentExpr<T,IOp,OperationRoot<T,Ranges...>,OpClass>
2021-01-23 19:40:15 +01:00
(mOrigDataPtr,*this,in))).template vec<IOp::VSIZE>());
template <class OpClass>
2021-01-15 01:05:58 +01:00
auto assign(const OpClass& in) const
2021-01-18 19:00:12 +01:00
-> decltype(this->template asx<IAssign<T>>(in));
2021-01-15 01:05:58 +01:00
template <class OpClass>
auto assignExpr(const OpClass& in) const
2021-01-18 19:00:12 +01:00
-> decltype(this->template asxExpr<IAssign<T>>(in));
2021-01-15 01:05:58 +01:00
template <class OpClass, class Index>
auto assign(const OpClass& in, const std::shared_ptr<Index>& i) const
2021-01-18 19:00:12 +01:00
-> decltype(this->template asx<IAssign<T>>(in,i));
2021-01-15 01:05:58 +01:00
template <class OpClass>
2019-05-07 16:40:40 +02:00
auto plus(const OpClass& in) const
2021-01-18 19:00:12 +01:00
-> decltype(this->template asx<IPlus<T>>(in));
2019-11-13 14:08:15 +01:00
template <class OpClass, class Index>
auto plus(const OpClass& in, const std::shared_ptr<Index>& i) const
2021-01-18 19:00:12 +01:00
-> decltype(this->template asx<IPlus<T>>(in,i));
2019-11-13 14:08:15 +01:00
2019-02-26 18:56:57 +01:00
template <class OpClass>
OperationRoot& operator=(const OpClass& in);
template <class OpClass>
OperationRoot& operator+=(const OpClass& in);
OperationRoot& operator=(const OperationRoot& in);
ParallelOperationRoot<T,Ranges...> par();
2019-01-15 17:41:43 +01:00
2018-02-12 18:26:56 +01:00
template <class ET>
2019-02-27 13:37:53 +01:00
inline T& get(ET pos) const;
2021-01-14 14:43:09 +01:00
template <typename V, class ET>
inline V& vget(ET pos) const;
2018-09-15 01:58:17 +02:00
template <class ET>
2019-02-26 18:56:57 +01:00
inline OperationRoot& set(ET pos);
2018-09-15 01:58:17 +02:00
2020-07-09 17:37:28 +02:00
MExt<None> rootSteps(std::intptr_t iPtrNum = 0) const; // nullptr for simple usage with decltype
template <class Expr>
Expr loop(Expr exp) const;
T* data() const;
2017-05-24 19:01:02 +02:00
2018-09-11 18:38:30 +02:00
template <class... Indices>
auto sl(const std::shared_ptr<Indices>&... inds)
-> Slice<T,typename Indices::RangeType...>;
};
template <typename T, class... Ranges>
class ParallelOperationRoot : public OperationTemplate<T,ParallelOperationRoot<T,Ranges...> >
{
public:
typedef T value_type;
typedef OperationBase<T,ParallelOperationRoot<T,Ranges...> > OT;
typedef ContainerRange<T,Ranges...> CRange;
typedef ContainerIndex<T,typename Ranges::IndexType...> IndexType;
static constexpr size_t SIZE = 1;
static constexpr bool CONT = true;
2021-01-18 19:00:12 +01:00
static constexpr bool VABLE = true;
private:
2017-12-25 13:44:55 +01:00
2018-02-13 15:38:03 +01:00
T* mDataPtr;
2019-02-26 18:56:57 +01:00
T* mOrigDataPtr;
IndexType mIndex;
public:
ParallelOperationRoot(MutableMultiArrayBase<T,Ranges...>& ma,
const std::shared_ptr<typename Ranges::IndexType>&... indices);
ParallelOperationRoot(T* data, const IndexType& ind);
template <class IOp, class OpClass>
auto asx(const OpClass& in) const
-> decltype(mIndex.pifor(1,in.loop(AssignmentExpr<T,IOp,ParallelOperationRoot<T,Ranges...>,OpClass,OpIndexAff::TARGET>
2021-01-24 02:10:06 +01:00
(mOrigDataPtr,*this,in))).template vec<IOp::VSIZE>());
template <class IOp, class OpClass>
auto asxExpr(const OpClass& in) const
-> decltype(in.loop(AssignmentExpr<T,IOp,ParallelOperationRoot<T,Ranges...>,OpClass>(mOrigDataPtr,*this,in)));
template <class IOp, class OpClass, class Index>
auto asx(const OpClass& in, const std::shared_ptr<Index>& i) const
-> decltype(i->pifor(1,in.loop(AssignmentExpr<T,IOp,ParallelOperationRoot<T,Ranges...>,OpClass>
2021-01-24 02:10:06 +01:00
(mOrigDataPtr,*this,in))).template vec<IOp::VSIZE>());
template <class OpClass>
auto assign(const OpClass& in) const
-> decltype(this->template asx<IAssign<T>>(in));
template <class OpClass>
auto assignExpr(const OpClass& in) const
-> decltype(this->template asxExpr<IAssign<T>>(in));
template <class OpClass, class Index>
auto assign(const OpClass& in, const std::shared_ptr<Index>& i) const
-> decltype(this->template asx<IAssign<T>>(in,i));
template <class OpClass>
auto plus(const OpClass& in) const
-> decltype(this->template asx<IPlus<T>>(in));
template <class OpClass, class Index>
auto plus(const OpClass& in, const std::shared_ptr<Index>& i) const
-> decltype(this->template asx<IPlus<T>>(in,i));
template <class OpClass>
ParallelOperationRoot& operator=(const OpClass& in);
template <class OpClass>
ParallelOperationRoot& operator+=(const OpClass& in);
ParallelOperationRoot& operator=(const ParallelOperationRoot& in);
template <class ET>
inline T& get(ET pos) const;
2021-01-14 14:43:09 +01:00
template <typename V, class ET>
inline V& vget(ET pos) const;
template <class ET>
inline ParallelOperationRoot& set(ET pos);
2020-07-09 17:37:28 +02:00
MExt<None> rootSteps(std::intptr_t iPtrNum = 0) const; // nullptr for simple usage with decltype
template <class Expr>
Expr loop(Expr exp) const;
T* data() const;
2017-02-16 11:20:40 +01:00
};
template <typename T>
class OperationValue : public OperationTemplate<T,OperationValue<T> >
{
public:
typedef T value_type;
typedef OperationBase<T,OperationValue<T> > OT;
typedef ContainerRange<T,NullRange> CRange;
typedef ContainerIndex<T,NullIndex> IndexType;
static constexpr size_t SIZE = 0;
2019-02-27 13:37:53 +01:00
static constexpr bool CONT = true;
2021-01-18 19:00:12 +01:00
static constexpr bool VABLE = false;
OperationValue(const T& val);
template <class ET>
2019-02-27 13:37:53 +01:00
inline const T& get(ET pos) const;
2021-01-14 14:43:09 +01:00
template <typename V, class ET>
inline V vget(ET pos) const;
2018-09-15 01:58:17 +02:00
template <class ET>
2019-02-26 18:56:57 +01:00
inline OperationValue& set(ET pos);
2018-09-15 01:58:17 +02:00
None rootSteps(std::intptr_t iPtrNum = 0) const; // nullptr for simple usage with decltype
template <class Expr>
Expr loop(Expr exp) const;
private:
T mVal;
};
2021-01-18 19:00:12 +01:00
template <class Op>
inline constexpr bool isVAble()
{
return Op::VABLE;
}
template <class Op1, class Op2, class... Ops>
inline constexpr bool isVAble()
{
return Op1::VABLE and isVAble<Op2,Ops...>();
}
2017-09-11 12:54:24 +02:00
2017-08-10 15:12:26 +02:00
template <typename T, class OpFunction, class... Ops>
2018-01-05 13:56:16 +01:00
class Operation : public OperationTemplate<T,Operation<T,OpFunction,Ops...> >
2017-03-22 11:44:33 +01:00
{
public:
2017-08-27 17:52:50 +02:00
typedef T value_type;
typedef OperationBase<T,Operation<T,OpFunction,Ops...> > OT;
2017-08-10 15:12:26 +02:00
typedef OpFunction F;
2018-02-13 15:38:03 +01:00
static constexpr size_t SIZE = RootSum<Ops...>::SIZE;
static constexpr bool FISSTATIC = OpFunction::FISSTATIC;
2019-02-27 13:37:53 +01:00
static constexpr bool CONT = false;
2021-01-18 19:00:12 +01:00
static constexpr bool VABLE = isVAble<Ops...>();
private:
2018-02-13 15:38:03 +01:00
std::tuple<Ops...> mOps;
std::shared_ptr<OpFunction> mF; // only if non-static
public:
2018-02-13 21:36:41 +01:00
typedef decltype(PackNum<sizeof...(Ops)-1>::template mkSteps<Ops...>(0, mOps)) ETuple;
2017-08-11 15:26:10 +02:00
Operation(const Ops&... ops);
Operation(std::shared_ptr<OpFunction> ff, const Ops&... ops);
2017-03-22 11:44:33 +01:00
2018-02-13 15:38:03 +01:00
template <class ET>
2019-05-20 18:32:19 +02:00
inline auto get(ET pos) const;
2021-01-14 14:43:09 +01:00
template <typename V, class ET>
inline auto vget(ET pos) const;
2018-09-15 01:58:17 +02:00
template <class ET>
2019-02-26 18:56:57 +01:00
inline Operation& set(ET pos);
2018-09-15 01:58:17 +02:00
auto rootSteps(std::intptr_t iPtrNum = 0) const // nullptr for simple usage with decltype
2018-02-13 15:38:03 +01:00
-> decltype(PackNum<sizeof...(Ops)-1>::mkSteps(iPtrNum, mOps));
template <class Expr>
auto loop(Expr exp) const
-> decltype(PackNum<sizeof...(Ops)-1>::mkLoop( mOps, exp));
T* data() const { assert(0); return nullptr; }
2017-03-22 11:44:33 +01:00
};
2018-03-19 18:38:53 +01:00
2018-09-12 20:56:55 +02:00
namespace
{
template <bool FISSTATIC>
struct OpMaker
{
template <class OpFunction, class... Ops>
static inline auto mkOperation(const std::shared_ptr<OpFunction>& f, const Ops&... ops)
-> Operation<typename OpFunction::value_type,OpFunction,Ops...>
{
return Operation<typename OpFunction::value_type,OpFunction,Ops...>(f,ops...);
}
};
template <>
struct OpMaker<true>
{
template <class OpFunction, class... Ops>
static inline auto mkOperation(const std::shared_ptr<OpFunction>& f, const Ops&... ops)
-> Operation<typename OpFunction::value_type,OpFunction,Ops...>
{
return Operation<typename OpFunction::value_type,OpFunction,Ops...>(ops...);
}
};
}
template <class OpFunction, class... Ops>
2018-09-12 20:56:55 +02:00
auto mkOperation(const std::shared_ptr<OpFunction>& f, const Ops&... ops)
-> Operation<typename OpFunction::value_type,OpFunction,Ops...>
{
2018-09-12 20:56:55 +02:00
return OpMaker<OpFunction::FISSTATIC>::mkOperation(f, ops...);
}
2018-07-19 13:00:34 +02:00
2017-11-04 22:49:55 +01:00
2017-11-02 21:20:31 +01:00
template <typename T, class Op, class IndexType>
2017-11-04 22:49:55 +01:00
class Contraction : public OperationTemplate<T,Contraction<T,Op,IndexType> >
{
public:
typedef T value_type;
typedef OperationBase<T,Contraction<T,Op,IndexType> > OT;
static constexpr size_t SIZE = Op::SIZE;
2019-02-27 13:37:53 +01:00
static constexpr bool CONT = Op::CONT;
2021-01-18 19:00:12 +01:00
static constexpr bool VABLE = Op::VABLE;
2019-02-27 13:37:53 +01:00
private:
2019-02-26 18:56:57 +01:00
Op mOp;
std::shared_ptr<IndexType> mInd;
public:
typedef decltype(mOp.rootSteps(0)) ETuple;
2017-11-02 21:20:31 +01:00
Contraction(const Op& op, std::shared_ptr<IndexType> ind);
2018-02-13 15:38:03 +01:00
template <class ET>
2019-02-27 13:37:53 +01:00
inline auto get(ET pos) const
-> decltype(mOp.template get<ET>(pos));
2021-01-14 14:43:09 +01:00
template <typename V, class ET>
inline auto vget(ET pos) const
-> decltype(mOp.template vget<V,ET>(pos));
2018-09-15 01:58:17 +02:00
template <class ET>
2019-02-26 18:56:57 +01:00
inline Contraction& set(ET pos);
2018-09-15 01:58:17 +02:00
2020-08-26 17:05:44 +02:00
T* data() const { assert(0); return nullptr; }
auto rootSteps(std::intptr_t iPtrNum = 0) const // nullptr for simple usage with decltype
-> decltype(mOp.rootSteps(iPtrNum));
template <class Expr>
auto loop(Expr exp) const
-> decltype(mInd->iforh(1,mOp.loop(exp)));
};
2018-05-15 21:18:21 +02:00
2018-09-14 17:50:19 +02:00
template <typename T, class Op, class... Indices>
2018-09-15 01:58:17 +02:00
// class SliceContraction : public OperationTemplate
//<MultiArray<T,typename Indices::RangeType...>,
//SliceContraction<MultiArray<T,typename Indices::RangeType...>,Op,Indices...> >
class SliceContraction : public OperationTemplate<T,SliceContraction<T,Op,Indices...> >
2018-09-14 17:50:19 +02:00
{
public:
2018-09-15 01:58:17 +02:00
typedef MultiArray<T,typename Indices::RangeType...> value_type;
typedef OperationTemplate<T,SliceContraction<T,Op,Indices...> > OT;
2018-09-14 17:50:19 +02:00
static constexpr size_t SIZE = Op::SIZE;
2019-02-27 13:37:53 +01:00
static constexpr bool CONT = false;
2021-01-18 19:00:12 +01:00
static constexpr bool VABLE = false;
2018-09-14 17:50:19 +02:00
private:
2019-02-26 18:56:57 +01:00
mutable Op mOp;
mutable std::shared_ptr<MultiArray<T,typename Indices::RangeType...> > mCont;
2018-09-15 01:58:17 +02:00
mutable OperationRoot<T,typename Indices::RangeType...> mTarOp;
2018-09-14 17:50:19 +02:00
public:
typedef decltype(mOp.rootSteps(0)) ETuple;
2018-09-15 16:33:49 +02:00
SliceContraction(const Op& op, std::shared_ptr<Indices>... ind);
2018-09-14 17:50:19 +02:00
template <class ET>
inline const value_type& get(ET pos) const;
2018-09-15 01:58:17 +02:00
template <class ET>
2019-02-26 18:56:57 +01:00
inline SliceContraction& set(ET pos);
2018-09-15 01:58:17 +02:00
2018-09-14 17:50:19 +02:00
auto rootSteps(std::intptr_t iPtrNum = 0) const // nullptr for simple usage with decltype
-> decltype(mOp.rootSteps(iPtrNum));
template <class Expr>
auto loop(Expr exp) const -> decltype(mOp.loop(exp)); // no loop
};
2017-08-11 15:26:10 +02:00
}
2017-02-16 11:20:40 +01:00
#include "type_operations.h"
2017-02-16 11:20:40 +01:00
#endif