99 lines
2.9 KiB
C++
99 lines
2.9 KiB
C++
#ifndef __ANTHEM__UTILS_H
|
|
#define __ANTHEM__UTILS_H
|
|
|
|
#include <iostream>
|
|
|
|
#include <clingo.hh>
|
|
|
|
#include <anthem/Context.h>
|
|
#include <anthem/input/Location.h>
|
|
|
|
namespace anthem
|
|
{
|
|
|
|
////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
// Utils
|
|
//
|
|
////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
|
|
template<class T1, class T2>
|
|
T1 location_cast(const T2 &location);
|
|
|
|
////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
|
|
template<>
|
|
inline input::Location location_cast(const Clingo::Location &location)
|
|
{
|
|
return {location.begin_file(), location.end_file(), location.begin_line(), location.end_line(),
|
|
location.begin_column(), location.end_column()};
|
|
}
|
|
|
|
////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
|
|
inline void throwErrorAtLocation(const Clingo::Location &clingoLocation, const char *errorMessage,
|
|
Context &context)
|
|
{
|
|
const auto location = location_cast<input::Location>(clingoLocation);
|
|
|
|
// TODO: think about removing this to avoid double error messages
|
|
context.logger.log(output::Priority::Error, location, errorMessage);
|
|
|
|
throw std::runtime_error(errorMessage);
|
|
}
|
|
|
|
////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
|
|
inline bool isPrefix(const char *prefix, const char *string)
|
|
{
|
|
const auto prefixLength = std::strlen(prefix);
|
|
const auto stringLength = std::strlen(string);
|
|
|
|
if (stringLength < prefixLength)
|
|
return false;
|
|
|
|
return std::strncmp(prefix, string, prefixLength) == 0;
|
|
}
|
|
|
|
////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
|
|
constexpr const auto AuxiliaryHeadVariablePrefix = "V";
|
|
constexpr const auto AuxiliaryBodyVariablePrefix = "X";
|
|
constexpr const auto AnonymousVariablePrefix = "A";
|
|
constexpr const auto UserVariablePrefix = "_";
|
|
|
|
////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
|
|
inline bool isReservedVariableName(const char *variableName)
|
|
{
|
|
if (!isPrefix(AuxiliaryBodyVariablePrefix, variableName)
|
|
&& !isPrefix(AuxiliaryHeadVariablePrefix, variableName)
|
|
&& !isPrefix(AnonymousVariablePrefix, variableName))
|
|
{
|
|
return false;
|
|
}
|
|
|
|
assert(std::strlen(AuxiliaryBodyVariablePrefix) == std::strlen(AuxiliaryHeadVariablePrefix));
|
|
assert(std::strlen(AuxiliaryBodyVariablePrefix) == std::strlen(AnonymousVariablePrefix));
|
|
assert(std::strlen(AuxiliaryBodyVariablePrefix) == std::strlen(UserVariablePrefix));
|
|
|
|
const auto prefixLength = std::strlen(AuxiliaryBodyVariablePrefix);
|
|
|
|
if (strlen(variableName) == prefixLength)
|
|
return false;
|
|
|
|
const char *suffix = variableName + prefixLength;
|
|
|
|
for (const auto *i = suffix; *i != '\0'; i++)
|
|
if (!std::isdigit(*i))
|
|
return false;
|
|
|
|
return true;
|
|
}
|
|
|
|
////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
|
|
}
|
|
|
|
#endif
|