diff --git a/include/anthem/Utils.h b/include/anthem/Utils.h index f10f1f5..2f6456e 100644 --- a/include/anthem/Utils.h +++ b/include/anthem/Utils.h @@ -13,6 +13,7 @@ namespace anthem constexpr const auto HeadVariablePrefix = "V"; constexpr const auto BodyVariablePrefix = "X"; constexpr const auto UserVariablePrefix = "U"; +constexpr const auto IntegerVariablePrefix = "N"; //////////////////////////////////////////////////////////////////////////////////////////////////// diff --git a/include/anthem/output/AST.h b/include/anthem/output/AST.h index 7aa1957..20e04eb 100644 --- a/include/anthem/output/AST.h +++ b/include/anthem/output/AST.h @@ -35,6 +35,7 @@ struct PrintContext std::map userVariableIDs; std::map headVariableIDs; std::map bodyVariableIDs; + std::map integerVariableIDs; const Context &context; }; @@ -339,6 +340,9 @@ inline output::ColorStream &print(output::ColorStream &stream, const VariableDec return (stream << output::Variable(variableName.c_str())); }; + if (variableDeclaration.domain == Domain::Integer) + return printVariableDeclaration(IntegerVariablePrefix, printContext.integerVariableIDs); + switch (variableDeclaration.type) { case VariableDeclaration::Type::UserDefined: