Lua profiling framework. No source location transfer yet.

2025-03-20 07:40:02 +08:00 · 2017-11-05 15:04:55 +01:00
parent 3dc7d04ab4
commit 0fb5f012ce
4 changed files with 158 additions and 0 deletions
@@ -0,0 +1,132 @@
+#ifndef __TRACYLUA_HPP__
+#define __TRACYLUA_HPP__
+
+#include <string.h>
+
+#include "common/TracySystem.hpp"
+#include "client/TracyProfiler.hpp"
+
+namespace tracy
+{
+
+#ifndef TRACY_ENABLE
+
+namespace detail
+{
+static inline int noop( lua_State* L ) { return 0; }
+}
+
+static inline void LuaRegister( lua_State* L )
+{
+    lua_newtable( L );
+    lua_pushcfunction( L, detail::noop );
+    lua_setfield( L, -2, "ZoneBegin" );
+    lua_pushcfunction( L, detail::noop );
+    lua_setfield( L, -2, "ZoneEnd" );
+    lua_setglobal( L, "tracy" );
+}
+
+static inline void LuaRemove( char* script )
+{
+    while( *script )
+    {
+        if( strncmp( script, "tracy.Zone", 10 ) == 0 )
+        {
+            if( strncmp( script + 10, "End()", 5 ) == 0 )
+            {
+                memset( script, ' ', 15 );
+                script += 15;
+            }
+            else if( strncmp( script + 10, "Begin()", 7 ) == 0 )
+            {
+                memset( script, ' ', 17 );
+                script += 17;
+            }
+            else
+            {
+                script += 10;
+            }
+        }
+        else
+        {
+            script++;
+        }
+    }
+}
+
+#else
+
+namespace detail
+{
+
+static inline int LuaZoneBegin( lua_State* L )
+{
+    const uint32_t color = 0x00DD6666;
+
+    lua_Debug dbg;
+    lua_getstack( L, 1, &dbg );
+    lua_getinfo( L, "Snl", &dbg );
+
+    const uint32_t line = dbg.currentline;
+    const auto fsz = strlen( dbg.name );
+    const auto ssz = strlen( dbg.source );
+
+    // Data layout:
+    //  4b  payload size
+    //  4b  color
+    //  4b  source line
+    //  fsz function name
+    //  1b  null terminator
+    //  ssz source file name
+    const uint32_t sz = 4 + 4 + 4 + fsz + 1 + ssz;
+    auto ptr = (char*)tracy_malloc( sz );
+    memcpy( ptr, &sz, 4 );
+    memcpy( ptr + 4, &color, 4 );
+    memcpy( ptr + 8, &line, 4 );
+    memcpy( ptr + 12, dbg.name, fsz+1 );
+    memcpy( ptr + 12 + fsz + 1, dbg.source, ssz );
+
+    Magic magic;
+    auto& token = s_token.ptr;
+    auto& tail = token->get_tail_index();
+    auto item = token->enqueue_begin<moodycamel::CanAlloc>( magic );
+    item->hdr.type = QueueType::ZoneBeginAllocSrcLoc;
+    item->zoneBegin.time = Profiler::GetTime( item->zoneBegin.cpu );
+    item->zoneBegin.thread = GetThreadHandle();
+    item->zoneBegin.srcloc = (uint64)ptr;
+    tail.store( magic + 1, std::memory_order_release );
+    return 0;
+}
+
+static inline int LuaZoneEnd( lua_State* L )
+{
+    Magic magic;
+    auto& token = s_token.ptr;
+    auto& tail = token->get_tail_index();
+    auto item = token->enqueue_begin<moodycamel::CanAlloc>( magic );
+    item->hdr.type = QueueType::ZoneEnd;
+    item->zoneEnd.time = Profiler::GetTime( item->zoneEnd.cpu );
+    item->zoneEnd.thread = GetThreadHandle();
+    tail.store( magic + 1, std::memory_order_release );
+    return 0;
+}
+
+}
+
+static inline void LuaRegister( lua_State* L )
+{
+    lua_newtable( L );
+    lua_pushcfunction( L, detail::LuaZoneBegin );
+    lua_setfield( L, -2, "ZoneBegin" );
+    lua_pushcfunction( L, detail::LuaZoneEnd );
+    lua_setfield( L, -2, "ZoneEnd" );
+    lua_setglobal( L, "tracy" );
+}
+
+static inline void LuaRemove( char* script ) {}
+
+#endif
+
+}
+
+#endif