Profiling.h

Go to the documentation of this file.

 
#ifndef PBAT_PROFILING_PROFILING_H
#define PBAT_PROFILING_PROFILING_H
 
#include "PhysicsBasedAnimationToolkitExport.h"
 
#if defined(PBAT_HAS_TRACY_PROFILER)
    #if not defined(__CUDACC__)
        #define PBAT_CAN_USE_TRACY_CPP
        #include <tracy/Tracy.hpp>
        #define PBAT_PROFILE_SCOPE             ZoneScoped
        #define PBAT_PROFILE_NAMED_SCOPE(name) ZoneScopedN(name)
    #else
        #define PBAT_CAN_USE_TRACY_C
        #include <tracy/TracyC.h>
        #define PBAT_PROFILE_SCOPE
        #define PBAT_PROFILE_NAMED_SCOPE(name)
        #define PBAT_PROFILE_CUDA_HOST_SCOPE_START(var)             TracyCZone(var, true)
        #define PBAT_PROFILE_NAMED_CUDA_HOST_SCOPE_START(var, name) TracyCZoneN(var, name, true)
        #define PBAT_PROFILE_CUDA_HOST_SCOPE_END(var)               TracyCZoneEnd(var)
    #endif // __CUDACC__
#else
    #define PBAT_PROFILE_SCOPE
    #define PBAT_PROFILE_NAMED_SCOPE(name)
    #define PBAT_PROFILE_CUDA_HOST_SCOPE_START(var)
    #define PBAT_PROFILE_CUDA_HOST_SCOPE_START(var, name)
    #define PBAT_PROFILE_CUDA_HOST_SCOPE_END(var)
#endif // PBAT_CAN_USE_TRACY








 
#include <map>
#include <string>
#include <string_view>
#include <type_traits>
 
namespace pbat {
namespace profiling {

PBAT_API void BeginFrame(std::string_view name);

PBAT_API void EndFrame(std::string_view name);

PBAT_API bool IsConnectedToServer();

template <class Func, class... Args>
std::invoke_result_t<Func, Args...> Profile(std::string const& zoneName, Func&& f, Args&&... args)
{
#ifdef PBAT_CAN_USE_TRACY_CPP
    static auto constexpr line = (uint32_t)TracyLine;
    struct SourceLocationData
    {
        SourceLocationData(std::string_view zoneNameView)
            : name(zoneNameView), function(TracyFunction), file(TracyFile), data()
        {
            data.name     = name.data();
            data.function = function.data();
            data.file     = file.data();
            data.line     = line;
            data.color    = 0;
        }
        std::string name;
        std::string function;
        std::string file;
        tracy::SourceLocationData data;
    };
    static std::map<std::string, SourceLocationData> zones{};
    auto it = zones.find(zoneName);
    if (it == zones.end())
    {
        bool inserted{false};
        std::tie(it, inserted) = zones.insert({zoneName, SourceLocationData(zoneName)});
        assert(inserted);
    }
    SourceLocationData const& data = it->second;
    tracy::ScopedZone zone(&(data.data));
#endif // PBAT_CAN_USE_TRACY
    return f(std::forward<Args>(args)...);
}
 
} // namespace profiling
} // namespace pbat
 
#endif // PBAT_PROFILING_PROFILING_H