Integrate custom paint with the DocumentLifecycle

sky/engine/core/painting/PaintingContext.h

Index: sky/engine/core/painting/PaintingContext.h
diff --git a/sky/engine/core/painting/PaintingContext.h b/sky/engine/core/painting/PaintingContext.h
index 084724e533f94f5444c4fafceaeee829cdabecc9..2c082522213f507a79707e6c1bb73e0b4388426c 100644
--- a/sky/engine/core/painting/PaintingContext.h
+++ b/sky/engine/core/painting/PaintingContext.h
@@ -5,7 +5,6 @@
 #ifndef SKY_ENGINE_CORE_PAINTING_PAINTINGCONTEXT_H_
 #define SKY_ENGINE_CORE_PAINTING_PAINTINGCONTEXT_H_
 
-#include "base/callback.h"
 #include "sky/engine/core/painting/Paint.h"
 #include "sky/engine/platform/graphics/DisplayList.h"
 #include "sky/engine/tonic/dart_wrappable.h"
@@ -13,17 +12,13 @@
 #include "sky/engine/wtf/RefCounted.h"
 
 namespace blink {
+class Element;
 
 class PaintingContext : public RefCounted<PaintingContext>, public DartWrappable {
     DEFINE_WRAPPERTYPEINFO();
 public:
-    typedef base::Callback<void (RefPtr<PaintingContext>)> CommitCallback;
-
     ~PaintingContext() override;
-    static PassRefPtr<PaintingContext> create(const FloatSize& size, const CommitCallback& commitCallback)
-    {
-        return adoptRef(new PaintingContext(size, commitCallback));
-    }
+    static PassRefPtr<PaintingContext> create(PassRefPtr<Element> element, const FloatSize& size);
 
     double height() const { return m_size.height(); }
     double width() const { return m_size.width(); }
@@ -31,17 +26,11 @@ public:
     void drawCircle(double x, double y, double radius, Paint* paint);
     void commit();
 
-    PassRefPtr<DisplayList> takeDisplayList()
-    {
-        ASSERT(!m_canvas);
-        return m_displayList.release();
-    }
-
 private:
-    PaintingContext(const FloatSize& size, const CommitCallback& commitCallback);
+    PaintingContext(PassRefPtr<Element> element, const FloatSize& size);
 
+    RefPtr<Element> m_element;
     FloatSize m_size;
-    CommitCallback m_commitCallback;
     RefPtr<DisplayList> m_displayList;
     SkCanvas* m_canvas;
 };