Add native proc profiling

OpenDreamProject · Dec 8, 2024 · c9f8c51 · c9f8c51
1 parent ca18c19
commit c9f8c51
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 2 deletions.
diff --git a/OpenDreamRuntime/DreamThread.cs b/OpenDreamRuntime/DreamThread.cs
@@ -211,7 +211,10 @@ public static DreamValue Run(DreamProc proc, DreamObject src, DreamObject? usr,
             var context = new DreamThread(proc.ToString());
 
             if (proc is NativeProc nativeProc) {
-                return nativeProc.Call(context, src, usr, new(arguments));
+                var zone = Profiler.BeginZone(filePath:"Native Proc", lineNumber:0, memberName:nativeProc.Name);
+                var result = nativeProc.Call(context, src, usr, new(arguments));
+                zone?.Dispose();
+                return result;
             }
 
             var state = proc.CreateState(context, src, usr, new DreamProcArguments(arguments));

diff --git a/OpenDreamRuntime/Procs/AsyncNativeProc.cs b/OpenDreamRuntime/Procs/AsyncNativeProc.cs
@@ -26,7 +26,7 @@ public sealed class State : ProcState {
             private AsyncNativeProc? _proc;
             public override DreamProc? Proc => _proc;
 
-            public override (string SourceFile, int Line) TracyLocationId => ("Native Proc", 0);
+            public override (string SourceFile, int Line) TracyLocationId => ("Async Native Proc", 0);
 
             private Func<State, Task<DreamValue>> _taskFunc;
             private Task? _task;

diff --git a/OpenDreamRuntime/Procs/DMProc.cs b/OpenDreamRuntime/Procs/DMProc.cs
@@ -488,7 +488,9 @@ public void SetReturn(DreamValue value) {
         public ProcStatus Call(DreamProc proc, DreamObject? src, DreamProcArguments arguments) {
             if (proc is NativeProc p) {
                 // Skip a whole song and dance.
+                var zone = Profiler.BeginZone(filePath:"Native Proc", lineNumber:0, memberName:p.Name);
                 Push(p.Call(Thread, src, Usr, arguments));
+                zone?.Dispose();
                 return ProcStatus.Continue;
             }