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.
This commit is contained in:
Bartosz Taudul 2018-08-04 16:25:11 +02:00
parent e174e2c12a
commit d2c866377e
3 changed files with 84 additions and 0 deletions

View File

@ -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

View File

@ -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" />

View File

@ -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" />