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

Unified Diff: webrtc/test/testsupport/frame_reader.cc

Issue 2708103002: Revert of Add optional visualization file writers to VideoProcessor tests. (Closed)
Patch Set: Created 3 years, 10 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 side-by-side diff with in-line comments
Download patch
« no previous file with comments | « webrtc/test/testsupport/frame_reader.h ('k') | webrtc/test/testsupport/frame_reader_unittest.cc » ('j') | no next file with comments »
Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
Index: webrtc/test/testsupport/frame_reader.cc
diff --git a/webrtc/test/testsupport/frame_reader.cc b/webrtc/test/testsupport/frame_reader.cc
new file mode 100644
index 0000000000000000000000000000000000000000..2593afd8c8cb002137fe4216f36d777a5fe19a5d
--- /dev/null
+++ b/webrtc/test/testsupport/frame_reader.cc
@@ -0,0 +1,84 @@
+/*
+ * Copyright (c) 2011 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 "webrtc/test/testsupport/frame_reader.h"
+
+#include <assert.h>
+
+#include "webrtc/api/video/i420_buffer.h"
+#include "webrtc/test/frame_utils.h"
+#include "webrtc/test/testsupport/fileutils.h"
+
+namespace webrtc {
+namespace test {
+
+FrameReaderImpl::FrameReaderImpl(std::string input_filename,
+ int width, int height)
+ : input_filename_(input_filename),
+ width_(width), height_(height),
+ input_file_(NULL) {
+}
+
+FrameReaderImpl::~FrameReaderImpl() {
+ Close();
+}
+
+bool FrameReaderImpl::Init() {
+ if (width_ <= 0 || height_ <= 0) {
+ fprintf(stderr, "Frame width and height must be >0, was %d x %d\n",
+ width_, height_);
+ return false;
+ }
+ frame_length_in_bytes_ =
+ width_ * height_ + 2 * ((width_ + 1) / 2) * ((height_ + 1) / 2);
+
+ input_file_ = fopen(input_filename_.c_str(), "rb");
+ if (input_file_ == NULL) {
+ fprintf(stderr, "Couldn't open input file for reading: %s\n",
+ input_filename_.c_str());
+ return false;
+ }
+ // Calculate total number of frames.
+ size_t source_file_size = GetFileSize(input_filename_);
+ if (source_file_size <= 0u) {
+ fprintf(stderr, "Found empty file: %s\n", input_filename_.c_str());
+ return false;
+ }
+ number_of_frames_ = static_cast<int>(source_file_size /
+ frame_length_in_bytes_);
+ return true;
+}
+
+void FrameReaderImpl::Close() {
+ if (input_file_ != NULL) {
+ fclose(input_file_);
+ input_file_ = NULL;
+ }
+}
+
+rtc::scoped_refptr<I420Buffer> FrameReaderImpl::ReadFrame() {
+ if (input_file_ == NULL) {
+ fprintf(stderr, "FrameReader is not initialized (input file is NULL)\n");
+ return nullptr;
+ }
+ rtc::scoped_refptr<I420Buffer> buffer(
+ ReadI420Buffer(width_, height_, input_file_));
+ if (!buffer && ferror(input_file_)) {
+ fprintf(stderr, "Error reading from input file: %s\n",
+ input_filename_.c_str());
+ }
+ return buffer;
+}
+
+size_t FrameReaderImpl::FrameLength() { return frame_length_in_bytes_; }
+int FrameReaderImpl::NumberOfFrames() { return number_of_frames_; }
+
+} // namespace test
+} // namespace webrtc
« no previous file with comments | « webrtc/test/testsupport/frame_reader.h ('k') | webrtc/test/testsupport/frame_reader_unittest.cc » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698