summaryrefslogtreecommitdiffstats
path: root/gfx/layers/ImageDataSerializer.cpp
diff options
context:
space:
mode:
authorDaniel Baumann <daniel.baumann@progress-linux.org>2024-04-21 11:44:51 +0000
committerDaniel Baumann <daniel.baumann@progress-linux.org>2024-04-21 11:44:51 +0000
commit9e3c08db40b8916968b9f30096c7be3f00ce9647 (patch)
treea68f146d7fa01f0134297619fbe7e33db084e0aa /gfx/layers/ImageDataSerializer.cpp
parentInitial commit. (diff)
downloadthunderbird-9e3c08db40b8916968b9f30096c7be3f00ce9647.tar.xz
thunderbird-9e3c08db40b8916968b9f30096c7be3f00ce9647.zip
Adding upstream version 1:115.7.0.upstream/1%115.7.0upstream
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'gfx/layers/ImageDataSerializer.cpp')
-rw-r--r--gfx/layers/ImageDataSerializer.cpp359
1 files changed, 359 insertions, 0 deletions
diff --git a/gfx/layers/ImageDataSerializer.cpp b/gfx/layers/ImageDataSerializer.cpp
new file mode 100644
index 0000000000..70d5493185
--- /dev/null
+++ b/gfx/layers/ImageDataSerializer.cpp
@@ -0,0 +1,359 @@
+/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
+/* vim: set ts=8 sts=2 et sw=2 tw=80: */
+/* 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 "ImageDataSerializer.h"
+
+#include "YCbCrUtils.h" // for YCbCr conversions
+#include "gfx2DGlue.h" // for SurfaceFormatToImageFormat
+#include "mozilla/Assertions.h" // for MOZ_ASSERT, etc
+#include "mozilla/gfx/2D.h" // for DataSourceSurface, Factory
+#include "mozilla/gfx/Logging.h" // for gfxDebug
+#include "mozilla/gfx/Tools.h" // for GetAlignedStride, etc
+#include "mozilla/gfx/Types.h"
+#include "mozilla/mozalloc.h" // for operator delete, etc
+
+namespace mozilla {
+namespace layers {
+namespace ImageDataSerializer {
+
+using namespace gfx;
+
+int32_t ComputeRGBStride(SurfaceFormat aFormat, int32_t aWidth) {
+#ifdef XP_MACOSX
+ // Some drivers require an alignment of 32 bytes for efficient texture upload.
+ return GetAlignedStride<32>(aWidth, BytesPerPixel(aFormat));
+#else
+ return GetAlignedStride<4>(aWidth, BytesPerPixel(aFormat));
+#endif
+}
+
+int32_t GetRGBStride(const RGBDescriptor& aDescriptor) {
+ return ComputeRGBStride(aDescriptor.format(), aDescriptor.size().width);
+}
+
+uint32_t ComputeRGBBufferSize(IntSize aSize, SurfaceFormat aFormat) {
+ MOZ_ASSERT(aSize.height >= 0 && aSize.width >= 0);
+
+ // This takes care of checking whether there could be overflow
+ // with enough margin for the metadata.
+ if (!gfx::Factory::AllowedSurfaceSize(aSize)) {
+ return 0;
+ }
+
+ // Note we're passing height instad of the bpp parameter, but the end
+ // result is the same - and the bpp was already taken care of in the
+ // ComputeRGBStride function.
+ int32_t bufsize = GetAlignedStride<16>(ComputeRGBStride(aFormat, aSize.width),
+ aSize.height);
+
+ if (bufsize < 0) {
+ // This should not be possible thanks to Factory::AllowedSurfaceSize
+ return 0;
+ }
+
+ return bufsize;
+}
+
+// Minimum required shmem size in bytes
+uint32_t ComputeYCbCrBufferSize(const gfx::IntSize& aYSize, int32_t aYStride,
+ const gfx::IntSize& aCbCrSize,
+ int32_t aCbCrStride) {
+ MOZ_ASSERT(aYSize.height >= 0 && aYSize.width >= 0);
+
+ if (aYSize.height < 0 || aYSize.width < 0 || aCbCrSize.height < 0 ||
+ aCbCrSize.width < 0 ||
+ !gfx::Factory::AllowedSurfaceSize(IntSize(aYStride, aYSize.height)) ||
+ !gfx::Factory::AllowedSurfaceSize(
+ IntSize(aCbCrStride, aCbCrSize.height))) {
+ return 0;
+ }
+
+ // Overflow checks are performed in AllowedSurfaceSize
+ return GetAlignedStride<4>(aYSize.height, aYStride) +
+ 2 * GetAlignedStride<4>(aCbCrSize.height, aCbCrStride);
+}
+
+uint32_t ComputeYCbCrBufferSize(const gfx::IntSize& aYSize, int32_t aYStride,
+ const gfx::IntSize& aCbCrSize,
+ int32_t aCbCrStride, uint32_t aYOffset,
+ uint32_t aCbOffset, uint32_t aCrOffset) {
+ MOZ_ASSERT(aYSize.height >= 0 && aYSize.width >= 0);
+
+ if (aYSize.height < 0 || aYSize.width < 0 || aCbCrSize.height < 0 ||
+ aCbCrSize.width < 0 ||
+ !gfx::Factory::AllowedSurfaceSize(IntSize(aYStride, aYSize.height)) ||
+ !gfx::Factory::AllowedSurfaceSize(
+ IntSize(aCbCrStride, aCbCrSize.height))) {
+ return 0;
+ }
+
+ uint32_t yLength = GetAlignedStride<4>(aYStride, aYSize.height);
+ uint32_t cbCrLength = GetAlignedStride<4>(aCbCrStride, aCbCrSize.height);
+ if (yLength == 0 || cbCrLength == 0) {
+ return 0;
+ }
+
+ CheckedInt<uint32_t> yEnd = aYOffset;
+ yEnd += yLength;
+ CheckedInt<uint32_t> cbEnd = aCbOffset;
+ cbEnd += cbCrLength;
+ CheckedInt<uint32_t> crEnd = aCrOffset;
+ crEnd += cbCrLength;
+
+ if (!yEnd.isValid() || !cbEnd.isValid() || !crEnd.isValid() ||
+ yEnd.value() > aCbOffset || cbEnd.value() > aCrOffset) {
+ return 0;
+ }
+
+ return crEnd.value();
+}
+
+uint32_t ComputeYCbCrBufferSize(uint32_t aBufferSize) {
+ return GetAlignedStride<4>(aBufferSize, 1);
+}
+
+void ComputeYCbCrOffsets(int32_t yStride, int32_t yHeight, int32_t cbCrStride,
+ int32_t cbCrHeight, uint32_t& outYOffset,
+ uint32_t& outCbOffset, uint32_t& outCrOffset) {
+ outYOffset = 0;
+ outCbOffset = outYOffset + GetAlignedStride<4>(yStride, yHeight);
+ outCrOffset = outCbOffset + GetAlignedStride<4>(cbCrStride, cbCrHeight);
+}
+
+gfx::SurfaceFormat FormatFromBufferDescriptor(
+ const BufferDescriptor& aDescriptor) {
+ switch (aDescriptor.type()) {
+ case BufferDescriptor::TRGBDescriptor:
+ return aDescriptor.get_RGBDescriptor().format();
+ case BufferDescriptor::TYCbCrDescriptor:
+ return gfx::SurfaceFormat::YUV;
+ default:
+ MOZ_CRASH("GFX: FormatFromBufferDescriptor");
+ }
+}
+
+gfx::IntSize SizeFromBufferDescriptor(const BufferDescriptor& aDescriptor) {
+ switch (aDescriptor.type()) {
+ case BufferDescriptor::TRGBDescriptor:
+ return aDescriptor.get_RGBDescriptor().size();
+ case BufferDescriptor::TYCbCrDescriptor: {
+ return aDescriptor.get_YCbCrDescriptor().display().Size();
+ }
+ default:
+ MOZ_CRASH("GFX: SizeFromBufferDescriptor");
+ }
+}
+
+gfx::IntRect RectFromBufferDescriptor(const BufferDescriptor& aDescriptor) {
+ switch (aDescriptor.type()) {
+ case BufferDescriptor::TRGBDescriptor: {
+ auto size = aDescriptor.get_RGBDescriptor().size();
+ return gfx::IntRect(0, 0, size.Width(), size.Height());
+ }
+ case BufferDescriptor::TYCbCrDescriptor:
+ return aDescriptor.get_YCbCrDescriptor().display();
+ default:
+ MOZ_CRASH("GFX: RectFromBufferDescriptor");
+ }
+}
+
+Maybe<gfx::IntSize> YSizeFromBufferDescriptor(
+ const BufferDescriptor& aDescriptor) {
+ switch (aDescriptor.type()) {
+ case BufferDescriptor::TRGBDescriptor:
+ return Nothing();
+ case BufferDescriptor::TYCbCrDescriptor:
+ return Some(aDescriptor.get_YCbCrDescriptor().ySize());
+ default:
+ MOZ_CRASH("GFX: YSizeFromBufferDescriptor");
+ }
+}
+
+Maybe<gfx::IntSize> CbCrSizeFromBufferDescriptor(
+ const BufferDescriptor& aDescriptor) {
+ switch (aDescriptor.type()) {
+ case BufferDescriptor::TRGBDescriptor:
+ return Nothing();
+ case BufferDescriptor::TYCbCrDescriptor:
+ return Some(aDescriptor.get_YCbCrDescriptor().cbCrSize());
+ default:
+ MOZ_CRASH("GFX: CbCrSizeFromBufferDescriptor");
+ }
+}
+
+Maybe<int32_t> YStrideFromBufferDescriptor(
+ const BufferDescriptor& aDescriptor) {
+ switch (aDescriptor.type()) {
+ case BufferDescriptor::TRGBDescriptor:
+ return Nothing();
+ case BufferDescriptor::TYCbCrDescriptor:
+ return Some(aDescriptor.get_YCbCrDescriptor().yStride());
+ default:
+ MOZ_CRASH("GFX: YStrideFromBufferDescriptor");
+ }
+}
+
+Maybe<int32_t> CbCrStrideFromBufferDescriptor(
+ const BufferDescriptor& aDescriptor) {
+ switch (aDescriptor.type()) {
+ case BufferDescriptor::TRGBDescriptor:
+ return Nothing();
+ case BufferDescriptor::TYCbCrDescriptor:
+ return Some(aDescriptor.get_YCbCrDescriptor().cbCrStride());
+ default:
+ MOZ_CRASH("GFX: CbCrStrideFromBufferDescriptor");
+ }
+}
+
+Maybe<gfx::YUVColorSpace> YUVColorSpaceFromBufferDescriptor(
+ const BufferDescriptor& aDescriptor) {
+ switch (aDescriptor.type()) {
+ case BufferDescriptor::TRGBDescriptor:
+ return Nothing();
+ case BufferDescriptor::TYCbCrDescriptor:
+ return Some(aDescriptor.get_YCbCrDescriptor().yUVColorSpace());
+ default:
+ MOZ_CRASH("GFX: YUVColorSpaceFromBufferDescriptor");
+ }
+}
+
+Maybe<gfx::ColorDepth> ColorDepthFromBufferDescriptor(
+ const BufferDescriptor& aDescriptor) {
+ switch (aDescriptor.type()) {
+ case BufferDescriptor::TRGBDescriptor:
+ return Nothing();
+ case BufferDescriptor::TYCbCrDescriptor:
+ return Some(aDescriptor.get_YCbCrDescriptor().colorDepth());
+ default:
+ MOZ_CRASH("GFX: ColorDepthFromBufferDescriptor");
+ }
+}
+
+Maybe<gfx::ColorRange> ColorRangeFromBufferDescriptor(
+ const BufferDescriptor& aDescriptor) {
+ switch (aDescriptor.type()) {
+ case BufferDescriptor::TRGBDescriptor:
+ return Nothing();
+ case BufferDescriptor::TYCbCrDescriptor:
+ return Some(aDescriptor.get_YCbCrDescriptor().colorRange());
+ default:
+ MOZ_CRASH("GFX: YUVFullRangeFromBufferDescriptor");
+ }
+}
+
+Maybe<StereoMode> StereoModeFromBufferDescriptor(
+ const BufferDescriptor& aDescriptor) {
+ switch (aDescriptor.type()) {
+ case BufferDescriptor::TRGBDescriptor:
+ return Nothing();
+ case BufferDescriptor::TYCbCrDescriptor:
+ return Some(aDescriptor.get_YCbCrDescriptor().stereoMode());
+ default:
+ MOZ_CRASH("GFX: StereoModeFromBufferDescriptor");
+ }
+}
+
+Maybe<gfx::ChromaSubsampling> ChromaSubsamplingFromBufferDescriptor(
+ const BufferDescriptor& aDescriptor) {
+ switch (aDescriptor.type()) {
+ case BufferDescriptor::TRGBDescriptor:
+ return Nothing();
+ case BufferDescriptor::TYCbCrDescriptor:
+ return Some(aDescriptor.get_YCbCrDescriptor().chromaSubsampling());
+ default:
+ MOZ_CRASH("GFX: ChromaSubsamplingFromBufferDescriptor");
+ }
+}
+
+uint8_t* GetYChannel(uint8_t* aBuffer, const YCbCrDescriptor& aDescriptor) {
+ return aBuffer + aDescriptor.yOffset();
+}
+
+uint8_t* GetCbChannel(uint8_t* aBuffer, const YCbCrDescriptor& aDescriptor) {
+ return aBuffer + aDescriptor.cbOffset();
+}
+
+uint8_t* GetCrChannel(uint8_t* aBuffer, const YCbCrDescriptor& aDescriptor) {
+ return aBuffer + aDescriptor.crOffset();
+}
+
+already_AddRefed<DataSourceSurface> DataSourceSurfaceFromYCbCrDescriptor(
+ uint8_t* aBuffer, const YCbCrDescriptor& aDescriptor,
+ gfx::DataSourceSurface* aSurface) {
+ const gfx::IntRect display = aDescriptor.display();
+ const gfx::IntSize size = display.Size();
+ RefPtr<DataSourceSurface> result;
+ if (aSurface) {
+ MOZ_ASSERT(aSurface->GetSize() == size);
+ MOZ_ASSERT(aSurface->GetFormat() == gfx::SurfaceFormat::B8G8R8X8);
+ if (aSurface->GetSize() == size &&
+ aSurface->GetFormat() == gfx::SurfaceFormat::B8G8R8X8) {
+ result = aSurface;
+ }
+ }
+
+ if (!result) {
+ result =
+ Factory::CreateDataSourceSurface(size, gfx::SurfaceFormat::B8G8R8X8);
+ }
+ if (NS_WARN_IF(!result)) {
+ return nullptr;
+ }
+
+ DataSourceSurface::MappedSurface map;
+ if (NS_WARN_IF(!result->Map(DataSourceSurface::MapType::WRITE, &map))) {
+ return nullptr;
+ }
+
+ layers::PlanarYCbCrData ycbcrData;
+ ycbcrData.mYChannel = GetYChannel(aBuffer, aDescriptor);
+ ycbcrData.mYStride = aDescriptor.yStride();
+ ycbcrData.mCbChannel = GetCbChannel(aBuffer, aDescriptor);
+ ycbcrData.mCrChannel = GetCrChannel(aBuffer, aDescriptor);
+ ycbcrData.mCbCrStride = aDescriptor.cbCrStride();
+ ycbcrData.mPictureRect = aDescriptor.display();
+ ycbcrData.mYUVColorSpace = aDescriptor.yUVColorSpace();
+ ycbcrData.mColorDepth = aDescriptor.colorDepth();
+ ycbcrData.mChromaSubsampling = aDescriptor.chromaSubsampling();
+
+ gfx::ConvertYCbCrToRGB(ycbcrData, gfx::SurfaceFormat::B8G8R8X8, size,
+ map.mData, map.mStride);
+
+ result->Unmap();
+ return result.forget();
+}
+
+void ConvertAndScaleFromYCbCrDescriptor(uint8_t* aBuffer,
+ const YCbCrDescriptor& aDescriptor,
+ const gfx::SurfaceFormat& aDestFormat,
+ const gfx::IntSize& aDestSize,
+ unsigned char* aDestBuffer,
+ int32_t aStride) {
+ MOZ_ASSERT(aBuffer);
+
+ layers::PlanarYCbCrData ycbcrData;
+ ycbcrData.mYChannel = GetYChannel(aBuffer, aDescriptor);
+ ycbcrData.mYStride = aDescriptor.yStride();
+ ycbcrData.mCbChannel = GetCbChannel(aBuffer, aDescriptor);
+ ycbcrData.mCrChannel = GetCrChannel(aBuffer, aDescriptor);
+ ycbcrData.mCbCrStride = aDescriptor.cbCrStride();
+ ycbcrData.mPictureRect = aDescriptor.display();
+ ycbcrData.mYUVColorSpace = aDescriptor.yUVColorSpace();
+ ycbcrData.mColorDepth = aDescriptor.colorDepth();
+ ycbcrData.mChromaSubsampling = aDescriptor.chromaSubsampling();
+
+ gfx::ConvertYCbCrToRGB(ycbcrData, aDestFormat, aDestSize, aDestBuffer,
+ aStride);
+}
+
+gfx::IntSize GetCroppedCbCrSize(const YCbCrDescriptor& aDescriptor) {
+ return ChromaSize(aDescriptor.display().Size(),
+ aDescriptor.chromaSubsampling());
+}
+
+} // namespace ImageDataSerializer
+} // namespace layers
+} // namespace mozilla