PixelClassificationRFImgLvl1.h

// PixelClassificationRFImgLvl1.h:
// -------------------------------
//

#ifndef __IPSDKIPLMACHINELEARNING_PIXELCLASSIFICATIONRFIMGLVL1_H__
#define __IPSDKIPLMACHINELEARNING_PIXELCLASSIFICATIONRFIMGLVL1_H__

#include <IPSDKIPL/IPSDKIPLMachineLearning/IPSDKIPLMachineLearningExports.h>
#include <IPSDKBaseProcessing/Processor/ProcessorHdrMacros.h>
#include <IPSDKIPL/IPSDKIPLMachineLearning/OutputInitializer/RFInitializerUtils.h>
#include <IPSDKImageProcessing/OutputInitializer/Image/ImageInitializerOperators.h>

// Attributes headers
#include <IPSDKIPL/IPSDKIPLAttributes/Attribute/DataItem/InRandomForestModel.h>
#include <IPSDKIPL/IPSDKIPLAttributes/Attribute/Image/Input/InImg.h>
#include <IPSDKIPL/IPSDKIPLAttributes/Attribute/Image/Output/OutBinLabImg.h>
#include <IPSDKIPL/IPSDKIPLAttributes/Attribute/Image/Output/Working/Optional/OutOptWk1Img.h>
#include <IPSDKIPL/IPSDKIPLAttributes/Attribute/Image/Output/Working/Optional/OutOptWk2Img.h>
#include <IPSDKIPL/IPSDKIPLAttributes/Attribute/Image/Output/Working/Optional/OutOptWk3Img.h>
#include <IPSDKIPL/IPSDKIPLAttributes/Attribute/Image/Output/Optional/OutOptRealImg.h>

#include <IPSDKIPL/IPSDKIPLAttributes/Attribute/Value/InOptMemoryRatio.h>
#include <IPSDKIPL/IPSDKIPLAttributes/Attribute/Value/InOptProbabilities.h>

namespace ipsdk {
namespace imaproc {
namespace ml {


// declare processor class
IPSDK_DECLARE_LVL1_PROCESSOR(REP_ASYNC1, IPSDKIPLMachineLearning, PixelClassificationRFImg,
                             ipsdk::imaproc::matchSize(eMatchImageSizeType::eMIST_3d,_pInImg, _pOutBinLabImg),
                             (((ipsdk)(imaproc)(attr))(InImg))
                             (((ipsdk)(imaproc)(attr))(InRandomForestModel))
                             (((ipsdk)(imaproc)(attr))(InOptMemoryRatio))
                             (((ipsdk)(imaproc)(attr))(InOptProbabilities))
                             (((ipsdk)(imaproc)(attr))(OutOptWk1Img))
                             (((ipsdk)(imaproc)(attr))(OutOptWk2Img))
                             (((ipsdk)(imaproc)(attr))(OutOptWk3Img))
                             (((ipsdk)(imaproc)(attr))(OutBinLabImg)(customOutput)(outputBinaryLabel(_pInImg, _pInRandomForestModel)))
                             (((ipsdk)(imaproc)(attr))(OutOptRealImg)))

// methods
public:
    processor::RetrievalResult provideProcessor(const boost::shared_ptr<core::BaseAction>& pProcessor,
                                                const core::BasePriority& priority,
                                                core::ActionProviderPtr& pProvider);

    processor::RetrievalResult computePixelClassification(const core::BasePriority& priority, core::ActionProviderPtr& pProvider);

    processor::RetrievalResult computeGetRoiBefore(const core::BasePriority& priority, core::ActionProviderPtr& pProvider);

    processor::RetrievalResult computeGetRoiAfter(const core::BasePriority& priority, core::ActionProviderPtr& pProvider);

    processor::RetrievalResult computePutRoi(const core::BasePriority& priority, core::ActionProviderPtr& pProvider);

    processor::RetrievalResult copy(const core::BasePriority& priority, core::ActionProviderPtr& pProvider,
                                    const ipsdk::image::ImageConstPtr pInImg, const ipsdk::image::ImagePtr pOutImg);

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

// attributes
protected:
    ipUInt64 _numberSplit;
    ipUInt64 _sizeSplit;
    ipUInt64 _minY;
    ipUInt64 _sizeY;
    ipUInt64 _offsetBefore;
    ipUInt64 _offsetAfter;
    ipUInt64 _currentSplit;
    ipUInt64 _currentProcess;

};


} // end of namespace ml
} // end of namespace imaproc
} // end of namespace ipsdk

#endif // __IPSDKIPLMACHINELEARNING_PIXELCLASSIFICATIONRFIMGLVL1_H__