KMeansAssignImgLvl1.h

// KMeansAssignImgLvl1.h:
// ----------------------
//

#ifndef __IPSDKIPLCLASSIFICATION_KMEANSASSIGNIMGLVL1_H__
#define __IPSDKIPLCLASSIFICATION_KMEANSASSIGNIMGLVL1_H__

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

// Attributes headers
#include <IPSDKIPL/IPSDKIPLAttributes/Attribute/DataItem/InClustersCenters.h>
#include <IPSDKIPL/IPSDKIPLAttributes/Attribute/DataItem/OutKMeansAssignmentInfo.h>
#include <IPSDKIPL/IPSDKIPLAttributes/Attribute/DataItem/OutOptKMeansAssignmentInfo.h>
#include <IPSDKIPL/IPSDKIPLAttributes/Attribute/Image/Input/InHomogeneousImg.h>
#include <IPSDKIPL/IPSDKIPLAttributes/Attribute/Image/Output/OutClassImg.h>
#include <IPSDKIPL/IPSDKIPLAttributes/DataItem/Doubles.h>
#include <IPSDKIPL/IPSDKIPLAttributes/Attribute/Image/Input/Optional/InOptSingleGreyMaskImg.h>

namespace ipsdk {
namespace imaproc {
namespace classif {


// declare processor class
IPSDK_DECLARE_LVL1_PROCESSOR(ASYNC1, IPSDKIPLClassification, KMeansAssignImg,
                             ipsdk::imaproc::matchSize(eMatchImageSizeType::eMIST_3d, _pInHomogeneousImg, _pOutClassImg) &&
                             ipsdk::imaproc::classif::clustersCentersCollMatchImgSize(_pInClustersCenters, _pInHomogeneousImg) &&
                             ipsdk::imaproc::customImageProperty(_pOutClassImg, allowedClassImgBuffersFromNbClusters(_pInClustersCenters)) &&
                             ipsdk::processor::If(ipsdk::processor::isSet(_pInOptSingleGreyMaskImg),
                                 ipsdk::imaproc::matchSize(eMatchImageSizeType::eMIST_3d, _pInHomogeneousImg, _pInOptSingleGreyMaskImg),
                                 ipsdk::processor::none()),
                             (((ipsdk)(imaproc)(attr))(InHomogeneousImg))
                             (((ipsdk)(imaproc)(attr))(InClustersCenters))
                             (((ipsdk)(imaproc)(attr))(InOptSingleGreyMaskImg))
                             (((ipsdk)(imaproc)(attr))(OutOptKMeansAssignmentInfo))
                             (((ipsdk)(imaproc)(attr))(OutClassImg)(customOutput)(outputKMeans(_pInHomogeneousImg, _pInClustersCenters))))

// methods
public:

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

    void updateKMeansAssignmentInfo();

// attributes
protected:
    attr::DoublesPtr                _pFlattenedClustersCenters;

    // declaration of dispatch results collection
    IPSDK_DECLARE_STATIC_RESULT_COLL(attr::OutKMeansAssignmentInfo, OutKMeansAssignmentInfoColl);
};


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

#endif // __IPSDKIPLCLASSIFICATION_KMEANSASSIGNIMGLVL1_H__