Use SPSC queue for frame images.

This commit is contained in:
Bartosz Taudul 2021-10-20 22:56:19 +02:00
parent ff3382391e
commit 02e76faff7
No known key found for this signature in database
GPG Key ID: B7FE2008B7575DF3
2 changed files with 28 additions and 60 deletions

View File

@ -1146,8 +1146,7 @@ Profiler::Profiler()
, m_serialQueue( 1024*1024 )
, m_serialDequeue( 1024*1024 )
#ifndef TRACY_NO_FRAME_IMAGE
, m_fiQueue( 16 )
, m_fiDequeue( 16 )
, m_fiQueue( 64 )
#endif
, m_frameCount( 0 )
, m_isConnected( false )
@ -1785,60 +1784,30 @@ void Profiler::CompressWorker()
for(;;)
{
const auto shouldExit = ShouldExit();
FrameImageQueueItem fi;
if( m_fiQueue.try_dequeue( fi ) )
{
bool lockHeld = true;
while( !m_fiLock.try_lock() )
{
if( m_shutdownManual.load( std::memory_order_relaxed ) )
{
lockHeld = false;
break;
}
}
if( !m_fiQueue.empty() ) m_fiQueue.swap( m_fiDequeue );
if( lockHeld )
{
m_fiLock.unlock();
}
}
const auto sz = m_fiDequeue.size();
if( sz > 0 )
{
auto fi = m_fiDequeue.data();
auto end = fi + sz;
while( fi != end )
{
const auto w = fi->w;
const auto h = fi->h;
const auto w = fi.w;
const auto h = fi.h;
const auto csz = size_t( w * h / 2 );
auto etc1buf = (char*)tracy_malloc( csz );
CompressImageDxt1( (const char*)fi->image, etc1buf, w, h );
tracy_free( fi->image );
CompressImageDxt1( (const char*)fi.image, etc1buf, w, h );
tracy_free( fi.image );
TracyLfqPrepare( QueueType::FrameImage );
MemWrite( &item->frameImageFat.image, (uint64_t)etc1buf );
MemWrite( &item->frameImageFat.frame, fi->frame );
MemWrite( &item->frameImageFat.frame, fi.frame );
MemWrite( &item->frameImageFat.w, w );
MemWrite( &item->frameImageFat.h, h );
uint8_t flip = fi->flip;
uint8_t flip = fi.flip;
MemWrite( &item->frameImageFat.flip, flip );
TracyLfqCommit;
fi++;
}
m_fiDequeue.clear();
}
else
{
if( shouldExit ) return;
std::this_thread::sleep_for( std::chrono::milliseconds( 20 ) );
}
if( shouldExit )
{
return;
}
}
}
#endif

View File

@ -8,6 +8,7 @@
#include <time.h>
#include "tracy_concurrentqueue.h"
#include "tracy_readerwriterqueue.h"
#include "TracyCallstack.hpp"
#include "TracySysTime.hpp"
#include "TracyFastVector.hpp"
@ -243,15 +244,14 @@ public:
auto ptr = (char*)tracy_malloc( sz );
memcpy( ptr, image, sz );
profiler.m_fiLock.lock();
auto fi = profiler.m_fiQueue.prepare_next();
fi->image = ptr;
fi->frame = uint32_t( profiler.m_frameCount.load( std::memory_order_relaxed ) - offset );
fi->w = w;
fi->h = h;
fi->flip = flip;
profiler.m_fiQueue.commit_next();
profiler.m_fiLock.unlock();
profiler.m_fiQueue.emplace( FrameImageQueueItem {
ptr,
uint32_t( profiler.m_frameCount.load( std::memory_order_relaxed ) - offset ),
w,
h,
0,
flip
} );
#endif
}
@ -810,8 +810,7 @@ private:
TracyMutex m_serialLock;
#ifndef TRACY_NO_FRAME_IMAGE
FastVector<FrameImageQueueItem> m_fiQueue, m_fiDequeue;
TracyMutex m_fiLock;
ReaderWriterQueue<FrameImageQueueItem> m_fiQueue;
#endif
std::atomic<uint64_t> m_frameCount;