mirror of
https://github.com/wolfpld/tracy.git
synced 2024-09-20 05:42:18 +00:00
d2c866377e
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.
81 lines
2.0 KiB
C++
81 lines
2.0 KiB
C++
#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
|