CommandLineHelpers.h

Go to the documentation of this file.

// This file is part of the MercuryDPM project (https://www.mercurydpm.org).
// Copyright (c), The MercuryDPM Developers Team. All rights reserved.
// License: BSD 3-Clause License; see the LICENSE file in the root directory.
 
#ifndef MERCURYDPM_COMMANDLINE_HELPERS_H
#define MERCURYDPM_COMMANDLINE_HELPERS_H
 
#include "Math/ExtendedMath.h"
 
namespace helpers
{
    std::size_t countLeadingDashes(const std::string & s);
 
    bool removeFromCommandline(int& argc, char* argv[], const std::string & varName, int nArgs);
 
    bool readFromCommandLine(int argc, char *argv[], const std::string & varName);
 
    template<typename T>
    T readFromCommandLine(int argc, char *argv[], const std::string & varName, T value)
    {
        const size_t nDashes = countLeadingDashes(varName);
        for (int i = 0; i < argc-1; ++i) {
            if (varName == argv[i]) {
                value = atof(argv[i+1]);
                logger(INFO, "readFromCommandLine: % set to % ", varName.substr(nDashes), value);
                return value;
            }
        }
        //if the variable is not found
        logger(INFO, "readFromCommandLine: % set to default value % ", varName.substr(nDashes), value);
        return value;
    }
 
    template<typename T, size_t n>
    std::array<T,n> readArrayFromCommandLine(int argc, char *argv[], const std::string & varName, std::array<T,n> value)
    {
        const size_t nDashes = countLeadingDashes(varName);
        for (int i = 0; i < argc-1; ++i) {
            if (varName == argv[i]) {
                int j = i + 1;
                std::stringstream out;
                for (auto& v : value) {
                    v = atof(argv[j]);
                    out << v << ' ';
                    ++j;
                }
                logger(INFO, "readArrayFromCommandLine: % set to % ", varName.substr(nDashes), out.str());
                return value;
            }
        }
        //if the variable is not found
        std::stringstream out;
        for (auto& v : value) out << v << ' ';
        logger(INFO, "readArrayFromCommandLine: % set to default value % ", varName.substr(nDashes), out.str());
        return value;
    }
 
    template<typename T>
    std::vector<T> readVectorFromCommandLine(int argc, char *argv[], const std::string & varName, size_t n, std::vector<T> values)
    {
        const size_t nDashes = countLeadingDashes(varName);
        for (int i = 0; i < argc-1; ++i) {
            if (varName == argv[i]) {
                // read until the next argument starts
                values.resize(0);
                std::stringstream out;
                for (int j = i+1; j < argc && argv[j][0] != '-'; ++j) {
                    values.push_back(atof(argv[j]));
                    out << values.back() << ' ';
                }
                logger(INFO, "readVectorFromCommandLine: % set to % ", varName.substr(nDashes), out.str());
                return values;
            }
        }
        //if the variable is not found
        std::stringstream out;
        for (auto& v: values) out << v << ' ';
        logger(INFO, "readVectorFromCommandLine: % set to default value % ", varName.substr(nDashes), out.str());
        return values;
    }
 
    template<>
    std::string readFromCommandLine<std::string>(int argc, char* argv[], const std::string & varName, std::string value);
}
 
#endif // MERCURYDPM_COMMANDLINE_HELPERS_H