/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */

/* This Source Code Form is subject to the terms of the Mozilla Public

 * License, v. 2.0. If a copy of the MPL was not distributed with this

 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

#include <gtest/gtest.h>

#include <stdint.h>

#include "VideoUtils.h"

using namespace mozilla;

struct TestData

{

  const char16_t* const mCodecParameterString;

  const bool mExpectedValue;

};

TEST(ExtractVPXCodecDetails, TestDataLength) {

  TestData tests[] =

  {

    {u"vp09.00.11.08", true }, // valid case

    {u"vp09.00.11.08.00", true }, // valid case, have extra optional field

    {u"vp09.02.10.10.01.09.16.09.01", true}, // maximum length valid case

    {u"vp09", false }, // lack of mandatory fields

    {u"vp09.00", false }, // lack of mandatory fields

    {u"vp09.00.11", false }, // lack of mandatory fields

    {u"vp09.02.10.10.01.09.16.09.01.00", false} // more than 9 fields, invalid case.

  };

  for (const auto& data : tests) {

    uint8_t profile = 0;

    uint8_t level = 0;

    uint8_t bitDepth = 0;

    bool result = ExtractVPXCodecDetails(nsString(data.mCodecParameterString), profile, level, bitDepth);

    EXPECT_EQ(result, data.mExpectedValue) << NS_ConvertUTF16toUTF8(data.mCodecParameterString).get();

  }

}

TEST(ExtractVPXCodecDetails, TestInputData) {

  TestData tests[] =

  {

    {u"vp09.02..08", false}, // malformed

    {u"vp9.02.10.08", false}, // invalid 4CC

    {u"vp09.03.11.08", false }, // profile should < 3

    {u"vp09.00.63.08.00", false }, // invalid level

    {u"vp09.02.10.13", false}, // invalid bitDepth

    {u"vp09.02.10.10.04", false}, // invalid chromasubsampling, should < 4

    {u"vp09.02.10.10.01.00", false}, // invalid Colour primaries, should not be 0,3 or < 23.

    {u"vp09.02.10.10.01.03", false}, // invalid Colour primaries.

    {u"vp09.02.10.10.01.23", false}, // invalid Colour primaries.

    {u"vp09.02.10.10.01.09.00", false}, // invalid Transfer characteristics, should not be 0,3 or < 19.

    {u"vp09.02.10.10.01.09.03", false}, // invalid Transfer characteristics.

    {u"vp09.02.10.10.01.09.19", false}, // invalid Transfer characteristics.

    {u"vp09.02.10.10.01.09.16.12", false}, // invalid Matrix coefficients, should not be 3 or < 12.

    {u"vp09.02.10.10.01.09.16.03", false}, // invalid matrix.

    {u"vp09.02.10.10.01.09.16.09.02", false}, // invalid range, should < 2.

    // Test if matrixCoefficients is 0 (RGB), then chroma subsampling MUST be 3 (4:4:4).

    {u"vp09.02.10.08.03.09.16.00.00", true} // invalid combination.

  };

  for (const auto& data : tests) {

    uint8_t profile = 0;

    uint8_t level = 0;

    uint8_t bitDepth = 0;

    bool result = ExtractVPXCodecDetails(nsString(data.mCodecParameterString), profile, level, bitDepth);

    EXPECT_EQ(result, data.mExpectedValue) << NS_ConvertUTF16toUTF8(data.mCodecParameterString).get();

  }

}

TEST(ExtractVPXCodecDetails, TestParsingOutput) {

    uint8_t profile = 0;

    uint8_t level = 0;

    uint8_t bitDepth = 0;

    uint8_t chromaSubsampling = 0;

    VideoColorSpace colorSpace;

    auto data = u"vp09.01.11.08";

    bool result = ExtractVPXCodecDetails(nsString(data),

                                         profile,

                                         level,

                                         bitDepth,

                                         chromaSubsampling,

                                         colorSpace);

    EXPECT_EQ(result, true);

    EXPECT_EQ(profile, 1);

    EXPECT_EQ(level, 11);

    EXPECT_EQ(bitDepth, 8);

    // Should keep spec defined default value.

    EXPECT_EQ(chromaSubsampling, 1);

    EXPECT_EQ(colorSpace.mPrimaryId, 1);

    EXPECT_EQ(colorSpace.mTransferId, 1);

    EXPECT_EQ(colorSpace.mMatrixId, 1);

    EXPECT_EQ(colorSpace.mRangeId, 0);

    data = u"vp09.02.10.10.01.09.16.09.01";

    result = ExtractVPXCodecDetails(nsString(data),

                                    profile,

                                    level,

                                    bitDepth,

                                    chromaSubsampling,

                                    colorSpace);

    EXPECT_EQ(result, true);

    EXPECT_EQ(profile, 2);

    EXPECT_EQ(level, 10);

    EXPECT_EQ(bitDepth, 10);

    EXPECT_EQ(chromaSubsampling, 1);

    EXPECT_EQ(colorSpace.mPrimaryId, 9);

    EXPECT_EQ(colorSpace.mTransferId, 16);

    EXPECT_EQ(colorSpace.mMatrixId, 9);

    EXPECT_EQ(colorSpace.mRangeId, 1);

}