Change Canvas2DLayerBridge to stay alive until last mailbox is returned.

Source/core/platform/graphics/chromium/Canvas2DLayerBridge.cpp

 /*
  * Copyright (C) 2012 Google Inc. All rights reserved.
  *
  * Redistribution and use in source and binary forms, with or without
  * modification, are permitted provided that the following conditions
  * are met:
  *
  * 1.  Redistributions of source code must retain the above copyright
  *     notice, this list of conditions and the following disclaimer.
  * 2.  Redistributions in binary form must reproduce the above copyright
@@ skipping 41 matching lines @@
         return 0;
     gr->resetContext();
     SkImage::Info info;
     info.fWidth = size.width();
     info.fHeight = size.height();
     info.fColorType = SkImage::kPMColor_ColorType;
     info.fAlphaType = SkImage::kPremul_AlphaType;
     return SkSurface::NewRenderTarget(gr, info);
 }
 
-PassOwnPtr<Canvas2DLayerBridge> Canvas2DLayerBridge::create(PassRefPtr<GraphicsContext3D> context, const IntSize& size, OpacityMode opacityMode)
+Canvas2DLayerBridgePtr Canvas2DLayerBridge::create(PassRefPtr<GraphicsContext3D> context, const IntSize& size, OpacityMode opacityMode)
 {
     TRACE_EVENT_INSTANT0("test_gpu", "Canvas2DLayerBridgeCreation");
     SkAutoTUnref<SkSurface> surface(createSurface(context.get(), size));
-    if (!surface.get())
-        return PassOwnPtr<Canvas2DLayerBridge>();
+    if (!surface.get()) {
+        Canvas2DLayerBridgePtr ptr;
+        return ptr;
+    }
     SkDeferredCanvas* canvas = SkDeferredCanvas::Create(surface.get());
-    OwnPtr<Canvas2DLayerBridge> layerBridge = adoptPtr(new Canvas2DLayerBridge(context, canvas, opacityMode));
-    return layerBridge.release();
+    Canvas2DLayerBridgePtr layerBridge(new Canvas2DLayerBridge(context, canvas, opacityMode));
+    return layerBridge;
 }
 
 Canvas2DLayerBridge::Canvas2DLayerBridge(PassRefPtr<GraphicsContext3D> context, SkDeferredCanvas* canvas, OpacityMode opacityMode)
     : m_canvas(canvas)
     , m_context(context)
     , m_bytesAllocated(0)
     , m_didRecordDrawCommand(false)
     , m_surfaceIsValid(true)
     , m_framesPending(0)
     , m_rateLimitingEnabled(false)
+    , m_destructionInProgress(false)
     , m_next(0)
     , m_prev(0)
     , m_lastImageId(0)
 {
     ASSERT(m_canvas);
     // Used by browser tests to detect the use of a Canvas2DLayerBridge.
     TRACE_EVENT_INSTANT0("test_gpu", "Canvas2DLayerBridgeCreation");
     m_layer = adoptPtr(WebKit::Platform::current()->compositorSupport()->createExternalTextureLayer(this));
     m_layer->setOpaque(opacityMode == Opaque);
     m_layer->setBlendBackgroundColor(opacityMode != Opaque);
     GraphicsLayer::registerContentsLayer(m_layer->layer());
     m_layer->setRateLimitContext(m_rateLimitingEnabled);
     m_canvas->setNotificationClient(this);
 }
 
+void Canvas2DLayerBridge::destroy()
+{
+    ASSERT(!m_destructionInProgress);
+    GraphicsLayer::unregisterContentsLayer(m_layer->layer());
+    m_canvas->setNotificationClient(0);
+    m_layer->clearTexture();
+    // Orphan the layer so the compositor will stop requesting new texture mailboxes.
+    m_layer->layer()->removeFromParent();

[Justin Novosad, 2013/08/20 15:51:10]

This line appears to be necessary if we want the mailboxes to be released
without having to produce new ones. Open issue: are texture layers always leaf
nodes in the compositor tree? If not, perhaps we need to worry about re
attaching the children of this layer to the parent layer?

[danakj, 2013/08/20 16:57:08]

You could also set the texture mailbox on the layer to be empty, then that would
be passed to the compositor thread and the old one would be released.

[piman, 2013/08/20 17:03:07]

I'll let the blink experts confirm, but the fact that you
GraphicsLayer::unregisterContentsLayer it either way means you're safe, I think.

+    Canvas2DLayerManager::get().layerToBeDestroyed(this);
+    m_destructionInProgress = true;
+    deleteIfPossible();
+}
+
+void Canvas2DLayerBridge::deleteIfPossible()
+{
+    // If we can't delete now, destruction is deferred until mailboxReleased
+    // is called on all remaining active mailboxes.
+    ASSERT(m_destructionInProgress);
+    bool deleteIsPossible = true;
+    Vector<MailboxInfo>::iterator mailboxInfo;
+    for (mailboxInfo = m_mailboxes.begin(); mailboxInfo < m_mailboxes.end(); mailboxInfo++) {
+        if (mailboxInfo->m_status == MailboxReleased) {
+            if (mailboxInfo->m_mailbox.syncPoint) {
+                context()->waitSyncPoint(mailboxInfo->m_mailbox.syncPoint);
+                mailboxInfo->m_mailbox.syncPoint = 0;
+            }
+            // Invalidate texture state in case the compositor altered it since the copy-on-write.
+            mailboxInfo->m_image->getTexture()->invalidateCachedState();
+            mailboxInfo->m_image.reset(0);
+            mailboxInfo->m_status = MailboxAvailable;
+        } else if (mailboxInfo->m_status == MailboxInUse) {
+            deleteIsPossible = false;
+        }
+    }
+    if (deleteIsPossible)
+        delete this;

[piman, 2013/08/20 17:03:07]

Could it make sense to make Canvas2DLayerBridge refcounted, and have each of the
MailboxInfo keep a ref while it's in the MailboxInUse state?
You could move the waitSyncPoint / m_image reset logic to the destructor and
that way you don't need to add deleteIfPossible logic in the various places.

[Stephen White, 2013/08/20 17:32:18]

+1000 to this. This poor-man's refcounting here looks confusing and difficult to
maintain.

+}
+
 Canvas2DLayerBridge::~Canvas2DLayerBridge()
 {
-    GraphicsLayer::unregisterContentsLayer(m_layer->layer());
-    Canvas2DLayerManager::get().layerToBeDestroyed(this);
-    m_canvas->setNotificationClient(0);
+    ASSERT(m_destructionInProgress);
     m_mailboxes.clear();
-    m_layer->clearTexture();
     m_layer.clear();
 }
 
 void Canvas2DLayerBridge::limitPendingFrames()
 {
+    ASSERT(!m_destructionInProgress);
     if (m_didRecordDrawCommand) {
         m_framesPending++;
         m_didRecordDrawCommand = false;
         if (m_framesPending > 1) {
             // Turn on the rate limiter if this layer tends to accumulate a
             // non-discardable multi-frame backlog of draw commands.
             setRateLimitingEnabled(true);
         }
         if (m_rateLimitingEnabled) {
             flush();
         }
     }
 }
 
 void Canvas2DLayerBridge::prepareForDraw()
 {
+    ASSERT(!m_destructionInProgress);
     ASSERT(m_layer);
     if (!isValid()) {
         if (m_canvas) {
             // drop pending commands because there is no surface to draw to
             m_canvas->silentFlush();
         }
         return;
     }
     m_context->makeContextCurrent();
 }
 
 void Canvas2DLayerBridge::storageAllocatedForRecordingChanged(size_t bytesAllocated)
 {
+    ASSERT(!m_destructionInProgress);
     intptr_t delta = (intptr_t)bytesAllocated - (intptr_t)m_bytesAllocated;
     m_bytesAllocated = bytesAllocated;
     Canvas2DLayerManager::get().layerAllocatedStorageChanged(this, delta);
 }
 
 size_t Canvas2DLayerBridge::storageAllocatedForRecording()
 {
+    ASSERT(!m_destructionInProgress);
     return m_canvas->storageAllocatedForRecording();
 }
 
 void Canvas2DLayerBridge::flushedDrawCommands()
 {
+    ASSERT(!m_destructionInProgress);
     storageAllocatedForRecordingChanged(storageAllocatedForRecording());
     m_framesPending = 0;
 }
 
 void Canvas2DLayerBridge::skippedPendingDrawCommands()
 {
+    ASSERT(!m_destructionInProgress);
     // Stop triggering the rate limiter if SkDeferredCanvas is detecting
     // and optimizing overdraw.
     setRateLimitingEnabled(false);
     flushedDrawCommands();
 }
 
 void Canvas2DLayerBridge::setRateLimitingEnabled(bool enabled)
 {
+    ASSERT(!m_destructionInProgress || !enabled);
     if (m_rateLimitingEnabled != enabled) {
         m_rateLimitingEnabled = enabled;
         m_layer->setRateLimitContext(m_rateLimitingEnabled);
     }
 }
 
 size_t Canvas2DLayerBridge::freeMemoryIfPossible(size_t bytesToFree)
 {
+    ASSERT(!m_destructionInProgress);
     size_t bytesFreed = m_canvas->freeMemoryIfPossible(bytesToFree);
     if (bytesFreed)
         Canvas2DLayerManager::get().layerAllocatedStorageChanged(this, -((intptr_t)bytesFreed));
     m_bytesAllocated -= bytesFreed;
     return bytesFreed;
 }
 
 void Canvas2DLayerBridge::flush()
 {
+    ASSERT(!m_destructionInProgress);
     if (m_canvas->hasPendingCommands()) {
         TRACE_EVENT0("cc", "Canvas2DLayerBridge::flush");
         m_canvas->flush();
     }
 }
 
 WebGraphicsContext3D* Canvas2DLayerBridge::context()
 {
     // Check on m_layer is necessary because context() may be called during
     // the destruction of m_layer
-    if (m_layer) {
+    if (m_layer && !m_destructionInProgress) {
         isValid(); // To ensure rate limiter is disabled if context is lost.
     }
     return m_context->webContext();
 }
 
 bool Canvas2DLayerBridge::isValid()
 {
+    ASSERT(!m_destructionInProgress);
     ASSERT(m_layer);
     if (m_context->webContext()->isContextLost() || !m_surfaceIsValid) {
         // Attempt to recover.
         m_layer->clearTexture();
         m_mailboxes.clear();
         RefPtr<GraphicsContext3D> sharedContext = SharedGraphicsContext3D::get();
         if (!sharedContext || sharedContext->webContext()->isContextLost()) {
             m_surfaceIsValid = false;
         } else {
             m_context = sharedContext;
             IntSize size(m_canvas->getTopDevice()->width(), m_canvas->getTopDevice()->height());
             SkAutoTUnref<SkSurface> surface(createSurface(m_context.get(), size));
             if (surface.get()) {
                 m_canvas->setSurface(surface.get());
                 m_surfaceIsValid = true;
                 // FIXME: draw sad canvas picture into new buffer crbug.com/243842
             } else {
                 // Surface allocation failed. Set m_surfaceIsValid to false to
                 // trigger subsequent retry.
                 m_surfaceIsValid = false;
             }
         }
     }
     if (!m_surfaceIsValid)
         setRateLimitingEnabled(false);
     return m_surfaceIsValid;
-
 }
 
 bool Canvas2DLayerBridge::prepareMailbox(WebKit::WebExternalTextureMailbox* outMailbox, WebKit::WebExternalBitmap* bitmap)
 {
+    ASSERT(!m_destructionInProgress)
     ASSERT(!bitmap);
     if (!isValid())
         return false;
     // Release to skia textures that were previouosly released by the
     // compositor. We do this before acquiring the next snapshot in
     // order to cap maximum gpu memory consumption.
     m_context->makeContextCurrent();
     flush();
     Vector<MailboxInfo>::iterator mailboxInfo;
     for (mailboxInfo = m_mailboxes.begin(); mailboxInfo < m_mailboxes.end(); mailboxInfo++) {
@@ skipping 37 matching lines @@
     m_context->bindTexture(GraphicsContext3D::TEXTURE_2D, 0);
     // Because we are changing the texture binding without going through skia,
     // we must dirty the context.
     m_context->grContext()->resetContext(kTextureBinding_GrGLBackendState);
 
     *outMailbox = mailboxInfo->m_mailbox;
     return true;
 }
 
 Canvas2DLayerBridge::MailboxInfo* Canvas2DLayerBridge::createMailboxInfo() {
+    ASSERT(!m_destructionInProgress);
     MailboxInfo* mailboxInfo;
     for (mailboxInfo = m_mailboxes.begin(); mailboxInfo < m_mailboxes.end(); mailboxInfo++) {
         if (mailboxInfo->m_status == MailboxAvailable) {
             return mailboxInfo;
         }
     }
 
     // No available mailbox: create one.
     m_mailboxes.grow(m_mailboxes.size() + 1);
     mailboxInfo = &m_mailboxes.last();
@@ skipping 13 matching lines @@
 {
     Vector<MailboxInfo>::iterator mailboxInfo;
     for (mailboxInfo = m_mailboxes.begin(); mailboxInfo < m_mailboxes.end(); mailboxInfo++) {
         if (!memcmp(mailboxInfo->m_mailbox.name, mailbox.name, sizeof(mailbox.name))) {
             mailboxInfo->m_mailbox.syncPoint = mailbox.syncPoint;
             ASSERT(mailboxInfo->m_status == MailboxInUse);
             mailboxInfo->m_status = MailboxReleased;
             return;
         }
     }
+    if (m_destructionInProgress)
+        deleteIfPossible();
 }
 
 WebKit::WebLayer* Canvas2DLayerBridge::layer()
 {
+    ASSERT(!m_destructionInProgress);
     ASSERT(m_layer);
     return m_layer->layer();
 }
 
 void Canvas2DLayerBridge::contextAcquired()
 {
+    ASSERT(!m_destructionInProgress);
     Canvas2DLayerManager::get().layerDidDraw(this);
     m_didRecordDrawCommand = true;
 }
 
 unsigned Canvas2DLayerBridge::backBufferTexture()
 {
+    ASSERT(!m_destructionInProgress);
     if (!isValid())
         return 0;
     contextAcquired();
     m_canvas->flush();
     m_context->flush();
     GrRenderTarget* renderTarget = reinterpret_cast<GrRenderTarget*>(m_canvas->getDevice()->accessRenderTarget());
     if (renderTarget) {
         return renderTarget->asTexture()->getTextureHandle();
     }
     return 0;
 }
 
 Canvas2DLayerBridge::MailboxInfo::MailboxInfo(const MailboxInfo& other) {
     // This copy constructor should only be used for Vector reallocation
     // Assuming 'other' is to be destroyed, we swap m_image ownership
     // rather than do a refcount dance.
     memcpy(&m_mailbox, &other.m_mailbox, sizeof(m_mailbox));
     m_image.swap(const_cast<SkAutoTUnref<SkImage>*>(&other.m_image));
     m_status = other.m_status;
 }
 
 }