89 lines
3.2 KiB
C++
89 lines
3.2 KiB
C++
/* This file is a part of plaYUVer project
|
|
* Copyright (C) 2014-2021 by Joao Carreira (jfmcarreira@gmail.com)
|
|
* Luis Lucas (luisfrlucas@gmail.com)
|
|
*
|
|
* This program is free software; you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation; either version 2 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License along
|
|
* with this program; if not, write to the Free Software Foundation, Inc.,
|
|
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
|
|
*/
|
|
|
|
/**
|
|
* \file CalypFrameQualityTests.cpp
|
|
* \brief CalypFrame quality metrics tests
|
|
*/
|
|
|
|
#include "CalypFrame.h"
|
|
#include "CalypStream.h"
|
|
#include "gtest/gtest.h"
|
|
|
|
class CalypFrameQualityTest : public ::testing::Test
|
|
{
|
|
protected:
|
|
// Per-test-case set-up.
|
|
// Called before the first test in this test case.
|
|
// Can be omitted if not needed.
|
|
static void SetUpTestCase()
|
|
{
|
|
pcStreamPast = std::make_unique<CalypStream>();
|
|
pcStreamPast->open( std::string( CALYP_TEST_DATA_DIR ) + std::string( "/BasketballDrill_F10_832x480_yuv420p.yuv" ), 832, 480, ClpPixelFormats::YUV420p, 8,
|
|
CLP_BIG_ENDIAN, 1, CalypStream::Type::Input );
|
|
pcFramePast = pcStreamPast->getCurrFrame();
|
|
|
|
pcStreamFuture = std::make_unique<CalypStream>();
|
|
pcStreamFuture->open( std::string( CALYP_TEST_DATA_DIR ) + std::string( "/BasketballDrill_F15_832x480_yuv420p.yuv" ), 832, 480, ClpPixelFormats::YUV420p, 8,
|
|
CLP_BIG_ENDIAN, 1, CalypStream::Type::Input );
|
|
pcFrameFuture = pcStreamFuture->getCurrFrame();
|
|
}
|
|
|
|
// Per-test-case tear-down.
|
|
// Called after the last test in this test case.
|
|
// Can be omitted if not needed.
|
|
static void TearDownTestCase()
|
|
{
|
|
pcStreamPast = nullptr;
|
|
pcStreamFuture = nullptr;
|
|
}
|
|
|
|
// Some expensive resource shared by all tests.
|
|
static std::unique_ptr<CalypStream> pcStreamPast;
|
|
static CalypFrame* pcFramePast;
|
|
|
|
static std::unique_ptr<CalypStream> pcStreamFuture;
|
|
static CalypFrame* pcFrameFuture;
|
|
};
|
|
|
|
std::unique_ptr<CalypStream> CalypFrameQualityTest::pcStreamPast{ nullptr };
|
|
std::unique_ptr<CalypStream> CalypFrameQualityTest::pcStreamFuture{ nullptr };
|
|
CalypFrame* CalypFrameQualityTest::pcFramePast{ nullptr };
|
|
CalypFrame* CalypFrameQualityTest::pcFrameFuture{ nullptr };
|
|
|
|
TEST_F( CalypFrameQualityTest, MSE )
|
|
{
|
|
for( int c = 0; c < 3; c++ )
|
|
EXPECT_EQ( pcFramePast->getQuality( CalypFrame::MSE_METRIC, pcFramePast, c ), 0 );
|
|
|
|
EXPECT_NE( pcFrameFuture->getQuality( CalypFrame::MSE_METRIC, pcFramePast, 0 ), 0 );
|
|
}
|
|
|
|
TEST_F( CalypFrameQualityTest, PSNR )
|
|
{
|
|
for( int c = 0; c < 3; c++ )
|
|
EXPECT_EQ( pcFramePast->getQuality( CalypFrame::PSNR_METRIC, pcFramePast, c ), 100 );
|
|
}
|
|
|
|
TEST_F( CalypFrameQualityTest, SSIM )
|
|
{
|
|
for( int c = 0; c < 3; c++ )
|
|
EXPECT_EQ( pcFramePast->getQuality( CalypFrame::SSIM_METRIC, pcFramePast, c ), 1.0 );
|
|
}
|