generic-library/opencv
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

added parallel version of CvtColorLoop - main inner function of cvtColor

Browse Source
This commit is contained in:
Ilya Lavrenov 2012-09-28 17:42:48 +04:00
parent c6e741196a
commit 9ee87bd04b
1 changed files with 32 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

44
modules/imgproc/src/color.cpp
View File

@ -156,25 +156,45 @@ template<> struct ColorChannel<float>
///////////////////////////// Top-level template function //////////////////////////////// ///////////////////////////// Top-level template function ////////////////////////////////
template<class Cvt> void CvtColorLoop(const Mat& srcmat, Mat& dstmat, const Cvt& cvt) template <typename Cvt>
class CvtColorLoop_Invoker :
public ParallelLoopBody
{ {
typedef typename Cvt::channel_type _Tp; typedef typename Cvt::channel_type _Tp;
Size sz = srcmat.size(); public:
const uchar* src = srcmat.data;
uchar* dst = dstmat.data; CvtColorLoop_Invoker(const Mat& _src, Mat& _dst, const Cvt& _cvt) :
size_t srcstep = srcmat.step, dststep = dstmat.step; ParallelLoopBody(), src(_src), dst(_dst), cvt(_cvt)
if( srcmat.isContinuous() && dstmat.isContinuous() )
{ {
sz.width *= sz.height;
sz.height = 1;
} }
virtual void operator()(const Range& range) const
{
int i = range.start;
const uchar* yS = src.data + src.step * i;
uchar* yD = dst.data + dst.step * i;
for( ; sz.height--; src += srcstep, dst += dststep ) for ( ; i < range.end; ++i, yS += src.step, yD += dst.step )
cvt((const _Tp*)src, (_Tp*)dst, sz.width); cvt((const _Tp*)yS, (_Tp*)yD, src.cols);
}
private:
const Mat src;
Mat dst;
const Cvt cvt;
CvtColorLoop_Invoker(const CvtColorLoop_Invoker&);
const CvtColorLoop_Invoker& operator= (const CvtColorLoop_Invoker&);
};
template <typename Cvt>
void CvtColorLoop(const Mat& src, Mat& dst, const Cvt& cvt)
{
Range range(0, src.rows);
CvtColorLoop_Invoker<Cvt> invoker(src, dst, cvt);
parallel_for_(range, invoker);
} }
////////////////// Various 3/4-channel to 3/4-channel RGB transformations ///////////////// ////////////////// Various 3/4-channel to 3/4-channel RGB transformations /////////////////
template<typename _Tp> struct RGB2RGB template<typename _Tp> struct RGB2RGB