From d2c866377ed712615e822c80cd794039e39d0b3a Mon Sep 17 00:00:00 2001
From: Bartosz Taudul <wolf.pld@gmail.com>
Date: Sat, 4 Aug 2018 16:25:11 +0200
Subject: [PATCH] Extract unique string discovery from worker.

This class is responsible for handling data sets that should be grouped
together, but which may come with different name pointers.

It is a generalization of the plot merging functionality.
---
 server/TracyStringDiscovery.hpp              | 80 ++++++++++++++++++++
 standalone/build/win32/Tracy.vcxproj         |  1 +
 standalone/build/win32/Tracy.vcxproj.filters |  3 +
 3 files changed, 84 insertions(+)
 create mode 100644 server/TracyStringDiscovery.hpp

diff --git a/server/TracyStringDiscovery.hpp b/server/TracyStringDiscovery.hpp
new file mode 100644
index 00000000..36e5cd84
--- /dev/null
+++ b/server/TracyStringDiscovery.hpp
@@ -0,0 +1,80 @@
+#ifndef __TRACYSTRINGDISCOVERY_HPP__
+#define __TRACYSTRINGDISCOVERY_HPP__
+
+#include "../common/TracyForceInline.hpp"
+#include "tracy_flat_hash_map.hpp"
+#include "TracyCharUtil.hpp"
+#include "TracyVector.hpp"
+
+namespace tracy
+{
+
+template<typename T>
+class StringDiscovery
+{
+public:
+    tracy_force_inline Vector<T>& Data() { return m_data; }
+    tracy_force_inline const Vector<T>& Data() const { return m_data; }
+
+    tracy_force_inline bool IsPending() const { return !m_pending.empty(); }
+
+    // Merge( destination, postponed )
+    tracy_force_inline bool StringDiscovered( uint64_t name, const StringLocation& sl, std::function<void(T,T)> Merge )
+    {
+        auto pit = m_pending.find( name );
+        assert( pit != m_pending.end() );
+
+        auto it = m_rev.find( sl.ptr );
+        bool add = it == m_rev.end();
+        if( add )
+        {
+            m_map.emplace( name, pit->second );
+            m_rev.emplace( sl.ptr, pit->second );
+            m_data.push_back( pit->second );
+        }
+        else
+        {
+            auto item = it->second;
+            m_map.emplace( name, item );
+            Merge( item, pit->second );
+        }
+
+        m_pending.erase( pit );
+
+        return add;
+    }
+
+    tracy_force_inline T Retrieve( uint64_t name, std::function<T(uint64_t)> Create, std::function<void(uint64_t)> Query )
+    {
+        auto it = m_map.find( name );
+        if( it == m_map.end() )
+        {
+            auto pit = m_pending.find( name );
+            if( pit == m_pending.end() )
+            {
+                T item = Create( name );
+                m_pending.emplace( name, item );
+                Query( name );
+                return item;
+            }
+            else
+            {
+                return pit->second;
+            }
+        }
+        else
+        {
+            return it->second;
+        }
+    }
+
+private:
+    Vector<T> m_data;
+    flat_hash_map<uint64_t, T, nohash<uint64_t>> m_pending;
+    flat_hash_map<uint64_t, T, nohash<uint64_t>> m_map;
+    flat_hash_map<const char*, T, charutil::HasherPOT, charutil::Comparator> m_rev;
+};
+
+}
+
+#endif
diff --git a/standalone/build/win32/Tracy.vcxproj b/standalone/build/win32/Tracy.vcxproj
index fd2651ad..c5a324aa 100644
--- a/standalone/build/win32/Tracy.vcxproj
+++ b/standalone/build/win32/Tracy.vcxproj
@@ -139,6 +139,7 @@
     <ClInclude Include="..\..\..\server\TracyMemory.hpp" />
     <ClInclude Include="..\..\..\server\TracyPopcnt.hpp" />
     <ClInclude Include="..\..\..\server\TracySlab.hpp" />
+    <ClInclude Include="..\..\..\server\TracyStringDiscovery.hpp" />
     <ClInclude Include="..\..\..\server\TracyVarArray.hpp" />
     <ClInclude Include="..\..\..\server\TracyVector.hpp" />
     <ClInclude Include="..\..\..\server\TracyVersion.hpp" />
diff --git a/standalone/build/win32/Tracy.vcxproj.filters b/standalone/build/win32/Tracy.vcxproj.filters
index 39b02feb..9b336b01 100644
--- a/standalone/build/win32/Tracy.vcxproj.filters
+++ b/standalone/build/win32/Tracy.vcxproj.filters
@@ -188,6 +188,9 @@
     <ClInclude Include="..\..\..\server\TracyVersion.hpp">
       <Filter>server</Filter>
     </ClInclude>
+    <ClInclude Include="..\..\..\server\TracyStringDiscovery.hpp">
+      <Filter>server</Filter>
+    </ClInclude>
   </ItemGroup>
   <ItemGroup>
     <Natvis Include="DebugVis.natvis" />