6e2d012b58
This CL adds support for reading .y4m files to the infra in video_quality_analysis.cc, adding new functions ExtractFrameFromYuvFile() and ExtractFrameFromY4mFile(), instad of the previous ExtractFrameFromI420(). The decision as to which one to use is taken from the file extension, if it is .y4m then is considered a YUV4MPEG file, otherwise is taken as a raw .yuv file. It also removes the pseudo duplicated function GetNextI420Frame(), that is used from psnr_ssim_analyzer.c, and adds support for y4m files there. Tested/validated via local compile-run. YUV4MPEG is a trivial container with a file header and a per-frame header, see [1] [1] http://wiki.multimedia.cx/index.php?title=YUV4MPEG2 BUG=https://code.google.com/p/chromium/issues/detail?id=343504 git-svn-id: http://webrtc.googlecode.com/svn/trunk@5702 4adac7df-926f-26a2-2b94-8c16560cd09d
114 lines
4.8 KiB
C++
114 lines
4.8 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 WEBRTC_TOOLS_FRAME_ANALYZER_VIDEO_QUALITY_ANALYSIS_H_
|
|
#define WEBRTC_TOOLS_FRAME_ANALYZER_VIDEO_QUALITY_ANALYSIS_H_
|
|
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
#include "third_party/libyuv/include/libyuv/compare.h"
|
|
#include "third_party/libyuv/include/libyuv/convert.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 {
|
|
std::vector<AnalysisResult> frames;
|
|
};
|
|
|
|
enum VideoAnalysisMetricsType {kPSNR, kSSIM};
|
|
|
|
// 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 should provide a statistics file along
|
|
// with the test video. The stats file contains the connection between the
|
|
// actual frames in the test file and their position in the reference video, so
|
|
// that the analysis could run with the right frames from both videos. The stats
|
|
// file should be in the form 'frame_xxxx yyyy', where xxxx is the consecutive
|
|
// number of the frame in the test video, and yyyy is the equivalent frame in
|
|
// the reference video. The stats file could be produced by
|
|
// tools/barcode_tools/barcode_decoder.py. This script decodes the barcodes
|
|
// integrated in every video and generates the stats file. If three was some
|
|
// problem with the decoding there would be 'Barcode error' instead of yyyy.
|
|
void RunAnalysis(const char* reference_file_name, const char* test_file_name,
|
|
const char* stats_file_name, int width, int height,
|
|
ResultsContainer* results);
|
|
|
|
// Compute PSNR or SSIM for an I420 frame (all planes). When we are calculating
|
|
// PSNR values, the max return value (in the case where the test and reference
|
|
// frames are exactly the same) will be 48. In the case of SSIM the max return
|
|
// value will be 1.
|
|
double CalculateMetrics(VideoAnalysisMetricsType video_metrics_type,
|
|
const uint8* ref_frame, const uint8* test_frame,
|
|
int width, int height);
|
|
|
|
// 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);
|
|
|
|
// Calculates max repeated and skipped frames and prints them to stdout in a
|
|
// format that is compatible with Chromium performance numbers.
|
|
void PrintMaxRepeatedAndSkippedFrames(const std::string& label,
|
|
const std::string& stats_file_name);
|
|
|
|
// Similar to the above, but will print to the specified file handle.
|
|
void PrintMaxRepeatedAndSkippedFrames(FILE* output, const std::string& label,
|
|
const std::string& stats_file_name);
|
|
|
|
// Gets the next line from an open stats file.
|
|
bool GetNextStatsLine(FILE* stats_file, char* line);
|
|
|
|
// Calculates the size of a I420 frame if given the width and height.
|
|
int GetI420FrameSize(int width, int height);
|
|
|
|
// Extract the sequence of the frame in the video. I.e. if line is
|
|
// frame_0023 0284, we will get 23.
|
|
int ExtractFrameSequenceNumber(std::string line);
|
|
|
|
// Checks if there is 'Barcode error' for the given line.
|
|
bool IsThereBarcodeError(std::string line);
|
|
|
|
// Extract the frame number in the reference video. I.e. if line is
|
|
// frame_0023 0284, we will get 284.
|
|
int ExtractDecodedFrameNumber(std::string line);
|
|
|
|
// Extracts an I420 frame at position frame_number from the raw YUV file.
|
|
bool ExtractFrameFromYuvFile(const char* i420_file_name, int width, int height,
|
|
int frame_number, uint8* result_frame);
|
|
|
|
// Extracts an I420 frame at position frame_number from the Y4M file. The first
|
|
// frame has corresponded |frame_number| 0.
|
|
bool ExtractFrameFromY4mFile(const char* i420_file_name, int width, int height,
|
|
int frame_number, uint8* result_frame);
|
|
|
|
|
|
} // namespace test
|
|
} // namespace webrtc
|
|
|
|
#endif // WEBRTC_TOOLS_FRAME_ANALYZER_VIDEO_QUALITY_ANALYSIS_H_
|