#include "OpStrHelper.h"

#ifdef WIN32

#include <windows.h>

#endif

#define CP_GBK (936) // 简体中文code page

std::wstring OpStrHelper::ansi_2_wstring(const char* in)

{

std::wstring out;

auto len = strlen(in);

#ifdef WIN32

//assert( _mbslen((const unsigned char*)in) == strlen(in) ); // ANSI 要求必须是纯英文

int wbufferlen = (int)::MultiByteToWideChar(CP_ACP, MB_PRECOMPOSED, in, (int)len, NULL, 0);

wchar_t* pwbuffer = new wchar_t[wbufferlen + 4];

wbufferlen = (int)::MultiByteToWideChar(CP_ACP, MB_PRECOMPOSED, in, (int)len, pwbuffer, wbufferlen + 2);

pwbuffer[wbufferlen] = '\0';

out.assign(pwbuffer, wbufferlen);

delete[] pwbuffer;

#elif defined __LINUX__

iconv_t env;

env = iconv_open("WCHAR_T", "MS-ANSI");

size_t out_len = (len + 1) * sizeof(wchar_t);

wchar_t* pbuffer = new wchar_t[len + 1];

memset(pbuffer, 0, out_len);

wchar_t* pResult = pbuffer;

iconv(env, (char**)&in, (size_t*)&len, (char**)&pbuffer, (size_t*)&out_len);

iconv_close(env);

out = pResult;

delete[] pResult;

#endif

return out;

}

std::wstring OpStrHelper::gbk_2_wstring(const char* in)

{

std::wstring out;

auto len = strlen(in);

#ifdef WIN32

int wbufferlen = (int)::MultiByteToWideChar(CP_GBK, MB_PRECOMPOSED, in, (int)len, NULL, 0);

wchar_t* pwbuffer = new wchar_t[wbufferlen + 4];

wbufferlen = (int)::MultiByteToWideChar(CP_GBK, MB_PRECOMPOSED, in, (int)len, pwbuffer, wbufferlen + 2);

pwbuffer[wbufferlen] = '\0';

out.assign(pwbuffer, wbufferlen);

delete[] pwbuffer;

#elif defined __LINUX__

iconv_t env;

env = iconv_open("WCHAR_T", "GBK");

size_t out_len = (len + 1) * sizeof(wchar_t);

wchar_t* pbuffer = new wchar_t[len + 1];

memset(pbuffer, 0, out_len);

wchar_t* pResult = pbuffer;

iconv(env, (char**)&in, (size_t*)&len, (char**)&pbuffer, (size_t*)&out_len);

iconv_close(env);

out = pResult;

delete[] pResult;

#endif

return out;

}

std::wstring OpStrHelper::utf8_2_wstring(const char* in)

{

std::wstring out;

auto len = strlen(in);

wchar_t* pBuf = new wchar_t[len + 1];

size_t out_len = (len + 1) * sizeof(wchar_t);

memset(pBuf, 0, (len + 1) * sizeof(wchar_t));

wchar_t* pResult = pBuf;

#ifdef WIN32

out_len = ::MultiByteToWideChar(CP_UTF8, 0, in, len, pBuf, len * sizeof(wchar_t));

out.assign(pResult, out_len);

#elif defined __LINUX__

iconv_t env;

env = iconv_open("WCHAR_T", "UTF-8");

iconv(env, (char**)&in, (size_t*)&len, (char**)&pBuf, (size_t*)&out_len);

iconv_close(env);

out = pResult;

#endif

delete[] pResult;

pResult = NULL;

return out;

}

std::string OpStrHelper::wstring_2_ansi(const wchar_t* in)

{

std::string out;

auto len = wcslen(in);

#ifdef WIN32

int bufferlen = (int)::WideCharToMultiByte(CP_ACP, 0, in, (int)len, NULL, 0, NULL, NULL);

char* pBuffer = new char[bufferlen + 4];

int out_len = ::WideCharToMultiByte(CP_ACP, 0, in, (int)len, pBuffer, bufferlen + 2, NULL, NULL);

pBuffer[bufferlen] = '\0';

out.assign(pBuffer, out_len);

delete[] pBuffer;

//assert( _mbslen((const unsigned char*)out.c_str()) == strlen(out.c_str()) );

#elif defined __LINUX__

iconv_t env;

env = iconv_open("MS-ANSI", "WCHAR_T");

size_t out_len = len * 2 + 1;

size_t in_len = len * sizeof(wchar_t);

char* pbuffer = new char[out_len];

memset(pbuffer, 0, out_len);

char* pResult = pbuffer;

iconv(env, (char**)&in, (size_t*)&in_len, (char**)&pbuffer, (size_t*)&out_len);

iconv_close(env);

out = pResult;

delete[] pResult;

#endif

return out;

}

std::string OpStrHelper::wstring_2_utf8(const wchar_t* in)

{

std::string out;

auto len = wcslen(in);

size_t out_len = len * 3 + 1;

char* pBuf = new char[out_len];

char* pResult = pBuf;

memset(pBuf, 0, out_len);

#ifdef WIN32

out_len = ::WideCharToMultiByte(CP_UTF8, 0, in, len, pBuf, len * 3, NULL, NULL);

out.assign(pResult, out_len);

#elif defined __LINUX__

iconv_t env;

env = iconv_open("UTF-8", "WCHAR_T");

size_t in_len = len * sizeof(wchar_t);

iconv(env, (char**)&in, (size_t*)&in_len, (char**)&pBuf, (size_t*)&out_len);

iconv_close(env);

out = pResult;

#endif

delete[] pResult;

pResult = NULL;

return out;

}

std::string OpStrHelper::wstring_2_gbk(const wchar_t* in)

{

std::string out;

auto len = wcslen(in);

#ifdef WIN32

int bufferlen = (int)::WideCharToMultiByte(CP_GBK, 0, in, (int)len, NULL, 0, NULL, NULL);

char* pBuffer = new char[bufferlen + 4];

int out_len = ::WideCharToMultiByte(CP_GBK, 0, in, (int)len, pBuffer, bufferlen + 2, NULL, NULL);

pBuffer[bufferlen] = '\0';

out.assign(pBuffer, out_len);

delete[] pBuffer;

#elif defined __LINUX__

iconv_t env;

env = iconv_open("GBK", "WCHAR_T");

size_t out_len = len * 2 + 1;

size_t in_len = len * sizeof(wchar_t);

char* pbuffer = new char[out_len];

memset(pbuffer, 0, out_len);

char* pResult = pbuffer;

iconv(env, (char**)&in, (size_t*)&in_len, (char**)&pbuffer, (size_t*)&out_len);

iconv_close(env);

out = pResult;

delete[] pResult;

#endif

return out;

}