This commit is contained in:
David Rose 2011-08-30 02:55:15 +00:00
parent 9150bd2f6c
commit 9c4990c88a
2 changed files with 118 additions and 0 deletions

View File

@ -0,0 +1,85 @@
// Filename: wstring_encode.cxx
// Created by: drose (29Aug11)
//
////////////////////////////////////////////////////////////////////
//
// PANDA 3D SOFTWARE
// Copyright (c) Carnegie Mellon University. All rights reserved.
//
// All use of this software is subject to the terms of the revised BSD
// license. You should have received a copy of this license along
// with this source code in a file named "LICENSE."
//
////////////////////////////////////////////////////////////////////
#include "wstring_encode.h"
#ifdef _WIN32
#include <windows.h>
#endif // _WIN32
#ifdef _WIN32
////////////////////////////////////////////////////////////////////
// Function: wstring_to_string
// Description: Encodes std::wstring to std::string using UTF-8.
////////////////////////////////////////////////////////////////////
bool
wstring_to_string(string &result, const wstring &source) {
bool success = false;
int size = WideCharToMultiByte(CP_UTF8, 0, source.data(), source.length(),
NULL, 0, NULL, NULL);
if (size > 0) {
char *buffer = new char[size + 1];
int rc = WideCharToMultiByte(CP_UTF8, 0, source.data(), source.length(),
buffer, size, NULL, NULL);
if (rc != 0) {
buffer[size] = 0;
result = buffer;
success = true;
}
delete[] buffer;
}
return success;
}
#endif // _WIN32
#ifdef _WIN32
////////////////////////////////////////////////////////////////////
// Function: string_to_wstring
// Description: Decodes std::string to std::wstring using UTF-8.
////////////////////////////////////////////////////////////////////
bool
string_to_wstring(wstring &result, const string &source) {
bool success = false;
int size = MultiByteToWideChar(CP_UTF8, 0, source.data(), source.length(),
NULL, 0);
if (size > 0) {
wchar_t *buffer = new wchar_t[size + 1];
int rc = MultiByteToWideChar(CP_UTF8, 0, source.data(), source.length(),
buffer, size);
if (rc != 0) {
buffer[size] = 0;
result = buffer;
success = true;
}
delete[] buffer;
}
return success;
}
#endif // _WIN32
////////////////////////////////////////////////////////////////////
// Function: wstring ostream operator
// Description: Converts the wstring to utf-8 for output.
////////////////////////////////////////////////////////////////////
ostream &
operator << (ostream &out, const wstring &str) {
string result;
if (wstring_to_string(result, str)) {
out << result;
}
return out;
}

View File

@ -0,0 +1,33 @@
// Filename: wstring_encode.h
// Created by: drose (29Jun09)
//
////////////////////////////////////////////////////////////////////
//
// PANDA 3D SOFTWARE
// Copyright (c) Carnegie Mellon University. All rights reserved.
//
// All use of this software is subject to the terms of the revised BSD
// license. You should have received a copy of this license along
// with this source code in a file named "LICENSE."
//
////////////////////////////////////////////////////////////////////
#ifndef WSTRING_ENCODE_H
#define WSTRING_ENCODE_H
#include <string>
using namespace std;
// Presently, these two functions are implemented only for Windows,
// which is the only place they are needed. (Only Windows requires
// wstrings for filenames.)
#ifdef _WIN32
bool wstring_to_string(string &result, const wstring &source);
bool string_to_wstring(wstring &result, const string &source);
ostream &operator << (ostream &out, const wstring &str);
#endif // _WIN32
#endif