StackRegistrationImgLvl2.h

// StackRegistrationImgLvl2.h:
// ---------------------------
//

#ifndef __IPSDKIPLREGISTRATION_STACKREGISTRATIONIMGLVL2_H__
#define __IPSDKIPLREGISTRATION_STACKREGISTRATIONIMGLVL2_H__

#include <IPSDKIPL/IPSDKIPLRegistration/IPSDKIPLRegistrationExports.h>
#include <IPSDKBaseProcessing/Processor/ProcessorHdrMacros.h>
#include <IPSDKIPL/IPSDKIPLAttributes/DataItem/WarpMotionTransform2d.h>
#include <IPSDKIPL/IPSDKIPLAttributes/DataItem/Features2dRegistrationResult.h>
#include <IPSDKIPL/IPSDKIPLAttributes/DataItem/PlanIndexedFeatures2dGreySignature.h>

// Attributes headers
#include <IPSDKIPL/IPSDKIPLAttributes/Attribute/Image/Output/OutImg.h>
#include <IPSDKIPL/IPSDKIPLAttributes/Attribute/SubImage/Input/3d/InSubImg3d.h>
#include <IPSDKIPL/IPSDKIPLAttributes/Attribute/Value/InOptTheta.h>
#include <IPSDKIPL/IPSDKIPLAttributes/Attribute/Value/InOptMaxTranslation.h>
#include <IPSDKIPL/IPSDKIPLAttributes/Attribute/Enumerate/InOptInterpolationPolicy.h>
#include <IPSDKIPL/IPSDKIPLAttributes/Attribute/DataItem/InOptRegistrationEstimationConfig.h>
#include <IPSDKIPL/IPSDKIPLAttributes/Attribute/DataItem/InOptWarpMotionTransform2dColl.h>
#include <IPSDKIPL/IPSDKIPLAttributes/Attribute/DataItem/OutWarpMotionTransform2dColl.h>
#include <IPSDKIPL/IPSDKIPLAttributes/Attribute/Value/InOptCorrelationThreshold2d.h>
#include <IPSDKIPL/IPSDKIPLAttributes/Attribute/Image/Output/Working/OutWk1RealImg.h>
#include <IPSDKIPL/IPSDKIPLAttributes/Attribute/Image/Output/Working/OutWk2RealImg.h>
#include <IPSDKIPL/IPSDKIPLAttributes/Attribute/Image/Output/Working/OutWk3RealImg.h>
#include <IPSDKIPL/IPSDKIPLAttributes/Attribute/Image/Output/Working/OutWk4RealImg.h>

namespace ipsdk {
namespace imaproc {
namespace registration {


// declare processor class
IPSDK_DECLARE_LVL2_PROCESSOR(REP_ASYNC1, IPSDKIPLRegistration, StackRegistrationImg,
                             ipsdk::imaproc::matchSize(_pInSubImg3d, _pOutImg),
                             (((ipsdk)(imaproc)(attr))(InSubImg3d))
                             (((ipsdk)(imaproc)(attr))(InOptTheta))
                             (((ipsdk)(imaproc)(attr))(InOptMaxTranslation))
                             (((ipsdk)(imaproc)(attr))(InOptInterpolationPolicy))
                             (((ipsdk)(imaproc)(attr))(InOptCorrelationThreshold2d))
                             (((ipsdk)(imaproc)(attr))(InOptRegistrationEstimationConfig))
                             (((ipsdk)(imaproc)(attr))(InOptWarpMotionTransform2dColl))
                             (((ipsdk)(imaproc)(attr))(OutWarpMotionTransform2dColl))
                             (((ipsdk)(imaproc)(attr))(OutWk1RealImg))
                             (((ipsdk)(imaproc)(attr))(OutWk2RealImg))
                             (((ipsdk)(imaproc)(attr))(OutWk3RealImg))
                             (((ipsdk)(imaproc)(attr))(OutWk4RealImg))
                             (((ipsdk)(imaproc)(attr))(OutImg)))

// methods
public:

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

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

    processor::RetrievalResult extractFeatures(const core::BasePriority& priority, core::ActionProviderPtr& pProvider);
    processor::RetrievalResult estimTransformation(const core::BasePriority& priority, core::ActionProviderPtr& pProvider, const ipsdk::ipUInt64 sliceIdx);
    processor::RetrievalResult copy(const core::BasePriority& priority, core::ActionProviderPtr& pProvider);
    processor::RetrievalResult warp(const core::BasePriority& priority, core::ActionProviderPtr& pProvider, const ipsdk::ipUInt64 sliceIdx);

// attributes
protected:
    std::vector<ipsdk::imaproc::attr::WarpMotionTransform2dPtr> _vTransforms;

    ipsdk::ipReal32                                             _displacement;

    ipsdk::ipReal64                                             _maxTranslation;

    ipsdk::imaproc::attr::Features2dGreySignaturePtr            _pFeaturesFixed;
    ipsdk::imaproc::attr::Features2dGreySignaturePtr            _pFeaturesMoving;

    ipsdk::imaproc::attr::Features2dRegistrationResultPtr       _pFeatures2dRegistrationResult;

    ipsdk::ipUInt8                                              _stepIdx;

    ipsdk::ipUInt64                                             _sizeZ;

    ipsdk::imaproc::attr::PlanIndexedFeatures2dGreySignaturePtr _pPIFeatures;

    ipBool                                                      _bCopyTransformFromInput;
};


} // end of namespace registration
} // end of namespace imaproc
} // end of namespace ipsdk

#endif // __IPSDKIPLREGISTRATION_STACKREGISTRATIONIMGLVL2_H__