Amebis/stdex
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

stream: add write_array variants for zero-terminated strings

Browse Source 
Signed-off-by: Simon Rozman <simon@rozman.si>
This commit is contained in:
Simon Rozman 2023-08-24 13:57:57 +02:00
parent 73d03d81af
commit be83f36082
1 changed files with 40 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
include/stdex/stream.hpp
View File

@ -394,6 +394,26 @@ namespace stdex
return write(array, mul(size, count)) / size;
}
///
/// Writes array of characters to the stream
///
/// \param[in] wstr String to write. Must be zero-terminated.
/// \param[in] charset Charset to convert string to
///
/// \return Number of code units written
///
size_t write_array(_In_z_ const wchar_t* wstr, _In_ charset_id charset)
{
if (!ok()) _Unlikely_
return 0;
size_t num_chars = stdex::strlen(wstr);
if (charset != charset_id::utf16) {
std::string str(wstr2str(wstr, num_chars, charset));
return write_array(str.data(), sizeof(char), str.size());
}
return write_array(wstr, sizeof(wchar_t), num_chars);
}
///
/// Writes array of characters to the stream
///
@ -415,6 +435,26 @@ namespace stdex
return write_array(wstr, sizeof(wchar_t), num_chars);
}
///
/// Writes array of characters to the stream
///
/// \param[in] wstr String to write
/// \param[in] charset Charset to convert string to
///
/// \return Number of code units written
///
template<class _Traits = std::char_traits<wchar_t>, class _Ax = std::allocator<wchar_t>>
size_t write_array(_In_ const std::basic_string<wchar_t, _Traits, _Ax>& wstr, _In_ charset_id charset)
{
if (!ok()) _Unlikely_
return 0;
if (charset != charset_id::utf16) {
std::string str(wstr2str(wstr, charset));
return write_array(str.data(), sizeof(char), str.size());
}
return write_array(wstr.data(), sizeof(wchar_t), wstr.size());
}
///
/// Reads length-prefixed string from the stream
///