control.cpp

来自「我的红黑树的c++实现。主要特点是可以用dot工具把红黑树画出来」· C++ 代码 · 共 44 行

// file control.cpp
//created by alpha 2008.11.3

#include "../headers/control.h"
#include <map>
#include <string>
#include <iostream>

using namespace std;

int controlInsert = 0;
int controlKeepDot = 0;

map<string, int> args;
typedef map<string, int>::value_type argsValType;

void controlInit()           // enrol arguments
{
	args.insert(argsValType("-ins", 0));
	args.insert(argsValType("-dot", 0));
}

void controlArgs(int argc, char *argv[])
{
	controlInit();
	int i = 1;
	for(; i < argc; i++)
	{	
		string current = argv[i];
		if (args.count(current))
			args[current] = 1;
		else
		{
			cerr<<"no this argument!\n";
			cout<<"help:\n";
			cout<<"-ins: show process of rbtree_insert;\n";
			cout<<"-dot: keep dot document;\n";
			exit(0);
		}
	}
	controlInsert = args["-ins"];
	controlKeepDot = args["-dot"];
}