diff options
Diffstat (limited to 'src/app.hpp')
-rw-r--r-- | src/app.hpp | 26 |
1 files changed, 22 insertions, 4 deletions
diff --git a/src/app.hpp b/src/app.hpp index c3c2060..8f30d96 100644 --- a/src/app.hpp +++ b/src/app.hpp @@ -53,8 +53,8 @@ struct id_allocator { id_allocator &operator=(id_allocator const &); id_allocator &operator=(id_allocator &&) = delete; - // Allocate a new ID - unsigned operator()(std::string const &name) { + // Insert and return a new ID + unsigned generate_id(std::string const &name) { unsigned sid = this->next++; this->surfaces[sid] = name; // this->pending_surfaces.insert({sid}); @@ -64,16 +64,34 @@ struct id_allocator { } // Lookup by ID or by name - optional<unsigned> operator[](std::string const &name) { + optional<unsigned> lookup(std::string const &name) const { auto i = this->names.find(name); return i == this->names.end() ? nullopt : optional<unsigned>(i->second); } - optional<std::string> operator[](unsigned id) { + optional<std::string> lookup(unsigned id) const { auto i = this->surfaces.find(id); return i == this->surfaces.end() ? nullopt : optional<std::string>(i->second); } + + // Remove a surface id and name + // I don't think I will need this, do I? + void remove_id(std::string const &name) { + auto i = this->names.find(name); + if (i != this->names.end()) { + this->surfaces.erase(i->second); + this->names.erase(i); + } + } + + void remove_id(unsigned id) { + auto i = this->surfaces.find(id); + if (i != this->surfaces.end()) { + this->names.erase(i->second); + this->surfaces.erase(i); + } + } }; struct App { |