diff options
author | Treeston <treeston.mmoc@gmail.com> | 2018-12-25 01:47:24 +0100 |
---|---|---|
committer | Shauren <shauren.trinity@gmail.com> | 2021-11-16 21:04:53 +0100 |
commit | 4f8fd5690355a4202e8f29eb175bb782d4417cef (patch) | |
tree | 05adf69d5aaff5438defd7550b635a3d1f98767e /src/common/Utilities/Containers.h | |
parent | 4177030dcfae9d5a866e4d77d47d333172a3efd6 (diff) |
Core/Utils: Fix Unicode handling
(cherry picked from commit 6d6077e36fe9c5cb8ea8e4f981d637e72ee87037)
Diffstat (limited to 'src/common/Utilities/Containers.h')
-rw-r--r-- | src/common/Utilities/Containers.h | 30 |
1 files changed, 30 insertions, 0 deletions
diff --git a/src/common/Utilities/Containers.h b/src/common/Utilities/Containers.h index ca4a00e4c29..cf4d31d4dc2 100644 --- a/src/common/Utilities/Containers.h +++ b/src/common/Utilities/Containers.h @@ -21,6 +21,8 @@ #include "Define.h" #include "Random.h" #include <algorithm> +#include <exception> +#include <iterator> #include <utility> #include <vector> @@ -38,6 +40,34 @@ namespace Trinity return std::addressof(not_ptr); } + template <class T> + class BufferWriteGuard + { + public: + using iterator_category = std::output_iterator_tag; + using value_type = void; + using pointer = T*; + using reference = T&; + using difference_type = std::ptrdiff_t; + + BufferWriteGuard(T* buf, size_t n) : _buf(buf), _n(n) {} + + T& operator*() const { check(); return *_buf; } + BufferWriteGuard& operator++() { check(); ++_buf; --_n; return *this; } + T* operator++(int) { check(); T* b = _buf; ++_buf; --_n; return b; } + + size_t size() const { return _n; } + + private: + T* _buf; + size_t _n; + void check() + { + if (!_n) + throw std::out_of_range("index"); + } + }; + namespace Containers { // replace with std::size in C++17 |