mirror of
https://github.com/mollyim/webrtc.git
synced 2025-05-14 14:20:45 +01:00
9d96e92316
This CL creates a test fixture for the videoprocessor integration tests and exposes it as part of the public API. It also rewrites the current versions of the tests to build on this new paradigm. The motivation for this is to easily allow projects that build on top of webrtc to add integration-level tests for their own custom codec implementations in a way that does not link them too tightly to the internal implementations of said tests. Bug: None Change-Id: I7cf9f29322a6934b3cfc32da02ea7dfa5858c2b2 Reviewed-on: https://webrtc-review.googlesource.com/72481 Commit-Queue: Kári Helgason <kthelgason@webrtc.org> Reviewed-by: Rasmus Brandt <brandtr@webrtc.org> Reviewed-by: Fredrik Solenberg <solenberg@webrtc.org> Reviewed-by: Patrik Höglund <phoglund@webrtc.org> Cr-Commit-Position: refs/heads/master@{#23118}
110 lines
4.4 KiB
C++
110 lines
4.4 KiB
C++
/*
|
|
* Copyright (c) 2017 The WebRTC project authors. All Rights Reserved.
|
|
*
|
|
* Use of this source code is governed by a BSD-style license
|
|
* that can be found in the LICENSE file in the root of the source
|
|
* tree. An additional intellectual property rights grant can be found
|
|
* in the file PATENTS. All contributing project authors may
|
|
* be found in the AUTHORS file in the root of the source tree.
|
|
*/
|
|
|
|
#ifndef MODULES_VIDEO_CODING_CODECS_TEST_VIDEOCODEC_TEST_FIXTURE_IMPL_H_
|
|
#define MODULES_VIDEO_CODING_CODECS_TEST_VIDEOCODEC_TEST_FIXTURE_IMPL_H_
|
|
|
|
#include <memory>
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
#include "api/test/videocodec_test_fixture.h"
|
|
#include "api/video_codecs/video_decoder_factory.h"
|
|
#include "api/video_codecs/video_encoder_factory.h"
|
|
#include "common_types.h" // NOLINT(build/include)
|
|
#include "common_video/h264/h264_common.h"
|
|
#include "modules/video_coding/codecs/test/stats.h"
|
|
#include "modules/video_coding/codecs/test/test_config.h"
|
|
#include "modules/video_coding/codecs/test/videoprocessor.h"
|
|
#include "modules/video_coding/utility/ivf_file_writer.h"
|
|
#include "test/testsupport/frame_reader.h"
|
|
#include "test/testsupport/frame_writer.h"
|
|
|
|
namespace webrtc {
|
|
namespace test {
|
|
|
|
// Integration test for video processor. It does rate control and frame quality
|
|
// analysis using frame statistics collected by video processor and logs the
|
|
// results. If thresholds are specified it checks that corresponding metrics
|
|
// are in desirable range.
|
|
class VideoCodecTestFixtureImpl : public VideoCodecTestFixture {
|
|
// Verifies that all H.264 keyframes contain SPS/PPS/IDR NALUs.
|
|
public:
|
|
class H264KeyframeChecker : public TestConfig::EncodedFrameChecker {
|
|
public:
|
|
void CheckEncodedFrame(webrtc::VideoCodecType codec,
|
|
const EncodedImage& encoded_frame) const override;
|
|
};
|
|
|
|
explicit VideoCodecTestFixtureImpl(TestConfig config);
|
|
VideoCodecTestFixtureImpl(
|
|
TestConfig config,
|
|
std::unique_ptr<VideoDecoderFactory> decoder_factory,
|
|
std::unique_ptr<VideoEncoderFactory> encoder_factory);
|
|
~VideoCodecTestFixtureImpl() override;
|
|
|
|
void RunTest(const std::vector<RateProfile>& rate_profiles,
|
|
const std::vector<RateControlThresholds>* rc_thresholds,
|
|
const std::vector<QualityThresholds>* quality_thresholds,
|
|
const BitstreamThresholds* bs_thresholds,
|
|
const VisualizationParams* visualization_params) override;
|
|
|
|
Stats GetStats() override;
|
|
|
|
private:
|
|
class CpuProcessTime;
|
|
|
|
void CreateEncoderAndDecoder();
|
|
void DestroyEncoderAndDecoder();
|
|
void SetUpAndInitObjects(rtc::TaskQueue* task_queue,
|
|
int initial_bitrate_kbps,
|
|
int initial_framerate_fps,
|
|
const VisualizationParams* visualization_params);
|
|
void ReleaseAndCloseObjects(rtc::TaskQueue* task_queue);
|
|
|
|
void ProcessAllFrames(rtc::TaskQueue* task_queue,
|
|
const std::vector<RateProfile>& rate_profiles);
|
|
void AnalyzeAllFrames(
|
|
const std::vector<RateProfile>& rate_profiles,
|
|
const std::vector<RateControlThresholds>* rc_thresholds,
|
|
const std::vector<QualityThresholds>* quality_thresholds,
|
|
const BitstreamThresholds* bs_thresholds);
|
|
|
|
void VerifyVideoStatistic(const VideoStatistics& video_stat,
|
|
const RateControlThresholds* rc_thresholds,
|
|
const QualityThresholds* quality_thresholds,
|
|
const BitstreamThresholds* bs_thresholds,
|
|
size_t target_bitrate_kbps,
|
|
float input_framerate_fps);
|
|
|
|
void PrintSettings(rtc::TaskQueue* task_queue) const;
|
|
std::unique_ptr<VideoDecoderFactory> CreateDecoderFactory();
|
|
std::unique_ptr<VideoEncoderFactory> CreateEncoderFactory();
|
|
|
|
// Codecs.
|
|
std::unique_ptr<VideoDecoderFactory> decoder_factory_;
|
|
std::unique_ptr<VideoEncoderFactory> encoder_factory_;
|
|
std::unique_ptr<VideoEncoder> encoder_;
|
|
VideoProcessor::VideoDecoderList decoders_;
|
|
|
|
// Helper objects.
|
|
TestConfig config_;
|
|
Stats stats_;
|
|
std::unique_ptr<FrameReader> source_frame_reader_;
|
|
VideoProcessor::IvfFileWriterList encoded_frame_writers_;
|
|
VideoProcessor::FrameWriterList decoded_frame_writers_;
|
|
std::unique_ptr<VideoProcessor> processor_;
|
|
std::unique_ptr<CpuProcessTime> cpu_process_time_;
|
|
};
|
|
|
|
} // namespace test
|
|
} // namespace webrtc
|
|
|
|
#endif // MODULES_VIDEO_CODING_CODECS_TEST_VIDEOCODEC_TEST_FIXTURE_IMPL_H_
|