/****************************************************************************

**

** Copyright (C) 2009 Nokia Corporation and/or its subsidiary(-ies).

** All rights reserved.

** Contact: Nokia Corporation (qt-info@nokia.com)

**

** This file is part of the test suite of the Qt Toolkit.

**

** $QT_BEGIN_LICENSE:LGPL$

** No Commercial Usage

** This file contains pre-release code and may not be distributed.

** You may use this file in accordance with the terms and conditions

** contained in the Technology Preview License Agreement accompanying

** this package.

**

** GNU Lesser General Public License Usage

** Alternatively, this file may be used under the terms of the GNU Lesser

** General Public License version 2.1 as published by the Free Software

** Foundation and appearing in the file LICENSE.LGPL included in the

** packaging of this file.  Please review the following information to

** ensure the GNU Lesser General Public License version 2.1 requirements

** will be met: http://www.gnu.org/licenses/old-licenses/lgpl-2.1.html.

**

** In addition, as a special exception, Nokia gives you certain additional

** rights.  These rights are described in the Nokia Qt LGPL Exception

** version 1.1, included in the file LGPL_EXCEPTION.txt in this package.

**

** If you have questions regarding the use of this file, please contact

** Nokia at qt-info@nokia.com.

**

**

**

**

**

**

**

**

** $QT_END_LICENSE$

**

****************************************************************************/

#include <QtTest/QtTest>

#include <QtMultimedia/QVideoFrame>

#include <QtGui/QImage>

#include <QtCore/QPointer>

class tst_QVideoFrame : public QObject

{

    Q_OBJECT

public:

    tst_QVideoFrame();

    ~tst_QVideoFrame();

public slots:

    void initTestCase();

    void cleanupTestCase();

    void init();

    void cleanup();

private slots:

    void create_data();

    void create();

    void createInvalid_data();

    void createInvalid();

    void createFromBuffer_data();

    void createFromBuffer();

    void createFromImage_data();

    void createFromImage();

    void createFromIncompatibleImage();

    void createNull();

    void destructor();

    void copy_data();

    void copy();

    void assign_data();

    void assign();

    void map_data();

    void map();

    void mapImage_data();

    void mapImage();

    void imageDetach();

};

Q_DECLARE_METATYPE(QImage::Format)

Q_DECLARE_METATYPE(QVideoFrame)

class QtTestVideoBuffer : public QObject, public QAbstractVideoBuffer

{

    Q_OBJECT

public:

    QtTestVideoBuffer()

        : QAbstractVideoBuffer(NoHandle) {}

    explicit QtTestVideoBuffer(QAbstractVideoBuffer::HandleType type)

        : QAbstractVideoBuffer(type) {}

    MapMode mapMode() const { return NotMapped; }

    uchar *map(MapMode, int *, int *) { return 0; }

    void unmap() {}

};

tst_QVideoFrame::tst_QVideoFrame()

{

}

tst_QVideoFrame::~tst_QVideoFrame()

{

}

void tst_QVideoFrame::initTestCase()

{

}

void tst_QVideoFrame::cleanupTestCase()

{

}

void tst_QVideoFrame::init()

{

}

void tst_QVideoFrame::cleanup()

{

}

void tst_QVideoFrame::create_data()

{

    QTest::addColumn<QSize>("size");

    QTest::addColumn<QVideoFrame::PixelFormat>("pixelFormat");

    QTest::addColumn<int>("bytes");

    QTest::addColumn<int>("bytesPerLine");

    QTest::newRow("64x64 ARGB32")

            << QSize(64, 64)

            << QVideoFrame::Format_ARGB32

            << 16384

            << 256;

    QTest::newRow("32x256 YUV420P")

            << QSize(32, 256)

            << QVideoFrame::Format_YUV420P

            << 13288

            << 32;

}

void tst_QVideoFrame::create()

{

    QFETCH(QSize, size);

    QFETCH(QVideoFrame::PixelFormat, pixelFormat);

    QFETCH(int, bytes);

    QFETCH(int, bytesPerLine);

    QVideoFrame frame(bytes, size, bytesPerLine, pixelFormat);

    QVERIFY(frame.isValid());

    QCOMPARE(frame.handleType(), QAbstractVideoBuffer::NoHandle);

    QCOMPARE(frame.pixelFormat(), pixelFormat);

    QCOMPARE(frame.size(), size);

    QCOMPARE(frame.width(), size.width());

    QCOMPARE(frame.height(), size.height());

    QCOMPARE(frame.fieldType(), QVideoFrame::ProgressiveFrame);

    QCOMPARE(frame.startTime(), qint64(-1));

    QCOMPARE(frame.endTime(), qint64(-1));

}

void tst_QVideoFrame::createInvalid_data()

{

    QTest::addColumn<QSize>("size");

    QTest::addColumn<QVideoFrame::PixelFormat>("pixelFormat");

    QTest::addColumn<int>("bytes");

    QTest::addColumn<int>("bytesPerLine");

    QTest::newRow("64x64 ARGB32 0 size")

            << QSize(64, 64)

            << QVideoFrame::Format_ARGB32

            << 0

            << 45;

    QTest::newRow("32x256 YUV420P negative size")

            << QSize(32, 256)

            << QVideoFrame::Format_YUV420P

            << -13288

            << 32;

}

void tst_QVideoFrame::createInvalid()

{

    QFETCH(QSize, size);

    QFETCH(QVideoFrame::PixelFormat, pixelFormat);

    QFETCH(int, bytes);

    QFETCH(int, bytesPerLine);

    QVideoFrame frame(bytes, size, bytesPerLine, pixelFormat);

    QVERIFY(!frame.isValid());

    QCOMPARE(frame.handleType(), QAbstractVideoBuffer::NoHandle);

    QCOMPARE(frame.pixelFormat(), pixelFormat);

    QCOMPARE(frame.size(), size);

    QCOMPARE(frame.width(), size.width());

    QCOMPARE(frame.height(), size.height());

    QCOMPARE(frame.fieldType(), QVideoFrame::ProgressiveFrame);

    QCOMPARE(frame.startTime(), qint64(-1));

    QCOMPARE(frame.endTime(), qint64(-1));

}

void tst_QVideoFrame::createFromBuffer_data()

{

    QTest::addColumn<QAbstractVideoBuffer::HandleType>("handleType");

    QTest::addColumn<QSize>("size");

    QTest::addColumn<QVideoFrame::PixelFormat>("pixelFormat");

    QTest::newRow("64x64 ARGB32 no handle")

            << QAbstractVideoBuffer::NoHandle

            << QSize(64, 64)

            << QVideoFrame::Format_ARGB32;

    QTest::newRow("64x64 ARGB32 gl handle")

            << QAbstractVideoBuffer::GLTextureHandle

            << QSize(64, 64)

            << QVideoFrame::Format_ARGB32;

    QTest::newRow("64x64 ARGB32 user handle")

            << QAbstractVideoBuffer::UserHandle

            << QSize(64, 64)

            << QVideoFrame::Format_ARGB32;

}

void tst_QVideoFrame::createFromBuffer()

{

    QFETCH(QAbstractVideoBuffer::HandleType, handleType);

    QFETCH(QSize, size);

    QFETCH(QVideoFrame::PixelFormat, pixelFormat);

    QVideoFrame frame(new QtTestVideoBuffer(handleType), size, pixelFormat);

    QVERIFY(frame.isValid());

    QCOMPARE(frame.handleType(), handleType);

    QCOMPARE(frame.pixelFormat(), pixelFormat);

    QCOMPARE(frame.size(), size);

    QCOMPARE(frame.width(), size.width());

    QCOMPARE(frame.height(), size.height());

    QCOMPARE(frame.fieldType(), QVideoFrame::ProgressiveFrame);

    QCOMPARE(frame.startTime(), qint64(-1));

    QCOMPARE(frame.endTime(), qint64(-1));

}

void tst_QVideoFrame::createFromImage_data()

{

    QTest::addColumn<QSize>("size");

    QTest::addColumn<QImage::Format>("imageFormat");

    QTest::addColumn<QVideoFrame::PixelFormat>("pixelFormat");

    QTest::newRow("64x64 RGB32")

            << QSize(64, 64)

            << QImage::Format_RGB32

            << QVideoFrame::Format_RGB32;

    QTest::newRow("12x45 RGB16")

            << QSize(12, 45)

            << QImage::Format_RGB16

            << QVideoFrame::Format_RGB565;

    QTest::newRow("19x46 ARGB32_Premultiplied")

            << QSize(19, 46)

            << QImage::Format_ARGB32_Premultiplied

            << QVideoFrame::Format_ARGB32_Premultiplied;

}

void tst_QVideoFrame::createFromImage()

{

    QFETCH(QSize, size);

    QFETCH(QImage::Format, imageFormat);

    QFETCH(QVideoFrame::PixelFormat, pixelFormat);

    const QImage image(size.width(), size.height(), imageFormat);

    QVideoFrame frame(image);

    QVERIFY(frame.isValid());

    QCOMPARE(frame.handleType(), QAbstractVideoBuffer::NoHandle);

    QCOMPARE(frame.pixelFormat(), pixelFormat);

    QCOMPARE(frame.size(), size);

    QCOMPARE(frame.width(), size.width());

    QCOMPARE(frame.height(), size.height());

    QCOMPARE(frame.fieldType(), QVideoFrame::ProgressiveFrame);

    QCOMPARE(frame.startTime(), qint64(-1));

    QCOMPARE(frame.endTime(), qint64(-1));

}

void tst_QVideoFrame::createFromIncompatibleImage()

{

    const QImage image(64, 64, QImage::Format_Mono);

    QVideoFrame frame(image);

    QVERIFY(!frame.isValid());

    QCOMPARE(frame.handleType(), QAbstractVideoBuffer::NoHandle);

    QCOMPARE(frame.pixelFormat(), QVideoFrame::Format_Invalid);

    QCOMPARE(frame.size(), QSize(64, 64));

    QCOMPARE(frame.width(), 64);

    QCOMPARE(frame.height(), 64);

    QCOMPARE(frame.fieldType(), QVideoFrame::ProgressiveFrame);

    QCOMPARE(frame.startTime(), qint64(-1));

    QCOMPARE(frame.endTime(), qint64(-1));

}

void tst_QVideoFrame::createNull()

{

    QVideoFrame frame;

    QVERIFY(!frame.isValid());

    QCOMPARE(frame.handleType(), QAbstractVideoBuffer::NoHandle);

    QCOMPARE(frame.pixelFormat(), QVideoFrame::Format_Invalid);

    QCOMPARE(frame.size(), QSize());

    QCOMPARE(frame.width(), -1);

    QCOMPARE(frame.height(), -1);

    QCOMPARE(frame.fieldType(), QVideoFrame::ProgressiveFrame);

    QCOMPARE(frame.startTime(), qint64(-1));

    QCOMPARE(frame.endTime(), qint64(-1));

}

void tst_QVideoFrame::destructor()

{

    QPointer<QtTestVideoBuffer> buffer = new QtTestVideoBuffer;

    {

        QVideoFrame frame(buffer, QSize(4, 1), QVideoFrame::Format_ARGB32);

    }

    QVERIFY(buffer.isNull());

}

void tst_QVideoFrame::copy_data()

{

    QTest::addColumn<QAbstractVideoBuffer::HandleType>("handleType");

    QTest::addColumn<QSize>("size");

    QTest::addColumn<QVideoFrame::PixelFormat>("pixelFormat");

    QTest::addColumn<QVideoFrame::FieldType>("fieldType");

    QTest::addColumn<qint64>("startTime");

    QTest::addColumn<qint64>("endTime");

    QTest::newRow("64x64 ARGB32")

            << QAbstractVideoBuffer::GLTextureHandle

            << QSize(64, 64)

            << QVideoFrame::Format_ARGB32

            << QVideoFrame::TopField

            << qint64(63641740)

            << qint64(63641954);

    QTest::newRow("32x256 YUV420P")

            << QAbstractVideoBuffer::UserHandle

            << QSize(32, 256)

            << QVideoFrame::Format_YUV420P

            << QVideoFrame::InterlacedFrame

            << qint64(12345)

            << qint64(12389);

}

void tst_QVideoFrame::copy()

{

    QFETCH(QAbstractVideoBuffer::HandleType, handleType);

    QFETCH(QSize, size);

    QFETCH(QVideoFrame::PixelFormat, pixelFormat);

    QFETCH(QVideoFrame::FieldType, fieldType);

    QFETCH(qint64, startTime);

    QFETCH(qint64, endTime);

    QPointer<QtTestVideoBuffer> buffer = new QtTestVideoBuffer(handleType);

    {

        QVideoFrame frame(buffer, size, pixelFormat);

        frame.setFieldType(QVideoFrame::FieldType(fieldType));

        frame.setStartTime(startTime);

        frame.setEndTime(endTime);

        QVERIFY(frame.isValid());

        QCOMPARE(frame.handleType(), handleType);

        QCOMPARE(frame.pixelFormat(), pixelFormat);

        QCOMPARE(frame.size(), size);

        QCOMPARE(frame.width(), size.width());

        QCOMPARE(frame.height(), size.height());

        QCOMPARE(frame.fieldType(), fieldType);

        QCOMPARE(frame.startTime(), startTime);

        QCOMPARE(frame.endTime(), endTime);

        {

            QVideoFrame otherFrame(frame);

            QVERIFY(!buffer.isNull());

            QVERIFY(otherFrame.isValid());

            QCOMPARE(otherFrame.handleType(), handleType);

            QCOMPARE(otherFrame.pixelFormat(), pixelFormat);

            QCOMPARE(otherFrame.size(), size);

            QCOMPARE(otherFrame.width(), size.width());

            QCOMPARE(otherFrame.height(), size.height());

            QCOMPARE(otherFrame.fieldType(), fieldType);

            QCOMPARE(otherFrame.startTime(), startTime);

            QCOMPARE(otherFrame.endTime(), endTime);

            otherFrame.setEndTime(-1);

            QVERIFY(!buffer.isNull());

            QVERIFY(otherFrame.isValid());    

            QCOMPARE(otherFrame.handleType(), handleType);

            QCOMPARE(otherFrame.pixelFormat(), pixelFormat);

            QCOMPARE(otherFrame.size(), size);

            QCOMPARE(otherFrame.width(), size.width());

            QCOMPARE(otherFrame.height(), size.height());

            QCOMPARE(otherFrame.fieldType(), fieldType);

            QCOMPARE(otherFrame.startTime(), startTime);

            QCOMPARE(otherFrame.endTime(), qint64(-1));

        }

        QVERIFY(!buffer.isNull());

        QVERIFY(frame.isValid());

        QCOMPARE(frame.handleType(), handleType);

        QCOMPARE(frame.pixelFormat(), pixelFormat);

        QCOMPARE(frame.size(), size);

        QCOMPARE(frame.width(), size.width());

        QCOMPARE(frame.height(), size.height());

        QCOMPARE(frame.fieldType(), fieldType);

        QCOMPARE(frame.startTime(), startTime);

        QCOMPARE(frame.endTime(), qint64(-1));  // Explicitly shared.

    }

    QVERIFY(buffer.isNull());

}

void tst_QVideoFrame::assign_data()

{

    QTest::addColumn<QAbstractVideoBuffer::HandleType>("handleType");

    QTest::addColumn<QSize>("size");

    QTest::addColumn<QVideoFrame::PixelFormat>("pixelFormat");

    QTest::addColumn<QVideoFrame::FieldType>("fieldType");

    QTest::addColumn<qint64>("startTime");

    QTest::addColumn<qint64>("endTime");

    QTest::newRow("64x64 ARGB32")

            << QAbstractVideoBuffer::GLTextureHandle

            << QSize(64, 64)

            << QVideoFrame::Format_ARGB32

            << QVideoFrame::TopField

            << qint64(63641740)

            << qint64(63641954);

    QTest::newRow("32x256 YUV420P")

            << QAbstractVideoBuffer::UserHandle

            << QSize(32, 256)

            << QVideoFrame::Format_YUV420P

            << QVideoFrame::InterlacedFrame

            << qint64(12345)

            << qint64(12389);

}

void tst_QVideoFrame::assign()

{

    QFETCH(QAbstractVideoBuffer::HandleType, handleType);

    QFETCH(QSize, size);

    QFETCH(QVideoFrame::PixelFormat, pixelFormat);

    QFETCH(QVideoFrame::FieldType, fieldType);

    QFETCH(qint64, startTime);

    QFETCH(qint64, endTime);

    QPointer<QtTestVideoBuffer> buffer = new QtTestVideoBuffer(handleType);

    QVideoFrame frame;

    {

        QVideoFrame otherFrame(buffer, size, pixelFormat);

        otherFrame.setFieldType(fieldType);

        otherFrame.setStartTime(startTime);

        otherFrame.setEndTime(endTime);

        frame = otherFrame;

        QVERIFY(!buffer.isNull());

        QVERIFY(otherFrame.isValid());

        QCOMPARE(otherFrame.handleType(), handleType);

        QCOMPARE(otherFrame.pixelFormat(), pixelFormat);

        QCOMPARE(otherFrame.size(), size);

        QCOMPARE(otherFrame.width(), size.width());

        QCOMPARE(otherFrame.height(), size.height());

        QCOMPARE(otherFrame.fieldType(), fieldType);

        QCOMPARE(otherFrame.startTime(), startTime);

        QCOMPARE(otherFrame.endTime(), endTime);

        otherFrame.setStartTime(-1);

        QVERIFY(!buffer.isNull());

        QVERIFY(otherFrame.isValid());

        QCOMPARE(otherFrame.handleType(), handleType);

        QCOMPARE(otherFrame.pixelFormat(), pixelFormat);

        QCOMPARE(otherFrame.size(), size);

        QCOMPARE(otherFrame.width(), size.width());

        QCOMPARE(otherFrame.height(), size.height());

        QCOMPARE(otherFrame.fieldType(), fieldType);

        QCOMPARE(otherFrame.startTime(), qint64(-1));

        QCOMPARE(otherFrame.endTime(), endTime);

    }

    QVERIFY(!buffer.isNull());

    QVERIFY(frame.isValid());

    QCOMPARE(frame.handleType(), handleType);

    QCOMPARE(frame.pixelFormat(), pixelFormat);

    QCOMPARE(frame.size(), size);

    QCOMPARE(frame.width(), size.width());

    QCOMPARE(frame.height(), size.height());

    QCOMPARE(frame.fieldType(), fieldType);

    QCOMPARE(frame.startTime(), qint64(-1));

    QCOMPARE(frame.endTime(), endTime);

    frame = QVideoFrame();

    QVERIFY(buffer.isNull());

    QVERIFY(!frame.isValid());

    QCOMPARE(frame.handleType(), QAbstractVideoBuffer::NoHandle);

    QCOMPARE(frame.pixelFormat(), QVideoFrame::Format_Invalid);

    QCOMPARE(frame.size(), QSize());

    QCOMPARE(frame.width(), -1);

    QCOMPARE(frame.height(), -1);

    QCOMPARE(frame.fieldType(), QVideoFrame::ProgressiveFrame);

    QCOMPARE(frame.startTime(), qint64(-1));

    QCOMPARE(frame.endTime(), qint64(-1));

}

void tst_QVideoFrame::map_data()

{

    QTest::addColumn<QSize>("size");

    QTest::addColumn<int>("numBytes");

    QTest::addColumn<int>("bytesPerLine");