public void visitStatement(StatementContext statement) { if (statement.expression() != null) { new ExpressionVisitor(symbolTable, method).visitExpression(statement.expression()); } else if (statement.assignment() != null) { new AssignmentVisitor(symbolTable, method).visitAssignment(statement.assignment()); } else if (statement.block() != null) { visitBlock(statement.block()); } else if (statement.jumpStatement() != null) { visitJumpStatement(statement.jumpStatement()); } else if (statement.IF() != null) { new IfStatementVisitor(this, symbolTable, method).visitIfStatement(statement); } else if (statement.forStatement() != null) { new CycleStatementVisitor(this, symbolTable, method) .visitForStatement(statement.forStatement()); } else if (statement.WHILE() != null) { new CycleStatementVisitor(this, symbolTable, method).visitWhileStatement(statement); } }
private void checkReturn(BlockContext block) { if (returnType == ValueType.VOID) { method.visitInsn(Opcodes.RETURN); return; } if (block.blockStatement() != null) { int lastStatement = block.blockStatement().size() - 1; BlockStatementContext blockStatement = block.blockStatement(lastStatement); if (blockStatement.statement() != null) { StatementContext statement = blockStatement.statement(); if (statement.jumpStatement() != null) { JumpStatementContext jumpStatement = statement.jumpStatement(); if (jumpStatement.returnSt != null) { return; } } } } throw new GenerationException( "Last statement of function with non-null return type must be return. String: " + block.getText()); }