Files
pytorch/c10/util/Unicode.cpp
Scott Wolchok 44cc873fba [PyTorch] Autoformat c10 (#56830)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/56830

Opt into formatting on GitHub and format everything. This is a trial run before turning on formatting for more and eventually all of the codebase.

Test Plan: CI

Reviewed By: zertosh

Differential Revision: D27979080

fbshipit-source-id: a80f0c48691c08ae8ca0af06377b87e6a2351151
2021-04-30 21:23:28 -07:00

50 lines
1.1 KiB
C++

#include <c10/util/Unicode.h>
namespace c10 {
#if defined(_WIN32)
std::wstring u8u16(const std::string& str) {
if (str.empty()) {
return std::wstring();
}
int size_needed = MultiByteToWideChar(
CP_UTF8, 0, str.c_str(), static_cast<int>(str.size()), NULL, 0);
TORCH_CHECK(size_needed > 0, "Error converting the content to Unicode");
std::wstring wstr(size_needed, 0);
MultiByteToWideChar(
CP_UTF8,
0,
str.c_str(),
static_cast<int>(str.size()),
&wstr[0],
size_needed);
return wstr;
}
std::string u16u8(const std::wstring& wstr) {
if (wstr.empty()) {
return std::string();
}
int size_needed = WideCharToMultiByte(
CP_UTF8,
0,
wstr.c_str(),
static_cast<int>(wstr.size()),
NULL,
0,
NULL,
NULL);
TORCH_CHECK(size_needed > 0, "Error converting the content to UTF8");
std::string str(size_needed, 0);
WideCharToMultiByte(
CP_UTF8,
0,
wstr.c_str(),
static_cast<int>(wstr.size()),
&str[0],
size_needed,
NULL,
NULL);
return str;
}
#endif
} // namespace c10