mirror of
https://github.com/zeek/zeek.git
synced 2025-10-06 00:28:21 +00:00
75 lines
1.4 KiB
C++
75 lines
1.4 KiB
C++
// See the file "COPYING" in the main distribution directory for copyright.
|
|
|
|
#include "zeek/script_opt/CPP/Tracker.h"
|
|
|
|
#include "zeek/Desc.h"
|
|
#include "zeek/script_opt/CPP/Util.h"
|
|
#include "zeek/script_opt/ProfileFunc.h"
|
|
|
|
namespace zeek::detail
|
|
{
|
|
|
|
using namespace std;
|
|
|
|
template <class T> void CPPTracker<T>::AddKey(IntrusivePtr<T> key, p_hash_type h)
|
|
{
|
|
if ( HasKey(key) )
|
|
return;
|
|
|
|
if ( h == 0 )
|
|
h = Hash(key);
|
|
|
|
if ( map2.count(h) == 0 )
|
|
{
|
|
auto index = keys.size();
|
|
keys.push_back(key);
|
|
|
|
map2[h] = index;
|
|
reps[h] = key.get();
|
|
}
|
|
|
|
ASSERT(h != 0); // check for hash botches
|
|
|
|
map[key.get()] = h;
|
|
}
|
|
|
|
template <class T> string CPPTracker<T>::KeyName(const T* key)
|
|
{
|
|
ASSERT(HasKey(key));
|
|
|
|
auto hash = map[key];
|
|
ASSERT(hash != 0);
|
|
|
|
auto rep = reps[hash];
|
|
auto gi = gi_s.find(rep);
|
|
if ( gi != gi_s.end() )
|
|
return gi->second->Name();
|
|
|
|
auto index = map2[hash];
|
|
string ind = Fmt(index);
|
|
string full_name;
|
|
|
|
if ( single_global )
|
|
full_name = base_name + "__CPP[" + ind + "]";
|
|
else
|
|
full_name = base_name + "_" + ind + "__CPP";
|
|
|
|
return full_name;
|
|
}
|
|
|
|
template <class T> p_hash_type CPPTracker<T>::Hash(IntrusivePtr<T> key) const
|
|
{
|
|
ODesc d;
|
|
d.SetDeterminism(true);
|
|
key->Describe(&d);
|
|
string desc = d.Description();
|
|
auto h = hash<string>{}(base_name + desc);
|
|
return p_hash_type(h);
|
|
}
|
|
|
|
// Instantiate the templates we'll need.
|
|
template class CPPTracker<Type>;
|
|
template class CPPTracker<Attributes>;
|
|
template class CPPTracker<Expr>;
|
|
|
|
} // zeek::detail
|