#ifdef _MSC_VER # pragma warning( disable: 4244 4267 ) // conversion from don't care to whatever, possible loss of data #endif #ifdef _WIN32 # include #else # include #endif #include #include #include #ifdef __MINGW32__ # define __STDC_FORMAT_MACROS #endif #include #include #include #include "../common/TracyProtocol.hpp" #include "../common/TracySystem.hpp" #include "TracyFileRead.hpp" #include "TracyFileWrite.hpp" #include "TracySort.hpp" #include "TracyTaskDispatch.hpp" #include "TracyVersion.hpp" #include "TracyWorker.hpp" #include "TracyYield.hpp" namespace tracy { static tracy_force_inline uint64_t PackFileLine( uint32_t fileIdx, uint32_t line ) { return ( uint64_t( fileIdx ) << 32 ) | line; } static tracy_force_inline uint32_t UnpackFileLine( uint64_t packed, uint32_t& line ) { line = packed & 0xFFFFFFFF; return packed >> 32; } static bool SourceFileValid( const char* fn, uint64_t olderThan ) { struct stat buf; if( stat( fn, &buf ) == 0 && ( buf.st_mode & S_IFREG ) != 0 ) { return (uint64_t)buf.st_mtime < olderThan; } return false; } static const uint8_t FileHeader[8] { 't', 'r', 'a', 'c', 'y', Version::Major, Version::Minor, Version::Patch }; enum { FileHeaderMagic = 5 }; static const int CurrentVersion = FileVersion( Version::Major, Version::Minor, Version::Patch ); static const int MinSupportedVersion = FileVersion( 0, 6, 0 ); static void UpdateLockCountLockable( LockMap& lockmap, size_t pos ) { auto& timeline = lockmap.timeline; bool isContended = lockmap.isContended; uint8_t lockingThread; uint8_t lockCount; uint64_t waitList; if( pos == 0 ) { lockingThread = 0; lockCount = 0; waitList = 0; } else { const auto& tl = timeline[pos-1]; lockingThread = tl.lockingThread; lockCount = tl.lockCount; waitList = tl.waitList; } const auto end = timeline.size(); while( pos != end ) { auto& tl = timeline[pos]; const auto tbit = uint64_t( 1 ) << tl.ptr->thread; switch( (LockEvent::Type)tl.ptr->type ) { case LockEvent::Type::Wait: waitList |= tbit; break; case LockEvent::Type::Obtain: assert( lockCount < std::numeric_limits::max() ); assert( ( waitList & tbit ) != 0 ); waitList &= ~tbit; lockingThread = tl.ptr->thread; lockCount++; break; case LockEvent::Type::Release: assert( lockCount > 0 ); lockCount--; break; default: break; } tl.lockingThread = lockingThread; tl.waitList = waitList; tl.lockCount = lockCount; if( !isContended ) isContended = lockCount != 0 && waitList != 0; pos++; } lockmap.isContended = isContended; } static void UpdateLockCountSharedLockable( LockMap& lockmap, size_t pos ) { auto& timeline = lockmap.timeline; bool isContended = lockmap.isContended; uint8_t lockingThread; uint8_t lockCount; uint64_t waitShared; uint64_t waitList; uint64_t sharedList; if( pos == 0 ) { lockingThread = 0; lockCount = 0; waitShared = 0; waitList = 0; sharedList = 0; } else { const auto& tl = timeline[pos-1]; const auto tlp = (const LockEventShared*)(const LockEvent*)tl.ptr; lockingThread = tl.lockingThread; lockCount = tl.lockCount; waitShared = tlp->waitShared; waitList = tl.waitList; sharedList = tlp->sharedList; } const auto end = timeline.size(); // ObtainShared and ReleaseShared should assert on lockCount == 0, but // due to the async retrieval of data from threads that's not possible. while( pos != end ) { auto& tl = timeline[pos]; const auto tlp = (LockEventShared*)(LockEvent*)tl.ptr; const auto tbit = uint64_t( 1 ) << tlp->thread; switch( (LockEvent::Type)tlp->type ) { case LockEvent::Type::Wait: waitList |= tbit; break; case LockEvent::Type::WaitShared: waitShared |= tbit; break; case LockEvent::Type::Obtain: assert( lockCount < std::numeric_limits::max() ); assert( ( waitList & tbit ) != 0 ); waitList &= ~tbit; lockingThread = tlp->thread; lockCount++; break; case LockEvent::Type::Release: assert( lockCount > 0 ); lockCount--; break; case LockEvent::Type::ObtainShared: assert( ( waitShared & tbit ) != 0 ); assert( ( sharedList & tbit ) == 0 ); waitShared &= ~tbit; sharedList |= tbit; break; case LockEvent::Type::ReleaseShared: assert( ( sharedList & tbit ) != 0 ); sharedList &= ~tbit; break; default: break; } tl.lockingThread = lockingThread; tlp->waitShared = waitShared; tl.waitList = waitList; tlp->sharedList = sharedList; tl.lockCount = lockCount; if( !isContended ) isContended = ( lockCount != 0 && ( waitList != 0 || waitShared != 0 ) ) || ( sharedList != 0 && waitList != 0 ); pos++; } lockmap.isContended = isContended; } static inline void UpdateLockCount( LockMap& lockmap, size_t pos ) { if( lockmap.type == LockType::Lockable ) { UpdateLockCountLockable( lockmap, pos ); } else { UpdateLockCountSharedLockable( lockmap, pos ); } } static tracy_force_inline void WriteTimeOffset( FileWrite& f, int64_t& refTime, int64_t time ) { int64_t timeOffset = time - refTime; refTime += timeOffset; f.Write( &timeOffset, sizeof( timeOffset ) ); } static tracy_force_inline int64_t ReadTimeOffset( FileRead& f, int64_t& refTime ) { int64_t timeOffset; f.Read( timeOffset ); refTime += timeOffset; return refTime; } static tracy_force_inline void UpdateLockRange( LockMap& lockmap, const LockEvent& ev, int64_t lt ) { auto& range = lockmap.range[ev.thread]; if( range.start > lt ) range.start = lt; if( range.end < lt ) range.end = lt; } LoadProgress Worker::s_loadProgress; Worker::Worker( const char* addr, int port ) : m_addr( addr ) , m_port( port ) , m_hasData( false ) , m_stream( LZ4_createStreamDecode() ) , m_buffer( new char[TargetFrameSize*3 + 1] ) , m_bufferOffset( 0 ) , m_pendingStrings( 0 ) , m_pendingThreads( 0 ) , m_pendingExternalNames( 0 ) , m_pendingSourceLocation( 0 ) , m_pendingCallstackFrames( 0 ) , m_pendingCallstackSubframes( 0 ) , m_pendingCodeInformation( 0 ) , m_callstackFrameStaging( nullptr ) , m_traceVersion( CurrentVersion ) , m_loadTime( 0 ) { m_data.sourceLocationExpand.push_back( 0 ); m_data.localThreadCompress.InitZero(); m_data.callstackPayload.push_back( nullptr ); m_data.zoneExtra.push_back( ZoneExtra {} ); memset( m_gpuCtxMap, 0, sizeof( m_gpuCtxMap ) ); #ifndef TRACY_NO_STATISTICS m_data.sourceLocationZonesReady = true; m_data.callstackSamplesReady = true; m_data.ghostZonesReady = true; m_data.ctxUsageReady = true; m_data.symbolSamplesReady = true; #endif m_thread = std::thread( [this] { SetThreadName( "Tracy Worker" ); Exec(); } ); m_threadNet = std::thread( [this] { SetThreadName( "Tracy Network" ); Network(); } ); } Worker::Worker( const std::string& program, const std::vector& timeline, const std::vector& messages, const std::vector& plots ) : m_hasData( true ) , m_delay( 0 ) , m_resolution( 0 ) , m_captureProgram( program ) , m_captureTime( 0 ) , m_pid( 0 ) , m_samplingPeriod( 0 ) , m_stream( nullptr ) , m_buffer( nullptr ) , m_traceVersion( CurrentVersion ) { m_data.sourceLocationExpand.push_back( 0 ); m_data.localThreadCompress.InitZero(); m_data.callstackPayload.push_back( nullptr ); m_data.zoneExtra.push_back( ZoneExtra {} ); m_data.lastTime = 0; if( !timeline.empty() ) { m_data.lastTime = timeline.back().timestamp; } if( !messages.empty() ) { if( m_data.lastTime < (int64_t)messages.back().timestamp ) m_data.lastTime = messages.back().timestamp; } if( !plots.empty() ) { for( auto& v : plots ) { if( m_data.lastTime < v.data.back().first ) m_data.lastTime = v.data.back().first; } } for( auto& v : timeline ) { if( !v.isEnd ) { SourceLocation srcloc { StringRef(), StringRef( StringRef::Idx, StoreString( v.name.c_str(), v.name.size() ).idx ), StringRef(), 0, 0 }; int key; auto it = m_data.sourceLocationPayloadMap.find( &srcloc ); if( it == m_data.sourceLocationPayloadMap.end() ) { auto slptr = m_slab.Alloc(); memcpy( slptr, &srcloc, sizeof( srcloc ) ); uint32_t idx = m_data.sourceLocationPayload.size(); m_data.sourceLocationPayloadMap.emplace( slptr, idx ); m_data.sourceLocationPayload.push_back( slptr ); key = -int16_t( idx + 1 ); #ifndef TRACY_NO_STATISTICS auto res = m_data.sourceLocationZones.emplace( key, SourceLocationZones() ); m_data.srclocZonesLast.first = key; m_data.srclocZonesLast.second = &res.first->second; #else auto res = m_data.sourceLocationZonesCnt.emplace( key, 0 ); m_data.srclocCntLast.first = key; m_data.srclocCntLast.second = &res.first->second; #endif } else { key = -int16_t( it->second + 1 ); } auto zone = AllocZoneEvent(); zone->SetStartSrcLoc( v.timestamp, key ); zone->SetEnd( -1 ); zone->SetChild( -1 ); if( !v.text.empty() ) { auto& extra = RequestZoneExtra( *zone ); extra.text = StringIdx( StoreString( v.text.c_str(), v.text.size() ).idx ); } m_threadCtxData = NoticeThread( v.tid ); NewZone( zone, v.tid ); } else { auto td = NoticeThread( v.tid ); td->zoneIdStack.pop_back(); auto& stack = td->stack; auto zone = stack.back_and_pop(); zone->SetEnd( v.timestamp ); #ifndef TRACY_NO_STATISTICS auto slz = GetSourceLocationZones( zone->SrcLoc() ); auto& ztd = slz->zones.push_next(); ztd.SetZone( zone ); ztd.SetThread( CompressThread( v.tid ) ); #else CountZoneStatistics( zone ); #endif } } for( auto& v : messages ) { auto msg = m_slab.Alloc(); msg->time = v.timestamp; msg->ref = StringRef( StringRef::Type::Idx, StoreString( v.message.c_str(), v.message.size() ).idx ); msg->thread = CompressThread( v.tid ); msg->color = 0xFFFFFFFF; msg->callstack.SetVal( 0 ); InsertMessageData( msg ); } for( auto& v : plots ) { uint64_t nptr = (uint64_t)&v.name; auto it = m_data.strings.find( nptr ); if( it == m_data.strings.end() ) { const auto sl = StoreString( v.name.c_str(), v.name.size() ); m_data.strings.emplace( nptr, sl.ptr ); } auto plot = m_slab.AllocInit(); plot->name = nptr; plot->type = PlotType::User; plot->format = v.format; double min = v.data.begin()->second; double max = v.data.begin()->second; plot->data.reserve_exact( v.data.size(), m_slab ); size_t idx = 0; for( auto& p : v.data ) { plot->data[idx].time.SetVal( p.first ); plot->data[idx].val = p.second; idx++; if( min > p.second ) min = p.second; else if( max < p.second ) max = p.second; } plot->min = min; plot->max = max; m_data.plots.Data().push_back( plot ); } for( auto& t : m_threadMap ) { char buf[64]; sprintf( buf, "%" PRIu64, t.first ); AddThreadString( t.first, buf, strlen( buf ) ); } m_data.framesBase = m_data.frames.Retrieve( 0, [this] ( uint64_t name ) { auto fd = m_slab.AllocInit(); fd->name = name; fd->continuous = 1; return fd; }, [this] ( uint64_t name ) { assert( name == 0 ); char tmp[6] = "Frame"; HandleFrameName( name, tmp, 5 ); } ); m_data.framesBase->frames.push_back( FrameEvent{ 0, -1, -1 } ); m_data.framesBase->frames.push_back( FrameEvent{ 0, -1, -1 } ); } Worker::Worker( FileRead& f, EventType::Type eventMask, bool bgTasks ) : m_hasData( true ) , m_stream( nullptr ) , m_buffer( nullptr ) { auto loadStart = std::chrono::high_resolution_clock::now(); m_data.callstackPayload.push_back( nullptr ); int fileVer = 0; uint8_t hdr[8]; f.Read( hdr, sizeof( hdr ) ); if( memcmp( FileHeader, hdr, FileHeaderMagic ) == 0 ) { fileVer = FileVersion( hdr[FileHeaderMagic], hdr[FileHeaderMagic+1], hdr[FileHeaderMagic+2] ); if( fileVer > CurrentVersion ) { throw UnsupportedVersion( fileVer ); } if( fileVer < MinSupportedVersion ) { throw LegacyVersion( fileVer ); } f.Read( m_delay ); } else { throw LegacyVersion( FileVersion( 0, 2, 0 ) ); } m_traceVersion = fileVer; s_loadProgress.total.store( 11, std::memory_order_relaxed ); s_loadProgress.subTotal.store( 0, std::memory_order_relaxed ); s_loadProgress.progress.store( LoadProgress::Initialization, std::memory_order_relaxed ); f.Read5( m_resolution, m_timerMul, m_data.lastTime, m_data.frameOffset, m_pid ); if( fileVer >= FileVersion( 0, 6, 5 ) ) { f.Read( m_samplingPeriod ); } else { m_samplingPeriod = 0; } if( fileVer >= FileVersion( 0, 6, 7 ) ) { f.Read( m_data.cpuArch ); } if( fileVer >= FileVersion( 0, 6, 12 ) ) { f.Read( m_data.cpuId ); f.Read( m_data.cpuManufacturer, 12 ); m_data.cpuManufacturer[12] = '\0'; } uint64_t sz; { f.Read( sz ); assert( sz < 1024 ); char tmp[1024]; f.Read( tmp, sz ); m_captureName = std::string( tmp, tmp+sz ); } { f.Read( sz ); assert( sz < 1024 ); char tmp[1024]; f.Read( tmp, sz ); m_captureProgram = std::string( tmp, tmp+sz ); f.Read( m_captureTime ); } { f.Read( sz ); assert( sz < 1024 ); char tmp[1024]; f.Read( tmp, sz ); m_hostInfo = std::string( tmp, tmp+sz ); } if( fileVer >= FileVersion( 0, 6, 2 ) ) { f.Read( sz ); m_data.cpuTopology.reserve( sz ); for( uint64_t i=0; i> {} ).first; package.second.reserve( psz ); for( uint64_t j=0; j {} ).first; core.second.reserve( csz ); for( uint64_t k=0; k(); uint64_t fsz; f.Read3( ptr->name, ptr->continuous, fsz ); ptr->frames.reserve_exact( fsz, m_slab ); int64_t refTime = 0; if( ptr->continuous ) { for( uint64_t j=0; jframes[j].start = ReadTimeOffset( f, refTime ); ptr->frames[j].end = -1; f.Read( &ptr->frames[j].frameImage, sizeof( int32_t ) ); } } else { for( uint64_t j=0; jframes[j].start = ReadTimeOffset( f, refTime ); ptr->frames[j].end = ReadTimeOffset( f, refTime ); f.Read( &ptr->frames[j].frameImage, sizeof( int32_t ) ); } } for( uint64_t j=0; j 0 ) { ptr->min = std::min( ptr->min, timeSpan ); ptr->max = std::max( ptr->max, timeSpan ); ptr->total += timeSpan; ptr->sumSq += double( timeSpan ) * timeSpan; } } m_data.frames.Data()[i] = ptr; } m_data.framesBase = m_data.frames.Data()[0]; assert( m_data.framesBase->name == 0 ); unordered_flat_map pointerMap; f.Read( sz ); m_data.stringMap.reserve( sz ); m_data.stringData.reserve_exact( sz, m_slab ); for( uint64_t i=0; i( ssz+1 ); f.Read( dst, ssz ); dst[ssz] = '\0'; m_data.stringMap.emplace( charutil::StringKey { dst, ssz }, i ); m_data.stringData[i] = ( dst ); pointerMap.emplace( ptr, dst ); } f.Read( sz ); for( uint64_t i=0; isecond ); } } f.Read( sz ); for( uint64_t i=0; isecond ); } } f.Read( sz ); for( uint64_t i=0; isecond, it2->second ) ); } } m_data.localThreadCompress.Load( f, fileVer ); m_data.externalThreadCompress.Load( f, fileVer ); f.Read( sz ); for( uint64_t i=0; i(); f.Read( srcloc, sizeof( SourceLocationBase ) ); srcloc->namehash = 0; m_data.sourceLocationPayload[i] = srcloc; m_data.sourceLocationPayloadMap.emplace( srcloc, int16_t( i ) ); } #ifndef TRACY_NO_STATISTICS m_data.sourceLocationZones.reserve( sle + sz ); f.Read( sz ); for( uint64_t i=0; isecond.zones.reserve( cnt ); } #else f.Read( sz ); for( uint64_t i=0; i(); auto& lockmap = *lockmapPtr; uint32_t id; uint64_t tsz; f.Read( id ); if( fileVer >= FileVersion( 0, 6, 6 ) ) { f.Read( lockmap.customName ); } f.Read6( lockmap.srcloc, lockmap.type, lockmap.valid, lockmap.timeAnnounce, lockmap.timeTerminate, tsz ); lockmap.isContended = false; lockmap.threadMap.reserve( tsz ); lockmap.threadList.reserve( tsz ); for( uint64_t i=0; i(); const auto lt = ReadTimeOffset( f, refTime ); lev->SetTime( lt ); int16_t srcloc; f.Read( srcloc ); lev->SetSrcLoc( srcloc ); f.Read( &lev->thread, sizeof( LockEvent::thread ) + sizeof( LockEvent::type ) ); *ptr++ = { lev }; UpdateLockRange( lockmap, *lev, lt ); } } else { for( uint64_t i=0; i(); const auto lt = ReadTimeOffset( f, refTime ); lev->SetTime( lt ); int16_t srcloc; f.Read( srcloc ); lev->SetSrcLoc( srcloc ); f.Read( &lev->thread, sizeof( LockEventShared::thread ) + sizeof( LockEventShared::type ) ); *ptr++ = { lev }; UpdateLockRange( lockmap, *lev, lt ); } } UpdateLockCount( lockmap, 0 ); m_data.lockMap.emplace( id, lockmapPtr ); } } else { for( uint64_t i=0; i= FileVersion( 0, 6, 6 ) ) { f.Skip( sizeof( LockMap::customName ) ); } f.Skip( sizeof( uint32_t ) + sizeof( LockMap::srcloc ) ); f.Read( type ); f.Skip( sizeof( LockMap::valid ) + sizeof( LockMap::timeAnnounce ) + sizeof( LockMap::timeTerminate ) ); f.Read( tsz ); f.Skip( tsz * sizeof( uint64_t ) ); f.Read( tsz ); f.Skip( tsz * ( sizeof( int64_t ) + sizeof( int16_t ) + sizeof( LockEvent::thread ) + sizeof( LockEvent::type ) ) ); } } s_loadProgress.subTotal.store( 0, std::memory_order_relaxed ); s_loadProgress.progress.store( LoadProgress::Messages, std::memory_order_relaxed ); unordered_flat_map msgMap; f.Read( sz ); if( eventMask & EventType::Messages ) { m_data.messages.reserve_exact( sz, m_slab ); int64_t refTime = 0; for( uint64_t i=0; i(); msgdata->time = ReadTimeOffset( f, refTime ); f.Read3( msgdata->ref, msgdata->color, msgdata->callstack ); m_data.messages[i] = msgdata; msgMap.emplace( ptr, msgdata ); } } else { f.Skip( sz * ( sizeof( uint64_t ) + sizeof( MessageData::time ) + sizeof( MessageData::ref ) + sizeof( MessageData::color ) + sizeof( MessageData::callstack ) ) ); } if( fileVer >= FileVersion( 0, 6, 3 ) ) { f.Read( sz ); assert( sz != 0 ); m_data.zoneExtra.reserve_exact( sz, m_slab ); f.Read( m_data.zoneExtra.data(), sz * sizeof( ZoneExtra ) ); } else { m_data.zoneExtra.push_back( ZoneExtra {} ); } s_loadProgress.progress.store( LoadProgress::Zones, std::memory_order_relaxed ); f.Read( sz ); s_loadProgress.subTotal.store( sz, std::memory_order_relaxed ); s_loadProgress.subProgress.store( 0, std::memory_order_relaxed ); f.Read( sz ); m_data.zoneChildren.reserve_exact( sz, m_slab ); memset( m_data.zoneChildren.data(), 0, sizeof( Vector> ) * sz ); int32_t childIdx = 0; f.Read( sz ); m_data.threads.reserve_exact( sz, m_slab ); for( uint64_t i=0; i(); uint64_t tid; f.Read2( tid, td->count ); td->id = tid; m_data.zonesCnt += td->count; if( fileVer < FileVersion( 0, 6, 3 ) ) { uint64_t tsz; f.Read( tsz ); if( tsz != 0 ) { int64_t refTime = 0; ReadTimelinePre063( f, td->timeline, tsz, refTime, childIdx, fileVer ); } } else { uint32_t tsz; f.Read( tsz ); if( tsz != 0 ) { ReadTimeline( f, td->timeline, tsz, 0, childIdx ); } } uint64_t msz; f.Read( msz ); if( eventMask & EventType::Messages ) { const auto ctid = CompressThread( tid ); td->messages.reserve_exact( msz, m_slab ); for( uint64_t j=0; jmessages[j] = md; md->thread = ctid; } } else { f.Skip( msz * sizeof( uint64_t ) ); } if( fileVer >= FileVersion( 0, 6, 4 ) ) { uint64_t ssz; f.Read( ssz ); if( ssz != 0 ) { if( eventMask & EventType::Samples ) { m_data.samplesCnt += ssz; int64_t refTime = 0; td->samples.reserve_exact( ssz, m_slab ); auto ptr = td->samples.data(); for( uint64_t j=0; jtime.SetVal( ReadTimeOffset( f, refTime ) ); f.Read( &ptr->callstack, sizeof( ptr->callstack ) ); ptr++; } } else { f.Skip( ssz * ( 8 + 3 ) ); } } } m_data.threads[i] = td; m_threadMap.emplace( tid, td ); } s_loadProgress.progress.store( LoadProgress::GpuZones, std::memory_order_relaxed ); f.Read( sz ); s_loadProgress.subTotal.store( sz, std::memory_order_relaxed ); s_loadProgress.subProgress.store( 0, std::memory_order_relaxed ); f.Read( sz ); m_data.gpuChildren.reserve_exact( sz, m_slab ); memset( m_data.gpuChildren.data(), 0, sizeof( Vector> ) * sz ); childIdx = 0; f.Read( sz ); m_data.gpuData.reserve_exact( sz, m_slab ); for( uint64_t i=0; i(); if( fileVer >= FileVersion( 0, 7, 1 ) ) { uint8_t calibration; f.Read5( ctx->thread, calibration, ctx->count, ctx->period, ctx->type ); ctx->hasCalibration = calibration; } else { uint8_t accuracy; if( fileVer >= FileVersion( 0, 6, 14 ) ) { f.Read5( ctx->thread, accuracy, ctx->count, ctx->period, ctx->type ); } else { f.Read4( ctx->thread, accuracy, ctx->count, ctx->period ); ctx->type = ctx->thread == 0 ? GpuContextType::Vulkan : GpuContextType::OpenGl; } ctx->hasCalibration = false; } ctx->hasPeriod = ctx->period != 1.f; m_data.gpuCnt += ctx->count; uint64_t tdsz; f.Read( tdsz ); for( uint64_t j=0; jthreadData.emplace( tid, GpuCtxThreadData {} ).first; ReadTimeline( f, td->second.timeline, tsz, refTime, refGpuTime, childIdx ); } } m_data.gpuData[i] = ctx; } s_loadProgress.progress.store( LoadProgress::Plots, std::memory_order_relaxed ); f.Read( sz ); if( eventMask & EventType::Plots ) { m_data.plots.Data().reserve( sz ); s_loadProgress.subTotal.store( sz, std::memory_order_relaxed ); for( uint64_t i=0; i(); uint64_t psz; f.Read6( pd->type, pd->format, pd->name, pd->min, pd->max, psz ); pd->data.reserve_exact( psz, m_slab ); auto ptr = pd->data.data(); int64_t refTime = 0; for( uint64_t j=0; jval ); refTime += t; ptr->time = refTime; ptr++; } m_data.plots.Data().push_back_no_space_check( pd ); } } else { for( uint64_t i=0; icsFree, timeAlloc, timeFree, threadAlloc, threadFree ); mem->SetPtr( ptr ); mem->SetSize( size ); mem->SetCsAlloc( csAlloc.Val() ); refTime += timeAlloc; mem->SetTimeThreadAlloc( refTime, threadAlloc ); if( timeFree >= 0 ) { mem->SetTimeThreadFree( timeFree + refTime, threadFree ); frees[fidx++] = i; } else { mem->SetTimeThreadFree( timeFree, threadFree ); active.emplace( ptr, i ); } mem++; } f.Read3( m_data.memory.high, m_data.memory.low, m_data.memory.usage ); if( sz != 0 ) { reconstructMemAllocPlot = true; } } else { f.Skip( 2 * sizeof( uint64_t ) ); f.Skip( sz * ( sizeof( uint64_t ) + sizeof( uint64_t ) + sizeof( Int24 ) + sizeof( Int24 ) + sizeof( int64_t ) * 2 + sizeof( uint16_t ) * 2 ) ); f.Skip( sizeof( MemData::high ) + sizeof( MemData::low ) + sizeof( MemData::usage ) ); } s_loadProgress.subTotal.store( 0, std::memory_order_relaxed ); s_loadProgress.progress.store( LoadProgress::CallStacks, std::memory_order_relaxed ); f.Read( sz ); m_data.callstackPayload.reserve( sz ); if( fileVer >= FileVersion( 0, 6, 8 ) ) { for( uint64_t i=0; i ) + csz * sizeof( CallstackFrameId ); auto mem = (char*)m_slab.AllocRaw( memsize ); auto data = (CallstackFrameId*)mem; f.Read( data, csz * sizeof( CallstackFrameId ) ); auto arr = (VarArray*)( mem + csz * sizeof( CallstackFrameId ) ); new(arr) VarArray( csz, data ); m_data.callstackPayload.push_back_no_space_check( arr ); } } else { for( uint64_t i=0; i ) + csz * sizeof( CallstackFrameId ); auto mem = (char*)m_slab.AllocRaw( memsize ); auto data = (CallstackFrameId*)mem; f.Read( data, csz * sizeof( CallstackFrameId ) ); auto arr = (VarArray*)( mem + csz * sizeof( CallstackFrameId ) ); new(arr) VarArray( csz, data ); m_data.callstackPayload.push_back_no_space_check( arr ); } } if( fileVer >= FileVersion( 0, 6, 5 ) ) { f.Read( sz ); m_data.callstackFrameMap.reserve( sz ); for( uint64_t i=0; i(); f.Read3( id, frameData->size, frameData->imageName ); frameData->data = m_slab.Alloc( frameData->size ); f.Read( frameData->data, sizeof( CallstackFrame ) * frameData->size ); m_data.callstackFrameMap.emplace( id, frameData ); } } else { f.Read( sz ); m_data.callstackFrameMap.reserve( sz ); for( uint64_t i=0; i(); f.Read2( id, frameData->size ); frameData->data = m_slab.Alloc( frameData->size ); for( uint8_t j=0; jsize; j++ ) { f.Read3( frameData->data[j].name, frameData->data[j].file, frameData->data[j].line ); frameData->data[j].symAddr = 0; } m_data.callstackFrameMap.emplace( id, frameData ); } } f.Read( sz ); if( sz > 0 ) { m_data.appInfo.reserve_exact( sz, m_slab ); f.Read( m_data.appInfo.data(), sizeof( m_data.appInfo[0] ) * sz ); } s_loadProgress.subTotal.store( 0, std::memory_order_relaxed ); s_loadProgress.progress.store( LoadProgress::FrameImages, std::memory_order_relaxed ); if( eventMask & EventType::FrameImages ) { f.Read( sz ); m_data.frameImage.reserve_exact( sz, m_slab ); s_loadProgress.subTotal.store( sz, std::memory_order_relaxed ); if( sz != 0 ) { struct JobData { enum State : int { InProgress, Available, DataReady }; FrameImage* fi; char* buf = nullptr; size_t bufsz = 0; char* outbuf = nullptr; size_t outsz = 0; ZSTD_CCtx* ctx = ZSTD_createCCtx(); alignas(64) std::atomic state = Available; }; // Leave one thread for file reader, second thread for dispatch (this thread) // Minimum 2 threads to have at least two buffers (one in use, second one filling up) const auto jobs = std::max( std::thread::hardware_concurrency() - 2, 2 ); auto td = std::make_unique( jobs ); auto data = std::make_unique( jobs ); for( uint64_t i=0; i(); f.Read3( fi->w, fi->h, fi->flip ); const auto sz = size_t( fi->w * fi->h / 2 ); int idx = -1; for(;;) { for( int j=0; jcsz ); memcpy( tmp, data[j].outbuf, data[j].fi->csz ); data[j].fi->ptr = tmp; } idx = j; break; } } if( idx >= 0 ) break; YieldThread(); } if( data[idx].bufsz < sz ) { data[idx].bufsz = sz; delete[] data[idx].buf; data[idx].buf = new char[sz]; } f.Read( data[idx].buf, sz ); data[idx].fi = fi; data[idx].state.store( JobData::InProgress, std::memory_order_release ); td->Queue( [this, &data, idx, fi, fileVer] { if( fileVer <= FileVersion( 0, 6, 9 ) ) m_texcomp.Rdo( data[idx].buf, fi->w * fi->h / 16 ); fi->csz = m_texcomp.Pack( data[idx].ctx, data[idx].outbuf, data[idx].outsz, data[idx].buf, fi->w * fi->h / 2 ); data[idx].state.store( JobData::DataReady, std::memory_order_release ); } ); m_data.frameImage[i] = fi; } td->Sync(); td.reset(); for( int i=0; icsz ); memcpy( tmp, data[i].outbuf, data[i].fi->csz ); data[i].fi->ptr = tmp; } ZSTD_freeCCtx( data[i].ctx ); delete[] data[i].buf; delete[] data[i].outbuf; } const auto& frames = GetFramesBase()->frames; const auto fsz = uint32_t( frames.size() ); for( uint32_t i=0; iframeRef = i; } } } } else { f.Read( sz ); s_loadProgress.subTotal.store( sz, std::memory_order_relaxed ); for( uint64_t i=0; iframes ) { v.frameImage = -1; } } s_loadProgress.subTotal.store( 0, std::memory_order_relaxed ); s_loadProgress.progress.store( LoadProgress::ContextSwitches, std::memory_order_relaxed ); if( eventMask & EventType::ContextSwitches ) { f.Read( sz ); s_loadProgress.subTotal.store( sz, std::memory_order_relaxed ); m_data.ctxSwitch.reserve( sz ); for( uint64_t i=0; i(); data->v.reserve_exact( csz, m_slab ); int64_t runningTime = 0; int64_t refTime = 0; auto ptr = data->v.data(); for( uint64_t j=0; jSetWakeup( refTime ); refTime += deltaStart; ptr->SetStartCpu( refTime, cpu ); if( diff > 0 ) runningTime += diff; refTime += diff; ptr->SetEndReasonState( refTime, reason, state ); ptr++; } data->runningTime = runningTime; m_data.ctxSwitch.emplace( thread, data ); } } else { f.Read( sz ); s_loadProgress.subTotal.store( sz, std::memory_order_relaxed ); for( uint64_t i=0; iSetStartThread( refTime, thread ); refTime += deltaEnd; ptr->SetEnd( refTime ); ptr++; } cnt += sz; } s_loadProgress.subProgress.store( cnt, std::memory_order_relaxed ); } } else { for( int i=0; i<256; i++ ) { f.Read( sz ); f.Skip( sz * ( sizeof( int64_t ) * 2 + sizeof( uint16_t ) ) ); } } f.Read( sz ); for( uint64_t i=0; i= FileVersion( 0, 6, 7 ) ) { if( fileVer >= FileVersion( 0, 6, 11 ) ) { f.Read( sz ); m_data.symbolLoc.reserve_exact( sz, m_slab ); f.Read( sz ); m_data.symbolLocInline.reserve_exact( sz, m_slab ); f.Read( sz ); m_data.symbolMap.reserve( sz ); int symIdx = 0; int symInlineIdx = 0; for( uint64_t i=0; i= FileVersion( 0, 6, 5 ) ) { f.Read( sz ); m_data.symbolMap.reserve( sz ); for( uint64_t i=0; i= FileVersion( 0, 6, 7 ) ) { f.Read( sz ); if( eventMask & EventType::SymbolCode ) { uint64_t ssz = 0; m_data.symbolCode.reserve( sz ); for( uint64_t i=0; i= FileVersion( 0, 6, 9 ) ) { f.Read( sz ); if( eventMask & EventType::SymbolCode ) { m_data.locationCodeAddressList.reserve( sz ); for( uint64_t i=0; i data; data.reserve_exact( lsz, m_slab ); uint64_t ref = 0; for( uint16_t j=0; j= FileVersion( 0, 6, 13 ) ) { f.Read( sz ); if( eventMask & EventType::SourceCache ) { m_data.sourceFileCache.reserve( sz ); for( uint64_t i=0; i( len+1 ); f.Read( key, len ); key[len] = '\0'; f.Read( len ); auto data = (char*)m_slab.AllocBig( len ); f.Read( data, len ); m_data.sourceFileCache.emplace( key, MemoryBlock { data, len } ); } } else { for( uint64_t i=0; i( std::chrono::high_resolution_clock::now() - loadStart ).count(); if( !bgTasks ) { m_backgroundDone.store( true, std::memory_order_relaxed ); } else { m_backgroundDone.store( false, std::memory_order_relaxed ); #ifndef TRACY_NO_STATISTICS m_threadBackground = std::thread( [this, reconstructMemAllocPlot, eventMask] { std::vector jobs; if( !m_data.ctxSwitch.empty() ) { jobs.emplace_back( std::thread( [this] { ReconstructContextSwitchUsage(); } ) ); } if( reconstructMemAllocPlot ) { jobs.emplace_back( std::thread( [this] { ReconstructMemAllocPlot(); } ) ); } std::function>&, uint16_t)> ProcessTimeline; ProcessTimeline = [this, &ProcessTimeline] ( Vector>& _vec, uint16_t thread ) { if( m_shutdown.load( std::memory_order_relaxed ) ) return; assert( _vec.is_magic() ); auto& vec = *(Vector*)( &_vec ); for( auto& zone : vec ) { if( zone.IsEndValid() ) ReconstructZoneStatistics( zone, thread ); if( zone.HasChildren() ) ProcessTimeline( GetZoneChildrenMutable( zone.Child() ), thread ); } }; jobs.emplace_back( std::thread( [this, ProcessTimeline] { for( auto& t : m_data.threads ) { if( m_shutdown.load( std::memory_order_relaxed ) ) return; if( !t->timeline.empty() ) { // Don't touch thread compression cache in a thread. ProcessTimeline( t->timeline, m_data.localThreadCompress.DecompressMustRaw( t->id ) ); } } } ) ); if( eventMask & EventType::Samples ) { jobs.emplace_back( std::thread( [this] { unordered_flat_map counts; uint32_t total = 0; for( auto& t : m_data.threads ) total += t->samples.size(); if( total != 0 ) { for( auto& t : m_data.threads ) { if( m_shutdown.load( std::memory_order_relaxed ) ) return; for( auto& sd : t->samples ) { const auto cs = sd.callstack.Val(); auto it = counts.find( cs ); if( it == counts.end() ) { counts.emplace( cs, 1 ); } else { it->second++; } const auto& callstack = GetCallstack( cs ); auto& ip = callstack[0]; auto frame = GetCallstackFrame( ip ); if( frame ) { const auto symAddr = frame->data[0].symAddr; auto it = m_data.instructionPointersMap.find( symAddr ); if( it == m_data.instructionPointersMap.end() ) { m_data.instructionPointersMap.emplace( symAddr, unordered_flat_map { { ip, 1 } } ); } else { auto fit = it->second.find( ip ); if( fit == it->second.end() ) { it->second.emplace( ip, 1 ); } else { fit->second++; } } } } } for( auto& v : counts ) UpdateSampleStatistics( v.first, v.second, false ); } std::lock_guard lock( m_data.lock ); m_data.callstackSamplesReady = true; } ) ); jobs.emplace_back( std::thread( [this] { uint32_t gcnt = 0; for( auto& t : m_data.threads ) { if( m_shutdown.load( std::memory_order_relaxed ) ) return; for( auto& sd : t->samples ) { gcnt += AddGhostZone( GetCallstack( sd.callstack.Val() ), &t->ghostZones, sd.time.Val() ); } } std::lock_guard lock( m_data.lock ); m_data.ghostZonesReady = true; m_data.ghostCnt = gcnt; } ) ); jobs.emplace_back( std::thread( [this] { for( auto& t : m_data.threads ) { for( auto& v : t->samples ) { const auto& time = v.time; const auto cs = v.callstack.Val(); const auto& callstack = GetCallstack( cs ); auto& ip = callstack[0]; auto frame = GetCallstackFrame( ip ); if( frame ) { const auto symAddr = frame->data[0].symAddr; auto it = m_data.symbolSamples.find( symAddr ); if( it == m_data.symbolSamples.end() ) { m_data.symbolSamples.emplace( symAddr, Vector( SampleDataRange { time, ip } ) ); } else { it->second.push_back_non_empty( SampleDataRange { time, ip } ); } } } } for( auto& v : m_data.symbolSamples ) { pdqsort_branchless( v.second.begin(), v.second.end(), []( const auto& lhs, const auto& rhs ) { return lhs.time.Val() < rhs.time.Val(); } ); } std::lock_guard lock( m_data.lock ); m_data.symbolSamplesReady = true; } ) ); } for( auto& job : jobs ) job.join(); for( auto& v : m_data.sourceLocationZones ) { if( m_shutdown.load( std::memory_order_relaxed ) ) return; auto& zones = v.second.zones; #ifdef NO_PARALLEL_SORT pdqsort_branchless( zones.begin(), zones.end(), []( const auto& lhs, const auto& rhs ) { return lhs.Zone()->Start() < rhs.Zone()->Start(); } ); #else std::sort( std::execution::par_unseq, zones.begin(), zones.end(), []( const auto& lhs, const auto& rhs ) { return lhs.Zone()->Start() < rhs.Zone()->Start(); } ); #endif } { std::lock_guard lock( m_data.lock ); m_data.sourceLocationZonesReady = true; } m_backgroundDone.store( true, std::memory_order_relaxed ); } ); #else m_backgroundDone.store( true, std::memory_order_relaxed ); #endif } } Worker::~Worker() { Shutdown(); if( m_threadNet.joinable() ) m_threadNet.join(); if( m_thread.joinable() ) m_thread.join(); if( m_threadBackground.joinable() ) m_threadBackground.join(); delete[] m_buffer; LZ4_freeStreamDecode( (LZ4_streamDecode_t*)m_stream ); delete[] m_frameImageBuffer; delete[] m_tmpBuf; for( auto& v : m_data.threads ) { v->timeline.~Vector(); v->stack.~Vector(); v->messages.~Vector(); v->zoneIdStack.~Vector(); v->samples.~Vector(); #ifndef TRACY_NO_STATISTICS v->childTimeStack.~Vector(); v->ghostZones.~Vector(); #endif } for( auto& v : m_data.gpuData ) { for( auto& vt : v->threadData ) { vt.second.timeline.~Vector(); vt.second.stack.~Vector(); } } for( auto& v : m_data.plots.Data() ) { v->~PlotData(); } for( auto& v : m_data.frames.Data() ) { v->~FrameData(); } for( auto& v : m_data.lockMap ) { v.second->~LockMap(); } for( auto& v : m_data.zoneChildren ) { v.~Vector(); } for( auto& v : m_data.ctxSwitch ) { v.second->v.~Vector(); } for( auto& v : m_data.gpuChildren ) { v.~Vector(); } #ifndef TRACY_NO_STATISTICS for( auto& v : m_data.ghostChildren ) { v.~Vector(); } #endif } uint64_t Worker::GetLockCount() const { uint64_t cnt = 0; for( auto& l : m_data.lockMap ) { cnt += l.second->timeline.size(); } return cnt; } uint64_t Worker::GetPlotCount() const { uint64_t cnt = 0; for( auto& p : m_data.plots.Data() ) { if( p->type == PlotType::User ) { cnt += p->data.size(); } } return cnt; } uint64_t Worker::GetTracyPlotCount() const { uint64_t cnt = 0; for( auto& p : m_data.plots.Data() ) { if( p->type != PlotType::User ) { cnt += p->data.size(); } } return cnt; } uint64_t Worker::GetContextSwitchCount() const { uint64_t cnt = 0; for( auto& v : m_data.ctxSwitch ) { cnt += v.second->v.size(); } return cnt; } uint64_t Worker::GetContextSwitchPerCpuCount() const { uint64_t cnt = 0; for( int i=0; isecond; } void Worker::GetCpuUsageAtTime( int64_t time, int& own, int& other ) const { own = other = 0; if( time < 0 || time > m_data.lastTime ) return; #ifndef TRACY_NO_STATISTICS // Remove this check when real-time ctxUsage contruction is implemented. if( !m_data.ctxUsage.empty() ) { auto it = std::upper_bound( m_data.ctxUsage.begin(), m_data.ctxUsage.end(), time, [] ( const auto& l, const auto& r ) { return l < r.Time(); } ); if( it == m_data.ctxUsage.begin() || it == m_data.ctxUsage.end() ) return; --it; own = it->Own(); other = it->Other(); return; } #endif int cntOwn = 0; int cntOther = 0; for( int i=0; iIsEndValid() && it->Start() <= time ) { if( GetPidFromTid( DecompressThreadExternal( it->Thread() ) ) == m_pid ) { cntOwn++; } else { cntOther++; } } } } own = cntOwn; other = cntOther; } const ContextSwitch* const Worker::GetContextSwitchDataImpl( uint64_t thread ) { auto it = m_data.ctxSwitch.find( thread ); if( it != m_data.ctxSwitch.end() ) { m_data.ctxSwitchLast.first = thread; m_data.ctxSwitchLast.second = it->second; return it->second; } else { return nullptr; } } size_t Worker::GetFullFrameCount( const FrameData& fd ) const { const auto sz = fd.frames.size(); assert( sz != 0 ); if( fd.continuous ) { if( IsConnected() ) { return sz - 1; } else { return sz; } } else { const auto& last = fd.frames.back(); if( last.end >= 0 ) { return sz; } else { return sz - 1; } } } int64_t Worker::GetFrameTime( const FrameData& fd, size_t idx ) const { if( fd.continuous ) { if( idx < fd.frames.size() - 1 ) { return fd.frames[idx+1].start - fd.frames[idx].start; } else { assert( m_data.lastTime != 0 ); return m_data.lastTime - fd.frames.back().start; } } else { const auto& frame = fd.frames[idx]; if( frame.end >= 0 ) { return frame.end - frame.start; } else { return m_data.lastTime - fd.frames.back().start; } } } int64_t Worker::GetFrameBegin( const FrameData& fd, size_t idx ) const { assert( idx < fd.frames.size() ); return fd.frames[idx].start; } int64_t Worker::GetFrameEnd( const FrameData& fd, size_t idx ) const { if( fd.continuous ) { if( idx < fd.frames.size() - 1 ) { return fd.frames[idx+1].start; } else { return m_data.lastTime; } } else { if( fd.frames[idx].end >= 0 ) { return fd.frames[idx].end; } else { return m_data.lastTime; } } } const FrameImage* Worker::GetFrameImage( const FrameData& fd, size_t idx ) const { assert( idx < fd.frames.size() ); const auto& v = fd.frames[idx].frameImage; if( v < 0 ) return nullptr; return m_data.frameImage[v]; } std::pair Worker::GetFrameRange( const FrameData& fd, int64_t from, int64_t to ) { auto zitbegin = std::lower_bound( fd.frames.begin(), fd.frames.end(), from, [] ( const auto& lhs, const auto& rhs ) { return lhs.start < rhs; } ); if( zitbegin == fd.frames.end() ) zitbegin--; const auto zitend = std::lower_bound( zitbegin, fd.frames.end(), to, [] ( const auto& lhs, const auto& rhs ) { return lhs.start < rhs; } ); int zbegin = std::distance( fd.frames.begin(), zitbegin ); if( zbegin > 0 && zitbegin->start != from ) --zbegin; const int zend = std::distance( fd.frames.begin(), zitend ); return std::make_pair( zbegin, zend ); } const CallstackFrameData* Worker::GetCallstackFrame( const CallstackFrameId& ptr ) const { assert( ptr.custom == 0 ); auto it = m_data.callstackFrameMap.find( ptr ); if( it == m_data.callstackFrameMap.end() ) { return nullptr; } else { return it->second; } } #ifndef TRACY_NO_STATISTICS const CallstackFrameData* Worker::GetParentCallstackFrame( const CallstackFrameId& ptr ) const { assert( ptr.custom == 1 ); auto it = m_data.parentCallstackFrameMap.find( ptr ); if( it == m_data.parentCallstackFrameMap.end() ) { return nullptr; } else { return it->second; } } const Vector* Worker::GetSamplesForSymbol( uint64_t symAddr ) const { assert( m_data.symbolSamplesReady ); auto it = m_data.symbolSamples.find( symAddr ); if( it == m_data.symbolSamples.end() ) return nullptr; return &it->second; } #endif const SymbolData* Worker::GetSymbolData( uint64_t sym ) const { auto it = m_data.symbolMap.find( sym ); if( it == m_data.symbolMap.end() ) { return nullptr; } else { return &it->second; } } bool Worker::HasSymbolCode( uint64_t sym ) const { return m_data.symbolCode.find( sym ) != m_data.symbolCode.end(); } const char* Worker::GetSymbolCode( uint64_t sym, uint32_t& len ) const { auto it = m_data.symbolCode.find( sym ); if( it == m_data.symbolCode.end() ) return nullptr; len = it->second.len; return it->second.data; } uint64_t Worker::GetSymbolForAddress( uint64_t address ) const { auto it = std::lower_bound( m_data.symbolLoc.begin(), m_data.symbolLoc.end(), address, [] ( const auto& l, const auto& r ) { return l.addr + l.len < r; } ); if( it == m_data.symbolLoc.end() || address < it->addr ) return 0; return it->addr; } uint64_t Worker::GetSymbolForAddress( uint64_t address, uint32_t& offset ) const { auto it = std::lower_bound( m_data.symbolLoc.begin(), m_data.symbolLoc.end(), address, [] ( const auto& l, const auto& r ) { return l.addr + l.len < r; } ); if( it == m_data.symbolLoc.end() || address < it->addr ) return 0; offset = address - it->addr; return it->addr; } StringIdx Worker::GetLocationForAddress( uint64_t address, uint32_t& line ) const { auto it = m_data.codeAddressToLocation.find( address ); if( it == m_data.codeAddressToLocation.end() ) { line = 0; return StringIdx(); } else { const auto idx = UnpackFileLine( it->second, line ); return StringIdx( idx ); } } const Vector* Worker::GetAddressesForLocation( uint32_t fileStringIdx, uint32_t line ) const { auto it = m_data.locationCodeAddressList.find( PackFileLine( fileStringIdx, line ) ); if( it == m_data.locationCodeAddressList.end() ) { return nullptr; } else { return &it->second; } } const uint64_t* Worker::GetInlineSymbolList( uint64_t sym, uint32_t len ) const { auto it = std::lower_bound( m_data.symbolLocInline.begin(), m_data.symbolLocInline.end(), sym ); if( it == m_data.symbolLocInline.end() ) return nullptr; if( *it >= sym + len ) return nullptr; return it; } int64_t Worker::GetZoneEnd( const ZoneEvent& ev ) { auto ptr = &ev; for(;;) { if( ptr->IsEndValid() ) return ptr->End(); if( !ptr->HasChildren() ) return ptr->Start(); auto& children = GetZoneChildren( ptr->Child() ); if( children.is_magic() ) { auto& c = *(Vector*)&children; ptr = &c.back(); } else { ptr = children.back(); } } } int64_t Worker::GetZoneEnd( const GpuEvent& ev ) { auto ptr = &ev; for(;;) { if( ptr->GpuEnd() >= 0 ) return ptr->GpuEnd(); if( ptr->Child() < 0 ) return ptr->GpuStart() >= 0 ? ptr->GpuStart() : m_data.lastTime; auto& children = GetGpuChildren( ptr->Child() ); if( children.is_magic() ) { auto& c = *(Vector*)&children; ptr = &c.back(); } else { ptr = children.back(); } } } uint32_t Worker::FindStringIdx( const char* str ) const { if( !str ) return 0; charutil::StringKey key = { str, strlen( str ) }; auto sit = m_data.stringMap.find( key ); if( sit == m_data.stringMap.end() ) { return 0; } else { return sit->second; } } const char* Worker::GetString( uint64_t ptr ) const { const auto it = m_data.strings.find( ptr ); if( it == m_data.strings.end() || it->second == nullptr ) { return "???"; } else { return it->second; } } const char* Worker::GetString( const StringRef& ref ) const { if( ref.isidx ) { assert( ref.active ); return m_data.stringData[ref.str]; } else { if( ref.active ) { return GetString( ref.str ); } else { return "???"; } } } const char* Worker::GetString( const StringIdx& idx ) const { assert( idx.Active() ); return m_data.stringData[idx.Idx()]; } static const char* BadExternalThreadNames[] = { "ntdll.dll", "???", nullptr }; const char* Worker::GetThreadName( uint64_t id ) const { const auto it = m_data.threadNames.find( id ); if( it == m_data.threadNames.end() ) { const auto eit = m_data.externalNames.find( id ); if( eit == m_data.externalNames.end() ) { return "???"; } else { return eit->second.second; } } else { // Client should send additional information about thread name, to make this check unnecessary const auto txt = it->second; if( txt[0] >= '0' && txt[0] <= '9' && (uint64_t)atoi( txt ) == id ) { const auto eit = m_data.externalNames.find( id ); if( eit != m_data.externalNames.end() ) { const char* ext = eit->second.second; const char** ptr = BadExternalThreadNames; while( *ptr ) { if( strcmp( *ptr, ext ) == 0 ) return txt; ptr++; } return ext; } } return txt; } } bool Worker::IsThreadLocal( uint64_t id ) { auto td = RetrieveThread( id ); return td && ( td->count > 0 || !td->samples.empty() ); } const SourceLocation& Worker::GetSourceLocation( int16_t srcloc ) const { if( srcloc < 0 ) { return *m_data.sourceLocationPayload[-srcloc-1]; } else { const auto it = m_data.sourceLocation.find( m_data.sourceLocationExpand[srcloc] ); assert( it != m_data.sourceLocation.end() ); return it->second; } } std::pair Worker::GetExternalName( uint64_t id ) const { const auto it = m_data.externalNames.find( id ); if( it == m_data.externalNames.end() ) { return std::make_pair( "???", "???" ); } else { return it->second; } } const char* Worker::GetZoneName( const SourceLocation& srcloc ) const { if( srcloc.name.active ) { return GetString( srcloc.name ); } else { return GetString( srcloc.function ); } } const char* Worker::GetZoneName( const ZoneEvent& ev ) const { auto& srcloc = GetSourceLocation( ev.SrcLoc() ); return GetZoneName( ev, srcloc ); } const char* Worker::GetZoneName( const ZoneEvent& ev, const SourceLocation& srcloc ) const { if( HasZoneExtra( ev ) && GetZoneExtra( ev ).name.Active() ) { return GetString( GetZoneExtra( ev ).name ); } else if( srcloc.name.active ) { return GetString( srcloc.name ); } else { return GetString( srcloc.function ); } } const char* Worker::GetZoneName( const GpuEvent& ev ) const { auto& srcloc = GetSourceLocation( ev.SrcLoc() ); return GetZoneName( ev, srcloc ); } const char* Worker::GetZoneName( const GpuEvent& ev, const SourceLocation& srcloc ) const { if( srcloc.name.active ) { return GetString( srcloc.name ); } else { return GetString( srcloc.function ); } } static bool strstr_nocase( const char* l, const char* r ) { const auto lsz = strlen( l ); const auto rsz = strlen( r ); auto ll = (char*)alloca( lsz + 1 ); auto rl = (char*)alloca( rsz + 1 ); for( size_t i=0; i Worker::GetMatchingSourceLocation( const char* query, bool ignoreCase ) const { std::vector match; const auto sz = m_data.sourceLocationExpand.size(); for( size_t i=1; isecond; const auto str = GetString( srcloc.name.active ? srcloc.name : srcloc.function ); bool found = false; if( ignoreCase ) { found = strstr_nocase( str, query ); } else { found = strstr( str, query ) != nullptr; } if( found ) { match.push_back( (int16_t)i ); } } for( auto& srcloc : m_data.sourceLocationPayload ) { const auto str = GetString( srcloc->name.active ? srcloc->name : srcloc->function ); bool found = false; if( ignoreCase ) { found = strstr_nocase( str, query ); } else { found = strstr( str, query ) != nullptr; } if( found ) { auto it = m_data.sourceLocationPayloadMap.find( (const SourceLocation*)srcloc ); assert( it != m_data.sourceLocationPayloadMap.end() ); match.push_back( -int16_t( it->second + 1 ) ); } } return match; } #ifndef TRACY_NO_STATISTICS const Worker::SourceLocationZones& Worker::GetZonesForSourceLocation( int16_t srcloc ) const { assert( AreSourceLocationZonesReady() ); static const SourceLocationZones empty; auto it = m_data.sourceLocationZones.find( srcloc ); return it != m_data.sourceLocationZones.end() ? it->second : empty; } const SymbolStats* Worker::GetSymbolStats( uint64_t symAddr ) const { assert( AreCallstackSamplesReady() ); auto it = m_data.symbolStats.find( symAddr ); if( it == m_data.symbolStats.end() ) { return nullptr; } else { return &it->second; } } const unordered_flat_map* Worker::GetSymbolInstructionPointers( uint64_t symAddr ) const { assert( AreCallstackSamplesReady() ); auto it = m_data.instructionPointersMap.find( symAddr ); if( it == m_data.instructionPointersMap.end() ) { return nullptr; } else { return &it->second; } } #endif void Worker::Network() { auto ShouldExit = [this] { return m_shutdown.load( std::memory_order_relaxed ); }; auto lz4buf = std::make_unique( LZ4Size ); for(;;) { { std::unique_lock lock( m_netWriteLock ); m_netWriteCv.wait( lock, [this] { return m_netWriteCnt > 0 || m_shutdown.load( std::memory_order_relaxed ); } ); if( m_shutdown.load( std::memory_order_relaxed ) ) goto close; m_netWriteCnt--; } auto buf = m_buffer + m_bufferOffset; lz4sz_t lz4sz; if( !m_sock.Read( &lz4sz, sizeof( lz4sz ), 10, ShouldExit ) ) goto close; if( !m_sock.Read( lz4buf.get(), lz4sz, 10, ShouldExit ) ) goto close; auto bb = m_bytes.load( std::memory_order_relaxed ); m_bytes.store( bb + sizeof( lz4sz ) + lz4sz, std::memory_order_relaxed ); auto sz = LZ4_decompress_safe_continue( (LZ4_streamDecode_t*)m_stream, lz4buf.get(), buf, lz4sz, TargetFrameSize ); assert( sz >= 0 ); bb = m_decBytes.load( std::memory_order_relaxed ); m_decBytes.store( bb + sz, std::memory_order_relaxed ); { std::lock_guard lock( m_netReadLock ); m_netRead.push_back( NetBuffer { m_bufferOffset, sz } ); m_netReadCv.notify_one(); } m_bufferOffset += sz; if( m_bufferOffset > TargetFrameSize * 2 ) m_bufferOffset = 0; } close: std::lock_guard lock( m_netReadLock ); m_netRead.push_back( NetBuffer { -1 } ); m_netReadCv.notify_one(); } void Worker::Exec() { auto ShouldExit = [this] { return m_shutdown.load( std::memory_order_relaxed ); }; for(;;) { if( m_shutdown.load( std::memory_order_relaxed ) ) { m_netWriteCv.notify_one(); return; }; if( m_sock.Connect( m_addr.c_str(), m_port ) ) break; std::this_thread::sleep_for( std::chrono::milliseconds( 10 ) ); } std::chrono::time_point t0; m_sock.Send( HandshakeShibboleth, HandshakeShibbolethSize ); uint32_t protocolVersion = ProtocolVersion; m_sock.Send( &protocolVersion, sizeof( protocolVersion ) ); HandshakeStatus handshake; if( !m_sock.Read( &handshake, sizeof( handshake ), 10, ShouldExit ) ) { m_handshake.store( HandshakeDropped, std::memory_order_relaxed ); goto close; } m_handshake.store( handshake, std::memory_order_relaxed ); switch( handshake ) { case HandshakeWelcome: break; case HandshakeProtocolMismatch: case HandshakeNotAvailable: default: goto close; } m_data.framesBase = m_data.frames.Retrieve( 0, [this] ( uint64_t name ) { auto fd = m_slab.AllocInit(); fd->name = name; fd->continuous = 1; return fd; }, [this] ( uint64_t name ) { assert( name == 0 ); char tmp[6] = "Frame"; HandleFrameName( name, tmp, 5 ); } ); { WelcomeMessage welcome; if( !m_sock.Read( &welcome, sizeof( welcome ), 10, ShouldExit ) ) { m_handshake.store( HandshakeDropped, std::memory_order_relaxed ); goto close; } m_timerMul = welcome.timerMul; m_data.baseTime = welcome.initBegin; const auto initEnd = TscTime( welcome.initEnd - m_data.baseTime ); m_data.framesBase->frames.push_back( FrameEvent{ 0, -1, -1 } ); m_data.framesBase->frames.push_back( FrameEvent{ initEnd, -1, -1 } ); m_data.lastTime = initEnd; m_delay = TscTime( welcome.delay ); m_resolution = TscTime( welcome.resolution ); m_pid = welcome.pid; m_samplingPeriod = welcome.samplingPeriod; m_onDemand = welcome.onDemand; m_captureProgram = welcome.programName; m_captureTime = welcome.epoch; m_ignoreMemFreeFaults = welcome.onDemand || welcome.isApple; m_data.cpuArch = (CpuArchitecture)welcome.cpuArch; m_data.cpuId = welcome.cpuId; memcpy( m_data.cpuManufacturer, welcome.cpuManufacturer, 12 ); m_data.cpuManufacturer[12] = '\0'; char dtmp[64]; time_t date = welcome.epoch; auto lt = localtime( &date ); strftime( dtmp, 64, "%F %T", lt ); char tmp[1024]; sprintf( tmp, "%s @ %s", welcome.programName, dtmp ); m_captureName = tmp; m_hostInfo = welcome.hostInfo; if( welcome.onDemand != 0 ) { OnDemandPayloadMessage onDemand; if( !m_sock.Read( &onDemand, sizeof( onDemand ), 10, ShouldExit ) ) { m_handshake.store( HandshakeDropped, std::memory_order_relaxed ); goto close; } m_data.frameOffset = onDemand.frames; m_data.framesBase->frames.push_back( FrameEvent{ TscTime( onDemand.currentTime - m_data.baseTime ), -1, -1 } ); } } m_serverQuerySpaceBase = m_serverQuerySpaceLeft = ( m_sock.GetSendBufSize() / ServerQueryPacketSize ) - ServerQueryPacketSize; // leave space for terminate request m_hasData.store( true, std::memory_order_release ); LZ4_setStreamDecode( (LZ4_streamDecode_t*)m_stream, nullptr, 0 ); m_connected.store( true, std::memory_order_relaxed ); { std::lock_guard lock( m_netWriteLock ); m_netWriteCnt = 2; m_netWriteCv.notify_one(); } t0 = std::chrono::high_resolution_clock::now(); for(;;) { if( m_shutdown.load( std::memory_order_relaxed ) ) { QueryTerminate(); goto close; } NetBuffer netbuf; { std::unique_lock lock( m_netReadLock ); m_netReadCv.wait( lock, [this] { return !m_netRead.empty(); } ); netbuf = m_netRead.front(); m_netRead.erase( m_netRead.begin() ); } if( netbuf.bufferOffset < 0 ) goto close; const char* ptr = m_buffer + netbuf.bufferOffset; const char* end = ptr + netbuf.size; { std::lock_guard lock( m_data.lock ); while( ptr < end ) { auto ev = (const QueueItem*)ptr; if( !DispatchProcess( *ev, ptr ) ) { if( m_failure != Failure::None ) HandleFailure( ptr, end ); QueryTerminate(); goto close; } } { std::lock_guard lock( m_netWriteLock ); m_netWriteCnt++; m_netWriteCv.notify_one(); } HandlePostponedPlots(); #ifndef TRACY_NO_STATISTICS if( m_data.newFramesWereReceived ) { HandlePostponedSamples(); HandlePostponedGhostZones(); m_data.newFramesWereReceived = false; } #endif if( m_data.newSymbolsWereAdded ) { m_data.newSymbolsWereAdded = false; #ifdef NO_PARALLEL_SORT pdqsort_branchless( m_data.symbolLoc.begin(), m_data.symbolLoc.end(), [] ( const auto& l, const auto& r ) { return l.addr < r.addr; } ); #else std::sort( std::execution::par_unseq, m_data.symbolLoc.begin(), m_data.symbolLoc.end(), [] ( const auto& l, const auto& r ) { return l.addr < r.addr; } ); #endif } if( m_data.newInlineSymbolsWereAdded ) { m_data.newInlineSymbolsWereAdded = false; #ifdef NO_PARALLEL_SORT pdqsort_branchless( m_data.symbolLocInline.begin(), m_data.symbolLocInline.end() ); #else std::sort( std::execution::par_unseq, m_data.symbolLocInline.begin(), m_data.symbolLocInline.end() ); #endif } if( !m_serverQueryQueue.empty() && m_serverQuerySpaceLeft > 0 ) { const auto toSend = std::min( m_serverQuerySpaceLeft, m_serverQueryQueue.size() ); m_sock.Send( m_serverQueryQueue.data(), toSend * ServerQueryPacketSize ); m_serverQuerySpaceLeft -= toSend; if( toSend == m_serverQueryQueue.size() ) { m_serverQueryQueue.clear(); } else { m_serverQueryQueue.erase( m_serverQueryQueue.begin(), m_serverQueryQueue.begin() + toSend ); } } } auto t1 = std::chrono::high_resolution_clock::now(); auto td = std::chrono::duration_cast( t1 - t0 ).count(); enum { MbpsUpdateTime = 200 }; if( td > MbpsUpdateTime ) { UpdateMbps( td ); t0 = t1; } if( m_terminate ) { if( m_pendingStrings != 0 || m_pendingThreads != 0 || m_pendingSourceLocation != 0 || m_pendingCallstackFrames != 0 || m_data.plots.IsPending() || m_pendingCallstackPtr != 0 || m_pendingExternalNames != 0 || m_pendingCallstackSubframes != 0 || m_pendingFrameImageData.image != nullptr || !m_pendingSymbols.empty() || !m_pendingSymbolCode.empty() || m_pendingCodeInformation != 0 || !m_serverQueryQueue.empty() || m_pendingSourceLocationPayload != 0 || m_pendingSingleString.ptr != nullptr || m_pendingSecondString.ptr != nullptr ) { continue; } if( !m_crashed && !m_disconnect ) { bool done = true; for( auto& v : m_data.threads ) { if( !v->stack.empty() ) { done = false; break; } } if( !done ) continue; } QueryTerminate(); UpdateMbps( 0 ); break; } } close: Shutdown(); m_netWriteCv.notify_one(); m_sock.Close(); m_connected.store( false, std::memory_order_relaxed ); } void Worker::UpdateMbps( int64_t td ) { const auto bytes = m_bytes.exchange( 0, std::memory_order_relaxed ); const auto decBytes = m_decBytes.exchange( 0, std::memory_order_relaxed ); std::lock_guard lock( m_mbpsData.lock ); if( td != 0 ) { m_mbpsData.mbps.erase( m_mbpsData.mbps.begin() ); m_mbpsData.mbps.emplace_back( bytes / ( td * 125.f ) ); } m_mbpsData.compRatio = decBytes == 0 ? 1 : float( bytes ) / decBytes; m_mbpsData.queue = m_serverQueryQueue.size(); m_mbpsData.transferred += bytes; } bool Worker::IsThreadStringRetrieved( uint64_t id ) { const auto name = GetThreadName( m_failureData.thread ); return strcmp( name, "???" ) != 0; } bool Worker::IsSourceLocationRetrieved( int16_t srcloc ) { auto& sl = GetSourceLocation( srcloc ); auto func = GetString( sl.function ); auto file = GetString( sl.file ); return strcmp( func, "???" ) != 0 && strcmp( file, "???" ) != 0; } bool Worker::HasAllFailureData() { if( m_failureData.thread != 0 && !IsThreadStringRetrieved( m_failureData.thread ) ) return false; if( m_failureData.srcloc != 0 && !IsSourceLocationRetrieved( m_failureData.srcloc ) ) return false; return true; } void Worker::HandleFailure( const char* ptr, const char* end ) { if( HasAllFailureData() ) return; for(;;) { while( ptr < end ) { auto ev = (const QueueItem*)ptr; DispatchFailure( *ev, ptr ); } if( HasAllFailureData() ) return; { std::lock_guard lock( m_netWriteLock ); m_netWriteCnt++; m_netWriteCv.notify_one(); } if( !m_serverQueryQueue.empty() && m_serverQuerySpaceLeft > 0 ) { const auto toSend = std::min( m_serverQuerySpaceLeft, m_serverQueryQueue.size() ); m_sock.Send( m_serverQueryQueue.data(), toSend * ServerQueryPacketSize ); m_serverQuerySpaceLeft -= toSend; if( toSend == m_serverQueryQueue.size() ) { m_serverQueryQueue.clear(); } else { m_serverQueryQueue.erase( m_serverQueryQueue.begin(), m_serverQueryQueue.begin() + toSend ); } } if( m_shutdown.load( std::memory_order_relaxed ) ) return; NetBuffer netbuf; { std::unique_lock lock( m_netReadLock ); m_netReadCv.wait( lock, [this] { return !m_netRead.empty(); } ); netbuf = m_netRead.front(); m_netRead.erase( m_netRead.begin() ); } if( netbuf.bufferOffset < 0 ) return; ptr = m_buffer + netbuf.bufferOffset; end = ptr + netbuf.size; } } void Worker::DispatchFailure( const QueueItem& ev, const char*& ptr ) { if( ev.hdr.idx >= (int)QueueType::StringData ) { ptr += sizeof( QueueHeader ) + sizeof( QueueStringTransfer ); if( ev.hdr.type == QueueType::FrameImageData || ev.hdr.type == QueueType::SymbolCode ) { uint32_t sz; memcpy( &sz, ptr, sizeof( sz ) ); ptr += sizeof( sz ) + sz; } else { uint16_t sz; memcpy( &sz, ptr, sizeof( sz ) ); ptr += sizeof( sz ); switch( ev.hdr.type ) { case QueueType::StringData: AddString( ev.stringTransfer.ptr, ptr, sz ); m_serverQuerySpaceLeft++; break; case QueueType::ThreadName: AddThreadString( ev.stringTransfer.ptr, ptr, sz ); m_serverQuerySpaceLeft++; break; case QueueType::PlotName: case QueueType::FrameName: case QueueType::ExternalName: m_serverQuerySpaceLeft++; break; default: break; } ptr += sz; } } else { ptr += QueueDataSize[ev.hdr.idx]; switch( ev.hdr.type ) { case QueueType::SourceLocation: AddSourceLocation( ev.srcloc ); m_serverQuerySpaceLeft++; break; case QueueType::CallstackFrameSize: case QueueType::SymbolInformation: case QueueType::ParamPingback: m_serverQuerySpaceLeft++; break; default: break; } } } void Worker::Query( ServerQuery type, uint64_t data, uint32_t extra ) { ServerQueryPacket query { type, data, extra }; if( m_serverQueryQueue.empty() && m_serverQuerySpaceLeft > 0 ) { m_serverQuerySpaceLeft--; m_sock.Send( &query, ServerQueryPacketSize ); } else { m_serverQueryQueue.push_back( query ); } } void Worker::QueryTerminate() { ServerQueryPacket query { ServerQueryTerminate, 0, 0 }; m_sock.Send( &query, ServerQueryPacketSize ); } bool Worker::DispatchProcess( const QueueItem& ev, const char*& ptr ) { if( ev.hdr.idx >= (int)QueueType::StringData ) { ptr += sizeof( QueueHeader ) + sizeof( QueueStringTransfer ); if( ev.hdr.type == QueueType::FrameImageData || ev.hdr.type == QueueType::SymbolCode ) { uint32_t sz; memcpy( &sz, ptr, sizeof( sz ) ); ptr += sizeof( sz ); switch( ev.hdr.type ) { case QueueType::FrameImageData: AddFrameImageData( ev.stringTransfer.ptr, ptr, sz ); break; case QueueType::SymbolCode: AddSymbolCode( ev.stringTransfer.ptr, ptr, sz ); m_serverQuerySpaceLeft++; break; default: assert( false ); break; } ptr += sz; } else { uint16_t sz; memcpy( &sz, ptr, sizeof( sz ) ); ptr += sizeof( sz ); switch( ev.hdr.type ) { case QueueType::StringData: AddString( ev.stringTransfer.ptr, ptr, sz ); m_serverQuerySpaceLeft++; break; case QueueType::ThreadName: AddThreadString( ev.stringTransfer.ptr, ptr, sz ); m_serverQuerySpaceLeft++; break; case QueueType::PlotName: HandlePlotName( ev.stringTransfer.ptr, ptr, sz ); m_serverQuerySpaceLeft++; break; case QueueType::SourceLocationPayload: AddSourceLocationPayload( ev.stringTransfer.ptr, ptr, sz ); break; case QueueType::CallstackPayload: AddCallstackPayload( ev.stringTransfer.ptr, ptr, sz ); break; case QueueType::FrameName: HandleFrameName( ev.stringTransfer.ptr, ptr, sz ); m_serverQuerySpaceLeft++; break; case QueueType::CallstackAllocPayload: AddCallstackAllocPayload( ev.stringTransfer.ptr, ptr, sz ); break; case QueueType::ExternalName: AddExternalName( ev.stringTransfer.ptr, ptr, sz ); m_serverQuerySpaceLeft++; break; case QueueType::ExternalThreadName: AddExternalThreadName( ev.stringTransfer.ptr, ptr, sz ); break; default: assert( false ); break; } ptr += sz; } return true; } else { uint16_t sz; switch( ev.hdr.type ) { case QueueType::SingleStringData: ptr += sizeof( QueueHeader ); memcpy( &sz, ptr, sizeof( sz ) ); ptr += sizeof( sz ); AddSingleString( ptr, sz ); ptr += sz; return true; case QueueType::SecondStringData: ptr += sizeof( QueueHeader ); memcpy( &sz, ptr, sizeof( sz ) ); ptr += sizeof( sz ); AddSecondString( ptr, sz ); ptr += sz; return true; default: ptr += QueueDataSize[ev.hdr.idx]; return Process( ev ); } } } void Worker::CheckSourceLocation( uint64_t ptr ) { if( m_data.checkSrclocLast != ptr ) { m_data.checkSrclocLast = ptr; if( m_data.sourceLocation.find( ptr ) == m_data.sourceLocation.end() ) { NewSourceLocation( ptr ); } } } void Worker::NewSourceLocation( uint64_t ptr ) { static const SourceLocation emptySourceLocation = {}; m_data.sourceLocation.emplace( ptr, emptySourceLocation ); m_pendingSourceLocation++; m_sourceLocationQueue.push_back( ptr ); Query( ServerQuerySourceLocation, ptr ); } int16_t Worker::ShrinkSourceLocationReal( uint64_t srcloc ) { auto it = m_sourceLocationShrink.find( srcloc ); if( it != m_sourceLocationShrink.end() ) { m_data.shrinkSrclocLast.first = srcloc; m_data.shrinkSrclocLast.second = it->second; return it->second; } else { return NewShrinkedSourceLocation( srcloc ); } } int16_t Worker::NewShrinkedSourceLocation( uint64_t srcloc ) { assert( m_data.sourceLocationExpand.size() < std::numeric_limits::max() ); const auto sz = int16_t( m_data.sourceLocationExpand.size() ); m_data.sourceLocationExpand.push_back( srcloc ); #ifndef TRACY_NO_STATISTICS auto res = m_data.sourceLocationZones.emplace( sz, SourceLocationZones() ); m_data.srclocZonesLast.first = sz; m_data.srclocZonesLast.second = &res.first->second; #else auto res = m_data.sourceLocationZonesCnt.emplace( sz, 0 ); m_data.srclocCntLast.first = sz; m_data.srclocCntLast.second = &res.first->second; #endif m_sourceLocationShrink.emplace( srcloc, sz ); m_data.shrinkSrclocLast.first = srcloc; m_data.shrinkSrclocLast.second = sz; return sz; } void Worker::InsertMessageData( MessageData* msg ) { if( m_data.messages.empty() ) { m_data.messages.push_back( msg ); } else if( m_data.messages.back()->time < msg->time ) { m_data.messages.push_back_non_empty( msg ); } else { auto mit = std::lower_bound( m_data.messages.begin(), m_data.messages.end(), msg->time, [] ( const auto& lhs, const auto& rhs ) { return lhs->time < rhs; } ); m_data.messages.insert( mit, msg ); } auto td = m_threadCtxData; if( !td ) td = m_threadCtxData = NoticeThread( m_threadCtx ); auto vec = &td->messages; if( vec->empty() ) { vec->push_back( msg ); } else if( vec->back()->time < msg->time ) { vec->push_back_non_empty( msg ); } else { auto tmit = std::lower_bound( vec->begin(), vec->end(), msg->time, [] ( const auto& lhs, const auto& rhs ) { return lhs->time < rhs; } ); vec->insert( tmit, msg ); } } ThreadData* Worker::NoticeThreadReal( uint64_t thread ) { auto it = m_threadMap.find( thread ); if( it != m_threadMap.end() ) { m_data.threadDataLast.first = thread; m_data.threadDataLast.second = it->second; return it->second; } else { return NewThread( thread ); } } ThreadData* Worker::RetrieveThreadReal( uint64_t thread ) { auto it = m_threadMap.find( thread ); if( it != m_threadMap.end() ) { m_data.threadDataLast.first = thread; m_data.threadDataLast.second = it->second; return it->second; } else { return nullptr; } } #ifndef TRACY_NO_STATISTICS Worker::SourceLocationZones* Worker::GetSourceLocationZonesReal( uint16_t srcloc ) { auto it = m_data.sourceLocationZones.find( srcloc ); assert( it != m_data.sourceLocationZones.end() ); m_data.srclocZonesLast.first = srcloc; m_data.srclocZonesLast.second = &it->second; return &it->second; } #else uint64_t* Worker::GetSourceLocationZonesCntReal( uint16_t srcloc ) { auto it = m_data.sourceLocationZonesCnt.find( srcloc ); assert( it != m_data.sourceLocationZonesCnt.end() ); m_data.srclocCntLast.first = srcloc; m_data.srclocCntLast.second = &it->second; return &it->second; } #endif const ThreadData* Worker::GetThreadData( uint64_t tid ) const { auto it = m_threadMap.find( tid ); if( it == m_threadMap.end() ) return nullptr; return it->second; } ThreadData* Worker::NewThread( uint64_t thread ) { CheckThreadString( thread ); auto td = m_slab.AllocInit(); td->id = thread; td->count = 0; td->nextZoneId = 0; #ifndef TRACY_NO_STATISTICS td->ghostIdx = 0; #endif m_data.threads.push_back( td ); m_threadMap.emplace( thread, td ); m_data.threadDataLast.first = thread; m_data.threadDataLast.second = td; return td; } void Worker::NewZone( ZoneEvent* zone, uint64_t thread ) { m_data.zonesCnt++; auto td = m_threadCtxData; if( !td ) td = m_threadCtxData = NoticeThread( thread ); td->count++; const auto ssz = td->stack.size(); if( ssz == 0 ) { td->stack.push_back( zone ); td->timeline.push_back( zone ); } else { auto& back = td->stack.data()[ssz-1]; if( !back->HasChildren() ) { back->SetChild( int32_t( m_data.zoneChildren.size() ) ); if( m_data.zoneVectorCache.empty() ) { m_data.zoneChildren.push_back( Vector>( zone ) ); } else { Vector> vze = std::move( m_data.zoneVectorCache.back_and_pop() ); assert( !vze.empty() ); vze.clear(); vze.push_back_non_empty( zone ); m_data.zoneChildren.push_back( std::move( vze ) ); } } else { const auto backChild = back->Child(); assert( !m_data.zoneChildren[backChild].empty() ); m_data.zoneChildren[backChild].push_back_non_empty( zone ); } td->stack.push_back_non_empty( zone ); } td->zoneIdStack.push_back( td->nextZoneId ); td->nextZoneId = 0; #ifndef TRACY_NO_STATISTICS td->childTimeStack.push_back( 0 ); #endif } void Worker::InsertLockEvent( LockMap& lockmap, LockEvent* lev, uint64_t thread, int64_t time ) { if( m_data.lastTime < time ) m_data.lastTime = time; NoticeThread( thread ); auto it = lockmap.threadMap.find( thread ); if( it == lockmap.threadMap.end() ) { assert( lockmap.threadList.size() < MaxLockThreads ); it = lockmap.threadMap.emplace( thread, lockmap.threadList.size() ).first; lockmap.threadList.emplace_back( thread ); } lev->thread = it->second; assert( lev->thread == it->second ); auto& timeline = lockmap.timeline; if( timeline.empty() ) { timeline.push_back( { lev } ); UpdateLockCount( lockmap, timeline.size() - 1 ); } else { assert( timeline.back().ptr->Time() <= time ); timeline.push_back_non_empty( { lev } ); UpdateLockCount( lockmap, timeline.size() - 1 ); } auto& range = lockmap.range[it->second]; if( range.start > time ) range.start = time; if( range.end < time ) range.end = time; } bool Worker::CheckString( uint64_t ptr ) { if( ptr == 0 ) return true; if( m_data.strings.find( ptr ) != m_data.strings.end() ) return true; m_data.strings.emplace( ptr, "???" ); m_pendingStrings++; Query( ServerQueryString, ptr ); return false; } void Worker::CheckThreadString( uint64_t id ) { if( m_data.threadNames.find( id ) != m_data.threadNames.end() ) return; m_data.threadNames.emplace( id, "???" ); m_pendingThreads++; if( m_sock.IsValid() ) Query( ServerQueryThreadString, id ); } void Worker::CheckExternalName( uint64_t id ) { if( m_data.externalNames.find( id ) != m_data.externalNames.end() ) return; m_data.externalNames.emplace( id, std::make_pair( "???", "???" ) ); m_pendingExternalNames += 2; Query( ServerQueryExternalName, id ); } void Worker::AddSourceLocation( const QueueSourceLocation& srcloc ) { assert( m_pendingSourceLocation > 0 ); m_pendingSourceLocation--; const auto ptr = m_sourceLocationQueue.front(); m_sourceLocationQueue.erase( m_sourceLocationQueue.begin() ); auto it = m_data.sourceLocation.find( ptr ); assert( it != m_data.sourceLocation.end() ); CheckString( srcloc.name ); if( CheckString( srcloc.file ) ) { StringRef ref( StringRef::Ptr, srcloc.file ); if( srcloc.file != 0 && m_checkedFileStrings.find( ref ) == m_checkedFileStrings.end() && m_pendingFileStrings.find( ref ) == m_pendingFileStrings.end() ) { CacheSource( ref ); } } else { StringRef ref( StringRef::Ptr, srcloc.file ); assert( m_checkedFileStrings.find( ref ) == m_checkedFileStrings.end() ); if( m_pendingFileStrings.find( ref ) == m_pendingFileStrings.end() ) { m_pendingFileStrings.emplace( ref ); } } CheckString( srcloc.function ); const uint32_t color = ( srcloc.r << 16 ) | ( srcloc.g << 8 ) | srcloc.b; it->second = SourceLocation { srcloc.name == 0 ? StringRef() : StringRef( StringRef::Ptr, srcloc.name ), StringRef( StringRef::Ptr, srcloc.function ), StringRef( StringRef::Ptr, srcloc.file ), srcloc.line, color }; } void Worker::AddSourceLocationPayload( uint64_t ptr, const char* data, size_t sz ) { const auto start = data; assert( m_pendingSourceLocationPayload == 0 ); uint32_t color, line; memcpy( &color, data, 4 ); memcpy( &line, data + 4, 4 ); data += 8; auto end = data; while( *end ) end++; const auto func = StoreString( data, end - data ); end++; data = end; while( *end ) end++; const auto source = StoreString( data, end - data ); end++; const auto nsz = sz - ( end - start ); color = ( ( color & 0x00FF0000 ) >> 16 ) | ( ( color & 0x0000FF00 ) ) | ( ( color & 0x000000FF ) << 16 ); SourceLocation srcloc { nsz == 0 ? StringRef() : StringRef( StringRef::Idx, StoreString( end, nsz ).idx ), StringRef( StringRef::Idx, func.idx ), StringRef( StringRef::Idx, source.idx ), line, color }; auto it = m_data.sourceLocationPayloadMap.find( &srcloc ); if( it == m_data.sourceLocationPayloadMap.end() ) { auto slptr = m_slab.Alloc(); memcpy( slptr, &srcloc, sizeof( srcloc ) ); uint32_t idx = m_data.sourceLocationPayload.size(); m_data.sourceLocationPayloadMap.emplace( slptr, idx ); m_pendingSourceLocationPayload = -int16_t( idx + 1 ); m_data.sourceLocationPayload.push_back( slptr ); const auto key = -int16_t( idx + 1 ); #ifndef TRACY_NO_STATISTICS auto res = m_data.sourceLocationZones.emplace( key, SourceLocationZones() ); m_data.srclocZonesLast.first = key; m_data.srclocZonesLast.second = &res.first->second; #else auto res = m_data.sourceLocationZonesCnt.emplace( key, 0 ); m_data.srclocCntLast.first = key; m_data.srclocCntLast.second = &res.first->second; #endif } else { m_pendingSourceLocationPayload = -int16_t( it->second + 1 ); } } void Worker::AddString( uint64_t ptr, const char* str, size_t sz ) { assert( m_pendingStrings > 0 ); m_pendingStrings--; auto it = m_data.strings.find( ptr ); assert( it != m_data.strings.end() && strcmp( it->second, "???" ) == 0 ); const auto sl = StoreString( str, sz ); it->second = sl.ptr; StringRef ref( StringRef::Ptr, ptr ); auto sit = m_pendingFileStrings.find( ref ); if( sit != m_pendingFileStrings.end() ) { m_pendingFileStrings.erase( sit ); CacheSource( ref ); } } void Worker::AddThreadString( uint64_t id, const char* str, size_t sz ) { assert( m_pendingThreads > 0 ); m_pendingThreads--; auto it = m_data.threadNames.find( id ); assert( it != m_data.threadNames.end() && strcmp( it->second, "???" ) == 0 ); const auto sl = StoreString( str, sz ); it->second = sl.ptr; } void Worker::AddSingleString( const char* str, size_t sz ) { assert( m_pendingSingleString.ptr == nullptr ); m_pendingSingleString = StoreString( str, sz ); } void Worker::AddSecondString( const char* str, size_t sz ) { assert( m_pendingSecondString.ptr == nullptr ); m_pendingSecondString = StoreString( str, sz ); } void Worker::AddExternalName( uint64_t ptr, const char* str, size_t sz ) { assert( m_pendingExternalNames > 0 ); m_pendingExternalNames--; auto it = m_data.externalNames.find( ptr ); assert( it != m_data.externalNames.end() && strcmp( it->second.first, "???" ) == 0 ); const auto sl = StoreString( str, sz ); it->second.first = sl.ptr; } void Worker::AddExternalThreadName( uint64_t ptr, const char* str, size_t sz ) { assert( m_pendingExternalNames > 0 ); m_pendingExternalNames--; auto it = m_data.externalNames.find( ptr ); assert( it != m_data.externalNames.end() && strcmp( it->second.second, "???" ) == 0 ); const auto sl = StoreString( str, sz ); it->second.second = sl.ptr; } void Worker::AddFrameImageData( uint64_t ptr, const char* data, size_t sz ) { assert( m_pendingFrameImageData.image == nullptr ); assert( sz % 8 == 0 ); // Input data buffer cannot be changed, as it is used as LZ4 dictionary. if( m_frameImageBufferSize < sz ) { m_frameImageBufferSize = sz; delete[] m_frameImageBuffer; m_frameImageBuffer = new char[sz]; } auto src = (uint8_t*)data; auto dst = (uint8_t*)m_frameImageBuffer; memcpy( dst, src, sz ); m_texcomp.FixOrder( (char*)dst, sz/8 ); m_texcomp.Rdo( (char*)dst, sz/8 ); m_pendingFrameImageData.image = m_texcomp.Pack( m_frameImageBuffer, sz, m_pendingFrameImageData.csz, m_slab ); } void Worker::AddSymbolCode( uint64_t ptr, const char* data, size_t sz ) { auto it = m_pendingSymbolCode.find( ptr ); assert( it != m_pendingSymbolCode.end() ); m_pendingSymbolCode.erase( it ); auto code = (char*)m_slab.AllocBig( sz ); memcpy( code, data, sz ); m_data.symbolCode.emplace( ptr, MemoryBlock{ code, uint32_t( sz ) } ); m_data.symbolCodeSize += sz; if( m_data.cpuArch == CpuArchUnknown ) return; csh handle; cs_err rval = CS_ERR_ARCH; switch( m_data.cpuArch ) { case CpuArchX86: rval = cs_open( CS_ARCH_X86, CS_MODE_32, &handle ); break; case CpuArchX64: rval = cs_open( CS_ARCH_X86, CS_MODE_64, &handle ); break; case CpuArchArm32: rval = cs_open( CS_ARCH_ARM, CS_MODE_ARM, &handle ); break; case CpuArchArm64: rval = cs_open( CS_ARCH_ARM64, CS_MODE_ARM, &handle ); break; default: assert( false ); break; } if( rval != CS_ERR_OK ) return; cs_insn* insn; size_t cnt = cs_disasm( handle, (const uint8_t*)code, sz, ptr, 0, &insn ); if( cnt > 0 ) { m_pendingCodeInformation += cnt; for( size_t i=0; i ) + sz * sizeof( CallstackFrameId ); auto mem = (char*)m_slab.AllocRaw( memsize ); auto data = (CallstackFrameId*)mem; auto dst = data; auto src = (uint64_t*)_data; for( size_t i=0; i*)( mem + sz * sizeof( CallstackFrameId ) ); new(arr) VarArray( sz, data ); uint32_t idx; auto it = m_data.callstackMap.find( arr ); if( it == m_data.callstackMap.end() ) { idx = m_data.callstackPayload.size(); m_data.callstackMap.emplace( arr, idx ); m_data.callstackPayload.push_back( arr ); for( auto& frame : *arr ) { auto fit = m_data.callstackFrameMap.find( frame ); if( fit == m_data.callstackFrameMap.end() ) { m_pendingCallstackFrames++; Query( ServerQueryCallstackFrame, GetCanonicalPointer( frame ) ); } } } else { idx = it->second; m_slab.Unalloc( memsize ); } m_pendingCallstackPtr = ptr; m_pendingCallstackId = idx; } void Worker::AddCallstackAllocPayload( uint64_t ptr, const char* data, size_t _sz ) { CallstackFrameId stack[64]; uint8_t sz; memcpy( &sz, data, 1 ); data++; assert( sz <= 64 ); for( uint8_t i=0; i(); memcpy( frame, &cf, sizeof( CallstackFrame ) ); auto frameData = m_slab.AllocInit(); frameData->data = frame; frameData->size = 1; id.idx = m_callstackAllocNextIdx++; id.sel = 1; id.custom = 0; m_data.callstackFrameMap.emplace( id, frameData ); m_data.revFrameMap.emplace( frameData, id ); } else { id = it->second; } stack[i] = id; } VarArray* arr; size_t memsize; if( m_pendingCallstackPtr != 0 ) { const auto nativeCs = m_data.callstackPayload[m_pendingCallstackId]; const auto nsz = nativeCs->size(); const auto tsz = sz + nsz; memsize = sizeof( VarArray ) + tsz * sizeof( CallstackFrameId ); auto mem = (char*)m_slab.AllocRaw( memsize ); memcpy( mem, stack, sizeof( CallstackFrameId ) * sz ); memcpy( mem + sizeof( CallstackFrameId ) * sz, nativeCs->data(), sizeof( CallstackFrameId ) * nsz ); arr = (VarArray*)( mem + tsz * sizeof( CallstackFrameId ) ); new(arr) VarArray( tsz, (CallstackFrameId*)mem ); } else { memsize = sizeof( VarArray ) + sz * sizeof( CallstackFrameId ); auto mem = (char*)m_slab.AllocRaw( memsize ); memcpy( mem, stack, sizeof( CallstackFrameId ) * sz ); arr = (VarArray*)( mem + sz * sizeof( CallstackFrameId ) ); new(arr) VarArray( sz, (CallstackFrameId*)mem ); } uint32_t idx; auto it = m_data.callstackMap.find( arr ); if( it == m_data.callstackMap.end() ) { idx = m_data.callstackPayload.size(); m_data.callstackMap.emplace( arr, idx ); m_data.callstackPayload.push_back( arr ); for( auto& frame : *arr ) { auto fit = m_data.callstackFrameMap.find( frame ); if( fit == m_data.callstackFrameMap.end() ) { m_pendingCallstackFrames++; Query( ServerQueryCallstackFrame, GetCanonicalPointer( frame ) ); } } } else { idx = it->second; m_slab.Unalloc( memsize ); } m_pendingCallstackPtr = ptr; m_pendingCallstackId = idx; } void Worker::InsertPlot( PlotData* plot, int64_t time, double val ) { if( plot->data.empty() ) { plot->min = val; plot->max = val; plot->data.push_back( { Int48( time ), val } ); } else if( plot->data.back().time.Val() < time ) { if( plot->min > val ) plot->min = val; else if( plot->max < val ) plot->max = val; plot->data.push_back_non_empty( { Int48( time ), val } ); } else { if( plot->min > val ) plot->min = val; else if( plot->max < val ) plot->max = val; if( plot->postpone.empty() ) { plot->postponeTime = std::chrono::duration_cast( std::chrono::high_resolution_clock::now().time_since_epoch() ).count(); plot->postpone.push_back( { Int48( time ), val } ); } else { plot->postpone.push_back_non_empty( { Int48( time ), val } ); } } } void Worker::HandlePlotName( uint64_t name, const char* str, size_t sz ) { const auto sl = StoreString( str, sz ); m_data.plots.StringDiscovered( name, sl, m_data.strings, [this] ( PlotData* dst, PlotData* src ) { for( auto& v : src->data ) { InsertPlot( dst, v.time.Val(), v.val ); } } ); } void Worker::HandleFrameName( uint64_t name, const char* str, size_t sz ) { const auto sl = StoreString( str, sz ); m_data.frames.StringDiscovered( name, sl, m_data.strings, [] ( FrameData* dst, FrameData* src ) { auto sz = dst->frames.size(); dst->frames.insert( dst->frames.end(), src->frames.begin(), src->frames.end() ); std::inplace_merge( dst->frames.begin(), dst->frames.begin() + sz, dst->frames.end(), [] ( const auto& lhs, const auto& rhs ) { return lhs.start < rhs.start; } ); } ); } void Worker::HandlePostponedPlots() { for( auto& plot : m_data.plots.Data() ) { auto& src = plot->postpone; if( src.empty() ) continue; if( std::chrono::duration_cast( std::chrono::high_resolution_clock::now().time_since_epoch() ).count() - plot->postponeTime < 100 ) continue; auto& dst = plot->data; #ifdef NO_PARALLEL_SORT pdqsort_branchless( src.begin(), src.end(), [] ( const auto& l, const auto& r ) { return l.time.Val() < r.time.Val(); } ); #else std::sort( std::execution::par_unseq, src.begin(), src.end(), [] ( const auto& l, const auto& r ) { return l.time.Val() < r.time.Val(); } ); #endif const auto ds = std::lower_bound( dst.begin(), dst.end(), src.front().time.Val(), [] ( const auto& l, const auto& r ) { return l.time.Val() < r; } ); const auto dsd = std::distance( dst.begin(), ds ) ; const auto de = std::lower_bound( ds, dst.end(), src.back().time.Val(), [] ( const auto& l, const auto& r ) { return l.time.Val() < r; } ); const auto ded = std::distance( dst.begin(), de ); dst.insert( de, src.begin(), src.end() ); std::inplace_merge( dst.begin() + dsd, dst.begin() + ded, dst.begin() + ded + src.size(), [] ( const auto& l, const auto& r ) { return l.time.Val() < r.time.Val(); } ); src.clear(); } } #ifndef TRACY_NO_STATISTICS void Worker::HandlePostponedSamples() { assert( m_data.newFramesWereReceived ); if( m_data.postponedSamples.empty() ) return; auto it = m_data.postponedSamples.begin(); do { UpdateSampleStatisticsPostponed( it ); } while( it != m_data.postponedSamples.end() ); } void Worker::GetStackWithInlines( Vector& ret, const VarArray& cs ) { ret.clear(); int idx = cs.size() - 1; do { auto& entry = cs[idx]; const auto frame = GetCallstackFrame( entry ); if( frame ) { uint8_t i = frame->size; do { i--; ret.push_back( InlineStackData { frame->data[i].symAddr, entry, i } ); } while( i != 0 ); } else { ret.push_back( InlineStackData{ GetCanonicalPointer( entry ), entry, 0 } ); } } while( idx-- > 0 ); } int Worker::AddGhostZone( const VarArray& cs, Vector* vec, uint64_t t ) { static Vector stack; GetStackWithInlines( stack, cs ); if( !vec->empty() && vec->back().end.Val() > (int64_t)t ) { const auto refBackTime = vec->back().end.Val(); auto tmp = vec; for(;;) { auto& back = tmp->back(); if( back.end.Val() != refBackTime ) break; back.end.SetVal( t ); if( back.child < 0 ) break; tmp = &m_data.ghostChildren[back.child]; } } const uint64_t refBackTime = vec->empty() ? 0 : vec->back().end.Val(); int gcnt = 0; int idx = 0; while( !vec->empty() && idx < stack.size() ) { auto& back = vec->back(); const auto& backKey = m_data.ghostFrames[back.frame.Val()]; const auto backFrame = GetCallstackFrame( backKey.frame ); if( !backFrame ) break; const auto& inlineFrame = backFrame->data[backKey.inlineFrame]; if( inlineFrame.symAddr != stack[idx].symAddr ) break; if( back.end.Val() != refBackTime ) break; back.end.SetVal( t + m_samplingPeriod ); if( ++idx == stack.size() ) break; if( back.child < 0 ) { back.child = m_data.ghostChildren.size(); vec = &m_data.ghostChildren.push_next(); } else { vec = &m_data.ghostChildren[back.child]; } } while( idx < stack.size() ) { gcnt++; uint32_t fid; GhostKey key { stack[idx].frame, stack[idx].inlineFrame }; auto it = m_data.ghostFramesMap.find( key ); if( it == m_data.ghostFramesMap.end() ) { fid = uint32_t( m_data.ghostFrames.size() ); m_data.ghostFrames.push_back( key ); m_data.ghostFramesMap.emplace( key, fid ); } else { fid = it->second; } auto& zone = vec->push_next(); zone.start.SetVal( t ); zone.end.SetVal( t + m_samplingPeriod ); zone.frame.SetVal( fid ); if( ++idx == stack.size() ) { zone.child = -1; } else { zone.child = m_data.ghostChildren.size(); vec = &m_data.ghostChildren.push_next(); } } return gcnt; } void Worker::HandlePostponedGhostZones() { assert( m_data.newFramesWereReceived ); if( !m_data.ghostZonesPostponed ) return; bool postponed = false; for( auto& td : m_data.threads ) { while( td->ghostIdx != td->samples.size() ) { const auto& sample = td->samples[td->ghostIdx]; const auto& cs = GetCallstack( sample.callstack.Val() ); const auto cssz = cs.size(); uint16_t i; for( i=0; ighostIdx++; m_data.ghostCnt += AddGhostZone( cs, &td->ghostZones, sample.time.Val() ); } } m_data.ghostZonesPostponed = postponed; } #endif uint32_t Worker::GetSingleStringIdx() { assert( m_pendingSingleString.ptr != nullptr ); const auto idx = m_pendingSingleString.idx; m_pendingSingleString.ptr = nullptr; return idx; } uint32_t Worker::GetSecondStringIdx() { assert( m_pendingSecondString.ptr != nullptr ); const auto idx = m_pendingSecondString.idx; m_pendingSecondString.ptr = nullptr; return idx; } StringLocation Worker::StoreString( const char* str, size_t sz ) { StringLocation ret; charutil::StringKey key = { str, sz }; auto sit = m_data.stringMap.find( key ); if( sit == m_data.stringMap.end() ) { auto ptr = m_slab.Alloc( sz+1 ); memcpy( ptr, str, sz ); ptr[sz] = '\0'; ret.ptr = ptr; ret.idx = m_data.stringData.size(); m_data.stringMap.emplace( charutil::StringKey { ptr, sz }, m_data.stringData.size() ); m_data.stringData.push_back( ptr ); } else { ret.ptr = sit->first.ptr; ret.idx = sit->second; } return ret; } bool Worker::Process( const QueueItem& ev ) { switch( ev.hdr.type ) { case QueueType::ThreadContext: ProcessThreadContext( ev.threadCtx ); break; case QueueType::ZoneBegin: ProcessZoneBegin( ev.zoneBegin ); break; case QueueType::ZoneBeginCallstack: ProcessZoneBeginCallstack( ev.zoneBegin ); break; case QueueType::ZoneBeginAllocSrcLoc: ProcessZoneBeginAllocSrcLoc( ev.zoneBeginLean ); break; case QueueType::ZoneBeginAllocSrcLocCallstack: ProcessZoneBeginAllocSrcLocCallstack( ev.zoneBeginLean ); break; case QueueType::ZoneEnd: ProcessZoneEnd( ev.zoneEnd ); break; case QueueType::ZoneValidation: ProcessZoneValidation( ev.zoneValidation ); break; case QueueType::FrameMarkMsg: ProcessFrameMark( ev.frameMark ); break; case QueueType::FrameMarkMsgStart: ProcessFrameMarkStart( ev.frameMark ); break; case QueueType::FrameMarkMsgEnd: ProcessFrameMarkEnd( ev.frameMark ); break; case QueueType::FrameImage: ProcessFrameImage( ev.frameImage ); break; case QueueType::SourceLocation: AddSourceLocation( ev.srcloc ); m_serverQuerySpaceLeft++; break; case QueueType::ZoneText: ProcessZoneText(); break; case QueueType::ZoneName: ProcessZoneName(); break; case QueueType::ZoneValue: ProcessZoneValue( ev.zoneValue ); break; case QueueType::LockAnnounce: ProcessLockAnnounce( ev.lockAnnounce ); break; case QueueType::LockTerminate: ProcessLockTerminate( ev.lockTerminate ); break; case QueueType::LockWait: ProcessLockWait( ev.lockWait ); break; case QueueType::LockObtain: ProcessLockObtain( ev.lockObtain ); break; case QueueType::LockRelease: ProcessLockRelease( ev.lockRelease ); break; case QueueType::LockSharedWait: ProcessLockSharedWait( ev.lockWait ); break; case QueueType::LockSharedObtain: ProcessLockSharedObtain( ev.lockObtain ); break; case QueueType::LockSharedRelease: ProcessLockSharedRelease( ev.lockRelease ); break; case QueueType::LockMark: ProcessLockMark( ev.lockMark ); break; case QueueType::LockName: ProcessLockName( ev.lockName ); break; case QueueType::PlotData: ProcessPlotData( ev.plotData ); break; case QueueType::PlotConfig: ProcessPlotConfig( ev.plotConfig ); break; case QueueType::Message: ProcessMessage( ev.message ); break; case QueueType::MessageLiteral: ProcessMessageLiteral( ev.messageLiteral ); break; case QueueType::MessageColor: ProcessMessageColor( ev.messageColor ); break; case QueueType::MessageLiteralColor: ProcessMessageLiteralColor( ev.messageColorLiteral ); break; case QueueType::MessageCallstack: ProcessMessageCallstack( ev.message ); break; case QueueType::MessageLiteralCallstack: ProcessMessageLiteralCallstack( ev.messageLiteral ); break; case QueueType::MessageColorCallstack: ProcessMessageColorCallstack( ev.messageColor ); break; case QueueType::MessageLiteralColorCallstack: ProcessMessageLiteralColorCallstack( ev.messageColorLiteral ); break; case QueueType::MessageAppInfo: ProcessMessageAppInfo( ev.message ); break; case QueueType::GpuNewContext: ProcessGpuNewContext( ev.gpuNewContext ); break; case QueueType::GpuZoneBegin: ProcessGpuZoneBegin( ev.gpuZoneBegin, false ); break; case QueueType::GpuZoneBeginCallstack: ProcessGpuZoneBeginCallstack( ev.gpuZoneBegin, false ); break; case QueueType::GpuZoneEnd: ProcessGpuZoneEnd( ev.gpuZoneEnd, false ); break; case QueueType::GpuZoneBeginSerial: ProcessGpuZoneBegin( ev.gpuZoneBegin, true ); break; case QueueType::GpuZoneBeginCallstackSerial: ProcessGpuZoneBeginCallstack( ev.gpuZoneBegin, true ); break; case QueueType::GpuZoneEndSerial: ProcessGpuZoneEnd( ev.gpuZoneEnd, true ); break; case QueueType::GpuTime: ProcessGpuTime( ev.gpuTime ); break; case QueueType::GpuCalibration: ProcessGpuCalibration( ev.gpuCalibration ); break; case QueueType::MemAlloc: ProcessMemAlloc( ev.memAlloc ); break; case QueueType::MemFree: ProcessMemFree( ev.memFree ); break; case QueueType::MemAllocCallstack: ProcessMemAllocCallstack( ev.memAlloc ); break; case QueueType::MemFreeCallstack: ProcessMemFreeCallstack( ev.memFree ); break; case QueueType::CallstackMemory: ProcessCallstackMemory(); break; case QueueType::Callstack: ProcessCallstack(); break; case QueueType::CallstackAlloc: ProcessCallstackAlloc(); break; case QueueType::CallstackSample: ProcessCallstackSample( ev.callstackSample ); break; case QueueType::CallstackFrameSize: ProcessCallstackFrameSize( ev.callstackFrameSize ); m_serverQuerySpaceLeft++; break; case QueueType::CallstackFrame: ProcessCallstackFrame( ev.callstackFrame ); break; case QueueType::SymbolInformation: ProcessSymbolInformation( ev.symbolInformation ); m_serverQuerySpaceLeft++; break; case QueueType::CodeInformation: ProcessCodeInformation( ev.codeInformation ); m_serverQuerySpaceLeft++; break; case QueueType::Terminate: m_terminate = true; break; case QueueType::KeepAlive: break; case QueueType::Crash: m_crashed = true; break; case QueueType::CrashReport: ProcessCrashReport( ev.crashReport ); break; case QueueType::SysTimeReport: ProcessSysTime( ev.sysTime ); break; case QueueType::ContextSwitch: ProcessContextSwitch( ev.contextSwitch ); break; case QueueType::ThreadWakeup: ProcessThreadWakeup( ev.threadWakeup ); break; case QueueType::TidToPid: ProcessTidToPid( ev.tidToPid ); break; case QueueType::ParamSetup: ProcessParamSetup( ev.paramSetup ); break; case QueueType::ParamPingback: m_serverQuerySpaceLeft++; break; case QueueType::CpuTopology: ProcessCpuTopology( ev.cpuTopology ); break; default: assert( false ); break; } return m_failure == Failure::None; } void Worker::ProcessThreadContext( const QueueThreadContext& ev ) { m_refTimeThread = 0; if( m_threadCtx != ev.thread ) { m_threadCtx = ev.thread; m_threadCtxData = RetrieveThread( ev.thread ); } } void Worker::ProcessZoneBeginImpl( ZoneEvent* zone, const QueueZoneBegin& ev ) { CheckSourceLocation( ev.srcloc ); const auto refTime = m_refTimeThread + ev.time; m_refTimeThread = refTime; const auto start = TscTime( refTime - m_data.baseTime ); zone->SetStartSrcLoc( start, ShrinkSourceLocation( ev.srcloc ) ); zone->SetEnd( -1 ); zone->SetChild( -1 ); if( m_data.lastTime < start ) m_data.lastTime = start; NewZone( zone, m_threadCtx ); } ZoneEvent* Worker::AllocZoneEvent() { ZoneEvent* ret; #ifndef TRACY_NO_STATISTICS ret = m_slab.Alloc(); #else if( m_zoneEventPool.empty() ) { ret = m_slab.Alloc(); } else { ret = m_zoneEventPool.back_and_pop(); } #endif ret->extra = 0; return ret; } void Worker::ProcessZoneBegin( const QueueZoneBegin& ev ) { auto zone = AllocZoneEvent(); ProcessZoneBeginImpl( zone, ev ); } void Worker::ProcessZoneBeginCallstack( const QueueZoneBegin& ev ) { auto zone = AllocZoneEvent(); ProcessZoneBeginImpl( zone, ev ); auto& next = m_nextCallstack[m_threadCtx]; next.type = NextCallstackType::Zone; next.zone = zone; } void Worker::ProcessZoneBeginAllocSrcLocImpl( ZoneEvent* zone, const QueueZoneBeginLean& ev ) { assert( m_pendingSourceLocationPayload != 0 ); const auto refTime = m_refTimeThread + ev.time; m_refTimeThread = refTime; const auto start = TscTime( refTime - m_data.baseTime ); zone->SetStartSrcLoc( start, m_pendingSourceLocationPayload ); zone->SetEnd( -1 ); zone->SetChild( -1 ); if( m_data.lastTime < start ) m_data.lastTime = start; NewZone( zone, m_threadCtx ); m_pendingSourceLocationPayload = 0; } void Worker::ProcessZoneBeginAllocSrcLoc( const QueueZoneBeginLean& ev ) { auto zone = AllocZoneEvent(); ProcessZoneBeginAllocSrcLocImpl( zone, ev ); } void Worker::ProcessZoneBeginAllocSrcLocCallstack( const QueueZoneBeginLean& ev ) { auto zone = AllocZoneEvent(); ProcessZoneBeginAllocSrcLocImpl( zone, ev ); auto& next = m_nextCallstack[m_threadCtx]; next.type = NextCallstackType::Zone; next.zone = zone; } void Worker::ProcessZoneEnd( const QueueZoneEnd& ev ) { auto td = m_threadCtxData; assert( td ); if( td->zoneIdStack.empty() ) { ZoneDoubleEndFailure( m_threadCtx, td->timeline.empty() ? nullptr : td->timeline.back() ); return; } auto zoneId = td->zoneIdStack.back_and_pop(); if( zoneId != td->nextZoneId ) { ZoneStackFailure( m_threadCtx, td->stack.back() ); return; } td->nextZoneId = 0; auto& stack = td->stack; assert( !stack.empty() ); auto zone = stack.back_and_pop(); assert( zone->End() == -1 ); const auto refTime = m_refTimeThread + ev.time; m_refTimeThread = refTime; const auto timeEnd = TscTime( refTime - m_data.baseTime ); zone->SetEnd( timeEnd ); assert( timeEnd >= zone->Start() ); if( m_data.lastTime < timeEnd ) m_data.lastTime = timeEnd; if( zone->HasChildren() ) { auto& childVec = m_data.zoneChildren[zone->Child()]; const auto sz = childVec.size(); if( sz <= 8 * 1024 ) { Vector> fitVec; #ifndef TRACY_NO_STATISTICS fitVec.reserve_exact( sz, m_slab ); memcpy( fitVec.data(), childVec.data(), sz * sizeof( short_ptr ) ); #else fitVec.set_magic(); auto& fv = *((Vector*)&fitVec); fv.reserve_exact( sz, m_slab ); auto dst = fv.data(); for( auto& ze : childVec ) { ZoneEvent* src = ze; memcpy( dst++, src, sizeof( ZoneEvent ) ); m_zoneEventPool.push_back( src ); } #endif fitVec.swap( childVec ); m_data.zoneVectorCache.push_back( std::move( fitVec ) ); } } #ifndef TRACY_NO_STATISTICS assert( !td->childTimeStack.empty() ); const auto timeSpan = timeEnd - zone->Start(); if( timeSpan > 0 ) { auto slz = GetSourceLocationZones( zone->SrcLoc() ); auto& ztd = slz->zones.push_next(); ztd.SetZone( zone ); ztd.SetThread( CompressThread( m_threadCtx ) ); if( slz->min > timeSpan ) slz->min = timeSpan; if( slz->max < timeSpan ) slz->max = timeSpan; slz->total += timeSpan; slz->sumSq += double( timeSpan ) * timeSpan; const auto selfSpan = timeSpan - td->childTimeStack.back_and_pop(); if( slz->selfMin > selfSpan ) slz->selfMin = selfSpan; if( slz->selfMax < selfSpan ) slz->selfMax = selfSpan; slz->selfTotal += selfSpan; if( !td->childTimeStack.empty() ) { td->childTimeStack.back() += timeSpan; } } else { td->childTimeStack.pop_back(); } #else CountZoneStatistics( zone ); #endif } void Worker::ZoneStackFailure( uint64_t thread, const ZoneEvent* ev ) { m_failure = Failure::ZoneStack; m_failureData.thread = thread; m_failureData.srcloc = ev->SrcLoc(); } void Worker::ZoneDoubleEndFailure( uint64_t thread, const ZoneEvent* ev ) { m_failure = Failure::ZoneDoubleEnd; m_failureData.thread = thread; m_failureData.srcloc = ev ? ev->SrcLoc() : 0; } void Worker::ZoneTextFailure( uint64_t thread ) { m_failure = Failure::ZoneText; m_failureData.thread = thread; m_failureData.srcloc = 0; } void Worker::ZoneNameFailure( uint64_t thread ) { m_failure = Failure::ZoneName; m_failureData.thread = thread; m_failureData.srcloc = 0; } void Worker::MemFreeFailure( uint64_t thread ) { m_failure = Failure::MemFree; m_failureData.thread = thread; m_failureData.srcloc = 0; } void Worker::FrameEndFailure() { m_failure = Failure::FrameEnd; m_failureData.thread = 0; m_failureData.srcloc = 0; } void Worker::FrameImageIndexFailure() { m_failure = Failure::FrameImageIndex; m_failureData.thread = 0; m_failureData.srcloc = 0; } void Worker::FrameImageTwiceFailure() { m_failure = Failure::FrameImageTwice; m_failureData.thread = 0; m_failureData.srcloc = 0; } void Worker::ProcessZoneValidation( const QueueZoneValidation& ev ) { auto td = m_threadCtxData; if( !td ) td = m_threadCtxData = NoticeThread( m_threadCtx ); td->nextZoneId = ev.id; } void Worker::ProcessFrameMark( const QueueFrameMark& ev ) { auto fd = m_data.frames.Retrieve( ev.name, [this] ( uint64_t name ) { auto fd = m_slab.AllocInit(); fd->name = name; fd->continuous = 1; return fd; }, [this] ( uint64_t name ) { Query( ServerQueryFrameName, name ); } ); int32_t frameImage = -1; if( ev.name == 0 ) { auto fis = m_frameImageStaging.find( fd->frames.size() ); if( fis != m_frameImageStaging.end() ) { frameImage = fis->second; m_frameImageStaging.erase( fis ); } } assert( fd->continuous == 1 ); const auto time = TscTime( ev.time - m_data.baseTime ); assert( fd->frames.empty() || fd->frames.back().start <= time ); fd->frames.push_back( FrameEvent{ time, -1, frameImage } ); if( m_data.lastTime < time ) m_data.lastTime = time; #ifndef TRACY_NO_STATISTICS const auto timeSpan = GetFrameTime( *fd, fd->frames.size() - 1 ); if( timeSpan > 0 ) { fd->min = std::min( fd->min, timeSpan ); fd->max = std::max( fd->max, timeSpan ); fd->total += timeSpan; fd->sumSq += double( timeSpan ) * timeSpan; } #endif } void Worker::ProcessFrameMarkStart( const QueueFrameMark& ev ) { auto fd = m_data.frames.Retrieve( ev.name, [this] ( uint64_t name ) { auto fd = m_slab.AllocInit(); fd->name = name; fd->continuous = 0; return fd; }, [this] ( uint64_t name ) { Query( ServerQueryFrameName, name ); } ); assert( fd->continuous == 0 ); const auto time = TscTime( ev.time - m_data.baseTime ); assert( fd->frames.empty() || ( fd->frames.back().end <= time && fd->frames.back().end != -1 ) ); fd->frames.push_back( FrameEvent{ time, -1, -1 } ); if( m_data.lastTime < time ) m_data.lastTime = time; } void Worker::ProcessFrameMarkEnd( const QueueFrameMark& ev ) { auto fd = m_data.frames.Retrieve( ev.name, [this] ( uint64_t name ) { auto fd = m_slab.AllocInit(); fd->name = name; fd->continuous = 0; return fd; }, [this] ( uint64_t name ) { Query( ServerQueryFrameName, name ); } ); assert( fd->continuous == 0 ); const auto time = TscTime( ev.time - m_data.baseTime ); if( fd->frames.empty() ) { FrameEndFailure(); return; } assert( fd->frames.back().end == -1 ); fd->frames.back().end = time; if( m_data.lastTime < time ) m_data.lastTime = time; #ifndef TRACY_NO_STATISTICS const auto timeSpan = GetFrameTime( *fd, fd->frames.size() - 1 ); if( timeSpan > 0 ) { fd->min = std::min( fd->min, timeSpan ); fd->max = std::max( fd->max, timeSpan ); fd->total += timeSpan; fd->sumSq += double( timeSpan ) * timeSpan; } #endif } void Worker::ProcessFrameImage( const QueueFrameImage& ev ) { assert( m_pendingFrameImageData.image != nullptr ); auto& frames = m_data.framesBase->frames; const auto fidx = int64_t( ev.frame ) - int64_t( m_data.frameOffset ) + 1; if( m_onDemand && fidx <= 1 ) { m_pendingFrameImageData.image = nullptr; return; } else if( fidx <= 0 ) { FrameImageIndexFailure(); return; } auto fi = m_slab.Alloc(); fi->ptr = m_pendingFrameImageData.image; fi->csz = m_pendingFrameImageData.csz; fi->w = ev.w; fi->h = ev.h; fi->frameRef = uint32_t( fidx ); fi->flip = ev.flip; const auto idx = m_data.frameImage.size(); m_data.frameImage.push_back( fi ); m_pendingFrameImageData.image = nullptr; if( fidx >= (int64_t)frames.size() ) { if( m_frameImageStaging.find( fidx ) != m_frameImageStaging.end() ) { FrameImageTwiceFailure(); return; } m_frameImageStaging.emplace( fidx, idx ); } else if( frames[fidx].frameImage >= 0 ) { FrameImageTwiceFailure(); } else { frames[fidx].frameImage = idx; } } void Worker::ProcessZoneText() { auto td = RetrieveThread( m_threadCtx ); if( !td || td->stack.empty() || td->nextZoneId != td->zoneIdStack.back() ) { ZoneTextFailure( m_threadCtx ); return; } const auto ptr = m_pendingSingleString.ptr; const auto idx = GetSingleStringIdx(); td->nextZoneId = 0; auto& stack = td->stack; auto zone = stack.back(); auto& extra = RequestZoneExtra( *zone ); if( !extra.text.Active() ) { extra.text = StringIdx( idx ); } else { const auto str0 = GetString( extra.text ); const auto str1 = ptr; const auto len0 = strlen( str0 ); const auto len1 = strlen( str1 ); const auto bsz = len0+len1+1; if( m_tmpBufSize < bsz ) { delete[] m_tmpBuf; m_tmpBuf = new char[bsz]; m_tmpBufSize = bsz; } char* buf = m_tmpBuf; memcpy( buf, str0, len0 ); buf[len0] = '\n'; memcpy( buf+len0+1, str1, len1 ); extra.text = StringIdx( StoreString( buf, bsz ).idx ); } } void Worker::ProcessZoneName() { auto td = RetrieveThread( m_threadCtx ); if( !td || td->stack.empty() || td->nextZoneId != td->zoneIdStack.back() ) { ZoneNameFailure( m_threadCtx ); return; } td->nextZoneId = 0; auto& stack = td->stack; auto zone = stack.back(); auto& extra = RequestZoneExtra( *zone ); extra.name = StringIdx( GetSingleStringIdx() ); } void Worker::ProcessZoneValue( const QueueZoneValue& ev ) { char tmp[32]; const auto tsz = sprintf( tmp, "%" PRIu64, ev.value ); auto td = RetrieveThread( m_threadCtx ); if( !td || td->stack.empty() || td->nextZoneId != td->zoneIdStack.back() ) { ZoneTextFailure( m_threadCtx ); return; } td->nextZoneId = 0; auto& stack = td->stack; auto zone = stack.back(); auto& extra = RequestZoneExtra( *zone ); if( !extra.text.Active() ) { extra.text = StringIdx( StoreString( tmp, tsz ).idx ); } else { const auto str0 = GetString( extra.text ); const auto len0 = strlen( str0 ); const auto bsz = len0+tsz+1; if( m_tmpBufSize < bsz ) { delete[] m_tmpBuf; m_tmpBuf = new char[bsz]; m_tmpBufSize = bsz; } char* buf = m_tmpBuf; memcpy( buf, str0, len0 ); buf[len0] = '\n'; memcpy( buf+len0+1, tmp, tsz ); extra.text = StringIdx( StoreString( buf, bsz ).idx ); } } void Worker::ProcessLockAnnounce( const QueueLockAnnounce& ev ) { auto it = m_data.lockMap.find( ev.id ); assert( it == m_data.lockMap.end() ); auto lm = m_slab.AllocInit(); lm->srcloc = ShrinkSourceLocation( ev.lckloc ); lm->type = ev.type; lm->timeAnnounce = TscTime( ev.time - m_data.baseTime ); lm->timeTerminate = 0; lm->valid = true; lm->isContended = false; m_data.lockMap.emplace( ev.id, lm ); CheckSourceLocation( ev.lckloc ); } void Worker::ProcessLockTerminate( const QueueLockTerminate& ev ) { auto it = m_data.lockMap.find( ev.id ); assert( it != m_data.lockMap.end() ); it->second->timeTerminate = TscTime( ev.time - m_data.baseTime ); } void Worker::ProcessLockWait( const QueueLockWait& ev ) { auto it = m_data.lockMap.find( ev.id ); assert( it != m_data.lockMap.end() ); auto& lock = *it->second; auto lev = lock.type == LockType::Lockable ? m_slab.Alloc() : m_slab.Alloc(); const auto refTime = m_refTimeSerial + ev.time; m_refTimeSerial = refTime; const auto time = TscTime( refTime - m_data.baseTime ); lev->SetTime( time ); lev->SetSrcLoc( 0 ); lev->type = LockEvent::Type::Wait; InsertLockEvent( lock, lev, ev.thread, time ); } void Worker::ProcessLockObtain( const QueueLockObtain& ev ) { auto it = m_data.lockMap.find( ev.id ); assert( it != m_data.lockMap.end() ); auto& lock = *it->second; auto lev = lock.type == LockType::Lockable ? m_slab.Alloc() : m_slab.Alloc(); const auto refTime = m_refTimeSerial + ev.time; m_refTimeSerial = refTime; const auto time = TscTime( refTime - m_data.baseTime ); lev->SetTime( time ); lev->SetSrcLoc( 0 ); lev->type = LockEvent::Type::Obtain; InsertLockEvent( lock, lev, ev.thread, time ); } void Worker::ProcessLockRelease( const QueueLockRelease& ev ) { auto it = m_data.lockMap.find( ev.id ); assert( it != m_data.lockMap.end() ); auto& lock = *it->second; auto lev = lock.type == LockType::Lockable ? m_slab.Alloc() : m_slab.Alloc(); const auto refTime = m_refTimeSerial + ev.time; m_refTimeSerial = refTime; const auto time = TscTime( refTime - m_data.baseTime ); lev->SetTime( time ); lev->SetSrcLoc( 0 ); lev->type = LockEvent::Type::Release; InsertLockEvent( lock, lev, ev.thread, time ); } void Worker::ProcessLockSharedWait( const QueueLockWait& ev ) { auto it = m_data.lockMap.find( ev.id ); assert( it != m_data.lockMap.end() ); auto& lock = *it->second; assert( lock.type == LockType::SharedLockable ); auto lev = m_slab.Alloc(); const auto refTime = m_refTimeSerial + ev.time; m_refTimeSerial = refTime; const auto time = TscTime( refTime - m_data.baseTime ); lev->SetTime( time ); lev->SetSrcLoc( 0 ); lev->type = LockEvent::Type::WaitShared; InsertLockEvent( lock, lev, ev.thread, time ); } void Worker::ProcessLockSharedObtain( const QueueLockObtain& ev ) { auto it = m_data.lockMap.find( ev.id ); assert( it != m_data.lockMap.end() ); auto& lock = *it->second; assert( lock.type == LockType::SharedLockable ); auto lev = m_slab.Alloc(); const auto refTime = m_refTimeSerial + ev.time; m_refTimeSerial = refTime; const auto time = TscTime( refTime - m_data.baseTime ); lev->SetTime( time ); lev->SetSrcLoc( 0 ); lev->type = LockEvent::Type::ObtainShared; InsertLockEvent( lock, lev, ev.thread, time ); } void Worker::ProcessLockSharedRelease( const QueueLockRelease& ev ) { auto it = m_data.lockMap.find( ev.id ); assert( it != m_data.lockMap.end() ); auto& lock = *it->second; assert( lock.type == LockType::SharedLockable ); auto lev = m_slab.Alloc(); const auto refTime = m_refTimeSerial + ev.time; m_refTimeSerial = refTime; const auto time = TscTime( refTime - m_data.baseTime ); lev->SetTime( time ); lev->SetSrcLoc( 0 ); lev->type = LockEvent::Type::ReleaseShared; InsertLockEvent( lock, lev, ev.thread, time ); } void Worker::ProcessLockMark( const QueueLockMark& ev ) { CheckSourceLocation( ev.srcloc ); auto lit = m_data.lockMap.find( ev.id ); assert( lit != m_data.lockMap.end() ); auto& lockmap = *lit->second; auto tid = lockmap.threadMap.find( ev.thread ); assert( tid != lockmap.threadMap.end() ); const auto thread = tid->second; auto it = lockmap.timeline.end(); for(;;) { --it; if( it->ptr->thread == thread ) { switch( it->ptr->type ) { case LockEvent::Type::Obtain: case LockEvent::Type::ObtainShared: case LockEvent::Type::Wait: case LockEvent::Type::WaitShared: it->ptr->SetSrcLoc( ShrinkSourceLocation( ev.srcloc ) ); return; default: break; } } } } void Worker::ProcessLockName( const QueueLockName& ev ) { auto lit = m_data.lockMap.find( ev.id ); assert( lit != m_data.lockMap.end() ); lit->second->customName = StringIdx( GetSingleStringIdx() ); } void Worker::ProcessPlotData( const QueuePlotData& ev ) { PlotData* plot = m_data.plots.Retrieve( ev.name, [this] ( uint64_t name ) { auto plot = m_slab.AllocInit(); plot->name = name; plot->type = PlotType::User; plot->format = PlotValueFormatting::Number; return plot; }, [this]( uint64_t name ) { Query( ServerQueryPlotName, name ); } ); const auto refTime = m_refTimeThread + ev.time; m_refTimeThread = refTime; const auto time = TscTime( refTime - m_data.baseTime ); if( m_data.lastTime < time ) m_data.lastTime = time; switch( ev.type ) { case PlotDataType::Double: InsertPlot( plot, time, ev.data.d ); break; case PlotDataType::Float: InsertPlot( plot, time, (double)ev.data.f ); break; case PlotDataType::Int: InsertPlot( plot, time, (double)ev.data.i ); break; default: assert( false ); break; } } void Worker::ProcessPlotConfig( const QueuePlotConfig& ev ) { PlotData* plot = m_data.plots.Retrieve( ev.name, [this] ( uint64_t name ) { auto plot = m_slab.AllocInit(); plot->name = name; plot->type = PlotType::User; return plot; }, [this]( uint64_t name ) { Query( ServerQueryPlotName, name ); } ); plot->format = (PlotValueFormatting)ev.type; } void Worker::ProcessMessage( const QueueMessage& ev ) { auto msg = m_slab.Alloc(); const auto time = TscTime( ev.time - m_data.baseTime ); msg->time = time; msg->ref = StringRef( StringRef::Type::Idx, GetSingleStringIdx() ); msg->thread = CompressThread( m_threadCtx ); msg->color = 0xFFFFFFFF; msg->callstack.SetVal( 0 ); if( m_data.lastTime < time ) m_data.lastTime = time; InsertMessageData( msg ); } void Worker::ProcessMessageLiteral( const QueueMessageLiteral& ev ) { CheckString( ev.text ); auto msg = m_slab.Alloc(); const auto time = TscTime( ev.time - m_data.baseTime ); msg->time = time; msg->ref = StringRef( StringRef::Type::Ptr, ev.text ); msg->thread = CompressThread( m_threadCtx ); msg->color = 0xFFFFFFFF; msg->callstack.SetVal( 0 ); if( m_data.lastTime < time ) m_data.lastTime = time; InsertMessageData( msg ); } void Worker::ProcessMessageColor( const QueueMessageColor& ev ) { auto msg = m_slab.Alloc(); const auto time = TscTime( ev.time - m_data.baseTime ); msg->time = time; msg->ref = StringRef( StringRef::Type::Idx, GetSingleStringIdx() ); msg->thread = CompressThread( m_threadCtx ); msg->color = 0xFF000000 | ( ev.r << 16 ) | ( ev.g << 8 ) | ev.b; msg->callstack.SetVal( 0 ); if( m_data.lastTime < time ) m_data.lastTime = time; InsertMessageData( msg ); } void Worker::ProcessMessageLiteralColor( const QueueMessageColorLiteral& ev ) { CheckString( ev.text ); auto msg = m_slab.Alloc(); const auto time = TscTime( ev.time - m_data.baseTime ); msg->time = time; msg->ref = StringRef( StringRef::Type::Ptr, ev.text ); msg->thread = CompressThread( m_threadCtx ); msg->color = 0xFF000000 | ( ev.r << 16 ) | ( ev.g << 8 ) | ev.b; msg->callstack.SetVal( 0 ); if( m_data.lastTime < time ) m_data.lastTime = time; InsertMessageData( msg ); } void Worker::ProcessMessageCallstack( const QueueMessage& ev ) { ProcessMessage( ev ); auto& next = m_nextCallstack[m_threadCtx]; next.type = NextCallstackType::Message; } void Worker::ProcessMessageLiteralCallstack( const QueueMessageLiteral& ev ) { ProcessMessageLiteral( ev ); auto& next = m_nextCallstack[m_threadCtx]; next.type = NextCallstackType::Message; } void Worker::ProcessMessageColorCallstack( const QueueMessageColor& ev ) { ProcessMessageColor( ev ); auto& next = m_nextCallstack[m_threadCtx]; next.type = NextCallstackType::Message; } void Worker::ProcessMessageLiteralColorCallstack( const QueueMessageColorLiteral& ev ) { ProcessMessageLiteralColor( ev ); auto& next = m_nextCallstack[m_threadCtx]; next.type = NextCallstackType::Message; } void Worker::ProcessMessageAppInfo( const QueueMessage& ev ) { m_data.appInfo.push_back( StringRef( StringRef::Type::Idx, GetSingleStringIdx() ) ); const auto time = TscTime( ev.time - m_data.baseTime ); if( m_data.lastTime < time ) m_data.lastTime = time; } void Worker::ProcessGpuNewContext( const QueueGpuNewContext& ev ) { assert( !m_gpuCtxMap[ev.context] ); assert( ev.type != GpuContextType::Invalid ); int64_t gpuTime; if( ev.period == 1.f ) { gpuTime = ev.gpuTime; } else { gpuTime = int64_t( double( ev.period ) * ev.gpuTime ); // precision loss } const auto cpuTime = TscTime( ev.cpuTime - m_data.baseTime ); auto gpu = m_slab.AllocInit(); memset( gpu->query, 0, sizeof( gpu->query ) ); gpu->timeDiff = cpuTime - gpuTime; gpu->thread = ev.thread; gpu->period = ev.period; gpu->count = 0; gpu->type = ev.type; gpu->hasPeriod = ev.period != 1.f; gpu->hasCalibration = ev.flags & GpuContextCalibration; gpu->calibratedGpuTime = gpuTime; gpu->calibratedCpuTime = cpuTime; gpu->calibrationMod = 1.; m_data.gpuData.push_back( gpu ); m_gpuCtxMap[ev.context] = gpu; } void Worker::ProcessGpuZoneBeginImpl( GpuEvent* zone, const QueueGpuZoneBegin& ev, bool serial ) { m_data.gpuCnt++; auto ctx = m_gpuCtxMap[ev.context]; assert( ctx ); CheckSourceLocation( ev.srcloc ); int64_t cpuTime; if( serial ) { cpuTime = m_refTimeSerial + ev.cpuTime; m_refTimeSerial = cpuTime; } else { cpuTime = m_refTimeThread + ev.cpuTime; m_refTimeThread = cpuTime; } const auto time = TscTime( cpuTime - m_data.baseTime ); zone->SetCpuStart( time ); zone->SetCpuEnd( -1 ); zone->SetGpuStart( -1 ); zone->SetGpuEnd( -1 ); zone->SetSrcLoc( ShrinkSourceLocation( ev.srcloc ) ); zone->callstack.SetVal( 0 ); zone->SetChild( -1 ); uint64_t ztid; if( ctx->thread == 0 ) { // Vulkan, OpenCL and Direct3D 12 contexts are not bound to any single thread. zone->SetThread( CompressThread( ev.thread ) ); ztid = ev.thread; } else { // OpenGL doesn't need per-zone thread id. It still can be sent, // because it may be needed for callstack collection purposes. zone->SetThread( 0 ); ztid = 0; } if( m_data.lastTime < time ) m_data.lastTime = time; auto td = ctx->threadData.find( ztid ); if( td == ctx->threadData.end() ) { td = ctx->threadData.emplace( ztid, GpuCtxThreadData {} ).first; } auto timeline = &td->second.timeline; auto& stack = td->second.stack; if( !stack.empty() ) { auto back = stack.back(); if( back->Child() < 0 ) { back->SetChild( int32_t( m_data.gpuChildren.size() ) ); m_data.gpuChildren.push_back( Vector>() ); } timeline = &m_data.gpuChildren[back->Child()]; } timeline->push_back( zone ); stack.push_back( zone ); assert( !ctx->query[ev.queryId] ); ctx->query[ev.queryId] = zone; } void Worker::ProcessGpuZoneBegin( const QueueGpuZoneBegin& ev, bool serial ) { auto zone = m_slab.Alloc(); ProcessGpuZoneBeginImpl( zone, ev, serial ); } void Worker::ProcessGpuZoneBeginCallstack( const QueueGpuZoneBegin& ev, bool serial ) { auto zone = m_slab.Alloc(); ProcessGpuZoneBeginImpl( zone, ev, serial ); auto& next = m_nextCallstack[ev.thread]; next.type = NextCallstackType::Gpu; next.gpu = zone; } void Worker::ProcessGpuZoneEnd( const QueueGpuZoneEnd& ev, bool serial ) { auto ctx = m_gpuCtxMap[ev.context]; assert( ctx ); auto td = ctx->threadData.find( ev.thread ); assert( td != ctx->threadData.end() ); assert( !td->second.stack.empty() ); auto zone = td->second.stack.back_and_pop(); assert( !ctx->query[ev.queryId] ); ctx->query[ev.queryId] = zone; int64_t cpuTime; if( serial ) { cpuTime = m_refTimeSerial + ev.cpuTime; m_refTimeSerial = cpuTime; } else { cpuTime = m_refTimeThread + ev.cpuTime; m_refTimeThread = cpuTime; } const auto time = TscTime( cpuTime - m_data.baseTime ); zone->SetCpuEnd( time ); if( m_data.lastTime < time ) m_data.lastTime = time; } void Worker::ProcessGpuTime( const QueueGpuTime& ev ) { auto ctx = m_gpuCtxMap[ev.context]; assert( ctx ); const int64_t tref = m_refTimeGpu + ev.gpuTime; m_refTimeGpu = tref; const int64_t t = std::max( 0, tref ); int64_t gpuTime; if( !ctx->hasPeriod ) { if( !ctx->hasCalibration ) { gpuTime = t + ctx->timeDiff; } else { gpuTime = int64_t( ( t - ctx->calibratedGpuTime ) * ctx->calibrationMod + ctx->calibratedCpuTime ); } } else { if( !ctx->hasCalibration ) { gpuTime = int64_t( double( ctx->period ) * t ) + ctx->timeDiff; // precision loss } else { gpuTime = int64_t( ( double( ctx->period ) * t - ctx->calibratedGpuTime ) * ctx->calibrationMod + ctx->calibratedCpuTime ); } } auto zone = ctx->query[ev.queryId]; assert( zone ); ctx->query[ev.queryId] = nullptr; if( zone->GpuStart() < 0 ) { zone->SetGpuStart( gpuTime ); if( m_data.lastTime < gpuTime ) m_data.lastTime = gpuTime; ctx->count++; } else { if( gpuTime < zone->GpuStart() ) { auto tmp = zone->GpuStart(); std::swap( gpuTime, tmp ); zone->SetGpuStart( tmp ); } zone->SetGpuEnd( gpuTime ); if( m_data.lastTime < gpuTime ) m_data.lastTime = gpuTime; } } void Worker::ProcessGpuCalibration( const QueueGpuCalibration& ev ) { auto ctx = m_gpuCtxMap[ev.context]; assert( ctx ); assert( ctx->hasCalibration ); int64_t gpuTime; if( !ctx->hasPeriod ) { gpuTime = ev.gpuTime; } else { gpuTime = int64_t( double( ctx->period ) * ev.gpuTime ); // precision loss } const auto cpuDelta = ev.cpuDelta; const auto gpuDelta = gpuTime - ctx->calibratedGpuTime; ctx->calibrationMod = double( cpuDelta ) / gpuDelta; ctx->calibratedGpuTime = gpuTime; ctx->calibratedCpuTime = TscTime( ev.cpuTime - m_data.baseTime ); } void Worker::ProcessMemAlloc( const QueueMemAlloc& ev ) { const auto refTime = m_refTimeSerial + ev.time; m_refTimeSerial = refTime; const auto time = TscTime( refTime - m_data.baseTime ); if( m_data.lastTime < time ) m_data.lastTime = time; NoticeThread( ev.thread ); assert( m_data.memory.active.find( ev.ptr ) == m_data.memory.active.end() ); assert( m_data.memory.data.empty() || m_data.memory.data.back().TimeAlloc() <= time ); m_data.memory.active.emplace( ev.ptr, m_data.memory.data.size() ); const auto ptr = ev.ptr; uint32_t lo; uint16_t hi; memcpy( &lo, ev.size, 4 ); memcpy( &hi, ev.size+4, 2 ); const uint64_t size = lo | ( uint64_t( hi ) << 32 ); auto& mem = m_data.memory.data.push_next(); mem.SetPtr( ptr ); mem.SetSize( size ); mem.SetTimeThreadAlloc( time, CompressThread( ev.thread ) ); mem.SetTimeThreadFree( -1, 0 ); mem.SetCsAlloc( 0 ); mem.csFree.SetVal( 0 ); const auto low = m_data.memory.low; const auto high = m_data.memory.high; const auto ptrend = ptr + size; m_data.memory.low = std::min( low, ptr ); m_data.memory.high = std::max( high, ptrend ); m_data.memory.usage += size; MemAllocChanged( time ); } bool Worker::ProcessMemFree( const QueueMemFree& ev ) { const auto refTime = m_refTimeSerial + ev.time; m_refTimeSerial = refTime; if( ev.ptr == 0 ) return false; auto it = m_data.memory.active.find( ev.ptr ); if( it == m_data.memory.active.end() ) { if( !m_ignoreMemFreeFaults ) { CheckThreadString( ev.thread ); MemFreeFailure( ev.thread ); } return false; } const auto time = TscTime( refTime - m_data.baseTime ); if( m_data.lastTime < time ) m_data.lastTime = time; NoticeThread( ev.thread ); m_data.memory.frees.push_back( it->second ); auto& mem = m_data.memory.data[it->second]; mem.SetTimeThreadFree( time, CompressThread( ev.thread ) ); m_data.memory.usage -= mem.Size(); m_data.memory.active.erase( it ); MemAllocChanged( time ); return true; } void Worker::ProcessMemAllocCallstack( const QueueMemAlloc& ev ) { m_lastMemActionCallstack = m_data.memory.data.size(); ProcessMemAlloc( ev ); m_lastMemActionWasAlloc = true; } void Worker::ProcessMemFreeCallstack( const QueueMemFree& ev ) { if( ProcessMemFree( ev ) ) { m_lastMemActionCallstack = m_data.memory.frees.back(); m_lastMemActionWasAlloc = false; } else { m_lastMemActionCallstack = std::numeric_limits::max(); } } void Worker::ProcessCallstackMemory() { assert( m_pendingCallstackPtr != 0 ); m_pendingCallstackPtr = 0; if( m_lastMemActionCallstack != std::numeric_limits::max() ) { auto& mem = m_data.memory.data[m_lastMemActionCallstack]; if( m_lastMemActionWasAlloc ) { mem.SetCsAlloc( m_pendingCallstackId ); } else { mem.csFree.SetVal( m_pendingCallstackId ); } } } void Worker::ProcessCallstack() { assert( m_pendingCallstackPtr != 0 ); m_pendingCallstackPtr = 0; auto nit = m_nextCallstack.find( m_threadCtx ); assert( nit != m_nextCallstack.end() ); auto& next = nit->second; switch( next.type ) { case NextCallstackType::Zone: { auto& extra = RequestZoneExtra( *next.zone ); extra.callstack.SetVal( m_pendingCallstackId ); break; } case NextCallstackType::Gpu: next.gpu->callstack.SetVal( m_pendingCallstackId ); break; case NextCallstackType::Crash: m_data.crashEvent.callstack = m_pendingCallstackId; break; case NextCallstackType::Message: { auto td = m_threadCtxData; if( !td ) td = m_threadCtxData = RetrieveThread( m_threadCtx ); assert( td ); td->messages.back()->callstack.SetVal( m_pendingCallstackId ); break; } default: assert( false ); break; } } void Worker::ProcessCallstackAlloc() { assert( m_pendingCallstackPtr != 0 ); m_pendingCallstackPtr = 0; auto nit = m_nextCallstack.find( m_threadCtx ); assert( nit != m_nextCallstack.end() ); auto& next = nit->second; switch( next.type ) { case NextCallstackType::Zone: { auto& extra = RequestZoneExtra( *next.zone ); extra.callstack.SetVal( m_pendingCallstackId ); break; } case NextCallstackType::Gpu: next.gpu->callstack.SetVal( m_pendingCallstackId ); break; case NextCallstackType::Crash: m_data.crashEvent.callstack = m_pendingCallstackId; break; case NextCallstackType::Message: { auto td = m_threadCtxData; if( !td ) td = m_threadCtxData = RetrieveThread( m_threadCtx ); assert( td ); td->messages.back()->callstack.SetVal( m_pendingCallstackId ); break; } default: assert( false ); break; } } void Worker::ProcessCallstackSample( const QueueCallstackSample& ev ) { assert( m_pendingCallstackPtr != 0 ); m_pendingCallstackPtr = 0; m_data.samplesCnt++; const auto refTime = m_refTimeCtx + ev.time; m_refTimeCtx = refTime; const auto t = TscTime( refTime - m_data.baseTime ); SampleData sd; sd.time.SetVal( t ); sd.callstack.SetVal( m_pendingCallstackId ); auto td = NoticeThread( ev.thread ); if( td->samples.empty() ) { td->samples.push_back( sd ); } else { assert( td->samples.back().time.Val() < t ); td->samples.push_back_non_empty( sd ); } #ifndef TRACY_NO_STATISTICS const auto& cs = GetCallstack( m_pendingCallstackId ); { const auto& ip = cs[0]; auto frame = GetCallstackFrame( ip ); if( frame ) { const auto symAddr = frame->data[0].symAddr; auto it = m_data.instructionPointersMap.find( symAddr ); if( it == m_data.instructionPointersMap.end() ) { m_data.instructionPointersMap.emplace( symAddr, unordered_flat_map { { ip, 1 } } ); } else { auto fit = it->second.find( ip ); if( fit == it->second.end() ) { it->second.emplace( ip, 1 ); } else { fit->second++; } } auto sit = m_data.symbolSamples.find( symAddr ); if( sit == m_data.symbolSamples.end() ) { m_data.symbolSamples.emplace( symAddr, Vector( SampleDataRange { sd.time, ip } ) ); } else { if( sit->second.back().time.Val() <= sd.time.Val() ) { sit->second.push_back_non_empty( SampleDataRange { sd.time, ip } ); } else { auto iit = std::upper_bound( sit->second.begin(), sit->second.end(), sd.time.Val(), [] ( const auto& lhs, const auto& rhs ) { return lhs < rhs.time.Val(); } ); sit->second.insert( iit, SampleDataRange { sd.time, ip } ); } } } else { auto it = m_data.pendingInstructionPointers.find( ip ); if( it == m_data.pendingInstructionPointers.end() ) { m_data.pendingInstructionPointers.emplace( ip, 1 ); } else { it->second++; } auto sit = m_data.pendingSymbolSamples.find( ip ); if( sit == m_data.pendingSymbolSamples.end() ) { m_data.pendingSymbolSamples.emplace( ip, Vector( SampleDataRange { sd.time, ip } ) ); } else { sit->second.push_back_non_empty( SampleDataRange { sd.time, ip } ); } } } const auto framesKnown = UpdateSampleStatistics( m_pendingCallstackId, 1, true ); assert( td->samples.size() > td->ghostIdx ); if( framesKnown && td->ghostIdx + 1 == td->samples.size() ) { td->ghostIdx++; m_data.ghostCnt += AddGhostZone( cs, &td->ghostZones, t ); } else { m_data.ghostZonesPostponed = true; } #endif } void Worker::ProcessCallstackFrameSize( const QueueCallstackFrameSize& ev ) { assert( !m_callstackFrameStaging ); assert( m_pendingCallstackSubframes == 0 ); assert( m_pendingCallstackFrames > 0 ); m_pendingCallstackFrames--; m_pendingCallstackSubframes = ev.size; #ifndef TRACY_NO_STATISTICS m_data.newFramesWereReceived = true; #endif const auto idx = GetSingleStringIdx(); // Frames may be duplicated due to recursion auto fmit = m_data.callstackFrameMap.find( PackPointer( ev.ptr ) ); if( fmit == m_data.callstackFrameMap.end() ) { m_callstackFrameStaging = m_slab.Alloc(); m_callstackFrameStaging->size = ev.size; m_callstackFrameStaging->data = m_slab.Alloc( ev.size ); m_callstackFrameStaging->imageName = StringIdx( idx ); m_callstackFrameStagingPtr = ev.ptr; } } void Worker::ProcessCallstackFrame( const QueueCallstackFrame& ev ) { assert( m_pendingCallstackSubframes > 0 ); const auto nitidx = GetSingleStringIdx(); const auto fitidx = GetSecondStringIdx(); if( m_callstackFrameStaging ) { const auto idx = m_callstackFrameStaging->size - m_pendingCallstackSubframes; const auto name = StringIdx( nitidx ); const auto file = StringIdx( fitidx ); m_callstackFrameStaging->data[idx].name = name; m_callstackFrameStaging->data[idx].file = file; m_callstackFrameStaging->data[idx].line = ev.line; m_callstackFrameStaging->data[idx].symAddr = ev.symAddr; if( ev.symAddr != 0 && m_data.symbolMap.find( ev.symAddr ) == m_data.symbolMap.end() && m_pendingSymbols.find( ev.symAddr ) == m_pendingSymbols.end() ) { m_pendingSymbols.emplace( ev.symAddr, SymbolPending { name, m_callstackFrameStaging->imageName, file, ev.line, ev.symLen, idx < m_callstackFrameStaging->size - 1 } ); Query( ServerQuerySymbol, ev.symAddr ); } StringRef ref( StringRef::Idx, fitidx ); auto cit = m_checkedFileStrings.find( ref ); if( cit == m_checkedFileStrings.end() ) CacheSource( ref ); const auto frameId = PackPointer( m_callstackFrameStagingPtr ); #ifndef TRACY_NO_STATISTICS auto it = m_data.pendingInstructionPointers.find( frameId ); if( it != m_data.pendingInstructionPointers.end() ) { if( ev.symAddr != 0 ) { auto sit = m_data.instructionPointersMap.find( ev.symAddr ); if( sit == m_data.instructionPointersMap.end() ) { m_data.instructionPointersMap.emplace( ev.symAddr, unordered_flat_map { { it->first, it->second } } ); } else { assert( sit->second.find( it->first ) == sit->second.end() ); sit->second.emplace( it->first, it->second ); } } m_data.pendingInstructionPointers.erase( it ); } auto pit = m_data.pendingSymbolSamples.find( frameId ); if( pit != m_data.pendingSymbolSamples.end() ) { if( ev.symAddr != 0 ) { auto sit = m_data.symbolSamples.find( ev.symAddr ); if( sit == m_data.symbolSamples.end() ) { pdqsort_branchless( pit->second.begin(), pit->second.end(), [] ( const auto& lhs, const auto& rhs ) { return lhs.time.Val() < rhs.time.Val(); } ); m_data.symbolSamples.emplace( ev.symAddr, std::move( pit->second ) ); } else { for( auto& v : pit->second ) { if( sit->second.back().time.Val() <= v.time.Val() ) { sit->second.push_back_non_empty( v ); } else { auto iit = std::upper_bound( sit->second.begin(), sit->second.end(), v.time.Val(), [] ( const auto& lhs, const auto& rhs ) { return lhs < rhs.time.Val(); } ); sit->second.insert( iit, v ); } } } } m_data.pendingSymbolSamples.erase( pit ); } #endif if( --m_pendingCallstackSubframes == 0 ) { assert( m_data.callstackFrameMap.find( frameId ) == m_data.callstackFrameMap.end() ); m_data.callstackFrameMap.emplace( frameId, m_callstackFrameStaging ); m_callstackFrameStaging = nullptr; } } else { m_pendingCallstackSubframes--; } } void Worker::ProcessSymbolInformation( const QueueSymbolInformation& ev ) { auto it = m_pendingSymbols.find( ev.symAddr ); assert( it != m_pendingSymbols.end() ); const auto idx = GetSingleStringIdx(); SymbolData sd; sd.name = it->second.name; sd.file = StringIdx( idx ); sd.line = ev.line; sd.imageName = it->second.imageName; sd.callFile = it->second.file; sd.callLine = it->second.line; sd.isInline = it->second.isInline; sd.size.SetVal( it->second.size ); m_data.symbolMap.emplace( ev.symAddr, std::move( sd ) ); if( it->second.size > 0 && it->second.size <= 64*1024 ) { assert( m_pendingSymbolCode.find( ev.symAddr ) == m_pendingSymbolCode.end() ); m_pendingSymbolCode.emplace( ev.symAddr ); Query( ServerQuerySymbolCode, ev.symAddr, it->second.size ); } if( !it->second.isInline ) { if( !m_data.newSymbolsWereAdded ) m_data.newSymbolsWereAdded = true; m_data.symbolLoc.push_back( SymbolLocation { ev.symAddr, it->second.size } ); } else { if( !m_data.newInlineSymbolsWereAdded ) m_data.newInlineSymbolsWereAdded = true; m_data.symbolLocInline.push_back( ev.symAddr ); } StringRef ref( StringRef::Idx, idx ); auto cit = m_checkedFileStrings.find( ref ); if( cit == m_checkedFileStrings.end() ) CacheSource( ref ); m_pendingSymbols.erase( it ); } void Worker::ProcessCodeInformation( const QueueCodeInformation& ev ) { assert( m_pendingCodeInformation > 0 ); m_pendingCodeInformation--; const auto idx = GetSingleStringIdx(); if( ev.line != 0 ) { assert( m_data.codeAddressToLocation.find( ev.ptr ) == m_data.codeAddressToLocation.end() ); const auto packed = PackFileLine( idx, ev.line ); m_data.codeAddressToLocation.emplace( ev.ptr, packed ); auto lit = m_data.locationCodeAddressList.find( packed ); if( lit == m_data.locationCodeAddressList.end() ) { m_data.locationCodeAddressList.emplace( packed, Vector( ev.ptr ) ); } else { const bool needSort = lit->second.back() > ev.ptr; lit->second.push_back( ev.ptr ); if( needSort ) pdqsort_branchless( lit->second.begin(), lit->second.end() ); } StringRef ref( StringRef::Idx, idx ); auto cit = m_checkedFileStrings.find( ref ); if( cit == m_checkedFileStrings.end() ) CacheSource( ref ); } } void Worker::ProcessCrashReport( const QueueCrashReport& ev ) { CheckString( ev.text ); auto& next = m_nextCallstack[m_threadCtx]; next.type = NextCallstackType::Crash; m_data.crashEvent.thread = m_threadCtx; m_data.crashEvent.time = TscTime( ev.time - m_data.baseTime ); m_data.crashEvent.message = ev.text; m_data.crashEvent.callstack = 0; } void Worker::ProcessSysTime( const QueueSysTime& ev ) { const auto time = TscTime( ev.time - m_data.baseTime ); if( m_data.lastTime < time ) m_data.lastTime = time; const auto val = ev.sysTime; if( !m_sysTimePlot ) { m_sysTimePlot = m_slab.AllocInit(); m_sysTimePlot->name = 0; m_sysTimePlot->type = PlotType::SysTime; m_sysTimePlot->format = PlotValueFormatting::Percentage; m_sysTimePlot->min = val; m_sysTimePlot->max = val; m_sysTimePlot->data.push_back( { time, val } ); m_data.plots.Data().push_back( m_sysTimePlot ); } else { assert( !m_sysTimePlot->data.empty() ); assert( m_sysTimePlot->data.back().time.Val() <= time ); if( m_sysTimePlot->min > val ) m_sysTimePlot->min = val; else if( m_sysTimePlot->max < val ) m_sysTimePlot->max = val; m_sysTimePlot->data.push_back_non_empty( { time, val } ); } } void Worker::ProcessContextSwitch( const QueueContextSwitch& ev ) { const auto refTime = m_refTimeCtx + ev.time; m_refTimeCtx = refTime; const auto time = TscTime( refTime - m_data.baseTime ); if( m_data.lastTime < time ) m_data.lastTime = time; if( ev.cpu >= m_data.cpuDataCount ) m_data.cpuDataCount = ev.cpu + 1; auto& cs = m_data.cpuData[ev.cpu].cs; if( ev.oldThread != 0 ) { auto it = m_data.ctxSwitch.find( ev.oldThread ); if( it != m_data.ctxSwitch.end() ) { auto& data = it->second->v; assert( !data.empty() ); auto& item = data.back(); assert( item.Start() <= time ); assert( item.End() == -1 ); item.SetEnd( time ); item.SetReason( ev.reason ); item.SetState( ev.state ); const auto dt = time - item.Start(); it->second->runningTime += dt; auto tdit = m_data.cpuThreadData.find( ev.oldThread ); if( tdit == m_data.cpuThreadData.end() ) { tdit = m_data.cpuThreadData.emplace( ev.oldThread, CpuThreadData {} ).first; } tdit->second.runningRegions++; tdit->second.runningTime += dt; } if( !cs.empty() ) { auto& cx = cs.back(); assert( m_data.externalThreadCompress.DecompressThread( cx.Thread() ) == ev.oldThread ); cx.SetEnd( time ); } } if( ev.newThread != 0 ) { auto it = m_data.ctxSwitch.find( ev.newThread ); if( it == m_data.ctxSwitch.end() ) { auto ctx = m_slab.AllocInit(); it = m_data.ctxSwitch.emplace( ev.newThread, ctx ).first; } auto& data = it->second->v; ContextSwitchData* item = nullptr; bool migration = false; if( !data.empty() && data.back().Reason() == ContextSwitchData::Wakeup ) { item = &data.back(); if( data.size() > 1 ) { migration = data[data.size()-2].Cpu() != ev.cpu; } } else { assert( data.empty() || (uint64_t)data.back().End() <= (uint64_t)time ); if( !data.empty() ) { migration = data.back().Cpu() != ev.cpu; } item = &data.push_next(); item->SetWakeup( time ); } item->SetStart( time ); item->SetEnd( -1 ); item->SetCpu( ev.cpu ); item->SetReason( -1 ); item->SetState( -1 ); auto& cx = cs.push_next(); cx.SetStart( time ); cx.SetEnd( -1 ); cx.SetThread( m_data.externalThreadCompress.CompressThread( ev.newThread ) ); CheckExternalName( ev.newThread ); if( migration ) { auto tdit = m_data.cpuThreadData.find( ev.newThread ); if( tdit == m_data.cpuThreadData.end() ) { tdit = m_data.cpuThreadData.emplace( ev.newThread, CpuThreadData {} ).first; } tdit->second.migrations++; } } } void Worker::ProcessThreadWakeup( const QueueThreadWakeup& ev ) { const auto refTime = m_refTimeCtx + ev.time; m_refTimeCtx = refTime; const auto time = TscTime( refTime - m_data.baseTime ); if( m_data.lastTime < time ) m_data.lastTime = time; auto it = m_data.ctxSwitch.find( ev.thread ); if( it == m_data.ctxSwitch.end() ) { auto ctx = m_slab.AllocInit(); it = m_data.ctxSwitch.emplace( ev.thread, ctx ).first; } auto& data = it->second->v; if( !data.empty() && !data.back().IsEndValid() ) return; // wakeup of a running thread auto& item = data.push_next(); item.SetWakeup( time ); item.SetStart( time ); item.SetEnd( -1 ); item.SetCpu( 0 ); item.SetReason( ContextSwitchData::Wakeup ); item.SetState( -1 ); } void Worker::ProcessTidToPid( const QueueTidToPid& ev ) { if( m_data.tidToPid.find( ev.tid ) == m_data.tidToPid.end() ) m_data.tidToPid.emplace( ev.tid, ev.pid ); } void Worker::ProcessParamSetup( const QueueParamSetup& ev ) { CheckString( ev.name ); m_params.push_back( Parameter { ev.idx, StringRef( StringRef::Ptr, ev.name ), bool( ev.isBool ), ev.val } ); } void Worker::ProcessCpuTopology( const QueueCpuTopology& ev ) { auto package = m_data.cpuTopology.find( ev.package ); if( package == m_data.cpuTopology.end() ) package = m_data.cpuTopology.emplace( ev.package, unordered_flat_map> {} ).first; auto core = package->second.find( ev.core ); if( core == package->second.end() ) core = package->second.emplace( ev.core, std::vector {} ).first; core->second.emplace_back( ev.thread ); assert( m_data.cpuTopologyMap.find( ev.thread ) == m_data.cpuTopologyMap.end() ); m_data.cpuTopologyMap.emplace( ev.thread, CpuThreadTopology { ev.package, ev.core } ); } void Worker::MemAllocChanged( int64_t time ) { const auto val = (double)m_data.memory.usage; if( !m_data.memory.plot ) { CreateMemAllocPlot(); m_data.memory.plot->min = val; m_data.memory.plot->max = val; m_data.memory.plot->data.push_back( { time, val } ); } else { assert( !m_data.memory.plot->data.empty() ); assert( m_data.memory.plot->data.back().time.Val() <= time ); if( m_data.memory.plot->min > val ) m_data.memory.plot->min = val; else if( m_data.memory.plot->max < val ) m_data.memory.plot->max = val; m_data.memory.plot->data.push_back_non_empty( { time, val } ); } } void Worker::CreateMemAllocPlot() { assert( !m_data.memory.plot ); m_data.memory.plot = m_slab.AllocInit(); m_data.memory.plot->name = 0; m_data.memory.plot->type = PlotType::Memory; m_data.memory.plot->format = PlotValueFormatting::Memory; m_data.memory.plot->data.push_back( { GetFrameBegin( *m_data.framesBase, 0 ), 0. } ); m_data.plots.Data().push_back( m_data.memory.plot ); } void Worker::ReconstructMemAllocPlot() { auto& mem = m_data.memory; #ifdef NO_PARALLEL_SORT pdqsort_branchless( mem.frees.begin(), mem.frees.end(), [&mem] ( const auto& lhs, const auto& rhs ) { return mem.data[lhs].TimeFree() < mem.data[rhs].TimeFree(); } ); #else std::sort( std::execution::par_unseq, mem.frees.begin(), mem.frees.end(), [&mem] ( const auto& lhs, const auto& rhs ) { return mem.data[lhs].TimeFree() < mem.data[rhs].TimeFree(); } ); #endif const auto psz = mem.data.size() + mem.frees.size() + 1; PlotData* plot; { std::lock_guard lock( m_data.lock ); plot = m_slab.AllocInit(); } plot->name = 0; plot->type = PlotType::Memory; plot->format = PlotValueFormatting::Memory; plot->data.reserve_exact( psz, m_slab ); auto aptr = mem.data.begin(); auto aend = mem.data.end(); auto fptr = mem.frees.begin(); auto fend = mem.frees.end(); double max = 0; double usage = 0; auto ptr = plot->data.data(); ptr->time = GetFrameBegin( *m_data.framesBase, 0 ); ptr->val = 0; ptr++; if( aptr != aend && fptr != fend ) { auto atime = aptr->TimeAlloc(); auto ftime = mem.data[*fptr].TimeFree(); for(;;) { if( atime < ftime ) { usage += int64_t( aptr->Size() ); assert( usage >= 0 ); if( max < usage ) max = usage; ptr->time = atime; ptr->val = usage; ptr++; aptr++; if( aptr == aend ) break; atime = aptr->TimeAlloc(); } else { usage -= int64_t( mem.data[*fptr].Size() ); assert( usage >= 0 ); if( max < usage ) max = usage; ptr->time = ftime; ptr->val = usage; ptr++; fptr++; if( fptr == fend ) break; ftime = mem.data[*fptr].TimeFree(); } } } while( aptr != aend ) { assert( aptr->TimeFree() < 0 ); int64_t time = aptr->TimeAlloc(); usage += int64_t( aptr->Size() ); assert( usage >= 0 ); if( max < usage ) max = usage; ptr->time = time; ptr->val = usage; ptr++; aptr++; } while( fptr != fend ) { const auto& memData = mem.data[*fptr]; int64_t time = memData.TimeFree(); usage -= int64_t( memData.Size() ); assert( usage >= 0 ); assert( max >= usage ); ptr->time = time; ptr->val = usage; ptr++; fptr++; } plot->min = 0; plot->max = max; std::lock_guard lock( m_data.lock ); m_data.plots.Data().insert( m_data.plots.Data().begin(), plot ); m_data.memory.plot = plot; } #ifndef TRACY_NO_STATISTICS void Worker::ReconstructContextSwitchUsage() { assert( m_data.cpuDataCount != 0 ); const auto cpucnt = m_data.cpuDataCount; auto& vec = m_data.ctxUsage; vec.push_back( ContextSwitchUsage( 0, 0, 0 ) ); struct Cpu { bool startDone; Vector::iterator it; Vector::iterator end; }; std::vector cpus; cpus.reserve( cpucnt ); for( int i=0; i::max(); bool atEnd = true; for( int i=0; iStart() : cpus[i].it->End(); if( ct < nextTime ) nextTime = ct; } } if( atEnd ) break; for( int i=0; iStart() : cpus[i].it->End(); if( nextTime != ct ) break; const auto isOwn = GetPidFromTid( DecompressThreadExternal( cpus[i].it->Thread() ) ) == m_pid; if( !cpus[i].startDone ) { if( isOwn ) { own++; assert( own <= cpucnt ); } else { other++; assert( other <= cpucnt ); } if( !cpus[i].it->IsEndValid() ) { cpus[i].it++; assert( cpus[i].it = cpus[i].end ); } else { cpus[i].startDone = true; } } else { if( isOwn ) { assert( own > 0 ); own--; } else { assert( other > 0 ); other--; } cpus[i].startDone = false; cpus[i].it++; } } } const auto& back = vec.back(); if( back.Other() != other || back.Own() != own ) { vec.push_back( ContextSwitchUsage( nextTime, other, own ) ); } } std::lock_guard lock( m_data.lock ); m_data.ctxUsageReady = true; } bool Worker::UpdateSampleStatistics( uint32_t callstack, uint32_t count, bool canPostpone ) { const auto& cs = GetCallstack( callstack ); const auto cssz = cs.size(); auto frames = (const CallstackFrameData**)alloca( cssz * sizeof( CallstackFrameData* ) ); for( uint16_t i=0; isecond += count; } } return false; } else { frames[i] = frame; } } if( canPostpone ) { auto it = m_data.postponedSamples.find( callstack ); if( it != m_data.postponedSamples.end() ) { count += it->second; m_data.postponedSamples.erase( it ); } } UpdateSampleStatisticsImpl( frames, cssz, count, cs ); return true; } void Worker::UpdateSampleStatisticsPostponed( decltype(Worker::DataBlock::postponedSamples.begin())& it ) { const auto& cs = GetCallstack( it->first ); const auto cssz = cs.size(); auto frames = (const CallstackFrameData**)alloca( cssz * sizeof( CallstackFrameData* ) ); for( uint16_t i=0; isecond, cs ); it = m_data.postponedSamples.erase( it ); } void Worker::UpdateSampleStatisticsImpl( const CallstackFrameData** frames, uint16_t framesCount, uint32_t count, const VarArray& cs ) { const auto fexcl = frames[0]; const auto fxsz = fexcl->size; const auto& frame0 = fexcl->data[0]; auto sym0 = m_data.symbolStats.find( frame0.symAddr ); if( sym0 == m_data.symbolStats.end() ) sym0 = m_data.symbolStats.emplace( frame0.symAddr, SymbolStats { 0, 0, unordered_flat_map() } ).first; sym0->second.excl += count; for( uint8_t f=1; fdata[f]; auto sym = m_data.symbolStats.find( frame.symAddr ); if( sym == m_data.symbolStats.end() ) sym = m_data.symbolStats.emplace( frame.symAddr, SymbolStats { 0, 0, unordered_flat_map() } ).first; sym->second.incl += count; } for( uint16_t c=1; csize; for( uint8_t f=0; fdata[f]; auto sym = m_data.symbolStats.find( frame.symAddr ); if( sym == m_data.symbolStats.end() ) sym = m_data.symbolStats.emplace( frame.symAddr, SymbolStats { 0, 0, unordered_flat_map() } ).first; sym->second.incl += count; } } CallstackFrameId parentFrameId; if( fxsz != 1 ) { auto cfdata = (CallstackFrame*)alloca( ( fxsz-1 ) * sizeof( CallstackFrame ) ); for( int i=0; idata[i+1]; } CallstackFrameData cfd; cfd.data = cfdata; cfd.size = fxsz-1; cfd.imageName = fexcl->imageName; auto it = m_data.revParentFrameMap.find( &cfd ); if( it == m_data.revParentFrameMap.end() ) { auto frame = m_slab.Alloc( fxsz-1 ); memcpy( frame, cfdata, ( fxsz-1 ) * sizeof( CallstackFrame ) ); auto frameData = m_slab.AllocInit(); frameData->data = frame; frameData->size = fxsz - 1; frameData->imageName = fexcl->imageName; parentFrameId.idx = m_callstackParentNextIdx++; parentFrameId.sel = 0; parentFrameId.custom = 1; m_data.parentCallstackFrameMap.emplace( parentFrameId, frameData ); m_data.revParentFrameMap.emplace( frameData, parentFrameId ); } else { parentFrameId = it->second; } } const auto sz = framesCount - ( fxsz == 1 ); const auto memsize = sizeof( VarArray ) + sz * sizeof( CallstackFrameId ); auto mem = (char*)m_slab.AllocRaw( memsize ); auto data = (CallstackFrameId*)mem; auto dst = data; if( fxsz == 1 ) { for( int i=0; i*)( mem + sz * sizeof( CallstackFrameId ) ); new(arr) VarArray( sz, data ); uint32_t idx; auto it = m_data.parentCallstackMap.find( arr ); if( it == m_data.parentCallstackMap.end() ) { idx = m_data.parentCallstackPayload.size(); m_data.parentCallstackMap.emplace( arr, idx ); m_data.parentCallstackPayload.push_back( arr ); } else { idx = it->second; m_slab.Unalloc( memsize ); } sym0 = m_data.symbolStats.find( frame0.symAddr ); auto sit = sym0->second.parents.find( idx ); if( sit == sym0->second.parents.end() ) { sym0->second.parents.emplace( idx, count ); } else { sit->second += count; } } #endif int64_t Worker::ReadTimeline( FileRead& f, ZoneEvent* zone, int64_t refTime, int32_t& childIdx ) { uint32_t sz; f.Read( sz ); return ReadTimelineHaveSize( f, zone, refTime, childIdx, sz ); } int64_t Worker::ReadTimelineHaveSize( FileRead& f, ZoneEvent* zone, int64_t refTime, int32_t& childIdx, uint32_t sz ) { if( sz == 0 ) { zone->SetChild( -1 ); return refTime; } else { const auto idx = childIdx; childIdx++; zone->SetChild( idx ); return ReadTimeline( f, m_data.zoneChildren[idx], sz, refTime, childIdx ); } } void Worker::ReadTimelinePre063( FileRead& f, ZoneEvent* zone, int64_t& refTime, int32_t& childIdx, int fileVer ) { uint64_t sz; f.Read( sz ); if( sz == 0 ) { zone->SetChild( -1 ); } else { const auto idx = childIdx; childIdx++; zone->SetChild( idx ); ReadTimelinePre063( f, m_data.zoneChildren[idx], sz, refTime, childIdx, fileVer ); } } void Worker::ReadTimeline( FileRead& f, GpuEvent* zone, int64_t& refTime, int64_t& refGpuTime, int32_t& childIdx ) { uint64_t sz; f.Read( sz ); ReadTimelineHaveSize( f, zone, refTime, refGpuTime, childIdx, sz ); } void Worker::ReadTimelineHaveSize( FileRead& f, GpuEvent* zone, int64_t& refTime, int64_t& refGpuTime, int32_t& childIdx, uint64_t sz ) { if( sz == 0 ) { zone->SetChild( -1 ); } else { const auto idx = childIdx; childIdx++; zone->SetChild( idx ); ReadTimeline( f, m_data.gpuChildren[idx], sz, refTime, refGpuTime, childIdx ); } } #ifndef TRACY_NO_STATISTICS void Worker::ReconstructZoneStatistics( ZoneEvent& zone, uint16_t thread ) { assert( zone.IsEndValid() ); auto timeSpan = zone.End() - zone.Start(); if( timeSpan > 0 ) { auto it = m_data.sourceLocationZones.find( zone.SrcLoc() ); assert( it != m_data.sourceLocationZones.end() ); auto& slz = it->second; auto& ztd = slz.zones.push_next(); ztd.SetZone( &zone ); ztd.SetThread( thread ); if( slz.min > timeSpan ) slz.min = timeSpan; if( slz.max < timeSpan ) slz.max = timeSpan; slz.total += timeSpan; slz.sumSq += double( timeSpan ) * timeSpan; if( zone.HasChildren() ) { auto& children = GetZoneChildren( zone.Child() ); assert( children.is_magic() ); auto& c = *(Vector*)( &children ); for( auto& v : c ) { const auto childSpan = std::max( int64_t( 0 ), v.End() - v.Start() ); timeSpan -= childSpan; } } if( slz.selfMin > timeSpan ) slz.selfMin = timeSpan; if( slz.selfMax < timeSpan ) slz.selfMax = timeSpan; slz.selfTotal += timeSpan; } } #else void Worker::CountZoneStatistics( ZoneEvent* zone ) { auto cnt = GetSourceLocationZonesCnt( zone->SrcLoc() ); (*cnt)++; } #endif int64_t Worker::ReadTimeline( FileRead& f, Vector>& _vec, uint32_t size, int64_t refTime, int32_t& childIdx ) { assert( size != 0 ); const auto lp = s_loadProgress.subProgress.load( std::memory_order_relaxed ); s_loadProgress.subProgress.store( lp + size, std::memory_order_relaxed ); auto& vec = *(Vector*)( &_vec ); vec.set_magic(); vec.reserve_exact( size, m_slab ); auto zone = vec.begin(); auto end = vec.end() - 1; int16_t srcloc; int64_t tstart, tend; uint32_t childSz, extra; f.Read4( srcloc, tstart, extra, childSz ); while( zone != end ) { refTime += tstart; zone->SetStartSrcLoc( refTime, srcloc ); zone->extra = extra; refTime = ReadTimelineHaveSize( f, zone, refTime, childIdx, childSz ); f.Read5( tend, srcloc, tstart, extra, childSz ); refTime += tend; zone->SetEnd( refTime ); #ifdef TRACY_NO_STATISTICS CountZoneStatistics( zone ); #endif zone++; } refTime += tstart; zone->SetStartSrcLoc( refTime, srcloc ); zone->extra = extra; refTime = ReadTimelineHaveSize( f, zone, refTime, childIdx, childSz ); f.Read( tend ); refTime += tend; zone->SetEnd( refTime ); #ifdef TRACY_NO_STATISTICS CountZoneStatistics( zone ); #endif return refTime; } void Worker::ReadTimelinePre063( FileRead& f, Vector>& _vec, uint64_t size, int64_t& refTime, int32_t& childIdx, int fileVer ) { assert( fileVer < FileVersion( 0, 6, 3 ) ); assert( size != 0 ); const auto lp = s_loadProgress.subProgress.load( std::memory_order_relaxed ); s_loadProgress.subProgress.store( lp + size, std::memory_order_relaxed ); auto& vec = *(Vector*)( &_vec ); vec.set_magic(); vec.reserve_exact( size, m_slab ); auto zone = vec.begin(); auto end = vec.end(); do { int16_t srcloc; f.Read( srcloc ); zone->SetSrcLoc( srcloc ); f.Read( &zone->_end_child1, sizeof( zone->_end_child1 ) ); ZoneExtra extra; f.Read( &extra.text, sizeof( extra.text ) ); f.Read( &extra.callstack, sizeof( extra.callstack ) ); f.Read( &extra.name, sizeof( extra.name ) ); zone->extra = 0; if( extra.callstack.Val() != 0 || extra.name.Active() || extra.text.Active() ) { memcpy( &AllocZoneExtra( *zone ), &extra, sizeof( ZoneExtra ) ); } refTime += zone->_end_child1; zone->SetStart( refTime - m_data.baseTime ); ReadTimelinePre063( f, zone, refTime, childIdx, fileVer ); int64_t end = ReadTimeOffset( f, refTime ); if( end >= 0 ) end -= m_data.baseTime; zone->SetEnd( end ); #ifdef TRACY_NO_STATISTICS CountZoneStatistics( zone ); #endif } while( ++zone != end ); } void Worker::ReadTimeline( FileRead& f, Vector>& _vec, uint64_t size, int64_t& refTime, int64_t& refGpuTime, int32_t& childIdx ) { assert( size != 0 ); const auto lp = s_loadProgress.subProgress.load( std::memory_order_relaxed ); s_loadProgress.subProgress.store( lp + size, std::memory_order_relaxed ); auto& vec = *(Vector*)( &_vec ); vec.set_magic(); vec.reserve_exact( size, m_slab ); auto zone = vec.begin(); auto end = vec.end(); do { int64_t tcpu, tgpu; int16_t srcloc; uint16_t thread; uint64_t childSz; f.Read6( tcpu, tgpu, srcloc, zone->callstack, thread, childSz ); zone->SetSrcLoc( srcloc ); zone->SetThread( thread ); refTime += tcpu; refGpuTime += tgpu; zone->SetCpuStart( refTime ); zone->SetGpuStart( refGpuTime ); ReadTimelineHaveSize( f, zone, refTime, refGpuTime, childIdx, childSz ); f.Read2( tcpu, tgpu ); refTime += tcpu; refGpuTime += tgpu; zone->SetCpuEnd( refTime ); zone->SetGpuEnd( refGpuTime ); } while( ++zone != end ); } void Worker::Disconnect() { Query( ServerQueryDisconnect, 0 ); m_disconnect = true; } void Worker::Write( FileWrite& f ) { f.Write( FileHeader, sizeof( FileHeader ) ); f.Write( &m_delay, sizeof( m_delay ) ); f.Write( &m_resolution, sizeof( m_resolution ) ); f.Write( &m_timerMul, sizeof( m_timerMul ) ); f.Write( &m_data.lastTime, sizeof( m_data.lastTime ) ); f.Write( &m_data.frameOffset, sizeof( m_data.frameOffset ) ); f.Write( &m_pid, sizeof( m_pid ) ); f.Write( &m_samplingPeriod, sizeof( m_samplingPeriod ) ); f.Write( &m_data.cpuArch, sizeof( m_data.cpuArch ) ); f.Write( &m_data.cpuId, sizeof( m_data.cpuId ) ); f.Write( m_data.cpuManufacturer, 12 ); uint64_t sz = m_captureName.size(); f.Write( &sz, sizeof( sz ) ); f.Write( m_captureName.c_str(), sz ); sz = m_captureProgram.size(); f.Write( &sz, sizeof( sz ) ); f.Write( m_captureProgram.c_str(), sz ); f.Write( &m_captureTime, sizeof( m_captureTime ) ); sz = m_hostInfo.size(); f.Write( &sz, sizeof( sz ) ); f.Write( m_hostInfo.c_str(), sz ); sz = m_data.cpuTopology.size(); f.Write( &sz, sizeof( sz ) ); for( auto& package : m_data.cpuTopology ) { sz = package.second.size(); f.Write( &package.first, sizeof( package.first ) ); f.Write( &sz, sizeof( sz ) ); for( auto& core : package.second ) { sz = core.second.size(); f.Write( &core.first, sizeof( core.first ) ); f.Write( &sz, sizeof( sz ) ); for( auto& thread : core.second ) { f.Write( &thread, sizeof( thread ) ); } } } f.Write( &m_data.crashEvent, sizeof( m_data.crashEvent ) ); sz = m_data.frames.Data().size(); f.Write( &sz, sizeof( sz ) ); for( auto& fd : m_data.frames.Data() ) { int64_t refTime = 0; f.Write( &fd->name, sizeof( fd->name ) ); f.Write( &fd->continuous, sizeof( fd->continuous ) ); sz = fd->frames.size(); f.Write( &sz, sizeof( sz ) ); if( fd->continuous ) { for( auto& fe : fd->frames ) { WriteTimeOffset( f, refTime, fe.start ); f.Write( &fe.frameImage, sizeof( fe.frameImage ) ); } } else { for( auto& fe : fd->frames ) { WriteTimeOffset( f, refTime, fe.start ); WriteTimeOffset( f, refTime, fe.end ); f.Write( &fe.frameImage, sizeof( fe.frameImage ) ); } } } sz = m_data.stringData.size(); f.Write( &sz, sizeof( sz ) ); for( auto& v : m_data.stringData ) { uint64_t ptr = (uint64_t)v; f.Write( &ptr, sizeof( ptr ) ); sz = strlen( v ); f.Write( &sz, sizeof( sz ) ); f.Write( v, sz ); } sz = m_data.strings.size(); f.Write( &sz, sizeof( sz ) ); for( auto& v : m_data.strings ) { f.Write( &v.first, sizeof( v.first ) ); uint64_t ptr = (uint64_t)v.second; f.Write( &ptr, sizeof( ptr ) ); } sz = m_data.threadNames.size(); f.Write( &sz, sizeof( sz ) ); for( auto& v : m_data.threadNames ) { f.Write( &v.first, sizeof( v.first ) ); uint64_t ptr = (uint64_t)v.second; f.Write( &ptr, sizeof( ptr ) ); } sz = m_data.externalNames.size(); f.Write( &sz, sizeof( sz ) ); for( auto& v : m_data.externalNames ) { f.Write( &v.first, sizeof( v.first ) ); uint64_t ptr = (uint64_t)v.second.first; f.Write( &ptr, sizeof( ptr ) ); ptr = (uint64_t)v.second.second; f.Write( &ptr, sizeof( ptr ) ); } m_data.localThreadCompress.Save( f ); m_data.externalThreadCompress.Save( f ); sz = m_data.sourceLocation.size(); f.Write( &sz, sizeof( sz ) ); for( auto& v : m_data.sourceLocation ) { f.Write( &v.first, sizeof( v.first ) ); f.Write( &v.second, sizeof( SourceLocationBase ) ); } sz = m_data.sourceLocationExpand.size(); f.Write( &sz, sizeof( sz ) ); for( auto& v : m_data.sourceLocationExpand ) { f.Write( &v, sizeof( v ) ); } sz = m_data.sourceLocationPayload.size(); f.Write( &sz, sizeof( sz ) ); for( auto& v : m_data.sourceLocationPayload ) { f.Write( v, sizeof( SourceLocationBase ) ); } #ifndef TRACY_NO_STATISTICS sz = m_data.sourceLocationZones.size(); f.Write( &sz, sizeof( sz ) ); for( auto& v : m_data.sourceLocationZones ) { int16_t id = v.first; uint64_t cnt = v.second.zones.size(); f.Write( &id, sizeof( id ) ); f.Write( &cnt, sizeof( cnt ) ); } #else sz = m_data.sourceLocationZonesCnt.size(); f.Write( &sz, sizeof( sz ) ); for( auto& v : m_data.sourceLocationZonesCnt ) { int16_t id = v.first; uint64_t cnt = v.second; f.Write( &id, sizeof( id ) ); f.Write( &cnt, sizeof( cnt ) ); } #endif sz = m_data.lockMap.size(); f.Write( &sz, sizeof( sz ) ); for( auto& v : m_data.lockMap ) { f.Write( &v.first, sizeof( v.first ) ); f.Write( &v.second->customName, sizeof( v.second->customName ) ); f.Write( &v.second->srcloc, sizeof( v.second->srcloc ) ); f.Write( &v.second->type, sizeof( v.second->type ) ); f.Write( &v.second->valid, sizeof( v.second->valid ) ); f.Write( &v.second->timeAnnounce, sizeof( v.second->timeAnnounce ) ); f.Write( &v.second->timeTerminate, sizeof( v.second->timeTerminate ) ); sz = v.second->threadList.size(); f.Write( &sz, sizeof( sz ) ); for( auto& t : v.second->threadList ) { f.Write( &t, sizeof( t ) ); } int64_t refTime = v.second->timeAnnounce; sz = v.second->timeline.size(); f.Write( &sz, sizeof( sz ) ); for( auto& lev : v.second->timeline ) { WriteTimeOffset( f, refTime, lev.ptr->Time() ); const int16_t srcloc = lev.ptr->SrcLoc(); f.Write( &srcloc, sizeof( srcloc ) ); f.Write( &lev.ptr->thread, sizeof( lev.ptr->thread ) ); f.Write( &lev.ptr->type, sizeof( lev.ptr->type ) ); } } { int64_t refTime = 0; sz = m_data.messages.size(); f.Write( &sz, sizeof( sz ) ); for( auto& v : m_data.messages ) { const auto ptr = (uint64_t)(MessageData*)v; f.Write( &ptr, sizeof( ptr ) ); WriteTimeOffset( f, refTime, v->time ); f.Write( &v->ref, sizeof( v->ref ) ); f.Write( &v->color, sizeof( v->color ) ); f.Write( &v->callstack, sizeof( v->callstack ) ); } } sz = m_data.zoneExtra.size(); f.Write( &sz, sizeof( sz ) ); f.Write( m_data.zoneExtra.data(), sz * sizeof( ZoneExtra ) ); sz = 0; for( auto& v : m_data.threads ) sz += v->count; f.Write( &sz, sizeof( sz ) ); sz = m_data.zoneChildren.size(); f.Write( &sz, sizeof( sz ) ); sz = m_data.threads.size(); f.Write( &sz, sizeof( sz ) ); for( auto& thread : m_data.threads ) { int64_t refTime = 0; f.Write( &thread->id, sizeof( thread->id ) ); f.Write( &thread->count, sizeof( thread->count ) ); WriteTimeline( f, thread->timeline, refTime ); sz = thread->messages.size(); f.Write( &sz, sizeof( sz ) ); for( auto& v : thread->messages ) { auto ptr = uint64_t( (MessageData*)v ); f.Write( &ptr, sizeof( ptr ) ); } sz = thread->samples.size(); f.Write( &sz, sizeof( sz ) ); refTime = 0; for( auto& v : thread->samples ) { WriteTimeOffset( f, refTime, v.time.Val() ); f.Write( &v.callstack, sizeof( v.callstack ) ); } } sz = 0; for( auto& v : m_data.gpuData ) sz += v->count; f.Write( &sz, sizeof( sz ) ); sz = m_data.gpuChildren.size(); f.Write( &sz, sizeof( sz ) ); sz = m_data.gpuData.size(); f.Write( &sz, sizeof( sz ) ); for( auto& ctx : m_data.gpuData ) { f.Write( &ctx->thread, sizeof( ctx->thread ) ); uint8_t calibration = ctx->hasCalibration; f.Write( &calibration, sizeof( calibration ) ); f.Write( &ctx->count, sizeof( ctx->count ) ); f.Write( &ctx->period, sizeof( ctx->period ) ); f.Write( &ctx->type, sizeof( ctx->type ) ); sz = ctx->threadData.size(); f.Write( &sz, sizeof( sz ) ); for( auto& td : ctx->threadData ) { int64_t refTime = 0; int64_t refGpuTime = 0; uint64_t tid = td.first; f.Write( &tid, sizeof( tid ) ); WriteTimeline( f, td.second.timeline, refTime, refGpuTime ); } } sz = m_data.plots.Data().size(); for( auto& plot : m_data.plots.Data() ) { if( plot->type == PlotType::Memory ) sz--; } f.Write( &sz, sizeof( sz ) ); for( auto& plot : m_data.plots.Data() ) { if( plot->type == PlotType::Memory ) continue; f.Write( &plot->type, sizeof( plot->type ) ); f.Write( &plot->format, sizeof( plot->format ) ); f.Write( &plot->name, sizeof( plot->name ) ); f.Write( &plot->min, sizeof( plot->min ) ); f.Write( &plot->max, sizeof( plot->max ) ); int64_t refTime = 0; sz = plot->data.size(); f.Write( &sz, sizeof( sz ) ); for( auto& v : plot->data ) { WriteTimeOffset( f, refTime, v.time.Val() ); f.Write( &v.val, sizeof( v.val ) ); } } { int64_t refTime = 0; sz = m_data.memory.data.size(); f.Write( &sz, sizeof( sz ) ); sz = m_data.memory.active.size(); f.Write( &sz, sizeof( sz ) ); sz = m_data.memory.frees.size(); f.Write( &sz, sizeof( sz ) ); for( auto& mem : m_data.memory.data ) { const auto ptr = mem.Ptr(); const auto size = mem.Size(); const Int24 csAlloc = mem.CsAlloc(); f.Write( &ptr, sizeof( ptr ) ); f.Write( &size, sizeof( size ) ); f.Write( &csAlloc, sizeof( csAlloc ) ); f.Write( &mem.csFree, sizeof( mem.csFree ) ); int64_t timeAlloc = mem.TimeAlloc(); uint16_t threadAlloc = mem.ThreadAlloc(); int64_t timeFree = mem.TimeFree(); uint16_t threadFree = mem.ThreadFree(); WriteTimeOffset( f, refTime, timeAlloc ); int64_t freeOffset = timeFree < 0 ? timeFree : timeFree - timeAlloc; f.Write( &freeOffset, sizeof( freeOffset ) ); f.Write( &threadAlloc, sizeof( threadAlloc ) ); f.Write( &threadFree, sizeof( threadFree ) ); } f.Write( &m_data.memory.high, sizeof( m_data.memory.high ) ); f.Write( &m_data.memory.low, sizeof( m_data.memory.low ) ); f.Write( &m_data.memory.usage, sizeof( m_data.memory.usage ) ); } sz = m_data.callstackPayload.size() - 1; f.Write( &sz, sizeof( sz ) ); for( size_t i=1; i<=sz; i++ ) { auto cs = m_data.callstackPayload[i]; uint16_t csz = cs->size(); f.Write( &csz, sizeof( csz ) ); f.Write( cs->data(), sizeof( CallstackFrameId ) * csz ); } sz = m_data.callstackFrameMap.size(); f.Write( &sz, sizeof( sz ) ); for( auto& frame : m_data.callstackFrameMap ) { f.Write( &frame.first, sizeof( CallstackFrameId ) ); f.Write( &frame.second->size, sizeof( frame.second->size ) ); f.Write( &frame.second->imageName, sizeof( frame.second->imageName ) ); f.Write( frame.second->data, sizeof( CallstackFrame ) * frame.second->size ); } sz = m_data.appInfo.size(); f.Write( &sz, sizeof( sz ) ); if( sz != 0 ) f.Write( m_data.appInfo.data(), sizeof( m_data.appInfo[0] ) * sz ); { TextureCompression texcomp; sz = m_data.frameImage.size(); f.Write( &sz, sizeof( sz ) ); for( auto& fi : m_data.frameImage ) { f.Write( &fi->w, sizeof( fi->w ) ); f.Write( &fi->h, sizeof( fi->h ) ); f.Write( &fi->flip, sizeof( fi->flip ) ); const auto image = texcomp.Unpack( *fi ); f.Write( image, fi->w * fi->h / 2 ); } } // Only save context switches relevant to active threads. std::vector::const_iterator> ctxValid; ctxValid.reserve( m_data.ctxSwitch.size() ); for( auto it = m_data.ctxSwitch.begin(); it != m_data.ctxSwitch.end(); ++it ) { auto td = RetrieveThread( it->first ); if( td && ( td->count > 0 || !td->samples.empty() ) ) { ctxValid.emplace_back( it ); } } sz = ctxValid.size(); f.Write( &sz, sizeof( sz ) ); for( auto& ctx : ctxValid ) { f.Write( &ctx->first, sizeof( ctx->first ) ); sz = ctx->second->v.size(); f.Write( &sz, sizeof( sz ) ); int64_t refTime = 0; for( auto& cs : ctx->second->v ) { WriteTimeOffset( f, refTime, cs.WakeupVal() ); WriteTimeOffset( f, refTime, cs.Start() ); WriteTimeOffset( f, refTime, cs.End() ); uint8_t cpu = cs.Cpu(); int8_t reason = cs.Reason(); int8_t state = cs.State(); f.Write( &cpu, sizeof( cpu ) ); f.Write( &reason, sizeof( reason ) ); f.Write( &state, sizeof( state ) ); } } sz = GetContextSwitchPerCpuCount(); f.Write( &sz, sizeof( sz ) ); for( int i=0; i<256; i++ ) { sz = m_data.cpuData[i].cs.size(); f.Write( &sz, sizeof( sz ) ); int64_t refTime = 0; for( auto& cx : m_data.cpuData[i].cs ) { WriteTimeOffset( f, refTime, cx.Start() ); WriteTimeOffset( f, refTime, cx.End() ); uint16_t thread = cx.Thread(); f.Write( &thread, sizeof( thread ) ); } } sz = m_data.tidToPid.size(); f.Write( &sz, sizeof( sz ) ); for( auto& v : m_data.tidToPid ) { f.Write( &v.first, sizeof( v.first ) ); f.Write( &v.second, sizeof( v.second ) ); } sz = m_data.cpuThreadData.size(); f.Write( &sz, sizeof( sz ) ); for( auto& v : m_data.cpuThreadData ) { f.Write( &v.first, sizeof( v.first ) ); f.Write( &v.second, sizeof( v.second ) ); } sz = m_data.symbolLoc.size(); f.Write( &sz, sizeof( sz ) ); sz = m_data.symbolLocInline.size(); f.Write( &sz, sizeof( sz ) ); sz = m_data.symbolMap.size(); f.Write( &sz, sizeof( sz ) ); for( auto& v : m_data.symbolMap ) { f.Write( &v.first, sizeof( v.first ) ); f.Write( &v.second, sizeof( v.second ) ); } sz = m_data.symbolCode.size(); f.Write( &sz, sizeof( sz ) ); for( auto& v : m_data.symbolCode ) { f.Write( &v.first, sizeof( v.first ) ); f.Write( &v.second.len, sizeof( v.second.len ) ); f.Write( v.second.data, v.second.len ); } sz = m_data.locationCodeAddressList.size(); f.Write( &sz, sizeof( sz ) ); for( auto& v : m_data.locationCodeAddressList ) { f.Write( &v.first, sizeof( v.first ) ); uint16_t lsz = uint16_t( v.second.size() ); f.Write( &lsz, sizeof( lsz ) ); uint64_t ref = 0; const uint64_t* ptr = v.second.data(); for( uint16_t i=0; i>& vec, int64_t& refTime ) { uint32_t sz = uint32_t( vec.size() ); f.Write( &sz, sizeof( sz ) ); if( vec.is_magic() ) { WriteTimelineImpl>( f, *(Vector*)( &vec ), refTime ); } else { WriteTimelineImpl>( f, vec, refTime ); } } template void Worker::WriteTimelineImpl( FileWrite& f, const V& vec, int64_t& refTime ) { Adapter a; for( auto& val : vec ) { auto& v = a(val); int16_t srcloc = v.SrcLoc(); f.Write( &srcloc, sizeof( srcloc ) ); int64_t start = v.Start(); WriteTimeOffset( f, refTime, start ); f.Write( &v.extra, sizeof( v.extra ) ); if( !v.HasChildren() ) { const uint32_t sz = 0; f.Write( &sz, sizeof( sz ) ); } else { WriteTimeline( f, GetZoneChildren( v.Child() ), refTime ); } WriteTimeOffset( f, refTime, v.End() ); } } void Worker::WriteTimeline( FileWrite& f, const Vector>& vec, int64_t& refTime, int64_t& refGpuTime ) { uint64_t sz = vec.size(); f.Write( &sz, sizeof( sz ) ); if( vec.is_magic() ) { WriteTimelineImpl>( f, *(Vector*)( &vec ), refTime, refGpuTime ); } else { WriteTimelineImpl>( f, vec, refTime, refGpuTime ); } } template void Worker::WriteTimelineImpl( FileWrite& f, const V& vec, int64_t& refTime, int64_t& refGpuTime ) { Adapter a; for( auto& val : vec ) { auto& v = a(val); WriteTimeOffset( f, refTime, v.CpuStart() ); WriteTimeOffset( f, refGpuTime, v.GpuStart() ); const int16_t srcloc = v.SrcLoc(); f.Write( &srcloc, sizeof( srcloc ) ); f.Write( &v.callstack, sizeof( v.callstack ) ); const uint16_t thread = v.Thread(); f.Write( &thread, sizeof( thread ) ); if( v.Child() < 0 ) { const uint64_t sz = 0; f.Write( &sz, sizeof( sz ) ); } else { WriteTimeline( f, GetGpuChildren( v.Child() ), refTime, refGpuTime ); } WriteTimeOffset( f, refTime, v.CpuEnd() ); WriteTimeOffset( f, refGpuTime, v.GpuEnd() ); } } static const char* s_failureReasons[] = { "", "Invalid order of zone begin and end events.", "Zone is ended twice.", "Zone text transfer destination doesn't match active zone.", "Zone name transfer destination doesn't match active zone.", "Memory free event without a matching allocation.", "Discontinuous frame begin/end mismatch.", "Frame image offset is invalid.", "Multiple frame images were sent for a single frame.", }; static_assert( sizeof( s_failureReasons ) / sizeof( *s_failureReasons ) == (int)Worker::Failure::NUM_FAILURES, "Missing failure reason description." ); const char* Worker::GetFailureString( Worker::Failure failure ) { return s_failureReasons[(int)failure]; } void Worker::SetParameter( size_t paramIdx, int32_t val ) { assert( paramIdx < m_params.size() ); m_params[paramIdx].val = val; const auto idx = uint64_t( m_params[paramIdx].idx ); const auto v = uint64_t( uint32_t( val ) ); Query( ServerQueryParameter, ( idx << 32 ) | v ); } const Worker::CpuThreadTopology* Worker::GetThreadTopology( uint32_t cpuThread ) const { auto it = m_data.cpuTopologyMap.find( cpuThread ); if( it == m_data.cpuTopologyMap.end() ) return nullptr; return &it->second; } ZoneExtra& Worker::AllocZoneExtra( ZoneEvent& ev ) { assert( ev.extra == 0 ); ev.extra = uint32_t( m_data.zoneExtra.size() ); auto& extra = m_data.zoneExtra.push_next(); memset( &extra, 0, sizeof( extra ) ); return extra; } ZoneExtra& Worker::RequestZoneExtra( ZoneEvent& ev ) { if( !HasZoneExtra( ev ) ) { return AllocZoneExtra( ev ); } else { return GetZoneExtraMutable( ev ); } } void Worker::CacheSource( const StringRef& str ) { assert( str.active ); assert( m_checkedFileStrings.find( str ) == m_checkedFileStrings.end() ); m_checkedFileStrings.emplace( str ); auto file = GetString( str ); if( SourceFileValid( file, GetCaptureTime() ) ) { // Possible duplication of pointer and index strings if( m_data.sourceFileCache.find( file ) == m_data.sourceFileCache.end() ) { FILE* f = fopen( file, "rb" ); fseek( f, 0, SEEK_END ); const auto sz = ftell( f ); fseek( f, 0, SEEK_SET ); auto src = (char*)m_slab.AllocBig( sz ); fread( src, 1, sz, f ); fclose( f ); m_data.sourceFileCache.emplace( file, MemoryBlock{ src, uint32_t( sz ) } ); } } } uint64_t Worker::GetSourceFileCacheSize() const { uint64_t cnt = 0; for( auto& v : m_data.sourceFileCache ) { cnt += v.second.len; } return cnt; } Worker::MemoryBlock Worker::GetSourceFileFromCache( const char* file ) const { auto it = m_data.sourceFileCache.find( file ); if( it == m_data.sourceFileCache.end() ) return MemoryBlock {}; return it->second; } }