goptions.cc

Go to the documentation of this file.

// goptions 
#include "goptions.h"
#include "goptionsConventions.h"
#include "gversion.h"
 
// gemc
#include "gutilities.h"
 
// c++
#include <iostream>
#include <cstring>
 
using namespace std;
 
GOptions::GOptions(int argc, char *argv[], const GOptions &user_defined_options) {
 
    executableName = gutilities::getFileFromPath(argv[0]);
    cout << endl;
 
    // Add user-defined options.
    addGOptions(user_defined_options);
 
    // switches for all everyone
    defineSwitch("gui", "use Graphical User Interface");
    defineSwitch("i", "use interactive batch mode");
    defineOption(
            GVariable("conf_yaml", "saved_configuration", "the yaml filename prefix where all used options are saved"),
            "The default value appends \"_saved_configuration\" to the executable name.");
 
    // version is a special option, not settable by the user
    // it is set by the gversion.h file
    // we add it here so it can be saved to the yaml file
    vector <GVariable> version = {
            {"release",      gversion,      "release version number"},
            {"release_date", grelease_date, "release date"},
            {"Reference",    greference,    "article reference"},
            {"Homepage",     gweb,          "homepage"},
            {"Author",       gauthor,       "author"}
    };
    defineOption(GVERSION_STRING, "version information", version, "Version information. Not settable by user.");
 
    string help = "Levels: \n \n";
    help += " - 0: (default) shush\n";
    help += " - 1: normal information\n";
    help += " - 2: detailed information\n \n";
    help += "Example: -verbosity.general=1 \n \n";
    help += "This option can be repeated.\n \n";
    defineOption("verbosity", "Sets the log verbosity for various classes", option_verbosity_names, help);
 
 
    help = "Debug information Types: \n \n";
    help += " - false: (default): do not print debug information\n";
    help += " - true: print debug information\n\n";
    help += "Example: -debug.general=true \n \n";
    help += "This option can be repeated.\n \n";
    defineOption("debug", "Sets the debug level for various classes", option_verbosity_names, help);
 
    // Process help/version command-line arguments.
    for (int i = 1; i < argc; i++) {
        if (strcmp(argv[i], "-h") == 0 || strcmp(argv[i], "--h") == 0 ||
            strcmp(argv[i], "-help") == 0 || strcmp(argv[i], "--help") == 0) {
            printHelp();
        } else if (strcmp(argv[i], "-hweb") == 0) {
            printWebHelp();
        } else if (strcmp(argv[i], "-v") == 0 || strcmp(argv[i], "--v") == 0 ||
                   strcmp(argv[i], "-version") == 0 || strcmp(argv[i], "--version") == 0) {
            print_version();
            exit(EXIT_SUCCESS);
        } else if (strcmp(argv[i], "help") == 0) {
            printOptionOrSwitchHelp(argv[i + 1]);
            exit(EXIT_SUCCESS);
        }
    }
 
    // finds and parse the yaml files
    yaml_files = findYamls(argc, argv);
    for (auto &yaml_file: yaml_files) {
        cout << " Parsing " << yaml_file << endl;
        setOptionsValuesFromYamlFile(yaml_file);
    }
 
    // Parse command-line arguments (supports both standard YAML–style and dot–notation).
    for (int i = 1; i < argc; i++) {
        string candidate = argv[i];
        if (candidate.empty()) continue;
        if (find(yaml_files.begin(), yaml_files.end(), candidate) != yaml_files.end()) continue;
        if (candidate[0] == '-') {
            string argStr = candidate.substr(1);
            size_t eqPos = argStr.find('=');
            if (eqPos != string::npos) {
                string keyPart = argStr.substr(0, eqPos);
                string valuePart = argStr.substr(eqPos + 1);
                size_t dotPos = keyPart.find('.');
                if (dotPos != string::npos) {
                    // Dot–notation detected (e.g. "debug.general=true")
                    string mainOption = keyPart.substr(0, dotPos);
                    string subOption = keyPart.substr(dotPos + 1);
                    if (doesOptionExist(mainOption)) {
                        auto it = getOptionIterator(mainOption);
                        it->set_sub_option_value(subOption, valuePart);
                    } else {
                        cerr << "The option " << mainOption << " is not known to this system." << endl;
                        exit(EC__NOOPTIONFOUND);
                    }
                } else {
                    // Standard option syntax.
                    if (doesOptionExist(keyPart)) {
                        setOptionValuesFromCommandLineArgument(keyPart, valuePart);
                    } else {
                        cerr << "The option " << keyPart << " is not known to this system." << endl;
                        exit(EC__NOOPTIONFOUND);
                    }
                }
            } else {
                // Treat as a switch.
                const string& possibleSwitch = argStr;
                if (switches.find(possibleSwitch) != switches.end()) {
                    switches[possibleSwitch].turnOn();
                } else {
                    cerr << "The switch " << possibleSwitch << " is not known to this system." << endl;
                    exit(EC__NOOPTIONFOUND);
                }
            }
        } else {
            cerr << "The command-line argument \"" << candidate << "\" is not valid." << endl;
            exit(EC__NOOPTIONFOUND);
        }
    }
 
    // Always print version information.
    print_version();
 
    // Save the final configuration to a YAML file.
    string yamlConf_filename = executableName + "." + getScalarString("conf_yaml") + ".yaml";
    cout << " Saving options to " << yamlConf_filename << endl << endl;
    yamlConf = new std::ofstream(yamlConf_filename);
    saveOptions();
}
 
void GOptions::defineSwitch(const std::string &name, const std::string &description) {
    if (switches.find(name) == switches.end()) {
        switches[name] = GSwitch(description);
    } else {
        std::cerr << FATALERRORL << "The " << YELLOWHHL << name << RSTHHR
                  << " switch is already present." << std::endl;
        exit(EC__DEFINED_SWITCHALREADYPRESENT);
    }
}
 
void GOptions::defineOption(const GVariable &gvar, const std::string &help) {
    if (doesOptionExist(gvar.name)) {
        std::cerr << FATALERRORL << "The " << YELLOWHHL << gvar.name << RSTHHR
                  << " option is already present." << std::endl;
        exit(EC__DEFINED_OPTION_ALREADY_PRESENT);
    } else {
        goptions.emplace_back(gvar, help);
    }
}
 
void GOptions::defineOption(const std::string &name, const std::string &description, const std::vector <GVariable> &g_vars,
                            const std::string &help) {
    if (doesOptionExist(name)) {
        std::cerr << FATALERRORL << "The " << YELLOWHHL << name << RSTHHR
                  << " option is already present." << std::endl;
        exit(EC__DEFINED_OPTION_ALREADY_PRESENT);
    } else {
        goptions.emplace_back(name, description, g_vars, help);
    }
}
 
int GOptions::getScalarInt(const std::string &tag) const {
    auto it = getOptionIterator(tag);
    if (it == goptions.end()) {
        cerr << FATALERRORL << "The option " << YELLOWHHL << tag << RSTHHR
             << " was not found." << endl;
        exit(EC__NOOPTIONFOUND);
    }
    return it->value.begin()->second.as<int>();
}
 
float GOptions::getScalarFloat(const std::string &tag) const {
    auto it = getOptionIterator(tag);
    if (it == goptions.end()) {
        cerr << FATALERRORL << "The option " << YELLOWHHL << tag << RSTHHR
             << " was not found." << endl;
        exit(EC__NOOPTIONFOUND);
    }
    return it->value.begin()->second.as<float>();
}
 
double GOptions::getScalarDouble(const std::string &tag) const {
    auto it = getOptionIterator(tag);
    if (it == goptions.end()) {
        cerr << FATALERRORL << "The option " << YELLOWHHL << tag << RSTHHR
             << " was not found." << endl;
        exit(EC__NOOPTIONFOUND);
    }
    return it->value.begin()->second.as<double>();
}
 
string GOptions::getScalarString(const std::string &tag) const {
    auto it = getOptionIterator(tag);
    if (it == goptions.end()) {
        cerr << FATALERRORL << "The option " << YELLOWHHL << tag << RSTHHR
             << " was not found." << endl;
        exit(EC__NOOPTIONFOUND);
    }
    return it->value.begin()->second.as<string>();
}
 
void GOptions::printOptionOrSwitchHelp(const std::string &tag) const {
    auto switchIt = switches.find(tag);
    if (switchIt != switches.end()) {
        cout << KGRN << "-" << tag << RST << ": " << switchIt->second.getDescription() << endl << endl;
        cout << TPOINTITEM << "Default value is " << (switchIt->second.getStatus() ? "on" : "off") << endl << endl;
        exit(EXIT_SUCCESS);
    }
    for (const auto &goption: goptions) {
        if (goption.name == tag) {
            goption.printHelp(true);
            exit(EXIT_SUCCESS);
        }
    }
    cerr << FATALERRORL << "The " << YELLOWHHL << tag << RSTHHR
         << " option is not known to this system." << endl;
    exit(EC__NOOPTIONFOUND);
}
 
vector <string> GOptions::findYamls(int argc, char *argv[]) {
    vector <string> yaml_files;
    for (int i = 1; i < argc; i++) {
        string arg = argv[i];
        size_t pos = arg.find(".yaml");
        if (pos != string::npos) yaml_files.push_back(arg);
        pos = arg.find(".yml");
        if (pos != string::npos) yaml_files.push_back(arg);
    }
    return yaml_files;
}
 
// checks if the option exists
bool GOptions::doesOptionExist(const std::string &tag) const {
 
    // [&tag] ensures we're referencing the original tag passed to the function
    return std::any_of(goptions.begin(), goptions.end(),
    [&tag](const auto& option) {
        return option.name == tag;
    });
}
 
void GOptions::setOptionsValuesFromYamlFile(const std::string &yaml) {
    YAML::Node config;
    try {
        config = YAML::LoadFile(yaml);
    } catch (YAML::ParserException &e) {
        cerr << FATALERRORL << "Error parsing " << YELLOWHHL << yaml << RSTHHR
             << " yaml file." << endl;
        cerr << e.what() << endl;
        cerr << "Try validating the yaml file with an online yaml validator, e.g., https://www.yamllint.com" << endl;
        exit(EC__YAML_PARSING_ERROR);
    }
 
    for (auto it = config.begin(); it != config.end(); ++it) {
        auto option_name = it->first.as<std::string>();
        auto option_it = getOptionIterator(option_name);
        if (option_it == goptions.end()) {
            if (switches.find(option_name) == switches.end()) {
                cerr << FATALERRORL << "The option or switch " << YELLOWHHL << option_name << RSTHHR
                     << " is not known to this system." << endl;
                exit(EC__NOOPTIONFOUND);
            } else {
                switches[option_name].turnOn();
            }
        } else {
            YAML::NodeType::value type = it->second.Type();
            switch (type) {
                case YAML::NodeType::Scalar:
                    option_it->set_scalar_value(it->second.as<std::string>());
                    break;
                case YAML::NodeType::Sequence:
                    option_it->set_value(it->second);
                    break;
                case YAML::NodeType::Map:
                    option_it->set_value(it->second);
                    break;
                default:
                    break;
            }
        }
    }
}
 
void GOptions::setOptionValuesFromCommandLineArgument(const std::string &optionName, const std::string &possibleYamlNode) {
    YAML::Node node = YAML::Load(possibleYamlNode);
    auto option_it = getOptionIterator(optionName);
    if (node.Type() == YAML::NodeType::Scalar) {
        option_it->set_scalar_value(possibleYamlNode);
    } else {
        option_it->set_value(node);
    }
}
 
std::vector<GOption>::iterator GOptions::getOptionIterator(const std::string &name) {
    return std::find_if(goptions.begin(), goptions.end(),
                        [&name](GOption &option) { return option.name == name; });
}
 
 
std::vector<GOption>::const_iterator GOptions::getOptionIterator(const std::string &name) const {
    return std::find_if(goptions.begin(), goptions.end(),
                        [&name](const GOption &option) { return option.name == name; });
}
 
bool GOptions::getSwitch(const std::string &tag) const {
    auto it = switches.find(tag);
    if (it != switches.end()) {
        return it->second.getStatus();
    } else {
        std::cerr << FATALERRORL << "The switch " << YELLOWHHL << tag << RSTHHR
                  << " was not found." << std::endl;
        exit(EC__NOOPTIONFOUND);
    }
}
 
YAML::Node GOptions::getOptionMapInNode(string option_name, string map_key) const {
    auto sequence_node = getOptionNode(option_name);
    for (auto seq_item : sequence_node) {
        for (auto map_item = seq_item.begin(); map_item != seq_item.end(); ++map_item) {
            if (map_item->first.as<string>() == map_key) {
                return map_item->second;
            }
        }
    }
    cerr << FATALERRORL << "The key " << YELLOWHHL << map_key << RSTHHR
         << " was not found in " << YELLOWHHL << option_name << RSTHHR << endl;
    exit(EC__NOOPTIONFOUND);
    return sequence_node;
}
 
template<typename T>
T GOptions::get_variable_in_option(const YAML::Node &node, const std::string &variable_name, const T &default_value) {
    if (node[variable_name]) {
        return node[variable_name].as<T>();
    }
    return default_value;
}
 
// Explicit template instantiations.
template int GOptions::get_variable_in_option<int>(const YAML::Node &node, const std::string &variable_name, const int &default_value);
template float GOptions::get_variable_in_option<float>(const YAML::Node &node, const std::string &variable_name, const float &default_value);
template double GOptions::get_variable_in_option<double>(const YAML::Node &node, const std::string &variable_name, const double &default_value);
template string GOptions::get_variable_in_option<string>(const YAML::Node &node, const std::string &variable_name, const string &default_value);
template bool GOptions::get_variable_in_option<bool>(const YAML::Node &node, const std::string &variable_name, const bool &default_value);
 
int GOptions::getVerbosityFor(const std::string &tag) const {
    YAML::Node verbosity_node = getOptionNode("verbosity");
    for (auto v : verbosity_node) {
        if (v.begin()->first.as<string>() == tag) {
            return v.begin()->second.as<int>();
        }
    }
 
    // not found. error
    std::cerr << KRED << " Invalid verbosity or debug requested: " << tag << RST << std::endl;
    exit(EC__NOOPTIONFOUND);
}
 
int GOptions::getDebugFor(const std::string &tag) const {
    YAML::Node debug_node = getOptionNode("debug");
    for (auto d : debug_node) {
        if (d.begin()->first.as<string>() == tag) {
            YAML::Node valNode = d.begin()->second;
            if (valNode.IsScalar()) {
                string s = valNode.as<string>();
                if (s == "true") return 1;
                if (s == "false") return 0;
            }
            try {
                return valNode.as<int>();
            } catch (const YAML::BadConversion &) {
                std::cerr << "Invalid debug value for " << tag << std::endl;
                exit(EC__BAD_CONVERSION);
            }
        }
    }
    // not found. error
    std::cerr << KRED << " Invalid verbosity or debug requested: " << tag << RST << std::endl;
    exit(EC__NOOPTIONFOUND);}
 
void GOptions::printHelp() const {
    long int fill_width = string(HELPFILLSPACE).size() + 1;
    cout.fill('.');
    cout << KGRN << KBOLD << " " << executableName << RST << " [options] [yaml files]" << endl << endl;
    cout << " Switches: " << endl << endl;
    for (auto &s : switches) {
        string help = "-" + s.first + RST + " ";
        cout << KGRN << " " << left;
        cout.width(fill_width);
        cout << help;
        cout << ": " << s.second.getDescription() << endl;
    }
    cout << endl;
    cout << " Options: " << endl << endl;
    for (auto &option : goptions) {
        option.printHelp(false);
    }
    cout << endl;
    cout << endl << " Help / Search / Introspection: " << endl << endl;
    vector<string> helps = {
            string("-h, --h, -help, --help") + RST,
            string("print this help and exit"),
            string("-hweb") + RST,
            string("print this help in web format and exit"),
            string("-v, --v, -version, --version") + RST,
            string("print the version and exit\n"),
            string("help <value>") + RST,
            string("print detailed help for option <value> and exit"),
            string("search <value>") + RST,
            string("list all options containing <value> in the description and exit\n")
    };
    unsigned half_help = helps.size() / 2;
    for (unsigned i = 0; i < half_help; i++) {
        cout << KGRN << " " << left;
        cout.width(fill_width);
        cout << helps[i * 2] << ": " << helps[i * 2 + 1] << endl;
    }
    cout << endl;
    cout << " Note: command line options overwrite YAML file(s)." << endl << endl;
    exit(EXIT_SUCCESS);
}
 
 
 
void GOptions::printWebHelp() const {
    exit(EXIT_SUCCESS);
}
 
void GOptions::saveOptions() const {
    for (auto &s : switches) {
        string status = s.second.getStatus() ? "true" : "false";
        *yamlConf << s.first + ": " + status << "," << endl;
    }
    for (const auto &option : goptions) {
        option.saveOption(yamlConf);
    }
    yamlConf->close();
}
 
void GOptions::print_version() {
    string asterisks = "**************************************************************";
    cout << endl << asterisks << endl;
    cout << " " << KGRN << KBOLD << executableName << RST << "  version: " << KGRN << gversion << RST << endl;
    cout << " Released on: " << KGRN << grelease_date << RST << endl;
    cout << " Reference: " << KGRN << greference << RST << endl;
    cout << " Homepage: " << KGRN << gweb << RST << endl;
    cout << " Author: " << KGRN << gauthor << RST << endl << endl;
    cout << asterisks << endl << endl;
}
 
GOptions &operator+=(GOptions &gopts, const GOptions &goptions_to_add) {
    gopts.addGOptions(goptions_to_add);
    return gopts;
}