cc: Route offscreen context creation for compositor to the browser.

content/browser/renderer_host/image_transport_factory.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 "content/browser/renderer_host/image_transport_factory.h"
 
 #include <algorithm>
 #include <map>
 
 #include "base/bind.h"
@@ skipping 354 matching lines @@
   if (surface)
     surface->OnUpdateVSyncParameters(timebase, interval);
 }
 
 class GpuProcessTransportFactory
     : public ui::ContextFactory,
       public ImageTransportFactory,
       public WebKit::WebGraphicsContext3D::WebGraphicsContextLostCallback {
  public:
   GpuProcessTransportFactory()
-      : ALLOW_THIS_IN_INITIALIZER_LIST(callback_factory_(this)) {
+      : lost_context_callbacks_(0),
+        ALLOW_THIS_IN_INITIALIZER_LIST(callback_factory_(this)) {
     output_surface_proxy_ = new BrowserCompositorOutputSurfaceProxy();
   }
 
   virtual ~GpuProcessTransportFactory() {
     DCHECK(per_compositor_data_.empty());
   }
 
   virtual WebGraphicsContext3DCommandBufferImpl* CreateOffscreenContext()
       OVERRIDE {
     base::WeakPtr<WebGraphicsContext3DSwapBuffersClient> swap_client;
@@ skipping 30 matching lines @@
   }
 
   virtual ui::ContextFactory* AsContextFactory() OVERRIDE {
     return this;
   }
 
   virtual gfx::GLSurfaceHandle CreateSharedSurfaceHandle() OVERRIDE {
     CreateSharedContextLazy();
     gfx::GLSurfaceHandle handle = gfx::GLSurfaceHandle(
         gfx::kNullPluginWindow, true);
-    handle.parent_gpu_process_id = shared_context_->GetGPUProcessID();
-    handle.parent_client_id = shared_context_->GetChannelID();
+    handle.parent_gpu_process_id =
+        shared_context_main_thread_->GetGPUProcessID();
+    handle.parent_client_id = shared_context_main_thread_->GetChannelID();
 
     return handle;
   }
 
   virtual void DestroySharedSurfaceHandle(
       gfx::GLSurfaceHandle surface) OVERRIDE {
   }
 
   virtual scoped_refptr<ui::Texture> CreateTransportClient(
       float device_scale_factor) {
-    if (!shared_context_.get())
+    if (!shared_context_main_thread_.get())
         return NULL;
     scoped_refptr<ImageTransportClientTexture> image(
-        new ImageTransportClientTexture(shared_context_.get(),
+        new ImageTransportClientTexture(shared_context_main_thread_.get(),
                                         device_scale_factor));
     return image;
   }
 
   virtual scoped_refptr<ui::Texture> CreateOwnedTexture(
       const gfx::Size& size,
       float device_scale_factor,
       unsigned int texture_id) OVERRIDE {
-    if (!shared_context_.get())
+    if (!shared_context_main_thread_.get())
         return NULL;
-    scoped_refptr<OwnedTexture> image(
-        new OwnedTexture(shared_context_.get(), size, device_scale_factor,
-                         texture_id));
+    scoped_refptr<OwnedTexture> image(new OwnedTexture(
+        shared_context_main_thread_.get(),
+        size,
+        device_scale_factor,
+        texture_id));
     return image;
   }
 
   virtual GLHelper* GetGLHelper() {
     if (!gl_helper_.get()) {
       CreateSharedContextLazy();
       WebKit::WebGraphicsContext3D* context_for_thread =
           CreateOffscreenContext();
       if (!context_for_thread)
         return NULL;
-      gl_helper_.reset(new GLHelper(shared_context_.get(),
+      gl_helper_.reset(new GLHelper(shared_context_main_thread_.get(),
                                     context_for_thread));
     }
     return gl_helper_.get();
   }
 
   virtual uint32 InsertSyncPoint() OVERRIDE {
-    if (!shared_context_.get())
+    if (!shared_context_main_thread_.get())
       return 0;
-    return shared_context_->insertSyncPoint();
+    return shared_context_main_thread_->insertSyncPoint();
   }
 
   virtual void AddObserver(ImageTransportFactoryObserver* observer) {
     observer_list_.AddObserver(observer);
   }
 
   virtual void RemoveObserver(ImageTransportFactoryObserver* observer) {
     observer_list_.RemoveObserver(observer);
   }
 
   // WebGraphicsContextLostCallback implementation, called for the shared
   // context.
   virtual void onContextLost() {
+    // Wait until we get the callback for each of our shared contexts.
+    ++lost_context_callbacks_;
+    int expected_callbacks = (shared_context_main_thread_ ? 1 : 0) +
+                             (shared_context_compositor_thread_ ? 1 : 0);
+    DCHECK(expected_callbacks);
+    if (lost_context_callbacks_ < expected_callbacks)
+      return;
+
     MessageLoop::current()->PostTask(
         FROM_HERE,
-        base::Bind(&GpuProcessTransportFactory::OnLostSharedContext,
+        base::Bind(&GpuProcessTransportFactory::OnLostSharedContexts,
                    callback_factory_.GetWeakPtr()));
   }
 
   void OnLostContext(ui::Compositor* compositor) {
     LOG(ERROR) << "Lost UI compositor context.";
     PerCompositorData* data = per_compositor_data_[compositor];
     DCHECK(data);
 
     // Prevent callbacks from other contexts in the same share group from
     // calling us again.
@@ skipping 52 matching lines @@
             factory,
             swap_client));
     if (!context->Initialize(
         attrs,
         false,
         CAUSE_FOR_GPU_LAUNCH_WEBGRAPHICSCONTEXT3DCOMMANDBUFFERIMPL_INITIALIZE))
       return NULL;
     return context.release();
   }
 
+  virtual WebKit::WebGraphicsContext3D* OffscreenContextForMainThread()
+      OVERRIDE {
+    CreateSharedContextLazy();
+    return shared_context_main_thread_.get();
+  }
+
+  virtual WebKit::WebGraphicsContext3D* OffscreenContextForCompositorThread()
+      OVERRIDE {
+    if (shared_context_compositor_thread_)
+      return shared_context_compositor_thread_.get();
+
+    shared_context_compositor_thread_.reset(CreateOffscreenContext());
+    if (!shared_context_compositor_thread_->makeContextCurrent())
+      shared_context_compositor_thread_.reset();
+    if (!shared_context_compositor_thread_)
+      return NULL;
+
+    shared_context_compositor_thread_->setContextLostCallback(this);
+    return shared_context_compositor_thread_.get();
+  }
+
   void CreateSharedContextLazy() {
-    if (shared_context_.get())
+    if (shared_context_main_thread_.get())
       return;
 
-    shared_context_.reset(CreateOffscreenContext());
-    if (!shared_context_.get()) {
+    shared_context_main_thread_.reset(CreateOffscreenContext());
+    if (!shared_context_main_thread_.get()) {
       // If we can't recreate contexts, we won't be able to show the UI. Better
       // crash at this point.
       LOG(FATAL) << "Failed to initialize UI shared context.";
     }
-    if (!shared_context_->makeContextCurrent()) {
+    if (!shared_context_main_thread_->makeContextCurrent()) {
       // If we can't recreate contexts, we won't be able to show the UI. Better
       // crash at this point.
       LOG(FATAL) << "Failed to make UI shared context current.";
     }
-    shared_context_->setContextLostCallback(this);
+    shared_context_main_thread_->setContextLostCallback(this);
   }
 
-  void OnLostSharedContext() {
+  void OnLostSharedContexts() {
     // Keep old resources around while we call the observers, but ensure that
     // new resources are created if needed.
-    scoped_ptr<WebGraphicsContext3DCommandBufferImpl> old_shared_context(
-        shared_context_.release());
+    scoped_ptr<WebGraphicsContext3DCommandBufferImpl>
+        old_shared_context_main_thread(shared_context_main_thread_.release());
+    scoped_ptr<WebGraphicsContext3DCommandBufferImpl>
+        old_shared_context_compositor_thread(
+            shared_context_compositor_thread_.release());
     scoped_ptr<GLHelper> old_helper(gl_helper_.release());
 
+    lost_context_callbacks_ = 0;
+
     FOR_EACH_OBSERVER(ImageTransportFactoryObserver,
         observer_list_,
         OnLostResources());
   }
 
   typedef std::map<ui::Compositor*, PerCompositorData*> PerCompositorDataMap;
   PerCompositorDataMap per_compositor_data_;
-  scoped_ptr<WebGraphicsContext3DCommandBufferImpl> shared_context_;
+  scoped_ptr<WebGraphicsContext3DCommandBufferImpl> shared_context_main_thread_;
+  scoped_ptr<WebGraphicsContext3DCommandBufferImpl>
+      shared_context_compositor_thread_;
+  int lost_context_callbacks_;
   scoped_ptr<GLHelper> gl_helper_;
   ObserverList<ImageTransportFactoryObserver> observer_list_;
   base::WeakPtrFactory<GpuProcessTransportFactory> callback_factory_;
   scoped_refptr<BrowserCompositorOutputSurfaceProxy> output_surface_proxy_;
 
   DISALLOW_COPY_AND_ASSIGN(GpuProcessTransportFactory);
 };
 
 void CompositorSwapClient::OnLostContext() {
   factory_->OnLostContext(compositor_);
@@ skipping 31 matching lines @@
   delete g_factory;
   g_factory = NULL;
 }
 
 // static
 ImageTransportFactory* ImageTransportFactory::GetInstance() {
   return g_factory;
 }
 
 }  // namespace content