Added new VideoFrameBuffer methods Data[YUV]() etc.

webrtc/common_video/i420_buffer_pool.cc

 /*
  *  Copyright (c) 2015 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/common_video/include/i420_buffer_pool.h"
 
 #include "webrtc/base/checks.h"
 
 namespace {
 
 // One extra indirection is needed to make |HasOneRef| work.
 class PooledI420Buffer : public webrtc::VideoFrameBuffer {
  public:
   explicit PooledI420Buffer(
       const rtc::scoped_refptr<webrtc::I420Buffer>& buffer)
       : buffer_(buffer) {}
 
  private:
   ~PooledI420Buffer() override {}
 
   int width() const override { return buffer_->width(); }
   int height() const override { return buffer_->height(); }
-  const uint8_t* data(webrtc::PlaneType type) const override {
-    return buffer_->data(type);
+  const uint8_t* DataY() const override { return buffer_->DataY(); }
+  const uint8_t* DataU() const override { return buffer_->DataU(); }
+  const uint8_t* DataV() const override { return buffer_->DataV(); }
+
+  bool IsMutable() { return HasOneRef(); }
+  // Make the IsMutable() check here instead of in |buffer_|, because the pool
+  // also has a reference to |buffer_|.
+  uint8_t* MutableDataY() override {
+    RTC_DCHECK(IsMutable());
+    return const_cast<uint8_t*>(buffer_->DataY());
   }
-  bool IsMutable() { return HasOneRef(); }
-  uint8_t* MutableData(webrtc::PlaneType type) override {
-    // Make the HasOneRef() check here instead of in |buffer_|, because the pool
-    // also has a reference to |buffer_|.
+  uint8_t* MutableDataU() override {
     RTC_DCHECK(IsMutable());
-    return const_cast<uint8_t*>(buffer_->data(type));
+    return const_cast<uint8_t*>(buffer_->DataU());
   }
-  int stride(webrtc::PlaneType type) const override {
-    return buffer_->stride(type);
+  uint8_t* MutableDataV() override {
+    RTC_DCHECK(IsMutable());
+    return const_cast<uint8_t*>(buffer_->DataV());
   }
+  int StrideY() const override { return buffer_->StrideY(); }
+  int StrideU() const override { return buffer_->StrideU(); }
+  int StrideV() const override { return buffer_->StrideV(); }
   void* native_handle() const override { return nullptr; }
 
   rtc::scoped_refptr<VideoFrameBuffer> NativeToI420Buffer() override {
     RTC_NOTREACHED();
     return nullptr;
   }
 
   friend class rtc::RefCountedObject<PooledI420Buffer>;
   rtc::scoped_refptr<webrtc::I420Buffer> buffer_;
 };
@@ skipping 35 matching lines @@
   // Allocate new buffer.
   rtc::scoped_refptr<I420Buffer> buffer = new rtc::RefCountedObject<I420Buffer>(
       width, height);
   if (zero_initialize_)
     buffer->InitializeData();
   buffers_.push_back(buffer);
   return new rtc::RefCountedObject<PooledI420Buffer>(buffers_.back());
 }
 
 }  // namespace webrtc