1 files changed, 27 insertions, 0 deletions
diff --git a/source/util/Diag.mc b/source/util/Diag.mc
index a09d3f8..0ab121f 100644
--- a/source/util/Diag.mc
+++ b/source/util/Diag.mc
@@ -96,4 +96,31 @@ module Diag {
     (:release)
     function overlay(dc as Graphics.Dc, frameStartMs as Number) as Void {
     }
+
+    // Peak memory while long work runs (sampled by the import job).
+    (:debug)
+    var peakMemory as Number = 0;
+
+    (:debug)
+    function sampleMemory() as Void {
+        var used = System.getSystemStats().usedMemory;
+        if (used > peakMemory) {
+            peakMemory = used;
+        }
+    }
+
+    (:release)
+    function sampleMemory() as Void {
+    }
+
+    // Reports an import's size, duration and peak memory for UI scripts.
+    (:debug)
+    function importDone(entries as Number, wallMs as Number) as Void {
+        System.println("DIAG import entries=" + entries + " wall_ms=" + wallMs + " mem_peak=" + peakMemory
+            + " mem_total=" + System.getSystemStats().totalMemory);
+    }
+
+    (:release)
+    function importDone(entries as Number, wallMs as Number) as Void {
+    }
 }