Use heimdalljs-graph to iterate nodes "smarter".

Author: Robert Jackson

app/components/slow-node-times.js

@@ -1,56 +1,87 @@
 import Ember from 'ember';
+import heimdallGraph from 'heimdalljs-graph';
 
 const {
   computed
 } = Ember;
 
+function selfTime(node) {
+  for (let [statName, value] of node.statsIterator()) {
+    if (statName === 'time.self') {
+      return value;
+    }
+  }
+}
+
+function nodeTime(node) {
+  let nodeTotal = selfTime(node);
+
+  for (let childNode of node.dfsIterator((n) => n.label.broccoliNode)) {
+    nodeTotal += selfTime(childNode);
+  }
+
+  return nodeTotal;
+}
+
 export default Ember.Component.extend({
   nodes: computed('data', 'filter', 'pluginNameFilter', 'groupByPluginName', function() {
     let data = this.get('data');
-    let groupByPluginName = this.get('groupByPluginName');
-    let pluginName = this.get('pluginNameFilter');
     if (!data) { return []; }
 
-    let nodes = data.nodes.filter((node) => {
-      if (!node.label.broccoliNode) { return false; }
-      if (pluginName && node.label.broccoliPluginName !== pluginName) { return false; }
+    let nodes = [];
+    let graph = heimdallGraph.loadFromJSON(data);
 
-      return true;
-    })
-
-    if (groupByPluginName) {
-      let pluginNameMap = nodes.reduce((memo, node) => {
-        let pluginName = node.label.broccoliPluginName;
-        memo[pluginName] = memo[pluginName] || { count: 0, time: 0 };
-        memo[pluginName].time += node.stats.time.self;
-        memo[pluginName].count++;
-        return memo;
-      }, {})
-
-      nodes = [];
-      for (let pluginName in pluginNameMap) {
+    for (let node of graph.dfsIterator()) {
+      if (node.label.broccoliNode) {
         nodes.push({
-          label: { name: pluginName, broccoliPluginName: pluginNameMap[pluginName].count },
-          stats: {
-            time: { self: pluginNameMap[pluginName].time}
-          }
-        });
+          label: node.label,
+          time: nodeTime(node)
+        })
       }
     }
 
-    let addonNodes = nodes
+    //let groupByPluginName = this.get('groupByPluginName');
+    //let pluginName = this.get('pluginNameFilter');
+    // nodes = data.nodes.filter((node) => {
+    //   if (!node.label.broccoliNode) { return false; }
+    //   if (pluginName && node.label.broccoliPluginName !== pluginName) { return false; }
+
+    //   return true;
+    // })
+
+    // if (groupByPluginName) {
+    //   let pluginNameMap = nodes.reduce((memo, node) => {
+    //     let pluginName = node.label.broccoliPluginName;
+    //     memo[pluginName] = memo[pluginName] || { count: 0, time: 0 };
+    //     memo[pluginName].time += node.stats.time.self;
+    //     memo[pluginName].count++;
+    //     return memo;
+    //   }, {})
+
+    //   nodes = [];
+    //   for (let pluginName in pluginNameMap) {
+    //     nodes.push({
+    //       label: { name: pluginName, broccoliPluginName: pluginNameMap[pluginName].count },
+    //       stats: {
+    //         time: { self: pluginNameMap[pluginName].time}
+    //       }
+    //     });
+    //   }
+    // }
+
+    let sortedNodes = nodes
         .sort((a, b) => {
-          return b.stats.time.self - a.stats.time.self;
+          return b.time - a.time;
         });
 
-    return addonNodes;
+    return sortedNodes;
   }),
 
   totalTime: computed('nodes', function() {
     let nodes = this.get('nodes');
 
     return nodes.reduce(function(previousValue, node){
-      return previousValue + node.stats.time.self;
+      return previousValue + node.time;
     }, 0);
   }),
 

app/templates/components/slow-node-times.hbs

@@ -15,7 +15,7 @@
       <tr>
         <td>{{node.label.name}}</td>
         <td>{{node.label.broccoliPluginName}}</td>
-        <td>{{ns-to-ms node.stats.time.self}}</td>
+        <td>{{ns-to-ms node.time}}</td>
       </tr>
     {{/each}}
   </tbody>

ember-cli-build.js

@@ -5,6 +5,9 @@ var EmberApp = require('ember-cli/lib/broccoli/ember-app');
 module.exports = function(defaults) {
   var app = new EmberApp(defaults, {
     // Add options here
+    trees: {
+      vendor: 'node_modules/heimdalljs-graph'
+    }
   });
 
   // Use `app.import` to add additional libraries to the generated
@@ -20,5 +23,7 @@ module.exports = function(defaults) {
   // please specify an object with the list of modules as keys
   // along with the exports of each module as its value.
 
+  app.import('vendor/dist/amd/heimdalljs-graph.js');
+
   return app.toTree();
 };