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

---

TalkNgbP2Loop.h

Go to the documentation of this file.

#ifndef Impala_Core_Array_Trait_TalkNgbP2Loop_h
#define Impala_Core_Array_Trait_TalkNgbP2Loop_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 TalkNgbP2Loop
{
public:

    typedef Pattern::TagLoop IteratorCategory;

    typedef Pattern::Tag2Phase PhaseCategory;

    typedef typename DstArrayT::StorType DstStorType;
    typedef typename DstArrayT::ArithType DstArithType;
    typedef typename SrcArrayT::StorType SrcStorType;
    typedef typename SrcArrayT::ArithType SrcArithType;

    TalkNgbP2Loop(bool verbose, int ngbWidth, int ngbHeight)
    {
        mNgbWidth = ngbWidth;
        mNgbHeight = ngbHeight;
        mFactor = ngbWidth * ngbHeight;
        mVerbose = verbose;
    }

    int
    Width()
    {
        return mNgbWidth;
    }

    int
    Height()
    {
        return mNgbHeight;
    }

    void
    Init(int x, int y, const SrcArithType& value)
    {
        if (mVerbose)
            std::cout << "  TalkNgbP2Loop::init(" << x << "," << y << ","
                      << value << ") " << std::endl;
        mResult = Element::E1Cast(0, DstArithType());
    }

    void
    Init2(int x, int y, const SrcArithType& value)
    {
        if (mVerbose)
            std::cout << "  TalkNgbP2Loop::init2(" << x << "," << y << ","
                      << value << ") " << std::endl;
    }

    void
    NextEl(int x, int y, const SrcArithType& value)
    {
        if (mVerbose)
            std::cout << "      TalkNgbP2Loop::next(" << x << "," << y << ","
                      << value << ") " << std::endl;
        mResult += value;
    }

    void
    NextEl2(int x, int y, const SrcArithType& value)
    {
        if (mVerbose)
            std::cout << "      TalkNgbP2Loop::next2(" << x << "," << y << ","
                      << value << ") " << std::endl;
        mResult += value;
    }

    DstArithType
    Result() const
    {
        return Element::E1Cast(mResult, DstArithType()) /
               Element::E1Cast(mFactor, DstArithType());
    }

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

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

#endif

---