131 lines
3.8 KiB
C++
131 lines
3.8 KiB
C++
#include <iostream>
|
|
|
|
#include <cxxopts.hpp>
|
|
|
|
#include <anthem/AST.h>
|
|
#include <anthem/Context.h>
|
|
#include <anthem/Translation.h>
|
|
|
|
int main(int argc, char **argv)
|
|
{
|
|
anthem::Context context;
|
|
|
|
cxxopts::Options options("anthem", "Translate ASP programs to the language of first-order theorem provers.");
|
|
|
|
options.add_options()
|
|
("h,help", "Display this help message")
|
|
("v,version", "Display version information")
|
|
("i,input", "Input files", cxxopts::value<std::vector<std::string>>())
|
|
("s,simplify", "Simplify the output")
|
|
("c,complete", "Perform completion")
|
|
("color", "Colorize output (always, never, auto)", cxxopts::value<std::string>()->default_value("auto"))
|
|
("parentheses", "Parenthesis style (normal, full)", cxxopts::value<std::string>()->default_value("normal"))
|
|
("p,log-priority", "Log messages starting from this priority (debug, info, warning, error)", cxxopts::value<std::string>()->default_value("info"));
|
|
|
|
options.parse_positional("input");
|
|
options.positional_help("[<input file...>]");
|
|
|
|
const auto printHelp =
|
|
[&]()
|
|
{
|
|
std::cout << options.help();
|
|
};
|
|
|
|
bool help;
|
|
bool version;
|
|
std::vector<std::string> inputFiles;
|
|
std::string colorPolicyString;
|
|
std::string parenthesisStyleString;
|
|
std::string logPriorityString;
|
|
|
|
try
|
|
{
|
|
const auto parseResult = options.parse(argc, argv);
|
|
|
|
help = (parseResult.count("help") > 0);
|
|
version = (parseResult.count("version") > 0);
|
|
|
|
if (parseResult.count("input") > 0)
|
|
inputFiles = parseResult["input"].as<std::vector<std::string>>();
|
|
|
|
context.performSimplification = (parseResult.count("simplify") > 0);
|
|
context.performCompletion = (parseResult.count("complete") > 0);
|
|
colorPolicyString = parseResult["color"].as<std::string>();
|
|
parenthesisStyleString = parseResult["parentheses"].as<std::string>();
|
|
logPriorityString = parseResult["log-priority"].as<std::string>();
|
|
}
|
|
catch (const std::exception &exception)
|
|
{
|
|
context.logger.log(anthem::output::Priority::Error) << exception.what();
|
|
context.logger.errorStream() << std::endl;
|
|
printHelp();
|
|
return EXIT_FAILURE;
|
|
}
|
|
|
|
if (help)
|
|
{
|
|
printHelp();
|
|
return EXIT_SUCCESS;
|
|
}
|
|
|
|
if (version)
|
|
{
|
|
std::cout << "anthem version 0.1.8" << std::endl;
|
|
return EXIT_SUCCESS;
|
|
}
|
|
|
|
if (colorPolicyString == "auto")
|
|
context.logger.setColorPolicy(anthem::output::ColorStream::ColorPolicy::Auto);
|
|
else if (colorPolicyString == "never")
|
|
context.logger.setColorPolicy(anthem::output::ColorStream::ColorPolicy::Never);
|
|
else if (colorPolicyString == "always")
|
|
context.logger.setColorPolicy(anthem::output::ColorStream::ColorPolicy::Always);
|
|
else
|
|
{
|
|
context.logger.log(anthem::output::Priority::Error) << "unknown color policy “" << colorPolicyString << "”";
|
|
context.logger.errorStream() << std::endl;
|
|
printHelp();
|
|
return EXIT_FAILURE;
|
|
}
|
|
|
|
if (parenthesisStyleString == "normal")
|
|
context.parenthesisStyle = anthem::ast::ParenthesisStyle::Normal;
|
|
else if (parenthesisStyleString == "full")
|
|
context.parenthesisStyle = anthem::ast::ParenthesisStyle::Full;
|
|
else
|
|
{
|
|
context.logger.log(anthem::output::Priority::Error) << "unknown parenthesis style “" << parenthesisStyleString << "”";
|
|
context.logger.errorStream() << std::endl;
|
|
printHelp();
|
|
return EXIT_FAILURE;
|
|
}
|
|
|
|
try
|
|
{
|
|
const auto logPriority = anthem::output::priorityFromName(logPriorityString.c_str());
|
|
context.logger.setLogPriority(logPriority);
|
|
}
|
|
catch (const std::exception &e)
|
|
{
|
|
context.logger.log(anthem::output::Priority::Error) << "unknown log priorty “" << logPriorityString << "”";
|
|
context.logger.errorStream() << std::endl;
|
|
printHelp();
|
|
return EXIT_FAILURE;
|
|
}
|
|
|
|
try
|
|
{
|
|
if (!inputFiles.empty())
|
|
anthem::translate(inputFiles, context);
|
|
else
|
|
anthem::translate("std::cin", std::cin, context);
|
|
}
|
|
catch (const std::exception &e)
|
|
{
|
|
context.logger.log(anthem::output::Priority::Error) << e.what();
|
|
return EXIT_FAILURE;
|
|
}
|
|
|
|
return EXIT_SUCCESS;
|
|
}
|