CmdLineArgs.cpp

Go to the documentation of this file.

/* Copyright (C) 2009 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"

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] != '-')
            continue;

        CStr name, value;

        // Check for "-arg=value"
        const char* eq = strchr(argv[i], '=');
        if (eq)
        {
            name = CStr(argv[i]+1, eq-argv[i]-1);
            value = CStr(eq+1);
        }
        else
        {
            name = CStr(argv[i]+1);
        }

        m_Args.push_back(make_pair(name, value));
    }
}

template<typename T>
struct first_equals
{
    T x;
    first_equals(const T& x) : x(x) {}
    template<typename S> bool operator()(const S& v) { return v.first == x; }
};

bool CmdLineArgs::Has(const char* name) const
{
    return find_if(m_Args.begin(), m_Args.end(), first_equals<CStr>(name)) != m_Args.end();
}

CStr CmdLineArgs::Get(const char* name) const
{
    ArgsT::const_iterator it = find_if(m_Args.begin(), m_Args.end(), first_equals<CStr>(name));
    if (it != m_Args.end())
        return it->second;
    else
        return "";
}

std::vector<CStr> CmdLineArgs::GetMultiple(const char* name) const
{
    std::vector<CStr> values;
    ArgsT::const_iterator it = m_Args.begin();
    while (1)
    {
        it = find_if(it, m_Args.end(), first_equals<CStr>(name));
        if (it == m_Args.end())
            break;
        values.push_back(it->second);
        ++it; // start searching from the next one in the next iteration
    }
    return values;
}

OsPath CmdLineArgs::GetArg0() const
{
    return m_Arg0;
}