зеркало из https://github.com/mozilla/gecko-dev.git
121 строка
4.0 KiB
C++
121 строка
4.0 KiB
C++
/* -*- Mode: C++; tab-width: 20; indent-tabs-mode: nil; c-basic-offset: 2 -*-
|
|
* 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 "libyuv.h"
|
|
#include "MacIOSurfaceHelpers.h"
|
|
#include "mozilla/gfx/MacIOSurface.h"
|
|
#include "YCbCrUtils.h"
|
|
|
|
namespace mozilla {
|
|
|
|
using namespace gfx;
|
|
|
|
namespace layers {
|
|
|
|
#define ALIGNED_32(x) ((x+31)&~31)
|
|
#define ALIGNEDPTR_32(x) reinterpret_cast<uint8_t*>((reinterpret_cast<uintptr_t>(x)+31)&~31)
|
|
|
|
static already_AddRefed<SourceSurface>
|
|
CreateSourceSurfaceFromLockedMacIOSurface(MacIOSurface* aSurface)
|
|
{
|
|
size_t bytesPerRow = aSurface->GetBytesPerRow();
|
|
size_t ioWidth = aSurface->GetDevicePixelWidth();
|
|
size_t ioHeight = aSurface->GetDevicePixelHeight();
|
|
SurfaceFormat ioFormat = aSurface->GetFormat();
|
|
|
|
if ((ioFormat == SurfaceFormat::NV12 || ioFormat == SurfaceFormat::YUV422) &&
|
|
(ioWidth > PlanarYCbCrImage::MAX_DIMENSION ||
|
|
ioHeight > PlanarYCbCrImage::MAX_DIMENSION)) {
|
|
return nullptr;
|
|
}
|
|
|
|
SurfaceFormat format =
|
|
(ioFormat == SurfaceFormat::NV12 || ioFormat == SurfaceFormat::YUV422)
|
|
? SurfaceFormat::B8G8R8X8
|
|
: SurfaceFormat::B8G8R8A8;
|
|
|
|
RefPtr<DataSourceSurface> dataSurface =
|
|
Factory::CreateDataSourceSurface(IntSize::Truncate(ioWidth, ioHeight), format);
|
|
if (NS_WARN_IF(!dataSurface)) {
|
|
return nullptr;
|
|
}
|
|
|
|
DataSourceSurface::MappedSurface mappedSurface;
|
|
if (!dataSurface->Map(DataSourceSurface::WRITE, &mappedSurface)) {
|
|
return nullptr;
|
|
}
|
|
|
|
if (ioFormat == SurfaceFormat::NV12) {
|
|
/* Extract and separate the CbCr planes */
|
|
size_t cbCrStride = aSurface->GetBytesPerRow(1);
|
|
size_t cbCrWidth = aSurface->GetDevicePixelWidth(1);
|
|
size_t cbCrHeight = aSurface->GetDevicePixelHeight(1);
|
|
|
|
auto cbPlane = MakeUnique<uint8_t[]>(cbCrWidth * cbCrHeight);
|
|
auto crPlane = MakeUnique<uint8_t[]>(cbCrWidth * cbCrHeight);
|
|
|
|
uint8_t* src = (uint8_t*)aSurface->GetBaseAddressOfPlane(1);
|
|
uint8_t* cbDest = cbPlane.get();
|
|
uint8_t* crDest = crPlane.get();
|
|
|
|
for (size_t i = 0; i < cbCrHeight; i++) {
|
|
uint8_t* rowSrc = src + cbCrStride * i;
|
|
for (size_t j = 0; j < cbCrWidth; j++) {
|
|
*cbDest = *rowSrc;
|
|
cbDest++;
|
|
rowSrc++;
|
|
*crDest = *rowSrc;
|
|
crDest++;
|
|
rowSrc++;
|
|
}
|
|
}
|
|
|
|
/* Convert to RGB */
|
|
PlanarYCbCrData data;
|
|
data.mYChannel = (uint8_t*)aSurface->GetBaseAddressOfPlane(0);
|
|
data.mYStride = aSurface->GetBytesPerRow(0);
|
|
data.mYSize = IntSize::Truncate(ioWidth, ioHeight);
|
|
data.mCbChannel = cbPlane.get();
|
|
data.mCrChannel = crPlane.get();
|
|
data.mCbCrStride = cbCrWidth;
|
|
data.mCbCrSize = IntSize::Truncate(cbCrWidth, cbCrHeight);
|
|
data.mPicSize = data.mYSize;
|
|
|
|
ConvertYCbCrToRGB(data, SurfaceFormat::B8G8R8X8, IntSize::Truncate(ioWidth, ioHeight), mappedSurface.mData, mappedSurface.mStride);
|
|
} else if (ioFormat == SurfaceFormat::YUV422) {
|
|
IntSize size = IntSize::Truncate(ioWidth, ioHeight);
|
|
libyuv::ConvertToARGB((uint8_t*)aSurface->GetBaseAddress(), 0 /* not used */,
|
|
mappedSurface.mData, mappedSurface.mStride,
|
|
0, 0,
|
|
size.width, size.height,
|
|
size.width, size.height,
|
|
libyuv::kRotate0, libyuv::FOURCC_UYVY);
|
|
} else {
|
|
unsigned char* ioData = (unsigned char*)aSurface->GetBaseAddress();
|
|
|
|
for (size_t i = 0; i < ioHeight; ++i) {
|
|
memcpy(mappedSurface.mData + i * mappedSurface.mStride,
|
|
ioData + i * bytesPerRow,
|
|
ioWidth * 4);
|
|
}
|
|
}
|
|
|
|
dataSurface->Unmap();
|
|
|
|
return dataSurface.forget();
|
|
}
|
|
|
|
already_AddRefed<SourceSurface>
|
|
CreateSourceSurfaceFromMacIOSurface(MacIOSurface* aSurface)
|
|
{
|
|
aSurface->Lock();
|
|
RefPtr<SourceSurface> result = CreateSourceSurfaceFromLockedMacIOSurface(aSurface);
|
|
aSurface->Unlock();
|
|
return result.forget();
|
|
}
|
|
|
|
} // namespace gfx
|
|
} // namespace mozilla
|