Add PERSISTENT_MAP usage for GpuMemoryBuffers.

content/common/gpu/client/gpu_memory_buffer_impl_unittest.cc

 // Copyright 2014 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 "content/common/gpu/client/gpu_memory_buffer_impl.h"
 
 #include "base/bind.h"
 #include "content/common/gpu/gpu_memory_buffer_factory.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
@@ skipping 61 matching lines @@
 
     // Check if destruction callback is executed when deleting the buffer.
     buffer.reset();
     EXPECT_EQ(0, buffer_count_);
   }
 }
 
 TEST_P(GpuMemoryBufferImplTest, Map) {
   const int kBufferId = 1;
 
-  gfx::Size buffer_size(2, 2);
+  // Use a multiple of 4 for both dimensions to support compressed formats.
+  gfx::Size buffer_size(4, 4);
 
   for (auto configuration : supported_configurations_) {
     if (configuration.usage != gfx::GpuMemoryBuffer::MAP)
       continue;
 
     scoped_ptr<GpuMemoryBufferImpl> buffer(
         GpuMemoryBufferImpl::CreateFromHandle(
             CreateGpuMemoryBuffer(kBufferId, buffer_size, configuration.format,
                                   configuration.usage),
             buffer_size, configuration.format,
@@ skipping 38 matching lines @@
                          data.get(), row_size_in_bytes),
                   0);
       }
     }
 
     buffer->Unmap();
     EXPECT_FALSE(buffer->IsMapped());
   }
 }
 
+TEST_P(GpuMemoryBufferImplTest, PersistentMap) {
+  const int kBufferId = 1;
+
+  // Use a multiple of 4 for both dimensions to support compressed formats.
+  gfx::Size buffer_size(4, 4);
+
+  for (auto configuration : supported_configurations_) {
+    if (configuration.usage != gfx::GpuMemoryBuffer::PERSISTENT_MAP)
+      continue;
+
+    scoped_ptr<GpuMemoryBufferImpl> buffer(
+        GpuMemoryBufferImpl::CreateFromHandle(
+            CreateGpuMemoryBuffer(kBufferId, buffer_size, configuration.format,
+                                  configuration.usage),
+            buffer_size, configuration.format,
+            base::Bind(&GpuMemoryBufferImplTest::DestroyGpuMemoryBuffer,
+                       base::Unretained(this), kBufferId)));
+    ASSERT_TRUE(buffer);
+    EXPECT_FALSE(buffer->IsMapped());
+
+    size_t num_planes =
+        GpuMemoryBufferImpl::NumberOfPlanesForGpuMemoryBufferFormat(
+            configuration.format);
+
+    // Map buffer into user space.
+    scoped_ptr<void* []> mapped_buffers(new void* [num_planes]);
+    bool rv = buffer->Map(mapped_buffers.get());
+    ASSERT_TRUE(rv);
+    EXPECT_TRUE(buffer->IsMapped());
+
+    // Get strides.
+    scoped_ptr<int[]> strides(new int[num_planes]);
+    buffer->GetStride(strides.get());
+
+    scoped_ptr<size_t[]> row_sizes_in_bytes(new size_t[num_planes]);
+    for (size_t plane = 0; plane < num_planes; ++plane) {
+      EXPECT_TRUE(GpuMemoryBufferImpl::RowSizeInBytes(
+          buffer_size.width(), configuration.format, plane,
+          &row_sizes_in_bytes[plane]));
+      EXPECT_GT(row_sizes_in_bytes[plane], 0u);
+    }

[reveman, 2015/05/20 14:26:08]

nit: please remove this loop and move the code into the loops below instead.
that's more consistent with the use of "data" in the loops below and the Map
test above.

[danakj, 2015/05/20 17:51:14]

Done.

+
+    // Copy and compare mapped buffers.
+    for (size_t plane = 0; plane < num_planes; ++plane) {
+      size_t row_size_in_bytes = row_sizes_in_bytes[plane];
+      scoped_ptr<char[]> data(new char[row_size_in_bytes]);
+      memset(data.get(), 0x2a + plane, row_size_in_bytes);
+
+      size_t height =
+          buffer_size.height() /
+          GpuMemoryBufferImpl::SubsamplingFactor(configuration.format, plane);
+      for (size_t y = 0; y < height; ++y) {
+        memcpy(static_cast<char*>(mapped_buffers[plane]) + y * strides[plane],
+               data.get(), row_size_in_bytes);
+        EXPECT_EQ(memcmp(static_cast<char*>(mapped_buffers[plane]) +
+                             y * strides[plane],
+                         data.get(), row_size_in_bytes),
+                  0);
+      }
+    }
+
+    buffer->Unmap();
+    EXPECT_FALSE(buffer->IsMapped());
+
+    // Remap the buffer, and compare again. It should contain the same data.
+    rv = buffer->Map(mapped_buffers.get());
+    ASSERT_TRUE(rv);
+    EXPECT_TRUE(buffer->IsMapped());
+
+    buffer->GetStride(strides.get());
+
+    for (size_t plane = 0; plane < num_planes; ++plane) {
+      size_t row_size_in_bytes = row_sizes_in_bytes[plane];
+      scoped_ptr<char[]> data(new char[row_size_in_bytes]);
+      memset(data.get(), 0x2a + plane, row_size_in_bytes);
+
+      size_t height =
+          buffer_size.height() /
+          GpuMemoryBufferImpl::SubsamplingFactor(configuration.format, plane);
+      for (size_t y = 0; y < height; ++y) {
+        EXPECT_EQ(memcmp(static_cast<char*>(mapped_buffers[plane]) +
+                             y * strides[plane],
+                         data.get(), row_size_in_bytes),
+                  0);
+      }
+    }
+  }
+}
+
 std::vector<gfx::GpuMemoryBufferType> GetSupportedGpuMemoryBufferTypes() {
   std::vector<gfx::GpuMemoryBufferType> supported_types;
   GpuMemoryBufferFactory::GetSupportedTypes(&supported_types);
   return supported_types;
 }
 
 INSTANTIATE_TEST_CASE_P(
     GpuMemoryBufferImplTests,
     GpuMemoryBufferImplTest,
     ::testing::ValuesIn(GetSupportedGpuMemoryBufferTypes()));
 
 }  // namespace
 }  // namespace content