[DevTools] Host paint profiles in PaintProfilerModel

third_party/WebKit/Source/devtools/front_end/layers/LayerTreeModel.js

 /*
  * Copyright (C) 2013 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:
  *
  *     * Redistributions of source code must retain the above copyright
  * notice, this list of conditions and the following disclaimer.
  *     * Redistributions in binary form must reproduce the above
@@ skipping 16 matching lines @@
  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  */
 
 /**
  * @unrestricted
  */
 Layers.LayerTreeModel = class extends SDK.SDKModel {
   constructor(target) {
     super(target);
+    this._layerTreeAgent = target.layerTreeAgent();
     target.registerLayerTreeDispatcher(new Layers.LayerTreeDispatcher(this));
+    this._paintProfilerModel = /** @type {!SDK.PaintProfilerModel} */ (target.model(SDK.PaintProfilerModel));
     var resourceTreeModel = target.model(SDK.ResourceTreeModel);
     if (resourceTreeModel) {
       resourceTreeModel.addEventListener(
           SDK.ResourceTreeModel.Events.MainFrameNavigated, this._onMainFrameNavigated, this);
     }
     /** @type {?SDK.LayerTreeBase} */
     this._layerTree = null;
   }
 
   disable() {
     if (!this._enabled)
       return;
     this._enabled = false;
-    this.target().layerTreeAgent().disable();
+    this._layerTreeAgent.disable();
   }
 
   enable() {
     if (this._enabled)
       return;
     this._enabled = true;
     this._forceEnable();
   }
 
   _forceEnable() {
     this._lastPaintRectByLayerId = {};
     if (!this._layerTree)
-      this._layerTree = new Layers.AgentLayerTree(this.target());
-    this.target().layerTreeAgent().enable();
+      this._layerTree = new Layers.AgentLayerTree(this);
+    this._layerTreeAgent.enable();
   }
 
   /**
    * @return {?SDK.LayerTreeBase}
    */
   layerTree() {
     return this._layerTree;
   }
 
   /**
@@ skipping 51 matching lines @@
 Layers.LayerTreeModel.Events = {
   LayerTreeChanged: Symbol('LayerTreeChanged'),
   LayerPainted: Symbol('LayerPainted'),
 };
 
 /**
  * @unrestricted
  */
 Layers.AgentLayerTree = class extends SDK.LayerTreeBase {
   /**
-   * @param {?SDK.Target} target
+   * @param {!Layers.LayerTreeModel} layerTreeModel
    */
-  constructor(target) {
-    super(target);
+  constructor(layerTreeModel) {
+    super(layerTreeModel.target());
+    this._layerTreeModel = layerTreeModel;
   }
 
   /**
    * @param {?Array.<!Protocol.LayerTree.Layer>} payload
    * @param {function()} callback
    */
   setLayers(payload, callback) {
     if (!payload) {
       onBackendNodeIdsResolved.call(this);
       return;
@@ skipping 28 matching lines @@
       return;
     var root;
     var oldLayersById = this._layersById;
     this._layersById = {};
     for (var i = 0; i < layers.length; ++i) {
       var layerId = layers[i].layerId;
       var layer = oldLayersById[layerId];
       if (layer)
         layer._reset(layers[i]);
       else
-        layer = new Layers.AgentLayer(this.target(), layers[i]);
+        layer = new Layers.AgentLayer(this._layerTreeModel, layers[i]);
       this._layersById[layerId] = layer;
       var backendNodeId = layers[i].backendNodeId;
       if (backendNodeId)
         layer._setNode(this.backendNodeIdToNode().get(backendNodeId));
       if (!this.contentRoot() && layer.drawsContent())
         this.setContentRoot(layer);
       var parentId = layer.parentId();
       if (parentId) {
         var parent = this._layersById[parentId];
         if (!parent)
@@ skipping 11 matching lines @@
     }
   }
 };
 
 /**
  * @implements {SDK.Layer}
  * @unrestricted
  */
 Layers.AgentLayer = class {
   /**
-   * @param {?SDK.Target} target
+   * @param {!Layers.LayerTreeModel} layerTreeModel
    * @param {!Protocol.LayerTree.Layer} layerPayload
    */
-  constructor(target, layerPayload) {
-    this._target = target;
+  constructor(layerTreeModel, layerPayload) {
+    this._layerTreeModel = layerTreeModel;
     this._reset(layerPayload);
   }
 
   /**
    * @override
    * @return {string}
    */
   id() {
     return this._layerPayload.layerId;
   }
@@ skipping 158 matching lines @@
    */
   scrollRects() {
     return this._scrollRects;
   }
 
   /**
    * @override
    * @param {function(!Array.<string>)} callback
    */
   requestCompositingReasons(callback) {
-    if (!this._target) {
-      callback([]);
-      return;
-    }
-
     var wrappedCallback = Protocol.inspectorBackend.wrapClientCallback(
         callback, 'Protocol.LayerTree.reasonsForCompositingLayer(): ', undefined, []);
-    this._target.layerTreeAgent().compositingReasons(this.id(), wrappedCallback);
+    this._layerTreeModel._layerTreeAgent.compositingReasons(this.id(), wrappedCallback);
   }
 
   /**
    * @override
    * @return {boolean}
    */
   drawsContent() {
     return this._layerPayload.drawsContent;
   }
 
   /**
    * @override
    * @return {number}
    */
   gpuMemoryUsage() {
     /**
      * @const
      */
     var bytesPerPixel = 4;
     return this.drawsContent() ? this.width() * this.height() * bytesPerPixel : 0;
   }
 
   /**
    * @override
    * @return {!Array<!Promise<?SDK.SnapshotWithRect>>}
    */
   snapshots() {
-    var rect = {x: 0, y: 0, width: this.width(), height: this.height()};
-    var promise = this._target.layerTreeAgent().makeSnapshot(
-        this.id(), (error, snapshotId) => error || !this._target ?
-            null :
-            {rect: rect, snapshot: new SDK.PaintProfilerSnapshot(this._target, snapshotId)});
+    var promise = this._layerTreeModel._paintProfilerModel.makeSnapshot(this.id()).then(snapshot => {
+      if (!snapshot)
+        return null;
+      return {rect: {x: 0, y: 0, width: this.width(), height: this.height()}, snapshot: snapshot};
+    });
     return [promise];
   }
 
   /**
    * @param {!Protocol.DOM.Rect} rect
    */
   _didPaint(rect) {
     this._lastPaintRect = rect;
     this._paintCount = this.paintCount() + 1;
     this._image = null;
@@ skipping 98 matching lines @@
 
   /**
    * @override
    * @param {!Protocol.LayerTree.LayerId} layerId
    * @param {!Protocol.DOM.Rect} clipRect
    */
   layerPainted(layerId, clipRect) {
     this._layerTreeModel._layerPainted(layerId, clipRect);
   }
 };