CudaSubImageSplitOperators.h

Go to the documentation of this file.

// CudaSubImageSplitOperators.h:
// -------------------------
//

#ifndef __IPSDKIMAGEPROCESSING_CUDASUBIMAGESPLITOPERATORS_H__
#define __IPSDKIMAGEPROCESSING_CUDASUBIMAGESPLITOPERATORS_H__

#include <IPSDKBaseProcessing/DataSplit/Static/StaticSplitOperators.h>
#include <IPSDKImageProcessing/DataSplit/SubImage/CudaSubImageSplit.h>
#include <IPSDKBaseProcessing/ProcessorDispatcher/Static/StaticProcessorDispatcher.h>
#include <IPSDKImage/Image/BaseImage.h>
#include <IPSDKImage/Image/Gpu/CudaImage.h>
#include <boost/make_shared.hpp>
#include <boost/utility/enable_if.hpp>
#include <boost/mpl/equal_to.hpp>
#include <boost/mpl/not_equal_to.hpp>

namespace ipsdk {
namespace imaproc {


template <typename TProcessor, typename TAttribute>
inline boost::shared_ptr<CudaSubImageSplit<TProcessor, TAttribute> >
splitGpu(const image::ImagePtr& pImage)
{
    typedef CudaSubImageSplit<TProcessor, TAttribute> DataSplit;
    boost::shared_ptr<DataSplit> pDataSplit = boost::make_shared<DataSplit>();

    const ipsdk::image::CudaImage& cudaImage = static_cast<const ipsdk::image::CudaImage&>(*pImage);
    //const ipUInt64 nbDevices = cudaImage.getNbDevices();

    pDataSplit->init(eSubImageSplitType::eSIST_GpuDevices,
                     //nbDevices,
                     pImage);

    return pDataSplit;
}
template <typename TProcessor, typename TAttribute>
inline boost::shared_ptr<CudaSubImageSplit<TProcessor, TAttribute> >
splitGpu(const image::ImageConstPtr& pImage)
{
    typedef CudaSubImageSplit<TProcessor, TAttribute> DataSplit;
    boost::shared_ptr<DataSplit> pDataSplit = boost::make_shared<DataSplit>();

    const ipsdk::image::CudaImage& cudaImage = static_cast<const ipsdk::image::CudaImage&>(*pImage);
    //const ipUInt64 nbDevices = cudaImage.getNbDevices();

    pDataSplit->init(eSubImageSplitType::eSIST_GpuDevices,
                     //nbDevices,
                     pImage);

    return pDataSplit;
}
template <typename TProcessor, typename TAttribute, typename TInputAttribute>
inline boost::shared_ptr<CudaSubImageSplit<TProcessor, TAttribute> >
splitGpu(const boost::shared_ptr<TInputAttribute>& pInputAttribute)
{
    return splitGpu<TProcessor, TAttribute>(pInputAttribute->getStorage());
}


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

#endif // __IPSDKIMAGEPROCESSING_CUDASUBIMAGESPLITOPERATORS_H__