VideoCapture: add support for GpuMemoryBuffer allocation and lifetime mgmt in VideoCaptureBufferPool

content/browser/renderer_host/media/video_capture_controller_unittest.cc

 // Copyright (c) 2012 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.
 
 // Unit test for VideoCaptureController.
 
 #include <string>
 
 #include "base/bind.h"
 #include "base/bind_helpers.h"
@@ skipping 114 matching lines @@
         base::MessageLoopProxy::current(),
         controller_->GetVideoCaptureFormat());
     client_a_.reset(new MockVideoCaptureControllerEventHandler(
         controller_.get()));
     client_b_.reset(new MockVideoCaptureControllerEventHandler(
         controller_.get()));
   }
 
   void TearDown() override { base::RunLoop().RunUntilIdle(); }
 
-  scoped_refptr<media::VideoFrame> WrapI420Buffer(
-      const scoped_refptr<media::VideoCaptureDevice::Client::Buffer>& buffer,
-      gfx::Size dimensions) {
+  scoped_refptr<media::VideoFrame> WrapI420Buffer(gfx::Size dimensions,
+                                                  uint8* data) {
     return media::VideoFrame::WrapExternalPackedMemory(
         media::VideoFrame::I420,
         dimensions,
         gfx::Rect(dimensions),
         dimensions,
-        reinterpret_cast<uint8*>(buffer->data()),
+        data,
         media::VideoFrame::AllocationSize(media::VideoFrame::I420, dimensions),
         base::SharedMemory::NULLHandle(),
         0,
         base::TimeDelta(),
         base::Closure());
   }
 
   scoped_refptr<media::VideoFrame> WrapMailboxBuffer(
       scoped_ptr<gpu::MailboxHolder> holder,
       const media::VideoFrame::ReleaseMailboxCB& release_cb,
@@ skipping 168 matching lines @@
                          session_200);
   ASSERT_EQ(3, controller_->GetClientCount());
 
   // Now, simulate an incoming captured buffer from the capture device. As a
   // side effect this will cause the first buffer to be shared with clients.
   uint8 buffer_no = 1;
   scoped_refptr<media::VideoCaptureDevice::Client::Buffer> buffer;
   buffer = device_->ReserveOutputBuffer(media::PIXEL_FORMAT_I420,
                                         capture_resolution);
   ASSERT_TRUE(buffer.get());
-  memset(buffer->data(), buffer_no++, buffer->size());
+  scoped_ptr<media::DataHandle> scoped_data_handle(buffer->GetDataHandle());
+  memset(scoped_data_handle->data(), buffer_no++, buffer->size());
   {
     InSequence s;
     EXPECT_CALL(*client_a_, DoBufferCreated(client_a_route_1)).Times(1);
     EXPECT_CALL(*client_a_, DoBufferReady(client_a_route_1,_)).Times(1);
   }
   {
     InSequence s;
     EXPECT_CALL(*client_b_, DoBufferCreated(client_b_route_1)).Times(1);
     EXPECT_CALL(*client_b_, DoBufferReady(client_b_route_1,_)).Times(1);
   }
   {
     InSequence s;
     EXPECT_CALL(*client_a_, DoBufferCreated(client_a_route_2)).Times(1);
     EXPECT_CALL(*client_a_, DoBufferReady(client_a_route_2,_)).Times(1);
   }
   device_->OnIncomingCapturedVideoFrame(
-      buffer,
-      WrapI420Buffer(buffer, capture_resolution),
+      buffer, WrapI420Buffer(capture_resolution,
+                             static_cast<uint8*>(scoped_data_handle->data())),
       base::TimeTicks());
   buffer = NULL;
 
   base::RunLoop().RunUntilIdle();
   Mock::VerifyAndClearExpectations(client_a_.get());
   Mock::VerifyAndClearExpectations(client_b_.get());
 
   // Second buffer which ought to use the same shared memory buffer. In this
   // case pretend that the Buffer pointer is held by the device for a long
   // delay. This shouldn't affect anything.
   buffer = device_->ReserveOutputBuffer(media::PIXEL_FORMAT_I420,
                                         capture_resolution);
+  scoped_data_handle = buffer->GetDataHandle().Pass();
   ASSERT_TRUE(buffer.get());
-  memset(buffer->data(), buffer_no++, buffer->size());
+  memset(scoped_data_handle->data(), buffer_no++, buffer->size());
   device_->OnIncomingCapturedVideoFrame(
-      buffer,
-      WrapI420Buffer(buffer, capture_resolution),
+      buffer, WrapI420Buffer(capture_resolution,
+                             static_cast<uint8*>(scoped_data_handle->data())),
       base::TimeTicks());
   buffer = NULL;
 
   // The buffer should be delivered to the clients in any order.
   EXPECT_CALL(*client_a_, DoBufferReady(client_a_route_1,_)).Times(1);
   EXPECT_CALL(*client_b_, DoBufferReady(client_b_route_1,_)).Times(1);
   EXPECT_CALL(*client_a_, DoBufferReady(client_a_route_2,_)).Times(1);
   base::RunLoop().RunUntilIdle();
   Mock::VerifyAndClearExpectations(client_a_.get());
   Mock::VerifyAndClearExpectations(client_b_.get());
 
   // Add a fourth client now that some buffers have come through.
   controller_->AddClient(client_b_route_2,
                          client_b_.get(),
                          base::kNullProcessHandle,
                          1,
                          session_1);
   Mock::VerifyAndClearExpectations(client_b_.get());
 
   // Third, fourth, and fifth buffers. Pretend they all arrive at the same time.
   for (int i = 0; i < kPoolSize; i++) {
     buffer = device_->ReserveOutputBuffer(media::PIXEL_FORMAT_I420,
                                           capture_resolution);
     ASSERT_TRUE(buffer.get());
-    memset(buffer->data(), buffer_no++, buffer->size());
+    scoped_data_handle = buffer->GetDataHandle().Pass();
+    memset(scoped_data_handle->data(), buffer_no++, buffer->size());
     device_->OnIncomingCapturedVideoFrame(
-        buffer,
-        WrapI420Buffer(buffer, capture_resolution),
+        buffer, WrapI420Buffer(capture_resolution,
+                               static_cast<uint8*>(scoped_data_handle->data())),
         base::TimeTicks());
     buffer = NULL;
   }
   // ReserveOutputBuffer ought to fail now, because the pool is depleted.
   ASSERT_FALSE(device_->ReserveOutputBuffer(media::PIXEL_FORMAT_I420,
                                             capture_resolution).get());
 
   // The new client needs to be told of 3 buffers; the old clients only 2.
   EXPECT_CALL(*client_b_, DoBufferCreated(client_b_route_2)).Times(kPoolSize);
   EXPECT_CALL(*client_b_, DoBufferReady(client_b_route_2,_)).Times(kPoolSize);
@@ skipping 13 matching lines @@
   // Now test the interaction of client shutdown and buffer delivery.
   // Kill A1 via renderer disconnect (synchronous).
   controller_->RemoveClient(client_a_route_1, client_a_.get());
   // Kill B1 via session close (posts a task to disconnect).
   EXPECT_CALL(*client_b_, DoEnded(client_b_route_1)).Times(1);
   controller_->StopSession(300);
   // Queue up another buffer.
   buffer = device_->ReserveOutputBuffer(media::PIXEL_FORMAT_I420,
                                         capture_resolution);
   ASSERT_TRUE(buffer.get());
-  memset(buffer->data(), buffer_no++, buffer->size());
+  scoped_data_handle = buffer->GetDataHandle().Pass();
+  memset(scoped_data_handle->data(), buffer_no++, buffer->size());
   device_->OnIncomingCapturedVideoFrame(
-      buffer,
-      WrapI420Buffer(buffer, capture_resolution),
+      buffer, WrapI420Buffer(capture_resolution,
+                             static_cast<uint8*>(scoped_data_handle->data())),
       base::TimeTicks());
   buffer = NULL;
   buffer = device_->ReserveOutputBuffer(media::PIXEL_FORMAT_I420,
                                         capture_resolution);
   {
     // Kill A2 via session close (posts a task to disconnect, but A2 must not
     // be sent either of these two buffers).
     EXPECT_CALL(*client_a_, DoEnded(client_a_route_2)).Times(1);
     controller_->StopSession(200);
   }
   ASSERT_TRUE(buffer.get());
-  memset(buffer->data(), buffer_no++, buffer->size());
+  memset(scoped_data_handle->data(), buffer_no++, buffer->size());
   device_->OnIncomingCapturedVideoFrame(
-      buffer,
-      WrapI420Buffer(buffer, capture_resolution),
+      buffer, WrapI420Buffer(capture_resolution,
+                             static_cast<uint8*>(scoped_data_handle->data())),
       base::TimeTicks());
   buffer = NULL;
   // B2 is the only client left, and is the only one that should
   // get the buffer.
   EXPECT_CALL(*client_b_, DoBufferReady(client_b_route_2,_)).Times(2);
   base::RunLoop().RunUntilIdle();
   Mock::VerifyAndClearExpectations(client_a_.get());
   Mock::VerifyAndClearExpectations(client_b_.get());
 
   // Allocate all buffers from the buffer pool, half as SHM buffer and half as
   // mailbox buffers.  Make sure of different counts though.
 #if defined(OS_ANDROID)
   int mailbox_buffers = 0;
 #else
   int mailbox_buffers = kPoolSize / 2;
   GLHelper* gl_helper = ImageTransportFactory::GetInstance()->GetGLHelper();
 #endif
   int shm_buffers = kPoolSize - mailbox_buffers;
   if (shm_buffers == mailbox_buffers) {
     shm_buffers--;
     mailbox_buffers++;
   }
 
   for (int i = 0; i < shm_buffers; ++i) {
     buffer = device_->ReserveOutputBuffer(media::PIXEL_FORMAT_I420,
                                           capture_resolution);
     ASSERT_TRUE(buffer.get());
+    scoped_data_handle = buffer->GetDataHandle().Pass();
     device_->OnIncomingCapturedVideoFrame(
-        buffer,
-        WrapI420Buffer(buffer, capture_resolution),
+        buffer, WrapI420Buffer(capture_resolution,
+                               static_cast<uint8*>(scoped_data_handle->data())),
         base::TimeTicks());
     buffer = NULL;
   }
   std::vector<uint32> mailbox_syncpoints(mailbox_buffers);
   std::vector<uint32> release_syncpoints(mailbox_buffers);
   for (int i = 0; i < mailbox_buffers; ++i) {
     buffer = device_->ReserveOutputBuffer(media::PIXEL_FORMAT_TEXTURE,
                                           capture_resolution);
     ASSERT_TRUE(buffer.get());
+    scoped_data_handle = buffer->GetDataHandle().Pass();
 #if !defined(OS_ANDROID)
     mailbox_syncpoints[i] = gl_helper->InsertSyncPoint();
 #endif
     device_->OnIncomingCapturedVideoFrame(
         buffer,
         WrapMailboxBuffer(make_scoped_ptr(new gpu::MailboxHolder(
                               gpu::Mailbox(), 0, mailbox_syncpoints[i])),
                           base::Bind(&CacheSyncPoint, &release_syncpoints[i]),
                           capture_resolution),
         base::TimeTicks());
@@ skipping 49 matching lines @@
   EXPECT_CALL(*client_b_, DoError(route_id)).Times(1);
   controller_->AddClient(
       route_id, client_b_.get(), base::kNullProcessHandle, 200, session_200);
   base::RunLoop().RunUntilIdle();
   Mock::VerifyAndClearExpectations(client_b_.get());
 
   scoped_refptr<media::VideoCaptureDevice::Client::Buffer> buffer =
       device_->ReserveOutputBuffer(media::PIXEL_FORMAT_I420,
                                    capture_resolution);
   ASSERT_TRUE(buffer.get());
+  scoped_ptr<media::DataHandle> scoped_data_handle(buffer->GetDataHandle());
 
   device_->OnIncomingCapturedVideoFrame(
-      buffer,
-      WrapI420Buffer(buffer, capture_resolution),
+      buffer, WrapI420Buffer(capture_resolution,
+                             static_cast<uint8*>(scoped_data_handle->data())),
       base::TimeTicks());
   buffer = NULL;
 
   base::RunLoop().RunUntilIdle();
 }
 
 // Exercises the OnError() codepath of VideoCaptureController, and tests the
 // behavior of various operations after the error state has been signalled.
 TEST_F(VideoCaptureControllerTest, ErrorAfterDeviceCreation) {
   media::VideoCaptureParams session_100;
@@ skipping 13 matching lines @@
   // Start the device. Then, before the first buffer, signal an error and
   // deliver the buffer. The error should be propagated to clients; the buffer
   // should not be.
   base::RunLoop().RunUntilIdle();
   Mock::VerifyAndClearExpectations(client_a_.get());
 
   const gfx::Size dims(320, 240);
   scoped_refptr<media::VideoCaptureDevice::Client::Buffer> buffer =
       device_->ReserveOutputBuffer(media::PIXEL_FORMAT_I420, dims);
   ASSERT_TRUE(buffer.get());
+  scoped_ptr<media::DataHandle> scoped_data_handle(buffer->GetDataHandle());
 
   device_->OnError("Test error");
   device_->OnIncomingCapturedVideoFrame(
       buffer,
-      WrapI420Buffer(buffer, dims),
+      WrapI420Buffer(dims, static_cast<uint8*>(scoped_data_handle->data())),
       base::TimeTicks());
   buffer = NULL;
 
   EXPECT_CALL(*client_a_, DoError(route_id)).Times(1);
   base::RunLoop().RunUntilIdle();
   Mock::VerifyAndClearExpectations(client_a_.get());
 
   // Second client connects after the error state. It also should get told of
   // the error.
   EXPECT_CALL(*client_b_, DoError(route_id)).Times(1);
@@ skipping 96 matching lines @@
     EXPECT_EQ(coded_size.width(), size_and_rotation.output_resolution.width());
     EXPECT_EQ(coded_size.height(),
               size_and_rotation.output_resolution.height());
 
     EXPECT_EQ(kSessionId, controller_->RemoveClient(route_id, client_a_.get()));
     Mock::VerifyAndClearExpectations(client_a_.get());
   }
 }
 
 }  // namespace content