getopt.cpp

来自「MPICH是MPI的重要研究,提供了一系列的接口函数,为并行计算的实现提供了编程」· C++ 代码 · 共 168 行

#include "GetOpt.h"
#include <tchar.h>
#include <stdio.h>

// Function name	: GetOpt
// Description	    : 
// Return type		: bool 
// Argument         : int &argc
// Argument         : LPTSTR *&argv
// Argument         : LPTSTR flag
bool GetOpt(int &argc, LPTSTR *&argv, LPTSTR flag)
{
	if (flag == NULL)
		return false;

	for (int i=0; i<argc; i++)
	{
		if (_tcsicmp(argv[i], flag) == 0)
		{
			for (int j=i; j<argc; j++)
			{
				argv[j] = argv[j+1];
			}
			argc -= 1;
			return true;
		}
	}
	return false;
}

// Function name	: GetOpt
// Description	    : 
// Return type		: bool 
// Argument         : int &argc
// Argument         : LPTSTR *&argv
// Argument         : LPTSTR flag
// Argument         : int *n
bool GetOpt(int &argc, LPTSTR *&argv, LPTSTR flag, int *n)
{
	if (flag == NULL)
		return false;

	for (int i=0; i<argc; i++)
	{
		if (_tcsicmp(argv[i], flag) == 0)
		{
			if (i+1 == argc)
				return false;
			*n = _ttoi(argv[i+1]);
			for (int j=i; j<argc-1; j++)
			{
				argv[j] = argv[j+2];
			}
			argc -= 2;
			return true;
		}
	}
	return false;
}

// Function name	: GetOpt
// Description	    : 
// Return type		: bool 
// Argument         : int &argc
// Argument         : LPTSTR *&argv
// Argument         : LPTSTR flag
// Argument         : long *n
bool GetOpt(int &argc, LPTSTR *&argv, LPTSTR flag, long *n)
{
	int i;
	if (GetOpt(argc, argv, flag, &i))
	{
		*n = (long)i;
		return true;
	}
	return false;
}

// Function name	: GetOpt
// Description	    : 
// Return type		: bool 
// Argument         : int &argc
// Argument         : LPTSTR *&argv
// Argument         : LPTSTR flag
// Argument         : unsigned long *u
bool GetOpt(int &argc, LPTSTR *&argv, LPTSTR flag, unsigned long *u)
{
	if (flag == NULL)
		return false;

	for (int i=0; i<argc; i++)
	{
		if (_tcsicmp(argv[i], flag) == 0)
		{
			if (i+1 == argc)
				return false;
			*u = (unsigned long)_ttol(argv[i+1]);
			for (int j=i; j<argc-1; j++)
			{
				argv[j] = argv[j+2];
			}
			argc -= 2;
			return true;
		}
	}
	return false;
}

// Function name	: GetOpt
// Description	    : 
// Return type		: bool 
// Argument         : int &argc
// Argument         : LPTSTR *&argv
// Argument         : LPTSTR flag
// Argument         : double *d
bool GetOpt(int &argc, LPTSTR *&argv, LPTSTR flag, double *d)
{
	if (flag == NULL)
		return false;

	for (int i=0; i<argc; i++)
	{
		if (_tcsicmp(argv[i], flag) == 0)
		{
			if (i+1 == argc)
				return false;
			*d = _tcstod(argv[i+1], NULL);
			for (int j=i; j<argc-1; j++)
			{
				argv[j] = argv[j+2];
			}
			argc -= 2;
			return true;
		}
	}
	return false;
}

// Function name	: GetOpt
// Description	    : 
// Return type		: bool 
// Argument         : int &argc
// Argument         : LPTSTR *&argv
// Argument         : LPTSTR flag
// Argument         : LPTSTR str
bool GetOpt(int &argc, LPTSTR *&argv, LPTSTR flag, LPTSTR str)
{
	if (flag == NULL)
		return false;

	for (int i=0; i<argc; i++)
	{
		if (_tcsicmp(argv[i], flag) == 0)
		{
			if (i+1 == argc)
				return false;
			_tcscpy(str, argv[i+1]);
			for (int j=i; j<argc-1; j++)
			{
				argv[j] = argv[j+2];
			}
			argc -= 2;
			return true;
		}
	}
	return false;
}