Reland: GpuMemoryBuffer interface change: Map(**) to Map() and memory(plane); stride(plane)

components/mus/gles2/mojo_gpu_memory_buffer.cc

 // Copyright 2015 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "components/mus/gles2/mojo_gpu_memory_buffer.h"
 
 #include "base/logging.h"
 #include "base/memory/shared_memory.h"
 #include "base/numerics/safe_conversions.h"
 #include "ui/gfx/buffer_format_util.h"
@@ skipping 25 matching lines @@
 
 MojoGpuMemoryBufferImpl* MojoGpuMemoryBufferImpl::FromClientBuffer(
     ClientBuffer buffer) {
   return reinterpret_cast<MojoGpuMemoryBufferImpl*>(buffer);
 }
 
 const unsigned char* MojoGpuMemoryBufferImpl::GetMemory() const {
   return static_cast<const unsigned char*>(shared_memory_->memory());
 }
 
-bool MojoGpuMemoryBufferImpl::Map(void** data) {
+bool MojoGpuMemoryBufferImpl::Map() {
   DCHECK(!mapped_);
   if (!shared_memory_->Map(gfx::BufferSizeForBufferFormat(size_, format_)))
     return false;
   mapped_ = true;
-  size_t offset = 0;
-  int num_planes =
-      static_cast<int>(gfx::NumberOfPlanesForBufferFormat(format_));
-  for (int i = 0; i < num_planes; ++i) {
-    data[i] = reinterpret_cast<uint8*>(shared_memory_->memory()) + offset;
-    offset +=
-        gfx::RowSizeForBufferFormat(size_.width(), format_, i) *
-        (size_.height() / gfx::SubsamplingFactorForBufferFormat(format_, i));
-  }
   return true;
 }
 
+void* MojoGpuMemoryBufferImpl::memory(size_t plane) {
+  DCHECK(mapped_);
+  DCHECK_LT(plane, gfx::NumberOfPlanesForBufferFormat(format_));
+  return reinterpret_cast<uint8*>(shared_memory_->memory()) +
+         gfx::BufferOffsetForBufferFormat(size_, format_, plane);
+}
+
 void MojoGpuMemoryBufferImpl::Unmap() {
   DCHECK(mapped_);
   shared_memory_->Unmap();
   mapped_ = false;
 }
 
 gfx::Size MojoGpuMemoryBufferImpl::GetSize() const {
   return size_;
 }
 
 gfx::BufferFormat MojoGpuMemoryBufferImpl::GetFormat() const {
   return format_;
 }
 
-void MojoGpuMemoryBufferImpl::GetStride(int* stride) const {
-  int num_planes =
-      static_cast<int>(gfx::NumberOfPlanesForBufferFormat(format_));
-  for (int i = 0; i < num_planes; ++i)
-    stride[i] = base::checked_cast<int>(
-        gfx::RowSizeForBufferFormat(size_.width(), format_, i));
+int MojoGpuMemoryBufferImpl::stride(size_t plane) const {
+  DCHECK_LT(plane, gfx::NumberOfPlanesForBufferFormat(format_));
+  return base::checked_cast<int>(gfx::RowSizeForBufferFormat(
+      size_.width(), format_, static_cast<int>(plane)));
 }
 
 gfx::GpuMemoryBufferId MojoGpuMemoryBufferImpl::GetId() const {
   return gfx::GpuMemoryBufferId(0);
 }
 
 gfx::GpuMemoryBufferHandle MojoGpuMemoryBufferImpl::GetHandle() const {
   gfx::GpuMemoryBufferHandle handle;
   handle.type = gfx::SHARED_MEMORY_BUFFER;
   handle.handle = shared_memory_->handle();
   return handle;
 }
 
 ClientBuffer MojoGpuMemoryBufferImpl::AsClientBuffer() {
   return reinterpret_cast<ClientBuffer>(this);
 }
 
 }  // namespace mus