ref: 4d307cd7ff7baaca93053be554db300de2b149f0
dir: /test/svc_test.h/
/* * Copyright (c) 2018 The WebM 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 VPX_TEST_SVC_TEST_H_ #define VPX_TEST_SVC_TEST_H_ #include "./vpx_config.h" #include "third_party/googletest/src/include/gtest/gtest.h" #include "test/codec_factory.h" #include "test/encode_test_driver.h" #include "test/i420_video_source.h" #include "test/util.h" #include "test/y4m_video_source.h" #include "vpx/vpx_codec.h" #include "vpx_ports/bitops.h" namespace svc_test { class OnePassCbrSvc : public ::libvpx_test::EncoderTest { public: explicit OnePassCbrSvc(const ::libvpx_test::CodecFactory *codec) : EncoderTest(codec), base_speed_setting_(0), speed_setting_(0), superframe_count_(0), temporal_layer_id_(0), number_temporal_layers_(0), number_spatial_layers_(0) { memset(&svc_params_, 0, sizeof(svc_params_)); memset(bits_in_buffer_model_, 0, sizeof(bits_in_buffer_model_[0]) * VPX_MAX_LAYERS); memset(layer_target_avg_bandwidth_, 0, sizeof(layer_target_avg_bandwidth_[0]) * VPX_MAX_LAYERS); } protected: virtual ~OnePassCbrSvc() {} virtual void SetConfig(const int num_temporal_layer) = 0; virtual void SetSvcConfig(const int num_spatial_layer, const int num_temporal_layer); virtual void PreEncodeFrameHookSetup(::libvpx_test::VideoSource *video, ::libvpx_test::Encoder *encoder); virtual void PostEncodeFrameHook(::libvpx_test::Encoder *encoder); virtual void AssignLayerBitrates(); virtual void MismatchHook(const vpx_image_t *, const vpx_image_t *) {} vpx_svc_extra_cfg_t svc_params_; int64_t bits_in_buffer_model_[VPX_MAX_LAYERS]; int layer_target_avg_bandwidth_[VPX_MAX_LAYERS]; int base_speed_setting_; int speed_setting_; int superframe_count_; int temporal_layer_id_; int number_temporal_layers_; int number_spatial_layers_; }; } // namespace svc_test #endif // VPX_TEST_SVC_TEST_H_