diff options
author | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-19 00:47:55 +0000 |
---|---|---|
committer | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-19 00:47:55 +0000 |
commit | 26a029d407be480d791972afb5975cf62c9360a6 (patch) | |
tree | f435a8308119effd964b339f76abb83a57c29483 /third_party/libwebrtc/rtc_tools/psnr_ssim_analyzer | |
parent | Initial commit. (diff) | |
download | firefox-26a029d407be480d791972afb5975cf62c9360a6.tar.xz firefox-26a029d407be480d791972afb5975cf62c9360a6.zip |
Adding upstream version 124.0.1.upstream/124.0.1
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'third_party/libwebrtc/rtc_tools/psnr_ssim_analyzer')
-rw-r--r-- | third_party/libwebrtc/rtc_tools/psnr_ssim_analyzer/psnr_ssim_analyzer.cc | 108 |
1 files changed, 108 insertions, 0 deletions
diff --git a/third_party/libwebrtc/rtc_tools/psnr_ssim_analyzer/psnr_ssim_analyzer.cc b/third_party/libwebrtc/rtc_tools/psnr_ssim_analyzer/psnr_ssim_analyzer.cc new file mode 100644 index 0000000000..bcbb1bbe7c --- /dev/null +++ b/third_party/libwebrtc/rtc_tools/psnr_ssim_analyzer/psnr_ssim_analyzer.cc @@ -0,0 +1,108 @@ +/* + * 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. + */ + +#include <stdio.h> +#include <stdlib.h> + +#include <algorithm> +#include <string> + +#include "absl/flags/flag.h" +#include "absl/flags/parse.h" +#include "absl/flags/usage.h" +#include "api/scoped_refptr.h" +#include "api/video/video_frame_buffer.h" +#include "rtc_tools/frame_analyzer/video_quality_analysis.h" +#include "rtc_tools/video_file_reader.h" + +ABSL_FLAG(std::string, + results_file, + "results.txt", + "The full name of the file where the results will be written"); +ABSL_FLAG(std::string, + reference_file, + "ref.yuv", + "The reference YUV file to compare against"); +ABSL_FLAG(std::string, + test_file, + "test.yuv", + "The test YUV file to run the analysis for"); + +void CompareFiles( + const rtc::scoped_refptr<webrtc::test::Video>& reference_video, + const rtc::scoped_refptr<webrtc::test::Video>& test_video, + const char* results_file_name) { + FILE* results_file = fopen(results_file_name, "w"); + + const size_t num_frames = std::min(reference_video->number_of_frames(), + test_video->number_of_frames()); + for (size_t i = 0; i < num_frames; ++i) { + const rtc::scoped_refptr<webrtc::I420BufferInterface> ref_buffer = + reference_video->GetFrame(i); + const rtc::scoped_refptr<webrtc::I420BufferInterface> test_buffer = + test_video->GetFrame(i); + + // Calculate the PSNR and SSIM. + double result_psnr = webrtc::test::Psnr(ref_buffer, test_buffer); + double result_ssim = webrtc::test::Ssim(ref_buffer, test_buffer); + fprintf(results_file, "Frame: %zu, PSNR: %f, SSIM: %f\n", i, result_psnr, + result_ssim); + } + + fclose(results_file); +} + +/* + * A tool running PSNR and SSIM analysis on two videos - a reference video and a + * test video. The two videos should be I420 Y4M videos. + * The tool just runs PSNR and SSIM on the corresponding frames in the test and + * the reference videos until either the first or the second video runs out of + * frames. The result is written in a results text file in the format: + * Frame: <frame_number>, PSNR: <psnr_value>, SSIM: <ssim_value> + * Frame: <frame_number>, ........ + * + * The max value for PSNR is 48.0 (between equal frames), as for SSIM it is 1.0. + * + * Usage: + * psnr_ssim_analyzer --reference_file=<name_of_file> --test_file=<name_of_file> + * --results_file=<name_of_file> + */ +int main(int argc, char* argv[]) { + absl::SetProgramUsageMessage( + "Runs PSNR and SSIM on two I420 videos and write the" + "results in a file.\n" + "Example usage:\n" + "./psnr_ssim_analyzer --reference_file=ref.yuv " + "--test_file=test.yuv --results_file=results.txt\n"); + absl::ParseCommandLine(argc, argv); + + rtc::scoped_refptr<webrtc::test::Video> reference_video = + webrtc::test::OpenY4mFile(absl::GetFlag(FLAGS_reference_file)); + rtc::scoped_refptr<webrtc::test::Video> test_video = + webrtc::test::OpenY4mFile(absl::GetFlag(FLAGS_test_file)); + + if (!reference_video || !test_video) { + fprintf(stderr, "Error opening video files\n"); + return 0; + } + if (reference_video->width() != test_video->width() || + reference_video->height() != test_video->height()) { + fprintf(stderr, + "Reference and test video files do not have same size: %dx%d " + "versus %dx%d\n", + reference_video->width(), reference_video->height(), + test_video->width(), test_video->height()); + return 0; + } + + CompareFiles(reference_video, test_video, + absl::GetFlag(FLAGS_results_file).c_str()); + return 0; +} |