2017-02-16 11:20:40 +01:00
|
|
|
// -*- C++ -*-
|
|
|
|
|
|
|
|
#ifndef __multi_array_operation_h__
|
|
|
|
#define __multi_array_operation_h__
|
|
|
|
|
|
|
|
#include <cstdlib>
|
|
|
|
#include <tuple>
|
|
|
|
|
|
|
|
#include "base_def.h"
|
|
|
|
#include "index_base.h"
|
|
|
|
|
|
|
|
namespace MultiArrayTools
|
|
|
|
{
|
|
|
|
|
|
|
|
template <typename T, class Range>
|
|
|
|
class MultiArrayOperationBase
|
|
|
|
{
|
|
|
|
public:
|
2017-02-21 21:47:40 +01:00
|
|
|
|
|
|
|
typedef decltype(MultiArray<T,Range>().begin()) IndexType;
|
2017-02-16 11:20:40 +01:00
|
|
|
|
2017-02-21 21:47:40 +01:00
|
|
|
MultiArrayOperationBase(MultiArray<T,Range>& ma, const Name& nm);
|
|
|
|
MultiArrayOperationBase& operator=(const MultiArrayOperationBase& in);
|
|
|
|
|
|
|
|
virtual ~MultiArrayOperationBase();
|
2017-02-16 11:20:40 +01:00
|
|
|
|
|
|
|
// execute AnyOperation
|
|
|
|
// exception if range types are inconsitent with names
|
|
|
|
template <class Range2>
|
|
|
|
MultiArrayOperationBase& operator=(const MultiArrayOperationBase<T, Range2>& in);
|
|
|
|
|
|
|
|
|
|
|
|
template <class Operation, class... Ranges>
|
2017-02-17 18:10:03 +01:00
|
|
|
MultiArrayOperation<T,Range,Operation,Ranges...>
|
|
|
|
operator()(Operation& op, MultiArrayOperationBase<T,Ranges>&... secs);
|
2017-02-16 11:20:40 +01:00
|
|
|
|
|
|
|
template <class Range2>
|
2017-02-17 18:10:03 +01:00
|
|
|
MultiArrayOperation<T,Range,std::plus<T>,Range2> operator+(MultiArrayOperationBase<T,Range2>& sec);
|
2017-02-16 11:20:40 +01:00
|
|
|
|
|
|
|
template <class Range2>
|
2017-02-17 18:10:03 +01:00
|
|
|
MultiArrayOperation<T,Range,std::minus<T>,Range2> operator-(MultiArrayOperationBase<T,Range2>& sec);
|
2017-02-16 11:20:40 +01:00
|
|
|
|
|
|
|
template <class Range2>
|
2017-02-17 18:10:03 +01:00
|
|
|
MultiArrayOperation<T,Range,std::multiplies<T>,Range2> operator*(MultiArrayOperationBase<T,Range2>& sec);
|
2017-02-16 11:20:40 +01:00
|
|
|
|
|
|
|
template <class Range2>
|
2017-02-17 18:10:03 +01:00
|
|
|
MultiArrayOperation<T,Range,std::divides<T>,Range2> operator/(MultiArrayOperationBase<T,Range2>& sec);
|
2017-02-16 11:20:40 +01:00
|
|
|
|
|
|
|
virtual size_t argNum() const;
|
|
|
|
|
|
|
|
IndefinitIndexBase* index();
|
|
|
|
|
|
|
|
virtual void linkIndicesTo(IndefinitIndexBase* target);
|
|
|
|
|
|
|
|
virtual T& get();
|
|
|
|
virtual const T& get() const;
|
|
|
|
|
|
|
|
protected:
|
|
|
|
|
2017-02-21 21:47:40 +01:00
|
|
|
// HERE !!!!!!
|
|
|
|
|
2017-02-16 11:20:40 +01:00
|
|
|
MultiArray<T,Range>& mArrayRef;
|
2017-02-21 21:47:40 +01:00
|
|
|
IndefinitIndexBase* mIibPtr = nullptr;
|
2017-02-16 11:20:40 +01:00
|
|
|
|
|
|
|
};
|
|
|
|
|
|
|
|
template <typename T, class Range, class Operation, class... Ranges>
|
|
|
|
class MultiArrayOperation : public MultiArrayOperationBase<T,Range>
|
|
|
|
{
|
|
|
|
public:
|
|
|
|
|
2017-02-17 18:10:03 +01:00
|
|
|
typedef MultiArrayOperationBase<T,Range> OB;
|
|
|
|
|
2017-02-16 11:20:40 +01:00
|
|
|
MultiArrayOperation(Operation& op, MultiArrayOperationBase<T,Ranges>&... secs);
|
|
|
|
virtual size_t argNum() const override;
|
|
|
|
|
|
|
|
virtual void linkIndicesTo(IndefinitIndexBase* target) override;
|
|
|
|
|
|
|
|
virtual T& get() override;
|
|
|
|
virtual const T& get() const override;
|
|
|
|
|
|
|
|
protected:
|
|
|
|
|
|
|
|
T mVal;
|
|
|
|
Operation mOp;
|
|
|
|
std::tuple<MultiArrayOperationBase<T,Ranges>... > mSecs;
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
}
|
|
|
|
|
2017-02-17 18:10:03 +01:00
|
|
|
#include "multi_array_operation.cc"
|
2017-02-16 11:20:40 +01:00
|
|
|
|
|
|
|
#endif
|