shift high level op code to separate headers
This commit is contained in:
Christian Zimmermann
parent
4beb352c0e
commit
e174ced2eb
6 changed files with 451 additions and 330 deletions
299
src/include/high_level_operation.cc.h
Normal file
299
src/include/high_level_operation.cc.h
Normal file
|
|
@ -0,0 +1,299 @@
|
||||||
|
|
||||||
|
#include "high_level_operation.h"
|
||||||
|
|
||||||
|
namespace MultiArrayTools
|
||||||
|
{
|
||||||
|
template <typename T, class Op>
|
||||||
|
DynamicO<T> mkDynOp1(const Op& op)
|
||||||
|
{
|
||||||
|
return DynamicO<T>(op);
|
||||||
|
}
|
||||||
|
|
||||||
|
template <class ROP>
|
||||||
|
template <class... Indices>
|
||||||
|
template <class Op, class... Ops>
|
||||||
|
void HighLevelOpBase<ROP>::RetT<Indices...>::appendOuterM(const Op& op, const Ops&... ops)
|
||||||
|
{
|
||||||
|
// does not check anything regarding input !!!
|
||||||
|
if(outer.init()){
|
||||||
|
outer = mkDynOp1<size_t>(mkMOp<size_t>(outer,op,ops...));
|
||||||
|
}
|
||||||
|
else {
|
||||||
|
outer = mkDynOp1<size_t>(mkMOp<size_t>(op,ops...));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
template <class ROP>
|
||||||
|
template <class... Indices>
|
||||||
|
void HighLevelOpBase<ROP>::RetT<Indices...>::appendOuterM()
|
||||||
|
{}
|
||||||
|
|
||||||
|
template <class ROP>
|
||||||
|
template <class... Indices>
|
||||||
|
void HighLevelOpBase<ROP>::RetT<Indices...>::appendOuter(const DynamicO<size_t>& in)
|
||||||
|
{
|
||||||
|
if(in.init()){
|
||||||
|
if(outer.init()){
|
||||||
|
outer = mkDynOp1<size_t>(mkMOp<size_t>(outer,in));
|
||||||
|
}
|
||||||
|
else {
|
||||||
|
outer = in;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
template <class ROP>
|
||||||
|
template <class... Indices>
|
||||||
|
void HighLevelOpBase<ROP>::RetT<Indices...>::appendOuter(const RetT& in)
|
||||||
|
{
|
||||||
|
appendOuter(in.outer);
|
||||||
|
}
|
||||||
|
|
||||||
|
template <class ROP>
|
||||||
|
HighLevelOpRoot<ROP>::HighLevelOpRoot(const ROP& op) : mOp(op) {}
|
||||||
|
|
||||||
|
template <class ROP>
|
||||||
|
bool HighLevelOpRoot<ROP>::root() const
|
||||||
|
{
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
template <class ROP>
|
||||||
|
auto HighLevelOpRoot<ROP>::create(const std::shared_ptr<CI> ind1,
|
||||||
|
const std::shared_ptr<CI> ind2)
|
||||||
|
-> typename B::RetT<CI,CI>
|
||||||
|
{
|
||||||
|
assert(0);
|
||||||
|
return typename B::RetT<CI,CI>();
|
||||||
|
}
|
||||||
|
|
||||||
|
template <class ROP>
|
||||||
|
const ROP* HighLevelOpRoot<ROP>::get() const
|
||||||
|
{
|
||||||
|
return &mOp;
|
||||||
|
}
|
||||||
|
|
||||||
|
namespace
|
||||||
|
{
|
||||||
|
template <size_t N>
|
||||||
|
struct Create
|
||||||
|
{
|
||||||
|
template <class... Indices>
|
||||||
|
struct cx
|
||||||
|
{
|
||||||
|
template <class ROP, class OpF, class... OPs>
|
||||||
|
struct ccx
|
||||||
|
{
|
||||||
|
template <size_t M, class... DOPs>
|
||||||
|
static inline void
|
||||||
|
cccx(typename HighLevelOpBase<ROP>::RetT<Indices...>& res,
|
||||||
|
const std::array<std::shared_ptr<HighLevelOpBase<ROP>>,M>& in,
|
||||||
|
const std::shared_ptr<Indices>&... inds,
|
||||||
|
const OPs&... ops,
|
||||||
|
const DOPs&... dops)
|
||||||
|
{
|
||||||
|
static_assert(N > 0, "N > 0 failed");
|
||||||
|
auto& inn = std::get<N>(in);
|
||||||
|
if(not inn->root()){
|
||||||
|
auto dop = inn->create(inds...);
|
||||||
|
auto op = *dop.op.data()->mOp;
|
||||||
|
typedef decltype(op) OP;
|
||||||
|
res.appendOuter(dop);
|
||||||
|
Create<N-1>::template cx<Indices...>::template ccx<ROP,OpF,OP,OPs...>::template cccx<M>
|
||||||
|
(res, in, inds..., op, ops..., dop, dops...);
|
||||||
|
}
|
||||||
|
else {
|
||||||
|
auto& op = *inn->get();
|
||||||
|
typedef typename std::remove_reference<decltype(op)>::type OP;
|
||||||
|
Create<N-1>::template cx<Indices...>::template ccx<ROP,OpF,OP,OPs...>::template cccx<M>
|
||||||
|
(res, in, inds..., op, ops..., dops...);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
};
|
||||||
|
};
|
||||||
|
};
|
||||||
|
|
||||||
|
template <>
|
||||||
|
struct Create<0>
|
||||||
|
{
|
||||||
|
template <class... Indices>
|
||||||
|
struct cx
|
||||||
|
{
|
||||||
|
template <class ROP, class OpF, class... OPs>
|
||||||
|
struct ccx
|
||||||
|
{
|
||||||
|
template <size_t M, class... DOPs>
|
||||||
|
static inline void
|
||||||
|
cccx(typename HighLevelOpBase<ROP>::RetT<Indices...>& res,
|
||||||
|
const std::array<std::shared_ptr<HighLevelOpBase<ROP>>,M>& in,
|
||||||
|
const std::shared_ptr<Indices>&... inds,
|
||||||
|
const OPs&... ops,
|
||||||
|
const DOPs&... dops)
|
||||||
|
{
|
||||||
|
auto& inn = std::get<0>(in);
|
||||||
|
if(not inn->root()){
|
||||||
|
auto dop = inn->create(inds...);
|
||||||
|
auto op = *dop.op.data()->mOp;
|
||||||
|
res.appendOuter(dop);
|
||||||
|
res.op = mkDynOutOp(mkFOp<OpF>(op,ops...), inds...);
|
||||||
|
res.appendOuterM(dop.op,dops.op...);
|
||||||
|
}
|
||||||
|
else {
|
||||||
|
auto& op = *inn->get();
|
||||||
|
res.op = mkDynOutOp(mkFOp<OpF>(op,ops...), inds...);
|
||||||
|
res.appendOuterM(dops.op...);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
};
|
||||||
|
};
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
template <class ROP, class OpF, size_t N>
|
||||||
|
HighLevelOp<ROP,OpF,N>::HighLevelOp(std::array<std::shared_ptr<HighLevelOpBase<ROP>>,N> in) : mIn(in) {}
|
||||||
|
|
||||||
|
template <class ROP, class OpF, size_t N>
|
||||||
|
bool HighLevelOp<ROP,OpF,N>::root() const
|
||||||
|
{
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
template <class ROP, class OpF, size_t N>
|
||||||
|
const ROP* HighLevelOp<ROP,OpF,N>::get() const
|
||||||
|
{ assert(0); return nullptr; }
|
||||||
|
|
||||||
|
template <class ROP, class OpF, size_t N>
|
||||||
|
auto HighLevelOp<ROP,OpF,N>::create(const std::shared_ptr<CI> ind1,
|
||||||
|
const std::shared_ptr<CI> ind2)
|
||||||
|
-> typename B::RetT<CI,CI>
|
||||||
|
{
|
||||||
|
typename B::RetT<CI,CI> res;
|
||||||
|
Create<N-1>::template cx<CI,CI>::template ccx<ROP,OpF>::template cccx<N>(res,mIn,ind1,ind2);
|
||||||
|
return res;
|
||||||
|
}
|
||||||
|
|
||||||
|
template <class ROP>
|
||||||
|
HighLevelOpHolder<ROP>::HighLevelOpHolder(const std::shared_ptr<HighLevelOpBase<ROP>>& op) : mOp(op) {}
|
||||||
|
|
||||||
|
template <class ROP>
|
||||||
|
bool HighLevelOpHolder<ROP>::root() const
|
||||||
|
{
|
||||||
|
return mOp->root();
|
||||||
|
}
|
||||||
|
|
||||||
|
template <class ROP>
|
||||||
|
auto HighLevelOpHolder<ROP>::create(const std::shared_ptr<CI> ind1,
|
||||||
|
const std::shared_ptr<CI> ind2) const
|
||||||
|
-> decltype(mOp->create(ind1,ind2))
|
||||||
|
{
|
||||||
|
return mOp->create(ind1,ind2);
|
||||||
|
}
|
||||||
|
|
||||||
|
template <class ROP>
|
||||||
|
auto HighLevelOpHolder<ROP>::get() const
|
||||||
|
-> decltype(mOp->get())
|
||||||
|
{
|
||||||
|
return mOp->get();
|
||||||
|
}
|
||||||
|
|
||||||
|
template <class ROP>
|
||||||
|
std::shared_ptr<HighLevelOpBase<ROP>> HighLevelOpHolder<ROP>::op() const
|
||||||
|
{
|
||||||
|
return mOp;
|
||||||
|
}
|
||||||
|
|
||||||
|
template <class ROP>
|
||||||
|
HighLevelOpHolder<ROP> HighLevelOpHolder<ROP>::operator*(const HighLevelOpHolder& in) const
|
||||||
|
{
|
||||||
|
return HighLevelOpHolder<ROP>
|
||||||
|
( std::make_shared<HighLevelOp<ROP,multipliesx<double,double>,2>>
|
||||||
|
( std::array<std::shared_ptr<HighLevelOpBase<ROP>>,2>({mOp, in.mOp}) ) );
|
||||||
|
}
|
||||||
|
|
||||||
|
template <class ROP>
|
||||||
|
HighLevelOpHolder<ROP> HighLevelOpHolder<ROP>::operator+(const HighLevelOpHolder& in) const
|
||||||
|
{
|
||||||
|
return HighLevelOpHolder<ROP>
|
||||||
|
( std::make_shared<HighLevelOp<ROP,plusx<double,double>,2>>
|
||||||
|
( std::array<std::shared_ptr<HighLevelOpBase<ROP>>,2>({mOp, in.mOp}) ) );
|
||||||
|
}
|
||||||
|
|
||||||
|
template <class ROP>
|
||||||
|
HighLevelOpHolder<ROP> HighLevelOpHolder<ROP>::operator-(const HighLevelOpHolder& in) const
|
||||||
|
{
|
||||||
|
return HighLevelOpHolder<ROP>
|
||||||
|
( std::make_shared<HighLevelOp<ROP,minusx<double,double>,2>>
|
||||||
|
( std::array<std::shared_ptr<HighLevelOpBase<ROP>>,2>({mOp, in.mOp}) ) );
|
||||||
|
}
|
||||||
|
|
||||||
|
template <class ROP>
|
||||||
|
HighLevelOpHolder<ROP> HighLevelOpHolder<ROP>::operator/(const HighLevelOpHolder& in) const
|
||||||
|
{
|
||||||
|
return HighLevelOpHolder<ROP>
|
||||||
|
( std::make_shared<HighLevelOp<ROP,dividesx<double,double>,2>>
|
||||||
|
( std::array<std::shared_ptr<HighLevelOpBase<ROP>>,2>({mOp, in.mOp}) ) );
|
||||||
|
}
|
||||||
|
|
||||||
|
template <class ROP>
|
||||||
|
template <class MIndex, class... Indices>
|
||||||
|
HighLevelOpHolder<ROP>& HighLevelOpHolder<ROP>::assign(const HighLevelOpHolder& in,
|
||||||
|
const std::shared_ptr<MIndex>& mi,
|
||||||
|
const std::shared_ptr<Indices>&... inds)
|
||||||
|
{
|
||||||
|
auto xx = mkArrayPtr<double>(nullr());
|
||||||
|
auto& opr = *mOp->get();
|
||||||
|
auto loop = mkPILoop
|
||||||
|
( [&opr,&in,&xx,&inds...,this](){
|
||||||
|
auto inx = in;
|
||||||
|
auto dop = inx.create(inds...);
|
||||||
|
auto gexp = mkDynOp1<size_t>(mkMOp<size_t>(dop.outer,dop.op));
|
||||||
|
auto xloop = mkILoop(std::make_tuple(*dop.op.data()->mOp),
|
||||||
|
std::make_tuple(inds...),
|
||||||
|
std::make_tuple(xx),
|
||||||
|
std::make_tuple(opr.assign( *dop.op.data()->mOp,
|
||||||
|
mkMIndex(inds...) )),
|
||||||
|
std::array<size_t,1>({1}), std::array<size_t,1>({0}));
|
||||||
|
return mkGetExpr(gexp, xloop); });
|
||||||
|
mi->pifor(1,loop)();
|
||||||
|
return *this;
|
||||||
|
}
|
||||||
|
|
||||||
|
template <class ROP>
|
||||||
|
template <class MIndex, class... Indices>
|
||||||
|
HighLevelOpHolder<ROP>& HighLevelOpHolder<ROP>::plus(const HighLevelOpHolder& in,
|
||||||
|
const std::shared_ptr<MIndex>& mi,
|
||||||
|
const std::shared_ptr<Indices>&... inds)
|
||||||
|
{
|
||||||
|
auto xx = mkArrayPtr<double>(nullr());
|
||||||
|
auto& opr = *mOp->get();
|
||||||
|
auto loop = mkPILoop
|
||||||
|
( [&opr,&in,&xx,&inds...,this](){
|
||||||
|
auto inx = in;
|
||||||
|
auto dop = inx.create(inds...);
|
||||||
|
auto gexp = mkDynOp1<size_t>(mkMOp<size_t>(dop.outer,dop.op));
|
||||||
|
auto xloop = mkILoop(std::make_tuple(*dop.op.data()->mOp),
|
||||||
|
std::make_tuple(inds...),
|
||||||
|
std::make_tuple(xx),
|
||||||
|
std::make_tuple(opr.plus( *dop.op.data()->mOp,
|
||||||
|
mkMIndex(inds...) )),
|
||||||
|
std::array<size_t,1>({1}), std::array<size_t,1>({0}));
|
||||||
|
return mkGetExpr(gexp, xloop); });
|
||||||
|
mi->pifor(1,loop)();
|
||||||
|
return *this;
|
||||||
|
}
|
||||||
|
|
||||||
|
template <class ROP>
|
||||||
|
HighLevelOpHolder<ROP> mkHLO(const ROP& op)
|
||||||
|
{
|
||||||
|
return HighLevelOpHolder<ROP>(std::make_shared<HighLevelOpRoot<ROP>>( op ) );
|
||||||
|
}
|
||||||
|
|
||||||
|
template <class ROP>
|
||||||
|
HighLevelOpHolder<ROP> exp(const HighLevelOpHolder<ROP>& in)
|
||||||
|
{
|
||||||
|
return HighLevelOpHolder<ROP>
|
||||||
|
( std::make_shared<HighLevelOp<ROP,x_exp<double>,1>>
|
||||||
|
( std::array<std::shared_ptr<HighLevelOpBase<ROP>>,1>( {in.op()} ) ) );
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
148
src/include/high_level_operation.h
Normal file
148
src/include/high_level_operation.h
Normal file
|
|
@ -0,0 +1,148 @@
|
||||||
|
|
||||||
|
#ifndef __high_level_operation_h__
|
||||||
|
#define __high_level_operation_h__
|
||||||
|
|
||||||
|
#include "base_def.h"
|
||||||
|
#include "ranges/rheader.h"
|
||||||
|
#include "dynamic_operation.h"
|
||||||
|
|
||||||
|
namespace MultiArrayTools
|
||||||
|
{
|
||||||
|
|
||||||
|
typedef ClassicRange CR;
|
||||||
|
typedef CR::IndexType CI;
|
||||||
|
typedef std::shared_ptr<CI> CIP;
|
||||||
|
|
||||||
|
template <typename T, class Op>
|
||||||
|
DynamicO<T> mkDynOp1(const Op& op);
|
||||||
|
|
||||||
|
template <class ROP>
|
||||||
|
class HighLevelOpBase
|
||||||
|
{
|
||||||
|
public:
|
||||||
|
|
||||||
|
template <class... Indices>
|
||||||
|
struct RetT
|
||||||
|
{
|
||||||
|
DynamicO<OpH<OperationRoot<double,typename Indices::RangeType...>>> op;
|
||||||
|
DynamicO<size_t> outer;
|
||||||
|
|
||||||
|
template <class Op, class... Ops>
|
||||||
|
void appendOuterM(const Op& op, const Ops&... ops);
|
||||||
|
|
||||||
|
void appendOuterM();
|
||||||
|
void appendOuter(const DynamicO<size_t>& in);
|
||||||
|
void appendOuter(const RetT& in);
|
||||||
|
|
||||||
|
};
|
||||||
|
|
||||||
|
virtual bool root() const = 0;
|
||||||
|
|
||||||
|
virtual RetT<CI,CI> create(const std::shared_ptr<CI> ind1,
|
||||||
|
const std::shared_ptr<CI> ind2) = 0;
|
||||||
|
|
||||||
|
virtual const ROP* get() const = 0;
|
||||||
|
|
||||||
|
};
|
||||||
|
|
||||||
|
template <class ROP>
|
||||||
|
class HighLevelOpRoot : public HighLevelOpBase<ROP>
|
||||||
|
{
|
||||||
|
private:
|
||||||
|
typedef HighLevelOpBase<ROP> B;
|
||||||
|
|
||||||
|
ROP mOp;
|
||||||
|
public:
|
||||||
|
|
||||||
|
HighLevelOpRoot(const ROP& op);
|
||||||
|
|
||||||
|
virtual bool root() const override final;
|
||||||
|
|
||||||
|
virtual typename B::RetT<CI,CI> create(const std::shared_ptr<CI> ind1,
|
||||||
|
const std::shared_ptr<CI> ind2) override final;
|
||||||
|
|
||||||
|
virtual const ROP* get() const override final;
|
||||||
|
|
||||||
|
|
||||||
|
};
|
||||||
|
|
||||||
|
template <class OpF, class... Ops>
|
||||||
|
auto mkFOp(const Ops&... ops)
|
||||||
|
{
|
||||||
|
return Operation<double,OpF,Ops...>(ops...);
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
template <class ROP, class OpF, size_t N>
|
||||||
|
class HighLevelOp : public HighLevelOpBase<ROP>
|
||||||
|
{
|
||||||
|
private:
|
||||||
|
std::array<std::shared_ptr<HighLevelOpBase<ROP>>,N> mIn;
|
||||||
|
|
||||||
|
public:
|
||||||
|
typedef HighLevelOpBase<ROP> B;
|
||||||
|
|
||||||
|
HighLevelOp(std::array<std::shared_ptr<HighLevelOpBase<ROP>>,N> in);
|
||||||
|
|
||||||
|
virtual bool root() const override final;
|
||||||
|
|
||||||
|
virtual const ROP* get() const override final;
|
||||||
|
|
||||||
|
virtual typename B::RetT<CI,CI> create(const std::shared_ptr<CI> ind1,
|
||||||
|
const std::shared_ptr<CI> ind2) override final;
|
||||||
|
|
||||||
|
};
|
||||||
|
|
||||||
|
|
||||||
|
template <class ROP>
|
||||||
|
class HighLevelOpHolder
|
||||||
|
{
|
||||||
|
private:
|
||||||
|
std::shared_ptr<HighLevelOpBase<ROP>> mOp;
|
||||||
|
|
||||||
|
public:
|
||||||
|
HighLevelOpHolder() = default;
|
||||||
|
HighLevelOpHolder(const HighLevelOpHolder& in) = default;
|
||||||
|
HighLevelOpHolder(HighLevelOpHolder&& in) = default;
|
||||||
|
HighLevelOpHolder& operator=(const HighLevelOpHolder& in) = default;
|
||||||
|
HighLevelOpHolder& operator=(HighLevelOpHolder&& in) = default;
|
||||||
|
|
||||||
|
HighLevelOpHolder(const std::shared_ptr<HighLevelOpBase<ROP>>& op);
|
||||||
|
|
||||||
|
bool root() const;
|
||||||
|
|
||||||
|
auto create(const std::shared_ptr<CI> ind1,
|
||||||
|
const std::shared_ptr<CI> ind2) const
|
||||||
|
-> decltype(mOp->create(ind1,ind2));
|
||||||
|
|
||||||
|
auto get() const -> decltype(mOp->get());
|
||||||
|
|
||||||
|
std::shared_ptr<HighLevelOpBase<ROP>> op() const;
|
||||||
|
HighLevelOpHolder operator*(const HighLevelOpHolder& in) const;
|
||||||
|
HighLevelOpHolder operator+(const HighLevelOpHolder& in) const;
|
||||||
|
HighLevelOpHolder operator-(const HighLevelOpHolder& in) const;
|
||||||
|
HighLevelOpHolder operator/(const HighLevelOpHolder& in) const;
|
||||||
|
|
||||||
|
template <class MIndex, class... Indices>
|
||||||
|
HighLevelOpHolder& assign(const HighLevelOpHolder& in,
|
||||||
|
const std::shared_ptr<MIndex>& mi,
|
||||||
|
const std::shared_ptr<Indices>&... inds);
|
||||||
|
|
||||||
|
template <class MIndex, class... Indices>
|
||||||
|
HighLevelOpHolder& plus(const HighLevelOpHolder& in,
|
||||||
|
const std::shared_ptr<MIndex>& mi,
|
||||||
|
const std::shared_ptr<Indices>&... inds);
|
||||||
|
};
|
||||||
|
|
||||||
|
|
||||||
|
template <class ROP>
|
||||||
|
HighLevelOpHolder<ROP> mkHLO(const ROP& op);
|
||||||
|
|
||||||
|
template <class ROP>
|
||||||
|
HighLevelOpHolder<ROP> exp(const HighLevelOpHolder<ROP>& in);
|
||||||
|
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
#endif
|
||||||
|
|
@ -9,5 +9,6 @@
|
||||||
#include "multi_array.cc.h"
|
#include "multi_array.cc.h"
|
||||||
#include "slice.cc.h"
|
#include "slice.cc.h"
|
||||||
#include "dynamic_operation.cc.h"
|
#include "dynamic_operation.cc.h"
|
||||||
|
#include "high_level_operation.cc.h"
|
||||||
//#include "expressions.cc.h"
|
//#include "expressions.cc.h"
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -12,6 +12,7 @@
|
||||||
#include "operation_def.h"
|
#include "operation_def.h"
|
||||||
#include "map_range.h"
|
#include "map_range.h"
|
||||||
#include "dynamic_operation.h"
|
#include "dynamic_operation.h"
|
||||||
|
#include "high_level_operation.h"
|
||||||
//#include "expressions.h"
|
//#include "expressions.h"
|
||||||
|
|
||||||
#include "multi_array_header.cc.h"
|
#include "multi_array_header.cc.h"
|
||||||
|
|
|
||||||
|
|
@ -1,5 +1,6 @@
|
||||||
|
|
||||||
#include "op4_unit_test.h"
|
#include "test_header.h"
|
||||||
|
#include <ctime>
|
||||||
|
|
||||||
namespace
|
namespace
|
||||||
{
|
{
|
||||||
|
|
|
||||||
|
|
@ -1,329 +0,0 @@
|
||||||
|
|
||||||
#include "test_header.h"
|
|
||||||
#include <ctime>
|
|
||||||
|
|
||||||
typedef OperationRoot<double,CR,DR> OpCD;
|
|
||||||
typedef OperationRoot<double,DR> OpD;
|
|
||||||
typedef CR::IndexType CI;
|
|
||||||
typedef std::shared_ptr<CI> CIP;
|
|
||||||
|
|
||||||
template <typename T, class Op>
|
|
||||||
DynamicO<T> mkDynOp1(const Op& op)
|
|
||||||
{
|
|
||||||
return DynamicO<T>(op);
|
|
||||||
}
|
|
||||||
|
|
||||||
template <class ROP>
|
|
||||||
class HighLevelOpBase
|
|
||||||
{
|
|
||||||
public:
|
|
||||||
|
|
||||||
typedef OperationRoot<double,CR,CR> OpCC;
|
|
||||||
typedef OperationRoot<double,CR,DR> OpCD;
|
|
||||||
typedef OperationRoot<double,DR> OpD;
|
|
||||||
|
|
||||||
template <class... Indices>
|
|
||||||
struct RetT
|
|
||||||
{
|
|
||||||
DynamicO<OpH<OperationRoot<double,typename Indices::RangeType...>>> op;
|
|
||||||
DynamicO<size_t> outer;
|
|
||||||
|
|
||||||
template <class Op, class... Ops>
|
|
||||||
void appendOuterM(const Op& op, const Ops&... ops)
|
|
||||||
{
|
|
||||||
// does not check anything regarding input !!!
|
|
||||||
if(outer.init()){
|
|
||||||
outer = mkDynOp1<size_t>(mkMOp<size_t>(outer,op,ops...));
|
|
||||||
}
|
|
||||||
else {
|
|
||||||
outer = mkDynOp1<size_t>(mkMOp<size_t>(op,ops...));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
void appendOuterM()
|
|
||||||
{}
|
|
||||||
|
|
||||||
void appendOuter(const DynamicO<size_t>& in)
|
|
||||||
{
|
|
||||||
if(in.init()){
|
|
||||||
if(outer.init()){
|
|
||||||
outer = mkDynOp1<size_t>(mkMOp<size_t>(outer,in));
|
|
||||||
}
|
|
||||||
else {
|
|
||||||
outer = in;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
void appendOuter(const RetT& in)
|
|
||||||
{
|
|
||||||
appendOuter(in.outer);
|
|
||||||
}
|
|
||||||
|
|
||||||
};
|
|
||||||
|
|
||||||
virtual bool root() const = 0;
|
|
||||||
|
|
||||||
virtual RetT<CI,CI> create(const std::shared_ptr<CI> ind1,
|
|
||||||
const std::shared_ptr<CI> ind2) = 0;
|
|
||||||
|
|
||||||
//virtual const OperationRoot<double,CR,DR>* get2() const = 0;
|
|
||||||
virtual const ROP* get() const = 0;
|
|
||||||
|
|
||||||
};
|
|
||||||
|
|
||||||
template <class ROP>
|
|
||||||
class HighLevelOpRoot : public HighLevelOpBase<ROP>
|
|
||||||
{
|
|
||||||
private:
|
|
||||||
typedef HighLevelOpBase<ROP> B;
|
|
||||||
|
|
||||||
ROP mOp;
|
|
||||||
public:
|
|
||||||
|
|
||||||
HighLevelOpRoot(const ROP& op) : mOp(op) {}
|
|
||||||
|
|
||||||
virtual bool root() const override final
|
|
||||||
{
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
virtual typename B::RetT<CI,CI> create(const std::shared_ptr<CI> ind1,
|
|
||||||
const std::shared_ptr<CI> ind2) override final
|
|
||||||
{
|
|
||||||
assert(0);
|
|
||||||
return typename B::RetT<CI,CI>();
|
|
||||||
}
|
|
||||||
|
|
||||||
virtual const ROP* get() const override final
|
|
||||||
{
|
|
||||||
return &mOp;
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
};
|
|
||||||
|
|
||||||
template <class OpF, class... Ops>
|
|
||||||
auto mkFOp(const Ops&... ops)
|
|
||||||
{
|
|
||||||
return Operation<double,OpF,Ops...>(ops...);
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
template <size_t N>
|
|
||||||
struct Create
|
|
||||||
{
|
|
||||||
template <class... Indices>
|
|
||||||
struct cx
|
|
||||||
{
|
|
||||||
template <class ROP, class OpF, class... OPs>
|
|
||||||
struct ccx
|
|
||||||
{
|
|
||||||
template <size_t M, class... DOPs>
|
|
||||||
static inline void
|
|
||||||
cccx(typename HighLevelOpBase<ROP>::RetT<Indices...>& res,
|
|
||||||
const std::array<std::shared_ptr<HighLevelOpBase<ROP>>,M>& in,
|
|
||||||
const std::shared_ptr<Indices>&... inds,
|
|
||||||
const OPs&... ops,
|
|
||||||
const DOPs&... dops)
|
|
||||||
{
|
|
||||||
static_assert(N > 0, "N > 0 failed");
|
|
||||||
auto& inn = std::get<N>(in);
|
|
||||||
if(not inn->root()){
|
|
||||||
auto dop = inn->create(inds...);
|
|
||||||
auto op = *dop.op.data()->mOp;
|
|
||||||
typedef decltype(op) OP;
|
|
||||||
res.appendOuter(dop);
|
|
||||||
Create<N-1>::template cx<Indices...>::template ccx<ROP,OpF,OP,OPs...>::template cccx<M>
|
|
||||||
(res, in, inds..., op, ops..., dop, dops...);
|
|
||||||
}
|
|
||||||
else {
|
|
||||||
auto& op = *inn->get();
|
|
||||||
typedef typename std::remove_reference<decltype(op)>::type OP;
|
|
||||||
Create<N-1>::template cx<Indices...>::template ccx<ROP,OpF,OP,OPs...>::template cccx<M>
|
|
||||||
(res, in, inds..., op, ops..., dops...);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
};
|
|
||||||
};
|
|
||||||
};
|
|
||||||
|
|
||||||
template <>
|
|
||||||
struct Create<0>
|
|
||||||
{
|
|
||||||
template <class... Indices>
|
|
||||||
struct cx
|
|
||||||
{
|
|
||||||
template <class ROP, class OpF, class... OPs>
|
|
||||||
struct ccx
|
|
||||||
{
|
|
||||||
template <size_t M, class... DOPs>
|
|
||||||
static inline void
|
|
||||||
cccx(typename HighLevelOpBase<ROP>::RetT<Indices...>& res,
|
|
||||||
const std::array<std::shared_ptr<HighLevelOpBase<ROP>>,M>& in,
|
|
||||||
const std::shared_ptr<Indices>&... inds,
|
|
||||||
const OPs&... ops,
|
|
||||||
const DOPs&... dops)
|
|
||||||
{
|
|
||||||
auto& inn = std::get<0>(in);
|
|
||||||
if(not inn->root()){
|
|
||||||
auto dop = inn->create(inds...);
|
|
||||||
auto op = *dop.op.data()->mOp;
|
|
||||||
res.appendOuter(dop);
|
|
||||||
res.op = mkDynOutOp(mkFOp<OpF>(op,ops...), inds...);
|
|
||||||
res.appendOuterM(dop.op,dops.op...);
|
|
||||||
}
|
|
||||||
else {
|
|
||||||
auto& op = *inn->get();
|
|
||||||
res.op = mkDynOutOp(mkFOp<OpF>(op,ops...), inds...);
|
|
||||||
res.appendOuterM(dops.op...);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
};
|
|
||||||
};
|
|
||||||
};
|
|
||||||
|
|
||||||
template <class ROP, class OpF, size_t N>
|
|
||||||
class HighLevelOp : public HighLevelOpBase<ROP>
|
|
||||||
{
|
|
||||||
private:
|
|
||||||
std::array<std::shared_ptr<HighLevelOpBase<ROP>>,N> mIn;
|
|
||||||
|
|
||||||
public:
|
|
||||||
typedef HighLevelOpBase<ROP> B;
|
|
||||||
|
|
||||||
HighLevelOp(std::array<std::shared_ptr<HighLevelOpBase<ROP>>,N> in) : mIn(in) {}
|
|
||||||
|
|
||||||
virtual bool root() const override final
|
|
||||||
{
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
virtual const ROP* get() const override final
|
|
||||||
{ assert(0); return nullptr; }
|
|
||||||
|
|
||||||
virtual typename B::RetT<CI,CI> create(const std::shared_ptr<CI> ind1,
|
|
||||||
const std::shared_ptr<CI> ind2) override final
|
|
||||||
{
|
|
||||||
typename B::RetT<CI,CI> res;
|
|
||||||
Create<N-1>::template cx<CI,CI>::template ccx<ROP,OpF>::template cccx<N>(res,mIn,ind1,ind2);
|
|
||||||
return res;
|
|
||||||
}
|
|
||||||
|
|
||||||
};
|
|
||||||
|
|
||||||
|
|
||||||
template <class ROP>
|
|
||||||
class HighLevelOpHolder
|
|
||||||
{
|
|
||||||
private:
|
|
||||||
std::shared_ptr<HighLevelOpBase<ROP>> mOp;
|
|
||||||
|
|
||||||
public:
|
|
||||||
HighLevelOpHolder() = default;
|
|
||||||
HighLevelOpHolder(const HighLevelOpHolder& in) = default;
|
|
||||||
HighLevelOpHolder(HighLevelOpHolder&& in) = default;
|
|
||||||
HighLevelOpHolder& operator=(const HighLevelOpHolder& in) = default;
|
|
||||||
HighLevelOpHolder& operator=(HighLevelOpHolder&& in) = default;
|
|
||||||
|
|
||||||
HighLevelOpHolder(const std::shared_ptr<HighLevelOpBase<ROP>>& op) : mOp(op) {}
|
|
||||||
|
|
||||||
bool root() const { return mOp->root(); }
|
|
||||||
|
|
||||||
auto create(const std::shared_ptr<CI> ind1,
|
|
||||||
const std::shared_ptr<CI> ind2) const { return mOp->create(ind1,ind2); }
|
|
||||||
|
|
||||||
auto get() const { return mOp->get(); }
|
|
||||||
|
|
||||||
std::shared_ptr<HighLevelOpBase<ROP>> op() const { return mOp; }
|
|
||||||
|
|
||||||
HighLevelOpHolder operator*(const HighLevelOpHolder& in) const
|
|
||||||
{
|
|
||||||
return HighLevelOpHolder<ROP>
|
|
||||||
( std::make_shared<HighLevelOp<ROP,multipliesx<double,double>,2>>
|
|
||||||
( std::array<std::shared_ptr<HighLevelOpBase<ROP>>,2>({mOp, in.mOp}) ) );
|
|
||||||
}
|
|
||||||
|
|
||||||
HighLevelOpHolder operator+(const HighLevelOpHolder& in) const
|
|
||||||
{
|
|
||||||
return HighLevelOpHolder<ROP>
|
|
||||||
( std::make_shared<HighLevelOp<ROP,plusx<double,double>,2>>
|
|
||||||
( std::array<std::shared_ptr<HighLevelOpBase<ROP>>,2>({mOp, in.mOp}) ) );
|
|
||||||
}
|
|
||||||
|
|
||||||
HighLevelOpHolder operator-(const HighLevelOpHolder& in) const
|
|
||||||
{
|
|
||||||
return HighLevelOpHolder<ROP>
|
|
||||||
( std::make_shared<HighLevelOp<ROP,minusx<double,double>,2>>
|
|
||||||
( std::array<std::shared_ptr<HighLevelOpBase<ROP>>,2>({mOp, in.mOp}) ) );
|
|
||||||
}
|
|
||||||
|
|
||||||
HighLevelOpHolder operator/(const HighLevelOpHolder& in) const
|
|
||||||
{
|
|
||||||
return HighLevelOpHolder<ROP>
|
|
||||||
( std::make_shared<HighLevelOp<ROP,dividesx<double,double>,2>>
|
|
||||||
( std::array<std::shared_ptr<HighLevelOpBase<ROP>>,2>({mOp, in.mOp}) ) );
|
|
||||||
}
|
|
||||||
|
|
||||||
template <class MIndex, class... Indices>
|
|
||||||
HighLevelOpHolder& assign(const HighLevelOpHolder& in,
|
|
||||||
const std::shared_ptr<MIndex>& mi,
|
|
||||||
const std::shared_ptr<Indices>&... inds)
|
|
||||||
{
|
|
||||||
auto xx = mkArrayPtr<double>(nullr());
|
|
||||||
auto& opr = *mOp->get();
|
|
||||||
auto loop = mkPILoop
|
|
||||||
( [&opr,&in,&xx,&inds...,this](){
|
|
||||||
auto inx = in;
|
|
||||||
auto dop = inx.create(inds...);
|
|
||||||
auto gexp = mkDynOp1<size_t>(mkMOp<size_t>(dop.outer,dop.op));
|
|
||||||
auto xloop = mkILoop(std::make_tuple(*dop.op.data()->mOp),
|
|
||||||
std::make_tuple(inds...),
|
|
||||||
std::make_tuple(xx),
|
|
||||||
std::make_tuple(opr.assign( *dop.op.data()->mOp,
|
|
||||||
mkMIndex(inds...) )),
|
|
||||||
std::array<size_t,1>({1}), std::array<size_t,1>({0}));
|
|
||||||
return mkGetExpr(gexp, xloop); });
|
|
||||||
mi->pifor(1,loop)();
|
|
||||||
return *this;
|
|
||||||
}
|
|
||||||
|
|
||||||
template <class MIndex, class... Indices>
|
|
||||||
HighLevelOpHolder& plus(const HighLevelOpHolder& in,
|
|
||||||
const std::shared_ptr<MIndex>& mi,
|
|
||||||
const std::shared_ptr<Indices>&... inds)
|
|
||||||
{
|
|
||||||
auto xx = mkArrayPtr<double>(nullr());
|
|
||||||
auto& opr = *mOp->get();
|
|
||||||
auto loop = mkPILoop
|
|
||||||
( [&opr,&in,&xx,&inds...,this](){
|
|
||||||
auto inx = in;
|
|
||||||
auto dop = inx.create(inds...);
|
|
||||||
auto gexp = mkDynOp1<size_t>(mkMOp<size_t>(dop.outer,dop.op));
|
|
||||||
auto xloop = mkILoop(std::make_tuple(*dop.op.data()->mOp),
|
|
||||||
std::make_tuple(inds...),
|
|
||||||
std::make_tuple(xx),
|
|
||||||
std::make_tuple(opr.plus( *dop.op.data()->mOp,
|
|
||||||
mkMIndex(inds...) )),
|
|
||||||
std::array<size_t,1>({1}), std::array<size_t,1>({0}));
|
|
||||||
return mkGetExpr(gexp, xloop); });
|
|
||||||
mi->pifor(1,loop)();
|
|
||||||
return *this;
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
typedef OperationRoot<double,CR,DR> OpType1;
|
|
||||||
|
|
||||||
template <class ROP>
|
|
||||||
HighLevelOpHolder<ROP> mkHLO(const ROP& op)
|
|
||||||
{
|
|
||||||
return HighLevelOpHolder<ROP>(std::make_shared<HighLevelOpRoot<ROP>>( op ) );
|
|
||||||
}
|
|
||||||
|
|
||||||
template <class ROP>
|
|
||||||
HighLevelOpHolder<ROP> exp(const HighLevelOpHolder<ROP>& in)
|
|
||||||
{
|
|
||||||
return HighLevelOpHolder<ROP>
|
|
||||||
( std::make_shared<HighLevelOp<ROP,x_exp<double>,1>>
|
|
||||||
( std::array<std::shared_ptr<HighLevelOpBase<ROP>>,1>( {in.op()} ) ) );
|
|
||||||
}
|
|
||||||
Loading…
Reference in a new issue