src/common/intern-pool.hpp


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40

#ifndef INTERN_POOL_HPP
#define INTERN_POOL_HPP

#include <cassert>

#include <map>
#include <vector>

#include "../strings/fstring.hpp"
#include "../strings/zstring.hpp"
#include "../strings/xstring.hpp"

class InternPool
{
    std::map<FString, size_t> known;
    std::vector<FString> names;
public:
    size_t intern(XString name_)
    {
        FString name = name_;
        // hm, I could change this to do aliases
        auto pair = known.insert({name, known.size()});
        if (pair.second)
            names.push_back(name);
        assert (known.size() == names.size());
        return pair.first->second;
    }

    ZString outtern(size_t sz) const
    {
        return names[sz];
    }

    size_t size() const
    {
        return known.size();
    }
};

#endif //INTERN_POOL_HPP