Re-use profiler visualization for extra networks

.eslintrc.js

@@ -78,6 +78,8 @@ module.exports = {
         //extraNetworks.js
         requestGet: "readonly",
         popup: "readonly",
+        // profilerVisualization.js
+        createVisualizationTable: "readonly",
         // from python
         localization: "readonly",
         // progrssbar.js

javascript/extraNetworks.js

@@ -528,12 +528,74 @@ function popupId(id) {
     popup(storedPopupIds[id]);
 }
 
+function extraNetworksFlattenMetadata(obj) {
+    const result = {};
+
+    // Convert any stringified JSON objects to actual objects
+    for (const key of Object.keys(obj)) {
+        if (typeof obj[key] === 'string') {
+            try {
+                const parsed = JSON.parse(obj[key]);
+                if (parsed && typeof parsed === 'object') {
+                    obj[key] = parsed;
+                }
+            } catch (error) {
+                continue;
+            }
+        }
+    }
+
+    // Flatten the object
+    for (const key of Object.keys(obj)) {
+        if (typeof obj[key] === 'object' && obj[key] !== null) {
+            const nested = extraNetworksFlattenMetadata(obj[key]);
+            for (const nestedKey of Object.keys(nested)) {
+                result[`${key}/${nestedKey}`] = nested[nestedKey];
+            }
+        } else {
+            result[key] = obj[key];
+        }
+    }
+
+    // Special case for handling modelspec keys
+    for (const key of Object.keys(result)) {
+        if (key.startsWith("modelspec.")) {
+            result[key.replaceAll(".", "/")] = result[key];
+            delete result[key];
+        }
+    }
+
+    // Add empty keys to designate hierarchy
+    for (const key of Object.keys(result)) {
+        const parts = key.split("/");
+        for (let i = 1; i < parts.length; i++) {
+            const parent = parts.slice(0, i).join("/");
+            if (!result[parent]) {
+                result[parent] = "";
+            }
+        }
+    }
+
+    return result;
+}
+
 function extraNetworksShowMetadata(text) {
+    try {
+        let parsed = JSON.parse(text);
+        if (parsed && typeof parsed === 'object') {
+            parsed = extraNetworksFlattenMetadata(parsed);
+            const table = createVisualizationTable(parsed, 0);
+            popup(table);
+            return;
+        }
+    } catch (error) { console.debug(error); }
+
     var elem = document.createElement('pre');
     elem.classList.add('popup-metadata');
     elem.textContent = text;
 
     popup(elem);
+    return;
 }
 
 function requestGet(url, data, handler, errorHandler) {

javascript/profilerVisualization.js

@@ -33,23 +33,33 @@ function createRow(table, cellName, items) {
     return res;
 }
 
-function showProfile(path, cutoff = 0.05) {
-    requestGet(path, {}, function(data) {
+function createVisualizationTable(data, cutoff = 0, sort = "") {
     var table = document.createElement('table');
     table.className = 'popup-table';
 
-        data.records['total'] = data.total;
-        var keys = Object.keys(data.records).sort(function(a, b) {
-            return data.records[b] - data.records[a];
+    var keys = Object.keys(data);
+    if (sort === "number") {
+        keys = keys.sort(function(a, b) {
+            return data[b] - data[a];
         });
+    } else {
+        keys = keys.sort();
+    }
     var items = keys.map(function(x) {
-            return {key: x, parts: x.split('/'), time: data.records[x]};
+        return {key: x, parts: x.split('/'), value: data[x]};
     });
     var maxLength = items.reduce(function(a, b) {
         return Math.max(a, b.parts.length);
     }, 0);
 
-        var cols = createRow(table, 'th', ['record', 'seconds']);
+    var cols = createRow(
+        table,
+        'th',
+        [
+            cutoff === 0 ? 'key' : 'record',
+            cutoff === 0 ? 'value' : 'seconds'
+        ]
+    );
     cols[0].colSpan = maxLength;
 
     function arraysEqual(a, b) {
@@ -60,21 +70,25 @@ function showProfile(path, cutoff = 0.05) {
         var matching = items.filter(function(x) {
             return x.parts[level] && !x.parts[level + 1] && arraysEqual(x.parts.slice(0, level), parent);
         });
-            var sorted = matching.sort(function(a, b) {
-                return b.time - a.time;
+        if (sort === "number") {
+            matching = matching.sort(function(a, b) {
+                return b.value - a.value;
             });
+        } else {
+            matching = matching.sort();
+        }
         var othersTime = 0;
         var othersList = [];
         var othersRows = [];
         var childrenRows = [];
-            sorted.forEach(function(x) {
-                var visible = x.time >= cutoff && !hide;
+        matching.forEach(function(x) {
+            var visible = (cutoff === 0 && !hide) || (x.value >= cutoff && !hide);
 
             var cells = [];
             for (var i = 0; i < maxLength; i++) {
                 cells.push(x.parts[i]);
             }
-                cells.push(x.time.toFixed(3));
+            cells.push(cutoff === 0 ? x.value : x.value.toFixed(3));
             var cols = createRow(table, 'td', cells);
             for (i = 0; i < level; i++) {
                 cols[i].className = 'muted';
@@ -85,10 +99,10 @@ function showProfile(path, cutoff = 0.05) {
                 tr.classList.add("hidden");
             }
 
-                if (x.time >= cutoff) {
+            if (cutoff === 0 || x.value >= cutoff) {
                 childrenRows.push(tr);
             } else {
-                    othersTime += x.time;
+                othersTime += x.value;
                 othersList.push(x.parts[level]);
                 othersRows.push(tr);
             }
@@ -147,6 +161,13 @@ function showProfile(path, cutoff = 0.05) {
 
     addLevel(0, []);
 
+    return table;
+}
+
+function showProfile(path, cutoff = 0.05) {
+    requestGet(path, {}, function(data) {
+        data.records['total'] = data.total;
+        const table = createVisualizationTable(data.records, cutoff, "number");
         popup(table);
     });
 }