im com while fixing (still NOT FINISHED)

This commit is contained in:
Christian Zimmermann 2018-01-14 19:15:05 +01:00
parent 740004d160
commit 96d9b4808e
4 changed files with 284 additions and 47 deletions

View file

@ -26,6 +26,44 @@ namespace MultiArrayTools
{ {
using namespace MultiArrayHelper; using namespace MultiArrayHelper;
} }
// OPERATIONS (STATIC)
template <typename T>
struct plus
{
static T&& apply(T&& a1, T&& a2)
{
return a1 + a2;
}
};
template <typename T>
struct minus
{
static T&& apply(T&& a1, T&& a2)
{
return a1 - a2;
}
};
template <typename T>
struct multiplies
{
static T&& apply(T&& a1, T&& a2)
{
return a1 * a2;
}
};
template <typename T>
struct divides
{
static T&& apply(T&& a1, T&& a2)
{
return a1 / a2;
}
};
template <typename T> template <typename T>
Block<T> makeBlock(const T* vec, size_t stepSize, size_t blockSize); Block<T> makeBlock(const T* vec, size_t stepSize, size_t blockSize);
@ -53,19 +91,19 @@ namespace MultiArrayTools
template <class Second> template <class Second>
auto operator+(const Second& in) const auto operator+(const Second& in) const
-> Operation<T,std::plus<T>,OperationClass,Second>; -> Operation<T,plus<T>,OperationClass,Second>;
template <class Second> template <class Second>
auto operator-(const Second& in) const auto operator-(const Second& in) const
-> Operation<T,std::minus<T>,OperationClass,Second>; -> Operation<T,minus<T>,OperationClass,Second>;
template <class Second> template <class Second>
auto operator*(const Second& in) const auto operator*(const Second& in) const
-> Operation<T,std::multiplies<T>,OperationClass,Second>; -> Operation<T,multiplies<T>,OperationClass,Second>;
template <class Second> template <class Second>
auto operator/(const Second& in) const auto operator/(const Second& in) const
-> Operation<T,std::divides<T>,OperationClass,Second>; -> Operation<T,divides<T>,OperationClass,Second>;
template <class IndexType> template <class IndexType>
auto c(std::shared_ptr<IndexType>& ind) const auto c(std::shared_ptr<IndexType>& ind) const
@ -88,22 +126,22 @@ namespace MultiArrayTools
//AssignmentExpr(const AssignmentExpr& in) = default; //AssignmentExpr(const AssignmentExpr& in) = default;
//AssignmentExpr& operator=(const AssignmentExpr& in) = default; //AssignmentExpr& operator=(const AssignmentExpr& in) = default;
OperationMaster* mMPtr; OperationMaster& mM;
const OpClass* mSecPtr; const OpClass& mSec;
public: public:
static size_t layer() { return 0; } static size_t layer() { return 0; }
static const size_t LAYER = 0; static const size_t LAYER = 0;
static const size_t SIZE = OpClass::SIZE; static const size_t SIZE = OpClass::SIZE;
typedef decltype(mSecPtr->rootSteps()) ETuple; typedef decltype(mSec.rootSteps()) ETuple;
AssignmentExpr(OperationMaster* mPtr, const OpClass* secPtr); AssignmentExpr(OperationMaster& m, const OpClass& sec);
AssignmentExpr(AssignmentExpr&& in) = default; AssignmentExpr(AssignmentExpr&& in) = default;
//AssignmentExpr& operator=(const AssignmentExpr&& in) = default; //AssignmentExpr& operator=(const AssignmentExpr&& in) = default;
inline void operator()(size_t start = 0); //inline void operator()(size_t start = 0);
inline void operator()(size_t start, const ETuple& last); inline void operator()(size_t start, const ETuple& last);
ETuple rootSteps(std::intptr_t iPtrNum = 0); ETuple rootSteps(std::intptr_t iPtrNum = 0);
@ -128,15 +166,19 @@ namespace MultiArrayTools
MBlock<T>& get(); MBlock<T>& get();
const Block<T>& get() const; const Block<T>& get() const;
T& get(size_t pos);
const T& get(size_t pos) const;
std::vector<BTSS> block(const IndexInfo* blockIndex, bool init = false) const; std::vector<BTSS> block(const IndexInfo* blockIndex, bool init = false) const;
const OperationMaster& block() const; const OperationMaster& block() const;
private: private:
std::shared_ptr<IndexType> mkIndex(std::shared_ptr<typename CRange::IndexType>& index); std::shared_ptr<IndexType> mkIndex(std::shared_ptr<typename CRange::IndexType>& index);
void performAssignment(std::intptr_t blockIndexNum); void performAssignment(std::intptr_t blockIndexNum);
OpClass const& mSecond; OpClass const& mSecond;
MutableMultiArrayBase<T,Ranges...>& mArrayRef; MutableMultiArrayBase<T,Ranges...>& mArrayRef;
T* mData;
std::shared_ptr<IndexType> mIndex; std::shared_ptr<IndexType> mIndex;
IndexInfo mIInfo; IndexInfo mIInfo;
mutable bType mBlock; mutable bType mBlock;
@ -163,12 +205,18 @@ namespace MultiArrayTools
const std::shared_ptr<typename Ranges::IndexType>&... indices); const std::shared_ptr<typename Ranges::IndexType>&... indices);
const Block<T>& get() const; const Block<T>& get() const;
template <class ET, size_t SITE>
inline const T& get(const ET& pos) const;
std::vector<BTSS> block(const IndexInfo* blockIndex, bool init = false) const; std::vector<BTSS> block(const IndexInfo* blockIndex, bool init = false) const;
const ConstOperationRoot& block() const; const ConstOperationRoot& block() const;
std::tuple<size_t> rootSteps(const IndexInfo* ii = nullptr) const; // nullptr for simple usage with decltype std::tuple<size_t> rootSteps(std::intptr_t iPtrNum = 0) const; // nullptr for simple usage with decltype
template <class Expr>
Expr&& loop(Expr&& exp) const;
private: private:
std::shared_ptr<IndexType> std::shared_ptr<IndexType>
@ -176,6 +224,7 @@ namespace MultiArrayTools
const std::shared_ptr<typename Ranges::IndexType>&... indices); const std::shared_ptr<typename Ranges::IndexType>&... indices);
MultiArrayBase<T,Ranges...> const& mArrayRef; MultiArrayBase<T,Ranges...> const& mArrayRef;
const T* mData;
std::shared_ptr<IndexType> mIndex; std::shared_ptr<IndexType> mIndex;
IndexInfo mIInfo; IndexInfo mIInfo;
mutable bType mBlock; mutable bType mBlock;
@ -205,11 +254,20 @@ namespace MultiArrayTools
const MBlock<T>& get() const; const MBlock<T>& get() const;
MBlock<T>& get(); MBlock<T>& get();
template <class ET, size_t SITE>
inline const T& get(const ET& pos) const;
template <class ET, size_t SITE>
inline T& get(const ET& pos);
OperationRoot& set(const IndexInfo* blockIndex); OperationRoot& set(const IndexInfo* blockIndex);
std::vector<BTSS> block(const IndexInfo* blockIndex, bool init = false) const; std::vector<BTSS> block(const IndexInfo* blockIndex, bool init = false) const;
const OperationRoot& block() const; const OperationRoot& block() const;
std::tuple<size_t> rootSteps(const IndexInfo* ii = nullptr) const; // nullptr for simple usage with decltype std::tuple<size_t> rootSteps(std::intptr_t iPtrNum = 0) const; // nullptr for simple usage with decltype
template <class Expr>
Expr&& loop(Expr&& exp) const;
private: private:
@ -218,6 +276,7 @@ namespace MultiArrayTools
const std::shared_ptr<typename Ranges::IndexType>&... indices); const std::shared_ptr<typename Ranges::IndexType>&... indices);
MutableMultiArrayBase<T,Ranges...>& mArrayRef; MutableMultiArrayBase<T,Ranges...>& mArrayRef;
T* mData;
std::shared_ptr<IndexType> mIndex; std::shared_ptr<IndexType> mIndex;
IndexInfo mIInfo; IndexInfo mIInfo;
mutable bType mBlock; mutable bType mBlock;
@ -283,15 +342,25 @@ namespace MultiArrayTools
mutable bType mRes; mutable bType mRes;
public: public:
typedef decltype(PackNum<sizeof...(Ops)-1>::mkStepTuple(0, mOps)) ETuple;
//typedef decltype(PackNum<sizeof...(Ops)-1>::template mkLoopType<Ops...>) LType;
Operation(const Ops&... ops); Operation(const Ops&... ops);
const BlockResult<T>& get() const; const BlockResult<T>& get() const;
template <class ET, size_t SITE>
inline T&& get(const ET& pos) const;
std::vector<BTSS> block(const IndexInfo* blockIndex, bool init = false) const; std::vector<BTSS> block(const IndexInfo* blockIndex, bool init = false) const;
const Operation& block() const; const Operation& block() const;
auto rootSteps(const IndexInfo* ii = nullptr) const // nullptr for simple usage with decltype auto rootSteps(std::intptr_t iPtrNum = 0) const // nullptr for simple usage with decltype
-> decltype(PackNum<sizeof...(Ops)-1>::mkStepTuple(ii, mOps)); -> decltype(PackNum<sizeof...(Ops)-1>::mkStepTuple(iPtrNum, mOps));
template <class Expr>
auto loop(Expr&& exp) const
-> decltype(PackNum<sizeof...(Ops)-1>::mkLoop( mOps, std::forward<Expr>( exp ) ))&&;
}; };
@ -314,16 +383,23 @@ namespace MultiArrayTools
mutable bType mRes; mutable bType mRes;
public: public:
typedef decltype(mOp.rootSteps(0)) ETuple;
Contraction(const Op& op, std::shared_ptr<IndexType> ind); Contraction(const Op& op, std::shared_ptr<IndexType> ind);
const BlockResult<T>& get() const; const BlockResult<T>& get() const;
template <class ET, size_t SITE>
inline T&& get(const ET& pos) const;
std::vector<BTSS> block(const IndexInfo* blockIndex, bool init = false) const; std::vector<BTSS> block(const IndexInfo* blockIndex, bool init = false) const;
const Contraction& block() const; const Contraction& block() const;
auto rootSteps(const IndexInfo* ii = nullptr) const // nullptr for simple usage with decltype auto rootSteps(std::intptr_t iPtrNum = 0) const // nullptr for simple usage with decltype
-> decltype(mOp.rootSteps(ii)); -> decltype(mOp.rootSteps(iPtrNum));
template <class Expr>
auto loop(Expr&& exp) const -> decltype(mInd->ifor(exp))&&;
}; };
} }
@ -393,33 +469,33 @@ namespace MultiArrayTools
template <typename T, class OperationClass> template <typename T, class OperationClass>
template <class Second> template <class Second>
auto OperationTemplate<T,OperationClass>::operator+(const Second& in) const auto OperationTemplate<T,OperationClass>::operator+(const Second& in) const
-> Operation<T,std::plus<T>,OperationClass,Second> -> Operation<T,plus<T>,OperationClass,Second>
{ {
return Operation<T,std::plus<T>,OperationClass,Second>(THIS(), in); return Operation<T,plus<T>,OperationClass,Second>(THIS(), in);
} }
template <typename T, class OperationClass> template <typename T, class OperationClass>
template <class Second> template <class Second>
auto OperationTemplate<T,OperationClass>::operator-(const Second& in) const auto OperationTemplate<T,OperationClass>::operator-(const Second& in) const
-> Operation<T,std::minus<T>,OperationClass,Second> -> Operation<T,minus<T>,OperationClass,Second>
{ {
return Operation<T,std::minus<T>,OperationClass,Second>(THIS(), in); return Operation<T,minus<T>,OperationClass,Second>(THIS(), in);
} }
template <typename T, class OperationClass> template <typename T, class OperationClass>
template <class Second> template <class Second>
auto OperationTemplate<T,OperationClass>::operator*(const Second& in) const auto OperationTemplate<T,OperationClass>::operator*(const Second& in) const
-> Operation<T,std::multiplies<T>,OperationClass,Second> -> Operation<T,multiplies<T>,OperationClass,Second>
{ {
return Operation<T,std::multiplies<T>,OperationClass,Second>(THIS(), in); return Operation<T,multiplies<T>,OperationClass,Second>(THIS(), in);
} }
template <typename T, class OperationClass> template <typename T, class OperationClass>
template <class Second> template <class Second>
auto OperationTemplate<T,OperationClass>::operator/(const Second& in) const auto OperationTemplate<T,OperationClass>::operator/(const Second& in) const
-> Operation<T,std::divides<T>,OperationClass,Second> -> Operation<T,divides<T>,OperationClass,Second>
{ {
return Operation<T,std::divides<T>,OperationClass,Second>(THIS(), in); return Operation<T,divides<T>,OperationClass,Second>(THIS(), in);
} }
template <typename T, class OperationClass> template <typename T, class OperationClass>
@ -430,6 +506,38 @@ namespace MultiArrayTools
return Contraction<T,OperationClass,IndexType>(THIS(), ind); return Contraction<T,OperationClass,IndexType>(THIS(), ind);
} }
/*****************************************
* OperationMaster::AssignmentExpr *
*****************************************/
template <typename T, class OpClass, class... Ranges>
OperationMaster<T,OpClass,Ranges...>::AssignmentExpr::
AssignmentExpr(OperationMaster& m, const OpClass& sec) :
mM(m), mSec(sec) {}
/*
template <typename T, class OpClass, class... Ranges>
inline void OperationMaster<T,OpClass,Ranges...>::AssignmentExpr::
operator()(size_t start)
{
}*/
template <typename T, class OpClass, class... Ranges>
inline void OperationMaster<T,OpClass,Ranges...>::AssignmentExpr::
operator()(size_t start, const ETuple& last)
{
mM.get(start) = mSec.template get<ETuple,OpClass::SIZE>(last);
}
template <typename T, class OpClass, class... Ranges>
typename OperationMaster<T,OpClass,Ranges...>::AssignmentExpr::ETuple
OperationMaster<T,OpClass,Ranges...>::AssignmentExpr::
rootSteps(std::intptr_t iPtrNum)
{
return mSec.rootSteps(iPtrNum);
}
/************************* /*************************
* OperationMaster * * OperationMaster *
@ -439,7 +547,8 @@ namespace MultiArrayTools
OperationMaster<T,OpClass,Ranges...>:: OperationMaster<T,OpClass,Ranges...>::
OperationMaster(MutableMultiArrayBase<T,Ranges...>& ma, const OpClass& second, OperationMaster(MutableMultiArrayBase<T,Ranges...>& ma, const OpClass& second,
std::shared_ptr<typename CRange::IndexType>& index) : std::shared_ptr<typename CRange::IndexType>& index) :
mSecond(second), mArrayRef(ma), mIndex(mkIndex(index)), mIInfo(*mIndex) mSecond(second), mArrayRef(ma), mData(mArrayRef.data()),
mIndex(mkIndex(index)), mIInfo(*mIndex)
{ {
auto blockIndex = seekBlockIndex( &mIInfo, second); auto blockIndex = seekBlockIndex( &mIInfo, second);
std::intptr_t blockIndexNum = blockIndex->getPtrNum(); std::intptr_t blockIndexNum = blockIndex->getPtrNum();
@ -455,7 +564,8 @@ namespace MultiArrayTools
OperationMaster(MutableMultiArrayBase<T,Ranges...>& ma, const OpClass& second, OperationMaster(MutableMultiArrayBase<T,Ranges...>& ma, const OpClass& second,
std::shared_ptr<typename CRange::IndexType>& index, std::shared_ptr<typename CRange::IndexType>& index,
const IndexInfo* blockIndex) : const IndexInfo* blockIndex) :
mSecond(second), mArrayRef(ma), mIndex(mkIndex(index)), mIInfo(*mIndex) mSecond(second), mArrayRef(ma), mData(mArrayRef.data()),
mIndex(mkIndex(index)), mIInfo(*mIndex)
{ {
std::intptr_t blockIndexNum = blockIndex->getPtrNum(); std::intptr_t blockIndexNum = blockIndex->getPtrNum();
second.block(blockIndex, true); second.block(blockIndex, true);
@ -482,9 +592,12 @@ namespace MultiArrayTools
#define XX_USE_NEW_LOOP_ROUTINE_XX #define XX_USE_NEW_LOOP_ROUTINE_XX
#ifdef XX_USE_NEW_LOOP_ROUTINE_XX #ifdef XX_USE_NEW_LOOP_ROUTINE_XX
// === N E W === // === N E W ===
AssignmentExpr ae(this, &mSecond); AssignmentExpr ae(*this, mSecond); // Expression to be executed within loop
const auto loop = mIndex->ifor(ae); //const auto hiddenLoop = mSecond.loop(AssignmentExpr(*this, mSecond)); // hidden loop within 'mSecond' e.g. contractions
loop(); const auto loop = mIndex->ifor
( mSecond.template loop<AssignmentExpr>
( std::move(ae) ) ); // init overall loop(s)
loop(); // execute overall loop(s) and so internal hidden loops and so the inherited expressions
#else #else
// === O L D === // === O L D ===
for(*mIndex = 0; mIndex->pos() != mIndex->max(); mIndex->pp(blockIndexNum) ){ for(*mIndex = 0; mIndex->pos() != mIndex->max(); mIndex->pp(blockIndexNum) ){
@ -506,6 +619,18 @@ namespace MultiArrayTools
return mBlock; return mBlock;
} }
template <typename T, class OpClass, class... Ranges>
T& OperationMaster<T,OpClass,Ranges...>::get(size_t pos)
{
return mData[pos];
}
template <typename T, class OpClass, class... Ranges>
const T& OperationMaster<T,OpClass,Ranges...>::get(size_t pos) const
{
return mData[pos];
}
template <typename T, class OpClass, class... Ranges> template <typename T, class OpClass, class... Ranges>
std::vector<BTSS> OperationMaster<T,OpClass,Ranges...>::block(const IndexInfo* blockIndex, bool init) const std::vector<BTSS> OperationMaster<T,OpClass,Ranges...>::block(const IndexInfo* blockIndex, bool init) const
{ {
@ -524,6 +649,7 @@ namespace MultiArrayTools
return *this; return *this;
} }
/**************************** /****************************
* ConstOperationRoot * * ConstOperationRoot *
****************************/ ****************************/
@ -533,7 +659,8 @@ namespace MultiArrayTools
ConstOperationRoot(const MultiArrayBase<T,Ranges...>& ma, ConstOperationRoot(const MultiArrayBase<T,Ranges...>& ma,
const std::shared_ptr<typename Ranges::IndexType>&... indices) : const std::shared_ptr<typename Ranges::IndexType>&... indices) :
//OperationTemplate<T,ConstOperationRoot<T,Ranges...> >(this), //OperationTemplate<T,ConstOperationRoot<T,Ranges...> >(this),
mArrayRef(ma), mIndex( mkIndex(ma,indices...) ), mIInfo(*mIndex) mArrayRef(ma), mData(mArrayRef.data()),
mIndex( mkIndex(ma,indices...) ), mIInfo(*mIndex)
{} {}
template <typename T, class... Ranges> template <typename T, class... Ranges>
@ -553,7 +680,14 @@ namespace MultiArrayTools
block(); block();
return mBlock; return mBlock;
} }
template <typename T, class... Ranges>
template <class ET, size_t SITE>
inline const T& ConstOperationRoot<T,Ranges...>::get(const ET& pos) const
{
return mData[std::get<SITE>(pos)];
}
template <typename T, class... Ranges> template <typename T, class... Ranges>
std::vector<BTSS> ConstOperationRoot<T,Ranges...>::block(const IndexInfo* blockIndex, bool init) const std::vector<BTSS> ConstOperationRoot<T,Ranges...>::block(const IndexInfo* blockIndex, bool init) const
{ {
@ -572,10 +706,18 @@ namespace MultiArrayTools
} }
template <typename T, class... Ranges> template <typename T, class... Ranges>
std::tuple<size_t> ConstOperationRoot<T,Ranges...>::rootSteps(const IndexInfo* ii) const std::tuple<size_t> ConstOperationRoot<T,Ranges...>::rootSteps(std::intptr_t iPtrNum) const
{ {
return std::tuple<size_t>(0ul); // !!!!!! return std::tuple<size_t>(0ul); // !!!!!!
} }
template <typename T, class... Ranges>
template <class Expr>
Expr&& ConstOperationRoot<T,Ranges...>::loop(Expr&& exp) const
{
return std::forward<Expr>(exp);
}
/*********************** /***********************
* OperationRoot * * OperationRoot *
@ -586,7 +728,8 @@ namespace MultiArrayTools
OperationRoot(MutableMultiArrayBase<T,Ranges...>& ma, OperationRoot(MutableMultiArrayBase<T,Ranges...>& ma,
const std::shared_ptr<typename Ranges::IndexType>&... indices) : const std::shared_ptr<typename Ranges::IndexType>&... indices) :
//OperationTemplate<T,OperationRoot<T,Ranges...> >(this), //OperationTemplate<T,OperationRoot<T,Ranges...> >(this),
mArrayRef(ma), mIndex( mkIndex( ma, indices... ) ), mIInfo(*mIndex), mArrayRef(ma), mData(mArrayRef.data()),
mIndex( mkIndex( ma, indices... ) ), mIInfo(*mIndex),
mBlockII(nullptr) mBlockII(nullptr)
{} {}
@ -627,6 +770,20 @@ namespace MultiArrayTools
return mBlock; return mBlock;
} }
template <typename T, class... Ranges>
template <class ET, size_t SITE>
inline const T& OperationRoot<T,Ranges...>::get(const ET& pos) const
{
return mData[std::get<SITE>( pos )];
}
template <typename T, class... Ranges>
template <class ET, size_t SITE>
inline T& OperationRoot<T,Ranges...>::get(const ET& pos)
{
return mData[std::get<SITE>( pos )];
}
template <typename T, class... Ranges> template <typename T, class... Ranges>
OperationRoot<T,Ranges...>& OperationRoot<T,Ranges...>&
OperationRoot<T,Ranges...>::set(const IndexInfo* blockIndex) OperationRoot<T,Ranges...>::set(const IndexInfo* blockIndex)
@ -653,10 +810,17 @@ namespace MultiArrayTools
} }
template <typename T, class... Ranges> template <typename T, class... Ranges>
std::tuple<size_t> OperationRoot<T,Ranges...>::rootSteps(const IndexInfo* ii) const std::tuple<size_t> OperationRoot<T,Ranges...>::rootSteps(std::intptr_t iPtrNum) const
{ {
return std::tuple<size_t>(0ul); // !!!!!! return std::tuple<size_t>(0ul); // !!!!!!
} }
template <typename T, class... Ranges>
template <class Expr>
Expr&& OperationRoot<T,Ranges...>::loop(Expr&& exp) const
{
return std::forward<Expr>(exp);
}
/******************* /*******************
* Operation * * Operation *
@ -674,6 +838,15 @@ namespace MultiArrayTools
return mRes; return mRes;
} }
template <typename T, class OpFunction, class... Ops>
template <class ET, size_t SITE>
inline T&& Operation<T,OpFunction,Ops...>::get(const ET& pos) const
{
typedef std::tuple<Ops const&...> OpTuple;
return std::forward<T>( PackNum<sizeof...(Ops)-2>::
template mkOpExpr<T,ET,OpTuple,OpFunction,SITE>(pos, mOps) );
}
template <typename T, class OpFunction, class... Ops> template <typename T, class OpFunction, class... Ops>
std::vector<BTSS> Operation<T,OpFunction,Ops...>::block(const IndexInfo* blockIndex, bool init) const std::vector<BTSS> Operation<T,OpFunction,Ops...>::block(const IndexInfo* blockIndex, bool init) const
{ {
@ -694,10 +867,21 @@ namespace MultiArrayTools
} }
template <typename T, class OpFunction, class... Ops> template <typename T, class OpFunction, class... Ops>
auto Operation<T,OpFunction,Ops...>::rootSteps(const IndexInfo* ii) const auto Operation<T,OpFunction,Ops...>::rootSteps(std::intptr_t iPtrNum) const
-> decltype(PackNum<sizeof...(Ops)-1>::mkStepTuple(ii, mOps)) -> decltype(PackNum<sizeof...(Ops)-1>::mkStepTuple(iPtrNum, mOps))
{ {
return PackNum<sizeof...(Ops)-1>::mkStepTuple(ii, mOps); return PackNum<sizeof...(Ops)-1>::mkStepTuple(iPtrNum, mOps);
}
template <typename T, class OpFunction, class... Ops>
template <class Expr>
auto Operation<T,OpFunction,Ops...>::loop(Expr&& exp) const
-> decltype(PackNum<sizeof...(Ops)-1>::mkLoop( mOps, std::forward<Expr>( exp ) ))&&
{
typedef decltype(PackNum<sizeof...(Ops)-1>::mkLoop( mOps, std::forward<Expr>( exp ) ) )
LType;
return std::forward<LType>
( PackNum<sizeof...(Ops)-1>::mkLoop( mOps, std::forward<Expr>( exp ) ) );
} }
@ -721,6 +905,14 @@ namespace MultiArrayTools
return mRes; return mRes;
} }
// forward loop !!!!
template <typename T, class Op, class IndexType>
template <class ET, size_t SITE>
inline T&& Contraction<T,Op,IndexType>::get(const ET& pos) const
{
return std::forward<T>( mOp.template get<SITE>(pos) );
}
template <typename T, class Op, class IndexType> template <typename T, class Op, class IndexType>
std::vector<BTSS> Contraction<T,Op,IndexType>::block(const IndexInfo* blockIndex, bool init) const std::vector<BTSS> Contraction<T,Op,IndexType>::block(const IndexInfo* blockIndex, bool init) const
{ {
@ -737,10 +929,19 @@ namespace MultiArrayTools
} }
template <typename T, class Op, class IndexType> template <typename T, class Op, class IndexType>
auto Contraction<T,Op,IndexType>::rootSteps(const IndexInfo* ii) const auto Contraction<T,Op,IndexType>::rootSteps(std::intptr_t iPtrNum) const
-> decltype(mOp.rootSteps(ii)) -> decltype(mOp.rootSteps(iPtrNum))
{ {
return mOp.rootSteps(ii); return mOp.rootSteps(iPtrNum);
}
template <typename T, class Op, class IndexType>
template <class Expr>
auto Contraction<T,Op,IndexType>::loop(Expr&& exp) const -> decltype(mInd->ifor(exp))&&
{
typedef decltype(mInd->ifor(exp)) LType;
LType&& loop = mInd->ifor(exp);
return std::forward<LType>( loop );
} }
} }

View file

@ -51,7 +51,7 @@ namespace MultiArrayHelper
} }
template <class... Ops> template <class... Ops>
static auto mkStepTuple(const IndexInfo* ii, std::tuple<Ops const&...> otp) static auto mkStepTuple(std::intptr_t ii, std::tuple<Ops const&...> otp)
-> decltype(std::tuple_cat( PackNum<N-1>::mkStepTuple(ii, otp), std::get<N>(otp).rootSteps(ii) )) -> decltype(std::tuple_cat( PackNum<N-1>::mkStepTuple(ii, otp), std::get<N>(otp).rootSteps(ii) ))
{ {
return std::tuple_cat( PackNum<N-1>::mkStepTuple(ii, otp), std::get<N>(otp).rootSteps(ii) ); return std::tuple_cat( PackNum<N-1>::mkStepTuple(ii, otp), std::get<N>(otp).rootSteps(ii) );
@ -65,6 +65,23 @@ namespace MultiArrayHelper
std::get<N>(out) = second.rootSteps( std::get<N>(siar) ); std::get<N>(out) = second.rootSteps( std::get<N>(siar) );
PackNum<N-1>::mkExt(out, siar, second); PackNum<N-1>::mkExt(out, siar, second);
} }
// call with -2 (instead of -1)
template <typename T, class ETuple, class OpTuple, class OpFunction, size_t START>
static T&& mkOpExpr(const ETuple& pos, const OpTuple& ops)
{
static const size_t NEXT = START - std::tuple_element<N+1,OpTuple>::type::SIZE;
return std::forward<T>
( OpFunction::apply( std::get<N+1>(ops).template get<START>(pos),
PackNum<N-1>::template mkOpExpr<ETuple,OpTuple,OpFunction,NEXT>(pos, ops) ) );
}
template <class OpTuple, class Expr>
static auto mkLoop( const OpTuple& ot, Expr&& exp )
-> decltype(std::get<N>(ot).loop( PackNum<N-1>::mkLoop(ot,exp) ))&&
{
return std::get<N>(ot).loop( PackNum<N-1>::mkLoop(ot,exp) );
}
}; };
template<> template<>
@ -105,7 +122,7 @@ namespace MultiArrayHelper
} }
template <class... Ops> template <class... Ops>
static auto mkStepTuple(const IndexInfo* ii, std::tuple<Ops const&...> otp) static auto mkStepTuple(std::intptr_t ii, std::tuple<Ops const&...> otp)
-> decltype(std::get<0>(otp).rootSteps(ii)) -> decltype(std::get<0>(otp).rootSteps(ii))
{ {
return std::get<0>(otp).rootSteps(ii); return std::get<0>(otp).rootSteps(ii);
@ -119,6 +136,23 @@ namespace MultiArrayHelper
std::get<0>(out) = second.rootSteps( std::get<0>(siar) ); std::get<0>(out) = second.rootSteps( std::get<0>(siar) );
} }
template <typename T, class ETuple, class OpTuple, class OpFunction, size_t START>
static T&& mkOpExpr(const ETuple& pos, const OpTuple& ops)
{
static const size_t NEXT = START - std::tuple_element<1,OpTuple>::type::SIZE;
return std::forward<T>
( OpFunction::apply( std::get<1>(ops).template get<START>(pos),
std::get<0>(ops).template get<NEXT>(pos) ) );
}
template <class OpTuple, class Expr>
static auto mkLoop( const OpTuple& ot, Expr&& exp )
-> decltype(std::get<0>(ot).loop( exp ))&&
{
return std::get<0>(ot).loop( exp );
}
}; };

View file

@ -254,7 +254,7 @@ namespace MultiArrayTools
-> For<SingleIndex<U,TYPE>,Expr> -> For<SingleIndex<U,TYPE>,Expr>
{ {
//static const size_t LAYER = typename Expr::LAYER; //static const size_t LAYER = typename Expr::LAYER;
return For<SingleIndex<U,TYPE>,Expr>(this, ex); return For<SingleIndex<U,TYPE>,Expr>(this, std::forward<Expr>( ex ));
} }

View file

@ -66,12 +66,14 @@ namespace MultiArrayHelper
template <class IndexClass, class Expr> template <class IndexClass, class Expr>
For<IndexClass,Expr>::For(const std::shared_ptr<IndexClass>& indPtr, For<IndexClass,Expr>::For(const std::shared_ptr<IndexClass>& indPtr,
Expr&& expr) : Expr&& expr) :
mIndPtr(indPtr.get()), mExpr(expr), mExt(expr.rootSteps( static_cast<std::intptr_t>( mIndPtr.get() ))) {} mIndPtr(indPtr.get()), mExpr(expr),
mExt(expr.rootSteps( reinterpret_cast<std::intptr_t>( mIndPtr.get() ))) {}
template <class IndexClass, class Expr> template <class IndexClass, class Expr>
For<IndexClass,Expr>::For(const IndexClass* indPtr, For<IndexClass,Expr>::For(const IndexClass* indPtr,
Expr&& expr) : Expr&& expr) :
mIndPtr(indPtr), mExpr(expr), mExt(expr.rootSteps( static_cast<std::intptr_t>( mIndPtr ) )) {} mIndPtr(indPtr), mExpr(std::forward<Expr>( expr )),
mExt(expr.rootSteps( reinterpret_cast<std::intptr_t>( mIndPtr ) )) {}
template <class IndexClass, class Expr> template <class IndexClass, class Expr>
inline void For<IndexClass,Expr>::operator()(size_t mlast, inline void For<IndexClass,Expr>::operator()(size_t mlast,