tracy/client/TracyScoped.hpp

#ifndef __TRACYSCOPED_HPP__
#define __TRACYSCOPED_HPP__

#include <stdint.h>
#include <string.h>

#include "../common/TracySystem.hpp"
#include "../common/TracyAlloc.hpp"
#include "TracyProfiler.hpp"

namespace tracy
{

class ScopedZone
{
public:
    tracy_force_inline ScopedZone( const SourceLocation* srcloc )
    {
        const auto thread = GetThreadHandle();
        m_thread = thread;
        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::ZoneBegin;
        item->zoneBegin.time = Profiler::GetTime( item->zoneBegin.cpu );
        item->zoneBegin.thread = thread;
        item->zoneBegin.srcloc = (uint64_t)srcloc;
        tail.store( magic + 1, std::memory_order_release );
    }

    tracy_force_inline ~ScopedZone()
    {
        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 = m_thread;
        tail.store( magic + 1, std::memory_order_release );
    }

    tracy_force_inline void Text( const char* txt, size_t size )
    {
        Magic magic;
        auto ptr = (char*)tracy_malloc( size+1 );
        memcpy( ptr, txt, size );
        ptr[size] = '\0';
        auto& token = s_token.ptr;
        auto& tail = token->get_tail_index();
        auto item = token->enqueue_begin<moodycamel::CanAlloc>( magic );
        item->hdr.type = QueueType::ZoneText;
        item->zoneText.thread = m_thread;
        item->zoneText.text = (uint64_t)ptr;
        tail.store( magic + 1, std::memory_order_release );
    }

    tracy_force_inline void Name( const char* name )
    {
        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::ZoneName;
        item->zoneName.thread = m_thread;
        item->zoneName.name = (uint64_t)name;
        tail.store( magic + 1, std::memory_order_release );
    }

private:
    uint64_t m_thread;
};

}

#endif