/*
	GETOPT FOR TCHAR, COPYRIGHT@2017~2019 BY LEO, VERSION 1.0
	TGETOPT.HPP
*/

#ifndef _TGETOPT_HPP
#define _TGETOPT_HPP USED

#if !( defined(_WINDOWS_H) || defined(_WINDOWS_) )
#include <windows.h>
#endif

#if !( defined(_TCHAR_H) || defined(_INC_TCHAR) )
#include   <tchar.h>
#endif

// TCHARFORMAT 格式化宏
#if defined(_UNICODE) || defined(UNICODE)
#define TCHARFORMAT WCHAR
#else
#define TCHARFORMAT CHAR
#endif

// 开关解析宏名
#define _OPT_TEOF -1
#define _OPT_TILL -2
#define _OPT_TERR -3

// 开关解析变量
INT OPTIND=1, OPTOPT, UNOPTIND=-1;
TCHAR* OPTARG;

//函数声明
INT   _tgetopt(INT, TCHAR**, TCHAR*);
TCHAR* V2T(VOID*, BOOL);
VOID*  T2V(TCHAR*, BOOL);

// TCHAR 版 getopt
INT _tgetopt(INT nargc, TCHAR* nargv[], TCHAR* ostr)
{
	static TCHAR* place = (TCHAR*)_T("");
	static TCHAR* lastostr = NULL;
	register TCHAR* oli;

	if(ostr!=lastostr)
	{
		lastostr=ostr;
		place=(TCHAR*)_T("");
	}

	if(!*place)
	{
		if(
		    (OPTIND>=nargc)                           ||
		    (*(place=nargv[OPTIND]) !=(TCHAR)_T('-')) ||
		    (!*(++place))
		)
		{
			if(*place !=(TCHAR)_T('-') && OPTIND <nargc)
			{
				place =(TCHAR*)_T("");
				if(UNOPTIND == -1)
				{
					UNOPTIND = OPTIND++;
					return _OPT_TILL;
				}
				else
				{
					return _OPT_TERR;
				}
			}

			place=(TCHAR*)_T("");
			return _OPT_TEOF;
		}
		if(*place == (TCHAR)_T('-') && *(place+1) == (TCHAR)_T('\0'))
		{
			++OPTIND;
			return _OPT_TEOF;
		}
	}

	if(
	    (OPTOPT=*place++) == (TCHAR)_T(':') ||
	    !(oli=(TCHAR*)_tcschr((TCHARFORMAT*)ostr, (TCHAR)OPTOPT))
	)
	{
		if(!*place)
		{
			++OPTIND;
		}
	}

	if(oli != NULL && *(++oli) !=(TCHAR)_T(':'))
	{
		OPTARG=NULL;
		if(!*place)
		{
			++OPTIND;
		}
	}
	else
	{
		if(*place)
		{
			OPTARG=place;
		}
		else if(nargc <= ++OPTIND)
		{
			place=(TCHAR*)_T("");
		}
		else
		{
			OPTARG=nargv[OPTIND];
		}
		place=(TCHAR*)_T("");
		++OPTIND;
	}
	return OPTOPT;
}

// VOID* 转 TCHAR* 
TCHAR* V2T(VOID* csDATA, BOOL wcsMODE)
{
#if defined(_UNICODE) || defined(UNICODE)

	if(wcsMODE == TRUE)
	{
		return (TCHAR*)csDATA;
	}
	else
	{
		INT wLEN = MultiByteToWideChar(CP_ACP, MB_PRECOMPOSED, (CHAR*)csDATA, -1, NULL, 0);
		WCHAR* wsDATA = (WCHAR*)malloc(wLEN + 1);
		MultiByteToWideChar(CP_ACP, MB_PRECOMPOSED, (CHAR*)csDATA, -1, wsDATA, wLEN);
		wsDATA[wLEN]=L'\0';
		return (TCHAR*)wsDATA;
	}

#else

	if(wcsMODE == TRUE)
	{
		INT aLEN = WideCharToMultiByte(CP_ACP, (DWORD)NULL, (WCHAR*)csDATA, -1, NULL, 0, NULL, NULL);
		CHAR* asDATA = (CHAR*)malloc(aLEN + 1);
		WideCharToMultiByte(CP_ACP, (DWORD)NULL, (WCHAR*)csDATA, -1, asDATA, aLEN, NULL, NULL);
		asDATA[aLEN] = '\0';
		return (TCHAR*)asDATA;
	}
	else
	{
		return (TCHAR*)csDATA;
	}

#endif
}

// TCHAR* 转 VOID* 
VOID* T2V(TCHAR* csDATA, BOOL wcsMODE)
{
#if defined(_UNICODE) || defined(UNICODE)

	if(wcsMODE == TRUE)
	{
		return (VOID*)csDATA;
	}
	else
	{
		INT aLEN = WideCharToMultiByte(CP_ACP, (DWORD)NULL, (WCHAR*)csDATA, -1, NULL, 0, NULL, NULL);
		CHAR* asDATA = (CHAR*)malloc(aLEN + 1);
		WideCharToMultiByte(CP_ACP, (DWORD)NULL, (WCHAR*)csDATA, -1, asDATA, aLEN, NULL, NULL);
		asDATA[aLEN] = '\0';
		return (VOID*)asDATA;
	}

#else

	if(wcsMODE == TRUE)
	{
		INT wLEN = MultiByteToWideChar(CP_ACP, MB_PRECOMPOSED, (CHAR*)csDATA, -1, NULL, 0);
		WCHAR* wsDATA = (WCHAR*)malloc(wLEN + 1);
		MultiByteToWideChar(CP_ACP, MB_PRECOMPOSED, (CHAR*)csDATA, -1, wsDATA, wLEN);
		wsDATA[wLEN]=L'\0';
		return (VOID*)wsDATA;
	}
	else
	{
		return (VOID*)csDATA;
	}

#endif
}

#endif
复制代码