| Index: webrtc/modules/video_coding/codecs/test/video_codec_test.h
|
| diff --git a/webrtc/modules/video_coding/codecs/test/video_codec_test.h b/webrtc/modules/video_coding/codecs/test/video_codec_test.h
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..65d20ee8984fdcfef911186f34f280cc4ed530a1
|
| --- /dev/null
|
| +++ b/webrtc/modules/video_coding/codecs/test/video_codec_test.h
|
| @@ -0,0 +1,101 @@
|
| +/*
|
| + * 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 WEBRTC_MODULES_VIDEO_CODING_CODECS_TEST_VIDEO_CODEC_TEST_H_
|
| +#define WEBRTC_MODULES_VIDEO_CODING_CODECS_TEST_VIDEO_CODEC_TEST_H_
|
| +
|
| +#include <memory>
|
| +
|
| +#include "webrtc/base/criticalsection.h"
|
| +#include "webrtc/base/event.h"
|
| +#include "webrtc/base/thread_annotations.h"
|
| +#include "webrtc/test/gtest.h"
|
| +#include "webrtc/video_decoder.h"
|
| +#include "webrtc/video_encoder.h"
|
| +
|
| +namespace webrtc {
|
| +
|
| +class VideoCodecTest : public ::testing::Test {
|
| + public:
|
| + VideoCodecTest()
|
| + : encode_complete_callback_(this),
|
| + decode_complete_callback_(this),
|
| + encoded_frame_event_(false /* manual reset */,
|
| + false /* initially signaled */),
|
| + decoded_frame_event_(false /* manual reset */,
|
| + false /* initially signaled */) {}
|
| +
|
| + protected:
|
| + class FakeEncodeCompleteCallback : public webrtc::EncodedImageCallback {
|
| + public:
|
| + explicit FakeEncodeCompleteCallback(VideoCodecTest* test) : test_(test) {}
|
| +
|
| + Result OnEncodedImage(const EncodedImage& frame,
|
| + const CodecSpecificInfo* codec_specific_info,
|
| + const RTPFragmentationHeader* fragmentation);
|
| +
|
| + private:
|
| + VideoCodecTest* const test_;
|
| + };
|
| +
|
| + class FakeDecodeCompleteCallback : public webrtc::DecodedImageCallback {
|
| + public:
|
| + explicit FakeDecodeCompleteCallback(VideoCodecTest* test) : test_(test) {}
|
| +
|
| + int32_t Decoded(VideoFrame& frame) override {
|
| + RTC_NOTREACHED();
|
| + return -1;
|
| + }
|
| + int32_t Decoded(VideoFrame& frame, int64_t decode_time_ms) override {
|
| + RTC_NOTREACHED();
|
| + return -1;
|
| + }
|
| + void Decoded(VideoFrame& frame,
|
| + rtc::Optional<int32_t> decode_time_ms,
|
| + rtc::Optional<uint8_t> qp) override;
|
| +
|
| + private:
|
| + VideoCodecTest* const test_;
|
| + };
|
| +
|
| + virtual VideoEncoder* CreateEncoder() = 0;
|
| + virtual VideoDecoder* CreateDecoder() = 0;
|
| + virtual VideoCodec codec_settings() = 0;
|
| +
|
| + void SetUp() override;
|
| +
|
| + bool WaitForEncodedFrame(EncodedImage* frame);
|
| + bool WaitForDecodedFrame(std::unique_ptr<VideoFrame>* frame,
|
| + rtc::Optional<uint8_t>* qp);
|
| +
|
| + std::unique_ptr<VideoFrame> input_frame_;
|
| +
|
| + std::unique_ptr<VideoEncoder> encoder_;
|
| + std::unique_ptr<VideoDecoder> decoder_;
|
| +
|
| + private:
|
| + void InitCodecs();
|
| +
|
| + FakeEncodeCompleteCallback encode_complete_callback_;
|
| + FakeDecodeCompleteCallback decode_complete_callback_;
|
| +
|
| + rtc::Event encoded_frame_event_;
|
| + rtc::CriticalSection encoded_frame_section_;
|
| + rtc::Optional<EncodedImage> encoded_frame_ GUARDED_BY(encoded_frame_section_);
|
| +
|
| + rtc::Event decoded_frame_event_;
|
| + rtc::CriticalSection decoded_frame_section_;
|
| + rtc::Optional<VideoFrame> decoded_frame_ GUARDED_BY(decoded_frame_section_);
|
| + rtc::Optional<uint8_t> decoded_qp_ GUARDED_BY(decoded_frame_section_);
|
| +};
|
| +
|
| +} // namespace webrtc
|
| +
|
| +#endif // WEBRTC_MODULES_VIDEO_CODING_CODECS_TEST_VIDEO_CODEC_TEST_H_
|
|
|