ASTNode (@Nullable ASTNode statement) { if (!is(statement, BASIC_FOR_STATEMENT)) { return null; } ASTNode semicolon = findChildByType(statement, JavaTokenType.SEMICOLON); if (semicolon == null) { return null; } for (ASTNode child = semicolon; child != null; child = child.getTreeNext()) { if (is(child, STATEMENT_SET)) { return child; } if (is(child, RPARENTH)) break; } return null; }