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 93 94 95 96 97 98 99 100 101 102 103 104 105
|
/* Copyright (C) 2025 Wildfire Games.
* This file is part of 0 A.D.
*
* 0 A.D. 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.
*
* 0 A.D. 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 0 A.D. If not, see <http://www.gnu.org/licenses/>.
*/
#ifndef INCLUDED_PS_STRINGBUILDER
#define INCLUDED_PS_STRINGBUILDER
#include <charconv>
#include <cstdint>
#include <span>
#include <string>
#include <string_view>
#include <type_traits>
#include <version>
#if defined(__cpp_lib_to_chars)
#include <charconv>
#else
#include <fmt/core.h>
#endif
namespace PS
{
/**
* A helper class to build string in-place without additional cost of
* allocations, locales and format parsing. Prefer to use the class with stack
* allocated buffers over standard tools in high-load cases.
* TODO: in case of overflow it might use a heap allocated storage.
*/
class StringBuilder
{
public:
StringBuilder(std::span<char> buffer) noexcept
: m_Buffer{buffer}, m_BufferBegin{buffer.data()}
{
ENSURE(!m_Buffer.empty());
}
template<typename T>
typename std::enable_if<std::is_arithmetic_v<T>>::type Append(const T value) noexcept
{
ENSURE(m_Buffer.size() > 0);
#if defined(__cpp_lib_to_chars)
const std::to_chars_result result{std::to_chars(m_Buffer.data(), m_Buffer.data() + m_Buffer.size() - 1, value)};
ENSURE(result.ec == std::errc());
m_Buffer = m_Buffer.subspan(result.ptr - m_Buffer.data());
// to_chars doesn't write terminating null.
m_Buffer.front() = 0;
#else
// TODO: switch to std::to_chars after minimal libcxx supports it.
const fmt::format_to_n_result result{
fmt::format_to_n(m_Buffer.data(), m_Buffer.size() - 1, "{}", value)};
ENSURE(m_Buffer.data() != result.out);
m_Buffer = m_Buffer.subspan(result.size);
// format_to_n doesn't write terminating null.
m_Buffer.front() = 0;
#endif
}
void Append(const char ch)
{
ENSURE(m_Buffer.size() > 1);
m_Buffer.front() = ch;
m_Buffer = m_Buffer.subspan(1);
m_Buffer.front() = 0;
}
void Append(const std::string_view str) noexcept
{
ENSURE(m_Buffer.size() >= static_cast<size_t>(str.size()) + 1);
std::copy(str.begin(), str.end(), m_Buffer.begin());
m_Buffer = m_Buffer.subspan(str.size());
m_Buffer.front() = 0;
}
/**
* Returns a string_view to a null-terminated string.
*/
std::string_view Str() noexcept
{
return {m_BufferBegin, m_Buffer.data()};
}
private:
std::span<char> m_Buffer;
char* m_BufferBegin;
};
} // namespace PS
#endif // INCLUDED_PS_STRINGBUILDER
|