Enable GLX/EGL backend switching while run HW video decode with libva.

content/common/gpu/media/vaapi_video_decode_accelerator.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.
 
 #include "base/bind.h"
 #include "base/debug/trace_event.h"
 #include "base/logging.h"
 #include "base/metrics/histogram.h"
 #include "base/stl_util.h"
 #include "base/strings/string_util.h"
@@ skipping 56 matching lines @@
 // at the end of decode (or when a new set of PictureBuffers is required).
 //
 // TFPPictures are used for output, contents of VASurfaces passed from decoder
 // are put into the associated pixmap memory and sent to client.
 class VaapiVideoDecodeAccelerator::TFPPicture {
  public:
   ~TFPPicture();
 
   static linked_ptr<TFPPicture> Create(
       const base::Callback<bool(void)>& make_context_current,
+#if defined(VIDEO_TEXTURE_GLX_BACKEND)
       const GLXFBConfig& fb_config,
+#elif defined(VIDEO_TEXTURE_EGL_BACKEND)
+      EGLDisplay egl_display,
+#endif
       Display* x_display,
       int32 picture_buffer_id,
       uint32 texture_id,
       gfx::Size size);
 
   int32 picture_buffer_id() {
     return picture_buffer_id_;
   }
 
   uint32 texture_id() {
@@ skipping 10 matching lines @@
 
   // Bind texture to pixmap. Needs to be called every frame.
   bool Bind();
 
  private:
   TFPPicture(const base::Callback<bool(void)>& make_context_current,
              Display* x_display,
              int32 picture_buffer_id,
              uint32 texture_id,
              gfx::Size size);
-
+#if defined(VIDEO_TEXTURE_GLX_BACKEND)
   bool Initialize(const GLXFBConfig& fb_config);
+#elif defined(VIDEO_TEXTURE_EGL_BACKEND)
+  bool Initialize(EGLDisplay egl_display);
+#endif
 
   base::Callback<bool(void)> make_context_current_;
 
   Display* x_display_;
 
   // Output id for the client.
   int32 picture_buffer_id_;
   uint32 texture_id_;
 
   gfx::Size size_;
 
   // Pixmaps bound to this texture.
   Pixmap x_pixmap_;
+#if defined(VIDEO_TEXTURE_GLX_BACKEND)
   GLXPixmap glx_pixmap_;
-
+#elif defined(VIDEO_TEXTURE_EGL_BACKEND)
+  EGLDisplay egl_display_;
+  EGLImageKHR egl_image_;
+#endif
   DISALLOW_COPY_AND_ASSIGN(TFPPicture);
 };
 
 VaapiVideoDecodeAccelerator::TFPPicture::TFPPicture(
     const base::Callback<bool(void)>& make_context_current,
     Display* x_display,
     int32 picture_buffer_id,
     uint32 texture_id,
     gfx::Size size)
     : make_context_current_(make_context_current),
       x_display_(x_display),
       picture_buffer_id_(picture_buffer_id),
       texture_id_(texture_id),
       size_(size),
       x_pixmap_(0),
-      glx_pixmap_(0) {
+#if defined(VIDEO_TEXTURE_GLX_BACKEND)
+      glx_pixmap_(0)
+#elif defined(VIDEO_TEXTURE_EGL_BACKEND)
+      egl_display_(0),
+      egl_image_(0)
+#endif
+    {
   DCHECK(!make_context_current_.is_null());
 };
 
 linked_ptr<VaapiVideoDecodeAccelerator::TFPPicture>
 VaapiVideoDecodeAccelerator::TFPPicture::Create(
     const base::Callback<bool(void)>& make_context_current,
+#if defined(VIDEO_TEXTURE_GLX_BACKEND)
     const GLXFBConfig& fb_config,
+#elif defined(VIDEO_TEXTURE_EGL_BACKEND)
+    EGLDisplay egl_display,
+#endif
     Display* x_display,
     int32 picture_buffer_id,
     uint32 texture_id,
     gfx::Size size) {
 
   linked_ptr<TFPPicture> tfp_picture(
       new TFPPicture(make_context_current, x_display, picture_buffer_id,
                      texture_id, size));
 
+#if defined(VIDEO_TEXTURE_GLX_BACKEND)
   if (!tfp_picture->Initialize(fb_config))
+#elif defined(VIDEO_TEXTURE_EGL_BACKEND)
+  if (!tfp_picture->Initialize(egl_display))
+#endif
     tfp_picture.reset();
 
   return tfp_picture;
 }
 
 bool VaapiVideoDecodeAccelerator::TFPPicture::Initialize(
+#if defined(VIDEO_TEXTURE_GLX_BACKEND)
     const GLXFBConfig& fb_config) {
+#elif defined(VIDEO_TEXTURE_EGL_BACKEND)
+    EGLDisplay egl_display) {
+#endif
   // Check for NULL prevents unittests from crashing on nonexistent ChildThread.
   DCHECK(ChildThread::current() == NULL ||
       ChildThread::current()->message_loop() == base::MessageLoop::current());
 
   if (!make_context_current_.Run())
     return false;
 
   XWindowAttributes win_attr;
   int screen = DefaultScreen(x_display_);
   XGetWindowAttributes(x_display_, RootWindow(x_display_, screen), &win_attr);
   //TODO(posciak): pass the depth required by libva, not the RootWindow's depth
   x_pixmap_ = XCreatePixmap(x_display_, RootWindow(x_display_, screen),
                             size_.width(), size_.height(), win_attr.depth);
   if (!x_pixmap_) {
     DVLOG(1) << "Failed creating an X Pixmap for TFP";
     return false;
   }
-
+#if defined(VIDEO_TEXTURE_GLX_BACKEND)
   static const int pixmap_attr[] = {
     GLX_TEXTURE_TARGET_EXT, GLX_TEXTURE_2D_EXT,
     GLX_TEXTURE_FORMAT_EXT, GLX_TEXTURE_FORMAT_RGB_EXT,
     GL_NONE,
   };
 
   glx_pixmap_ = glXCreatePixmap(x_display_, fb_config, x_pixmap_, pixmap_attr);
   if (!glx_pixmap_) {
     // x_pixmap_ will be freed in the destructor.
     DVLOG(1) << "Failed creating a GLX Pixmap for TFP";
     return false;
   }
+#elif defined(VIDEO_TEXTURE_EGL_BACKEND)
+  egl_display_ = egl_display;
+  EGLint image_attrs[] = { EGL_IMAGE_PRESERVED_KHR, 1 , EGL_NONE };
+
+  egl_image_ = eglCreateImageKHR(egl_display_,
+                                EGL_NO_CONTEXT,
+                                EGL_NATIVE_PIXMAP_KHR,
+                                (EGLClientBuffer)x_pixmap_,
+                                image_attrs);
+  if (!egl_image_) {
+    DVLOG(1) << "Failed creating a EGLImage from Pixmap for KHR";
+    return false;
+  }
+#endif
 
   return true;
 }
 
 VaapiVideoDecodeAccelerator::TFPPicture::~TFPPicture() {
   // Check for NULL prevents unittests from crashing on nonexistent ChildThread.
   DCHECK(ChildThread::current() == NULL ||
       ChildThread::current()->message_loop() == base::MessageLoop::current());
 
   // Unbind surface from texture and deallocate resources.
+#if defined(VIDEO_TEXTURE_GLX_BACKEND)
   if (glx_pixmap_ && make_context_current_.Run()) {
     glXReleaseTexImageEXT(x_display_, glx_pixmap_, GLX_FRONT_LEFT_EXT);
     glXDestroyPixmap(x_display_, glx_pixmap_);
   }
+#elif defined(VIDEO_TEXTURE_EGL_BACKEND)
+  if (egl_image_ && make_context_current_.Run()) {
+      eglDestroyImageKHR(egl_display_, egl_image_);
+  }
+#endif
 
   if (x_pixmap_)
     XFreePixmap(x_display_, x_pixmap_);
   XSync(x_display_, False);  // Needed to work around buggy vdpau-driver.
 }
 
 bool VaapiVideoDecodeAccelerator::TFPPicture::Bind() {
   DCHECK(x_pixmap_);
+#if defined(VIDEO_TEXTURE_GLX_BACKEND)
   DCHECK(glx_pixmap_);
+#elif defined(VIDEO_TEXTURE_EGL_BACKEND)
+  DCHECK(egl_image_);
+#endif
   // Check for NULL prevents unittests from crashing on nonexistent ChildThread.
   DCHECK(ChildThread::current() == NULL ||
       ChildThread::current()->message_loop() == base::MessageLoop::current());
 
   if (!make_context_current_.Run())
     return false;
 
   gfx::ScopedTextureBinder texture_binder(GL_TEXTURE_2D, texture_id_);
+#if defined(VIDEO_TEXTURE_GLX_BACKEND)
   glXBindTexImageEXT(x_display_, glx_pixmap_, GLX_FRONT_LEFT_EXT, NULL);
-
+#elif defined(VIDEO_TEXTURE_EGL_BACKEND)
+  glEGLImageTargetTexture2DOES(GL_TEXTURE_2D, egl_image_);
+#endif
   return true;
 }
 
 VaapiVideoDecodeAccelerator::TFPPicture*
     VaapiVideoDecodeAccelerator::TFPPictureById(int32 picture_buffer_id) {
   TFPPictures::iterator it = tfp_pictures_.find(picture_buffer_id);
   if (it == tfp_pictures_.end()) {
     DVLOG(1) << "Picture id " << picture_buffer_id << " does not exist";
     return NULL;
   }
 
   return it->second.get();
 }
 
 VaapiVideoDecodeAccelerator::VaapiVideoDecodeAccelerator(
+#if defined(VIDEO_TEXTURE_GLX_BACKEND)
     Display* x_display, GLXContext glx_context,
+#elif defined(VIDEO_TEXTURE_EGL_BACKEND)
+    EGLDisplay egl_display, EGLContext egl_context,
+#endif
     Client* client,
     const base::Callback<bool(void)>& make_context_current)
+#if defined(VIDEO_TEXTURE_GLX_BACKEND)
     : x_display_(x_display),
       glx_context_(glx_context),
+#elif defined(VIDEO_TEXTURE_EGL_BACKEND)
+    : x_display_(0),
+      egl_display_(egl_display),
+      egl_context_(egl_context),
+#endif
       make_context_current_(make_context_current),
       state_(kUninitialized),
       input_ready_(&lock_),
       surfaces_available_(&lock_),
       message_loop_(base::MessageLoop::current()),
       weak_this_(base::AsWeakPtr(this)),
       client_ptr_factory_(client),
       client_(client_ptr_factory_.GetWeakPtr()),
       decoder_thread_("VaapiDecoderThread"),
       num_frames_at_client_(0),
       num_stream_bufs_at_decoder_(0),
       finish_flush_pending_(false),
       awaiting_va_surfaces_recycle_(false),
       requested_num_pics_(0) {
   DCHECK(client);
 }
 
 VaapiVideoDecodeAccelerator::~VaapiVideoDecodeAccelerator() {
   DCHECK_EQ(message_loop_, base::MessageLoop::current());
 }
 
 class ScopedPtrXFree {
  public:
   void operator()(void* x) const {
     ::XFree(x);
   }
 };
-
+#if defined(VIDEO_TEXTURE_GLX_BACKEND)
 bool VaapiVideoDecodeAccelerator::InitializeFBConfig() {
   const int fbconfig_attr[] = {
     GLX_DRAWABLE_TYPE, GLX_PIXMAP_BIT,
     GLX_BIND_TO_TEXTURE_TARGETS_EXT, GLX_TEXTURE_2D_BIT_EXT,
     GLX_BIND_TO_TEXTURE_RGB_EXT, GL_TRUE,
     GLX_Y_INVERTED_EXT, GL_TRUE,
     GL_NONE,
   };
 
   int num_fbconfigs;
   scoped_ptr_malloc<GLXFBConfig, ScopedPtrXFree> glx_fb_configs(
       glXChooseFBConfig(x_display_, DefaultScreen(x_display_), fbconfig_attr,
                         &num_fbconfigs));
   if (!glx_fb_configs)
     return false;
   if (!num_fbconfigs)
     return false;
 
   fb_config_ = glx_fb_configs.get()[0];
   return true;
 }
-
+#endif
 bool VaapiVideoDecodeAccelerator::Initialize(
     media::VideoCodecProfile profile) {
   DCHECK_EQ(message_loop_, base::MessageLoop::current());
 
   base::AutoLock auto_lock(lock_);
   DCHECK_EQ(state_, kUninitialized);
   DVLOG(2) << "Initializing VAVDA, profile: " << profile;
 
   if (!make_context_current_.Run())
     return false;
 
+#if defined(VIDEO_TEXTURE_GLX_BACKEND)
   if (!InitializeFBConfig()) {
     DVLOG(1) << "Could not get a usable FBConfig";
     return false;
   }
+#elif defined(VIDEO_TEXTURE_EGL_BACKEND)
+  x_display_ = base::MessagePumpForUI::GetDefaultXDisplay();
+#endif
 
   vaapi_wrapper_ = VaapiWrapper::Create(
       profile, x_display_,
       base::Bind(&ReportToUMA, content::VaapiH264Decoder::VAAPI_ERROR));
 
   if (!vaapi_wrapper_.get()) {
     DVLOG(1) << "Failed initializing VAAPI";
     return false;
   }
 
@@ skipping 388 matching lines @@
                                      &va_surface_ids),
       "Failed creating VA Surfaces", PLATFORM_FAILURE, );
   DCHECK_EQ(va_surface_ids.size(), buffers.size());
 
   for (size_t i = 0; i < buffers.size(); ++i) {
     DVLOG(2) << "Assigning picture id: " << buffers[i].id()
              << " to texture id: " << buffers[i].texture_id()
              << " VASurfaceID: " << va_surface_ids[i];
 
     linked_ptr<TFPPicture> tfp_picture(
+#if defined(VIDEO_TEXTURE_GLX_BACKEND)
         TFPPicture::Create(make_context_current_, fb_config_, x_display_,
+#elif defined(VIDEO_TEXTURE_EGL_BACKEND)
+        TFPPicture::Create(make_context_current_, egl_display_, x_display_,
+#endif
                            buffers[i].id(), buffers[i].texture_id(),
                            requested_pic_size_));
 
     RETURN_AND_NOTIFY_ON_FAILURE(
         tfp_picture.get(), "Failed assigning picture buffer to a texture.",
         PLATFORM_FAILURE, );
 
     bool inserted = tfp_pictures_.insert(std::make_pair(
         buffers[i].id(), tfp_picture)).second;
     DCHECK(inserted);
@@ skipping 193 matching lines @@
   state_ = kUninitialized;
 }
 
 void VaapiVideoDecodeAccelerator::Destroy() {
   DCHECK_EQ(message_loop_, base::MessageLoop::current());
   Cleanup();
   delete this;
 }
 
 }  // namespace content