src/id_map.hh


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
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92

#ifndef id_map_hh_INCLUDED
#define id_map_hh_INCLUDED

#include "containers.hh"
#include "string.hh"
#include "vector.hh"

namespace Kakoune
{

template<typename Value, MemoryDomain domain = MemoryDomain::Undefined>
class IdMap
{
public:
    using value_type = std::pair<String, Value>;
    using container_type = Vector<value_type, domain>;
    using iterator = typename container_type::iterator;
    using const_iterator = typename container_type::const_iterator;

    IdMap() = default;
    IdMap(std::initializer_list<value_type> val) : m_content{val} {}

    void append(const value_type& value)
    {
        m_content.push_back(value);
    }

    void append(value_type&& value)
    {
        m_content.push_back(std::move(value));
    }

    iterator find(StringView id)
    {
        return Kakoune::find(transformed(m_content, get_id), id).base();
    }

    const_iterator find(StringView id) const
    {
        return Kakoune::find(transformed(m_content, get_id), id).base();
    }

    bool contains(StringView id) const
    {
        return find(id) != end();
    }

    void remove(StringView id)
    {
        auto it = find(id);
        if (it != end())
            m_content.erase(it);
    }

    void remove_all(StringView id)
    {
        auto it = std::remove_if(begin(), end(),
                                 [&](value_type& v){ return v.first == id; });
        m_content.erase(it, end());
    }

    Value& operator[](StringView id)
    {
        auto it = find(id);
        if (it != m_content.end())
            return it->second;

        append({ id.str(), Value{} });
        return (m_content.end()-1)->second;
    }

    const Value& operator[](StringView id) const
    {
        return (*const_cast<IdMap*>(this))[id];
    }

    static const String& get_id(const value_type& v) { return v.first; }

    bool empty() const { return m_content.empty(); }

    iterator       begin()       { return m_content.begin(); }
    iterator       end()         { return m_content.end(); }
    const_iterator begin() const { return m_content.begin(); }
    const_iterator end()   const { return m_content.end(); }

private:
    container_type m_content;
};

}

#endif // id_map_hh_INCLUDED