|| || || || || || || ||

---

TalkNgbExtra2PnLoop.h

Go to the documentation of this file.

#ifndef Impala_Core_Array_Trait_TalkNgbExtra2PnLoop_h
#define Impala_Core_Array_Trait_TalkNgbExtra2PnLoop_h

#include "Core/Array/Pattern/Categories.h"
#include "Core/Array/Pattern/Cnum.h"
#include "Core/Array/Element/E1Cast.h"

namespace Impala
{
namespace Core
{
namespace Array
{
namespace Trait
{


template<class DstArrayT, class SrcArrayT, class ExtraArrayT, class Extra2ArrayT>
class TalkNgbExtra2PnLoop
{
public:

    typedef Pattern::TagLoop IteratorCategory;

    typedef Pattern::TagNPhase PhaseCategory;

    typedef typename DstArrayT::StorType DstStorType;
    typedef typename DstArrayT::ArithType DstArithType;
    typedef typename SrcArrayT::StorType SrcStorType;
    typedef typename SrcArrayT::ArithType SrcArithType;
    typedef typename ExtraArrayT::StorType ExtraStorType;
    typedef typename ExtraArrayT::ArithType ExtraArithType;
    typedef typename Extra2ArrayT::StorType Extra2StorType;
    typedef typename Extra2ArrayT::ArithType Extra2ArithType;


    TalkNgbExtra2PnLoop(bool verbose, int ngbWidth, int ngbHeight)
    {
        mNgbWidth = ngbWidth;
        mNgbHeight = ngbHeight;
        mFactor = Element::E1Cast(ngbWidth * ngbHeight, DstArithType());
        mVerbose = verbose;
    }

    int
    Width()
    {
        return mNgbWidth;
    }

    int
    Height()
    {
        return mNgbHeight;
    }

    void
    Init(int phase, int x, int y, SrcArithType v1, ExtraArithType v2,
         Extra2ArithType v3)
    {
        if (mVerbose)
            std::cout << "  TalkNgbExtra2PnLoop::init(" << phase << ","
                      << x << "," << y << "," << v1 << "," << v2 << "," << v3
                      << ") " << std::endl;
        if (phase == 1)
            mResult = Element::E1Cast(0, DstArithType());
    }

    void
    NextEl(int x, int y, SrcArithType v1, ExtraArithType v2, Extra2ArithType v3)
    {
        if (mVerbose)
            std::cout << "      TalkNgbExtra2PnLoop::next(" << x << "," << y
                      << "," << v1 << "," << v2 << "," << v3 << ") " << std::endl;
        DstArithType tmp(v1, v2, v3);
        mResult += tmp;
    }

    void
    Done(int phase)
    {
    }

    bool
    HasNextPhase(int lastPhase)
    {
        if (mVerbose)
            std::cout << "  TalkNgbExtra2PnLoop::hasNext(" << lastPhase << ") "
                      << std::endl;
        return (lastPhase < 4);
    }

    DstArithType
    Result() const
    {
        return mResult / mFactor;
    }

private:
    int          mNgbWidth;
    int          mNgbHeight;
    DstArithType mResult;
    DstArithType mFactor;
    bool         mVerbose;
};

} // namespace Trait
} // namespace Array
} // namespace Core
} // namespace Impala

#endif

---