From a76622d17ab41fcca17eada9156d68d9dd0bee42 Mon Sep 17 00:00:00 2001
From: Bartosz Taudul <wolf.pld@gmail.com>
Date: Sat, 3 Aug 2019 14:35:01 +0200
Subject: [PATCH] Cache last searched ThreadData.

---
 server/TracyWorker.cpp |  6 +++++-
 server/TracyWorker.hpp | 10 ++++++++--
 2 files changed, 13 insertions(+), 3 deletions(-)

diff --git a/server/TracyWorker.cpp b/server/TracyWorker.cpp
index e675646e..3556d1ef 100644
--- a/server/TracyWorker.cpp
+++ b/server/TracyWorker.cpp
@@ -2131,11 +2131,13 @@ void Worker::InsertMessageData( MessageData* msg, uint64_t thread )
     }
 }
 
-ThreadData* Worker::NoticeThread( uint64_t thread )
+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
@@ -2153,6 +2155,8 @@ ThreadData* Worker::NewThread( uint64_t thread )
     td->nextZoneId = 0;
     m_data.threads.push_back( td );
     m_threadMap.emplace( thread, td );
+    m_data.threadDataLast.first = thread;
+    m_data.threadDataLast.second = td;
     return td;
 }
 
diff --git a/server/TracyWorker.hpp b/server/TracyWorker.hpp
index 2915c0de..2acbc89c 100644
--- a/server/TracyWorker.hpp
+++ b/server/TracyWorker.hpp
@@ -137,7 +137,7 @@ private:
 
     struct DataBlock
     {
-        DataBlock() : zonesCnt( 0 ), lastTime( 0 ), frameOffset( 0 ), threadLast( std::numeric_limits<uint64_t>::max(), 0 ) {}
+        DataBlock() : zonesCnt( 0 ), lastTime( 0 ), frameOffset( 0 ), threadLast( std::numeric_limits<uint64_t>::max(), 0 ), threadDataLast( std::numeric_limits<uint64_t>::max(), nullptr ) {}
 
         std::shared_mutex lock;
         StringDiscovery<FrameData*> frames;
@@ -177,6 +177,7 @@ private:
         flat_hash_map<uint64_t, uint16_t, nohash<uint64_t>> threadMap;
         Vector<uint64_t> threadExpand;
         std::pair<uint64_t, uint16_t> threadLast;
+        std::pair<uint64_t, ThreadData*> threadDataLast;
 
         Vector<Vector<ZoneEvent*>> zoneChildren;
         Vector<Vector<GpuEvent*>> gpuChildren;
@@ -430,8 +431,13 @@ private:
 
     void InsertMessageData( MessageData* msg, uint64_t thread );
 
+    ThreadData* NoticeThreadReal( uint64_t thread );
     ThreadData* NewThread( uint64_t thread );
-    ThreadData* NoticeThread( uint64_t thread );
+    ThreadData* NoticeThread( uint64_t thread )
+    {
+        if( m_data.threadDataLast.first == thread ) return m_data.threadDataLast.second;
+        return NoticeThreadReal( thread );
+    }
 
     tracy_force_inline void NewZone( ZoneEvent* zone, uint64_t thread );