| Index: content/browser/renderer_host/media/gpu_memory_buffer_tracker.cc
|
| diff --git a/content/browser/renderer_host/media/gpu_memory_buffer_tracker.cc b/content/browser/renderer_host/media/gpu_memory_buffer_tracker.cc
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..b4e601c972be0ffe173128e226275e13e47ee575
|
| --- /dev/null
|
| +++ b/content/browser/renderer_host/media/gpu_memory_buffer_tracker.cc
|
| @@ -0,0 +1,105 @@
|
| +// Copyright 2016 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/browser/renderer_host/media/gpu_memory_buffer_tracker.h"
|
| +
|
| +#include "base/memory/ptr_util.h"
|
| +#include "content/browser/gpu/browser_gpu_memory_buffer_manager.h"
|
| +#include "content/browser/renderer_host/media/gpu_memory_buffer_handle.h"
|
| +#include "content/public/browser/browser_thread.h"
|
| +#include "media/base/video_frame.h"
|
| +
|
| +namespace content {
|
| +
|
| +GpuMemoryBufferTracker::GpuMemoryBufferTracker()
|
| + : VideoCaptureBufferTracker() {}
|
| +
|
| +GpuMemoryBufferTracker::~GpuMemoryBufferTracker() {
|
| + for (const auto& gmb : gpu_memory_buffers_)
|
| + gmb->Unmap();
|
| +}
|
| +
|
| +bool GpuMemoryBufferTracker::Init(const gfx::Size& dimensions,
|
| + media::VideoPixelFormat format,
|
| + media::VideoPixelStorage storage_type,
|
| + base::Lock* lock) {
|
| + DVLOG(2) << "allocating GMB for " << dimensions.ToString();
|
| + // BrowserGpuMemoryBufferManager::current() may not be accessed on IO
|
| + // Thread.
|
| + DCHECK(!BrowserThread::CurrentlyOn(BrowserThread::IO));
|
| + DCHECK(BrowserGpuMemoryBufferManager::current());
|
| + // This class is only expected to be called with I420 buffer requests at
|
| + // this point.
|
| + DCHECK_EQ(format, media::PIXEL_FORMAT_I420);
|
| + set_dimensions(dimensions);
|
| + set_max_pixel_count(dimensions.GetArea());
|
| + set_pixel_format(format);
|
| + set_storage_type(storage_type);
|
| + // |dimensions| can be 0x0 for trackers that do not require memory backing.
|
| + if (dimensions.GetArea() == 0u)
|
| + return true;
|
| +
|
| + base::AutoUnlock auto_unlock(*lock);
|
| + const size_t num_planes = media::VideoFrame::NumPlanes(pixel_format());
|
| + for (size_t i = 0; i < num_planes; ++i) {
|
| + const gfx::Size& size =
|
| + media::VideoFrame::PlaneSize(pixel_format(), i, dimensions);
|
| + gpu_memory_buffers_.push_back(
|
| + BrowserGpuMemoryBufferManager::current()->AllocateGpuMemoryBuffer(
|
| + size, gfx::BufferFormat::R_8,
|
| + gfx::BufferUsage::GPU_READ_CPU_READ_WRITE,
|
| + gpu::kNullSurfaceHandle));
|
| +
|
| + DLOG_IF(ERROR, !gpu_memory_buffers_[i]) << "Allocating GpuMemoryBuffer";
|
| + if (!gpu_memory_buffers_[i] || !gpu_memory_buffers_[i]->Map())
|
| + return false;
|
| + }
|
| + return true;
|
| +}
|
| +
|
| +std::unique_ptr<VideoCaptureBufferHandle>
|
| +GpuMemoryBufferTracker::GetBufferHandle() {
|
| + DCHECK_EQ(gpu_memory_buffers_.size(),
|
| + media::VideoFrame::NumPlanes(pixel_format()));
|
| + return base::MakeUnique<GpuMemoryBufferBufferHandle>(this);
|
| +}
|
| +
|
| +bool GpuMemoryBufferTracker::ShareToProcess(
|
| + base::ProcessHandle process_handle,
|
| + base::SharedMemoryHandle* new_handle) {
|
| + NOTREACHED();
|
| + return false;
|
| +}
|
| +
|
| +bool GpuMemoryBufferTracker::ShareToProcess2(
|
| + int plane,
|
| + base::ProcessHandle process_handle,
|
| + gfx::GpuMemoryBufferHandle* new_handle) {
|
| + DCHECK_LE(plane, static_cast<int>(gpu_memory_buffers_.size()));
|
| +
|
| + const auto& current_gmb_handle = gpu_memory_buffers_[plane]->GetHandle();
|
| + switch (current_gmb_handle.type) {
|
| + case gfx::EMPTY_BUFFER:
|
| + NOTREACHED();
|
| + return false;
|
| + case gfx::SHARED_MEMORY_BUFFER: {
|
| + DCHECK(base::SharedMemory::IsHandleValid(current_gmb_handle.handle));
|
| + base::SharedMemory shared_memory(
|
| + base::SharedMemory::DuplicateHandle(current_gmb_handle.handle),
|
| + false);
|
| + shared_memory.ShareToProcess(process_handle, &new_handle->handle);
|
| + DCHECK(base::SharedMemory::IsHandleValid(new_handle->handle));
|
| + new_handle->type = gfx::SHARED_MEMORY_BUFFER;
|
| + return true;
|
| + }
|
| + case gfx::IO_SURFACE_BUFFER:
|
| + case gfx::OZONE_NATIVE_PIXMAP:
|
| + *new_handle = current_gmb_handle;
|
| + return true;
|
| + }
|
| + NOTREACHED();
|
| + return true;
|
| +}
|
| +
|
| +} // namespace content
|
|
|