Skip to content
Snippets Groups Projects
Verified Commit bd64c91c authored by Sébastien Villemot's avatar Sébastien Villemot
Browse files

Fix the interpretation of @#elseif clauses within an @#ifdef or @#ifndef

(cherry picked from commit 5d564eed)
parent 8207886e
No related branches found
No related tags found
No related merge requests found
Pipeline #4307 passed
......@@ -245,8 +245,23 @@ For::interpret(ostream &output, vector<filesystem::path> &paths)
void
If::interpret(ostream &output, vector<filesystem::path> &paths)
{
bool first_clause = true;
for (const auto & [expr, body] : expr_and_body)
try
{
if ((ifdef || ifndef) && first_clause)
{
first_clause = false;
VariablePtr vp = dynamic_pointer_cast<Variable>(expr);
assert(vp);
if ((ifdef && env.isVariableDefined(vp->getName()))
|| (ifndef && !env.isVariableDefined(vp->getName())))
{
interpretBody(body, output, paths);
break;
}
}
else
{
auto tmp = expr->eval();
RealPtr dp = dynamic_pointer_cast<Real>(tmp);
......@@ -260,6 +275,7 @@ If::interpret(ostream &output, vector<filesystem::path> &paths)
break;
}
}
}
catch (StackTrace &ex)
{
ex.push("@#if", location);
......@@ -286,31 +302,3 @@ If::interpretBody(const vector<DirectivePtr> &body, ostream &output, vector<file
statement->interpret(output, paths);
}
}
void
Ifdef::interpret(ostream &output, vector<filesystem::path> &paths)
{
for (const auto & [expr, body] : expr_and_body)
if (VariablePtr vp = dynamic_pointer_cast<Variable>(expr);
dynamic_pointer_cast<BaseType>(expr)
|| (vp && env.isVariableDefined(vp->getName())))
{
interpretBody(body, output, paths);
break;
}
printEndLineInfo(output);
}
void
Ifndef::interpret(ostream &output, vector<filesystem::path> &paths)
{
for (const auto & [expr, body] : expr_and_body)
if (VariablePtr vp = dynamic_pointer_cast<Variable>(expr);
dynamic_pointer_cast<BaseType>(expr)
|| (vp && !env.isVariableDefined(vp->getName())))
{
interpretBody(body, output, paths);
break;
}
printEndLineInfo(output);
}
......@@ -175,24 +175,25 @@ namespace macro
* If there is an `else` statement it is the last element in the vector. Its condition is true.
*/
const vector<pair<ExpressionPtr, vector<DirectivePtr>>> expr_and_body;
const bool ifdef, ifndef;
public:
If(vector<pair<ExpressionPtr, vector<DirectivePtr>>> expr_and_body_arg,
Environment &env_arg, Tokenizer::location location_arg) :
Directive(env_arg, move(location_arg)), expr_and_body{move(expr_and_body_arg)} { }
Environment &env_arg, Tokenizer::location location_arg,
bool ifdef_arg = false, bool ifndef_arg = false) :
Directive(env_arg, move(location_arg)), expr_and_body{move(expr_and_body_arg)},
ifdef{ifdef_arg}, ifndef{ifndef_arg} { }
void interpret(ostream &output, vector<filesystem::path> &paths) override;
protected:
void interpretBody(const vector<DirectivePtr> &body, ostream &output,
vector<filesystem::path> &paths);
};
class Ifdef : public If
{
public:
Ifdef(vector<pair<ExpressionPtr, vector<DirectivePtr>>> expr_and_body_arg,
Environment &env_arg, Tokenizer::location location_arg) :
If(move(expr_and_body_arg), env_arg, move(location_arg)) { }
void interpret(ostream &output, vector<filesystem::path> &paths) override;
If(move(expr_and_body_arg), env_arg, move(location_arg), true, false) { }
};
......@@ -201,8 +202,7 @@ namespace macro
public:
Ifndef(vector<pair<ExpressionPtr, vector<DirectivePtr>>> expr_and_body_arg,
Environment &env_arg, Tokenizer::location location_arg) :
If(move(expr_and_body_arg), env_arg, move(location_arg)) { }
void interpret(ostream &output, vector<filesystem::path> &paths) override;
If(move(expr_and_body_arg), env_arg, move(location_arg), false, true) { }
};
}
#endif
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment