From e0bcd3514f586b250a56c5049053476c4c002dc5 Mon Sep 17 00:00:00 2001 From: Carl Hetherington Date: Tue, 27 Sep 2022 00:18:36 +0200 Subject: [PATCH] Add EnumIndexedVector. --- src/lib/enum_indexed_vector.h | 92 +++++++++++++++++++++++++++++++++++ 1 file changed, 92 insertions(+) create mode 100644 src/lib/enum_indexed_vector.h diff --git a/src/lib/enum_indexed_vector.h b/src/lib/enum_indexed_vector.h new file mode 100644 index 000000000..5032a0bf2 --- /dev/null +++ b/src/lib/enum_indexed_vector.h @@ -0,0 +1,92 @@ +/* + Copyright (C) 2022 Carl Hetherington + + This file is part of DCP-o-matic. + + DCP-o-matic 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; either version 2 of the License, or + (at your option) any later version. + + DCP-o-matic 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 DCP-o-matic. If not, see . + +*/ + + +#ifndef DCPOMATIC_ENUM_INDEXED_VECTOR_H +#define DCPOMATIC_ENUM_INDEXED_VECTOR_H + + +#include + + +template +class EnumIndexedVector +{ +public: + EnumIndexedVector() + : _data(static_cast(Enum::COUNT)) + {} + + EnumIndexedVector(EnumIndexedVector const& other) + : _data(other._data) + {} + + EnumIndexedVector& operator=(EnumIndexedVector const& other) { + if (this == &other) { + return *this; + } + + _data = other._data; + return *this; + } + + typename std::vector::reference operator[](int index) { + return _data[index]; + } + + typename std::vector::const_reference operator[](int index) const { + return _data[index]; + } + + typename std::vector::reference operator[](Enum index) { + return _data[static_cast(index)]; + } + + typename std::vector::const_reference operator[](Enum index) const { + return _data[static_cast(index)]; + } + + void clear() { + std::fill(_data.begin(), _data.end(), {}); + } + + typename std::vector::const_iterator begin() const { + return _data.begin(); + } + + typename std::vector::const_iterator end() const { + return _data.end(); + } + + typename std::vector::iterator begin() { + return _data.begin(); + } + + typename std::vector::iterator end() { + return _data.end(); + } + +private: + std::vector _data; +}; + + +#endif + -- 2.30.2