mirror/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2024-11-28 05:06:29 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

YUV420i to RGB conversion is added to cvtColor

Browse Source
This commit is contained in:
Andrey Kamaev 2011-06-04 18:15:52 +00:00
parent d77915759a
commit cdda5ec491
4 changed files with 160 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
modules/imgproc/include/opencv2/imgproc/types_c.h
View File

@ -225,6 +225,9 @@ enum
CV_BayerGB2GRAY = 87,
CV_BayerRG2GRAY = 88,
CV_BayerGR2GRAY = 89,
CV_YUV420i2RGB = 90,
CV_YUV420i2BGR = 91,
CV_COLORCVT_MAX =100
};

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

@ -2648,10 +2648,130 @@ static void Bayer2RGB_VNG_8u( const Mat& srcmat, Mat& dstmat, int code )
}
}
}
///////////////////////////////////// YUV420i -> RGB /////////////////////////////////////
template<int R>
struct YUV420i2BGR888Invoker
{
Mat& dst;
const uchar* my1, *muv;
int width;
YUV420i2BGR888Invoker(Mat& _dst, int _width, const uchar* _y1, const uchar* _uv)
: dst(_dst), my1(_y1), muv(_uv), width(_width) {}
void operator()(const BlockedRange& range) const
{
//B = 1.164(Y - 16) + 2.018(U - 128)
//G = 1.164(Y - 16) - 0.813(V - 128) - 0.391(U - 128)
//R = 1.164(Y - 16) + 1.596(V - 128)
const uchar* y1 = my1 + range.begin() * width, *uv = muv + range.begin() * width / 2;
for (int j = range.begin(); j < range.end(); j+=2, y1+=width*2, uv+=width)
{
uchar* row1 = dst.ptr<uchar>(j);
uchar* row2 = dst.ptr<uchar>(j+1);
const uchar* y2 = y1 + width;
for(int i = 0; i < width; i+=2,row1+=6,row2+=6)
{
int cr = uv[i] - 128;
int cb = uv[i+1] - 128;
int ruv = 409 * cr + 128;
int guv = 128 - 100 * cb - 208 * cr;
int buv = 516 * cb + 128;
int y00 = (y1[i] - 16) * 298;
row1[0+R] = saturate_cast<uchar>((y00 + buv) >> 8);
row1[1] = saturate_cast<uchar>((y00 + guv) >> 8);
row1[2-R] = saturate_cast<uchar>((y00 + ruv) >> 8);
int y01 = (y1[i+1] - 16) * 298;
row1[3+R] = saturate_cast<uchar>((y01 + buv) >> 8);
row1[4] = saturate_cast<uchar>((y01 + guv) >> 8);
row1[5-R] = saturate_cast<uchar>((y01 + ruv) >> 8);
int y10 = (y2[i] - 16) * 298;
row2[0+R] = saturate_cast<uchar>((y10 + buv) >> 8);
row2[1] = saturate_cast<uchar>((y10 + guv) >> 8);
row2[2-R] = saturate_cast<uchar>((y10 + ruv) >> 8);
int y11 = (y2[i+1] - 16) * 298;
row2[3+R] = saturate_cast<uchar>((y11 + buv) >> 8);
row2[4] = saturate_cast<uchar>((y11 + guv) >> 8);
row2[5-R] = saturate_cast<uchar>((y11 + ruv) >> 8);
}
}
}
};
template<int R>
struct YUV420i2BGRA8888Invoker
{
Mat& dst;
const uchar* my1, *muv;
int width;
YUV420i2BGRA8888Invoker(Mat& _dst, int _width, const uchar* _y1, const uchar* _uv)
: dst(_dst), my1(_y1), muv(_uv), width(_width) {}
void operator()(const BlockedRange& range) const
{
//B = 1.164(Y - 16) + 2.018(U - 128)
//G = 1.164(Y - 16) - 0.813(V - 128) - 0.391(U - 128)
//R = 1.164(Y - 16) + 1.596(V - 128)
const uchar* y1 = my1 + range.begin() * width, *uv = muv + range.begin() * width / 2;
for (int j = range.begin(); j < range.end(); j+=2, y1+=width*2, uv+=width)
{
uchar* row1 = dst.ptr<uchar>(j);
uchar* row2 = dst.ptr<uchar>(j+1);
const uchar* y2 = y1 + width;
for(int i = 0; i < width; i+=2,row1+=8,row2+=8)
{
int cr = uv[i] - 128;
int cb = uv[i+1] - 128;
int ruv = 409 * cr + 128;
int guv = 128 - 100 * cb - 208 * cr;
int buv = 516 * cb + 128;
int y00 = (y1[i] - 16) * 298;
row1[0+R] = saturate_cast<uchar>((y00 + buv) >> 8);
row1[1] = saturate_cast<uchar>((y00 + guv) >> 8);
row1[2-R] = saturate_cast<uchar>((y00 + ruv) >> 8);
row1[3] = (uchar)0xff;
int y01 = (y1[i+1] - 16) * 298;
row1[4+R] = saturate_cast<uchar>((y01 + buv) >> 8);
row1[5] = saturate_cast<uchar>((y01 + guv) >> 8);
row1[6-R] = saturate_cast<uchar>((y01 + ruv) >> 8);
row1[7] = (uchar)0xff;
int y10 = (y2[i] - 16) * 298;
row2[0+R] = saturate_cast<uchar>((y10 + buv) >> 8);
row2[1] = saturate_cast<uchar>((y10 + guv) >> 8);
row2[2-R] = saturate_cast<uchar>((y10 + ruv) >> 8);
row2[3] = (uchar)0xff;
int y11 = (y2[i+1] - 16) * 298;
row2[4+R] = saturate_cast<uchar>((y11 + buv) >> 8);
row2[5] = saturate_cast<uchar>((y11 + guv) >> 8);
row2[6-R] = saturate_cast<uchar>((y11 + ruv) >> 8);
row2[7] = (uchar)0xff;
}
}
}
};
}//namespace cv
//////////////////////////////////////////////////////////////////////////////////////////
// The main function //
// The main function //
//////////////////////////////////////////////////////////////////////////////////////////
void cv::cvtColor( const InputArray& _src, OutputArray _dst, int code, int dcn )
@ -2992,6 +3112,40 @@ void cv::cvtColor( const InputArray& _src, OutputArray _dst, int code, int dcn )
Bayer2RGB_VNG_8u(src, dst, code);
}
break;
case CV_YUV420i2BGR: case CV_YUV420i2RGB:
{
if(dcn <= 0) dcn = 3;
CV_Assert( dcn == 3 || dcn == 4 );
CV_Assert( sz.width % 2 == 0 && sz.height % 3 == 0 && depth == CV_8U && src.isContinuous() );
Size dstSz(sz.width, sz.height * 2 / 3);
_dst.create( dstSz, CV_MAKETYPE(depth, dcn));
dst = _dst.getMat();
const uchar* y = src.ptr();
const uchar* uv = y + dstSz.area();
#ifdef HAVE_TEGRA_OPTIMIZATION
if (!tegra::YUV420i2BGR(y, uv, dst, CV_YUV420i2RGB == code))
#endif
{
if (CV_YUV420i2RGB == code)
{
if (dcn == 3)
parallel_for(BlockedRange(0, dstSz.height, 2), YUV420i2BGR888Invoker<2>(dst, dstSz.width, y, uv));
else
parallel_for(BlockedRange(0, dstSz.height, 2), YUV420i2BGRA8888Invoker<2>(dst, dstSz.width, y, uv));
}
else
{
if (dcn == 3)
parallel_for(BlockedRange(0, dstSz.height, 2), YUV420i2BGR888Invoker<0>(dst, dstSz.width, y, uv));
else
parallel_for(BlockedRange(0, dstSz.height, 2), YUV420i2BGRA8888Invoker<0>(dst, dstSz.width, y, uv));
}
}
}
break;
default:
CV_Error( CV_StsBadFlag, "Unknown/unsupported color conversion code" );
}
@ -3009,5 +3163,3 @@ cvCvtColor( const CvArr* srcarr, CvArr* dstarr, int code )
/* End of file. */

1
modules/imgproc/src/thresh.cpp
View File

@ -60,7 +60,6 @@ thresh_8u( const Mat& _src, Mat& _dst, uchar thresh, uchar maxval, int type )
}
#ifdef HAVE_TEGRA_OPTIMIZATION
#warning TEGRA OPTIMIZED THRESHOLD IS USED
switch( type )
{
case THRESH_BINARY:

3
modules/video/src/lkpyramid.cpp
View File

@ -51,9 +51,8 @@ void cv::calcOpticalFlowPyrLK( const InputArray& _prevImg, const InputArray& _ne
int flags )
{
#ifdef HAVE_TEGRA_OPTIMIZATION
#warning "TEGRA OPTIMIZED calcOpticalFlowPyrLK IS USED"
if (tegra::calcOpticalFlowPyrLK(_prevImg, _nextImg, _prevPts, _nextPts, _status, _err, winSize, maxLevel, criteria, derivLambda, flags))
return;
return;
#endif
Mat prevImg = _prevImg.getMat(), nextImg = _nextImg.getMat(), prevPtsMat = _prevPts.getMat();
derivLambda = std::min(std::max(derivLambda, 0.), 1.);