1793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler/*M/////////////////////////////////////////////////////////////////////////////////////// 2793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler// 3793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler// IMPORTANT: READ BEFORE DOWNLOADING, COPYING, INSTALLING OR USING. 4793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler// 5793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler// By downloading, copying, installing or using the software you agree to this license. 6793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler// If you do not agree to this license, do not download, install, 7793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler// copy or use the software. 8793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler// 9793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler// 10793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler// License Agreement 11793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler// For Open Source Computer Vision Library 12793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler// 13793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler// Copyright (C) 2000-2008, Intel Corporation, all rights reserved. 14793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler// Copyright (C) 2009, Willow Garage Inc., all rights reserved. 15793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler// Third party copyrights are property of their respective owners. 16793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler// 17793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler// Redistribution and use in source and binary forms, with or without modification, 18793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler// are permitted provided that the following conditions are met: 19793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler// 20793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler// * Redistribution's of source code must retain the above copyright notice, 21793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler// this list of conditions and the following disclaimer. 22793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler// 23793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler// * Redistribution's in binary form must reproduce the above copyright notice, 24793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler// this list of conditions and the following disclaimer in the documentation 25793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler// and/or other materials provided with the distribution. 26793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler// 27793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler// * The name of the copyright holders may not be used to endorse or promote products 28793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler// derived from this software without specific prior written permission. 29793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler// 30793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler// This software is provided by the copyright holders and contributors "as is" and 31793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler// any express or implied warranties, including, but not limited to, the implied 32793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler// warranties of merchantability and fitness for a particular purpose are disclaimed. 33793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler// In no event shall the Intel Corporation or contributors be liable for any direct, 34793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler// indirect, incidental, special, exemplary, or consequential damages 35793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler// (including, but not limited to, procurement of substitute goods or services; 36793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler// loss of use, data, or profits; or business interruption) however caused 37793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler// and on any theory of liability, whether in contract, strict liability, 38793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler// or tort (including negligence or otherwise) arising in any way out of 39793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler// the use of this software, even if advised of the possibility of such damage. 40793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler// 41793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler//M*/ 42793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler 43793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler#include "opencv2/opencv_modules.hpp" 44793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler 45793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler#ifndef HAVE_OPENCV_CUDEV 46793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler 47793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler#error "opencv_cudev is required" 48793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler 49793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler#else 50793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler 51793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler#include "opencv2/cudaarithm.hpp" 52793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler#include "opencv2/cudev.hpp" 53793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler#include "opencv2/core/private.cuda.hpp" 54793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler 55793ee12c6df9cad3806238d32528c49a3ff9331dNoah Preslerusing namespace cv; 56793ee12c6df9cad3806238d32528c49a3ff9331dNoah Preslerusing namespace cv::cuda; 57793ee12c6df9cad3806238d32528c49a3ff9331dNoah Preslerusing namespace cv::cudev; 58793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler 59793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler//////////////////////////////////////////////////////////////////////// 60793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler/// merge 61793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler 62793ee12c6df9cad3806238d32528c49a3ff9331dNoah Preslernamespace 63793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler{ 64793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler template <int cn, typename T> struct MergeFunc; 65793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler 66793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler template <typename T> struct MergeFunc<2, T> 67793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler { 68793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler static void call(const GpuMat* src, GpuMat& dst, Stream& stream) 69793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler { 70793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler gridMerge(zipPtr(globPtr<T>(src[0]), globPtr<T>(src[1])), 71793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler globPtr<typename MakeVec<T, 2>::type>(dst), 72793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler stream); 73793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler } 74793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler }; 75793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler 76793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler template <typename T> struct MergeFunc<3, T> 77793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler { 78793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler static void call(const GpuMat* src, GpuMat& dst, Stream& stream) 79793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler { 80793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler gridMerge(zipPtr(globPtr<T>(src[0]), globPtr<T>(src[1]), globPtr<T>(src[2])), 81793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler globPtr<typename MakeVec<T, 3>::type>(dst), 82793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler stream); 83793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler } 84793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler }; 85793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler 86793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler template <typename T> struct MergeFunc<4, T> 87793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler { 88793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler static void call(const GpuMat* src, GpuMat& dst, Stream& stream) 89793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler { 90793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler gridMerge(zipPtr(globPtr<T>(src[0]), globPtr<T>(src[1]), globPtr<T>(src[2]), globPtr<T>(src[3])), 91793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler globPtr<typename MakeVec<T, 4>::type>(dst), 92793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler stream); 93793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler } 94793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler }; 95793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler 96793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler void mergeImpl(const GpuMat* src, size_t n, cv::OutputArray _dst, Stream& stream) 97793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler { 98793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler CV_Assert( src != 0 ); 99793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler CV_Assert( n > 0 && n <= 4 ); 100793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler 101793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler const int depth = src[0].depth(); 102793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler const cv::Size size = src[0].size(); 103793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler 104793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler for (size_t i = 0; i < n; ++i) 105793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler { 106793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler CV_Assert( src[i].size() == size ); 107793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler CV_Assert( src[i].depth() == depth ); 108793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler CV_Assert( src[i].channels() == 1 ); 109793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler } 110793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler 111793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler if (n == 1) 112793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler { 113793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler src[0].copyTo(_dst, stream); 114793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler } 115793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler else 116793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler { 117793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler typedef void (*func_t)(const GpuMat* src, GpuMat& dst, Stream& stream); 118793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler static const func_t funcs[3][5] = 119793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler { 120793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler {MergeFunc<2, uchar>::call, MergeFunc<2, ushort>::call, MergeFunc<2, int>::call, 0, MergeFunc<2, double>::call}, 121793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler {MergeFunc<3, uchar>::call, MergeFunc<3, ushort>::call, MergeFunc<3, int>::call, 0, MergeFunc<3, double>::call}, 122793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler {MergeFunc<4, uchar>::call, MergeFunc<4, ushort>::call, MergeFunc<4, int>::call, 0, MergeFunc<4, double>::call} 123793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler }; 124793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler 125793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler const int channels = static_cast<int>(n); 126793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler 127793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler GpuMat dst = getOutputMat(_dst, size, CV_MAKE_TYPE(depth, channels), stream); 128793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler 129793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler const func_t func = funcs[channels - 2][CV_ELEM_SIZE(depth) / 2]; 130793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler 131793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler if (func == 0) 132793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler CV_Error(cv::Error::StsUnsupportedFormat, "Unsupported channel count or data type"); 133793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler 134793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler func(src, dst, stream); 135793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler 136793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler syncOutput(dst, _dst, stream); 137793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler } 138793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler } 139793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler} 140793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler 141793ee12c6df9cad3806238d32528c49a3ff9331dNoah Preslervoid cv::cuda::merge(const GpuMat* src, size_t n, OutputArray dst, Stream& stream) 142793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler{ 143793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler mergeImpl(src, n, dst, stream); 144793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler} 145793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler 146793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler 147793ee12c6df9cad3806238d32528c49a3ff9331dNoah Preslervoid cv::cuda::merge(const std::vector<GpuMat>& src, OutputArray dst, Stream& stream) 148793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler{ 149793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler mergeImpl(&src[0], src.size(), dst, stream); 150793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler} 151793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler 152793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler//////////////////////////////////////////////////////////////////////// 153793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler/// split 154793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler 155793ee12c6df9cad3806238d32528c49a3ff9331dNoah Preslernamespace 156793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler{ 157793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler template <int cn, typename T> struct SplitFunc; 158793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler 159793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler template <typename T> struct SplitFunc<2, T> 160793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler { 161793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler static void call(const GpuMat& src, GpuMat* dst, Stream& stream) 162793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler { 163793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler GlobPtrSz<T> dstarr[2] = 164793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler { 165793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler globPtr<T>(dst[0]), globPtr<T>(dst[1]) 166793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler }; 167793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler 168793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler gridSplit(globPtr<typename MakeVec<T, 2>::type>(src), dstarr, stream); 169793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler } 170793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler }; 171793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler 172793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler template <typename T> struct SplitFunc<3, T> 173793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler { 174793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler static void call(const GpuMat& src, GpuMat* dst, Stream& stream) 175793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler { 176793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler GlobPtrSz<T> dstarr[3] = 177793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler { 178793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler globPtr<T>(dst[0]), globPtr<T>(dst[1]), globPtr<T>(dst[2]) 179793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler }; 180793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler 181793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler gridSplit(globPtr<typename MakeVec<T, 3>::type>(src), dstarr, stream); 182793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler } 183793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler }; 184793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler 185793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler template <typename T> struct SplitFunc<4, T> 186793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler { 187793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler static void call(const GpuMat& src, GpuMat* dst, Stream& stream) 188793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler { 189793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler GlobPtrSz<T> dstarr[4] = 190793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler { 191793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler globPtr<T>(dst[0]), globPtr<T>(dst[1]), globPtr<T>(dst[2]), globPtr<T>(dst[3]) 192793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler }; 193793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler 194793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler gridSplit(globPtr<typename MakeVec<T, 4>::type>(src), dstarr, stream); 195793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler } 196793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler }; 197793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler 198793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler void splitImpl(const GpuMat& src, GpuMat* dst, Stream& stream) 199793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler { 200793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler typedef void (*func_t)(const GpuMat& src, GpuMat* dst, Stream& stream); 201793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler static const func_t funcs[3][5] = 202793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler { 203793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler {SplitFunc<2, uchar>::call, SplitFunc<2, ushort>::call, SplitFunc<2, int>::call, 0, SplitFunc<2, double>::call}, 204793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler {SplitFunc<3, uchar>::call, SplitFunc<3, ushort>::call, SplitFunc<3, int>::call, 0, SplitFunc<3, double>::call}, 205793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler {SplitFunc<4, uchar>::call, SplitFunc<4, ushort>::call, SplitFunc<4, int>::call, 0, SplitFunc<4, double>::call} 206793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler }; 207793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler 208793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler CV_Assert( dst != 0 ); 209793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler 210793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler const int depth = src.depth(); 211793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler const int channels = src.channels(); 212793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler 213793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler CV_Assert( channels <= 4 ); 214793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler 215793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler if (channels == 0) 216793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler return; 217793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler 218793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler if (channels == 1) 219793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler { 220793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler src.copyTo(dst[0], stream); 221793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler return; 222793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler } 223793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler 224793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler for (int i = 0; i < channels; ++i) 225793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler dst[i].create(src.size(), depth); 226793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler 227793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler const func_t func = funcs[channels - 2][CV_ELEM_SIZE(depth) / 2]; 228793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler 229793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler if (func == 0) 230793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler CV_Error(cv::Error::StsUnsupportedFormat, "Unsupported channel count or data type"); 231793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler 232793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler func(src, dst, stream); 233793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler } 234793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler} 235793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler 236793ee12c6df9cad3806238d32528c49a3ff9331dNoah Preslervoid cv::cuda::split(InputArray _src, GpuMat* dst, Stream& stream) 237793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler{ 238793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler GpuMat src = getInputMat(_src, stream); 239793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler splitImpl(src, dst, stream); 240793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler} 241793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler 242793ee12c6df9cad3806238d32528c49a3ff9331dNoah Preslervoid cv::cuda::split(InputArray _src, std::vector<GpuMat>& dst, Stream& stream) 243793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler{ 244793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler GpuMat src = getInputMat(_src, stream); 245793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler dst.resize(src.channels()); 246793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler if (src.channels() > 0) 247793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler splitImpl(src, &dst[0], stream); 248793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler} 249793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler 250793ee12c6df9cad3806238d32528c49a3ff9331dNoah Presler#endif 251