PatchBasedBilateral2dImgLvl3.h

// PatchBasedBilateral2dImgLvl3.h:
// -------------------------
//

#ifndef __IPSDKIPLFILTERING_PATCHBASEDBILATERAL2DIMGLVL3_H__
#define __IPSDKIPLFILTERING_PATCHBASEDBILATERAL2DIMGLVL3_H__

#include <IPSDKIPL/IPSDKIPLFiltering/IPSDKIPLFilteringExports.h>
#include <IPSDKImageProcessing/Algorithm/ImageProcessingAlgorithmHdrMacros.h>

// Attributes headers
#include <IPSDKIPL/IPSDKIPLAttributes/Attribute/Strip/YStrip2d/OutYStrip2d.h>
#include <IPSDKIPL/IPSDKIPLAttributes/Attribute/Strip/YStrip2dWithKernel/InKnlYStrip2d.h>
#include <IPSDKIPL/IPSDKIPLAttributes/Attribute/DataItem/InPatchBasedBilateralParams.h>

namespace ipsdk {
namespace imaproc {
namespace filter {


// declare processing algorithm
IPSDK_DECLARE_PROCESSING_ALGORITHM(IPSDKIPLFiltering, PatchBasedBilateral2dImg,
                                   ipsdk::imaproc::matchSize(_pInKnlYStrip2d, _pOutYStrip2d),
                                   (((ipsdk)(imaproc)(attr))(InKnlYStrip2d))
                                   (((ipsdk)(imaproc)(attr))(InPatchBasedBilateralParams))
                                   (((ipsdk)(imaproc)(attr))(OutYStrip2d)))

    // implementation of image processing data identifier retrieval method
    // for function selector usage
    IPSDK_DECLARE_ALGORITHM_FUNCTION_SELECTOR(((STRIP)(InKnlYStrip2d))
                                              ((STRIP)(OutYStrip2d)));
    
    // declaration of parsed data during processing
    IPSDK_DECLARE_ALGORITHM_PARSED_DATA(((LINESTRIP)(InKnlYStrip2d))
                                        ((LINESTRIP)(OutYStrip2d)));

// methods
public:

protected:
    ipsdk::core::ExecResult preProcess();

    template <typename BufInType, typename BufOutType,
        ipsdk::eInstructionSet::domain ISFma, ipsdk::eInstructionSet::domain IS>
    ipsdk::BoolResult genericProcessFunction(const ipsdk::processor::ProcessingInfo& info);

    template <typename BufInType, typename BufOutType,
              typename ComputationType,
              eDataType::domain eComputationType,
              ipsdk::eInstructionSet::domain ISFma, ipsdk::eInstructionSet::domain IS>
    ipsdk::BoolResult       computationTGenericProcessFunction(const ipsdk::processor::ProcessingInfo& info);

    
    template <typename ComputationType,
    eDataType::domain eComputationType>
    void
    retrieveIntermediaryBuffers(
        const ipsdk::processor::ProcessingInfo& info,
        const ipInt32 kernelRadius,
        const ipInt32 patchSz,
        HybridBufferPtr& pCurSquareDiffs,
        FragHybridBufferPtr*& pCurAvgXSquareDiffs,
        HybridBufferPtr*& pCurSimilarities,
        FragHybridBufferPtr*& pCurWeights, 
        HybridBufferPtr& pCurWeightSums,
        HybridBufferPtr& pCurWeightMaxima,
        HybridBufferPtr& pCurWeightedIntensitiesSums);

    template <typename BufInType, typename ComputationType, ipsdk::eInstructionSet::domain IS>
    void computeSimilaritiesFromPrevRow(
        const ipUInt32 nbElts,
        const ipUInt32 patchSz,
        const BufInType* pCurInBuf,
        const BufInType* pOtherInBuf,
        ComputationType* pCurSquareDiffsBuf,
        FragHybridBufferPtr pCurAvgXSquareDiffs,
        ComputationType* pCurSimilarities);

    template <typename ComputationType, ipsdk::eInstructionSet::domain ISFma, ipsdk::eInstructionSet::domain IS>
    void
    computeSimilarityWeightsForCurKnlElt(
        const ipUInt64 otherIx,
        const ipUInt64 nbElts,
        const ipReal64 sigma2,
        const ipReal64 twicePatchSimilaritySigmaSq,
        const ComputationType* pSimilaritiesBuf,
        ComputationType* pWeightsBuf);

    template <eInstructionSet::domain ISFma, eInstructionSet::domain IS, typename BufInType, typename ComputationType>
    void
    computeSimilarityWeights(
        HybridBufferPtr* pCurSimilarities,
        ComputationType* pCurSquareDiffsBuf,
        FragHybridBufferPtr* pCurAvgXSquareDiffs,
        FragHybridBufferPtr* pCurSimilarityWeights,
        const ipsdk::processor::ProcessingInfo& info);

// attributes
protected:
    HybridBufferPtr                                             _pInitialSquareDiffs;
    HybridBufferPtr                                             _pMainSquareDiffs;
    HybridBufferPtr                                             _pFinalSquareDiffs;

    boost::scoped_array<FragHybridBufferPtr>                    _pInitialAvgXSquareDiffs;
    boost::scoped_array<FragHybridBufferPtr>                    _pMainAvgXSquareDiffs;
    boost::scoped_array<FragHybridBufferPtr>                    _pFinalAvgXSquareDiffs;

    boost::scoped_array<HybridBufferPtr>                        _pInitialSimilarities;
    boost::scoped_array<HybridBufferPtr>                        _pMainSimilarities;
    boost::scoped_array<HybridBufferPtr>                        _pFinalSimilarities;

    boost::scoped_array<FragHybridBufferPtr>                    _pInitialSimilarityWeights;
    boost::scoped_array<FragHybridBufferPtr>                    _pMainSimilarityWeights;
    boost::scoped_array<FragHybridBufferPtr>                    _pFinalSimilarityWeights;

    std::vector<std::pair<ipInt32, ipInt32> >                   _ptsInKernel;
    ipUInt64                                                    _nbPointsInKernel;
    ipUInt64                                                    _nbHalfPointsInKernel;

    HybridBufferPtr                                             _pInitialWeightSums;
    HybridBufferPtr                                             _pInitialWeightMaxima;
    HybridBufferPtr                                             _pInitialWeightedIntensitiesSums;

    HybridBufferPtr                                             _pMainWeightSums;
    HybridBufferPtr                                             _pMainWeightMaxima;
    HybridBufferPtr                                             _pMainWeightedIntensitiesSums;

    HybridBufferPtr                                             _pFinalWeightSums;
    HybridBufferPtr                                             _pFinalWeightMaxima;
    HybridBufferPtr                                             _pFinalWeightedIntensitiesSums;
};


} // end of namespace filter
} // end of namespace imaproc
} // end of namespace ipsdk

#endif // __IPSDKIPLFILTERING_PATCHBASEDBILATERAL2DIMGLVL3_H__