PatchBasedBilateral3dImgLvl3.h

// PatchBasedBilateral3dImgLvl3.h:
// -------------------------
//

#ifndef __IPSDKIPLFILTERING_PATCHBASEDBILATERAL3DIMGLVL3_H__
#define __IPSDKIPLFILTERING_PATCHBASEDBILATERAL3DIMGLVL3_H__

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

// Attributes headers
#include <IPSDKIPL/IPSDKIPLAttributes/Attribute/Strip/ZStrip3d/OutZStrip3d.h>
#include <IPSDKIPL/IPSDKIPLAttributes/Attribute/Strip/ZStrip3d/OutZStrip3d1.h>
#include <IPSDKIPL/IPSDKIPLAttributes/Attribute/Strip/ZStrip3d/OutZStrip3d2.h>
#include <IPSDKIPL/IPSDKIPLAttributes/Attribute/Strip/ZStrip3dWithKernel/InKnlZStrip3d.h>
#include <IPSDKIPL/IPSDKIPLAttributes/Attribute/DataItem/InPatchBasedBilateralParams.h>

#include <boost/tuple/tuple.hpp>

namespace ipsdk {
namespace imaproc {
namespace filter {


// declare processing algorithm
IPSDK_DECLARE_PROCESSING_ALGORITHM(IPSDKIPLFiltering, PatchBasedBilateral3dImg,
                                   ipsdk::imaproc::matchSize(_pInKnlZStrip3d, _pOutZStrip3d),
                                   (((ipsdk)(imaproc)(attr))(InKnlZStrip3d))
                                   (((ipsdk)(imaproc)(attr))(InPatchBasedBilateralParams))

                                   (((ipsdk)(imaproc)(attr))(OutZStrip3d1))
                                   (((ipsdk)(imaproc)(attr))(OutZStrip3d2))
                                   (((ipsdk)(imaproc)(attr))(OutZStrip3d)))

    // implementation of image processing data identifier retrieval method
    // for function selector usage
    IPSDK_DECLARE_ALGORITHM_FUNCTION_SELECTOR(((STRIP)(InKnlZStrip3d))
                                              ((STRIP)(OutZStrip3d)));
    
    // declaration of parsed data during processing
    IPSDK_DECLARE_ALGORITHM_NOPARSED_DATA()

// methods
public:

protected:
    void getNoParseInfo(ipUInt64& nbDataByElement,
                        ipUInt64& nbMaxInitData,
                        ipUInt64& nbMaxFinalData) const;

    template <
        typename BufInType,
        typename BufOutType,
        typename ComputationType,
        ipsdk::eInstructionSet::domain ISFma,
        ipsdk::eInstructionSet::domain IS
    >
    void genericProcessFunctionForCurProcessingPhasis(
        std::deque<boost::shared_ptr<YStrip2dData<BufInType> > >& yStrip2dDataColl,
        ipInt64 outputBufferOffset,
        ipInt32 kernelRadius,
        ipInt32 patchHalfSz,
        ComputationType patchSimilaritySigma,
        ComputationType spaceSigma,
        ComputationType twiceNoiseSigmaSq);

    template <
        typename BufInType,
        typename ComputationType,
        ipsdk::eInstructionSet::domain IS>
    void computeMeanOfDiffSquaresForEltInKernel(
        ipInt64 otherIx,
        ipUInt64 nbTotData,
        ipInt32 kernelRadius,
        ipInt32 patchHalfSz,
        std::deque<boost::shared_ptr<YStrip2dData<BufInType> > > & yStrip2dDataColl);

    template <typename BufInType,
              typename BufOutType,
              typename ComputationType,
              ipsdk::eInstructionSet::domain ISFma,
              ipsdk::eInstructionSet::domain IS>
    void processEltInKernelForCurrentRow(
        ipInt64 otherIx,
        ipUInt64 nbTotData,
        ipUInt64 nbRealData,
        ipInt32 kernelRadius,
        ipInt32 patchHalfSz,
        ComputationType patchSimilaritySigma,
        ComputationType spaceSigma,
        ComputationType twiceNoiseSigmaSq,
        std::deque<boost::shared_ptr<YStrip2dData<BufInType> > >& yStrip2dDataColl);

    template <
        typename BufInType,
        typename BufOutType,
        typename ComputationType,
        eDataType::domain eComputationType,
        ipsdk::eInstructionSet::domain IS>
    void prepareBuffers(
        Parser3dInfo<BufInType>& parserInfo,
        ipInt32 kernelRadius,
        ipInt32 patchHalfSz,
        std::deque<boost::shared_ptr<YStrip2dData<BufInType> > > & yStrip2dDataColl);

    template <
        typename BufInType,
        typename BufOutType,
        typename ComputationType,
        ipsdk::eInstructionSet::domain IS>
    void updateBuffers(
        Parser3dInfo<BufInType>& parserInfo,
        std::deque<boost::shared_ptr<YStrip2dData<BufInType> > > & yStrip2dDataColl);

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

    template <typename BufInType>
    void updateToInitData(
        std::deque<boost::shared_ptr<YStrip2dData<BufInType> > >& yStrip2dDataColl);

    template <typename BufInType>
    void updateToMainData(
        std::deque<boost::shared_ptr<YStrip2dData<BufInType> > >& yStrip2dDataColl);

    template <typename BufInType>
    void updateToFinalData(
        std::deque<boost::shared_ptr<YStrip2dData<BufInType> > >& yStrip2dDataColl);

// attributes
protected:
    std::vector<boost::tuple<ipInt32, ipInt32, ipInt32> >       _ptsInKernel;

    HybridBufferPtr                             _pInitialSquareDiffs;
    HybridBufferPtr                             _pMainSquareDiffs;
    HybridBufferPtr                             _pFinalSquareDiffs;
    HybridBufferPtr                             _pCurSquareDiffs;

    FragHybridBufferPtr                         _pInitialAvgXSquareDiffs;
    FragHybridBufferPtr                         _pMainAvgXSquareDiffs;
    FragHybridBufferPtr                         _pFinalAvgXSquareDiffs;
    FragHybridBufferPtr                         _pCurAvgXSquareDiffs;

    FragHybridBufferPtr                         _pInitialAvgXYSquareDiffs;
    FragHybridBufferPtr                         _pMainAvgXYSquareDiffs;
    FragHybridBufferPtr                         _pFinalAvgXYSquareDiffs;
    FragHybridBufferPtr                         _pCurAvgXYSquareDiffs;

    HybridBufferPtr                             _pInitialAvgXYZSquareDiffs;
    HybridBufferPtr                             _pMainAvgXYZSquareDiffs;
    HybridBufferPtr                             _pFinalAvgXYZSquareDiffs;
    HybridBufferPtr                             _pCurAvgXYZSquareDiffs;

    processor::eProcessingPhasis                _curProcessingPhasis;
    ipUInt64                                    _curRowIdx;
    ipUInt64                                    _curPlanIdx;

    ipUInt64                                    _nbInitialData;
    ipUInt64                                    _nbMainData;
    ipUInt64                                    _nbFinalData;
    ipUInt64                                    _nbRealFinalData;
    ipUInt64                                    _nbCurData;
    ipUInt64                                    _nbRealCurData;
};


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

#endif // __IPSDKIPLFILTERING_PATCHBASEDBILATERAL3DIMGLVL3_H__