68 lines
2.2 KiB
C++
68 lines
2.2 KiB
C++
/*
|
|
* 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_
|