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
106
107
108
109
110
111
112
113
|
/* Copyright (C) 2018 Wildfire Games.
* This file is part of 0 A.D.
*
* 0 A.D. is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 2 of the License, or
* (at your option) any later version.
*
* 0 A.D. is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with 0 A.D. If not, see <http://www.gnu.org/licenses/>.
*/
#include "precompiled.h"
#include "CmdLineArgs.h"
#include "lib/sysdep/sysdep.h"
namespace
{
// Simple matcher for elements of the arguments container.
class IsKeyEqualTo
{
public:
IsKeyEqualTo(const CStr& value) : m_Value(value) {}
bool operator()(const std::pair<CStr, CStr>& p) const
{
return p.first == m_Value;
}
private:
const CStr m_Value;
};
} // namespace
CmdLineArgs::CmdLineArgs(int argc, const char* argv[])
{
if (argc >= 1)
{
std::string arg0(argv[0]);
// avoid OsPath complaining about mixing both types of separators,
// which happens when running in the VC2010 debugger
std::replace(arg0.begin(), arg0.end(), '/', SYS_DIR_SEP);
m_Arg0 = arg0;
}
for (int i = 1; i < argc; ++i)
{
// Only accept arguments that start with '-'
if (argv[i][0] != '-')
{
m_ArgsWithoutName.emplace_back(argv[i]);
continue;
}
// Allow -arg and --arg
char offset = argv[i][1] == '-' ? 2 : 1;
CStr name, value;
// Check for "-arg=value"
const char* eq = strchr(argv[i], '=');
if (eq)
{
name = CStr(argv[i]+offset, eq-argv[i]-offset);
value = CStr(eq+1);
}
else
name = CStr(argv[i]+offset);
m_Args.emplace_back(std::move(name), std::move(value));
}
}
bool CmdLineArgs::Has(const CStr& name) const
{
return std::any_of(m_Args.begin(), m_Args.end(), IsKeyEqualTo(name));
}
CStr CmdLineArgs::Get(const CStr& name) const
{
ArgsT::const_iterator it = std::find_if(m_Args.begin(), m_Args.end(), IsKeyEqualTo(name));
return it != m_Args.end() ? it->second : "";
}
std::vector<CStr> CmdLineArgs::GetMultiple(const CStr& name) const
{
std::vector<CStr> values;
ArgsT::const_iterator it = m_Args.begin();
while ((it = std::find_if(it, m_Args.end(), IsKeyEqualTo(name))) != m_Args.end())
{
values.push_back(it->second);
// Start searching from the next one in the next iteration
++it;
}
return values;
}
OsPath CmdLineArgs::GetArg0() const
{
return m_Arg0;
}
std::vector<CStr> CmdLineArgs::GetArgsWithoutName() const
{
return m_ArgsWithoutName;
}
|
