This commit is contained in:
Christian Zimmermann 2020-08-27 23:48:04 +02:00
parent 34161f10cb
commit 7afd193d09
3 changed files with 74 additions and 20 deletions

View file

@ -1,5 +1,6 @@
#include "dynamic_operation.h"
#include "helper_tools.h"
namespace MultiArrayTools
{
@ -43,23 +44,28 @@ namespace MultiArrayTools
template <typename T, class Operation, class... Ranges>
DynamicOuterOp<T,Operation,Ranges...>::DynamicOuterOp(const Operation& op,
const std::shared_ptr<typename Ranges::IndexType>&... inds)
: mOp(op)
: mOp(op),
mMa(std::make_shared<MultiArray<T,Ranges...>>(mkArray<T>(inds->range()...))),
mProto((*mMa)(inds...))
{
//...!!!
// setup operations
/*
auto ll = mkILoop(std::make_tuple(mProto.mOp,mOp), std::make_tuple(inds...),
std::make_tuple(mMa), std::make_tuple(mProto.mOp.assign( mOp )),
std::array<size_t,1>({0}), std::array<size_t,1>({0}));
*/
}
template <typename T, class Operation, class... Ranges>
OpH<ConstOperationRoot<T,Ranges...>> DynamicOuterOp<T,Operation,Ranges...>::get(const DExtT& pos) const
OpH<OperationRoot<T,Ranges...>> DynamicOuterOp<T,Operation,Ranges...>::get(const DExtT& pos) const
{
mOp.get(pos.expl<ET>());
//...
//mL();
// execute assignment... care about threads!!!
return OpH<ConstOperationRoot<T,Ranges...>>(); // empty
return mProto.mOp; // empty
}
template <typename T, class Operation, class... Ranges>
DynamicOperationBase<OpH<ConstOperationRoot<T,Ranges...>>>&
DynamicOperationBase<OpH<OperationRoot<T,Ranges...>>>&
DynamicOuterOp<T,Operation,Ranges...>::set(const DExtT& pos)
{
mOp.set(pos.expl<ET>());
@ -79,16 +85,24 @@ namespace MultiArrayTools
}
template <typename T, class Operation, class... Ranges>
const OpH<ConstOperationRoot<T,Ranges...>>* DynamicOuterOp<T,Operation,Ranges...>::data() const
const OpH<OperationRoot<T,Ranges...>>* DynamicOuterOp<T,Operation,Ranges...>::data() const
{
return &mProto;
}
template <typename T, class Operation, class... Ranges>
std::shared_ptr<DynamicOperationBase<OpH<ConstOperationRoot<T,Ranges...>>>>
std::shared_ptr<DynamicOperationBase<OpH<OperationRoot<T,Ranges...>>>>
DynamicOuterOp<T,Operation,Ranges...>::deepCopy() const
{
return std::make_shared<DynamicOuterOp<T,Operation,Ranges...>>(*this);
}
template <class Operation, class... Indices>
DynamicOuterOp<typename Operation::value_type,Operation,typename Indices::RangeType...>
mkDynOutOp(const Operation& op, const std::shared_ptr<Indices>&... inds)
{
return DynamicOuterOp<typename Operation::value_type,Operation,
typename Indices::RangeType...>(op, inds...);
}
} // namespace MultiArrayTools

View file

@ -61,16 +61,26 @@ namespace MultiArrayTools
template <class Op>
struct OpH
{
Op mOp;
Op mOp;
OpH(const Op& op) : mOp(op) {}
};
template <typename T, class Operation, class... Ranges>
class DynamicOuterOp : public DynamicOperationBase<OpH<ConstOperationRoot<T,Ranges...>>>
class DynamicOuterOp : public DynamicOperationBase<OpH<OperationRoot<T,Ranges...>>>
{
private:
Operation mOp;
//ConstOperationRoot<T,Ranges...> mProto;
OpH<ConstOperationRoot<T,Ranges...>> mProto;
//OperationRoot<T,Ranges...> mProto;
std::shared_ptr<MultiArray<T,Ranges...>> mMa;
OpH<OperationRoot<T,Ranges...>> mProto;
/*
typedef ILoop<std::tuple<OperationRoot<T,Ranges...>,Operation>,
std::tuple<std::shared_ptr<typename Ranges::IndexType>...>,
std::tuple<std::shared_ptr<MultiArray<T,Ranges...>>>,
std::tuple<decltype(mProto.mOp.assign( mOp ))>> LoopT;
LoopT mL;
*/
public:
typedef decltype(mOp.rootSteps()) ET;
//typedef decltype(std::declval<Operation>().rootSteps()) ET;
@ -83,14 +93,18 @@ namespace MultiArrayTools
DynamicOuterOp(const Operation& op, const std::shared_ptr<typename Ranges::IndexType>&... inds);
virtual OpH<ConstOperationRoot<T,Ranges...>> get(const DExtT& pos) const override final;
virtual DynamicOperationBase<OpH<ConstOperationRoot<T,Ranges...>>>& set(const DExtT& pos) override final;
virtual OpH<OperationRoot<T,Ranges...>> get(const DExtT& pos) const override final;
virtual DynamicOperationBase<OpH<OperationRoot<T,Ranges...>>>& set(const DExtT& pos) override final;
virtual DExtT rootSteps(std::intptr_t iPtrNum = 0) const override final;
virtual DynamicExpression loop(const DynamicExpression& exp) const override final;
virtual const OpH<ConstOperationRoot<T,Ranges...>>* data() const override final;
virtual std::shared_ptr<DynamicOperationBase<OpH<ConstOperationRoot<T,Ranges...>>>> deepCopy() const override final;
virtual const OpH<OperationRoot<T,Ranges...>>* data() const override final;
virtual std::shared_ptr<DynamicOperationBase<OpH<OperationRoot<T,Ranges...>>>> deepCopy() const override final;
};
template <class Operation, class... Indices>
DynamicOuterOp<typename Operation::value_type,Operation,typename Indices::RangeType...>
mkDynOutOp(const Operation& op, const std::shared_ptr<Indices>&... inds);
template <typename T>
class DynamicO : public OperationTemplate<T,DynamicO<T>>
{
@ -113,6 +127,9 @@ namespace MultiArrayTools
template <class Op>
DynamicO(const Op& op) : mOp(std::make_shared<DynamicOperation<T,Op>>(op)) {}
DynamicO(const std::shared_ptr<DynamicOperationBase<T>>& op) :
mOp(op) {}
template <class X>
inline T get(const DExtTX<X>& pos) const { return mOp->get(pos.reduce()); }
template <class X>

View file

@ -53,6 +53,7 @@ namespace
std::map<std::string,std::shared_ptr<IndexW>> imap;
std::shared_ptr<DR> dr1;
std::shared_ptr<DR> dr1a;
std::shared_ptr<DR> dr2;
std::shared_ptr<DR> dr3;
std::shared_ptr<DR> dr4;
@ -60,6 +61,8 @@ namespace
std::shared_ptr<DR> dr6;
std::shared_ptr<CR> cr1;
std::shared_ptr<CR::IndexType> ci4;
OpTest_Dyn()
{
cr1 = createRangeE<CR>(5);
@ -70,6 +73,7 @@ namespace
auto cr5 = createRangeE<CR>(13);
dr1 = createRangeE<DR>(cr2,cr2,cr3,cr4);
dr1a = createRangeE<DR>(cr2,cr2,cr3);
dr2 = createRangeE<DR>(cr3,cr3,cr4);
dr3 = createRangeE<DR>(cr2,cr5);
dr5 = createRangeE<DR>(cr5);
@ -94,7 +98,8 @@ namespace
imap["i2_2"] = mkIndexW(getIndex(cr2));
imap["i3_1"] = mkIndexW(getIndex(cr3));
imap["i3_2"] = mkIndexW(getIndex(cr3));
imap["i4_1"] = mkIndexW(getIndex(cr4));
ci4 = getIndex(cr4);
imap["i4_1"] = mkIndexW(ci4);
imap["i4_2"] = mkIndexW(getIndex(cr4));
imap["i5_1"] = mkIndexW(getIndex(cr5));
imap["i5_2"] = mkIndexW(getIndex(cr5));
@ -105,21 +110,39 @@ namespace
{
auto i1 = getIndex(cr1);
auto di1 = getIndex(dr1);
auto di1a = getIndex(dr1a);
auto di2 = getIndex(dr2);
auto di4 = getIndex(dr4);
auto mi = mkMIndex(i1,di1a);
(*di1)({imap["i2_1"],imap["i2_2"],imap["i3_1"],imap["i4_1"]});
(*di1a)({imap["i2_1"],imap["i2_2"],imap["i3_1"]});
(*di2)({imap["i3_1"],imap["i3_1"],imap["i4_2"]});
(*di4)({imap["i2_1"],imap["i3_1"],imap["i4_1"],imap["i4_2"]});
auto resx1 = res1;
auto resx2 = res1;
auto resx3 = res1;
auto resx4 = res1;
//auto xx = std::make_shared<decltype(resx4)>(resx4);
res1(i1,di4) = ma1(i1,di1) * ma2(i1,di2);
resx1(i1,di4) = mkDynOp(ma1(i1,di1)) * mkDynOp(ma2(i1,di2));
resx2(i1,di4) = mkDynOp(ma1(i1,di1) * ma2(i1,di2));
resx3(i1,di4) = mkDynOp(mkDynOp(ma1(i1,di1)) * mkDynOp(ma2(i1,di2)));
auto op1 = mkDynOutOp(ma1(i1,di1) * ma2(i1,di2), ci4);
/**auto opr = resx4(i1,di4);
auto loop = mkILoop(std::make_tuple(opr,op1), std::make_tuple(ci4),
std::make_tuple(xx), std::make_tuple(opr.assign( op1.data()->mOp )),
//std::make_tuple(), std::make_tuple(),
std::array<size_t,1>({1}), std::array<size_t,1>({0}));
*/
//std::array<size_t,0>(), std::array<size_t,0>());
//!!!!
//mi->ifor(1, loop)();
//mi->ifor(1, opr.assign( op1.data()->mOp ))();
auto i2_1 = imap.at("i2_1");
auto i2_2 = imap.at("i2_2");
auto i3_1 = imap.at("i3_1");