ZYM/BH-Python-Interpreter-2023
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: bug in If_stmt

Browse Source
This commit is contained in:
ZhuangYumin
2023-11-09 20:46:54 +08:00
parent c2fafb76bd
commit 29eaf69558
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/Evalvisitor.cpp
View File

@ -170,6 +170,8 @@ std::any EvalVisitor::visitFlow_stmt(Python3Parser::Flow_stmtContext *ctx) {
if (ctx->break_stmt()) return visitBreak_stmt(ctx->break_stmt());
if (ctx->continue_stmt()) return visitContinue_stmt(ctx->continue_stmt());
if (ctx->return_stmt()) return visitReturn_stmt(ctx->return_stmt());
throw FatalError("Unknown flow statement");
return nullptr;
}
std::any EvalVisitor::visitBreak_stmt(Python3Parser::Break_stmtContext *ctx) {
@ -198,6 +200,8 @@ std::any EvalVisitor::visitCompound_stmt(
if (ctx->if_stmt()) return visitIf_stmt(ctx->if_stmt());
if (ctx->while_stmt()) return visitWhile_stmt(ctx->while_stmt());
if (ctx->funcdef()) return visitFuncdef(ctx->funcdef());
throw FatalError("Unknown compound statement");
return nullptr;
}
std::any EvalVisitor::visitIf_stmt(Python3Parser::If_stmtContext *ctx) {
@ -209,7 +213,7 @@ std::any EvalVisitor::visitIf_stmt(Python3Parser::If_stmtContext *ctx) {
for (int i = 0; i < test_list.size(); i++)
if (Any2Bool(DeQuate(visitTest(test_list[i]), Variables)))
return visitSuite(suite_list[i]);
if (test_list.size() - suite_list.size() == 1)
if (test_list.size() - suite_list.size() == -1)
return visitSuite(suite_list[suite_list.size() - 1]);
return nullptr;
}