2020-11-16 14:47:43 +08:00
|
|
|
#ifndef SIMDSOFT__NTCVT_HPP
|
|
|
|
#define SIMDSOFT__NTCVT_HPP
|
|
|
|
|
|
|
|
#pragma once
|
|
|
|
#if !defined(WIN32_LEAN_AND_MEAN)
|
|
|
|
# define WIN32_LEAN_AND_MEAN
|
|
|
|
#endif
|
|
|
|
#include <Windows.h>
|
|
|
|
#include <string>
|
|
|
|
|
|
|
|
#if !defined(NTCVT_CP_DEFAULT)
|
|
|
|
#define NTCVT_CP_DEFAULT CP_ACP
|
|
|
|
#endif
|
|
|
|
|
|
|
|
namespace ntcvt
|
|
|
|
{
|
|
|
|
|
|
|
|
// different with resize, not all of fill new memory with '\0'
|
|
|
|
namespace buffer_traits
|
|
|
|
{
|
|
|
|
template <class _Elem, class _Traits = std::char_traits<_Elem>,
|
|
|
|
class _Alloc = std::allocator<_Elem>>
|
2021-05-28 14:55:27 +08:00
|
|
|
class intrusive_string : public std::basic_string<_Elem, _Traits, _Alloc>
|
2020-11-16 14:47:43 +08:00
|
|
|
{
|
|
|
|
public:
|
2021-05-29 16:50:38 +08:00
|
|
|
#if _MSC_VER >= 1920 // VS2019+
|
2020-11-16 14:47:43 +08:00
|
|
|
using _Alty = std::_Rebind_alloc_t<_Alloc, _Elem>;
|
|
|
|
using _Alty_traits = std::allocator_traits<_Alty>;
|
|
|
|
|
|
|
|
using _Scary_val = std::_String_val<
|
|
|
|
std::conditional_t<std::_Is_simple_alloc_v<_Alty>, std::_Simple_types<_Elem>,
|
|
|
|
std::_String_iter_types<
|
|
|
|
_Elem, typename _Alty_traits::size_type,
|
|
|
|
typename _Alty_traits::difference_type, typename _Alty_traits::pointer,
|
|
|
|
typename _Alty_traits::const_pointer, _Elem&, const _Elem&>>>;
|
|
|
|
#endif
|
|
|
|
// See also afxmfc CString::GetBufferSetLength
|
|
|
|
// Why do this hack?
|
|
|
|
// stupid: because the default c++ standard resize always fill with '\0'
|
|
|
|
// std::string: use memset (usually implemented with SIMD)
|
|
|
|
// std::wstring: for loop (slow performance)
|
|
|
|
// only works on msvc currently
|
2021-05-28 14:55:27 +08:00
|
|
|
_Elem* resize_nofill(int len)
|
2020-11-16 14:47:43 +08:00
|
|
|
{
|
|
|
|
this->reserve(len);
|
2021-05-28 14:55:27 +08:00
|
|
|
#if _MSC_VER >= 1920 // VS2019+
|
2020-11-16 14:47:43 +08:00
|
|
|
std::_Compressed_pair<_Alty, _Scary_val>* _Myval =
|
|
|
|
(std::_Compressed_pair<_Alty, _Scary_val>*)this;
|
|
|
|
_Myval->_Myval2._Mysize = len;
|
|
|
|
auto front = &this->front();
|
|
|
|
front[len] = '\0';
|
|
|
|
return front;
|
|
|
|
#else
|
|
|
|
this->_Eos(len);
|
|
|
|
return &this->front();
|
|
|
|
#endif
|
|
|
|
}
|
|
|
|
};
|
|
|
|
|
2021-05-28 14:55:27 +08:00
|
|
|
template <typename _Elem> static _Elem* prepare(std::basic_string<_Elem>& str, int size)
|
2020-11-16 14:47:43 +08:00
|
|
|
{
|
2021-05-28 14:55:27 +08:00
|
|
|
intrusive_string<_Elem>& helper = (intrusive_string<_Elem>&)str;
|
|
|
|
return helper.resize_nofill(size);
|
2020-11-16 14:47:43 +08:00
|
|
|
}
|
|
|
|
#if defined(_AFX)
|
2021-05-28 14:55:27 +08:00
|
|
|
template <typename _Elem> _Elem* prepare(CStringT<_Elem, StrTraitMFC_DLL<_Elem>>& str, int size)
|
2020-11-16 14:47:43 +08:00
|
|
|
{
|
|
|
|
return str.GetBufferSetLength(size);
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
} // namespace buffer_traits
|
|
|
|
|
|
|
|
template <typename _StringContType>
|
|
|
|
inline _StringContType wcbs2a(const wchar_t* wcb, int len, UINT cp = NTCVT_CP_DEFAULT)
|
|
|
|
{
|
|
|
|
if (len == -1)
|
|
|
|
len = lstrlenW(wcb);
|
|
|
|
_StringContType buffer;
|
|
|
|
int cch;
|
|
|
|
if (len > 0 && (cch = ::WideCharToMultiByte(cp, 0, wcb, len, NULL, 0, NULL, NULL)) > 0)
|
2021-05-28 14:55:27 +08:00
|
|
|
::WideCharToMultiByte(cp, 0, wcb, len, buffer_traits::prepare(buffer, cch), cch, NULL, NULL);
|
2020-11-16 14:47:43 +08:00
|
|
|
return buffer;
|
|
|
|
}
|
|
|
|
|
|
|
|
template <typename _StringContType>
|
|
|
|
inline _StringContType mcbs2w(const char* mcb, int len, UINT cp = NTCVT_CP_DEFAULT)
|
|
|
|
{
|
|
|
|
if (len == -1)
|
|
|
|
len = lstrlenA(mcb);
|
|
|
|
_StringContType buffer;
|
|
|
|
int cch;
|
|
|
|
if (len > 0 && (cch = ::MultiByteToWideChar(cp, 0, mcb, len, NULL, 0)) > 0)
|
2021-05-28 14:55:27 +08:00
|
|
|
::MultiByteToWideChar(cp, 0, mcb, len, buffer_traits::prepare(buffer, cch), cch);
|
2020-11-16 14:47:43 +08:00
|
|
|
|
|
|
|
return buffer;
|
|
|
|
}
|
|
|
|
|
|
|
|
inline int mcbs2w(const char* mcb, int len, wchar_t* wbuf, int wbuf_len, UINT cp = NTCVT_CP_DEFAULT)
|
|
|
|
{
|
|
|
|
if (len == -1)
|
|
|
|
len = lstrlenA(mcb);
|
|
|
|
int cch;
|
|
|
|
if (len > 0 && (cch = ::MultiByteToWideChar(cp, 0, mcb, len, NULL, 0)) > 0)
|
|
|
|
return ::MultiByteToWideChar(cp, 0, mcb, len, wbuf, wbuf_len);
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
inline wchar_t* mcbs2wdup(const char* mcb, int len, int* wbuf_len, UINT cp = NTCVT_CP_DEFAULT)
|
|
|
|
{
|
|
|
|
if (len == -1)
|
|
|
|
len = lstrlenA(mcb);
|
|
|
|
int cch;
|
|
|
|
if (len > 0 && (cch = ::MultiByteToWideChar(cp, 0, mcb, len, NULL, 0)) > 0)
|
|
|
|
{
|
|
|
|
wchar_t* wbuf = (wchar_t*)calloc(cch + 1, 2);
|
|
|
|
*wbuf_len = ::MultiByteToWideChar(cp, 0, mcb, len, wbuf, cch);
|
|
|
|
return wbuf;
|
|
|
|
}
|
|
|
|
|
|
|
|
return (wchar_t*)calloc(1, 2);
|
|
|
|
}
|
|
|
|
|
|
|
|
#if _HAS_CXX17
|
|
|
|
inline std::string from_chars(const std::wstring_view& wcb, UINT cp = NTCVT_CP_DEFAULT)
|
|
|
|
{
|
|
|
|
return wcbs2a<std::string>(wcb.data(), wcb.length(), cp);
|
|
|
|
}
|
|
|
|
|
|
|
|
inline std::wstring from_chars(const std::string_view& mcb, UINT cp = NTCVT_CP_DEFAULT)
|
|
|
|
{
|
|
|
|
return mcbs2w<std::wstring>(mcb.data(), mcb.length(), cp);
|
|
|
|
}
|
|
|
|
#else
|
|
|
|
inline std::string from_chars(const std::wstring& wcb, UINT cp = NTCVT_CP_DEFAULT)
|
|
|
|
{
|
|
|
|
return wcbs2a<std::string>(wcb.c_str(), wcb.length(), cp);
|
|
|
|
}
|
|
|
|
|
|
|
|
inline std::wstring from_chars(const std::string& mcb, UINT cp = NTCVT_CP_DEFAULT)
|
|
|
|
{
|
|
|
|
return mcbs2w<std::wstring>(mcb.c_str(), mcb.length(), cp);
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
|
|
|
inline std::string from_chars(const wchar_t* str, UINT cp = NTCVT_CP_DEFAULT)
|
|
|
|
{
|
|
|
|
return wcbs2a<std::string>(str, -1, cp);
|
|
|
|
}
|
|
|
|
|
|
|
|
inline std::wstring from_chars(const char* str, UINT cp = NTCVT_CP_DEFAULT)
|
|
|
|
{
|
|
|
|
return mcbs2w<std::wstring>(str, -1, cp);
|
|
|
|
}
|
|
|
|
|
|
|
|
// ntcs or std::string to CStringW
|
|
|
|
#if defined(_AFX)
|
|
|
|
inline std::string from_chars(const CStringW& wcb, UINT cp = NTCVT_CP_DEFAULT)
|
|
|
|
{
|
|
|
|
return wcbs2a<std::string>(wcb.GetString(), wcb.GetLength(), cp);
|
|
|
|
}
|
|
|
|
namespace afx
|
|
|
|
{
|
|
|
|
# if _HAS_CXX17
|
|
|
|
inline CStringW from_chars(std::string_view mcb, UINT cp = NTCVT_CP_DEFAULT)
|
|
|
|
{
|
|
|
|
return mcbs2w<CStringW>(mcb.data(), mcb.length(), cp);
|
|
|
|
}
|
|
|
|
# else
|
|
|
|
inline CStringW from_chars(const char* str, UINT cp = NTCVT_CP_DEFAULT)
|
|
|
|
{
|
|
|
|
return mcbs2w<CStringW>(str, -1, cp);
|
|
|
|
}
|
|
|
|
inline CStringW from_chars(const std::string& mcb, UINT cp = NTCVT_CP_DEFAULT)
|
|
|
|
{
|
|
|
|
return mcbs2w<CStringW>(mcb.c_str(), mcb.length(), cp);
|
|
|
|
}
|
|
|
|
# endif
|
|
|
|
} // namespace afx
|
|
|
|
#endif
|
|
|
|
} // namespace ntcvt
|
|
|
|
|
|
|
|
#endif
|