CudaSubImageSplit.h

// CudaSubImageSplit.h:
// --------------------
//

#ifndef __IPSDKIMAGEPROCESSING_CUDASUBIMAGESPLIT_H__
#define __IPSDKIMAGEPROCESSING_CUDASUBIMAGESPLIT_H__

// suppression warnings
// warning C4251: 'ipsdk::imaproc::CudaSubImageSplit<InOutStorageType>::_subImageSplitType' : class 'ipsdk::imaproc::eSubImageSplitType' needs to have dll-interface to be used by clients of class 'ipsdk::imaproc::CudaSubImageSplit<InOutStorageType>'
#pragma warning (push)
#pragma warning (disable : 4251)

#include <IPSDKImageProcessing/IPSDKImageProcessingExports.h>
#include <IPSDKImageProcessing/DataSplit/SubImage/BaseCudaSubImageSplit.h>
#include <IPSDKImageProcessing/DataSplit/SubImage/SubImageSplitTypes.h>
#include <IPSDKImage/Image/ImageTypes.h>
#include <IPSDKUtil/Tools/ProcessingResult.h>

namespace ipsdk {
namespace imaproc {


template <typename TProcessor, typename TAttribute>
class CudaSubImageSplit : public BaseCudaSubImageSplit<typename TAttribute::StorageType>
{
    // predefined public types
public:
    typedef TProcessor ProcessorType;

    typedef TAttribute AttributeType;

    typedef typename TAttribute::StorageType InputStorageType;

    typedef typename TAttribute::StorageType OutputStorageType;

public:
    CudaSubImageSplit() {}
    ~CudaSubImageSplit() {}

// methods from SubImageSplit
public:
    const std::string&      getAttributeName() const;

    const std::string&      getAttributeToolTip() const;

    void                    init(const eSubImageSplitType& subImageSplitType,
                                 const ipUInt64 nbDevices,
                                 const InputStorageType& pImage);

    void                    init(const eSubImageSplitType& subImageSplitType,
                                 const InputStorageType& pImage);

protected:
    bool                    checkProcessorType(const processor::BaseProcessor& processor);

    BoolResult              processAttribute(processor::BaseProcessor& processor,
                                             const ipUInt32 elementIdx,
                                             const core::BaseRequestOrigin& requestOrigin);
};


template <typename TProcessor, typename TAttribute>
inline const std::string&
CudaSubImageSplit<TProcessor, TAttribute>::getAttributeName() const
{
    return TAttribute::getObjectNameStr();
}

template <typename TProcessor, typename TAttribute>
inline const std::string&
CudaSubImageSplit<TProcessor, TAttribute>::getAttributeToolTip() const
{
    return TAttribute::getToolTipStr();
}

//template <typename TProcessor, typename TAttribute>
//inline void
//CudaSubImageSplit<TProcessor, TAttribute>::init(const eSubImageSplitType& subImageSplitType,
//                                               const ipUInt64 nbDevices,
//                                               const InputStorageType& pImage)
//{
//    // call of base class initialization method
//    this->initBase(subImageSplitType, nbDevices, pImage);
//}

template <typename TProcessor, typename TAttribute>
inline void
CudaSubImageSplit<TProcessor, TAttribute>::init(const eSubImageSplitType& subImageSplitType,
                                               const InputStorageType& pImage)
{
    // call of base class initialization method
    this->initBase(subImageSplitType, pImage);
}

template <typename TProcessor, typename TAttribute>
inline bool
CudaSubImageSplit<TProcessor, TAttribute>::checkProcessorType(const processor::BaseProcessor& processor)
{
    // check for consistency between processor and attribute
    BOOST_STATIC_ASSERT_MSG((processor::ExistsAttribute<TAttribute, typename TProcessor::AttributesInfo>::g_bValue),
        "This attribute is not part of processor class");

    return dynamic_cast<const TProcessor*>(&processor) != 0;
}

template <typename TProcessor, typename TAttribute>
inline BoolResult
CudaSubImageSplit<TProcessor, TAttribute>::processAttribute(processor::BaseProcessor& processor,
                                                           const ipUInt32 elementIdx,
                                                           const core::BaseRequestOrigin& requestOrigin)
{
    // cast of input processor object
    TProcessor& tProcessor = static_cast<TProcessor&>(processor);

    // retrieve sub image associated to element
    OutputStorageType pSubImage;
    BoolResult bGetSubImageRes = this->getSubImage(elementIdx, pSubImage);
    if (bGetSubImageRes == true) {

        // update of processor attribute value
        tProcessor.TProcessor::template set<TAttribute>(pSubImage);

        return true;
    }
    else
        return bGetSubImageRes;
}

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

#pragma warning (pop)

#endif // __IPSDKIMAGEPROCESSING_CUDASUBIMAGESPLIT_H__