This repository has been archived on 2023-07-19. You can view files and clone it, but cannot push or open issues or pull requests.
plasp/app/main.cpp

181 lines
5.0 KiB
C++
Raw Normal View History

#include <algorithm>
2016-05-20 15:29:24 +02:00
#include <iostream>
#include <string>
2016-05-20 15:29:24 +02:00
#include <boost/program_options.hpp>
#include <plasp/LanguageDetection.h>
#include <plasp/output/ColorStream.h>
#include <plasp/output/TranslatorException.h>
#include <plasp/pddl/Description.h>
#include <plasp/pddl/TranslatorASP.h>
2016-05-20 15:29:24 +02:00
#include <plasp/sas/Description.h>
#include <plasp/sas/TranslatorASP.h>
2016-05-20 15:29:24 +02:00
int main(int argc, char **argv)
{
namespace po = boost::program_options;
bool warningsAsErrors;
2016-05-20 15:29:24 +02:00
po::options_description description("Allowed options");
description.add_options()
2016-11-29 02:01:23 +01:00
("help,h", "Display this help message")
("version,v", "Display version information")
("input,i", po::value<std::vector<std::string>>(), "Input files (in PDDL or SAS format)")
("language,l", po::value<std::string>()->default_value("auto"), "Input language (pddl, sas, auto)")
("color", po::value<std::string>()->default_value("auto"), "Colorize output (always, never, auto)")
("log-priority,p", po::value<std::string>()->default_value("warning"), "Log messages starting from this priority (debug, info, warning, error)")
("warnings-as-errors", po::bool_switch(&warningsAsErrors), "Treat warnings as errors");
2016-05-20 15:29:24 +02:00
po::positional_options_description positionalOptionsDescription;
positionalOptionsDescription.add("input", -1);
po::variables_map variablesMap;
2016-05-24 16:24:37 +02:00
const auto printHelp =
[&]()
{
2016-11-29 02:01:23 +01:00
std::cout << "Usage: plasp [options] file..." << std::endl;
std::cout << "Translate PDDL to ASP." << std::endl << std::endl;
2016-05-24 16:24:37 +02:00
std::cout << description;
};
plasp::output::Logger logger;
2016-05-24 16:24:37 +02:00
try
{
po::store(po::command_line_parser(argc, argv)
.options(description)
.positional(positionalOptionsDescription)
.run(),
variablesMap);
po::notify(variablesMap);
}
catch (const po::error &e)
{
logger.log(plasp::output::Priority::Error, e.what());
std::cout << std::endl;
2016-05-24 16:24:37 +02:00
printHelp();
return EXIT_FAILURE;
}
2016-05-20 15:29:24 +02:00
if (variablesMap.count("help"))
{
2016-05-24 16:24:37 +02:00
printHelp();
return EXIT_SUCCESS;
}
if (variablesMap.count("version"))
{
2016-09-02 15:46:58 +02:00
std::cout << "plasp version 3.0.3-git" << std::endl;
2016-05-20 15:29:24 +02:00
return EXIT_SUCCESS;
}
if (warningsAsErrors)
logger.setAbortPriority(plasp::output::Priority::Warning);
const auto colorPolicy = variablesMap["color"].as<std::string>();
if (colorPolicy == "auto")
logger.setColorPolicy(plasp::output::ColorStream::ColorPolicy::Auto);
else if (colorPolicy == "never")
logger.setColorPolicy(plasp::output::ColorStream::ColorPolicy::Never);
else if (colorPolicy == "always")
logger.setColorPolicy(plasp::output::ColorStream::ColorPolicy::Always);
else
{
logger.log(plasp::output::Priority::Error, "unknown color policy “" + colorPolicy + "");
std::cout << std::endl;
printHelp();
return EXIT_FAILURE;
}
const auto logPriorityString = variablesMap["log-priority"].as<std::string>();
try
{
const auto logPriority = plasp::output::priorityFromName(logPriorityString.c_str());
logger.setLogPriority(logPriority);
}
catch (const std::exception &e)
{
logger.log(plasp::output::Priority::Error, ("unknown log priorty “" + logPriorityString + "").c_str());
logger.errorStream() << std::endl;
printHelp();
return EXIT_FAILURE;
}
try
{
2017-05-12 14:17:57 +02:00
tokenize::Tokenizer<tokenize::CaseInsensitiveTokenizerPolicy> tokenizer;
if (variablesMap.count("input"))
{
const auto &inputFiles = variablesMap["input"].as<std::vector<std::string>>();
std::for_each(inputFiles.cbegin(), inputFiles.cend(),
[&](const auto &inputFile)
{
2017-05-12 14:17:57 +02:00
tokenizer.read(inputFile);
});
}
else
2017-05-12 14:17:57 +02:00
tokenizer.read("std::cin", std::cin);
const auto detectLanguage =
[&]()
{
const auto languageName = variablesMap["language"].as<std::string>();
2016-11-29 02:01:23 +01:00
const auto language = plasp::Language::fromString(languageName);
if (language == plasp::Language::Type::Automatic)
2017-05-12 14:17:57 +02:00
return plasp::detectLanguage(tokenizer);
2016-11-29 02:01:23 +01:00
return language;
};
const auto language = detectLanguage();
if (language == plasp::Language::Type::Unknown)
{
logger.log(plasp::output::Priority::Error, "unknown input language");
2016-06-14 12:47:39 +02:00
std::cout << std::endl;
printHelp();
return EXIT_FAILURE;
}
if (language == plasp::Language::Type::PDDL)
{
2017-05-12 14:17:57 +02:00
auto context = plasp::pddl::Context(std::move(tokenizer), logger);
auto description = plasp::pddl::Description::fromContext(context);
const auto translator = plasp::pddl::TranslatorASP(description, logger.outputStream());
translator.translate();
}
else if (language == plasp::Language::Type::SAS)
{
2017-05-12 14:17:57 +02:00
const auto description = plasp::sas::Description::fromTokenizer(std::move(tokenizer));
const auto translator = plasp::sas::TranslatorASP(description, logger.outputStream());
translator.translate();
}
}
2017-05-12 14:17:57 +02:00
catch (const tokenize::TokenizerException &e)
{
logger.log(plasp::output::Priority::Error, e.location(), e.message().c_str());
2016-06-14 12:47:39 +02:00
return EXIT_FAILURE;
}
catch (const plasp::output::TranslatorException &e)
{
logger.log(plasp::output::Priority::Error, e.what());
2016-06-14 12:47:39 +02:00
return EXIT_FAILURE;
}
catch (const std::exception &e)
{
logger.log(plasp::output::Priority::Error, e.what());
2016-06-14 12:47:39 +02:00
return EXIT_FAILURE;
}
2016-05-20 15:29:24 +02:00
return EXIT_SUCCESS;
}