1// Copyright (C) 2023 The Qt Company Ltd.
2// SPDX-License-Identifier: LicenseRef-Qt-Commercial OR LGPL-3.0-only OR GPL-2.0-only OR GPL-3.0-only
3
4#include "qimagevideobuffer_p.h"
5#include "qvideoframeformat.h"
6
7QT_BEGIN_NAMESPACE
8
9namespace {
10
11QImage::Format fixImageFormat(QImage::Format format)
12{
13 switch (format) {
14 case QImage::Format_ARGB32_Premultiplied:
15 case QImage::Format_ARGB8565_Premultiplied:
16 case QImage::Format_ARGB6666_Premultiplied:
17 case QImage::Format_ARGB8555_Premultiplied:
18 case QImage::Format_ARGB4444_Premultiplied:
19 case QImage::Format_RGBA8888_Premultiplied:
20 case QImage::Format_A2BGR30_Premultiplied:
21 case QImage::Format_A2RGB30_Premultiplied:
22 case QImage::Format_RGBA64_Premultiplied:
23 case QImage::Format_RGBA16FPx4_Premultiplied:
24 case QImage::Format_RGBA32FPx4_Premultiplied:
25 return QImage::Format_ARGB32_Premultiplied;
26 case QImage::Format_ARGB32:
27 case QImage::Format_RGBA8888:
28 case QImage::Format_Alpha8:
29 case QImage::Format_RGBA64:
30 case QImage::Format_RGBA16FPx4:
31 case QImage::Format_RGBA32FPx4:
32 return QImage::Format_ARGB32;
33 case QImage::Format_Invalid:
34 return QImage::Format_Invalid;
35 default:
36 return QImage::Format_RGB32;
37 }
38}
39
40QImage fixImage(QImage image)
41{
42 if (image.format() == QImage::Format_Invalid)
43 return image;
44
45 const auto frameFormat = QVideoFrameFormat::pixelFormatFromImageFormat(format: image.format());
46 if (frameFormat != QVideoFrameFormat::Format_Invalid)
47 return image;
48
49 return image.convertToFormat(f: fixImageFormat(format: image.format()));
50}
51
52} // namespace
53
54QImageVideoBuffer::QImageVideoBuffer(QImage image) : m_image(fixImage(image: std::move(image))) { }
55
56QAbstractVideoBuffer::MapData QImageVideoBuffer::map(QVideoFrame::MapMode mode)
57{
58 MapData mapData;
59
60 if (!m_image.isNull()) {
61 mapData.planeCount = 1;
62 mapData.bytesPerLine[0] = m_image.bytesPerLine();
63 if (mode == QVideoFrame::ReadOnly)
64 mapData.data[0] = const_cast<uint8_t *>(m_image.constBits());
65 else
66 mapData.data[0] = m_image.bits();
67 mapData.dataSize[0] = m_image.sizeInBytes();
68 }
69
70 return mapData;
71}
72
73QImage QImageVideoBuffer::underlyingImage() const
74{
75 return m_image;
76}
77
78QT_END_NAMESPACE
79

Provided by KDAB

Privacy Policy
Learn Advanced QML with KDAB
Find out more

source code of qtmultimedia/src/multimedia/video/qimagevideobuffer.cpp