add custom profiling via Debug(Start|Stop)Profile

BeamMP · Jan 24, 2024 · 0e2855d · 0e2855d
1 parent 5cd7584
commit 0e2855d
Show file tree

Hide file tree

Showing 4 changed files with 14 additions and 0 deletions.
diff --git a/include/Profiling.h b/include/Profiling.h
@@ -21,6 +21,7 @@ struct Stats {
     double stddev;
     double min;
     double max;
+    size_t n;
 };
 
 /// Calculates and stores the moving average over K samples of execution time data

diff --git a/include/TLuaEngine.h b/include/TLuaEngine.h
@@ -256,6 +256,7 @@ class TLuaEngine : public std::enable_shared_from_this<TLuaEngine>, IThreaded {
         sol::table Lua_FS_ListDirectories(const std::string& Path);
 
         prof::UnitProfileCollection mProfile {};
+        std::unordered_map<std::string, prof::TimePoint> mProfileStarts;
 
         std::string mName;
         TLuaStateId mStateId;

diff --git a/src/Profiling.cpp b/src/Profiling.cpp
@@ -30,6 +30,7 @@ prof::Stats prof::UnitExecutionTime::stats() const {
     if (measurements->size() == 0) {
         return result;
     }
+    result.n = measurements->size();
     result.max = std::numeric_limits<double>::min();
     result.min = std::numeric_limits<double>::max();
     Duration sum {};

diff --git a/src/TLuaEngine.cpp b/src/TLuaEngine.cpp
@@ -859,9 +859,20 @@ TLuaEngine::StateThreadData::StateThreadData(const std::string& Name, TLuaStateI
             Result[name]["stddev"] = stat.stddev;
             Result[name]["min"] = stat.min;
             Result[name]["max"] = stat.max;
+            Result[name]["n"] = stat.n;
         }
         return Result;
     });
+    UtilTable.set_function("DebugStartProfile", [this](const std::string& name) {
+        mProfileStarts[name] = prof::now();
+    });
+    UtilTable.set_function("DebugStopProfile", [this](const std::string& name) {
+        if (!mProfileStarts.contains(name)) {
+            beammp_lua_errorf("DebugStopProfile('{}') failed, because a profile for '{}' wasn't started", name, name);
+            return;
+        }
+        mProfile.add_sample(name, prof::duration(mProfileStarts.at(name), prof::now()));
+    });
 
     auto HttpTable = StateView.create_named_table("Http");
     HttpTable.set_function("CreateConnection", [this](const std::string& host, uint16_t port) {