cnorxz/src/include/xpr/for.h

#ifndef __cxz_for_h__
#define __cxz_for_h__

#include "base/base.h"
#include "xpr_base.h"
#include "func.h"

namespace CNORXZ
{

    template <SizeT L, class Xpr, class F = NoF>
    class For : public XprInterface<For<L,Xpr,F>>
    {
    public:
	DEFAULT_MEMBERS(For);

	constexpr For(SizeT size, const IndexId<L>& id, const Xpr& xpr, F&& f);

	template <class PosT>
	inline decltype(auto) operator()(const PosT& last) const;
	
	inline decltype(auto) operator()() const;

	template <SizeT I>
	inline decltype(auto) rootSteps(const IndexId<I>& id) const;

    private:
	SizeT mSize = 0;
	IndexId<L> mId;
	Xpr mXpr;
	typedef decltype(mXpr.rootSteps(mId)) XPosT;
	XPosT mExt;
	F mF;
    };

    template <SizeT L, class Xpr, class F>
    constexpr decltype(auto) mkFor(SizeT size, const IndexId<L>& id, const Xpr& xpr, F&& f);

    template <SizeT L, class Xpr>
    constexpr decltype(auto) mkFor(SizeT size, const IndexId<L>& id, const Xpr& xpr);

    // unrolled loop:
    template <SizeT N, SizeT L, class Xpr, class F = NoF>
    class SFor : public XprInterface<SFor<N,L,Xpr,F>>
    {
    public:
	DEFAULT_MEMBERS(SFor);

	constexpr SFor(const IndexId<L>& id, const Xpr& xpr, F&& f);

	template <class PosT>
	constexpr decltype(auto) operator()(const PosT& last) const;
	
	constexpr decltype(auto) operator()() const;

	template <SizeT I>
	constexpr decltype(auto) rootSteps(const IndexId<I>& id) const;
	
    private:

	template <SizeT I, class PosT>
	constexpr decltype(auto) exec(const PosT& last) const;

	template <SizeT I>
	constexpr decltype(auto) exec() const;

	template <SizeT I, class PosT>
	inline void exec2(const PosT& last) const;

	template <SizeT I>
	inline void exec2() const;

	IndexId<L> mId;
	Xpr mXpr;
	typedef decltype(mXpr.rootSteps(mId)) XPosT;
	XPosT mExt;
	F mF;
	
    };

    template <SizeT N, SizeT L, class Xpr, class F>
    constexpr decltype(auto) mkSFor(const IndexId<L>& id, const Xpr& xpr, F&& f);

    template <SizeT N, SizeT L, class Xpr>
    constexpr decltype(auto) mkSFor(const IndexId<L>& id, const Xpr& xpr);

    // partial for:
    template <SizeT L1, SizeT L2, class Xpr, class F = NoF>
    class PFor : public XprInterface<PFor<L1,L2,Xpr,F>>
    {
    public:
	DEFAULT_MEMBERS(PFor);

	constexpr PFor(SizeT size, const IndexId<L1>& id1, const IndexId<L2>& id2,
		       const SizeT* map, const Xpr& xpr, F&& f);

	template <class PosT>
	inline decltype(auto) operator()(const PosT& last) const;
	
	inline decltype(auto) operator()() const;

	template <SizeT I>
	inline decltype(auto) rootSteps(const IndexId<I>& id) const;

    private:
	SizeT mSize = 0;
	IndexId<L1> mId1;
	IndexId<L2> mId2;
	Xpr mXpr;
	typedef decltype(mXpr.rootSteps(mId1)) XPosT1;
	typedef decltype(mXpr.rootSteps(mId2)) XPosT2;
	XPosT1 mExt1;
	XPosT2 mExt2;
	FPos mPart;
	F mF;
    };

    template <SizeT L, class Xpr, class F>
    constexpr decltype(auto) mkFor(SizeT size, const IndexId<L>& id, const Xpr& xpr, F&& f);

    template <SizeT L, class Xpr>
    constexpr decltype(auto) mkFor(SizeT size, const IndexId<L>& id, const Xpr& xpr);

    // multi-threading
    template <SizeT L, class Xpr, class F = NoF>
    class TFor : public XprInterface<TFor<L,Xpr,F>>
    {
    public:
	DEFAULT_MEMBERS(TFor);

	constexpr TFor(SizeT size, const IndexId<L>& id, const Xpr& xpr, F&& f);

	template <class PosT>
	inline decltype(auto) operator()(const PosT& last) const;
	
	inline decltype(auto) operator()() const;

	template <SizeT I>
	inline decltype(auto) rootSteps(const IndexId<I>& id) const;
	
    private:
	SizeT mSize = 0;
	IndexId<L> mId;
	Xpr mXpr;
	typedef decltype(mXpr.rootSteps(mId)) XPosT;
	XPosT mExt;
	F mF;
    };

    // Extension For (Vectorization)
    template <SizeT N, SizeT L, class Xpr>
    class EFor : public XprInterface<EFor<N,L,Xpr>>
    {
    public:
	DEFAULT_MEMBERS(EFor);

	constexpr EFor(const IndexId<L>& id, const Xpr& xpr);

	template <class PosT>
	constexpr decltype(auto) operator()(const PosT& last) const;

	constexpr decltype(auto) operator()() const;

	template <SizeT I>
	constexpr decltype(auto) rootSteps(const IndexId<I>& id) const;

    private:
	IndexId<L> mId;
	Xpr mXpr;
	typedef decltype(mXpr.rootSteps(mId)) XPosT;
	XPosT mExt;
    };
    
}

#endif