OLD | NEW |
| (Empty) |
1 /* | |
2 * Copyright (c) 2012 The WebRTC project authors. All Rights Reserved. | |
3 * | |
4 * Use of this source code is governed by a BSD-style license | |
5 * that can be found in the LICENSE file in the root of the source | |
6 * tree. An additional intellectual property rights grant can be found | |
7 * in the file PATENTS. All contributing project authors may | |
8 * be found in the AUTHORS file in the root of the source tree. | |
9 */ | |
10 | |
11 #include <stdio.h> | |
12 #include <stdlib.h> | |
13 | |
14 #include <map> | |
15 #include <string> | |
16 #include <vector> | |
17 | |
18 #include "webrtc/tools/frame_analyzer/video_quality_analysis.h" | |
19 #include "webrtc/tools/simple_command_line_parser.h" | |
20 | |
21 /* | |
22 * A command line tool running PSNR and SSIM on a reference video and a test | |
23 * video. The test video is a record of the reference video which can start at | |
24 * an arbitrary point. It is possible that there will be repeated frames or | |
25 * skipped frames as well. In order to have a way to compare corresponding | |
26 * frames from the two videos, two stats files should be provided. One for the | |
27 * reference video and one for the test video. The stats file | |
28 * is a text file assumed to be in the format: | |
29 * frame_xxxx yyyy where xxxx is the frame number in and yyyy is the | |
30 * corresponding barcode. The video files should be 1420 YUV videos. | |
31 * The tool prints the result to standard output in the Chromium perf format: | |
32 * RESULT <metric>:<label>= <values> | |
33 * | |
34 * The max value for PSNR is 48.0 (between equal frames), as for SSIM it is 1.0. | |
35 * | |
36 * Usage: | |
37 * frame_analyzer --label=<test_label> --reference_file=<name_of_file> | |
38 * --test_file_ref=<name_of_file> --stats_file_test=<name_of_file> | |
39 * --stats_file=<name_of_file> --width=<frame_width> | |
40 * --height=<frame_height> | |
41 */ | |
42 int main(int argc, char** argv) { | |
43 std::string program_name = argv[0]; | |
44 std::string usage = | |
45 "Compares the output video with the initially sent video." | |
46 "\nExample usage:\n" + | |
47 program_name + | |
48 " --reference_file=ref.yuv --test_file=test.yuv --width=320 " | |
49 "--height=240\n" | |
50 "Command line flags:\n" | |
51 " - width(int): The width of the reference and test files. Default: -1\n" | |
52 " - height(int): The height of the reference and test files. " | |
53 " Default: -1\n" | |
54 " - label(string): The label to use for the perf output." | |
55 " Default: MY_TEST\n" | |
56 " - stats_file_ref(string): The path to the stats file that will be" | |
57 " produced for the reference video file." | |
58 " Default: stats_ref.txt\n" | |
59 " - stats_file_test(string): The path to the stats file that will be" | |
60 " produced for the test video file." | |
61 " Default: stats_test.txt\n" | |
62 " - reference_file(string): The reference YUV file to compare against." | |
63 " Default: ref.yuv\n" | |
64 " - test_file(string): The test YUV file to run the analysis for." | |
65 " Default: test_file.yuv\n"; | |
66 | |
67 webrtc::test::CommandLineParser parser; | |
68 | |
69 // Init the parser and set the usage message | |
70 parser.Init(argc, argv); | |
71 parser.SetUsageMessage(usage); | |
72 | |
73 parser.SetFlag("width", "-1"); | |
74 parser.SetFlag("height", "-1"); | |
75 parser.SetFlag("label", "MY_TEST"); | |
76 parser.SetFlag("stats_file_ref", "stats_ref.txt"); | |
77 parser.SetFlag("stats_file_test", "stats_test.txt"); | |
78 parser.SetFlag("reference_file", "ref.yuv"); | |
79 parser.SetFlag("test_file", "test.yuv"); | |
80 parser.SetFlag("help", "false"); | |
81 | |
82 parser.ProcessFlags(); | |
83 if (parser.GetFlag("help") == "true") { | |
84 parser.PrintUsageMessage(); | |
85 exit(EXIT_SUCCESS); | |
86 } | |
87 parser.PrintEnteredFlags(); | |
88 | |
89 int width = strtol((parser.GetFlag("width")).c_str(), NULL, 10); | |
90 int height = strtol((parser.GetFlag("height")).c_str(), NULL, 10); | |
91 | |
92 if (width <= 0 || height <= 0) { | |
93 fprintf(stderr, "Error: width or height cannot be <= 0!\n"); | |
94 return -1; | |
95 } | |
96 | |
97 webrtc::test::ResultsContainer results; | |
98 | |
99 webrtc::test::RunAnalysis(parser.GetFlag("reference_file").c_str(), | |
100 parser.GetFlag("test_file").c_str(), | |
101 parser.GetFlag("stats_file_ref").c_str(), | |
102 parser.GetFlag("stats_file_test").c_str(), width, | |
103 height, &results); | |
104 | |
105 std::string label = parser.GetFlag("label"); | |
106 webrtc::test::PrintAnalysisResults(label, &results); | |
107 webrtc::test::PrintMaxRepeatedAndSkippedFrames( | |
108 label, parser.GetFlag("stats_file_ref"), | |
109 parser.GetFlag("stats_file_test")); | |
110 } | |
OLD | NEW |