summaryrefslogtreecommitdiff
path: root/lib/nes/common/containers/string-builder.hh
blob: 0f8a64ca24dd63e52a33cfe4246422605f4227e8 (plain)
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
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
#pragma once

#include "nes/common/types.hh"
#include "nes/common/containers/span.hh"
#include "nes/common/status.hh"
#include <string_view>

namespace nes
{
	enum class number_format
	{
		decimal,
		hexadecimal,
		octal,
		binary,
	};

	/// Builds a string into a fixed-length buffer.
	class string_builder
	{
		template<u32 Capacity>
		friend class string_buffer;

		span<char> buffer_;
		u32 length_{ 0 };

		enum class format_arg_type
		{
			string,
			character,
			signed_integer,
			unsigned_integer,
			boolean,
		};

		struct format_arg
		{
			format_arg_type type;
			union
			{
				std::string_view string;
				char character;
				i64 signed_integer;
				u64 unsigned_integer;
				bool boolean;
			};

			format_arg() = delete;

			explicit format_arg(char const* v) : type{ format_arg_type::string } { string = v; }
			explicit format_arg(std::string_view const v) : type{ format_arg_type::string } { string = v; }
			explicit format_arg(char const v) : type{ format_arg_type::character } { character = v; }
			explicit format_arg(unsigned char const v) : format_arg{ static_cast<char>(v) } {}
			explicit format_arg(long long const v) : type{ format_arg_type::signed_integer } { signed_integer = v; }
			explicit format_arg(unsigned long long const v) : type{ format_arg_type::unsigned_integer } { unsigned_integer = v; }
			explicit format_arg(short const v) : format_arg{ static_cast<signed long long>(v) } {}
			explicit format_arg(unsigned short const v) : format_arg{ static_cast<unsigned long long>(v) } {}
			explicit format_arg(int const v) : format_arg{ static_cast<signed long long>(v) } {}
			explicit format_arg(unsigned int const v) : format_arg{ static_cast<unsigned long long>(v) } {}
			explicit format_arg(long const v) : format_arg{ static_cast<signed long long>(v) } {}
			explicit format_arg(unsigned long const v) : format_arg{ static_cast<unsigned long long>(v) } {}
			explicit format_arg(bool const v) : type{ format_arg_type::boolean } { boolean = v; }
		};

	public:
		explicit string_builder(span<char>);

		/// Get the resulting string.
		auto get_result() const -> std::string_view { return std::string_view{ buffer_.get_data(), length_ }; }

		auto append_char(char) -> status;
		auto append_string(std::string_view) -> status;
		auto append_int(i64, number_format = number_format::decimal) -> status;
		auto append_int(u64, number_format = number_format::decimal) -> status;
		auto append_bool(bool) -> status;

		/// Print a templated string.
		///
		/// Placeholders are signified by `{<params>}` where `params` contains optional formatting parameters:
		///  - for integers either "d" (the default), "x", "o", or "b"
		///  - for any other type, no parameters are supported.
		///
		/// Curly braces can be escaped by repeating them twice like {{this}}.
		template<typename... Args>
		auto append_format(std::string_view const fmt, Args... args) -> status
		{
			constexpr u32 arg_count = sizeof...(Args);
			// Add another argument at the end so we don't have an empty array expression.
			const format_arg arg_wrappers[arg_count + 1] = { format_arg{ args }..., format_arg(0) };

			return append_format(fmt, arg_wrappers, arg_count);
		}

	private:
		auto append_format(std::string_view fmt, format_arg const* args, u32 arg_count) -> status;
		auto append_format_arg(format_arg, std::string_view params) -> status;
	};

	/// A string builder with a buffer embedded into it.
	template<u32 Capacity>
	class string_buffer : public string_builder
	{
		char buffer_[Capacity];

	public:
		explicit string_buffer()
			: string_builder{ buffer_ }
		{
		}

		string_buffer(string_buffer const& other)
			: string_builder{ buffer_ }
		{
			std::copy_n(other.buffer_, Capacity, buffer_);
			length_ = other.length_;
		}

		auto operator=(string_buffer const& other)
		{
			std::copy_n(other.buffer_, Capacity, buffer_);
			length_ = other.length_;
		}
	};
} // namespace nes