1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71
|
//===--- StmtUtils.cpp - Statement helper functions -----------------------===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
#include "StmtUtils.h"
#include "clang/AST/DeclObjC.h"
#include "clang/AST/Stmt.h"
#include "clang/AST/StmtCXX.h"
#include "clang/AST/StmtObjC.h"
#include "clang/Lex/Lexer.h"
using namespace clang;
SourceLocation
clang::tooling::getLexicalEndLocForDecl(const Decl *D, const SourceManager &SM,
const LangOptions &LangOpts) {
if (!isa<ObjCImplDecl>(D))
return D->getSourceRange().getEnd();
auto AtEnd = D->getSourceRange().getEnd();
auto AdjustedEnd =
Lexer::findNextTokenLocationAfterTokenAt(AtEnd, SM, LangOpts);
return AdjustedEnd.isValid() ? AdjustedEnd : AtEnd;
}
bool clang::tooling::isSemicolonRequiredAfter(const Stmt *S) {
if (isa<CompoundStmt>(S))
return false;
if (const auto *If = dyn_cast<IfStmt>(S))
return isSemicolonRequiredAfter(If->getElse() ? If->getElse()
: If->getThen());
if (const auto *While = dyn_cast<WhileStmt>(S))
return isSemicolonRequiredAfter(While->getBody());
if (const auto *For = dyn_cast<ForStmt>(S))
return isSemicolonRequiredAfter(For->getBody());
if (const auto *CXXFor = dyn_cast<CXXForRangeStmt>(S))
return isSemicolonRequiredAfter(CXXFor->getBody());
if (const auto *ObjCFor = dyn_cast<ObjCForCollectionStmt>(S))
return isSemicolonRequiredAfter(ObjCFor->getBody());
switch (S->getStmtClass()) {
case Stmt::SwitchStmtClass:
case Stmt::CXXTryStmtClass:
case Stmt::ObjCAtSynchronizedStmtClass:
case Stmt::ObjCAutoreleasePoolStmtClass:
case Stmt::ObjCAtTryStmtClass:
return false;
default:
return true;
}
}
static bool isAssignmentOperator(const Stmt *S) {
if (const auto *PseudoExpr = dyn_cast<PseudoObjectExpr>(S))
return isAssignmentOperator(PseudoExpr->getSyntacticForm());
if (const auto *BO = dyn_cast<BinaryOperator>(S))
return BO->isAssignmentOp();
return false;
}
bool clang::tooling::isLexicalExpression(const Stmt *S, const Stmt *Parent) {
if (!isa<Expr>(S))
return false;
// Assignment operators should be treated as statements unless they are a part
// of an expression.
if (isAssignmentOperator(S) && (!Parent || !isa<Expr>(Parent)))
return false;
return !cast<Expr>(S)->getType()->isVoidType();
}
|