121 lines
3.9 KiB
C++
121 lines
3.9 KiB
C++
/*
|
|
* Copyright (c) 2015 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_VIDEO_VIDEO_QUALITY_TEST_H_
|
|
#define WEBRTC_VIDEO_VIDEO_QUALITY_TEST_H_
|
|
|
|
#include <memory>
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
#include "webrtc/test/call_test.h"
|
|
#include "webrtc/test/frame_generator.h"
|
|
#include "webrtc/test/testsupport/trace_to_stderr.h"
|
|
|
|
namespace webrtc {
|
|
|
|
class VideoQualityTest : public test::CallTest {
|
|
public:
|
|
// Parameters are grouped into smaller structs to make it easier to set
|
|
// the desired elements and skip unused, using aggregate initialization.
|
|
// Unfortunately, C++11 (as opposed to C11) doesn't support unnamed structs,
|
|
// which makes the implementation of VideoQualityTest a bit uglier.
|
|
struct Params {
|
|
struct {
|
|
size_t width;
|
|
size_t height;
|
|
int32_t fps;
|
|
int min_bitrate_bps;
|
|
int target_bitrate_bps;
|
|
int max_bitrate_bps;
|
|
std::string codec;
|
|
int num_temporal_layers;
|
|
int selected_tl;
|
|
int min_transmit_bps;
|
|
|
|
Call::Config::BitrateConfig call_bitrate_config;
|
|
bool send_side_bwe;
|
|
bool fec;
|
|
} common;
|
|
struct { // Video-specific settings.
|
|
std::string clip_name;
|
|
} video;
|
|
struct { // Screenshare-specific settings.
|
|
bool enabled;
|
|
int32_t slide_change_interval;
|
|
int32_t scroll_duration;
|
|
} screenshare;
|
|
struct { // Analyzer settings.
|
|
std::string test_label;
|
|
double avg_psnr_threshold; // (*)
|
|
double avg_ssim_threshold; // (*)
|
|
int test_durations_secs;
|
|
std::string graph_data_output_filename;
|
|
std::string graph_title;
|
|
} analyzer;
|
|
FakeNetworkPipe::Config pipe;
|
|
bool logs;
|
|
struct { // Spatial scalability.
|
|
std::vector<VideoStream> streams; // If empty, one stream is assumed.
|
|
size_t selected_stream;
|
|
int num_spatial_layers;
|
|
int selected_sl;
|
|
// If empty, bitrates are generated in VP9Impl automatically.
|
|
std::vector<SpatialLayer> spatial_layers;
|
|
} ss;
|
|
};
|
|
// (*) Set to -1.1 if generating graph data for simulcast or SVC and the
|
|
// selected stream/layer doesn't have the same resolution as the largest
|
|
// stream/layer (to ignore the PSNR and SSIM calculation errors).
|
|
|
|
VideoQualityTest();
|
|
void RunWithAnalyzer(const Params& params);
|
|
void RunWithVideoRenderer(const Params& params);
|
|
|
|
static void FillScalabilitySettings(
|
|
Params* params,
|
|
const std::vector<std::string>& stream_descriptors,
|
|
size_t selected_stream,
|
|
int num_spatial_layers,
|
|
int selected_sl,
|
|
const std::vector<std::string>& sl_descriptors);
|
|
|
|
protected:
|
|
// No-op implementation to be able to instantiate this class from non-TEST_F
|
|
// locations.
|
|
void TestBody() override;
|
|
|
|
// Helper methods accessing only params_.
|
|
std::string GenerateGraphTitle() const;
|
|
void CheckParams();
|
|
|
|
// Helper static methods.
|
|
static VideoStream DefaultVideoStream(const Params& params);
|
|
static std::vector<int> ParseCSV(const std::string& str);
|
|
|
|
// Helper methods for setting up the call.
|
|
void CreateCapturer(VideoCaptureInput* input);
|
|
void SetupCommon(Transport* send_transport, Transport* recv_transport);
|
|
void SetupScreenshare();
|
|
|
|
// We need a more general capturer than the FrameGeneratorCapturer.
|
|
std::unique_ptr<test::VideoCapturer> capturer_;
|
|
std::unique_ptr<test::TraceToStderr> trace_to_stderr_;
|
|
std::unique_ptr<test::FrameGenerator> frame_generator_;
|
|
std::unique_ptr<VideoEncoder> encoder_;
|
|
VideoCodecUnion codec_settings_;
|
|
Clock* const clock_;
|
|
|
|
Params params_;
|
|
};
|
|
|
|
} // namespace webrtc
|
|
|
|
#endif // WEBRTC_VIDEO_VIDEO_QUALITY_TEST_H_
|