summaryrefslogtreecommitdiffstats
path: root/tests/auto/unit/mockbackend/qmockvideobuffer.h
blob: 033dd2cfb70c5beb334478c5f99cbb29eeebfbce (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
// Copyright (C) 2021 The Qt Company Ltd.
// SPDX-License-Identifier: LicenseRef-Qt-Commercial OR GPL-3.0-only

#ifndef QMOCKVIDEOBUFFER_H
#define QMOCKVIDEOBUFFER_H

#include "qimage.h"
#include "private/qhwvideobuffer_p.h"

class QMockVideoBuffer : public QHwVideoBuffer
{
public:
    QMockVideoBuffer(QImage image) : QHwVideoBuffer(QVideoFrame::NoHandle), m_image(image) { }

    MapData map(QVideoFrame::MapMode mode) override
    {
        MapData mapData;
        if (m_mapMode == QVideoFrame::NotMapped && !m_image.isNull()
            && mode != QVideoFrame::NotMapped) {
            m_mapMode = mode;

            mapData.planeCount = 1;
            mapData.bytesPerLine[0] = m_image.bytesPerLine();
            mapData.data[0] = m_image.bits();
            mapData.dataSize[0] = m_image.sizeInBytes();
        }

        return mapData;
    }

    void unmap() override { m_mapMode = QVideoFrame::NotMapped; }

private:
    QVideoFrame::MapMode m_mapMode = QVideoFrame::NotMapped;
    QImage m_image;
};

#endif // QMOCKVIDEOBUFFER_H