Hook up CSSPaintValue::image to CSS Paint API callback.

third_party/WebKit/Source/modules/csspaint/PaintWorkletGlobalScope.cpp

 // 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 "modules/csspaint/PaintWorkletGlobalScope.h"
 
 #include "bindings/core/v8/ScopedPersistent.h"
 #include "bindings/core/v8/V8BindingMacros.h"
 #include "bindings/core/v8/WorkerOrWorkletScriptController.h"
 #include "core/dom/ExceptionCode.h"
 #include "core/inspector/MainThreadDebugger.h"
 #include "modules/csspaint/CSSPaintDefinition.h"
+#include "modules/csspaint/CSSPaintImageGeneratorImpl.h"
+#include "platform/graphics/ImageBuffer.h"
+#include "platform/graphics/ImageBufferSurface.h"
+#include "platform/graphics/RecordingImageBufferSurface.h"
 
 namespace blink {
 
 // static
 RawPtr<PaintWorkletGlobalScope> PaintWorkletGlobalScope::create(LocalFrame* frame, const KURL& url, const String& userAgent, PassRefPtr<SecurityOrigin> securityOrigin, v8::Isolate* isolate)
 {
     RawPtr<PaintWorkletGlobalScope> paintWorkletGlobalScope = new PaintWorkletGlobalScope(frame, url, userAgent, securityOrigin, isolate);
     paintWorkletGlobalScope->scriptController()->initializeContextIfNeeded();
     MainThreadDebugger::instance()->contextCreated(paintWorkletGlobalScope->scriptController()->getScriptState(), paintWorkletGlobalScope->frame(), paintWorkletGlobalScope->getSecurityOrigin());
     return paintWorkletGlobalScope.release();
@@ skipping 73 matching lines @@
         return;
     }
 
     if (!paintValue->IsFunction()) {
         exceptionState.throwTypeError("The 'paint' property on the prototype is not a function.");
         return;
     }
 
     v8::Local<v8::Function> paint = v8::Local<v8::Function>::Cast(paintValue);
 
-    RawPtr<CSSPaintDefinition> definition = CSSPaintDefinition::create(scriptController()->getScriptState(), constructor, paint);
+    CSSPaintDefinition* definition = CSSPaintDefinition::create(scriptController()->getScriptState(), constructor, paint);
     m_paintDefinitions.set(name, definition);
+
+    // Set the definition on any pending generators.
+    Member<GeneratorHashSet> set = m_pendingGenerators.get(name);
+    if (set) {
+        for (const WeakMember<CSSPaintImageGeneratorImpl> generator : *set) {
+            if (generator) {
+                generator->setDefinition(definition);
+            }
+        }
+    }
+    m_pendingGenerators.remove(name);
 }
 
 CSSPaintDefinition* PaintWorkletGlobalScope::findDefinition(const String& name)
 {
     return m_paintDefinitions.get(name);
 }
 
+void PaintWorkletGlobalScope::addPendingGenerator(const String& name, CSSPaintImageGeneratorImpl* generator)
+{
+    Member<GeneratorHashSet>& set = m_pendingGenerators.add(name, nullptr).storedValue->value;
+    if (!set)
+        set = new GeneratorHashSet;
+    set->add(generator);
+}
+
 DEFINE_TRACE(PaintWorkletGlobalScope)
 {
     visitor->trace(m_paintDefinitions);
+    visitor->trace(m_pendingGenerators);
     WorkletGlobalScope::trace(visitor);
 }
 
 } // namespace blink