cnorxz/src/xfor/xfor.h

#ifndef __xfor_h__
#define __xfor_h__

#include <cstdlib>
#include <memory>
#include <tuple>
#include "xfor/for_utils.h"

namespace MultiArrayHelper
{
    
    template <class IndexClass, class Expr, class... Ops>
    class For
    {
    public:

	static size_t layer() { return typename Expr::layer() + 1; }
	
	For(For&& in) = default;
	For& operator=(For&& in) = default;
	
	template <typename... Args>
	For(const std::shared_ptr<IndexClass>& indPtr,
	    std::tuple<to_size_t<Ops>...>&& ext, const Args&... args);

	For(const std::shared_ptr<IndexClass>& indPtr,
	    Expr&& expr, std::tuple<to_size_t<Ops>...>&& ext);

	template <typename... Args>
	For(IndexClass* indPtr,
	    std::tuple<to_size_t<Ops>...>&& ext, const Args&... args);

	For(IndexClass* indPtr,
	    Expr&& expr, std::tuple<to_size_t<Ops>...>&& ext);

	
	inline void operator()(size_t mlast, const std::tuple<to_size_t<Ops>...>& last) const;
	inline void operator()(size_t mlast = 0) const;
	
    private:
	For() = default;
	
	IndexClass* mIndPtr;
	const Expr mExpr;
	const std::tuple<to_size_t<Ops>...> mExt;
    };

    template <size_t N>
    size_t exceptMax(size_t max) { return max; }

    template <>
    size_t exceptMax<1>(size_t max) { return 1; }

    
} // namespace MultiArrayHelper

/* ========================= *
 * ---   TEMPLATE CODE   --- *
 * ========================= */

namespace MultiArrayHelper
{

    template <class IndexClass, class Expr, class... Ops>
    template <typename... Args>
    For<IndexClass,Expr,Ops...>::For(const std::shared_ptr<IndexClass>& indPtr,
				     std::tuple<to_size_t<Ops>...>&& ext,
				     const Args&... args) :
	mIndPtr(indPtr.get()), mExpr(args...), mExt(ext) {}

    template <class IndexClass, class Expr, class... Ops>
    For<IndexClass,Expr,Ops...>::For(const std::shared_ptr<IndexClass>& indPtr,
				     Expr&& expr, std::tuple<to_size_t<Ops>...>&& ext) :
	mIndPtr(indPtr.get()), mExpr(expr), mExt(ext) {}

    template <class IndexClass, class Expr, class... Ops>
    template <typename... Args>
    For<IndexClass,Expr,Ops...>::For(IndexClass* indPtr,
				     std::tuple<to_size_t<Ops>...>&& ext,
				     const Args&... args) :
	mIndPtr(indPtr), mExpr(args...), mExt(ext) {}
    
    template <class IndexClass, class Expr, class... Ops>
    For<IndexClass,Expr,Ops...>::For(IndexClass* indPtr,
				     Expr&& expr, std::tuple<to_size_t<Ops>...>&& ext) :
	mIndPtr(indPtr), mExpr(expr), mExt(ext) {}
    
    template <class IndexClass, class Expr, class... Ops>
    inline void For<IndexClass,Expr,Ops...>::operator()(size_t mlast,
							const std::tuple<to_size_t<Ops>...>& last) const
    {
	static const size_t opNum = sizeof...(Ops);
	auto& ind = *mIndPtr;
	const size_t max = exceptMax<For<IndexClass,Expr,Ops...>::layer()>( ind.max() ); // blocking
	for(ind = 0; ind.pos() != max; ++ind){
	    const size_t mnpos = mlast * max + ind.pos();
	    const std::tuple<to_size_t<Ops>...> npos = std::move( XFPackNum<opNum-1>::mkPos(ind, mExt, last) );
	    mExpr(mnpos, npos);
	}
    }

    template <class IndexClass, class Expr, class... Ops>
    inline void For<IndexClass,Expr,Ops...>::operator()(size_t mlast) const
    {
	static const size_t opNum = sizeof...(Ops);
	std::tuple<to_size_t<Ops>...> last(to_size_t<Ops>(0)...);
	auto& ind = *mIndPtr;
	const size_t max = exceptMax<For<IndexClass,Expr,Ops...>::layer()>( ind.max() ); // blocking
	for(ind = 0; ind.pos() != max; ++ind){
	    const size_t mnpos = mlast * max + ind.pos();
	    const std::tuple<to_size_t<Ops>...> npos = std::move( XFPackNum<opNum-1>::mkPos(ind, mExt, last) );
	    mExpr(mnpos, npos);
	}
    }
    
} // namespace MultiArrayHelper

#endif