/*M///////////////////////////////////////////////////////////////////////////////////////
//
//  IMPORTANT: READ BEFORE DOWNLOADING, COPYING, INSTALLING OR USING.
//
//  By downloading, copying, installing or using the software you agree to this license.
//  If you do not agree to this license, do not download, install,
//  copy or use the software.
//
//
//                           License Agreement
//                For Open Source Computer Vision Library
//
// Copyright (C) 2000-2008, Intel Corporation, all rights reserved.
// Copyright (C) 2009, Willow Garage Inc., all rights reserved.
// Third party copyrights are property of their respective owners.
//
// Redistribution and use in source and binary forms, with or without modification,
// are permitted provided that the following conditions are met:
//
//   * Redistribution's of source code must retain the above copyright notice,
//     this list of conditions and the following disclaimer.
//
//   * Redistribution's in binary form must reproduce the above copyright notice,
//     this list of conditions and the following disclaimer in the documentation
//     and/or other materials provided with the distribution.
//
//   * The name of the copyright holders may not be used to endorse or promote products
//     derived from this software without specific prior written permission.
//
// This software is provided by the copyright holders and contributors "as is" and
// any express or implied warranties, including, but not limited to, the implied
// warranties of merchantability and fitness for a particular purpose are disclaimed.
// In no event shall the Intel Corporation or contributors be liable for any direct,
// indirect, incidental, special, exemplary, or consequential damages
// (including, but not limited to, procurement of substitute goods or services;
// loss of use, data, or profits; or business interruption) however caused
// and on any theory of liability, whether in contract, strict liability,
// or tort (including negligence or otherwise) arising in any way out of
// the use of this software, even if advised of the possibility of such damage.
//
//M*/

#include <internal_shared.hpp>
#include <opencv2/gpu/device/transform.hpp>
#include <opencv2/gpu/device/color.hpp>
#include <cvt_colot_internal.h>

namespace cv { namespace gpu { namespace device
{
    OPENCV_GPU_TRANSFORM_FUNCTOR_TRAITS(bgra_to_rgba_traits<uchar>::functor_type)
    {
        enum { smart_block_dim_x = 8 };
        enum { smart_block_dim_y = 8 };
        enum { smart_shift = 4 };
    };

    OPENCV_GPU_TRANSFORM_FUNCTOR_TRAITS(bgra_to_bgr555_traits::functor_type)
    {
        enum { smart_block_dim_y = 8 };
        enum { smart_shift = 4 };
    };
    OPENCV_GPU_TRANSFORM_FUNCTOR_TRAITS(rgba_to_bgr555_traits::functor_type)
    {
        enum { smart_block_dim_y = 8 };
        enum { smart_shift = 4 };
    };
    OPENCV_GPU_TRANSFORM_FUNCTOR_TRAITS(bgra_to_bgr565_traits::functor_type)
    {
        enum { smart_block_dim_y = 8 };
        enum { smart_shift = 4 };
    };
    OPENCV_GPU_TRANSFORM_FUNCTOR_TRAITS(rgba_to_bgr565_traits::functor_type)
    {
        enum { smart_block_dim_y = 8 };
        enum { smart_shift = 4 };
    };

    OPENCV_GPU_TRANSFORM_FUNCTOR_TRAITS(bgr555_to_bgra_traits::functor_type)
    {
        enum { smart_block_dim_y = 8 };
        enum { smart_shift = 4 };
    };
    OPENCV_GPU_TRANSFORM_FUNCTOR_TRAITS(bgr555_to_rgba_traits::functor_type)
    {
        enum { smart_block_dim_y = 8 };
        enum { smart_shift = 4 };
    };
    OPENCV_GPU_TRANSFORM_FUNCTOR_TRAITS(bgr565_to_bgra_traits::functor_type)
    {
        enum { smart_block_dim_y = 8 };
        enum { smart_shift = 4 };
    };
    OPENCV_GPU_TRANSFORM_FUNCTOR_TRAITS(bgr565_to_rgba_traits::functor_type)
    {
        enum { smart_block_dim_y = 8 };
        enum { smart_shift = 4 };
    };

    OPENCV_GPU_TRANSFORM_FUNCTOR_TRAITS(gray_to_bgra_traits<uchar>::functor_type)
    {
        enum { smart_block_dim_y = 8 };
        enum { smart_shift = 4 };
    };

    OPENCV_GPU_TRANSFORM_FUNCTOR_TRAITS(gray_to_bgr555_traits::functor_type)
    {
        enum { smart_shift = 4 };
    };
    OPENCV_GPU_TRANSFORM_FUNCTOR_TRAITS(gray_to_bgr565_traits::functor_type)
    {
        enum { smart_shift = 4 };
    };

    OPENCV_GPU_TRANSFORM_FUNCTOR_TRAITS(bgra_to_yuv4_traits<uchar>::functor_type)
    {
        enum { smart_block_dim_y = 8 };
        enum { smart_shift = 4 };
    };
    OPENCV_GPU_TRANSFORM_FUNCTOR_TRAITS(rgba_to_yuv4_traits<uchar>::functor_type)
    {
        enum { smart_block_dim_y = 8 };
        enum { smart_shift = 4 };
    };

    OPENCV_GPU_TRANSFORM_FUNCTOR_TRAITS(yuv4_to_bgra_traits<uchar>::functor_type)
    {
        enum { smart_block_dim_y = 8 };
        enum { smart_shift = 4 };
    };
    OPENCV_GPU_TRANSFORM_FUNCTOR_TRAITS(yuv4_to_rgba_traits<uchar>::functor_type)
    {
        enum { smart_block_dim_y = 8 };
        enum { smart_shift = 4 };
    };

    OPENCV_GPU_TRANSFORM_FUNCTOR_TRAITS(bgra_to_YCrCb4_traits<uchar>::functor_type)
    {
        enum { smart_block_dim_y = 8 };
        enum { smart_shift = 4 };
    };
    OPENCV_GPU_TRANSFORM_FUNCTOR_TRAITS(rgba_to_YCrCb4_traits<uchar>::functor_type)
    {
        enum { smart_block_dim_y = 8 };
        enum { smart_shift = 4 };
    };

    OPENCV_GPU_TRANSFORM_FUNCTOR_TRAITS(YCrCb4_to_bgra_traits<uchar>::functor_type)
    {
        enum { smart_block_dim_y = 8 };
        enum { smart_shift = 4 };
    };
    OPENCV_GPU_TRANSFORM_FUNCTOR_TRAITS(YCrCb4_to_rgba_traits<uchar>::functor_type)
    {
        enum { smart_block_dim_y = 8 };
        enum { smart_shift = 4 };
    };

    OPENCV_GPU_TRANSFORM_FUNCTOR_TRAITS(bgra_to_xyz4_traits<uchar>::functor_type)
    {
        enum { smart_block_dim_y = 8 };
        enum { smart_shift = 4 };
    };
    OPENCV_GPU_TRANSFORM_FUNCTOR_TRAITS(rgba_to_xyz4_traits<uchar>::functor_type)
    {
        enum { smart_block_dim_y = 8 };
        enum { smart_shift = 4 };
    };

    OPENCV_GPU_TRANSFORM_FUNCTOR_TRAITS(xyz4_to_bgra_traits<uchar>::functor_type)
    {
        enum { smart_block_dim_y = 8 };
        enum { smart_shift = 4 };
    };
    OPENCV_GPU_TRANSFORM_FUNCTOR_TRAITS(xyz4_to_rgba_traits<uchar>::functor_type)
    {
        enum { smart_block_dim_y = 8 };
        enum { smart_shift = 4 };
    };

    OPENCV_GPU_TRANSFORM_FUNCTOR_TRAITS(bgra_to_hsv4_traits<uchar>::functor_type)
    {
        enum { smart_block_dim_y = 8 };
        enum { smart_shift = 4 };
    };
    OPENCV_GPU_TRANSFORM_FUNCTOR_TRAITS(rgba_to_hsv4_traits<uchar>::functor_type)
    {
        enum { smart_block_dim_y = 8 };
        enum { smart_shift = 4 };
    };

    OPENCV_GPU_TRANSFORM_FUNCTOR_TRAITS(hsv4_to_bgra_traits<uchar>::functor_type)
    {
        enum { smart_block_dim_y = 8 };
        enum { smart_shift = 4 };
    };
    OPENCV_GPU_TRANSFORM_FUNCTOR_TRAITS(hsv4_to_rgba_traits<uchar>::functor_type)
    {
        enum { smart_block_dim_y = 8 };
        enum { smart_shift = 4 };
    };

    OPENCV_GPU_TRANSFORM_FUNCTOR_TRAITS(bgra_to_hls4_traits<uchar>::functor_type)
    {
        enum { smart_block_dim_y = 8 };
        enum { smart_shift = 4 };
    };
    OPENCV_GPU_TRANSFORM_FUNCTOR_TRAITS(rgba_to_hls4_traits<uchar>::functor_type)
    {
        enum { smart_block_dim_y = 8 };
        enum { smart_shift = 4 };
    };

    OPENCV_GPU_TRANSFORM_FUNCTOR_TRAITS(hls4_to_bgra_traits<uchar>::functor_type)
    {
        enum { smart_block_dim_y = 8 };
        enum { smart_shift = 4 };
    };
    OPENCV_GPU_TRANSFORM_FUNCTOR_TRAITS(hls4_to_rgba_traits<uchar>::functor_type)
    {
        enum { smart_block_dim_y = 8 };
        enum { smart_shift = 4 };
    };

#define OPENCV_GPU_IMPLEMENT_CVTCOLOR(name, traits) \
    void name(const DevMem2Db& src, const DevMem2Db& dst, cudaStream_t stream) \
    { \
        traits::functor_type functor = traits::create_functor(); \
        typedef typename traits::functor_type::argument_type src_t; \
        typedef typename traits::functor_type::result_type   dst_t; \
        cv::gpu::device::transform((DevMem2D_<src_t>)src, (DevMem2D_<dst_t>)dst, functor, WithOutMask(), stream); \
    }

#define OPENCV_GPU_IMPLEMENT_CVTCOLOR_ONE(name) \
    OPENCV_GPU_IMPLEMENT_CVTCOLOR(name, name ## _traits)

#define OPENCV_GPU_IMPLEMENT_CVTCOLOR_ALL(name) \
    OPENCV_GPU_IMPLEMENT_CVTCOLOR(name ## _8u, name ## _traits<uchar>) \
    OPENCV_GPU_IMPLEMENT_CVTCOLOR(name ## _16u, name ## _traits<ushort>) \
    OPENCV_GPU_IMPLEMENT_CVTCOLOR(name ## _32f, name ## _traits<float>)

#define OPENCV_GPU_IMPLEMENT_CVTCOLOR_8U32F(name) \
    OPENCV_GPU_IMPLEMENT_CVTCOLOR(name ## _8u, name ## _traits<uchar>) \
    OPENCV_GPU_IMPLEMENT_CVTCOLOR(name ## _32f, name ## _traits<float>) \
    OPENCV_GPU_IMPLEMENT_CVTCOLOR(name ## _full_8u, name ## _full_traits<uchar>) \
    OPENCV_GPU_IMPLEMENT_CVTCOLOR(name ## _full_32f, name ## _full_traits<float>)

    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ALL(bgr_to_rgb)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ALL(bgr_to_bgra)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ALL(bgr_to_rgba)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ALL(bgra_to_bgr)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ALL(bgra_to_rgb)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ALL(bgra_to_rgba)

    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ONE(bgr_to_bgr555)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ONE(bgr_to_bgr565)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ONE(rgb_to_bgr555)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ONE(rgb_to_bgr565)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ONE(bgra_to_bgr555)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ONE(bgra_to_bgr565)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ONE(rgba_to_bgr555)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ONE(rgba_to_bgr565)

    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ONE(bgr555_to_rgb)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ONE(bgr565_to_rgb)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ONE(bgr555_to_bgr)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ONE(bgr565_to_bgr)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ONE(bgr555_to_rgba)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ONE(bgr565_to_rgba)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ONE(bgr555_to_bgra)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ONE(bgr565_to_bgra)

    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ALL(gray_to_bgr)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ALL(gray_to_bgra)

    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ONE(gray_to_bgr555)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ONE(gray_to_bgr565)

    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ONE(bgr555_to_gray)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ONE(bgr565_to_gray)

    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ALL(rgb_to_gray)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ALL(bgr_to_gray)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ALL(rgba_to_gray)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ALL(bgra_to_gray)

    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ALL(rgb_to_yuv)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ALL(rgba_to_yuv)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ALL(rgb_to_yuv4)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ALL(rgba_to_yuv4)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ALL(bgr_to_yuv)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ALL(bgra_to_yuv)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ALL(bgr_to_yuv4)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ALL(bgra_to_yuv4)

    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ALL(yuv_to_rgb)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ALL(yuv_to_rgba)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ALL(yuv4_to_rgb)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ALL(yuv4_to_rgba)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ALL(yuv_to_bgr)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ALL(yuv_to_bgra)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ALL(yuv4_to_bgr)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ALL(yuv4_to_bgra)

    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ALL(rgb_to_YCrCb)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ALL(rgba_to_YCrCb)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ALL(rgb_to_YCrCb4)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ALL(rgba_to_YCrCb4)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ALL(bgr_to_YCrCb)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ALL(bgra_to_YCrCb)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ALL(bgr_to_YCrCb4)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ALL(bgra_to_YCrCb4)

    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ALL(YCrCb_to_rgb)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ALL(YCrCb_to_rgba)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ALL(YCrCb4_to_rgb)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ALL(YCrCb4_to_rgba)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ALL(YCrCb_to_bgr)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ALL(YCrCb_to_bgra)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ALL(YCrCb4_to_bgr)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ALL(YCrCb4_to_bgra)

    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ALL(rgb_to_xyz)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ALL(rgba_to_xyz)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ALL(rgb_to_xyz4)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ALL(rgba_to_xyz4)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ALL(bgr_to_xyz)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ALL(bgra_to_xyz)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ALL(bgr_to_xyz4)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ALL(bgra_to_xyz4)

    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ALL(xyz_to_rgb)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ALL(xyz4_to_rgb)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ALL(xyz_to_rgba)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ALL(xyz4_to_rgba)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ALL(xyz_to_bgr)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ALL(xyz4_to_bgr)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ALL(xyz_to_bgra)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_ALL(xyz4_to_bgra)

    OPENCV_GPU_IMPLEMENT_CVTCOLOR_8U32F(rgb_to_hsv)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_8U32F(rgba_to_hsv)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_8U32F(rgb_to_hsv4)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_8U32F(rgba_to_hsv4)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_8U32F(bgr_to_hsv)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_8U32F(bgra_to_hsv)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_8U32F(bgr_to_hsv4)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_8U32F(bgra_to_hsv4)

    OPENCV_GPU_IMPLEMENT_CVTCOLOR_8U32F(hsv_to_rgb)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_8U32F(hsv_to_rgba)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_8U32F(hsv4_to_rgb)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_8U32F(hsv4_to_rgba)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_8U32F(hsv_to_bgr)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_8U32F(hsv_to_bgra)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_8U32F(hsv4_to_bgr)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_8U32F(hsv4_to_bgra)

    OPENCV_GPU_IMPLEMENT_CVTCOLOR_8U32F(rgb_to_hls)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_8U32F(rgba_to_hls)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_8U32F(rgb_to_hls4)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_8U32F(rgba_to_hls4)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_8U32F(bgr_to_hls)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_8U32F(bgra_to_hls)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_8U32F(bgr_to_hls4)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_8U32F(bgra_to_hls4)

    OPENCV_GPU_IMPLEMENT_CVTCOLOR_8U32F(hls_to_rgb)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_8U32F(hls_to_rgba)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_8U32F(hls4_to_rgb)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_8U32F(hls4_to_rgba)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_8U32F(hls_to_bgr)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_8U32F(hls_to_bgra)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_8U32F(hls4_to_bgr)
    OPENCV_GPU_IMPLEMENT_CVTCOLOR_8U32F(hls4_to_bgra)

    #undef OPENCV_GPU_IMPLEMENT_CVTCOLOR
    #undef OPENCV_GPU_IMPLEMENT_CVTCOLOR_ONE
    #undef OPENCV_GPU_IMPLEMENT_CVTCOLOR_ALL
    #undef OPENCV_GPU_IMPLEMENT_CVTCOLOR_8U32F
}}} // namespace cv { namespace gpu { namespace device
