KMeansPPClusterInitLvl1.h

// KMeansPPClusterInitLvl1.h:
// -------------------------------
//

#ifndef __IPSDKIPLCLASSIFICATION_KMEANSPPCLUSTERINITLVL1_H__
#define __IPSDKIPLCLASSIFICATION_KMEANSPPCLUSTERINITLVL1_H__

#include <IPSDKIPL/IPSDKIPLClassification/IPSDKIPLClassificationExports.h>
#include <IPSDKBaseProcessing/Processor/ProcessorHdrMacros.h>
#include <IPSDKBaseProcessing/OutputInitializer/OutputInitializerOperators.h>
#include <IPSDKImageProcessing/OutputInitializer/Image/ImageInitializerOperators.h>

// Attributes headers
#include <IPSDKIPL/IPSDKIPLAttributes/Attribute/DataItem/OutClustersCenters.h>
#include <IPSDKIPL/IPSDKIPLAttributes/Attribute/Image/Input/InHomogeneousImg.h>
#include <IPSDKIPL/IPSDKIPLAttributes/Attribute/Value/InNbClusters.h>
#include <IPSDKIPL/IPSDKIPLAttributes/DataItem/Doubles.h>
#include <IPSDKIPL/IPSDKIPLAttributes/Attribute/Enumerate/InRandomAlgoType.h>
#include <IPSDKIPL/IPSDKIPLAttributes/Attribute/Image/InOut/InOutOptWk1RealImg.h>
#include <IPSDKIPL/IPSDKIPLAttributes/Attribute/Image/Output/Working/Optional/OutOptWk1RealImg.h>
#include <IPSDKIPL/IPSDKIPLAttributes/Attribute/DataItem/OutMaxValIdx.h>
#include <IPSDKIPL/IPSDKIPLAttributes/Attribute/DataItem/OutKMeansPPClusterInitInfo.h>
#include <IPSDKIPL/IPSDKIPLAttributes/Attribute/Image/Input/Optional/InOptSingleGreyMaskImg.h>

namespace ipsdk {
namespace imaproc {
namespace classif {


// declare processor class
IPSDK_DECLARE_LVL1_PROCESSOR(REP_ASYNC1, IPSDKIPLClassification, KMeansPPClusterInit,
                            ipsdk::processor::ifIsSet(_pInOutOptWk1RealImg,
                                ipsdk::imaproc::matchSize(eMatchImageSizeType::eMIST_3d, _pInHomogeneousImg, _pInOutOptWk1RealImg) &&
                                ipsdk::processor::ifIsSet(_pOutOptWk1RealImg,
                                    ipsdk::imaproc::matchSize(eMatchImageSizeType::eMIST_3d, _pInHomogeneousImg, _pOutOptWk1RealImg) &&
                                    ipsdk::imaproc::matchSizeAndType(_pInOutOptWk1RealImg, _pOutOptWk1RealImg))) &&
                             ipsdk::processor::ifIsSet(_pInOptSingleGreyMaskImg,
                                ipsdk::imaproc::matchSize(eMatchImageSizeType::eMIST_3d, _pInHomogeneousImg, _pInOptSingleGreyMaskImg)),
                             (((ipsdk)(imaproc)(attr))(InHomogeneousImg))
                             (((ipsdk)(imaproc)(attr))(InOptSingleGreyMaskImg))
                             (((ipsdk)(imaproc)(attr))(InNbClusters))
                             (((ipsdk)(imaproc)(attr))(InRandomAlgoType))
                             (((ipsdk)(imaproc)(attr))(InOutOptWk1RealImg))
                             (((ipsdk)(imaproc)(attr))(OutOptWk1RealImg))
                             (((ipsdk)(imaproc)(attr))(OutClustersCenters)(ipsdk::processor::allocate)))

// methods
public:

protected:

    template <typename T>
    void addClusterCenter(const ipUInt64 centerIdx, const ipUInt64 zIdx);

    void findRandomClusterCenter(ipUInt64& maxIdx, ipUInt64& zIdx);

    void mergeLvl2Results(ipUInt64& maxIdx, ipUInt64& zIdx);

    void clearIntermediateData();

// attributes
protected:
    // declaration of dispatch results collection
    IPSDK_DECLARE_STATIC_RESULT_COLL(attr::OutMaxValIdx, OutMaxValIdxColl);
    IPSDK_DECLARE_STATIC_RESULT_COLL(attr::OutKMeansPPClusterInitInfo, OutKMeansPPClusterInitInfoColl);

    image::ImageConstPtr                _pReinterpretedInImg;
    image::ImagePtr                     _pDistImg;
    image::ImagePtr                     _pCumulDistImg;
    attr::DoublesPtr                    _pFlattenedClustersCenters;
    attr::DoublesPtr                    _pCumDistColl;

    // Storage of _pInOptSingleGreyMaskImg
    image::ImageConstPtr                _pMaskImg;

};


} // end of namespace classif
} // end of namespace imaproc
} // end of namespace ipsdk

#endif // __IPSDKIPLCLASSIFICATION_KMEANSPPCLUSTERINITLVL1_H__