Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(29)

Side by Side Diff: webrtc/tools/frame_analyzer/frame_analyzer.cc

Issue 2965593002: Move webrtc/{tools => rtc_tools} (Closed)
Patch Set: Adding back root changes Created 3 years, 5 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
OLDNEW
(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 }
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698