Magic-Mirror/gecko-dev
1
0
Fork 0
mirror of https://github.com/mozilla/gecko-dev.git synced 2024-12-11 16:32:59 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity
80adc67aba
gecko-dev/dom/canvas/ImageBitmapColorUtils.cpp

2071 lines
72 KiB
C++
Raw Blame History

/* -*- 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_t clamp0(int32_t v) {
return ((-(v) >> 31) & (v));
}
static __inline int32_t clamp255(int32_t v) {
return (((255 - (v)) >> 31) | (v)) & 255;
}
static __inline uint32_t Clamp(int32_t val) {
int v = clamp0(val);
return (uint32_t)(clamp255(v));
}
#define YG 74 /* (int8_t)(1.164 * 64 + 0.5) */
#define UB 127 /* min(63,(int8_t)(2.018 * 64)) */
#define UG -25 /* (int8_t)(-0.391 * 64 - 0.5) */
#define UR 0
#define VB 0
#define VG -52 /* (int8_t)(-0.813 * 64 - 0.5) */
#define VR 102 /* (int8_t)(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_t y, uint8_t u, uint8_t v, uint8_t* b, uint8_t* g, uint8_t* r)
{
int32_t y1 = ((int32_t)(y) - 16) * YG;
*b = Clamp((int32_t)((u * UB + v * VB) - (BB) + y1) >> 6);
*g = Clamp((int32_t)((u * UG + v * VG) - (BG) + y1) >> 6);
*r = Clamp((int32_t)((u * UR + v * VR) - (BR) + y1) >> 6);
}
static __inline int
RGBToY(uint8_t r, uint8_t g, uint8_t b)
{
return (66 * r + 129 * g + 25 * b + 0x1080) >> 8;
}
static __inline int
RGBToU(uint8_t r, uint8_t g, uint8_t b)
{
return (112 * b - 74 * g - 38 * r + 0x8080) >> 8;
}
static __inline int
RGBToV(uint8_t r, uint8_t g, uint8_t 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
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
Reference in New Issue View Git Blame Copy Permalink