9bb55fc09b
This is a reland of d65e143801a7aaa9affdb939ea836aec1955cdcc The binary for frame_analyzer.cpp is precompiled and stored in the cloud, so it won't automatically pick up change to the source file. Therefore, restore all old code to be backwards compatible. Original change's description: > Update video_quality_analysis to align videos instead of using barcodes > > This CL is a follow-up to the previous CL > https://webrtc-review.googlesource.com/c/src/+/94773 that added generic > logic for aligning videos. This will allow us to easily extend > video_quality_analysis with new sophisticated video quality metrics. > Also, we can use any kind of video that does not necessarily need to > contain bar codes. Removing the need to decode barcodes also leads to a > big speedup for the tests. > > Bug: webrtc:9642 > Change-Id: I74b0d630b3e1ed44781ad024115ded3143e28f50 > Reviewed-on: https://webrtc-review.googlesource.com/94845 > Reviewed-by: Paulina Hensman <phensman@webrtc.org> > Reviewed-by: Patrik Höglund <phoglund@webrtc.org> > Commit-Queue: Magnus Jedvert <magjed@webrtc.org> > Cr-Commit-Position: refs/heads/master@{#24423} TBR=phensman@webrtc.org,phoglund@webrtc.org Bug: webrtc:9642 Change-Id: Id8d129ce103284504c67690f8363c03eaae3eee7 Reviewed-on: https://webrtc-review.googlesource.com/96000 Reviewed-by: Magnus Jedvert <magjed@webrtc.org> Reviewed-by: Patrik Höglund <phoglund@webrtc.org> Commit-Queue: Magnus Jedvert <magjed@webrtc.org> Cr-Commit-Position: refs/heads/master@{#24429}
104 lines
3.9 KiB
C++
104 lines
3.9 KiB
C++
/*
|
|
* Copyright (c) 2012 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 RTC_TOOLS_FRAME_ANALYZER_VIDEO_QUALITY_ANALYSIS_H_
|
|
#define RTC_TOOLS_FRAME_ANALYZER_VIDEO_QUALITY_ANALYSIS_H_
|
|
|
|
#include <string>
|
|
#include <utility>
|
|
#include <vector>
|
|
|
|
#include "api/video/i420_buffer.h"
|
|
#include "rtc_tools/y4m_file_reader.h"
|
|
|
|
namespace webrtc {
|
|
namespace test {
|
|
|
|
struct AnalysisResult {
|
|
AnalysisResult() {}
|
|
AnalysisResult(int frame_number, double psnr_value, double ssim_value)
|
|
: frame_number(frame_number),
|
|
psnr_value(psnr_value),
|
|
ssim_value(ssim_value) {}
|
|
int frame_number;
|
|
double psnr_value;
|
|
double ssim_value;
|
|
};
|
|
|
|
struct ResultsContainer {
|
|
ResultsContainer();
|
|
~ResultsContainer();
|
|
|
|
std::vector<AnalysisResult> frames;
|
|
int max_repeated_frames;
|
|
int max_skipped_frames;
|
|
int total_skipped_frames;
|
|
int decode_errors_ref;
|
|
int decode_errors_test;
|
|
};
|
|
|
|
// A function to run the PSNR and SSIM analysis on the test file. The test file
|
|
// comprises the frames that were captured during the quality measurement test.
|
|
// There may be missing or duplicate frames. Also the frames start at a random
|
|
// position in the original video. We also need to provide a map from test frame
|
|
// indices to reference frame indices.
|
|
std::vector<AnalysisResult> RunAnalysis(
|
|
const rtc::scoped_refptr<webrtc::test::Video>& reference_video,
|
|
const rtc::scoped_refptr<webrtc::test::Video>& test_video,
|
|
const std::vector<size_t>& test_frame_indices);
|
|
|
|
// Compute PSNR for an I420 buffer (all planes). The max return value (in the
|
|
// case where the test and reference frames are exactly the same) will be 48.
|
|
double Psnr(const rtc::scoped_refptr<I420BufferInterface>& ref_buffer,
|
|
const rtc::scoped_refptr<I420BufferInterface>& test_buffer);
|
|
|
|
// Compute SSIM for an I420 buffer (all planes). The max return value (in the
|
|
// case where the test and reference frames are exactly the same) will be 1.
|
|
double Ssim(const rtc::scoped_refptr<I420BufferInterface>& ref_buffer,
|
|
const rtc::scoped_refptr<I420BufferInterface>& test_buffer);
|
|
|
|
// Prints the result from the analysis in Chromium performance
|
|
// numbers compatible format to stdout. If the results object contains no frames
|
|
// no output will be written.
|
|
void PrintAnalysisResults(const std::string& label, ResultsContainer* results);
|
|
|
|
// Similar to the above, but will print to the specified file handle.
|
|
void PrintAnalysisResults(FILE* output,
|
|
const std::string& label,
|
|
ResultsContainer* results);
|
|
|
|
struct Cluster {
|
|
// Corresponding reference frame index for this cluster.
|
|
size_t index;
|
|
// The number of sequential frames that mapped to the same reference frame
|
|
// index.
|
|
int number_of_repeated_frames;
|
|
};
|
|
|
|
// Clusters sequentially repeated frames. For example, the sequence {100, 102,
|
|
// 102, 103} will be mapped to {{100, 1}, {102, 2}, {103, 1}}.
|
|
std::vector<Cluster> CalculateFrameClusters(const std::vector<size_t>& indices);
|
|
|
|
// Get number of max sequentially repeated frames in the test video. This number
|
|
// will be one if we only store unique frames in the test video.
|
|
int GetMaxRepeatedFrames(const std::vector<Cluster>& clusters);
|
|
|
|
// Get the longest sequence of skipped reference frames. This corresponds to the
|
|
// longest freeze in the test video.
|
|
int GetMaxSkippedFrames(const std::vector<Cluster>& clusters);
|
|
|
|
// Get total number of skipped frames in the test video.
|
|
int GetTotalNumberOfSkippedFrames(const std::vector<Cluster>& clusters);
|
|
|
|
} // namespace test
|
|
} // namespace webrtc
|
|
|
|
#endif // RTC_TOOLS_FRAME_ANALYZER_VIDEO_QUALITY_ANALYSIS_H_
|