Update test code to use I420Buffer when writing pixel data.

webrtc/test/frame_generator.cc

 /*
  *  Copyright (c) 2013 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/frame_generator.h"
 
 #include <math.h>
 #include <stdio.h>
 #include <string.h>
 
 #include <memory>
 
 #include "webrtc/base/checks.h"
 #include "webrtc/common_video/libyuv/include/webrtc_libyuv.h"
 #include "webrtc/system_wrappers/include/clock.h"
+#include "libyuv/convert.h"
 
 namespace webrtc {
 namespace test {
 namespace {
 
 class ChromaGenerator : public FrameGenerator {
  public:
   ChromaGenerator(size_t width, size_t height)
-      : angle_(0.0), width_(width), height_(height) {
-    assert(width > 0);
-    assert(height > 0);
+      : width_(width), height_(height), angle_(0.0) {
+    RTC_CHECK(width_ > 0);
+    RTC_CHECK(height_ > 0);
+    half_width_ = (width_ + 1) / 2;
+    y_size_ = width_ * height_;
+    uv_size_ = half_width_ * ((height_ + 1) / 2);
   }
 
   VideoFrame* NextFrame() override {
-    frame_.CreateEmptyFrame(static_cast<int>(width_),
-                            static_cast<int>(height_),
-                            static_cast<int>(width_),
-                            static_cast<int>((width_ + 1) / 2),
-                            static_cast<int>((width_ + 1) / 2));
     angle_ += 30.0;
     uint8_t u = fabs(sin(angle_)) * 0xFF;
     uint8_t v = fabs(cos(angle_)) * 0xFF;
 
-    memset(frame_.video_frame_buffer()->MutableDataY(), 0x80,
-           frame_.allocated_size(kYPlane));
-    memset(frame_.video_frame_buffer()->MutableDataU(), u,
-           frame_.allocated_size(kUPlane));
-    memset(frame_.video_frame_buffer()->MutableDataV(), v,
-           frame_.allocated_size(kVPlane));
-    return &frame_;
+    // Ensure stride == width.
+    rtc::scoped_refptr<I420Buffer> buffer(I420Buffer::Create(
+        static_cast<int>(width_), static_cast<int>(height_),
+        static_cast<int>(width_), static_cast<int>(half_width_),
+        static_cast<int>(half_width_)));
+
+    memset(buffer->MutableDataY(), 0x80, y_size_);
+    memset(buffer->MutableDataU(), u, uv_size_);
+    memset(buffer->MutableDataV(), v, uv_size_);
+
+    frame_.reset(new VideoFrame(buffer, 0, 0, webrtc::kVideoRotation_0));
+    return frame_.get();
   }
 
  private:
+  int width_;
+  int height_;
+  int half_width_;
   double angle_;
-  size_t width_;
-  size_t height_;
-  VideoFrame frame_;
+  size_t y_size_;
+  size_t uv_size_;
+  std::unique_ptr<VideoFrame> frame_;
 };
 
 class YuvFileGenerator : public FrameGenerator {
  public:
   YuvFileGenerator(std::vector<FILE*> files,
                    size_t width,
                    size_t height,
                    int frame_repeat_count)
       : file_index_(0),
         files_(files),
@@ skipping 14 matching lines @@
     for (FILE* file : files_)
       fclose(file);
   }
 
   VideoFrame* NextFrame() override {
     if (current_display_count_ == 0)
       ReadNextFrame();
     if (++current_display_count_ >= frame_display_count_)
       current_display_count_ = 0;
 
-    // If this is the last repeatition of this frame, it's OK to use the
-    // original instance, otherwise use a copy.
-    if (current_display_count_ == frame_display_count_)
-      return &last_read_frame_;
-
-    temp_frame_copy_.CopyFrame(last_read_frame_);
-    return &temp_frame_copy_;
+    temp_frame_.reset(
+        new VideoFrame(last_read_buffer_, 0, 0, webrtc::kVideoRotation_0));
+    return temp_frame_.get();
   }
 
+  // TODO(nisse): Have a frame reader in one place. And read directly
+  // into the planes of an I420Buffer, the extra copying below is silly.
   void ReadNextFrame() {
     size_t bytes_read =
         fread(frame_buffer_.get(), 1, frame_size_, files_[file_index_]);
     if (bytes_read < frame_size_) {
       // No more frames to read in this file, rewind and move to next file.
       rewind(files_[file_index_]);
       file_index_ = (file_index_ + 1) % files_.size();
       bytes_read = fread(frame_buffer_.get(), 1, frame_size_,
           files_[file_index_]);
       assert(bytes_read >= frame_size_);
     }
 
-    last_read_frame_.CreateEmptyFrame(
+    size_t half_width = (width_ + 1) / 2;
+    size_t size_y = width_ * height_;
+    size_t size_uv = half_width * ((height_ + 1) / 2);
+    last_read_buffer_ = I420Buffer::Create(
         static_cast<int>(width_), static_cast<int>(height_),
-        static_cast<int>(width_), static_cast<int>((width_ + 1) / 2),
-        static_cast<int>((width_ + 1) / 2));
-
-    ConvertToI420(kI420, frame_buffer_.get(), 0, 0, static_cast<int>(width_),
-                  static_cast<int>(height_), 0, kVideoRotation_0,
-                  &last_read_frame_);
+        static_cast<int>(width_), static_cast<int>(half_width),
+        static_cast<int>(half_width));
+    libyuv::I420Copy(
+        frame_buffer_.get(), static_cast<int>(width_),
+        frame_buffer_.get() + size_y, static_cast<int>(half_width),
+        frame_buffer_.get() + size_y + size_uv, static_cast<int>(half_width),
+        last_read_buffer_->MutableDataY(), last_read_buffer_->StrideY(),
+        last_read_buffer_->MutableDataU(), last_read_buffer_->StrideU(),
+        last_read_buffer_->MutableDataV(), last_read_buffer_->StrideV(),
+        static_cast<int>(width_), static_cast<int>(height_));
   }
 
  private:
   size_t file_index_;
   const std::vector<FILE*> files_;
   const size_t width_;
   const size_t height_;
   const size_t frame_size_;
   const std::unique_ptr<uint8_t[]> frame_buffer_;
   const int frame_display_count_;
   int current_display_count_;
-  VideoFrame last_read_frame_;
-  VideoFrame temp_frame_copy_;
+  rtc::scoped_refptr<I420Buffer> last_read_buffer_;
+  std::unique_ptr<VideoFrame> temp_frame_;
 };
 
 class ScrollingImageFrameGenerator : public FrameGenerator {
  public:
   ScrollingImageFrameGenerator(Clock* clock,
                                const std::vector<FILE*>& files,
                                size_t source_width,
                                size_t source_height,
                                size_t target_width,
                                size_t target_height,
@@ skipping 133 matching lines @@
     files.push_back(file);
   }
 
   return new ScrollingImageFrameGenerator(
       clock, files, source_width, source_height, target_width, target_height,
       scroll_time_ms, pause_time_ms);
 }
 
 }  // namespace test
 }  // namespace webrtc