deimos/ParaReader.cpp at master · chebastian/deimos · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
#include "ParaReader.h"


ParaReader::ParaReader()
{
	_keyValues = std::map<std::string, std::string>();
	_switches = std::vector<std::string>();
}


ParaReader::~ParaReader()
{
}

bool IsFlag(const std::string& arg)
{
	return arg.find('-') == 0;
}

ParaReader ParaReader::FromString(const int argc, char* argv[])
{
	auto reader = ParaReader();
	for (int i = 0; i < argc; i++)
	{
		if (IsFlag(argv[i]))
		{
			if (reader.HasSwitch(argv[i]))
			{
				reader.AddKeyValue(argv[i], "");
			}
			else
				reader.AddKeyValue(argv[i],argv[i+1]);
		}
	}

	return reader;
}

ParaReader ParaReader::FromStringWithSwitches(const int argc, char* argv[], std::vector<std::string> switches)
{
	auto reader = ParaReader();
	for (auto sw : switches)
		reader.AddSwitchWithName(sw);

	for (int i = 0; i < argc; i++)
	{
		if (IsFlag(argv[i]))
		{
			if (reader.HasSwitch(argv[i]))
			{
				reader.AddKeyValue(argv[i], "");
			}
			else if(i+1 < argc)
			{
				reader.AddKeyValue(argv[i],argv[i+1]);
			}
		}
	}

	return reader;

}

void ParaReader::AddSwitchWithName(const std::string& name)
{
	if (std::find(std::begin(_switches), std::end(_switches),name) != std::end(_switches))
	{
		return;
	}

	_switches.push_back(name);
}


void ParaReader::AddKeyValue(const std::string& flag, const std::string& value)
{
	_keyValues[flag] = value;
}

std::string ParaReader::GetValue(const std::string& flag)
{
	return _keyValues[flag];
}

int ParaReader::GetNumber(const std::string& flag)
{
	return std::stoi(GetValue(flag));
}

int ParaReader::GetNumberOrdefault(const std::string& flag, int def)
{
	if (_keyValues.find(flag) != _keyValues.end())
		return std::stoi( _keyValues[flag] );

	return def;
}

bool ParaReader::HasFlag(const std::string& flag)
{
	return _keyValues.find(flag) != _keyValues.end();
}
bool ParaReader::HasSwitch(const std::string& flag)
{
	return std::find(std::begin(_switches), std::end(_switches), flag) != std::end(_switches);
}