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

---

TalkNgbExtraP2Loop.h

Go to the documentation of this file.

#ifndef Impala_Core_Array_Trait_TalkNgbExtraP2Loop_h
#define Impala_Core_Array_Trait_TalkNgbExtraP2Loop_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 TalkNgbExtraP2Loop
{
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;
    typedef typename ExtraArrayT::StorType ExtraStorType;
    typedef typename ExtraArrayT::ArithType ExtraArithType;

    TalkNgbExtraP2Loop(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 x, int y, const SrcArithType& v1, const ExtraArithType& v2)
    {
        if (mVerbose)
            std::cout << "  TalkNgbExtraP2Loop::init(" << x << "," << y << ","
                      << v1 << "," << v2 << ") " << std::endl;
        mResult = Element::E1Cast(0, DstArithType());
    }

    void
    Init2(int x, int y, const SrcArithType& v1, const ExtraArithType& v2)
    {
        if (mVerbose)
            std::cout << "  TalkNgbExtraP2Loop::init2(" << x << "," << y << ","
                      << v1 << "," << v2 << ") " << std::endl;
    }

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

    void
    NextEl2(int x, int y, const SrcArithType& v1, const ExtraArithType& v2)
    {
        if (mVerbose)
            std::cout << "      TalkNgbExtraP2Loop::next2(" << x << "," << y
                      << "," << v1 << "," << v2 << ") " << std::endl;
        DstArithType tmp(v1, v2, 0);
        mResult += tmp;
    }

    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

---