mirror of
https://github.com/wolfpld/tracy.git
synced 2026-07-24 06:59:14 +00:00
f93d17a96f
Extension point so private/unsupported platforms can plug in their own implementations of the kernel/libc primitives Tracy depends on, without patching the `#if`/`#elif` chains. Projects supply a platform header via `-DTRACY_PLATFORM_HEADER="\"my_platform.h\""` at build time. Tracy includes it in any TU that needs the hooks. The header toggles per-category `TRACY_HAS_CUSTOM_*` macros and declares matching `tracy::Platform*` functions. Available hooks: - `TRACY_HAS_CUSTOM_THREAD_ID` → `PlatformGetThreadId` - `TRACY_HAS_CUSTOM_USER_INFO` → `PlatformGetHostname`, `PlatformGetUserLogin`, `PlatformGetUserFullName` - `TRACY_HAS_CUSTOM_SAFE_COPY` → `PlatformSafeMemcpy` - `TRACY_HAS_CUSTOM_ALLOCATOR` → `PlatformMalloc`, `PlatformFree`, `PlatformRealloc`, `PlatformAllocatorInit`, `PlatformAllocatorThreadInit`, `PlatformAllocatorFinalize`, `PlatformAllocatorThreadFinalize` Each hook is wired as the first arm of its respective `#if`/`#elif` chain, so existing supported platforms are unaffected. Template files in `examples/CustomPlatform/` and a new subsection in `manual/tracy.tex` document the mechanism.
52 lines
1.2 KiB
C++
52 lines
1.2 KiB
C++
#include "../common/TracyAlloc.hpp"
|
|
|
|
#if defined TRACY_USE_RPMALLOC || defined TRACY_HAS_CUSTOM_ALLOCATOR
|
|
|
|
#include <atomic>
|
|
|
|
#include "../common/TracyForceInline.hpp"
|
|
#include "../common/TracyYield.hpp"
|
|
|
|
namespace tracy
|
|
{
|
|
|
|
extern thread_local bool RpThreadInitDone;
|
|
extern std::atomic<int> RpInitDone;
|
|
extern std::atomic<int> RpInitLock;
|
|
|
|
tracy_no_inline static void InitAllocatorPlumbing()
|
|
{
|
|
const auto done = RpInitDone.load( std::memory_order_acquire );
|
|
if( !done )
|
|
{
|
|
int expected = 0;
|
|
while( !RpInitLock.compare_exchange_weak( expected, 1, std::memory_order_release, std::memory_order_relaxed ) ) { expected = 0; YieldThread(); }
|
|
const auto done = RpInitDone.load( std::memory_order_acquire );
|
|
if( !done )
|
|
{
|
|
#if defined TRACY_HAS_CUSTOM_ALLOCATOR
|
|
PlatformAllocatorInit();
|
|
#else
|
|
rpmalloc_initialize();
|
|
#endif
|
|
RpInitDone.store( 1, std::memory_order_release );
|
|
}
|
|
RpInitLock.store( 0, std::memory_order_release );
|
|
}
|
|
#if defined TRACY_HAS_CUSTOM_ALLOCATOR
|
|
PlatformAllocatorThreadInit();
|
|
#else
|
|
rpmalloc_thread_initialize();
|
|
#endif
|
|
RpThreadInitDone = true;
|
|
}
|
|
|
|
TRACY_API void InitAllocator()
|
|
{
|
|
if( !RpThreadInitDone ) InitAllocatorPlumbing();
|
|
}
|
|
|
|
}
|
|
|
|
#endif
|