mirror of
https://github.com/mollyim/webrtc.git
synced 2025-05-13 13:50:40 +01:00
dcd7fc7ea8
Bug: webrtc:12338 Change-Id: Ia8a9adea291d594e4f59a6a1203a7bfb0758adac Reviewed-on: https://webrtc-review.googlesource.com/c/src/+/227165 Commit-Queue: Artem Titov <titovartem@webrtc.org> Reviewed-by: Harald Alvestrand <hta@webrtc.org> Cr-Commit-Position: refs/heads/master@{#34684}
84 lines
3 KiB
C++
84 lines
3 KiB
C++
/*
|
|
* Copyright (c) 2011 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_STATS_IMPL_H_
|
|
#define MODULES_VIDEO_CODING_CODECS_TEST_VIDEOCODEC_TEST_STATS_IMPL_H_
|
|
|
|
#include <stddef.h>
|
|
|
|
#include <map>
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
#include "api/test/videocodec_test_stats.h" // NOLINT(build/include)
|
|
|
|
namespace webrtc {
|
|
namespace test {
|
|
|
|
// Statistics for a sequence of processed frames. This class is not thread safe.
|
|
class VideoCodecTestStatsImpl : public VideoCodecTestStats {
|
|
public:
|
|
VideoCodecTestStatsImpl();
|
|
~VideoCodecTestStatsImpl() override;
|
|
|
|
// Creates a FrameStatistics for the next frame to be processed.
|
|
void AddFrame(const FrameStatistics& frame_stat);
|
|
|
|
// Returns the FrameStatistics corresponding to `frame_number` or `timestamp`.
|
|
FrameStatistics* GetFrame(size_t frame_number, size_t spatial_idx);
|
|
FrameStatistics* GetFrameWithTimestamp(size_t timestamp, size_t spatial_idx);
|
|
|
|
// Implements VideoCodecTestStats.
|
|
std::vector<FrameStatistics> GetFrameStatistics() override;
|
|
std::vector<VideoStatistics> SliceAndCalcLayerVideoStatistic(
|
|
size_t first_frame_num,
|
|
size_t last_frame_num) override;
|
|
|
|
VideoStatistics SliceAndCalcAggregatedVideoStatistic(size_t first_frame_num,
|
|
size_t last_frame_num);
|
|
|
|
size_t Size(size_t spatial_idx);
|
|
|
|
void Clear();
|
|
|
|
private:
|
|
VideoCodecTestStats::FrameStatistics AggregateFrameStatistic(
|
|
size_t frame_num,
|
|
size_t spatial_idx,
|
|
bool aggregate_independent_layers);
|
|
|
|
size_t CalcLayerTargetBitrateKbps(size_t first_frame_num,
|
|
size_t last_frame_num,
|
|
size_t spatial_idx,
|
|
size_t temporal_idx,
|
|
bool aggregate_independent_layers);
|
|
|
|
VideoCodecTestStats::VideoStatistics SliceAndCalcVideoStatistic(
|
|
size_t first_frame_num,
|
|
size_t last_frame_num,
|
|
size_t spatial_idx,
|
|
size_t temporal_idx,
|
|
bool aggregate_independent_layers);
|
|
|
|
void GetNumberOfEncodedLayers(size_t first_frame_num,
|
|
size_t last_frame_num,
|
|
size_t* num_encoded_spatial_layers,
|
|
size_t* num_encoded_temporal_layers);
|
|
|
|
// layer_idx -> stats.
|
|
std::map<size_t, std::vector<FrameStatistics>> layer_stats_;
|
|
// layer_idx -> rtp_timestamp -> frame_num.
|
|
std::map<size_t, std::map<size_t, size_t>> rtp_timestamp_to_frame_num_;
|
|
};
|
|
|
|
} // namespace test
|
|
} // namespace webrtc
|
|
|
|
#endif // MODULES_VIDEO_CODING_CODECS_TEST_VIDEOCODEC_TEST_STATS_IMPL_H_
|