2011-06-22 04:20:39 +08:00
|
|
|
//===-- Transforms.h - Tranformations to ARC mode ---------------*- C++ -*-===//
|
|
|
|
//
|
|
|
|
// The LLVM Compiler Infrastructure
|
|
|
|
//
|
|
|
|
// This file is distributed under the University of Illinois Open Source
|
|
|
|
// License. See LICENSE.TXT for details.
|
|
|
|
//
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
|
|
|
|
#ifndef LLVM_CLANG_LIB_ARCMIGRATE_TRANSFORMS_H
|
|
|
|
#define LLVM_CLANG_LIB_ARCMIGRATE_TRANSFORMS_H
|
|
|
|
|
|
|
|
#include "clang/AST/RecursiveASTVisitor.h"
|
2011-11-04 23:58:08 +08:00
|
|
|
#include "clang/AST/ParentMap.h"
|
2011-06-22 04:20:39 +08:00
|
|
|
#include "llvm/ADT/DenseSet.h"
|
|
|
|
|
|
|
|
namespace clang {
|
|
|
|
class Decl;
|
|
|
|
class Stmt;
|
|
|
|
class BlockDecl;
|
|
|
|
class ObjCMethodDecl;
|
|
|
|
class FunctionDecl;
|
|
|
|
|
|
|
|
namespace arcmt {
|
|
|
|
class MigrationPass;
|
|
|
|
|
|
|
|
namespace trans {
|
|
|
|
|
2011-11-04 23:58:08 +08:00
|
|
|
class MigrationContext;
|
|
|
|
|
2011-06-22 04:20:39 +08:00
|
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
// Transformations.
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
|
|
|
|
void rewriteAutoreleasePool(MigrationPass &pass);
|
|
|
|
void rewriteUnbridgedCasts(MigrationPass &pass);
|
|
|
|
void makeAssignARCSafe(MigrationPass &pass);
|
2011-11-04 23:58:22 +08:00
|
|
|
void removeRetainReleaseDeallocFinalize(MigrationPass &pass);
|
|
|
|
void removeZeroOutPropsInDeallocFinalize(MigrationPass &pass);
|
2011-06-22 04:20:39 +08:00
|
|
|
void rewriteBlockObjCVariable(MigrationPass &pass);
|
|
|
|
void rewriteUnusedInitDelegate(MigrationPass &pass);
|
2011-07-18 15:44:45 +08:00
|
|
|
void checkAPIUses(MigrationPass &pass);
|
2011-06-22 04:20:39 +08:00
|
|
|
|
2011-11-04 23:58:22 +08:00
|
|
|
void removeEmptyStatementsAndDeallocFinalize(MigrationPass &pass);
|
2011-06-22 04:20:42 +08:00
|
|
|
|
2011-11-04 23:58:08 +08:00
|
|
|
class BodyContext {
|
|
|
|
MigrationContext &MigrateCtx;
|
|
|
|
ParentMap PMap;
|
|
|
|
Stmt *TopStmt;
|
|
|
|
|
|
|
|
public:
|
|
|
|
BodyContext(MigrationContext &MigrateCtx, Stmt *S)
|
|
|
|
: MigrateCtx(MigrateCtx), PMap(S), TopStmt(S) {}
|
|
|
|
|
|
|
|
MigrationContext &getMigrationContext() { return MigrateCtx; }
|
|
|
|
ParentMap &getParentMap() { return PMap; }
|
|
|
|
Stmt *getTopStmt() { return TopStmt; }
|
|
|
|
};
|
|
|
|
|
2011-11-07 02:57:57 +08:00
|
|
|
class ObjCImplementationContext {
|
|
|
|
MigrationContext &MigrateCtx;
|
|
|
|
ObjCImplementationDecl *ImpD;
|
|
|
|
|
|
|
|
public:
|
|
|
|
ObjCImplementationContext(MigrationContext &MigrateCtx,
|
|
|
|
ObjCImplementationDecl *D)
|
|
|
|
: MigrateCtx(MigrateCtx), ImpD(D) {}
|
|
|
|
|
|
|
|
MigrationContext &getMigrationContext() { return MigrateCtx; }
|
|
|
|
ObjCImplementationDecl *getImplementationDecl() { return ImpD; }
|
|
|
|
};
|
|
|
|
|
2011-11-04 23:58:08 +08:00
|
|
|
class ASTTraverser {
|
|
|
|
public:
|
|
|
|
virtual ~ASTTraverser();
|
2011-11-07 02:57:57 +08:00
|
|
|
virtual void traverseTU(MigrationContext &MigrateCtx) { }
|
2011-11-04 23:58:08 +08:00
|
|
|
virtual void traverseBody(BodyContext &BodyCtx) { }
|
2011-11-07 02:57:57 +08:00
|
|
|
virtual void traverseObjCImplementation(ObjCImplementationContext &ImplCtx) {}
|
2011-11-04 23:58:08 +08:00
|
|
|
};
|
|
|
|
|
|
|
|
class MigrationContext {
|
|
|
|
MigrationPass &Pass;
|
|
|
|
std::vector<ASTTraverser *> Traversers;
|
|
|
|
|
|
|
|
public:
|
|
|
|
explicit MigrationContext(MigrationPass &pass) : Pass(pass) {}
|
|
|
|
~MigrationContext();
|
|
|
|
|
|
|
|
MigrationPass &getPass() { return Pass; }
|
|
|
|
|
|
|
|
typedef std::vector<ASTTraverser *>::iterator traverser_iterator;
|
|
|
|
traverser_iterator traversers_begin() { return Traversers.begin(); }
|
|
|
|
traverser_iterator traversers_end() { return Traversers.end(); }
|
|
|
|
|
|
|
|
void addTraverser(ASTTraverser *traverser) {
|
|
|
|
Traversers.push_back(traverser);
|
|
|
|
}
|
|
|
|
|
2011-11-05 07:43:03 +08:00
|
|
|
bool isGCOwnedNonObjC(QualType T);
|
|
|
|
|
2011-11-04 23:58:08 +08:00
|
|
|
void traverse(TranslationUnitDecl *TU);
|
|
|
|
};
|
|
|
|
|
2011-11-07 02:57:57 +08:00
|
|
|
class PropertyRewriteTraverser : public ASTTraverser {
|
|
|
|
public:
|
|
|
|
virtual void traverseObjCImplementation(ObjCImplementationContext &ImplCtx);
|
|
|
|
};
|
|
|
|
|
2011-11-04 23:58:08 +08:00
|
|
|
// GC transformations
|
|
|
|
|
|
|
|
class GCCollectableCallsTraverser : public ASTTraverser {
|
|
|
|
public:
|
|
|
|
virtual void traverseBody(BodyContext &BodyCtx);
|
|
|
|
};
|
|
|
|
|
2011-06-22 04:20:39 +08:00
|
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
// Helpers.
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
|
2011-07-13 06:05:17 +08:00
|
|
|
/// \brief Determine whether we can add weak to the given type.
|
|
|
|
bool canApplyWeak(ASTContext &Ctx, QualType type);
|
|
|
|
|
2011-06-22 04:20:39 +08:00
|
|
|
/// \brief 'Loc' is the end of a statement range. This returns the location
|
|
|
|
/// immediately after the semicolon following the statement.
|
|
|
|
/// If no semicolon is found or the location is inside a macro, the returned
|
|
|
|
/// source location will be invalid.
|
|
|
|
SourceLocation findLocationAfterSemi(SourceLocation loc, ASTContext &Ctx);
|
|
|
|
|
2011-09-02 04:53:18 +08:00
|
|
|
/// \brief \arg Loc is the end of a statement range. This returns the location
|
|
|
|
/// of the semicolon following the statement.
|
|
|
|
/// If no semicolon is found or the location is inside a macro, the returned
|
|
|
|
/// source location will be invalid.
|
|
|
|
SourceLocation findSemiAfterLocation(SourceLocation loc, ASTContext &Ctx);
|
|
|
|
|
2011-06-22 04:20:39 +08:00
|
|
|
bool hasSideEffects(Expr *E, ASTContext &Ctx);
|
2011-07-15 07:32:04 +08:00
|
|
|
bool isGlobalVar(Expr *E);
|
2011-07-27 13:28:18 +08:00
|
|
|
/// \brief Returns "nil" or "0" if 'nil' macro is not actually defined.
|
|
|
|
StringRef getNilString(ASTContext &Ctx);
|
2011-06-22 04:20:39 +08:00
|
|
|
|
|
|
|
template <typename BODY_TRANS>
|
|
|
|
class BodyTransform : public RecursiveASTVisitor<BodyTransform<BODY_TRANS> > {
|
|
|
|
MigrationPass &Pass;
|
|
|
|
|
|
|
|
public:
|
|
|
|
BodyTransform(MigrationPass &pass) : Pass(pass) { }
|
|
|
|
|
2011-06-24 05:21:33 +08:00
|
|
|
bool TraverseStmt(Stmt *rootS) {
|
2011-07-18 15:44:45 +08:00
|
|
|
if (rootS)
|
|
|
|
BODY_TRANS(Pass).transformBody(rootS);
|
2011-06-22 04:20:39 +08:00
|
|
|
return true;
|
|
|
|
}
|
|
|
|
};
|
|
|
|
|
|
|
|
typedef llvm::DenseSet<Expr *> ExprSet;
|
|
|
|
|
|
|
|
void clearRefsIn(Stmt *S, ExprSet &refs);
|
|
|
|
template <typename iterator>
|
|
|
|
void clearRefsIn(iterator begin, iterator end, ExprSet &refs) {
|
|
|
|
for (; begin != end; ++begin)
|
|
|
|
clearRefsIn(*begin, refs);
|
|
|
|
}
|
|
|
|
|
|
|
|
void collectRefs(ValueDecl *D, Stmt *S, ExprSet &refs);
|
|
|
|
|
|
|
|
void collectRemovables(Stmt *S, ExprSet &exprs);
|
|
|
|
|
|
|
|
} // end namespace trans
|
|
|
|
|
|
|
|
} // end namespace arcmt
|
|
|
|
|
|
|
|
} // end namespace clang
|
|
|
|
|
|
|
|
#endif
|