1 /****************************************************************************
2 **
3 ** Copyright (C) 2015 The Qt Company Ltd.
4 ** Contact: http://www.qt.io/licensing/
5 **
6 ** This file is part of the examples of the Qt Toolkit.
7 **
8 ** $QT_BEGIN_LICENSE:BSD$
9 ** You may use this file under the terms of the BSD license as follows:
10 **
11 ** "Redistribution and use in source and binary forms, with or without
12 ** modification, are permitted provided that the following conditions are
13 ** met:
14 ** * Redistributions of source code must retain the above copyright
15 ** notice, this list of conditions and the following disclaimer.
16 ** * Redistributions in binary form must reproduce the above copyright
17 ** notice, this list of conditions and the following disclaimer in
18 ** the documentation and/or other materials provided with the
19 ** distribution.
20 ** * Neither the name of The Qt Company Ltd nor the names of its
21 ** contributors may be used to endorse or promote products derived
22 ** from this software without specific prior written permission.
23 **
24 **
25 ** THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
26 ** "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
27 ** LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
28 ** A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
29 ** OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
30 ** SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
31 ** LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
32 ** DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
33 ** THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
34 ** (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
35 ** OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE."
36 **
37 ** $QT_END_LICENSE$
38 **
39 ****************************************************************************/
40
41 #include "videowidgetsurface.h"
42
43 #include <QtMultimedia>
44
VideoWidgetSurface(QWidget * widget,QObject * parent)45 VideoWidgetSurface::VideoWidgetSurface(QWidget *widget, QObject *parent)
46 : QAbstractVideoSurface(parent)
47 , widget(widget)
48 , imageFormat(QImage::Format_Invalid)
49 {
50 }
51
52 //! [0]
supportedPixelFormats(QAbstractVideoBuffer::HandleType handleType) const53 QList<QVideoFrame::PixelFormat> VideoWidgetSurface::supportedPixelFormats(
54 QAbstractVideoBuffer::HandleType handleType) const
55 {
56 if (handleType == QAbstractVideoBuffer::NoHandle) {
57 return QList<QVideoFrame::PixelFormat>()
58 << QVideoFrame::Format_RGB32
59 << QVideoFrame::Format_ARGB32
60 << QVideoFrame::Format_ARGB32_Premultiplied
61 << QVideoFrame::Format_RGB565
62 << QVideoFrame::Format_RGB555;
63 } else {
64 return QList<QVideoFrame::PixelFormat>();
65 }
66 }
67 //! [0]
68
69 //! [1]
isFormatSupported(const QVideoSurfaceFormat & format,QVideoSurfaceFormat * similar) const70 bool VideoWidgetSurface::isFormatSupported(
71 const QVideoSurfaceFormat &format, QVideoSurfaceFormat *similar) const
72 {
73 Q_UNUSED(similar);
74
75 const QImage::Format imageFormat = QVideoFrame::imageFormatFromPixelFormat(format.pixelFormat());
76 const QSize size = format.frameSize();
77
78 return imageFormat != QImage::Format_Invalid
79 && !size.isEmpty()
80 && format.handleType() == QAbstractVideoBuffer::NoHandle;
81 }
82 //! [1]
83
84 //! [2]
start(const QVideoSurfaceFormat & format)85 bool VideoWidgetSurface::start(const QVideoSurfaceFormat &format)
86 {
87 const QImage::Format imageFormat = QVideoFrame::imageFormatFromPixelFormat(format.pixelFormat());
88 const QSize size = format.frameSize();
89
90 if (imageFormat != QImage::Format_Invalid && !size.isEmpty()) {
91 this->imageFormat = imageFormat;
92 imageSize = size;
93 sourceRect = format.viewport();
94
95 QAbstractVideoSurface::start(format);
96
97 widget->updateGeometry();
98 updateVideoRect();
99
100 return true;
101 } else {
102 return false;
103 }
104 }
105 //! [2]
106
107 //! [3]
stop()108 void VideoWidgetSurface::stop()
109 {
110 currentFrame = QVideoFrame();
111 targetRect = QRect();
112
113 QAbstractVideoSurface::stop();
114
115 widget->update();
116 }
117 //! [3]
118
119 //! [4]
present(const QVideoFrame & frame)120 bool VideoWidgetSurface::present(const QVideoFrame &frame)
121 {
122 if (surfaceFormat().pixelFormat() != frame.pixelFormat()
123 || surfaceFormat().frameSize() != frame.size()) {
124 setError(IncorrectFormatError);
125 stop();
126
127 return false;
128 } else {
129 currentFrame = frame;
130
131 widget->repaint(targetRect);
132
133 return true;
134 }
135 }
136 //! [4]
137
138 //! [5]
updateVideoRect()139 void VideoWidgetSurface::updateVideoRect()
140 {
141 QSize size = surfaceFormat().sizeHint();
142 size.scale(widget->size().boundedTo(size), Qt::KeepAspectRatio);
143
144 targetRect = QRect(QPoint(0, 0), size);
145 targetRect.moveCenter(widget->rect().center());
146 }
147 //! [5]
148
149 //! [6]
paint(QPainter * painter)150 void VideoWidgetSurface::paint(QPainter *painter)
151 {
152 if (currentFrame.map(QAbstractVideoBuffer::ReadOnly)) {
153 const QTransform oldTransform = painter->transform();
154
155 if (surfaceFormat().scanLineDirection() == QVideoSurfaceFormat::BottomToTop) {
156 painter->scale(1, -1);
157 painter->translate(0, -widget->height());
158 }
159
160 QImage image(
161 currentFrame.bits(),
162 currentFrame.width(),
163 currentFrame.height(),
164 currentFrame.bytesPerLine(),
165 imageFormat);
166
167 painter->drawImage(targetRect, image, sourceRect);
168
169 painter->setTransform(oldTransform);
170
171 currentFrame.unmap();
172 }
173 }
174 //! [6]
175