test/encode_test_driver.cc - platform/external/libaom - Git at Google


            
            /*

            
             * Copyright (c) 2016, Alliance for Open Media. All rights reserved

            
             *

            
             * This source code is subject to the terms of the BSD 2 Clause License and

            
             * the Alliance for Open Media Patent License 1.0. If the BSD 2 Clause License

            
             * was not distributed with this source code in the LICENSE file, you can

            
             * obtain it at www.aomedia.org/license/software. If the Alliance for Open

            
             * Media Patent License 1.0 was not distributed with this source code in the

            
             * PATENTS file, you can obtain it at www.aomedia.org/license/patent.

            
            */


            #include <string>


            #include "third_party/googletest/src/include/gtest/gtest.h"


            #include "./aom_config.h"

            
            #include "test/codec_factory.h"

            
            #include "test/decode_test_driver.h"

            
            #include "test/encode_test_driver.h"

            
            #include "test/register_state_check.h"

            
            #include "test/video_source.h"


            namespace libaom_test {

            
            void Encoder::InitEncoder(VideoSource *video) {

            
              aom_codec_err_t res;

            
              const aom_image_t *img = video->img();


              if (video->img() && !encoder_.priv) {

            
                cfg_.g_w = img->d_w;

            
                cfg_.g_h = img->d_h;

            
                cfg_.g_timebase = video->timebase();

            
                cfg_.rc_twopass_stats_in = stats_->buf();


                res = aom_codec_enc_init(&encoder_, CodecInterface(), &cfg_, init_flags_);

            
                ASSERT_EQ(AOM_CODEC_OK, res) << EncoderError();


            #if CONFIG_AV1_ENCODER

            
                if (CodecInterface() == &aom_codec_av1_cx_algo) {

            
                  // Default to 1 tile column for AV1.

            
                  const int log2_tile_columns = 0;

            
                  res = aom_codec_control_(&encoder_, AV1E_SET_TILE_COLUMNS,

            
                                           log2_tile_columns);

            
                  ASSERT_EQ(AOM_CODEC_OK, res) << EncoderError();

            
                }

            
            #endif

            
              }

            
            }


            void Encoder::EncodeFrame(VideoSource *video, const unsigned long frame_flags) {

            
              if (video->img())

            
                EncodeFrameInternal(*video, frame_flags);

            
              else

            
                Flush();


              // Handle twopass stats

            
              CxDataIterator iter = GetCxData();


              while (const aom_codec_cx_pkt_t *pkt = iter.Next()) {

            
                if (pkt->kind != AOM_CODEC_STATS_PKT) continue;


                stats_->Append(*pkt);

            
              }

            
            }


            void Encoder::EncodeFrameInternal(const VideoSource &video,

            
                                              const unsigned long frame_flags) {

            
              aom_codec_err_t res;

            
              const aom_image_t *img = video.img();


              // Handle frame resizing

            
              if (cfg_.g_w != img->d_w || cfg_.g_h != img->d_h) {

            
                cfg_.g_w = img->d_w;

            
                cfg_.g_h = img->d_h;

            
                res = aom_codec_enc_config_set(&encoder_, &cfg_);

            
                ASSERT_EQ(AOM_CODEC_OK, res) << EncoderError();

            
              }


              // Encode the frame

            
              API_REGISTER_STATE_CHECK(res = aom_codec_encode(&encoder_, img, video.pts(),

            
                                                              video.duration(), frame_flags,

            
                                                              deadline_));

            
              ASSERT_EQ(AOM_CODEC_OK, res) << EncoderError();

            
            }


            void Encoder::Flush() {

            
              const aom_codec_err_t res =

            
                  aom_codec_encode(&encoder_, NULL, 0, 0, 0, deadline_);

            
              if (!encoder_.priv)

            
                ASSERT_EQ(AOM_CODEC_ERROR, res) << EncoderError();

            
              else

            
                ASSERT_EQ(AOM_CODEC_OK, res) << EncoderError();

            
            }


            void EncoderTest::InitializeConfig() {

            
              const aom_codec_err_t res = codec_->DefaultEncoderConfig(&cfg_, 0);

            
              dec_cfg_ = aom_codec_dec_cfg_t();

            
              ASSERT_EQ(AOM_CODEC_OK, res);

            
            }


            void EncoderTest::SetMode(TestMode mode) {

            
              switch (mode) {

            
                case kRealTime: deadline_ = AOM_DL_REALTIME; break;


                case kOnePassGood:

            
                case kTwoPassGood: deadline_ = AOM_DL_GOOD_QUALITY; break;


                case kOnePassBest:

            
                case kTwoPassBest: deadline_ = AOM_DL_BEST_QUALITY; break;


                default: ASSERT_TRUE(false) << "Unexpected mode " << mode;

            
              }


              if (mode == kTwoPassGood || mode == kTwoPassBest)

            
                passes_ = 2;

            
              else

            
                passes_ = 1;

            
            }

            
            // The function should return "true" most of the time, therefore no early

            
            // break-out is implemented within the match checking process.

            
            static bool compare_img(const aom_image_t *img1, const aom_image_t *img2) {

            
              bool match = (img1->fmt == img2->fmt) && (img1->cs == img2->cs) &&

            
                           (img1->d_w == img2->d_w) && (img1->d_h == img2->d_h);


              const unsigned int width_y = img1->d_w;

            
              const unsigned int height_y = img1->d_h;

            
              unsigned int i;

            
              for (i = 0; i < height_y; ++i)

            
                match = (memcmp(img1->planes[AOM_PLANE_Y] + i * img1->stride[AOM_PLANE_Y],

            
                                img2->planes[AOM_PLANE_Y] + i * img2->stride[AOM_PLANE_Y],

            
                                width_y) == 0) &&

            
                        match;

            
              const unsigned int width_uv = (img1->d_w + 1) >> 1;

            
              const unsigned int height_uv = (img1->d_h + 1) >> 1;

            
              for (i = 0; i < height_uv; ++i)

            
                match = (memcmp(img1->planes[AOM_PLANE_U] + i * img1->stride[AOM_PLANE_U],

            
                                img2->planes[AOM_PLANE_U] + i * img2->stride[AOM_PLANE_U],

            
                                width_uv) == 0) &&

            
                        match;

            
              for (i = 0; i < height_uv; ++i)

            
                match = (memcmp(img1->planes[AOM_PLANE_V] + i * img1->stride[AOM_PLANE_V],

            
                                img2->planes[AOM_PLANE_V] + i * img2->stride[AOM_PLANE_V],

            
                                width_uv) == 0) &&

            
                        match;

            
              return match;

            
            }


            void EncoderTest::MismatchHook(const aom_image_t * /*img1*/,

            
                                           const aom_image_t * /*img2*/) {

            
              ASSERT_TRUE(0) << "Encode/Decode mismatch found";

            
            }


            void EncoderTest::RunLoop(VideoSource *video) {

            
              aom_codec_dec_cfg_t dec_cfg = aom_codec_dec_cfg_t();


              stats_.Reset();


              ASSERT_TRUE(passes_ == 1 || passes_ == 2);

            
              for (unsigned int pass = 0; pass < passes_; pass++) {

            
                last_pts_ = 0;


                if (passes_ == 1)

            
                  cfg_.g_pass = AOM_RC_ONE_PASS;

            
                else if (pass == 0)

            
                  cfg_.g_pass = AOM_RC_FIRST_PASS;

            
                else

            
                  cfg_.g_pass = AOM_RC_LAST_PASS;


                BeginPassHook(pass);

            
                Encoder *const encoder =

            
                    codec_->CreateEncoder(cfg_, deadline_, init_flags_, &stats_);

            
                ASSERT_TRUE(encoder != NULL);


                video->Begin();

            
                encoder->InitEncoder(video);

            
                ASSERT_FALSE(::testing::Test::HasFatalFailure());


                unsigned long dec_init_flags = 0;  // NOLINT

            
                // Use fragment decoder if encoder outputs partitions.

            
                // NOTE: fragment decoder and partition encoder are only supported by AOM.

            
                if (init_flags_ & AOM_CODEC_USE_OUTPUT_PARTITION)

            
                  dec_init_flags |= AOM_CODEC_USE_INPUT_FRAGMENTS;

            
                Decoder *const decoder = codec_->CreateDecoder(dec_cfg, dec_init_flags, 0);

            
                bool again;

            
                for (again = true; again; video->Next()) {

            
                  again = (video->img() != NULL);


                  PreEncodeFrameHook(video);

            
                  PreEncodeFrameHook(video, encoder);

            
                  encoder->EncodeFrame(video, frame_flags_);


                  CxDataIterator iter = encoder->GetCxData();


                  bool has_cxdata = false;

            
                  bool has_dxdata = false;

            
                  while (const aom_codec_cx_pkt_t *pkt = iter.Next()) {

            
                    pkt = MutateEncoderOutputHook(pkt);

            
                    again = true;

            
                    switch (pkt->kind) {

            
                      case AOM_CODEC_CX_FRAME_PKT:

            
                        has_cxdata = true;

            
                        if (decoder && DoDecode()) {

            
                          aom_codec_err_t res_dec = decoder->DecodeFrame(

            
                              (const uint8_t *)pkt->data.frame.buf, pkt->data.frame.sz);


                          if (!HandleDecodeResult(res_dec, *video, decoder)) break;


                          has_dxdata = true;

            
                        }

            
                        ASSERT_GE(pkt->data.frame.pts, last_pts_);

            
                        last_pts_ = pkt->data.frame.pts;

            
                        FramePktHook(pkt);

            
                        break;


                      case AOM_CODEC_PSNR_PKT: PSNRPktHook(pkt); break;


                      default: break;

            
                    }

            
                  }


                  // Flush the decoder when there are no more fragments.

            
                  if ((init_flags_ & AOM_CODEC_USE_OUTPUT_PARTITION) && has_dxdata) {

            
                    const aom_codec_err_t res_dec = decoder->DecodeFrame(NULL, 0);

            
                    if (!HandleDecodeResult(res_dec, *video, decoder)) break;

            
                  }


                  if (has_dxdata && has_cxdata) {

            
                    const aom_image_t *img_enc = encoder->GetPreviewFrame();

            
                    DxDataIterator dec_iter = decoder->GetDxData();

            
                    const aom_image_t *img_dec = dec_iter.Next();

            
                    if (img_enc && img_dec) {

            
                      const bool res = compare_img(img_enc, img_dec);

            
                      if (!res) {  // Mismatch

            
                        MismatchHook(img_enc, img_dec);

            
                      }

            
                    }

            
                    if (img_dec) DecompressedFrameHook(*img_dec, video->pts());

            
                  }

            
                  if (!Continue()) break;

            
                }


                EndPassHook();


                if (decoder) delete decoder;

            
                delete encoder;


                if (!Continue()) break;

            
              }

            
            }


            }  // namespace libaom_test
	/*
	* Copyright (c) 2016, Alliance for Open Media. All rights reserved
	*
	* This source code is subject to the terms of the BSD 2 Clause License and
	* the Alliance for Open Media Patent License 1.0. If the BSD 2 Clause License
	* was not distributed with this source code in the LICENSE file, you can
	* obtain it at www.aomedia.org/license/software. If the Alliance for Open
	* Media Patent License 1.0 was not distributed with this source code in the
	* PATENTS file, you can obtain it at www.aomedia.org/license/patent.
	*/


	#include <string>

	#include "third_party/googletest/src/include/gtest/gtest.h"

	#include "./aom_config.h"
	#include "test/codec_factory.h"
	#include "test/decode_test_driver.h"
	#include "test/encode_test_driver.h"
	#include "test/register_state_check.h"
	#include "test/video_source.h"

	namespace libaom_test {
	void Encoder::InitEncoder(VideoSource *video) {
	aom_codec_err_t res;
	const aom_image_t *img = video->img();

	if (video->img() && !encoder_.priv) {
	cfg_.g_w = img->d_w;
	cfg_.g_h = img->d_h;
	cfg_.g_timebase = video->timebase();
	cfg_.rc_twopass_stats_in = stats_->buf();

	res = aom_codec_enc_init(&encoder_, CodecInterface(), &cfg_, init_flags_);
	ASSERT_EQ(AOM_CODEC_OK, res) << EncoderError();

	#if CONFIG_AV1_ENCODER
	if (CodecInterface() == &aom_codec_av1_cx_algo) {
	// Default to 1 tile column for AV1.
	const int log2_tile_columns = 0;
	res = aom_codec_control_(&encoder_, AV1E_SET_TILE_COLUMNS,
	log2_tile_columns);
	ASSERT_EQ(AOM_CODEC_OK, res) << EncoderError();
	}
	#endif
	}
	}

	void Encoder::EncodeFrame(VideoSource *video, const unsigned long frame_flags) {
	if (video->img())
	EncodeFrameInternal(*video, frame_flags);
	else
	Flush();

	// Handle twopass stats
	CxDataIterator iter = GetCxData();

	while (const aom_codec_cx_pkt_t *pkt = iter.Next()) {
	if (pkt->kind != AOM_CODEC_STATS_PKT) continue;

	stats_->Append(*pkt);
	}
	}

	void Encoder::EncodeFrameInternal(const VideoSource &video,
	const unsigned long frame_flags) {
	aom_codec_err_t res;
	const aom_image_t *img = video.img();

	// Handle frame resizing
	if (cfg_.g_w != img->d_w \|\| cfg_.g_h != img->d_h) {
	cfg_.g_w = img->d_w;
	cfg_.g_h = img->d_h;
	res = aom_codec_enc_config_set(&encoder_, &cfg_);
	ASSERT_EQ(AOM_CODEC_OK, res) << EncoderError();
	}

	// Encode the frame
	API_REGISTER_STATE_CHECK(res = aom_codec_encode(&encoder_, img, video.pts(),
	video.duration(), frame_flags,
	deadline_));
	ASSERT_EQ(AOM_CODEC_OK, res) << EncoderError();
	}

	void Encoder::Flush() {
	const aom_codec_err_t res =
	aom_codec_encode(&encoder_, NULL, 0, 0, 0, deadline_);
	if (!encoder_.priv)
	ASSERT_EQ(AOM_CODEC_ERROR, res) << EncoderError();
	else
	ASSERT_EQ(AOM_CODEC_OK, res) << EncoderError();
	}

	void EncoderTest::InitializeConfig() {
	const aom_codec_err_t res = codec_->DefaultEncoderConfig(&cfg_, 0);
	dec_cfg_ = aom_codec_dec_cfg_t();
	ASSERT_EQ(AOM_CODEC_OK, res);
	}

	void EncoderTest::SetMode(TestMode mode) {
	switch (mode) {
	case kRealTime: deadline_ = AOM_DL_REALTIME; break;

	case kOnePassGood:
	case kTwoPassGood: deadline_ = AOM_DL_GOOD_QUALITY; break;

	case kOnePassBest:
	case kTwoPassBest: deadline_ = AOM_DL_BEST_QUALITY; break;

	default: ASSERT_TRUE(false) << "Unexpected mode " << mode;
	}

	if (mode == kTwoPassGood \|\| mode == kTwoPassBest)
	passes_ = 2;
	else
	passes_ = 1;
	}
	// The function should return "true" most of the time, therefore no early
	// break-out is implemented within the match checking process.
	static bool compare_img(const aom_image_t img1, const aom_image_t img2) {
	bool match = (img1->fmt == img2->fmt) && (img1->cs == img2->cs) &&
	(img1->d_w == img2->d_w) && (img1->d_h == img2->d_h);

	const unsigned int width_y = img1->d_w;
	const unsigned int height_y = img1->d_h;
	unsigned int i;
	for (i = 0; i < height_y; ++i)
	match = (memcmp(img1->planes[AOM_PLANE_Y] + i * img1->stride[AOM_PLANE_Y],
	img2->planes[AOM_PLANE_Y] + i * img2->stride[AOM_PLANE_Y],
	width_y) == 0) &&
	match;
	const unsigned int width_uv = (img1->d_w + 1) >> 1;
	const unsigned int height_uv = (img1->d_h + 1) >> 1;
	for (i = 0; i < height_uv; ++i)
	match = (memcmp(img1->planes[AOM_PLANE_U] + i * img1->stride[AOM_PLANE_U],
	img2->planes[AOM_PLANE_U] + i * img2->stride[AOM_PLANE_U],
	width_uv) == 0) &&
	match;
	for (i = 0; i < height_uv; ++i)
	match = (memcmp(img1->planes[AOM_PLANE_V] + i * img1->stride[AOM_PLANE_V],
	img2->planes[AOM_PLANE_V] + i * img2->stride[AOM_PLANE_V],
	width_uv) == 0) &&
	match;
	return match;
	}

	void EncoderTest::MismatchHook(const aom_image_t * /img1/,
	const aom_image_t * /img2/) {
	ASSERT_TRUE(0) << "Encode/Decode mismatch found";
	}

	void EncoderTest::RunLoop(VideoSource *video) {
	aom_codec_dec_cfg_t dec_cfg = aom_codec_dec_cfg_t();

	stats_.Reset();

	ASSERT_TRUE(passes_ == 1 \|\| passes_ == 2);
	for (unsigned int pass = 0; pass < passes_; pass++) {
	last_pts_ = 0;

	if (passes_ == 1)
	cfg_.g_pass = AOM_RC_ONE_PASS;
	else if (pass == 0)
	cfg_.g_pass = AOM_RC_FIRST_PASS;
	else
	cfg_.g_pass = AOM_RC_LAST_PASS;

	BeginPassHook(pass);
	Encoder *const encoder =
	codec_->CreateEncoder(cfg_, deadline_, init_flags_, &stats_);
	ASSERT_TRUE(encoder != NULL);

	video->Begin();
	encoder->InitEncoder(video);
	ASSERT_FALSE(::testing::Test::HasFatalFailure());

	unsigned long dec_init_flags = 0; // NOLINT
	// Use fragment decoder if encoder outputs partitions.
	// NOTE: fragment decoder and partition encoder are only supported by AOM.
	if (init_flags_ & AOM_CODEC_USE_OUTPUT_PARTITION)
	dec_init_flags \|= AOM_CODEC_USE_INPUT_FRAGMENTS;
	Decoder *const decoder = codec_->CreateDecoder(dec_cfg, dec_init_flags, 0);
	bool again;
	for (again = true; again; video->Next()) {
	again = (video->img() != NULL);

	PreEncodeFrameHook(video);
	PreEncodeFrameHook(video, encoder);
	encoder->EncodeFrame(video, frame_flags_);

	CxDataIterator iter = encoder->GetCxData();

	bool has_cxdata = false;
	bool has_dxdata = false;
	while (const aom_codec_cx_pkt_t *pkt = iter.Next()) {
	pkt = MutateEncoderOutputHook(pkt);
	again = true;
	switch (pkt->kind) {
	case AOM_CODEC_CX_FRAME_PKT:
	has_cxdata = true;
	if (decoder && DoDecode()) {
	aom_codec_err_t res_dec = decoder->DecodeFrame(
	(const uint8_t *)pkt->data.frame.buf, pkt->data.frame.sz);

	if (!HandleDecodeResult(res_dec, *video, decoder)) break;

	has_dxdata = true;
	}
	ASSERT_GE(pkt->data.frame.pts, last_pts_);
	last_pts_ = pkt->data.frame.pts;
	FramePktHook(pkt);
	break;

	case AOM_CODEC_PSNR_PKT: PSNRPktHook(pkt); break;

	default: break;
	}
	}

	// Flush the decoder when there are no more fragments.
	if ((init_flags_ & AOM_CODEC_USE_OUTPUT_PARTITION) && has_dxdata) {
	const aom_codec_err_t res_dec = decoder->DecodeFrame(NULL, 0);
	if (!HandleDecodeResult(res_dec, *video, decoder)) break;
	}

	if (has_dxdata && has_cxdata) {
	const aom_image_t *img_enc = encoder->GetPreviewFrame();
	DxDataIterator dec_iter = decoder->GetDxData();
	const aom_image_t *img_dec = dec_iter.Next();
	if (img_enc && img_dec) {
	const bool res = compare_img(img_enc, img_dec);
	if (!res) { // Mismatch
	MismatchHook(img_enc, img_dec);
	}
	}
	if (img_dec) DecompressedFrameHook(*img_dec, video->pts());
	}
	if (!Continue()) break;
	}

	EndPassHook();

	if (decoder) delete decoder;
	delete encoder;

	if (!Continue()) break;
	}
	}

	} // namespace libaom_test