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

---

PatGenConv2d.h

Go to the documentation of this file.

#ifndef Impala_Core_Array_Pattern_PatGenConv2d_h
#define Impala_Core_Array_Pattern_PatGenConv2d_h

#include "Core/Array/CheckBorderSize.h"
#include "Core/Array/Pattern/PatSetBorder.h"
#include "Core/Array/Pattern/FuncGenConv2d.h"

#ifdef PX_HORUS_USED
#include "Core/Array/Pattern/PxArrayFunc.h"
#include "Core/Array/Pattern/PxStateTrans.h"
#endif

namespace Impala
{
namespace Core
{
namespace Array
{
namespace Pattern
{


template<class DstArrayT, class SrcArrayT, class KerArrayT,
         class PixOpT, class RedOpT>
inline void
PatGenConv2d(DstArrayT*& dst, SrcArrayT* src, KerArrayT* ker,
             PixOpT& pixOp, RedOpT& redOp, BorderType borderType = BORDERMIRROR)
{
    int numX = ArrayCW(ker) / 2;
    int numY = ArrayCH(ker) / 2;
    SrcArrayT* checkedSrc = CheckBorderSize(src, numX, numY);

    if ((dst == 0) || (dst == checkedSrc))
        dst = ArrayClone<DstArrayT>(checkedSrc);

#ifdef PX_HORUS_USED
    if (!PxRunParallel()) {                         // run sequential
#endif
        /*
        if (borderType == BORDERCONSTANT)
            PatSetBorder(checkedSrc, numX, numY, borderType,
                         redOp.NeutralElement(), redOp.NeutralElement(),
                         redOp.NeutralElement(), redOp.NeutralElement(),
                         redOp.NeutralElement());
        else
        */
            PatSetBorder(checkedSrc, numX, numY);
        FuncGenConv2dDispatch(dst, checkedSrc, ker, pixOp, redOp);

#ifdef PX_HORUS_USED
    } else {                                        // run parallel
        PxArrayPreStateTrans(checkedSrc, PAR_PART, STRONG);
        PxArrayPreStateTrans(dst, PAR_PART, WEAK);
        PatSetBorder(PxArrayPD(checkedSrc), numX, numY);
        FuncGenConv2dDispatch(PxArrayPD(dst),
                        PxArrayPD(checkedSrc), ker, pixOp, redOp);
        if (PxArrayPD(checkedSrc) != PxArrayPD(src))
            PxArrayDeletePD(checkedSrc);
        PxArrayPostStateTrans(dst);

        if (!PxRunLazyParallel()) {
            PxArrayForceNonDistributed(src);
            PxArrayForceNonDistributed(dst);
        }
    }
#endif

    if (checkedSrc != src)
        delete checkedSrc;
}

} // namespace Pattern
} // namespace Array
} // namespace Core
} // namespace Impala

#endif

---