media: Implement zero-copy video playback for ffmpeg.

media/video/gpu_memory_buffer_video_frame_pool.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 "media/video/gpu_memory_buffer_video_frame_pool.h"
 
 #include <algorithm>
 #include <list>
 
 #include "base/bind.h"
@@ skipping 61 matching lines @@
 
 bool IsPowerOfTwo(size_t x) {
   return x != 0 && (x & (x - 1)) == 0;
 }
 
 inline size_t RoundUp(size_t value, size_t alignment) {
   DCHECK(IsPowerOfTwo(alignment));
   return ((value + (alignment - 1)) & ~(alignment - 1));
 }
 
-gfx::Size CodedSize(const gfx::Size& size, VideoPixelFormat format) {
-  switch (format) {
-    case PIXEL_FORMAT_I420:
-    case PIXEL_FORMAT_YV12:
-    case PIXEL_FORMAT_YV12A:
-    case PIXEL_FORMAT_NV12:
-      return gfx::Size(RoundUp(size.width(), 2), RoundUp(size.height(), 2));
-    case PIXEL_FORMAT_UYVY:
-      return gfx::Size(RoundUp(size.width(), 2), size.height());
-    default:
-      NOTREACHED();
-  }
-  return gfx::Size();
-}
-
 // All the resources needed to compose a frame.
 class FrameResources : public base::RefCountedThreadSafe<FrameResources> {
  public:
   static scoped_refptr<FrameResources> Create(
       GpuVideoAcceleratorFactories* gpu_factories,
       VideoPixelFormat format,
       const gfx::Size& size) {
     // Create the resources.
     scoped_ptr<GpuVideoAcceleratorFactories::ScopedGLContextLock> lock(
         gpu_factories->GetGLContextLock());
@@ skipping 367 matching lines @@
     : public base::RefCountedThreadSafe<
           GpuMemoryBufferVideoFramePool::PoolImpl>,
       public base::trace_event::MemoryDumpProvider {
  public:
   // |media_task_runner| is the media task runner associated with the
   // GL context provided by |gpu_factories|
   // |gpu_factories| is an interface to GPU related operation and can be
   // null if a GL context is not available.
   PoolImpl(const scoped_refptr<base::SingleThreadTaskRunner>& media_task_runner,
            GpuVideoAcceleratorFactories* gpu_factories)
-      : media_task_runner_(media_task_runner), gpu_factories_(gpu_factories) {
+      : media_task_runner_(media_task_runner),
+        gpu_factories_(gpu_factories),
+        usage_(GpuMemoryBufferVideoFramePool::Usage::GENERIC) {
     DCHECK(media_task_runner_);
     base::trace_event::MemoryDumpManager::GetInstance()->RegisterDumpProvider(
         this, "GpuMemoryBufferVideoFramePool", media_task_runner_);
   }
 
+  void SetUsage(GpuMemoryBufferVideoFramePool::Usage usage) { usage_ = usage; }
+
   std::unique_ptr<VideoFrameFuture> CreateFrame(VideoPixelFormat format,
                                                 const gfx::Size& coded_size,
                                                 const gfx::Rect& visible_rect,
                                                 const gfx::Size& natural_size,
                                                 base::TimeDelta timestamp) {
     DCHECK(media_task_runner_->BelongsToCurrentThread());
     if (!IsSupported(format)) {
       NOTREACHED();
       return nullptr;
     }
@@ skipping 86 matching lines @@
       case PIXEL_FORMAT_YV12:
       case PIXEL_FORMAT_YV12A:
       case PIXEL_FORMAT_NV12:
       case PIXEL_FORMAT_UYVY:
         return true;
       default:
         return false;
     }
   }
 
+  gfx::Size CodedSize(const gfx::Size& size, VideoPixelFormat format) {
+    static const int kFrameSizeAlignment = 16;
+    if (usage_ == GpuMemoryBufferVideoFramePool::Usage::FFMPEG) {
+      // libavcodec requires to access areas outside 0,0 - width,height, please
+      // see libavcodec/utils.c:avcodec_align_dimensions2().
+      int width = RoundUp(size.width(), kFrameSizeAlignment);
+      // H.264 uses edge emulation which requires at least 32.
+      width = std::max(width, kFrameSizeAlignment * 2);
+      // The *2 in alignment for height is because some formats (e.g. h264)
+      // allow interlaced coding, and then the size needs to be a multiple of
+      // two macroblocks (vertically).
+      int height = RoundUp(size.height(), kFrameSizeAlignment * 2);
+      // some of the optimized chroma MC needs two more rows.
+      height += 2;
+      return gfx::Size(width, height);
+    }
+
+    switch (format) {
+      case PIXEL_FORMAT_I420:
+      case PIXEL_FORMAT_YV12:
+      case PIXEL_FORMAT_YV12A:
+      case PIXEL_FORMAT_NV12:
+        return gfx::Size(RoundUp(size.width(), 2), RoundUp(size.height(), 2));
+      case PIXEL_FORMAT_UYVY:
+        return gfx::Size(RoundUp(size.width(), 2), size.height());
+      default:
+        NOTREACHED();
+    }
+    return gfx::Size();
+  }
+
   // Task runner associated to the GL context provided by |gpu_factories_|.
   scoped_refptr<base::SingleThreadTaskRunner> media_task_runner_;
 
   // Interface to GPU related operations.
   GpuVideoAcceleratorFactories* gpu_factories_;
 
+  GpuMemoryBufferVideoFramePool::Usage usage_;
+
   // Pool of resources.
   std::list<scoped_refptr<FrameResources>> resources_pool_;
 
   DISALLOW_COPY_AND_ASSIGN(PoolImpl);
 };
 
 GpuMemoryBufferVideoFramePool::GpuMemoryBufferVideoFramePool(
     const scoped_refptr<base::SingleThreadTaskRunner>& media_task_runner,
     GpuVideoAcceleratorFactories* gpu_factories)
     : pool_impl_(new PoolImpl(media_task_runner, gpu_factories)) {}
 
 GpuMemoryBufferVideoFramePool::~GpuMemoryBufferVideoFramePool() {
 }
 
 std::unique_ptr<VideoFrameFuture> GpuMemoryBufferVideoFramePool::CreateFrame(
     VideoPixelFormat format,
     const gfx::Size& coded_size,
     const gfx::Rect& visible_rect,
     const gfx::Size& natural_size,
     base::TimeDelta timestamp) {
   return pool_impl_->CreateFrame(format, coded_size, visible_rect, natural_size,
                                  timestamp);
 }
 
+void GpuMemoryBufferVideoFramePool::SetUsage(Usage usage) {
+  pool_impl_->SetUsage(usage);
+}
+
 // static
 size_t GpuMemoryBufferVideoFramePool::PlanesPerCopy(VideoPixelFormat format,
                                                     size_t plane) {
   return gfx::NumberOfPlanesForBufferFormat(
       GpuMemoryBufferFormat(format, plane));
 }
 
 }  // namespace media