flatset_type.hpp

Go to the documentation of this file.

/*
 * This file is part of OpenTTD.
 * OpenTTD is free software; you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation, version 2.
 * OpenTTD is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
 * See the GNU General Public License for more details. You should have received a copy of the GNU General Public License along with OpenTTD. If not, see <http://www.gnu.org/licenses/>.
 */
 
#ifndef FLATSET_TYPE_HPP
#define FLATSET_TYPE_HPP
 
template <class Tkey, class Tcompare = std::less<>>
class FlatSet {
  std::vector<Tkey> data; 
public:
  using const_iterator = std::vector<Tkey>::const_iterator;
 
  std::pair<const_iterator, bool> insert(const Tkey &key)
  {
    auto it = std::ranges::lower_bound(this->data, key, Tcompare{});
    if (it == std::end(this->data) || *it != key) return {this->data.emplace(it, key), true};
    return {it, false};
  }
  std::pair<const_iterator, bool> insert(const Tkey &key) {…}
 
  size_t erase(const Tkey &key)
  {
    auto it = std::ranges::lower_bound(this->data, key, Tcompare{});
    if (it == std::end(this->data) || *it != key) return 0;
 
    this->data.erase(it);
    return 1;
  }
  size_t erase(const Tkey &key) {…}
 
  bool contains(const Tkey &key)
  {
    return std::ranges::binary_search(this->data, key, Tcompare{});
  }
  bool contains(const Tkey &key) {…}
 
  const_iterator begin() const { return std::cbegin(this->data); }
  const_iterator end() const { return std::cend(this->data); }
 
  const_iterator cbegin() const { return std::cbegin(this->data); }
  const_iterator cend() const { return std::cend(this->data); }
 
  size_t size() const { return std::size(this->data); }
  bool empty() const { return this->data.empty(); }
 
  void clear() { this->data.clear(); }
 
  auto operator<=>(const FlatSet<Tkey, Tcompare> &) const = default;
};
class FlatSet {…};
 
#endif /* FLATSET_TYPE_HPP */