Revision control
++ // Copyright (C) 2023 The Qt Company Ltd.
++ // SPDX-License-Identifier: LicenseRef-Qt-Commercial OR LGPL-3.0-only OR GPL-2.0-only OR GPL-3.0-only
++
++ #include "qimagevideobuffer_p.h"
++ #include "qvideoframeformat.h"
++
++ QT_BEGIN_NAMESPACE
++
++ namespace {
++
++ QImage::Format fixImageFormat(QImage::Format format)
++ {
++ switch (format) {
++ case QImage::Format_ARGB32_Premultiplied:
++ case QImage::Format_ARGB8565_Premultiplied:
++ case QImage::Format_ARGB6666_Premultiplied:
++ case QImage::Format_ARGB8555_Premultiplied:
++ case QImage::Format_ARGB4444_Premultiplied:
++ case QImage::Format_RGBA8888_Premultiplied:
++ case QImage::Format_A2BGR30_Premultiplied:
++ case QImage::Format_A2RGB30_Premultiplied:
++ case QImage::Format_RGBA64_Premultiplied:
++ case QImage::Format_RGBA16FPx4_Premultiplied:
++ case QImage::Format_RGBA32FPx4_Premultiplied:
++ return QImage::Format_ARGB32_Premultiplied;
++ case QImage::Format_ARGB32:
++ case QImage::Format_RGBA8888:
++ case QImage::Format_Alpha8:
++ case QImage::Format_RGBA64:
++ case QImage::Format_RGBA16FPx4:
++ case QImage::Format_RGBA32FPx4:
++ return QImage::Format_ARGB32;
++ case QImage::Format_Invalid:
++ return QImage::Format_Invalid;
++ default:
++ return QImage::Format_RGB32;
++ }
++ }
++
++ QImage fixImage(QImage image)
++ {
++ const auto imageFormat = image.format();
++ if (imageFormat == QImage::Format_Invalid)
++ return image;
++
++ const auto frameFormat = QVideoFrameFormat::pixelFormatFromImageFormat(imageFormat);
++ if (frameFormat != QVideoFrameFormat::Format_Invalid)
++ return image;
++
++ return std::move(image).convertToFormat(fixImageFormat(imageFormat));
++ }
++
++ } // namespace
++
++ QImageVideoBuffer::QImageVideoBuffer(QImage image) : m_image(fixImage(std::move(image))) { }
++
++ QAbstractVideoBuffer::MapData QImageVideoBuffer::map(QVideoFrame::MapMode mode)
++ {
++ MapData mapData;
++
++ if (!m_image.isNull()) {
++ mapData.planeCount = 1;
++ mapData.bytesPerLine[0] = m_image.bytesPerLine();
++ if (mode == QVideoFrame::ReadOnly)
++ mapData.data[0] = const_cast<uint8_t *>(m_image.constBits());
++ else
++ mapData.data[0] = m_image.bits();
++ mapData.dataSize[0] = m_image.sizeInBytes();
++ }
++
++ return mapData;
++ }
++
++ QImage QImageVideoBuffer::underlyingImage() const
++ {
++ return m_image;
++ }
++
++ QT_END_NAMESPACE