patrick/anthem
1
0
Fork 0
Code Issues 15 Releases 21 Activity

Add suffix to variables with known domain

Browse Source 
This adds suffixes to integer and general variables, where the domain is
explicitly known.
This commit is contained in:
Patrick Lühne 2018-04-20 16:32:18 +02:00
parent aedb7e9bbd
commit d1ee6eca19
Signed by: patrick
GPG Key ID: 05F3611E97A70ABF
1 changed files with 17 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
include/anthem/output/AST.h
View File

@ -322,6 +322,22 @@ inline output::ColorStream &print(output::ColorStream &stream, const Variable &v
inline output::ColorStream &print(output::ColorStream &stream, const VariableDeclaration &variableDeclaration, PrintContext &printContext, bool) inline output::ColorStream &print(output::ColorStream &stream, const VariableDeclaration &variableDeclaration, PrintContext &printContext, bool)
{ {
const auto domainSuffix =
[&variableDeclaration]()
{
switch (variableDeclaration.domain)
{
case Domain::Unknown:
return "";
case Domain::General:
return "g";
case Domain::Integer:
return "i";
}
return "";
};
const auto printVariableDeclaration = const auto printVariableDeclaration =
[&](const auto *prefix, auto &variableIDs) -> output::ColorStream & [&](const auto *prefix, auto &variableIDs) -> output::ColorStream &
{ {
@ -334,7 +350,7 @@ inline output::ColorStream &print(output::ColorStream &stream, const VariableDec
matchingVariableID = emplaceResult.first; matchingVariableID = emplaceResult.first;
} }
const auto variableName = std::string(prefix) + std::to_string(matchingVariableID->second); const auto variableName = std::string(prefix) + std::to_string(matchingVariableID->second) + domainSuffix();
return (stream << output::Variable(variableName.c_str())); return (stream << output::Variable(variableName.c_str()));
}; };