mirror of
https://github.com/mozilla/gecko-dev.git
synced 2024-10-30 21:55:31 +00:00
2082 lines
72 KiB
C++
2082 lines
72 KiB
C++
/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
|
|
/* vim:set ts=2 sw=2 sts=2 et cindent: */
|
|
/* This Source Code Form is subject to the terms of the Mozilla Public
|
|
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
|
|
|
#include "ImageBitmapColorUtils.h"
|
|
#include "libyuv.h"
|
|
|
|
namespace mozilla {
|
|
namespace dom {
|
|
|
|
/*
|
|
* Utility function form libyuv source files.
|
|
*/
|
|
static __inline int32 clamp0(int32 v) {
|
|
return ((-(v) >> 31) & (v));
|
|
}
|
|
|
|
static __inline int32 clamp255(int32 v) {
|
|
return (((255 - (v)) >> 31) | (v)) & 255;
|
|
}
|
|
|
|
static __inline uint32 Clamp(int32 val) {
|
|
int v = clamp0(val);
|
|
return (uint32)(clamp255(v));
|
|
}
|
|
|
|
#define YG 74 /* (int8)(1.164 * 64 + 0.5) */
|
|
|
|
#define UB 127 /* min(63,(int8)(2.018 * 64)) */
|
|
#define UG -25 /* (int8)(-0.391 * 64 - 0.5) */
|
|
#define UR 0
|
|
|
|
#define VB 0
|
|
#define VG -52 /* (int8)(-0.813 * 64 - 0.5) */
|
|
#define VR 102 /* (int8)(1.596 * 64 + 0.5) */
|
|
|
|
// Bias
|
|
#define BB UB * 128 + VB * 128
|
|
#define BG UG * 128 + VG * 128
|
|
#define BR UR * 128 + VR * 128
|
|
|
|
static __inline void
|
|
YuvPixel(uint8 y, uint8 u, uint8 v, uint8* b, uint8* g, uint8* r)
|
|
{
|
|
int32 y1 = ((int32)(y) - 16) * YG;
|
|
*b = Clamp((int32)((u * UB + v * VB) - (BB) + y1) >> 6);
|
|
*g = Clamp((int32)((u * UG + v * VG) - (BG) + y1) >> 6);
|
|
*r = Clamp((int32)((u * UR + v * VR) - (BR) + y1) >> 6);
|
|
}
|
|
|
|
static __inline int
|
|
RGBToY(uint8 r, uint8 g, uint8 b)
|
|
{
|
|
return (66 * r + 129 * g + 25 * b + 0x1080) >> 8;
|
|
}
|
|
|
|
static __inline int
|
|
RGBToU(uint8 r, uint8 g, uint8 b)
|
|
{
|
|
return (112 * b - 74 * g - 38 * r + 0x8080) >> 8;
|
|
}
|
|
|
|
static __inline int
|
|
RGBToV(uint8 r, uint8 g, uint8 b)
|
|
{
|
|
return (112 * r - 94 * g - 18 * b + 0x8080) >> 8;
|
|
}
|
|
|
|
/*
|
|
* Generic functions.
|
|
*/
|
|
template<int aSrcRIndex, int aSrcGIndex, int aSrcBIndex,
|
|
int aDstRIndex, int aDstGIndex, int aDstBIndex, int aDstAIndex>
|
|
static int
|
|
RGBFamilyToRGBAFamily(const uint8_t* aSrcBuffer, int aSrcStride,
|
|
uint8_t* aDstBuffer, int aDstStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
static_assert(aSrcRIndex == 0 || aSrcRIndex == 2, "Wrong SrcR index.");
|
|
static_assert(aSrcGIndex == 1, "Wrong SrcG index.");
|
|
static_assert(aSrcBIndex == 0 || aSrcBIndex == 2, "Wrong SrcB index.");
|
|
static_assert(aDstRIndex == 0 || aDstRIndex == 2, "Wrong DstR index.");
|
|
static_assert(aDstGIndex == 1, "Wrong DstG index.");
|
|
static_assert(aDstBIndex == 0 || aDstBIndex == 2, "Wrong DstB index.");
|
|
static_assert(aDstAIndex == 3, "Wrong DstA index.");
|
|
|
|
for (int i = 0; i < aHeight; ++i) {
|
|
const uint8_t* srcBuffer = aSrcBuffer + aSrcStride * i;
|
|
uint8_t* dstBuffer = aDstBuffer + aDstStride * i;
|
|
|
|
for (int j = 0; j < aWidth; ++j) {
|
|
uint8_t r = *(srcBuffer + aSrcRIndex);
|
|
uint8_t g = *(srcBuffer + aSrcGIndex);
|
|
uint8_t b = *(srcBuffer + aSrcBIndex);
|
|
*(dstBuffer + aDstRIndex) = r;
|
|
*(dstBuffer + aDstGIndex) = g;
|
|
*(dstBuffer + aDstBIndex) = b;
|
|
*(dstBuffer + aDstAIndex) = 255;
|
|
srcBuffer += 3;
|
|
dstBuffer += 4;
|
|
}
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
template<int aSrcRIndex, int aSrcGIndex, int aSrcBIndex,
|
|
int aDstRIndex, int aDstGIndex, int aDstBIndex>
|
|
static int
|
|
RGBAFamilyToRGBFamily(const uint8_t* aSrcBuffer, int aSrcStride,
|
|
uint8_t* aDstBuffer, int aDstStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
static_assert(aSrcRIndex == 0 || aSrcRIndex == 2, "Wrong SrcR index.");
|
|
static_assert(aSrcGIndex == 1, "Wrong SrcG index.");
|
|
static_assert(aSrcBIndex == 0 || aSrcBIndex == 2, "Wrong SrcB index.");
|
|
static_assert(aDstRIndex == 0 || aDstRIndex == 2, "Wrong DstR index.");
|
|
static_assert(aDstGIndex == 1, "Wrong DstG index.");
|
|
static_assert(aDstBIndex == 0 || aDstBIndex == 2, "Wrong DstB index.");
|
|
|
|
for (int i = 0; i < aHeight; ++i) {
|
|
const uint8_t* srcBuffer = aSrcBuffer + aSrcStride * i;
|
|
uint8_t* dstBuffer = aDstBuffer + aDstStride * i;
|
|
|
|
for (int j = 0; j < aWidth; ++j) {
|
|
uint8_t r = *(srcBuffer + aSrcRIndex);
|
|
uint8_t g = *(srcBuffer + aSrcGIndex);
|
|
uint8_t b = *(srcBuffer + aSrcBIndex);
|
|
*(dstBuffer + aDstRIndex) = r;
|
|
*(dstBuffer + aDstGIndex) = g;
|
|
*(dstBuffer + aDstBIndex) = b;
|
|
srcBuffer += 4;
|
|
dstBuffer += 3;
|
|
}
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
template<int aPixel1YOffset, int aPixel1UOffset, int aPixel1VOffset,
|
|
int aPixel2YOffset, int aPixel2UOffset, int aPixel2VOffset,
|
|
int aYStep, int aUStep, int aVStep,
|
|
int aRIndex, int aGIndex, int aBIndex>
|
|
void
|
|
YUVFamilyToRGBFamily_Row(const uint8_t* aYBuffer,
|
|
const uint8_t* aUBuffer,
|
|
const uint8_t* aVBuffer,
|
|
uint8_t* aDstBuffer,
|
|
int aWidth)
|
|
{
|
|
static_assert(aRIndex == 0 || aRIndex == 2, "Wrong R index.");
|
|
static_assert(aGIndex == 1, "Wrong G index.");
|
|
static_assert(aBIndex == 0 || aBIndex == 2, "Wrong B index.");
|
|
|
|
for (int j = 0; j < aWidth - 1; j += 2) {
|
|
YuvPixel(aYBuffer[aPixel1YOffset], aUBuffer[aPixel1UOffset], aVBuffer[aPixel1VOffset],
|
|
aDstBuffer + aBIndex, aDstBuffer + aGIndex, aDstBuffer + aRIndex);
|
|
YuvPixel(aYBuffer[aPixel2YOffset], aUBuffer[aPixel2UOffset], aVBuffer[aPixel2VOffset],
|
|
aDstBuffer + aBIndex + 3, aDstBuffer + aGIndex + 3, aDstBuffer + aRIndex + 3);
|
|
aYBuffer += aYStep;
|
|
aUBuffer += aUStep;
|
|
aVBuffer += aVStep;
|
|
aDstBuffer += 6;
|
|
}
|
|
|
|
if (aWidth & 1) {
|
|
YuvPixel(aYBuffer[aPixel1YOffset], aUBuffer[aPixel1UOffset], aVBuffer[aPixel1VOffset],
|
|
aDstBuffer + aBIndex, aDstBuffer + aGIndex, aDstBuffer + aRIndex);
|
|
}
|
|
}
|
|
|
|
template<int aPixel1YOffset, int aPixel1UOffset, int aPixel1VOffset,
|
|
int aPixel2YOffset, int aPixel2UOffset, int aPixel2VOffset,
|
|
int aYStep, int aUStep, int aVStep,
|
|
int aRIndex, int aGIndex, int aBIndex, int aAIndex>
|
|
void
|
|
YUVFamilyToRGBAFamily_Row(const uint8_t* aYBuffer,
|
|
const uint8_t* aUBuffer,
|
|
const uint8_t* aVBuffer,
|
|
uint8_t* aDstBuffer,
|
|
int aWidth)
|
|
{
|
|
static_assert(aRIndex == 0 || aRIndex == 2, "Wrong R index.");
|
|
static_assert(aGIndex == 1, "Wrong G index.");
|
|
static_assert(aBIndex == 0 || aBIndex == 2, "Wrong B index.");
|
|
static_assert(aAIndex == 3, "Wrong A index.");
|
|
|
|
for (int j = 0; j < aWidth - 1; j += 2) {
|
|
YuvPixel(aYBuffer[aPixel1YOffset], aUBuffer[aPixel1UOffset], aVBuffer[aPixel1VOffset],
|
|
aDstBuffer + aBIndex, aDstBuffer + aGIndex, aDstBuffer + aRIndex);
|
|
YuvPixel(aYBuffer[aPixel2YOffset], aUBuffer[aPixel2UOffset], aVBuffer[aPixel2VOffset],
|
|
aDstBuffer + aBIndex + 4, aDstBuffer + aGIndex + 4, aDstBuffer + aRIndex + 4);
|
|
aDstBuffer[aAIndex] = 255;
|
|
aDstBuffer[aAIndex + 4] = 255;
|
|
|
|
aYBuffer += aYStep;
|
|
aUBuffer += aUStep;
|
|
aVBuffer += aVStep;
|
|
aDstBuffer += 8;
|
|
}
|
|
|
|
if (aWidth & 1) {
|
|
YuvPixel(aYBuffer[aPixel1YOffset], aUBuffer[aPixel1UOffset], aVBuffer[aPixel1VOffset],
|
|
aDstBuffer + aBIndex, aDstBuffer + aGIndex, aDstBuffer + aRIndex);
|
|
aDstBuffer[aAIndex] = 255;
|
|
}
|
|
}
|
|
|
|
template< int aRIndex, int aGIndex, int aBIndex>
|
|
static void
|
|
RGBFamilyToY_Row(const uint8_t* aSrcBuffer, uint8_t* aYBuffer, int aWidth)
|
|
{
|
|
static_assert(aRIndex == 0 || aRIndex == 2, "Wrong R index.");
|
|
static_assert(aGIndex == 1, "Wrong G index.");
|
|
static_assert(aBIndex == 0 || aBIndex == 2, "Wrong B index.");
|
|
|
|
for (int j = 0; j < aWidth - 1; j += 2) {
|
|
aYBuffer[0] = RGBToY(aSrcBuffer[aRIndex], aSrcBuffer[aGIndex], aSrcBuffer[aBIndex]);
|
|
aYBuffer[1] = RGBToY(aSrcBuffer[aRIndex + 3], aSrcBuffer[aGIndex + 3], aSrcBuffer[aBIndex + 3]);
|
|
|
|
aYBuffer += 2;
|
|
aSrcBuffer += 3 * 2;
|
|
}
|
|
|
|
if (aWidth & 1) {
|
|
aYBuffer[0] = RGBToY(aSrcBuffer[aRIndex], aSrcBuffer[aGIndex], aSrcBuffer[aBIndex]);
|
|
}
|
|
}
|
|
|
|
template< int aRIndex, int aGIndex, int aBIndex, int aUStep, int aVStep>
|
|
static void
|
|
RGBFamilyToUV_Row(const uint8_t* aSrcBuffer, int aSrcStride,
|
|
uint8_t* aUBuffer, uint8_t* aVBuffer, int aWidth)
|
|
{
|
|
static_assert(aRIndex == 0 || aRIndex == 2, "Wrong R index.");
|
|
static_assert(aGIndex == 1, "Wrong G index.");
|
|
static_assert(aBIndex == 0 || aBIndex == 2, "Wrong B index.");
|
|
|
|
uint8_t averageR = 0;
|
|
uint8_t averageG = 0;
|
|
uint8_t averageB = 0;
|
|
|
|
const uint8_t* aSrcBufferNextRow = aSrcBuffer + aSrcStride;
|
|
for (int j = 0; j < aWidth - 1; j += 2) {
|
|
averageR = (aSrcBuffer[aRIndex] + aSrcBuffer[aRIndex + 3] + aSrcBufferNextRow[aRIndex] + aSrcBufferNextRow[aRIndex + 3]) >> 2;
|
|
averageG = (aSrcBuffer[aGIndex] + aSrcBuffer[aGIndex + 3] + aSrcBufferNextRow[aGIndex] + aSrcBufferNextRow[aGIndex + 3]) >> 2;
|
|
averageB = (aSrcBuffer[aBIndex] + aSrcBuffer[aBIndex + 3] + aSrcBufferNextRow[aBIndex] + aSrcBufferNextRow[aBIndex + 3]) >> 2;
|
|
|
|
aUBuffer[0] = RGBToU(averageR, averageG, averageB);
|
|
aVBuffer[0] = RGBToV(averageR, averageG, averageB);
|
|
|
|
aUBuffer += aUStep;
|
|
aVBuffer += aVStep;
|
|
aSrcBuffer += 3 * 2;
|
|
aSrcBufferNextRow += 3 * 2;
|
|
}
|
|
|
|
if (aWidth & 1) {
|
|
averageR = (aSrcBuffer[aRIndex] + aSrcBufferNextRow[aRIndex]) >> 1;
|
|
averageG = (aSrcBuffer[aGIndex] + aSrcBufferNextRow[aGIndex]) >> 1;
|
|
averageB = (aSrcBuffer[aBIndex] + aSrcBufferNextRow[aBIndex]) >> 1;
|
|
|
|
aUBuffer[0] = RGBToU(averageR, averageG, averageB);
|
|
aVBuffer[0] = RGBToV(averageR, averageG, averageB);
|
|
}
|
|
}
|
|
|
|
template< int aRIndex, int aGIndex, int aBIndex>
|
|
static void
|
|
RGBAFamilyToY_Row(const uint8_t* aSrcBuffer, uint8_t* aYBuffer, int aWidth)
|
|
{
|
|
static_assert(aRIndex == 0 || aRIndex == 2, "Wrong R index.");
|
|
static_assert(aGIndex == 1, "Wrong G index.");
|
|
static_assert(aBIndex == 0 || aBIndex == 2, "Wrong B index.");
|
|
|
|
for (int j = 0; j < aWidth - 1; j += 2) {
|
|
aYBuffer[0] = RGBToY(aSrcBuffer[aRIndex], aSrcBuffer[aGIndex], aSrcBuffer[aBIndex]);
|
|
aYBuffer[1] = RGBToY(aSrcBuffer[aRIndex + 4], aSrcBuffer[aGIndex + 4], aSrcBuffer[aBIndex + 4]);
|
|
|
|
aYBuffer += 2;
|
|
aSrcBuffer += 4 * 2;
|
|
}
|
|
|
|
if (aWidth & 1) {
|
|
aYBuffer[0] = RGBToY(aSrcBuffer[aRIndex], aSrcBuffer[aGIndex], aSrcBuffer[aBIndex]);
|
|
}
|
|
}
|
|
|
|
template< int aRIndex, int aGIndex, int aBIndex, int aUStep, int aVStep>
|
|
static void
|
|
RGBAFamilyToUV_Row(const uint8_t* aSrcBuffer, int aSrcStride,
|
|
uint8_t* aUBuffer, uint8_t* aVBuffer, int aWidth)
|
|
{
|
|
static_assert(aRIndex == 0 || aRIndex == 2, "Wrong R index.");
|
|
static_assert(aGIndex == 1, "Wrong G index.");
|
|
static_assert(aBIndex == 0 || aBIndex == 2, "Wrong B index.");
|
|
|
|
uint8_t averageR = 0;
|
|
uint8_t averageG = 0;
|
|
uint8_t averageB = 0;
|
|
|
|
const uint8_t* aSrcBufferNextRow = aSrcBuffer + aSrcStride;
|
|
for (int j = 0; j < aWidth - 1; j += 2) {
|
|
averageR = (aSrcBuffer[aRIndex] + aSrcBuffer[aRIndex + 4] + aSrcBufferNextRow[aRIndex] + aSrcBufferNextRow[aRIndex + 4]) >> 2;
|
|
averageG = (aSrcBuffer[aGIndex] + aSrcBuffer[aGIndex + 4] + aSrcBufferNextRow[aGIndex] + aSrcBufferNextRow[aGIndex + 4]) >> 2;
|
|
averageB = (aSrcBuffer[aBIndex] + aSrcBuffer[aBIndex + 4] + aSrcBufferNextRow[aBIndex] + aSrcBufferNextRow[aBIndex + 4]) >> 2;
|
|
|
|
aUBuffer[0] = RGBToU(averageR, averageG, averageB);
|
|
aVBuffer[0] = RGBToV(averageR, averageG, averageB);
|
|
|
|
aUBuffer += aUStep;
|
|
aVBuffer += aVStep;
|
|
aSrcBuffer += 4 * 2;
|
|
aSrcBufferNextRow += 4 * 2;
|
|
}
|
|
|
|
if (aWidth & 1) {
|
|
averageR = (aSrcBuffer[aRIndex] + aSrcBufferNextRow[aRIndex]) >> 1;
|
|
averageG = (aSrcBuffer[aGIndex] + aSrcBufferNextRow[aGIndex]) >> 1;
|
|
averageB = (aSrcBuffer[aBIndex] + aSrcBufferNextRow[aBIndex]) >> 1;
|
|
|
|
aUBuffer[0] = RGBToU(averageR, averageG, averageB);
|
|
aVBuffer[0] = RGBToV(averageR, averageG, averageB);
|
|
}
|
|
}
|
|
|
|
/*
|
|
* RGB family -> RGBA family.
|
|
*/
|
|
int
|
|
RGB24ToRGBA32(const uint8_t* aSrcBuffer, int aSrcStride,
|
|
uint8_t* aDstBuffer, int aDstStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
return RGBFamilyToRGBAFamily<0, 1, 2, 0, 1, 2, 3>(aSrcBuffer, aSrcStride,
|
|
aDstBuffer, aDstStride,
|
|
aWidth, aHeight);
|
|
}
|
|
|
|
int
|
|
BGR24ToRGBA32(const uint8_t* aSrcBuffer, int aSrcStride,
|
|
uint8_t* aDstBuffer, int aDstStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
return RGBFamilyToRGBAFamily<2, 1, 0, 0, 1, 2, 3>(aSrcBuffer, aSrcStride,
|
|
aDstBuffer, aDstStride,
|
|
aWidth, aHeight);
|
|
}
|
|
|
|
int
|
|
RGB24ToBGRA32(const uint8_t* aSrcBuffer, int aSrcStride,
|
|
uint8_t* aDstBuffer, int aDstStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
return RGBFamilyToRGBAFamily<0, 1, 2, 2, 1, 0, 3>(aSrcBuffer, aSrcStride,
|
|
aDstBuffer, aDstStride,
|
|
aWidth, aHeight);
|
|
}
|
|
|
|
int
|
|
BGR24ToBGRA32(const uint8_t* aSrcBuffer, int aSrcStride,
|
|
uint8_t* aDstBuffer, int aDstStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
return RGBFamilyToRGBAFamily<2, 1, 0, 2, 1, 0, 3>(aSrcBuffer, aSrcStride,
|
|
aDstBuffer, aDstStride,
|
|
aWidth, aHeight);
|
|
}
|
|
|
|
/*
|
|
* RGBA family -> RGB family.
|
|
*/
|
|
int
|
|
RGBA32ToRGB24(const uint8_t* aSrcBuffer, int aSrcStride,
|
|
uint8_t* aDstBuffer, int aDstStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
return RGBAFamilyToRGBFamily<0, 1, 2, 0, 1, 2>(aSrcBuffer, aSrcStride,
|
|
aDstBuffer, aDstStride,
|
|
aWidth, aHeight);
|
|
}
|
|
|
|
int
|
|
BGRA32ToRGB24(const uint8_t* aSrcBuffer, int aSrcStride,
|
|
uint8_t* aDstBuffer, int aDstStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
return RGBAFamilyToRGBFamily<2, 1, 0, 0, 1, 2>(aSrcBuffer, aSrcStride,
|
|
aDstBuffer, aDstStride,
|
|
aWidth, aHeight);
|
|
}
|
|
|
|
int
|
|
RGBA32ToBGR24(const uint8_t* aSrcBuffer, int aSrcStride,
|
|
uint8_t* aDstBuffer, int aDstStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
return RGBAFamilyToRGBFamily<0, 1, 2, 2, 1, 0>(aSrcBuffer, aSrcStride,
|
|
aDstBuffer, aDstStride,
|
|
aWidth, aHeight);
|
|
}
|
|
|
|
int
|
|
BGRA32ToBGR24(const uint8_t* aSrcBuffer, int aSrcStride,
|
|
uint8_t* aDstBuffer, int aDstStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
return RGBAFamilyToRGBFamily<2, 1, 0, 2, 1, 0>(aSrcBuffer, aSrcStride,
|
|
aDstBuffer, aDstStride,
|
|
aWidth, aHeight);
|
|
}
|
|
|
|
/*
|
|
* Among RGB family.
|
|
*/
|
|
int
|
|
RGB24Copy(const uint8_t* aSrcBuffer, int aSrcStride,
|
|
uint8_t* aDstBuffer, int aDstStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
MOZ_ASSERT(aSrcStride == aDstStride, "RGB24Copy: aSrcStride != aDstStride");
|
|
|
|
const uint32_t length = aHeight * aDstStride;
|
|
memcpy(aDstBuffer, aSrcBuffer, length);
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
RGB24ToBGR24(const uint8_t* aSrcBuffer, int aSrcStride,
|
|
uint8_t* aDstBuffer, int aDstStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
for (int i = 0; i < aHeight; ++i) {
|
|
const uint8_t* srcBuffer = aSrcBuffer + aSrcStride * i;
|
|
uint8_t* dstBuffer = aDstBuffer + aDstStride * i;
|
|
|
|
for (int j = 0; j < aWidth; ++j) {
|
|
*(dstBuffer + 0) = *(srcBuffer + 2);
|
|
*(dstBuffer + 1) = *(srcBuffer + 1);
|
|
*(dstBuffer + 2) = *(srcBuffer + 0);
|
|
srcBuffer += 3;
|
|
dstBuffer += 3;
|
|
}
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
/*
|
|
* YUV family -> RGB family.
|
|
*/
|
|
int
|
|
YUV444PToRGB24(const uint8_t* aYBuffer, int aYStride,
|
|
const uint8_t* aUBuffer, int aUStride,
|
|
const uint8_t* aVBuffer, int aVStride,
|
|
uint8_t* aDstBuffer, int aDstStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
for (int i = 0; i < aHeight; ++i) {
|
|
const uint8_t* yBuffer = aYBuffer + aYStride * i;
|
|
const uint8_t* uBuffer = aUBuffer + aUStride * i;
|
|
const uint8_t* vBuffer = aVBuffer + aVStride * i;
|
|
uint8_t* dstBuffer = aDstBuffer + aDstStride * i;
|
|
|
|
YUVFamilyToRGBFamily_Row<0, 0, 0, 1, 1, 1, 2, 2, 2, 0, 1, 2>(yBuffer,
|
|
uBuffer,
|
|
vBuffer,
|
|
dstBuffer,
|
|
aWidth);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
YUV422PToRGB24(const uint8_t* aYBuffer, int aYStride,
|
|
const uint8_t* aUBuffer, int aUStride,
|
|
const uint8_t* aVBuffer, int aVStride,
|
|
uint8_t* aDstBuffer, int aDstStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
for (int i = 0; i < aHeight; ++i) {
|
|
const uint8_t* yBuffer = aYBuffer + aYStride * i;
|
|
const uint8_t* uBuffer = aUBuffer + aUStride * i;
|
|
const uint8_t* vBuffer = aVBuffer + aVStride * i;
|
|
uint8_t* dstBuffer = aDstBuffer + aDstStride * i;
|
|
|
|
YUVFamilyToRGBFamily_Row<0, 0, 0, 1, 0, 0, 2, 1, 1, 0, 1, 2>(yBuffer,
|
|
uBuffer,
|
|
vBuffer,
|
|
dstBuffer,
|
|
aWidth);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
YUV420PToRGB24(const uint8_t* aYBuffer, int aYStride,
|
|
const uint8_t* aUBuffer, int aUStride,
|
|
const uint8_t* aVBuffer, int aVStride,
|
|
uint8_t* aDstBuffer, int aDstStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
for (int i = 0; i < aHeight; ++i) {
|
|
const uint8_t* yBuffer = aYBuffer + aYStride * i;
|
|
const uint8_t* uBuffer = aUBuffer + aUStride * (i / 2);
|
|
const uint8_t* vBuffer = aVBuffer + aVStride * (i / 2);
|
|
uint8_t* dstBuffer = aDstBuffer + aDstStride * i;
|
|
|
|
YUVFamilyToRGBFamily_Row<0, 0, 0, 1, 0, 0, 2, 1, 1, 0, 1, 2>(yBuffer,
|
|
uBuffer,
|
|
vBuffer,
|
|
dstBuffer,
|
|
aWidth);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
NV12ToRGB24(const uint8_t* aYBuffer, int aYStride,
|
|
const uint8_t* aUVBuffer, int aUVStride,
|
|
uint8_t* aDstBuffer, int aDstStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
for (int i = 0; i < aHeight; ++i) {
|
|
const uint8_t* yBuffer = aYBuffer + aYStride * i;
|
|
const uint8_t* uBuffer = aUVBuffer + aUVStride * (i / 2);
|
|
const uint8_t* vBuffer = aUVBuffer + aUVStride * (i / 2) + 1;
|
|
uint8_t* dstBuffer = aDstBuffer + aDstStride * i;
|
|
|
|
YUVFamilyToRGBFamily_Row<0, 0, 0, 1, 0, 0, 2, 2, 2, 0, 1, 2>(yBuffer,
|
|
uBuffer,
|
|
vBuffer,
|
|
dstBuffer,
|
|
aWidth);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
NV21ToRGB24(const uint8_t* aYBuffer, int aYStride,
|
|
const uint8_t* aVUBuffer, int aVUStride,
|
|
uint8_t* aDstBuffer, int aDstStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
for (int i = 0; i < aHeight; ++i) {
|
|
const uint8_t* yBuffer = aYBuffer + aYStride * i;
|
|
const uint8_t* uBuffer = aVUBuffer + aVUStride * (i / 2) + 1;
|
|
const uint8_t* vBuffer = aVUBuffer + aVUStride * (i / 2);
|
|
uint8_t* dstBuffer = aDstBuffer + aDstStride * i;
|
|
|
|
YUVFamilyToRGBFamily_Row<0, 0, 0, 1, 0, 0, 2, 2, 2, 0, 1, 2>(yBuffer,
|
|
uBuffer,
|
|
vBuffer,
|
|
dstBuffer,
|
|
aWidth);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
YUV444PToBGR24(const uint8_t* aYBuffer, int aYStride,
|
|
const uint8_t* aUBuffer, int aUStride,
|
|
const uint8_t* aVBuffer, int aVStride,
|
|
uint8_t* aDstBuffer, int aDstStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
for (int i = 0; i < aHeight; ++i) {
|
|
const uint8_t* yBuffer = aYBuffer + aYStride * i;
|
|
const uint8_t* uBuffer = aUBuffer + aUStride * i;
|
|
const uint8_t* vBuffer = aVBuffer + aVStride * i;
|
|
uint8_t* dstBuffer = aDstBuffer + aDstStride * i;
|
|
|
|
YUVFamilyToRGBFamily_Row<0, 0, 0, 1, 1, 1, 2, 2, 2, 2, 1, 0>(yBuffer,
|
|
uBuffer,
|
|
vBuffer,
|
|
dstBuffer,
|
|
aWidth);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
YUV422PToBGR24(const uint8_t* aYBuffer, int aYStride,
|
|
const uint8_t* aUBuffer, int aUStride,
|
|
const uint8_t* aVBuffer, int aVStride,
|
|
uint8_t* aDstBuffer, int aDstStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
for (int i = 0; i < aHeight; ++i) {
|
|
const uint8_t* yBuffer = aYBuffer + aYStride * i;
|
|
const uint8_t* uBuffer = aUBuffer + aUStride * i;
|
|
const uint8_t* vBuffer = aVBuffer + aVStride * i;
|
|
uint8_t* dstBuffer = aDstBuffer + aDstStride * i;
|
|
|
|
YUVFamilyToRGBFamily_Row<0, 0, 0, 1, 0, 0, 2, 1, 1, 2, 1, 0>(yBuffer,
|
|
uBuffer,
|
|
vBuffer,
|
|
dstBuffer,
|
|
aWidth);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
YUV420PToBGR24(const uint8_t* aYBuffer, int aYStride,
|
|
const uint8_t* aUBuffer, int aUStride,
|
|
const uint8_t* aVBuffer, int aVStride,
|
|
uint8_t* aDstBuffer, int aDstStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
for (int i = 0; i < aHeight; ++i) {
|
|
const uint8_t* yBuffer = aYBuffer + aYStride * i;
|
|
const uint8_t* uBuffer = aUBuffer + aUStride * (i / 2);
|
|
const uint8_t* vBuffer = aVBuffer + aVStride * (i / 2);
|
|
uint8_t* dstBuffer = aDstBuffer + aDstStride * i;
|
|
|
|
YUVFamilyToRGBFamily_Row<0, 0, 0, 1, 0, 0, 2, 1, 1, 2, 1, 0>(yBuffer,
|
|
uBuffer,
|
|
vBuffer,
|
|
dstBuffer,
|
|
aWidth);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
NV12ToBGR24(const uint8_t* aYBuffer, int aYStride,
|
|
const uint8_t* aUVBuffer, int aUVStride,
|
|
uint8_t* aDstBuffer, int aDstStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
for (int i = 0; i < aHeight; ++i) {
|
|
const uint8_t* yBuffer = aYBuffer + aYStride * i;
|
|
const uint8_t* uBuffer = aUVBuffer + aUVStride * (i / 2);
|
|
const uint8_t* vBuffer = aUVBuffer + aUVStride * (i / 2) + 1;
|
|
uint8_t* dstBuffer = aDstBuffer + aDstStride * i;
|
|
|
|
YUVFamilyToRGBFamily_Row<0, 0, 0, 1, 0, 0, 2, 2, 2, 2, 1, 0>(yBuffer,
|
|
uBuffer,
|
|
vBuffer,
|
|
dstBuffer,
|
|
aWidth);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
NV21ToBGR24(const uint8_t* aYBuffer, int aYStride,
|
|
const uint8_t* aVUBuffer, int aVUStride,
|
|
uint8_t* aDstBuffer, int aDstStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
for (int i = 0; i < aHeight; ++i) {
|
|
const uint8_t* yBuffer = aYBuffer + aYStride * i;
|
|
const uint8_t* uBuffer = aVUBuffer + aVUStride * (i / 2) + 1;
|
|
const uint8_t* vBuffer = aVUBuffer + aVUStride * (i / 2);
|
|
uint8_t* dstBuffer = aDstBuffer + aDstStride * i;
|
|
|
|
YUVFamilyToRGBFamily_Row<0, 0, 0, 1, 0, 0, 2, 2, 2, 2, 1, 0>(yBuffer,
|
|
uBuffer,
|
|
vBuffer,
|
|
dstBuffer,
|
|
aWidth);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
/*
|
|
* YUV family -> RGBA family.
|
|
*/
|
|
int
|
|
YUV444PToRGBA32(const uint8_t* aYBuffer, int aYStride,
|
|
const uint8_t* aUBuffer, int aUStride,
|
|
const uint8_t* aVBuffer, int aVStride,
|
|
uint8_t* aDstBuffer, int aDstStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
for (int i = 0; i < aHeight; ++i) {
|
|
const uint8_t* yBuffer = aYBuffer + aYStride * i;
|
|
const uint8_t* uBuffer = aUBuffer + aUStride * i;
|
|
const uint8_t* vBuffer = aVBuffer + aVStride * i;
|
|
uint8_t* dstBuffer = aDstBuffer + aDstStride * i;
|
|
|
|
YUVFamilyToRGBAFamily_Row<0, 0, 0, 1, 1, 1, 2, 2, 2, 0, 1, 2, 3>(yBuffer,
|
|
uBuffer,
|
|
vBuffer,
|
|
dstBuffer,
|
|
aWidth);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
YUV422PToRGBA32(const uint8_t* aYBuffer, int aYStride,
|
|
const uint8_t* aUBuffer, int aUStride,
|
|
const uint8_t* aVBuffer, int aVStride,
|
|
uint8_t* aDstBuffer, int aDstStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
for (int i = 0; i < aHeight; ++i) {
|
|
const uint8_t* yBuffer = aYBuffer + aYStride * i;
|
|
const uint8_t* uBuffer = aUBuffer + aUStride * i;
|
|
const uint8_t* vBuffer = aVBuffer + aVStride * i;
|
|
uint8_t* dstBuffer = aDstBuffer + aDstStride * i;
|
|
|
|
YUVFamilyToRGBAFamily_Row<0, 0, 0, 1, 0, 0, 2, 1, 1, 0, 1, 2, 3>(yBuffer,
|
|
uBuffer,
|
|
vBuffer,
|
|
dstBuffer,
|
|
aWidth);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
YUV420PToRGBA32(const uint8_t* aYBuffer, int aYStride,
|
|
const uint8_t* aUBuffer, int aUStride,
|
|
const uint8_t* aVBuffer, int aVStride,
|
|
uint8_t* aDstBuffer, int aDstStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
for (int i = 0; i < aHeight; ++i) {
|
|
const uint8_t* yBuffer = aYBuffer + aYStride * i;
|
|
const uint8_t* uBuffer = aUBuffer + aUStride * (i / 2);
|
|
const uint8_t* vBuffer = aVBuffer + aVStride * (i / 2);
|
|
uint8_t* dstBuffer = aDstBuffer + aDstStride * i;
|
|
|
|
YUVFamilyToRGBAFamily_Row<0, 0, 0, 1, 0, 0, 2, 1, 1, 0, 1, 2, 3>(yBuffer,
|
|
uBuffer,
|
|
vBuffer,
|
|
dstBuffer,
|
|
aWidth);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
NV12ToRGBA32(const uint8_t* aYBuffer, int aYStride,
|
|
const uint8_t* aUVBuffer, int aUVStride,
|
|
uint8_t* aDstBuffer, int aDstStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
for (int i = 0; i < aHeight; ++i) {
|
|
const uint8_t* yBuffer = aYBuffer + aYStride * i;
|
|
const uint8_t* uBuffer = aUVBuffer + aUVStride * (i / 2);
|
|
const uint8_t* vBuffer = aUVBuffer + aUVStride * (i / 2) + 1;
|
|
uint8_t* dstBuffer = aDstBuffer + aDstStride * i;
|
|
|
|
YUVFamilyToRGBAFamily_Row<0, 0, 0, 1, 0, 0, 2, 2, 2, 0, 1, 2, 3>(yBuffer,
|
|
uBuffer,
|
|
vBuffer,
|
|
dstBuffer,
|
|
aWidth);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
NV21ToRGBA32(const uint8_t* aYBuffer, int aYStride,
|
|
const uint8_t* aVUBuffer, int aVUStride,
|
|
uint8_t* aDstBuffer, int aDstStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
for (int i = 0; i < aHeight; ++i) {
|
|
const uint8_t* yBuffer = aYBuffer + aYStride * i;
|
|
const uint8_t* uBuffer = aVUBuffer + aVUStride * (i / 2) + 1;
|
|
const uint8_t* vBuffer = aVUBuffer + aVUStride * (i / 2);
|
|
uint8_t* dstBuffer = aDstBuffer + aDstStride * i;
|
|
|
|
YUVFamilyToRGBAFamily_Row<0, 0, 0, 1, 0, 0, 2, 2, 2, 0, 1, 2, 3>(yBuffer,
|
|
uBuffer,
|
|
vBuffer,
|
|
dstBuffer,
|
|
aWidth);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
YUV444PToBGRA32(const uint8_t* aYBuffer, int aYStride,
|
|
const uint8_t* aUBuffer, int aUStride,
|
|
const uint8_t* aVBuffer, int aVStride,
|
|
uint8_t* aDstBuffer, int aDstStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
for (int i = 0; i < aHeight; ++i) {
|
|
const uint8_t* yBuffer = aYBuffer + aYStride * i;
|
|
const uint8_t* uBuffer = aUBuffer + aUStride * i;
|
|
const uint8_t* vBuffer = aVBuffer + aVStride * i;
|
|
uint8_t* dstBuffer = aDstBuffer + aDstStride * i;
|
|
|
|
YUVFamilyToRGBAFamily_Row<0, 0, 0, 1, 1, 1, 2, 2, 2, 2, 1, 0, 3>(yBuffer,
|
|
uBuffer,
|
|
vBuffer,
|
|
dstBuffer,
|
|
aWidth);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
YUV422PToBGRA32(const uint8_t* aYBuffer, int aYStride,
|
|
const uint8_t* aUBuffer, int aUStride,
|
|
const uint8_t* aVBuffer, int aVStride,
|
|
uint8_t* aDstBuffer, int aDstStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
for (int i = 0; i < aHeight; ++i) {
|
|
const uint8_t* yBuffer = aYBuffer + aYStride * i;
|
|
const uint8_t* uBuffer = aUBuffer + aUStride * i;
|
|
const uint8_t* vBuffer = aVBuffer + aVStride * i;
|
|
uint8_t* dstBuffer = aDstBuffer + aDstStride * i;
|
|
|
|
YUVFamilyToRGBAFamily_Row<0, 0, 0, 1, 0, 0, 2, 1, 1, 2, 1, 0, 3>(yBuffer,
|
|
uBuffer,
|
|
vBuffer,
|
|
dstBuffer,
|
|
aWidth);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
YUV420PToBGRA32(const uint8_t* aYBuffer, int aYStride,
|
|
const uint8_t* aUBuffer, int aUStride,
|
|
const uint8_t* aVBuffer, int aVStride,
|
|
uint8_t* aDstBuffer, int aDstStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
for (int i = 0; i < aHeight; ++i) {
|
|
const uint8_t* yBuffer = aYBuffer + aYStride * i;
|
|
const uint8_t* uBuffer = aUBuffer + aUStride * (i / 2);
|
|
const uint8_t* vBuffer = aVBuffer + aVStride * (i / 2);
|
|
uint8_t* dstBuffer = aDstBuffer + aDstStride * i;
|
|
|
|
YUVFamilyToRGBAFamily_Row<0, 0, 0, 1, 0, 0, 2, 1, 1, 2, 1, 0, 3>(yBuffer,
|
|
uBuffer,
|
|
vBuffer,
|
|
dstBuffer,
|
|
aWidth);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
NV12ToBGRA32(const uint8_t* aYBuffer, int aYStride,
|
|
const uint8_t* aUVBuffer, int aUVStride,
|
|
uint8_t* aDstBuffer, int aDstStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
for (int i = 0; i < aHeight; ++i) {
|
|
const uint8_t* yBuffer = aYBuffer + aYStride * i;
|
|
const uint8_t* uBuffer = aUVBuffer + aUVStride * (i / 2);
|
|
const uint8_t* vBuffer = aUVBuffer + aUVStride * (i / 2) + 1;
|
|
uint8_t* dstBuffer = aDstBuffer + aDstStride * i;
|
|
|
|
YUVFamilyToRGBAFamily_Row<0, 0, 0, 1, 0, 0, 2, 2, 2, 2, 1, 0, 3>(yBuffer,
|
|
uBuffer,
|
|
vBuffer,
|
|
dstBuffer,
|
|
aWidth);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
NV21ToBGRA32(const uint8_t* aYBuffer, int aYStride,
|
|
const uint8_t* aVUBuffer, int aVUStride,
|
|
uint8_t* aDstBuffer, int aDstStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
for (int i = 0; i < aHeight; ++i) {
|
|
const uint8_t* yBuffer = aYBuffer + aYStride * i;
|
|
const uint8_t* uBuffer = aVUBuffer + aVUStride * (i / 2) + 1;
|
|
const uint8_t* vBuffer = aVUBuffer + aVUStride * (i / 2);
|
|
uint8_t* dstBuffer = aDstBuffer + aDstStride * i;
|
|
|
|
YUVFamilyToRGBAFamily_Row<0, 0, 0, 1, 0, 0, 2, 2, 2, 2, 1, 0, 3>(yBuffer,
|
|
uBuffer,
|
|
vBuffer,
|
|
dstBuffer,
|
|
aWidth);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
/*
|
|
* RGB family -> YUV family.
|
|
*/
|
|
int
|
|
RGB24ToYUV444P(const uint8_t* aSrcBuffer, int aSrcStride,
|
|
uint8_t* aYBuffer, int aYStride,
|
|
uint8_t* aUBuffer, int aUStride,
|
|
uint8_t* aVBuffer, int aVStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
for (int i = 0; i < aHeight; ++i) {
|
|
const uint8_t* srcBuffer = aSrcBuffer + aSrcStride * i;
|
|
uint8_t* yBuffer = aYBuffer + aYStride * i;
|
|
uint8_t* uBuffer = aUBuffer + aUStride * i;
|
|
uint8_t* vBuffer = aVBuffer + aVStride * i;
|
|
|
|
for (int j = 0; j < aWidth; ++j) {
|
|
yBuffer[0] = RGBToY(srcBuffer[0], srcBuffer[1], srcBuffer[2]);
|
|
uBuffer[0] = RGBToU(srcBuffer[0], srcBuffer[1], srcBuffer[2]);
|
|
vBuffer[0] = RGBToV(srcBuffer[0], srcBuffer[1], srcBuffer[2]);
|
|
|
|
yBuffer += 1;
|
|
uBuffer += 1;
|
|
vBuffer += 1;
|
|
srcBuffer += 3;
|
|
}
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
RGB24ToYUV422P(const uint8_t* aSrcBuffer, int aSrcStride,
|
|
uint8_t* aYBuffer, int aYStride,
|
|
uint8_t* aUBuffer, int aUStride,
|
|
uint8_t* aVBuffer, int aVStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
for (int i = 0; i < aHeight; ++i) {
|
|
const uint8_t* srcBuffer = aSrcBuffer + aSrcStride * i;
|
|
uint8_t* yBuffer = aYBuffer + aYStride * i;
|
|
uint8_t* uBuffer = aUBuffer + aUStride * i;
|
|
uint8_t* vBuffer = aVBuffer + aVStride * i;
|
|
|
|
RGBFamilyToY_Row<0, 1, 2>(srcBuffer, yBuffer, aWidth);
|
|
|
|
// Pass 0 as the aSrcStride so we don't sample next row's RGB information.
|
|
RGBFamilyToUV_Row<0, 1, 2, 1, 1>(srcBuffer, 0, uBuffer, vBuffer, aWidth);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
RGB24ToYUV420P(const uint8_t* aSrcBuffer, int aSrcStride,
|
|
uint8_t* aYBuffer, int aYStride,
|
|
uint8_t* aUBuffer, int aUStride,
|
|
uint8_t* aVBuffer, int aVStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
for (int i = 0; i < aHeight - 1; i += 2) {
|
|
const uint8_t* srcBuffer = aSrcBuffer + aSrcStride * i;
|
|
uint8_t* yBuffer = aYBuffer + aYStride * i;
|
|
uint8_t* uBuffer = aUBuffer + aUStride * (i / 2);
|
|
uint8_t* vBuffer = aVBuffer + aVStride * (i / 2);
|
|
|
|
RGBFamilyToY_Row<0, 1, 2>(srcBuffer, yBuffer, aWidth);
|
|
RGBFamilyToY_Row<0, 1, 2>(srcBuffer + aSrcStride, yBuffer + aYStride, aWidth);
|
|
RGBFamilyToUV_Row<0, 1, 2, 1, 1>(srcBuffer, aSrcStride, uBuffer, vBuffer, aWidth);
|
|
}
|
|
|
|
if (aHeight & 1) {
|
|
const int i = aHeight - 1;
|
|
const uint8_t* srcBuffer = aSrcBuffer + aSrcStride * i;
|
|
uint8_t* yBuffer = aYBuffer + aYStride * i;
|
|
uint8_t* uBuffer = aUBuffer + aUStride * (i / 2);
|
|
uint8_t* vBuffer = aVBuffer + aVStride * (i / 2);
|
|
|
|
RGBFamilyToY_Row<0, 1, 2>(srcBuffer, yBuffer, aWidth);
|
|
|
|
// Pass 0 as the aSrcStride so we don't sample next row's RGB information.
|
|
RGBFamilyToUV_Row<0, 1, 2, 1, 1>(srcBuffer, 0, uBuffer, vBuffer, aWidth);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
RGB24ToNV12(const uint8_t* aSrcBuffer, int aSrcStride,
|
|
uint8_t* aYBuffer, int aYStride,
|
|
uint8_t* aUVBuffer, int aUVStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
for (int i = 0; i < aHeight - 1; i += 2) {
|
|
const uint8_t* srcBuffer = aSrcBuffer + aSrcStride * i;
|
|
uint8_t* yBuffer = aYBuffer + aYStride * i;
|
|
uint8_t* uBuffer = aUVBuffer + aUVStride * (i / 2);
|
|
uint8_t* vBuffer = aUVBuffer + aUVStride * (i / 2) + 1;
|
|
|
|
RGBFamilyToY_Row<0, 1, 2>(srcBuffer, yBuffer, aWidth);
|
|
RGBFamilyToY_Row<0, 1, 2>(srcBuffer + aSrcStride, yBuffer + aYStride, aWidth);
|
|
RGBFamilyToUV_Row<0, 1, 2, 2, 2>(srcBuffer, aSrcStride, uBuffer, vBuffer, aWidth);
|
|
}
|
|
|
|
if (aHeight & 1) {
|
|
const int i = aHeight - 1;
|
|
const uint8_t* srcBuffer = aSrcBuffer + aSrcStride * i;
|
|
uint8_t* yBuffer = aYBuffer + aYStride * i;
|
|
uint8_t* uBuffer = aUVBuffer + aUVStride * (i / 2);
|
|
uint8_t* vBuffer = aUVBuffer + aUVStride * (i / 2) + 1;
|
|
|
|
RGBFamilyToY_Row<0, 1, 2>(srcBuffer, yBuffer, aWidth);
|
|
|
|
// Pass 0 as the aSrcStride so we don't sample next row's RGB information.
|
|
RGBFamilyToUV_Row<0, 1, 2, 2, 2>(srcBuffer, 0, uBuffer, vBuffer, aWidth);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
RGB24ToNV21(const uint8_t* aSrcBuffer, int aSrcStride,
|
|
uint8_t* aYBuffer, int aYStride,
|
|
uint8_t* aVUBuffer, int aVUStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
for (int i = 0; i < aHeight - 1; i += 2) {
|
|
const uint8_t* srcBuffer = aSrcBuffer + aSrcStride * i;
|
|
uint8_t* yBuffer = aYBuffer + aYStride * i;
|
|
uint8_t* uBuffer = aVUBuffer + aVUStride * (i / 2) + 1;
|
|
uint8_t* vBuffer = aVUBuffer + aVUStride * (i / 2);
|
|
|
|
RGBFamilyToY_Row<0, 1, 2>(srcBuffer, yBuffer, aWidth);
|
|
RGBFamilyToY_Row<0, 1, 2>(srcBuffer + aSrcStride, yBuffer + aYStride, aWidth);
|
|
RGBFamilyToUV_Row<0, 1, 2, 2, 2>(srcBuffer, aSrcStride, uBuffer, vBuffer, aWidth);
|
|
}
|
|
|
|
if (aHeight & 1) {
|
|
const int i = aHeight - 1;
|
|
const uint8_t* srcBuffer = aSrcBuffer + aSrcStride * i;
|
|
uint8_t* yBuffer = aYBuffer + aYStride * i;
|
|
uint8_t* uBuffer = aVUBuffer + aVUStride * (i / 2) + 1;
|
|
uint8_t* vBuffer = aVUBuffer + aVUStride * (i / 2);
|
|
|
|
RGBFamilyToY_Row<0, 1, 2>(srcBuffer, yBuffer, aWidth);
|
|
|
|
// Pass 0 as the aSrcStride so we don't sample next row's RGB information.
|
|
RGBFamilyToUV_Row<0, 1, 2, 2, 2>(srcBuffer, 0, uBuffer, vBuffer, aWidth);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
BGR24ToYUV444P(const uint8_t* aSrcBuffer, int aSrcStride,
|
|
uint8_t* aYBuffer, int aYStride,
|
|
uint8_t* aUBuffer, int aUStride,
|
|
uint8_t* aVBuffer, int aVStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
for (int i = 0; i < aHeight; ++i) {
|
|
const uint8_t* srcBuffer = aSrcBuffer + aSrcStride * i;
|
|
uint8_t* yBuffer = aYBuffer + aYStride * i;
|
|
uint8_t* uBuffer = aUBuffer + aUStride * i;
|
|
uint8_t* vBuffer = aVBuffer + aVStride * i;
|
|
|
|
for (int j = 0; j < aWidth; ++j) {
|
|
yBuffer[0] = RGBToY(srcBuffer[2], srcBuffer[1], srcBuffer[0]);
|
|
uBuffer[0] = RGBToU(srcBuffer[2], srcBuffer[1], srcBuffer[0]);
|
|
vBuffer[0] = RGBToV(srcBuffer[2], srcBuffer[1], srcBuffer[0]);
|
|
|
|
yBuffer += 1;
|
|
uBuffer += 1;
|
|
vBuffer += 1;
|
|
srcBuffer += 3;
|
|
}
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
BGR24ToYUV422P(const uint8_t* aSrcBuffer, int aSrcStride,
|
|
uint8_t* aYBuffer, int aYStride,
|
|
uint8_t* aUBuffer, int aUStride,
|
|
uint8_t* aVBuffer, int aVStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
for (int i = 0; i < aHeight; ++i) {
|
|
const uint8_t* srcBuffer = aSrcBuffer + aSrcStride * i;
|
|
uint8_t* yBuffer = aYBuffer + aYStride * i;
|
|
uint8_t* uBuffer = aUBuffer + aUStride * i;
|
|
uint8_t* vBuffer = aVBuffer + aVStride * i;
|
|
|
|
RGBFamilyToY_Row<2, 1, 0>(srcBuffer, yBuffer, aWidth);
|
|
|
|
// Pass 0 as the aSrcStride so we don't sample next row's RGB information.
|
|
RGBFamilyToUV_Row<2, 1, 0, 1, 1>(srcBuffer, 0, uBuffer, vBuffer, aWidth);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
BGR24ToYUV420P(const uint8_t* aSrcBuffer, int aSrcStride,
|
|
uint8_t* aYBuffer, int aYStride,
|
|
uint8_t* aUBuffer, int aUStride,
|
|
uint8_t* aVBuffer, int aVStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
for (int i = 0; i < aHeight - 1; i += 2) {
|
|
const uint8_t* srcBuffer = aSrcBuffer + aSrcStride * i;
|
|
uint8_t* yBuffer = aYBuffer + aYStride * i;
|
|
uint8_t* uBuffer = aUBuffer + aUStride * (i / 2);
|
|
uint8_t* vBuffer = aVBuffer + aVStride * (i / 2);
|
|
|
|
RGBFamilyToY_Row<2, 1, 0>(srcBuffer, yBuffer, aWidth);
|
|
RGBFamilyToY_Row<2, 1, 0>(srcBuffer + aSrcStride, yBuffer + aYStride, aWidth);
|
|
RGBFamilyToUV_Row<2, 1, 0, 1, 1>(srcBuffer, aSrcStride, uBuffer, vBuffer, aWidth);
|
|
}
|
|
|
|
if (aHeight & 1) {
|
|
const int i = aHeight - 1;
|
|
const uint8_t* srcBuffer = aSrcBuffer + aSrcStride * i;
|
|
uint8_t* yBuffer = aYBuffer + aYStride * i;
|
|
uint8_t* uBuffer = aUBuffer + aUStride * (i / 2);
|
|
uint8_t* vBuffer = aVBuffer + aVStride * (i / 2);
|
|
|
|
RGBFamilyToY_Row<2, 1, 0>(srcBuffer, yBuffer, aWidth);
|
|
|
|
// Pass 0 as the aSrcStride so we don't sample next row's RGB information.
|
|
RGBFamilyToUV_Row<2, 1, 0, 1, 1>(srcBuffer, 0, uBuffer, vBuffer, aWidth);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
BGR24ToNV12(const uint8_t* aSrcBuffer, int aSrcStride,
|
|
uint8_t* aYBuffer, int aYStride,
|
|
uint8_t* aUVBuffer, int aUVStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
for (int i = 0; i < aHeight - 1; i += 2) {
|
|
const uint8_t* srcBuffer = aSrcBuffer + aSrcStride * i;
|
|
uint8_t* yBuffer = aYBuffer + aYStride * i;
|
|
uint8_t* uBuffer = aUVBuffer + aUVStride * (i / 2);
|
|
uint8_t* vBuffer = aUVBuffer + aUVStride * (i / 2) + 1;
|
|
|
|
RGBFamilyToY_Row<2, 1, 0>(srcBuffer, yBuffer, aWidth);
|
|
RGBFamilyToY_Row<2, 1, 0>(srcBuffer + aSrcStride, yBuffer + aYStride, aWidth);
|
|
RGBFamilyToUV_Row<2, 1, 0, 2, 2>(srcBuffer, aSrcStride, uBuffer, vBuffer, aWidth);
|
|
}
|
|
|
|
if (aHeight & 1) {
|
|
const int i = aHeight - 1;
|
|
const uint8_t* srcBuffer = aSrcBuffer + aSrcStride * i;
|
|
uint8_t* yBuffer = aYBuffer + aYStride * i;
|
|
uint8_t* uBuffer = aUVBuffer + aUVStride * (i / 2);
|
|
uint8_t* vBuffer = aUVBuffer + aUVStride * (i / 2) + 1;
|
|
|
|
RGBFamilyToY_Row<2, 1, 0>(srcBuffer, yBuffer, aWidth);
|
|
|
|
// Pass 0 as the aSrcStride so we don't sample next row's RGB information.
|
|
RGBFamilyToUV_Row<2, 1, 0, 2, 2>(srcBuffer, 0, uBuffer, vBuffer, aWidth);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
BGR24ToNV21(const uint8_t* aSrcBuffer, int aSrcStride,
|
|
uint8_t* aYBuffer, int aYStride,
|
|
uint8_t* aVUBuffer, int aVUStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
for (int i = 0; i < aHeight - 1; i += 2) {
|
|
const uint8_t* srcBuffer = aSrcBuffer + aSrcStride * i;
|
|
uint8_t* yBuffer = aYBuffer + aYStride * i;
|
|
uint8_t* uBuffer = aVUBuffer + aVUStride * (i / 2) + 1;
|
|
uint8_t* vBuffer = aVUBuffer + aVUStride * (i / 2);
|
|
|
|
RGBFamilyToY_Row<2, 1, 0>(srcBuffer, yBuffer, aWidth);
|
|
RGBFamilyToY_Row<2, 1, 0>(srcBuffer + aSrcStride, yBuffer + aYStride, aWidth);
|
|
RGBFamilyToUV_Row<2, 1, 0, 2, 2>(srcBuffer, aSrcStride, uBuffer, vBuffer, aWidth);
|
|
}
|
|
|
|
if (aHeight & 1) {
|
|
const int i = aHeight - 1;
|
|
const uint8_t* srcBuffer = aSrcBuffer + aSrcStride * i;
|
|
uint8_t* yBuffer = aYBuffer + aYStride * i;
|
|
uint8_t* uBuffer = aVUBuffer + aVUStride * (i / 2) + 1;
|
|
uint8_t* vBuffer = aVUBuffer + aVUStride * (i / 2);
|
|
|
|
RGBFamilyToY_Row<2, 1, 0>(srcBuffer, yBuffer, aWidth);
|
|
|
|
// Pass 0 as the aSrcStride so we don't sample next row's RGB information.
|
|
RGBFamilyToUV_Row<2, 1, 0, 2, 2>(srcBuffer, 0, uBuffer, vBuffer, aWidth);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
/*
|
|
* RGBA family -> YUV family.
|
|
*/
|
|
int
|
|
RGBA32ToYUV444P(const uint8_t* aSrcBuffer, int aSrcStride,
|
|
uint8_t* aYBuffer, int aYStride,
|
|
uint8_t* aUBuffer, int aUStride,
|
|
uint8_t* aVBuffer, int aVStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
for (int i = 0; i < aHeight; ++i) {
|
|
const uint8_t* srcBuffer = aSrcBuffer + aSrcStride * i;
|
|
uint8_t* yBuffer = aYBuffer + aYStride * i;
|
|
uint8_t* uBuffer = aUBuffer + aUStride * i;
|
|
uint8_t* vBuffer = aVBuffer + aVStride * i;
|
|
|
|
for (int j = 0; j < aWidth; ++j) {
|
|
yBuffer[0] = RGBToY(srcBuffer[0], srcBuffer[1], srcBuffer[2]);
|
|
uBuffer[0] = RGBToU(srcBuffer[0], srcBuffer[1], srcBuffer[2]);
|
|
vBuffer[0] = RGBToV(srcBuffer[0], srcBuffer[1], srcBuffer[2]);
|
|
|
|
yBuffer += 1;
|
|
uBuffer += 1;
|
|
vBuffer += 1;
|
|
srcBuffer += 4;
|
|
}
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
RGBA32ToYUV422P(const uint8_t* aSrcBuffer, int aSrcStride,
|
|
uint8_t* aYBuffer, int aYStride,
|
|
uint8_t* aUBuffer, int aUStride,
|
|
uint8_t* aVBuffer, int aVStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
for (int i = 0; i < aHeight; ++i) {
|
|
const uint8_t* srcBuffer = aSrcBuffer + aSrcStride * i;
|
|
uint8_t* yBuffer = aYBuffer + aYStride * i;
|
|
uint8_t* uBuffer = aUBuffer + aUStride * i;
|
|
uint8_t* vBuffer = aVBuffer + aVStride * i;
|
|
|
|
RGBAFamilyToY_Row<0, 1, 2>(srcBuffer, yBuffer, aWidth);
|
|
|
|
// Pass 0 as the aSrcStride so we don't sample next row's RGB information.
|
|
RGBAFamilyToUV_Row<0, 1, 2, 1, 1>(srcBuffer, 0, uBuffer, vBuffer, aWidth);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
RGBA32ToYUV420P(const uint8_t* aSrcBuffer, int aSrcStride,
|
|
uint8_t* aYBuffer, int aYStride,
|
|
uint8_t* aUBuffer, int aUStride,
|
|
uint8_t* aVBuffer, int aVStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
for (int i = 0; i < aHeight - 1; i += 2) {
|
|
const uint8_t* srcBuffer = aSrcBuffer + aSrcStride * i;
|
|
uint8_t* yBuffer = aYBuffer + aYStride * i;
|
|
uint8_t* uBuffer = aUBuffer + aUStride * (i / 2);
|
|
uint8_t* vBuffer = aVBuffer + aVStride * (i / 2);
|
|
|
|
RGBAFamilyToY_Row<0, 1, 2>(srcBuffer, yBuffer, aWidth);
|
|
RGBAFamilyToY_Row<0, 1, 2>(srcBuffer + aSrcStride, yBuffer + aYStride, aWidth);
|
|
RGBAFamilyToUV_Row<0, 1, 2, 1, 1>(srcBuffer, aSrcStride, uBuffer, vBuffer, aWidth);
|
|
}
|
|
|
|
if (aHeight & 1) {
|
|
const int i = aHeight - 1;
|
|
const uint8_t* srcBuffer = aSrcBuffer + aSrcStride * i;
|
|
uint8_t* yBuffer = aYBuffer + aYStride * i;
|
|
uint8_t* uBuffer = aUBuffer + aUStride * (i / 2);
|
|
uint8_t* vBuffer = aVBuffer + aVStride * (i / 2);
|
|
|
|
RGBAFamilyToY_Row<0, 1, 2>(srcBuffer, yBuffer, aWidth);
|
|
|
|
// Pass 0 as the aSrcStride so we don't sample next row's RGB information.
|
|
RGBAFamilyToUV_Row<0, 1, 2, 1, 1>(srcBuffer, 0, uBuffer, vBuffer, aWidth);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
RGBA32ToNV12(const uint8_t* aSrcBuffer, int aSrcStride,
|
|
uint8_t* aYBuffer, int aYStride,
|
|
uint8_t* aUVBuffer, int aUVStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
for (int i = 0; i < aHeight - 1; i += 2) {
|
|
const uint8_t* srcBuffer = aSrcBuffer + aSrcStride * i;
|
|
uint8_t* yBuffer = aYBuffer + aYStride * i;
|
|
uint8_t* uBuffer = aUVBuffer + aUVStride * (i / 2);
|
|
uint8_t* vBuffer = aUVBuffer + aUVStride * (i / 2) + 1;
|
|
|
|
RGBAFamilyToY_Row<0, 1, 2>(srcBuffer, yBuffer, aWidth);
|
|
RGBAFamilyToY_Row<0, 1, 2>(srcBuffer + aSrcStride, yBuffer + aYStride, aWidth);
|
|
RGBAFamilyToUV_Row<0, 1, 2, 2, 2>(srcBuffer, aSrcStride, uBuffer, vBuffer, aWidth);
|
|
}
|
|
|
|
if (aHeight & 1) {
|
|
const int i = aHeight - 1;
|
|
const uint8_t* srcBuffer = aSrcBuffer + aSrcStride * i;
|
|
uint8_t* yBuffer = aYBuffer + aYStride * i;
|
|
uint8_t* uBuffer = aUVBuffer + aUVStride * (i / 2);
|
|
uint8_t* vBuffer = aUVBuffer + aUVStride * (i / 2) + 1;
|
|
|
|
RGBAFamilyToY_Row<0, 1, 2>(srcBuffer, yBuffer, aWidth);
|
|
|
|
// Pass 0 as the aSrcStride so we don't sample next row's RGB information.
|
|
RGBAFamilyToUV_Row<0, 1, 2, 2, 2>(srcBuffer, 0, uBuffer, vBuffer, aWidth);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
RGBA32ToNV21(const uint8_t* aSrcBuffer, int aSrcStride,
|
|
uint8_t* aYBuffer, int aYStride,
|
|
uint8_t* aVUBuffer, int aVUStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
for (int i = 0; i < aHeight - 1; i += 2) {
|
|
const uint8_t* srcBuffer = aSrcBuffer + aSrcStride * i;
|
|
uint8_t* yBuffer = aYBuffer + aYStride * i;
|
|
uint8_t* uBuffer = aVUBuffer + aVUStride * (i / 2) + 1;
|
|
uint8_t* vBuffer = aVUBuffer + aVUStride * (i / 2);
|
|
|
|
RGBAFamilyToY_Row<0, 1, 2>(srcBuffer, yBuffer, aWidth);
|
|
RGBAFamilyToY_Row<0, 1, 2>(srcBuffer + aSrcStride, yBuffer + aYStride, aWidth);
|
|
RGBAFamilyToUV_Row<0, 1, 2, 2, 2>(srcBuffer, aSrcStride, uBuffer, vBuffer, aWidth);
|
|
}
|
|
|
|
if (aHeight & 1) {
|
|
const int i = aHeight - 1;
|
|
const uint8_t* srcBuffer = aSrcBuffer + aSrcStride * i;
|
|
uint8_t* yBuffer = aYBuffer + aYStride * i;
|
|
uint8_t* uBuffer = aVUBuffer + aVUStride * (i / 2) + 1;
|
|
uint8_t* vBuffer = aVUBuffer + aVUStride * (i / 2);
|
|
|
|
RGBAFamilyToY_Row<0, 1, 2>(srcBuffer, yBuffer, aWidth);
|
|
|
|
// Pass 0 as the aSrcStride so we don't sample next row's RGB information.
|
|
RGBAFamilyToUV_Row<0, 1, 2, 2, 2>(srcBuffer, 0, uBuffer, vBuffer, aWidth);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
BGRA32ToYUV444P(const uint8_t* aSrcBuffer, int aSrcStride,
|
|
uint8_t* aYBuffer, int aYStride,
|
|
uint8_t* aUBuffer, int aUStride,
|
|
uint8_t* aVBuffer, int aVStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
for (int i = 0; i < aHeight; ++i) {
|
|
const uint8_t* srcBuffer = aSrcBuffer + aSrcStride * i;
|
|
uint8_t* yBuffer = aYBuffer + aYStride * i;
|
|
uint8_t* uBuffer = aUBuffer + aUStride * i;
|
|
uint8_t* vBuffer = aVBuffer + aVStride * i;
|
|
|
|
for (int j = 0; j < aWidth; ++j) {
|
|
yBuffer[0] = RGBToY(srcBuffer[2], srcBuffer[1], srcBuffer[0]);
|
|
uBuffer[0] = RGBToU(srcBuffer[2], srcBuffer[1], srcBuffer[0]);
|
|
vBuffer[0] = RGBToV(srcBuffer[2], srcBuffer[1], srcBuffer[0]);
|
|
|
|
yBuffer += 1;
|
|
uBuffer += 1;
|
|
vBuffer += 1;
|
|
srcBuffer += 4;
|
|
}
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
BGRA32ToYUV422P(const uint8_t* aSrcBuffer, int aSrcStride,
|
|
uint8_t* aYBuffer, int aYStride,
|
|
uint8_t* aUBuffer, int aUStride,
|
|
uint8_t* aVBuffer, int aVStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
for (int i = 0; i < aHeight; ++i) {
|
|
const uint8_t* srcBuffer = aSrcBuffer + aSrcStride * i;
|
|
uint8_t* yBuffer = aYBuffer + aYStride * i;
|
|
uint8_t* uBuffer = aUBuffer + aUStride * i;
|
|
uint8_t* vBuffer = aVBuffer + aVStride * i;
|
|
|
|
RGBAFamilyToY_Row<2, 1, 0>(srcBuffer, yBuffer, aWidth);
|
|
|
|
// Pass 0 as the aSrcStride so we don't sample next row's RGB information.
|
|
RGBAFamilyToUV_Row<2, 1, 0, 1, 1>(srcBuffer, 0, uBuffer, vBuffer, aWidth);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
BGRA32ToYUV420P(const uint8_t* aSrcBuffer, int aSrcStride,
|
|
uint8_t* aYBuffer, int aYStride,
|
|
uint8_t* aUBuffer, int aUStride,
|
|
uint8_t* aVBuffer, int aVStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
for (int i = 0; i < aHeight - 1; i += 2) {
|
|
const uint8_t* srcBuffer = aSrcBuffer + aSrcStride * i;
|
|
uint8_t* yBuffer = aYBuffer + aYStride * i;
|
|
uint8_t* uBuffer = aUBuffer + aUStride * (i / 2);
|
|
uint8_t* vBuffer = aVBuffer + aVStride * (i / 2);
|
|
|
|
RGBAFamilyToY_Row<2, 1, 0>(srcBuffer, yBuffer, aWidth);
|
|
RGBAFamilyToY_Row<2, 1, 0>(srcBuffer + aSrcStride, yBuffer + aYStride, aWidth);
|
|
RGBAFamilyToUV_Row<2, 1, 0, 1, 1>(srcBuffer, aSrcStride, uBuffer, vBuffer, aWidth);
|
|
}
|
|
|
|
if (aHeight & 1) {
|
|
const int i = aHeight - 1;
|
|
const uint8_t* srcBuffer = aSrcBuffer + aSrcStride * i;
|
|
uint8_t* yBuffer = aYBuffer + aYStride * i;
|
|
uint8_t* uBuffer = aUBuffer + aUStride * (i / 2);
|
|
uint8_t* vBuffer = aVBuffer + aVStride * (i / 2);
|
|
|
|
RGBAFamilyToY_Row<2, 1, 0>(srcBuffer, yBuffer, aWidth);
|
|
|
|
// Pass 0 as the aSrcStride so we don't sample next row's RGB information.
|
|
RGBAFamilyToUV_Row<2, 1, 0, 1, 1>(srcBuffer, 0, uBuffer, vBuffer, aWidth);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
BGRA32ToNV12(const uint8_t* aSrcBuffer, int aSrcStride,
|
|
uint8_t* aYBuffer, int aYStride,
|
|
uint8_t* aUVBuffer, int aUVStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
for (int i = 0; i < aHeight - 1; i += 2) {
|
|
const uint8_t* srcBuffer = aSrcBuffer + aSrcStride * i;
|
|
uint8_t* yBuffer = aYBuffer + aYStride * i;
|
|
uint8_t* uBuffer = aUVBuffer + aUVStride * (i / 2);
|
|
uint8_t* vBuffer = aUVBuffer + aUVStride * (i / 2) + 1;
|
|
|
|
RGBAFamilyToY_Row<2, 1, 0>(srcBuffer, yBuffer, aWidth);
|
|
RGBAFamilyToY_Row<2, 1, 0>(srcBuffer + aSrcStride, yBuffer + aYStride, aWidth);
|
|
RGBAFamilyToUV_Row<2, 1, 0, 2, 2>(srcBuffer, aSrcStride, uBuffer, vBuffer, aWidth);
|
|
}
|
|
|
|
if (aHeight & 1) {
|
|
const int i = aHeight - 1;
|
|
const uint8_t* srcBuffer = aSrcBuffer + aSrcStride * i;
|
|
uint8_t* yBuffer = aYBuffer + aYStride * i;
|
|
uint8_t* uBuffer = aUVBuffer + aUVStride * (i / 2);
|
|
uint8_t* vBuffer = aUVBuffer + aUVStride * (i / 2) + 1;
|
|
|
|
RGBAFamilyToY_Row<2, 1, 0>(srcBuffer, yBuffer, aWidth);
|
|
|
|
// Pass 0 as the aSrcStride so we don't sample next row's RGB information.
|
|
RGBAFamilyToUV_Row<2, 1, 0, 2, 2>(srcBuffer, 0, uBuffer, vBuffer, aWidth);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
BGRA32ToNV21(const uint8_t* aSrcBuffer, int aSrcStride,
|
|
uint8_t* aYBuffer, int aYStride,
|
|
uint8_t* aVUBuffer, int aVUStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
for (int i = 0; i < aHeight - 1; i += 2) {
|
|
const uint8_t* srcBuffer = aSrcBuffer + aSrcStride * i;
|
|
uint8_t* yBuffer = aYBuffer + aYStride * i;
|
|
uint8_t* uBuffer = aVUBuffer + aVUStride * (i / 2) + 1;
|
|
uint8_t* vBuffer = aVUBuffer + aVUStride * (i / 2);
|
|
|
|
RGBAFamilyToY_Row<2, 1, 0>(srcBuffer, yBuffer, aWidth);
|
|
RGBAFamilyToY_Row<2, 1, 0>(srcBuffer + aSrcStride, yBuffer + aYStride, aWidth);
|
|
RGBAFamilyToUV_Row<2, 1, 0, 2, 2>(srcBuffer, aSrcStride, uBuffer, vBuffer, aWidth);
|
|
}
|
|
|
|
if (aHeight & 1) {
|
|
const int i = aHeight - 1;
|
|
const uint8_t* srcBuffer = aSrcBuffer + aSrcStride * i;
|
|
uint8_t* yBuffer = aYBuffer + aYStride * i;
|
|
uint8_t* uBuffer = aVUBuffer + aVUStride * (i / 2) + 1;
|
|
uint8_t* vBuffer = aVUBuffer + aVUStride * (i / 2);
|
|
|
|
RGBAFamilyToY_Row<2, 1, 0>(srcBuffer, yBuffer, aWidth);
|
|
|
|
// Pass 0 as the aSrcStride so we don't sample next row's RGB information.
|
|
RGBAFamilyToUV_Row<2, 1, 0, 2, 2>(srcBuffer, 0, uBuffer, vBuffer, aWidth);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
/*
|
|
* RGBA/RGB family -> HSV.
|
|
* Reference:
|
|
* (1) https://en.wikipedia.org/wiki/HSL_and_HSV
|
|
* (2) OpenCV implementation:
|
|
* http://docs.opencv.org/3.1.0/de/d25/imgproc_color_conversions.html
|
|
*/
|
|
const float EPSILON = 1e-10f;
|
|
|
|
template<int aRIndex, int aGIndex, int aBIndex, int aSrcStep>
|
|
int
|
|
RGBFamilyToHSV(const uint8_t* aSrcBuffer, int aSrcStride,
|
|
float* aDstBuffer, int aDstStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
for (int i = 0; i < aHeight; ++i) {
|
|
const uint8_t* srcBuffer = aSrcBuffer + aSrcStride * i;
|
|
float* dstBuffer = (float*)((uint8_t*)(aDstBuffer) + aDstStride * i);
|
|
|
|
for (int j = 0; j < aWidth; ++j) {
|
|
const float r = (float)(srcBuffer[aRIndex]) / 255.0f;
|
|
const float g = (float)(srcBuffer[aGIndex]) / 255.0f;
|
|
const float b = (float)(srcBuffer[aBIndex]) / 255.0f;
|
|
float& h = dstBuffer[0];
|
|
float& s = dstBuffer[1];
|
|
float& v = dstBuffer[2];
|
|
|
|
float min = r;
|
|
if (g < min) min = g;
|
|
if (b < min) min = b;
|
|
|
|
float max = r;
|
|
if (g > max) max = g;
|
|
if (b > max) max = b;
|
|
|
|
const float diff = max - min + EPSILON; // Prevent dividing by zero.
|
|
|
|
// Calculate v.
|
|
v = max;
|
|
|
|
// Calculate s.
|
|
if (max == 0.0f) {
|
|
s = 0.0f;
|
|
} else {
|
|
s = diff / v;
|
|
}
|
|
|
|
// Calculate h.
|
|
if (max == r) {
|
|
h = 60.0f * (g - b) / diff;
|
|
} else if (max == g) {
|
|
h = 60.0f * (b - r) / diff + 120.0f;
|
|
} else if (max == b) {
|
|
h = 60.0f * (r - g) / diff + 240.0f;
|
|
}
|
|
|
|
if (h < 0.0f) {
|
|
h += 360.0f;
|
|
}
|
|
|
|
// Step one pixel.
|
|
srcBuffer += aSrcStep;
|
|
dstBuffer += 3;
|
|
}
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
static const int sector_data[][3]= {{0,3,1}, {2,0,1}, {1,0,3}, {1,2,0}, {3,1,0}, {0,1,2}};
|
|
|
|
// If the destination is a RGB24 or BGR24, set the aAIndex to be 0, 1 or 2,
|
|
// so that the r, g or b value will be set to 255 first than to the right value.
|
|
template<int aRIndex, int aGIndex, int aBIndex, int aAIndex, int aDstStep>
|
|
int
|
|
HSVToRGBAFamily(const float* aSrcBuffer, int aSrcStride,
|
|
uint8_t* aDstBuffer, int aDstStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
static_assert(aRIndex == 0 || aRIndex == 2, "Wrong R index.");
|
|
static_assert(aGIndex == 1, "Wrong G index.");
|
|
static_assert(aBIndex == 0 || aBIndex == 2, "Wrong B index.");
|
|
static_assert(aAIndex == 0 || aAIndex == 1 || aAIndex == 2 || aAIndex == 3, "Wrong A index.");
|
|
|
|
for (int i = 0; i < aHeight; ++i) {
|
|
const float* srcBuffer = (const float*)((const uint8_t*)(aSrcBuffer) + aSrcStride * i);
|
|
uint8_t* dstBuffer = aDstBuffer + aDstStride * i;
|
|
|
|
for (int j = 0; j < aWidth; ++j) {
|
|
const float h = srcBuffer[0];
|
|
const float s = srcBuffer[1];
|
|
const float v = srcBuffer[2];
|
|
|
|
// Calculate h-prime which should be in range [0, 6). -> h should be in
|
|
// range [0, 360).
|
|
float hPrime = h / 60.0f;
|
|
if (hPrime < 0.0f)
|
|
do hPrime += 6.0f; while (hPrime < 0.0f);
|
|
else if (hPrime >= 6.0f)
|
|
do hPrime -= 6.0f; while (hPrime >= 6.0f);
|
|
const int sector = floor(hPrime);
|
|
const float hMod1 = hPrime - sector;
|
|
|
|
float values[4];
|
|
values[0] = v;
|
|
values[1] = v * (1.0f - s);
|
|
values[2] = v * (1.0f - s * hMod1);
|
|
values[3] = v * (1.0f - s * (1.0f - hMod1));
|
|
|
|
dstBuffer[aAIndex] = 255;
|
|
dstBuffer[aRIndex] = Clamp(values[sector_data[sector][0]] * 255.0f);
|
|
dstBuffer[aGIndex] = Clamp(values[sector_data[sector][1]] * 255.0f);
|
|
dstBuffer[aBIndex] = Clamp(values[sector_data[sector][2]] * 255.0f);
|
|
|
|
// Step one pixel.
|
|
srcBuffer += 3;
|
|
dstBuffer += aDstStep;
|
|
}
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
RGBA32ToHSV(const uint8_t* aSrcBuffer, int aSrcStride,
|
|
float* aDstBuffer, int aDstStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
return RGBFamilyToHSV<0, 1, 2, 4>(aSrcBuffer, aSrcStride,
|
|
aDstBuffer, aDstStride,
|
|
aWidth, aHeight);
|
|
}
|
|
|
|
int
|
|
BGRA32ToHSV(const uint8_t* aSrcBuffer, int aSrcStride,
|
|
float* aDstBuffer, int aDstStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
return RGBFamilyToHSV<2, 1, 0, 4>(aSrcBuffer, aSrcStride,
|
|
aDstBuffer, aDstStride,
|
|
aWidth, aHeight);
|
|
}
|
|
|
|
int
|
|
RGB24ToHSV(const uint8_t* aSrcBuffer, int aSrcStride,
|
|
float* aDstBuffer, int aDstStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
return RGBFamilyToHSV<0, 1, 2, 3>(aSrcBuffer, aSrcStride,
|
|
aDstBuffer, aDstStride,
|
|
aWidth, aHeight);
|
|
}
|
|
|
|
int
|
|
BGR24ToHSV(const uint8_t* aSrcBuffer, int aSrcStride,
|
|
float* aDstBuffer, int aDstStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
return RGBFamilyToHSV<2, 1, 0, 3>(aSrcBuffer, aSrcStride,
|
|
aDstBuffer, aDstStride,
|
|
aWidth, aHeight);
|
|
}
|
|
|
|
int
|
|
HSVToRGBA32(const float* aSrcBuffer, int aSrcStride,
|
|
uint8_t* aDstBuffer, int aDstStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
return HSVToRGBAFamily<0, 1, 2, 3, 4>(aSrcBuffer, aSrcStride,
|
|
aDstBuffer, aDstStride,
|
|
aWidth, aHeight);
|
|
}
|
|
|
|
int
|
|
HSVToBGRA32(const float* aSrcBuffer, int aSrcStride,
|
|
uint8_t* aDstBuffer, int aDstStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
return HSVToRGBAFamily<2, 1, 0, 3, 4>(aSrcBuffer, aSrcStride,
|
|
aDstBuffer, aDstStride,
|
|
aWidth, aHeight);
|
|
}
|
|
|
|
int
|
|
HSVToRGB24(const float* aSrcBuffer, int aSrcStride,
|
|
uint8_t* aDstBuffer, int aDstStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
return HSVToRGBAFamily<0, 1, 2, 0, 3>(aSrcBuffer, aSrcStride,
|
|
aDstBuffer, aDstStride,
|
|
aWidth, aHeight);
|
|
}
|
|
|
|
int
|
|
HSVToBGR24(const float* aSrcBuffer, int aSrcStride,
|
|
uint8_t* aDstBuffer, int aDstStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
return HSVToRGBAFamily<2, 1, 0, 0, 3>(aSrcBuffer, aSrcStride,
|
|
aDstBuffer, aDstStride,
|
|
aWidth, aHeight);
|
|
}
|
|
|
|
/*
|
|
* RGBA/RGB family -> Lab.
|
|
* Reference:
|
|
* (1) https://en.wikipedia.org/wiki/SRGB
|
|
* (2) https://en.wikipedia.org/wiki/Lab_color_space
|
|
* (3) OpenCV implementation:
|
|
* http://docs.opencv.org/3.1.0/de/d25/imgproc_color_conversions.html
|
|
*/
|
|
static const float sRGBToXYZ_D65[] = {0.412453f, 0.357580f, 0.180423f,
|
|
0.212671f, 0.715160f, 0.072169f,
|
|
0.019334f, 0.119193f, 0.950227f};
|
|
static const float XYZTosRGB_D65[] = {3.240479f, -1.53715f, -0.498535f,
|
|
-0.969256f, 1.875991f, 0.041556f,
|
|
0.055648f, -0.204043f, 1.057311f};
|
|
static const float whitept_D65[] = {0.950456f, 1.0f, 1.088754f};
|
|
static const float _magic = std::pow((6.0 / 29.0), 3.0); // should be around 0.008856.
|
|
static const float _1_3 = 1.0f / 3.0f;
|
|
static const float _a = std::pow((29.0 / 6.0), 2.0) / 3.0; // should be around 7.787.
|
|
static const float _b = 16.0f / 116.0f; // should be around 0.1379.
|
|
|
|
template<int aRIndex, int aGIndex, int aBIndex, int aSrcStep>
|
|
int
|
|
RGBFamilyToLab(const uint8_t* aSrcBuffer, int aSrcStride,
|
|
float* aDstBuffer, int aDstStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
static_assert(aRIndex == 0 || aRIndex == 2, "Wrong R index.");
|
|
static_assert(aGIndex == 1, "Wrong G index.");
|
|
static_assert(aBIndex == 0 || aBIndex == 2, "Wrong B index.");
|
|
|
|
const float C0 = sRGBToXYZ_D65[0] / whitept_D65[0],
|
|
C1 = sRGBToXYZ_D65[1] / whitept_D65[0],
|
|
C2 = sRGBToXYZ_D65[2] / whitept_D65[0],
|
|
C3 = sRGBToXYZ_D65[3] / whitept_D65[1],
|
|
C4 = sRGBToXYZ_D65[4] / whitept_D65[1],
|
|
C5 = sRGBToXYZ_D65[5] / whitept_D65[1],
|
|
C6 = sRGBToXYZ_D65[6] / whitept_D65[2],
|
|
C7 = sRGBToXYZ_D65[7] / whitept_D65[2],
|
|
C8 = sRGBToXYZ_D65[8] / whitept_D65[2];
|
|
|
|
for (int i = 0; i < aHeight; ++i) {
|
|
const uint8_t* srcBuffer = aSrcBuffer + aSrcStride * i;
|
|
float* dstBuffer = (float*)((uint8_t*)(aDstBuffer) + aDstStride * i);
|
|
|
|
for (int j = 0; j < aWidth; ++j) {
|
|
float r = (float)(srcBuffer[aRIndex]) / 255.0f;
|
|
float g = (float)(srcBuffer[aGIndex]) / 255.0f;
|
|
float b = (float)(srcBuffer[aBIndex]) / 255.0f;
|
|
|
|
// gamma correction of sRGB
|
|
r = r <= 0.04045f ? r / 12.92f : std::pow((r + 0.055) / 1.055, 2.4);
|
|
g = g <= 0.04045f ? g / 12.92f : std::pow((g + 0.055) / 1.055, 2.4);
|
|
b = b <= 0.04045f ? b / 12.92f : std::pow((b + 0.055) / 1.055, 2.4);
|
|
|
|
const float X = C0 * r + C1 * g + C2 * b;
|
|
const float Y = C3 * r + C4 * g + C5 * b;
|
|
const float Z = C6 * r + C7 * g + C8 * b;
|
|
|
|
const float FX = X > _magic ? std::pow(X, _1_3) : (_a * X + _b);
|
|
const float FY = Y > _magic ? std::pow(Y, _1_3) : (_a * Y + _b);
|
|
const float FZ = Z > _magic ? std::pow(Z, _1_3) : (_a * Z + _b);
|
|
|
|
dstBuffer[0] = 116.0f * FY - 16.0f;
|
|
dstBuffer[1] = 500.0f * (FX - FY);
|
|
dstBuffer[2] = 200.0f * (FY - FZ);
|
|
|
|
// Step one pixel.
|
|
srcBuffer += aSrcStep;
|
|
dstBuffer += 3;
|
|
}
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
// If the destination is a RGB24 or BGR24, set the aAIndex to be 0, 1 or 2,
|
|
// so that the r, g or b value will be set to 255 first than to the right value.
|
|
template<int aRIndex, int aGIndex, int aBIndex, int aAIndex, int aDstStep>
|
|
int
|
|
LabToRGBAFamily(const float* aSrcBuffer, int aSrcStride,
|
|
uint8_t* aDstBuffer, int aDstStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
static_assert(aRIndex == 0 || aRIndex == 2, "Wrong R index.");
|
|
static_assert(aGIndex == 1, "Wrong G index.");
|
|
static_assert(aBIndex == 0 || aBIndex == 2, "Wrong B index.");
|
|
static_assert(aAIndex == 0 || aAIndex == 1 || aAIndex == 2 || aAIndex == 3, "Wrong A index.");
|
|
|
|
const float C0 = XYZTosRGB_D65[0] * whitept_D65[0],
|
|
C1 = XYZTosRGB_D65[1] * whitept_D65[1],
|
|
C2 = XYZTosRGB_D65[2] * whitept_D65[2],
|
|
C3 = XYZTosRGB_D65[3] * whitept_D65[0],
|
|
C4 = XYZTosRGB_D65[4] * whitept_D65[1],
|
|
C5 = XYZTosRGB_D65[5] * whitept_D65[2],
|
|
C6 = XYZTosRGB_D65[6] * whitept_D65[0],
|
|
C7 = XYZTosRGB_D65[7] * whitept_D65[1],
|
|
C8 = XYZTosRGB_D65[8] * whitept_D65[2];
|
|
|
|
for (int i = 0; i < aHeight; ++i) {
|
|
const float* srcBuffer = (const float*)((const uint8_t*)(aSrcBuffer) + aSrcStride * i);
|
|
uint8_t* dstBuffer = aDstBuffer + aDstStride * i;
|
|
|
|
for (int j = 0; j < aWidth; ++j) {
|
|
const float L = srcBuffer[0];
|
|
const float a = srcBuffer[1];
|
|
const float b = srcBuffer[2];
|
|
|
|
const float FY = (L + 16.0f) / 116.0f;
|
|
const float FX = (a / 500.0f) + FY;
|
|
const float FZ = FY - (b / 200.0f);
|
|
|
|
const float X = FX > 6.0f / 29.0f ? std::pow((double)FX, 3.0) : 3.0 * std::pow((6.0 / 29.0), 2.0) * (FX - (4.0 / 29.0));
|
|
const float Y = FY > 6.0f / 29.0f ? std::pow((double)FY, 3.0) : 3.0 * std::pow((6.0 / 29.0), 2.0) * (FY - (4.0 / 29.0));
|
|
const float Z = FZ > 6.0f / 29.0f ? std::pow((double)FZ, 3.0) : 3.0 * std::pow((6.0 / 29.0), 2.0) * (FZ - (4.0 / 29.0));
|
|
|
|
const float r0 = C0 * X + C1 * Y + C2 * Z;
|
|
const float g0 = C3 * X + C4 * Y + C5 * Z;
|
|
const float b0 = C6 * X + C7 * Y + C8 * Z;
|
|
|
|
// Apply gamma curve of sRGB to the linear rgb values.
|
|
dstBuffer[aAIndex] = 255;
|
|
dstBuffer[aRIndex] = Clamp((r0 <= 0.0031308f ? r0 * 12.92f : 1.055 * std::pow((double)r0, 1.0 / 2.4) - 0.055) * 255.0);
|
|
dstBuffer[aGIndex] = Clamp((g0 <= 0.0031308f ? g0 * 12.92f : 1.055 * std::pow((double)g0, 1.0 / 2.4) - 0.055) * 255.0);
|
|
dstBuffer[aBIndex] = Clamp((b0 <= 0.0031308f ? b0 * 12.92f : 1.055 * std::pow((double)b0, 1.0 / 2.4) - 0.055) * 255.0);
|
|
|
|
// Step one pixel.
|
|
srcBuffer += 3;
|
|
dstBuffer += aDstStep;
|
|
}
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
RGBA32ToLab(const uint8_t* aSrcBuffer, int aSrcStride,
|
|
float* aDstBuffer, int aDstStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
return RGBFamilyToLab<0, 1, 2, 4>(aSrcBuffer, aSrcStride,
|
|
aDstBuffer, aDstStride,
|
|
aWidth, aHeight);
|
|
}
|
|
|
|
int
|
|
BGRA32ToLab(const uint8_t* aSrcBuffer, int aSrcStride,
|
|
float* aDstBuffer, int aDstStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
return RGBFamilyToLab<2, 1, 0, 4>(aSrcBuffer, aSrcStride,
|
|
aDstBuffer, aDstStride,
|
|
aWidth, aHeight);
|
|
}
|
|
|
|
int
|
|
RGB24ToLab(const uint8_t* aSrcBuffer, int aSrcStride,
|
|
float* aDstBuffer, int aDstStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
return RGBFamilyToLab<0, 1, 2, 3>(aSrcBuffer, aSrcStride,
|
|
aDstBuffer, aDstStride,
|
|
aWidth, aHeight);
|
|
}
|
|
|
|
int
|
|
BGR24ToLab(const uint8_t* aSrcBuffer, int aSrcStride,
|
|
float* aDstBuffer, int aDstStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
return RGBFamilyToLab<2, 1, 0, 3>(aSrcBuffer, aSrcStride,
|
|
aDstBuffer, aDstStride,
|
|
aWidth, aHeight);
|
|
}
|
|
|
|
int
|
|
LabToRGBA32(const float* aSrcBuffer, int aSrcStride,
|
|
uint8_t* aDstBuffer, int aDstStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
return LabToRGBAFamily<0, 1, 2, 3, 4>(aSrcBuffer, aSrcStride,
|
|
aDstBuffer, aDstStride,
|
|
aWidth, aHeight);
|
|
}
|
|
|
|
int
|
|
LabToBGRA32(const float* aSrcBuffer, int aSrcStride,
|
|
uint8_t* aDstBuffer, int aDstStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
return LabToRGBAFamily<2, 1, 0, 3, 4>(aSrcBuffer, aSrcStride,
|
|
aDstBuffer, aDstStride,
|
|
aWidth, aHeight);
|
|
}
|
|
|
|
int
|
|
LabToRGB24(const float* aSrcBuffer, int aSrcStride,
|
|
uint8_t* aDstBuffer, int aDstStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
return LabToRGBAFamily<0, 1, 2, 0, 3>(aSrcBuffer, aSrcStride,
|
|
aDstBuffer, aDstStride,
|
|
aWidth, aHeight);
|
|
}
|
|
|
|
int
|
|
LabToBGR24(const float* aSrcBuffer, int aSrcStride,
|
|
uint8_t* aDstBuffer, int aDstStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
return LabToRGBAFamily<2, 1, 0, 0, 3>(aSrcBuffer, aSrcStride,
|
|
aDstBuffer, aDstStride,
|
|
aWidth, aHeight);
|
|
}
|
|
|
|
/*
|
|
* RGBA/RGB family -> Gray8.
|
|
* Reference:
|
|
* (1) OpenCV implementation:
|
|
* http://docs.opencv.org/3.1.0/de/d25/imgproc_color_conversions.html
|
|
*/
|
|
template<int aRIndex, int aGIndex, int aBIndex, int aSrcStep>
|
|
int
|
|
RGBFamilyToGray8(const uint8_t* aSrcBuffer, int aSrcStride,
|
|
uint8_t* aDstBuffer, int aDstStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
static_assert(aRIndex == 0 || aRIndex == 2, "Wrong R index.");
|
|
static_assert(aGIndex == 1, "Wrong G index.");
|
|
static_assert(aBIndex == 0 || aBIndex == 2, "Wrong B index.");
|
|
|
|
for (int i = 0; i < aHeight; ++i) {
|
|
const uint8_t* srcBuffer = aSrcBuffer + aSrcStride * i;
|
|
uint8_t* dstBuffer = aDstBuffer + aDstStride * i;
|
|
|
|
for (int j = 0; j < aWidth; ++j) {
|
|
dstBuffer[j] = 0.299 * srcBuffer[aRIndex] +
|
|
0.587 * srcBuffer[aGIndex] +
|
|
0.114 * srcBuffer[aBIndex];
|
|
srcBuffer += aSrcStep;
|
|
}
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
RGB24ToGray8(const uint8_t* aSrcBuffer, int aSrcStride,
|
|
uint8_t* aDstBuffer, int aDstStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
return RGBFamilyToGray8<0, 1, 2, 3>(aSrcBuffer, aSrcStride,
|
|
aDstBuffer, aDstStride,
|
|
aWidth, aHeight);
|
|
}
|
|
|
|
int
|
|
BGR24ToGray8(const uint8_t* aSrcBuffer, int aSrcStride,
|
|
uint8_t* aDstBuffer, int aDstStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
return RGBFamilyToGray8<2, 1, 0, 3>(aSrcBuffer, aSrcStride,
|
|
aDstBuffer, aDstStride,
|
|
aWidth, aHeight);
|
|
}
|
|
|
|
int
|
|
RGBA32ToGray8(const uint8_t* aSrcBuffer, int aSrcStride,
|
|
uint8_t* aDstBuffer, int aDstStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
return RGBFamilyToGray8<0, 1, 2, 4>(aSrcBuffer, aSrcStride,
|
|
aDstBuffer, aDstStride,
|
|
aWidth, aHeight);
|
|
}
|
|
|
|
int
|
|
BGRA32ToGray8(const uint8_t* aSrcBuffer, int aSrcStride,
|
|
uint8_t* aDstBuffer, int aDstStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
return RGBFamilyToGray8<2, 1, 0, 4>(aSrcBuffer, aSrcStride,
|
|
aDstBuffer, aDstStride,
|
|
aWidth, aHeight);
|
|
}
|
|
|
|
/*
|
|
* YUV family -> Gray8.
|
|
* Reference:
|
|
* (1) OpenCV implementation:
|
|
* http://docs.opencv.org/3.1.0/de/d25/imgproc_color_conversions.html
|
|
*/
|
|
int
|
|
YUVFamilyToGray8(const uint8_t* aSrcYBuffer, int aSrcYStride,
|
|
uint8_t* aDstBuffer, int aDstStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
for (int i = 0; i < aHeight; ++i) {
|
|
const uint8_t* srcYBuffer = aSrcYBuffer + aSrcYStride * i;
|
|
uint8_t* dstBuffer = aDstBuffer + aDstStride * i;
|
|
|
|
memcpy(dstBuffer, srcYBuffer, aDstStride);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
YUV444PToGray8(const uint8_t* aYBuffer, int aYStride,
|
|
const uint8_t*, int,
|
|
const uint8_t*, int,
|
|
uint8_t* aDstBuffer, int aDstStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
return YUVFamilyToGray8(aYBuffer, aYStride,
|
|
aDstBuffer, aDstStride,
|
|
aWidth, aHeight);
|
|
}
|
|
|
|
int
|
|
YUV422PToGray8(const uint8_t* aYBuffer, int aYStride,
|
|
const uint8_t*, int,
|
|
const uint8_t*, int,
|
|
uint8_t* aDstBuffer, int aDstStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
return YUVFamilyToGray8(aYBuffer, aYStride,
|
|
aDstBuffer, aDstStride,
|
|
aWidth, aHeight);
|
|
}
|
|
|
|
int
|
|
YUV420PToGray8(const uint8_t* aYBuffer, int aYStride,
|
|
const uint8_t*, int,
|
|
const uint8_t*, int,
|
|
uint8_t* aDstBuffer, int aDstStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
return YUVFamilyToGray8(aYBuffer, aYStride,
|
|
aDstBuffer, aDstStride,
|
|
aWidth, aHeight);
|
|
}
|
|
|
|
int
|
|
NV12ToGray8(const uint8_t* aYBuffer, int aYStride,
|
|
const uint8_t*, int,
|
|
uint8_t* aDstBuffer, int aDstStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
return YUVFamilyToGray8(aYBuffer, aYStride,
|
|
aDstBuffer, aDstStride,
|
|
aWidth, aHeight);
|
|
}
|
|
|
|
int
|
|
NV21ToGray8(const uint8_t* aYBuffer, int aYStride,
|
|
const uint8_t*, int,
|
|
uint8_t* aDstBuffer, int aDstStride,
|
|
int aWidth, int aHeight)
|
|
{
|
|
return YUVFamilyToGray8(aYBuffer, aYStride,
|
|
aDstBuffer, aDstStride,
|
|
aWidth, aHeight);
|
|
}
|
|
|
|
} // namespace dom
|
|
} // namespace mozilla
|