Add optional visualization file writers to VideoProcessor tests.

webrtc/test/testsupport/yuv_frame_writer.cc

Index: webrtc/test/testsupport/yuv_frame_writer.cc
diff --git a/webrtc/test/testsupport/frame_writer.cc b/webrtc/test/testsupport/yuv_frame_writer.cc
similarity index 46%
rename from webrtc/test/testsupport/frame_writer.cc
rename to webrtc/test/testsupport/yuv_frame_writer.cc
index 1b9e8a82efda2f1dd41e78a7ed0a8059e5847a3d..3c0076152e84b4fd0731b2e3707ad7f3b82a7fd5 100644
--- a/webrtc/test/testsupport/frame_writer.cc
+++ b/webrtc/test/testsupport/yuv_frame_writer.cc
@@ -1,5 +1,5 @@
 /*
- *  Copyright (c) 2011 The WebRTC project authors. All Rights Reserved.
+ *  Copyright (c) 2017 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
@@ -8,32 +8,36 @@
  *  be found in the AUTHORS file in the root of the source tree.
  */
 
+#include "webrtc/base/checks.h"
 #include "webrtc/test/testsupport/frame_writer.h"
 
-#include <assert.h>
-
 namespace webrtc {
 namespace test {
 
-FrameWriterImpl::FrameWriterImpl(std::string output_filename,
-                                 size_t frame_length_in_bytes)
+YuvFrameWriterImpl::YuvFrameWriterImpl(std::string output_filename,
+                                       int width,
+                                       int height)
     : output_filename_(output_filename),
-      frame_length_in_bytes_(frame_length_in_bytes),
-      output_file_(NULL) {
-}
+      frame_length_in_bytes_(0),
+      width_(width),
+      height_(height),
+      output_file_(nullptr) {}
 
-FrameWriterImpl::~FrameWriterImpl() {
+YuvFrameWriterImpl::~YuvFrameWriterImpl() {
   Close();
 }
 
-bool FrameWriterImpl::Init() {
-  if (frame_length_in_bytes_ <= 0) {
-    fprintf(stderr, "Frame length must be >0, was %zu\n",
-            frame_length_in_bytes_);
+bool YuvFrameWriterImpl::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);
+
   output_file_ = fopen(output_filename_.c_str(), "wb");
-  if (output_file_ == NULL) {
+  if (output_file_ == nullptr) {
     fprintf(stderr, "Couldn't open output file for writing: %s\n",
             output_filename_.c_str());
     return false;
@@ -41,23 +45,15 @@ bool FrameWriterImpl::Init() {
   return true;
 }
 
-void FrameWriterImpl::Close() {
-  if (output_file_ != NULL) {
-    fclose(output_file_);
-    output_file_ = NULL;
-  }
-}
-
-size_t FrameWriterImpl::FrameLength() { return frame_length_in_bytes_; }
-
-bool FrameWriterImpl::WriteFrame(uint8_t* frame_buffer) {
-  assert(frame_buffer);
-  if (output_file_ == NULL) {
-    fprintf(stderr, "FrameWriter is not initialized (output file is NULL)\n");
+bool YuvFrameWriterImpl::WriteFrame(uint8_t* frame_buffer) {
+  RTC_DCHECK(frame_buffer);
+  if (output_file_ == nullptr) {
+    fprintf(stderr,
+            "YuvFrameWriterImpl is not initialized (output file is NULL)\n");
     return false;
   }
-  size_t bytes_written = fwrite(frame_buffer, 1, frame_length_in_bytes_,
-                                output_file_);
+  size_t bytes_written =
+      fwrite(frame_buffer, 1, frame_length_in_bytes_, output_file_);
   if (bytes_written != frame_length_in_bytes_) {
     fprintf(stderr, "Failed to write %zu bytes to file %s\n",
             frame_length_in_bytes_, output_filename_.c_str());
@@ -66,5 +62,16 @@ bool FrameWriterImpl::WriteFrame(uint8_t* frame_buffer) {
   return true;
 }
 
+void YuvFrameWriterImpl::Close() {
+  if (output_file_ != nullptr) {
+    fclose(output_file_);
+    output_file_ = nullptr;
+  }
+}
+
+size_t YuvFrameWriterImpl::FrameLength() {
+  return frame_length_in_bytes_;
+}
+
 }  // namespace test
 }  // namespace webrtc