[WIP] Plumbing for paint

third_party/WebKit/Source/modules/csspaint/CSSPaintImageGeneratorImpl.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/CSSPaintImageGeneratorImpl.h"
+
+#include "core/dom/Document.h"
+#include "core/frame/LocalDOMWindow.h"
+#include "modules/csspaint/CSSPaintDefinition.h"
+#include "modules/csspaint/PaintWorklet.h"
+#include "modules/csspaint/WindowPaintWorklet.h"
+#include "platform/graphics/Image.h"
+
+namespace blink {
+
+PassRefPtrWillBeRawPtr<CSSPaintImageGenerator> CSSPaintImageGeneratorImpl::create(const String& name, Document& document, Observer* observer)
+{
+    // TODO(ikilpatrick): lazy create worklet global scope.
+    LocalDOMWindow* domWindow = document.domWindow();
+    PaintWorklet* paintWorklet = WindowPaintWorklet::from(*domWindow).paintWorklet(&document);
+
+    CSSPaintDefinition* paintDefinition = paintWorklet->findDefinition(name);
+    RefPtrWillBeRawPtr<CSSPaintImageGeneratorImpl> generator;
+    if (!paintDefinition) {
+        generator = adoptRefWillBeNoop(new CSSPaintImageGeneratorImpl(observer));
+        paintWorklet->addPendingGenerator(name, generator);
+    } else {
+        generator = adoptRefWillBeNoop(new CSSPaintImageGeneratorImpl(paintDefinition));
+    }
+
+    return generator.release();
+}
+
+CSSPaintImageGeneratorImpl::CSSPaintImageGeneratorImpl(PassRefPtrWillBeRawPtr<CSSPaintDefinition> definition)
+    : m_definition(definition)
+{
+}
+
+CSSPaintImageGeneratorImpl::CSSPaintImageGeneratorImpl(Observer* observer)
+    : m_observer(observer)
+{
+}
+
+void CSSPaintImageGeneratorImpl::setDefinition(PassRefPtrWillBeRawPtr<CSSPaintDefinition> definition)
+{
+    ASSERT(!m_definition);
+    m_definition = definition;
+
+    ASSERT(m_observer);
+    m_observer->ready();
+}
+
+PassRefPtr<Image> CSSPaintImageGeneratorImpl::paint(const IntSize& size)
+{
+    return m_definition ? m_definition->paint(size) : nullptr;
+}
+
+DEFINE_TRACE(CSSPaintImageGeneratorImpl)
+{
+    visitor->trace(m_definition);
+    CSSPaintImageGenerator::trace(visitor);
+}
+
+} // namespace blink