microsoft
/
clang
зеркало из https://github.com/microsoft/clang.git
1
0
Форкнуть 0
Код Задачи Пакеты Проекты Релизы Вики Активность

add some helper methods for removing and replacing text, this makes the 

rewriter more robust.


git-svn-id: https://llvm.org/svn/llvm-project/cfe/trunk@46622 91177308-0d34-0410-b5e6-96231b3b80d8
This commit is contained in:
Chris Lattner 2008-01-31 19:51:04 +00:00
Родитель f3dd57e537
Коммит aadaf78d65
3 изменённых файлов: 67 добавлений и 57 удалений
Показать статистику Скачать Patch файл Скачать Diff файл Показать все файлы Свернуть все файлы

89
Driver/RewriteTest.cpp
Просмотреть файл

@ -102,7 +102,7 @@ namespace {
} }
void InsertText(SourceLocation Loc, const char *StrData, unsigned StrLen) { void InsertText(SourceLocation Loc, const char *StrData, unsigned StrLen) {
// If replacement succeeded or warning disabled return with no warning. // If insertion succeeded or warning disabled return with no warning.
if (!Rewrite.InsertText(Loc, StrData, StrLen) || if (!Rewrite.InsertText(Loc, StrData, StrLen) ||
SilenceRewriteMacroWarning) SilenceRewriteMacroWarning)
return; return;
@ -110,6 +110,23 @@ namespace {
Diags.Report(Context->getFullLoc(Loc), RewriteFailedDiag); Diags.Report(Context->getFullLoc(Loc), RewriteFailedDiag);
} }
void RemoveText(SourceLocation Loc, unsigned StrLen) {
// If removal succeeded or warning disabled return with no warning.
if (!Rewrite.RemoveText(Loc, StrLen) || SilenceRewriteMacroWarning)
return;
Diags.Report(Context->getFullLoc(Loc), RewriteFailedDiag);
}
void ReplaceText(SourceLocation Start, unsigned OrigLength,
const char *NewStr, unsigned NewLength) {
// If removal succeeded or warning disabled return with no warning.
if (!Rewrite.ReplaceText(Start, OrigLength, NewStr, NewLength) ||
SilenceRewriteMacroWarning)
return;
Diags.Report(Context->getFullLoc(Start), RewriteFailedDiag);
}
// Syntactic Rewriting. // Syntactic Rewriting.
void RewritePrologue(SourceLocation Loc); void RewritePrologue(SourceLocation Loc);
@ -410,7 +427,7 @@ void RewriteTest::RewriteInclude() {
// replace import with include // replace import with include
SourceLocation ImportLoc = SourceLocation ImportLoc =
LocStart.getFileLocWithOffset(BufPtr-MainBufStart); LocStart.getFileLocWithOffset(BufPtr-MainBufStart);
Rewrite.ReplaceText(ImportLoc, ImportLen, "include", IncludeLen); ReplaceText(ImportLoc, ImportLen, "include", IncludeLen);
BufPtr += ImportLen; BufPtr += ImportLen;
} }
} }
@ -443,7 +460,7 @@ void RewriteTest::RewriteTabs() {
SourceLocation::getFileLoc(MainFileID, BufPtr-MainBufStart); SourceLocation::getFileLoc(MainFileID, BufPtr-MainBufStart);
// Rewrite the single tab character into a sequence of spaces. // Rewrite the single tab character into a sequence of spaces.
Rewrite.ReplaceText(TabLoc, 1, " ", Spaces); ReplaceText(TabLoc, 1, " ", Spaces);
} }
} }
@ -478,7 +495,7 @@ void RewriteTest::RewriteForwardClassDecl(ObjCClassDecl *ClassDecl) {
} }
// Replace the @class with typedefs corresponding to the classes. // Replace the @class with typedefs corresponding to the classes.
Rewrite.ReplaceText(startLoc, semiPtr-startBuf+1, ReplaceText(startLoc, semiPtr-startBuf+1,
typedefString.c_str(), typedefString.size()); typedefString.c_str(), typedefString.size());
} }
@ -488,7 +505,7 @@ void RewriteTest::RewriteMethodDeclaration(ObjCMethodDecl *Method) {
if (SM->getLineNumber(LocEnd) > SM->getLineNumber(LocStart)) { if (SM->getLineNumber(LocEnd) > SM->getLineNumber(LocStart)) {
InsertText(LocStart, "/* ", 3); InsertText(LocStart, "/* ", 3);
Rewrite.ReplaceText(LocEnd, 1, ";*/ ", 4); ReplaceText(LocEnd, 1, ";*/ ", 4);
} else { } else {
InsertText(LocStart, "// ", 3); InsertText(LocStart, "// ", 3);
} }
@ -500,7 +517,7 @@ void RewriteTest::RewriteProperties(int nProperties, ObjCPropertyDecl **Properti
ObjCPropertyDecl *Property = Properties[i]; ObjCPropertyDecl *Property = Properties[i];
SourceLocation Loc = Property->getLocation(); SourceLocation Loc = Property->getLocation();
Rewrite.ReplaceText(Loc, 0, "// ", 3); ReplaceText(Loc, 0, "// ", 3);
// FIXME: handle properties that are declared across multiple lines. // FIXME: handle properties that are declared across multiple lines.
} }
@ -510,7 +527,7 @@ void RewriteTest::RewriteCategoryDecl(ObjCCategoryDecl *CatDecl) {
SourceLocation LocStart = CatDecl->getLocStart(); SourceLocation LocStart = CatDecl->getLocStart();
// FIXME: handle category headers that are declared across multiple lines. // FIXME: handle category headers that are declared across multiple lines.
Rewrite.ReplaceText(LocStart, 0, "// ", 3); ReplaceText(LocStart, 0, "// ", 3);
for (ObjCCategoryDecl::instmeth_iterator I = CatDecl->instmeth_begin(), for (ObjCCategoryDecl::instmeth_iterator I = CatDecl->instmeth_begin(),
E = CatDecl->instmeth_end(); I != E; ++I) E = CatDecl->instmeth_end(); I != E; ++I)
@ -520,7 +537,7 @@ void RewriteTest::RewriteCategoryDecl(ObjCCategoryDecl *CatDecl) {
RewriteMethodDeclaration(*I); RewriteMethodDeclaration(*I);
// Lastly, comment out the @end. // Lastly, comment out the @end.
Rewrite.ReplaceText(CatDecl->getAtEndLoc(), 0, "// ", 3); ReplaceText(CatDecl->getAtEndLoc(), 0, "// ", 3);
} }
void RewriteTest::RewriteProtocolDecl(ObjCProtocolDecl *PDecl) { void RewriteTest::RewriteProtocolDecl(ObjCProtocolDecl *PDecl) {
@ -529,7 +546,7 @@ void RewriteTest::RewriteProtocolDecl(ObjCProtocolDecl *PDecl) {
SourceLocation LocStart = PDecl->getLocStart(); SourceLocation LocStart = PDecl->getLocStart();
// FIXME: handle protocol headers that are declared across multiple lines. // FIXME: handle protocol headers that are declared across multiple lines.
Rewrite.ReplaceText(LocStart, 0, "// ", 3); ReplaceText(LocStart, 0, "// ", 3);
for (ObjCProtocolDecl::instmeth_iterator I = PDecl->instmeth_begin(), for (ObjCProtocolDecl::instmeth_iterator I = PDecl->instmeth_begin(),
E = PDecl->instmeth_end(); I != E; ++I) E = PDecl->instmeth_end(); I != E; ++I)
@ -540,7 +557,7 @@ void RewriteTest::RewriteProtocolDecl(ObjCProtocolDecl *PDecl) {
// Lastly, comment out the @end. // Lastly, comment out the @end.
SourceLocation LocEnd = PDecl->getAtEndLoc(); SourceLocation LocEnd = PDecl->getAtEndLoc();
Rewrite.ReplaceText(LocEnd, 0, "// ", 3); ReplaceText(LocEnd, 0, "// ", 3);
// Must comment out @optional/@required // Must comment out @optional/@required
const char *startBuf = SM->getCharacterData(LocStart); const char *startBuf = SM->getCharacterData(LocStart);
@ -549,14 +566,14 @@ void RewriteTest::RewriteProtocolDecl(ObjCProtocolDecl *PDecl) {
if (*p == '@' && !strncmp(p+1, "optional", strlen("optional"))) { if (*p == '@' && !strncmp(p+1, "optional", strlen("optional"))) {
std::string CommentedOptional = "/* @optional */"; std::string CommentedOptional = "/* @optional */";
SourceLocation OptionalLoc = LocStart.getFileLocWithOffset(p-startBuf); SourceLocation OptionalLoc = LocStart.getFileLocWithOffset(p-startBuf);
Rewrite.ReplaceText(OptionalLoc, strlen("@optional"), ReplaceText(OptionalLoc, strlen("@optional"),
CommentedOptional.c_str(), CommentedOptional.size()); CommentedOptional.c_str(), CommentedOptional.size());
} }
else if (*p == '@' && !strncmp(p+1, "required", strlen("required"))) { else if (*p == '@' && !strncmp(p+1, "required", strlen("required"))) {
std::string CommentedRequired = "/* @required */"; std::string CommentedRequired = "/* @required */";
SourceLocation OptionalLoc = LocStart.getFileLocWithOffset(p-startBuf); SourceLocation OptionalLoc = LocStart.getFileLocWithOffset(p-startBuf);
Rewrite.ReplaceText(OptionalLoc, strlen("@required"), ReplaceText(OptionalLoc, strlen("@required"),
CommentedRequired.c_str(), CommentedRequired.size()); CommentedRequired.c_str(), CommentedRequired.size());
} }
@ -568,7 +585,7 @@ void RewriteTest::RewriteForwardProtocolDecl(ObjCForwardProtocolDecl *PDecl) {
if (LocStart.isInvalid()) if (LocStart.isInvalid())
assert(false && "Invalid SourceLocation"); assert(false && "Invalid SourceLocation");
// FIXME: handle forward protocol that are declared across multiple lines. // FIXME: handle forward protocol that are declared across multiple lines.
Rewrite.ReplaceText(LocStart, 0, "// ", 3); ReplaceText(LocStart, 0, "// ", 3);
} }
void RewriteTest::RewriteObjCMethodDecl(ObjCMethodDecl *OMD, void RewriteTest::RewriteObjCMethodDecl(ObjCMethodDecl *OMD,
@ -667,7 +684,7 @@ void RewriteTest::RewriteImplementationDecl(NamedDecl *OID) {
const char *startBuf = SM->getCharacterData(LocStart); const char *startBuf = SM->getCharacterData(LocStart);
const char *endBuf = SM->getCharacterData(LocEnd); const char *endBuf = SM->getCharacterData(LocEnd);
Rewrite.ReplaceText(LocStart, endBuf-startBuf, ReplaceText(LocStart, endBuf-startBuf,
ResultStr.c_str(), ResultStr.size()); ResultStr.c_str(), ResultStr.size());
} }
@ -682,7 +699,7 @@ void RewriteTest::RewriteImplementationDecl(NamedDecl *OID) {
const char *startBuf = SM->getCharacterData(LocStart); const char *startBuf = SM->getCharacterData(LocStart);
const char *endBuf = SM->getCharacterData(LocEnd); const char *endBuf = SM->getCharacterData(LocEnd);
Rewrite.ReplaceText(LocStart, endBuf-startBuf, ReplaceText(LocStart, endBuf-startBuf,
ResultStr.c_str(), ResultStr.size()); ResultStr.c_str(), ResultStr.size());
} }
if (IMD) if (IMD)
@ -722,7 +739,7 @@ void RewriteTest::RewriteInterfaceDecl(ObjCInterfaceDecl *ClassDecl) {
RewriteMethodDeclaration(*I); RewriteMethodDeclaration(*I);
// Lastly, comment out the @end. // Lastly, comment out the @end.
Rewrite.ReplaceText(ClassDecl->getAtEndLoc(), 0, "// ", 3); ReplaceText(ClassDecl->getAtEndLoc(), 0, "// ", 3);
} }
Stmt *RewriteTest::RewriteObjCIvarRefExpr(ObjCIvarRefExpr *IV) { Stmt *RewriteTest::RewriteObjCIvarRefExpr(ObjCIvarRefExpr *IV) {
@ -899,7 +916,7 @@ Stmt *RewriteTest::RewriteBreakStmt(BreakStmt *S) {
SourceLocation startLoc = S->getLocStart(); SourceLocation startLoc = S->getLocStart();
buf = "goto __break_label_"; buf = "goto __break_label_";
buf += utostr(ObjCBcLabelNo.back()); buf += utostr(ObjCBcLabelNo.back());
Rewrite.ReplaceText(startLoc, strlen("break"), buf.c_str(), buf.size()); ReplaceText(startLoc, strlen("break"), buf.c_str(), buf.size());
return 0; return 0;
} }
@ -916,7 +933,7 @@ Stmt *RewriteTest::RewriteContinueStmt(ContinueStmt *S) {
SourceLocation startLoc = S->getLocStart(); SourceLocation startLoc = S->getLocStart();
buf = "goto __continue_label_"; buf = "goto __continue_label_";
buf += utostr(ObjCBcLabelNo.back()); buf += utostr(ObjCBcLabelNo.back());
Rewrite.ReplaceText(startLoc, strlen("continue"), buf.c_str(), buf.size()); ReplaceText(startLoc, strlen("continue"), buf.c_str(), buf.size());
return 0; return 0;
} }
@ -1004,7 +1021,7 @@ Stmt *RewriteTest::RewriteObjCForCollectionStmt(ObjCForCollectionStmt *S,
startCollectionBuf += 3; startCollectionBuf += 3;
// Replace: "for (type element in" with string constructed thus far. // Replace: "for (type element in" with string constructed thus far.
Rewrite.ReplaceText(startLoc, startCollectionBuf - startBuf, ReplaceText(startLoc, startCollectionBuf - startBuf,
buf.c_str(), buf.size()); buf.c_str(), buf.size());
// Replace ')' in for '(' type elem in collection ')' with ';' // Replace ')' in for '(' type elem in collection ')' with ';'
SourceLocation rightParenLoc = S->getRParenLoc(); SourceLocation rightParenLoc = S->getRParenLoc();
@ -1046,7 +1063,7 @@ Stmt *RewriteTest::RewriteObjCForCollectionStmt(ObjCForCollectionStmt *S,
buf += elementTypeAsString; buf += elementTypeAsString;
buf += ")enumState.itemsPtr[counter++];"; buf += ")enumState.itemsPtr[counter++];";
// Replace ')' in for '(' type elem in collection ')' with all of these. // Replace ')' in for '(' type elem in collection ')' with all of these.
Rewrite.ReplaceText(lparenLoc, 1, buf.c_str(), buf.size()); ReplaceText(lparenLoc, 1, buf.c_str(), buf.size());
/// __continue_label: ; /// __continue_label: ;
/// } while (counter < limit); /// } while (counter < limit);
@ -1101,7 +1118,7 @@ Stmt *RewriteTest::RewriteObjCSynchronizedStmt(ObjCAtSynchronizedStmt *S) {
std::string buf; std::string buf;
buf = "objc_sync_enter"; buf = "objc_sync_enter";
Rewrite.ReplaceText(startLoc, 13, buf.c_str(), buf.size()); ReplaceText(startLoc, 13, buf.c_str(), buf.size());
SourceLocation endLoc = S->getSynchExpr()->getLocEnd(); SourceLocation endLoc = S->getSynchExpr()->getLocEnd();
const char *endBuf = SM->getCharacterData(endLoc); const char *endBuf = SM->getCharacterData(endLoc);
endBuf++; endBuf++;
@ -1115,7 +1132,7 @@ Stmt *RewriteTest::RewriteObjCSynchronizedStmt(ObjCAtSynchronizedStmt *S) {
buf += "id volatile _rethrow = 0;\n"; buf += "id volatile _rethrow = 0;\n";
buf += "objc_exception_try_enter(&_stack);\n"; buf += "objc_exception_try_enter(&_stack);\n";
buf += "if (!_setjmp(_stack.buf)) /* @try block continue */\n"; buf += "if (!_setjmp(_stack.buf)) /* @try block continue */\n";
Rewrite.ReplaceText(rparenLoc, 1, buf.c_str(), buf.size()); ReplaceText(rparenLoc, 1, buf.c_str(), buf.size());
startLoc = S->getSynchBody()->getLocEnd(); startLoc = S->getSynchBody()->getLocEnd();
startBuf = SM->getCharacterData(startLoc); startBuf = SM->getCharacterData(startLoc);
@ -1130,7 +1147,7 @@ Stmt *RewriteTest::RewriteObjCSynchronizedStmt(ObjCAtSynchronizedStmt *S) {
buf += "}\n"; buf += "}\n";
buf += "}"; buf += "}";
Rewrite.ReplaceText(lastCurlyLoc, 1, buf.c_str(), buf.size()); ReplaceText(lastCurlyLoc, 1, buf.c_str(), buf.size());
return 0; return 0;
} }
@ -1150,7 +1167,7 @@ Stmt *RewriteTest::RewriteObjCTryStmt(ObjCAtTryStmt *S) {
buf += "objc_exception_try_enter(&_stack);\n"; buf += "objc_exception_try_enter(&_stack);\n";
buf += "if (!_setjmp(_stack.buf)) /* @try block continue */\n"; buf += "if (!_setjmp(_stack.buf)) /* @try block continue */\n";
Rewrite.ReplaceText(startLoc, 4, buf.c_str(), buf.size()); ReplaceText(startLoc, 4, buf.c_str(), buf.size());
startLoc = S->getTryBody()->getLocEnd(); startLoc = S->getTryBody()->getLocEnd();
startBuf = SM->getCharacterData(startLoc); startBuf = SM->getCharacterData(startLoc);
@ -1190,8 +1207,7 @@ Stmt *RewriteTest::RewriteObjCTryStmt(ObjCAtTryStmt *S) {
QualType t = dyn_cast<ValueDecl>(declStmt->getDecl())->getType(); QualType t = dyn_cast<ValueDecl>(declStmt->getDecl())->getType();
if (t == Context->getObjCIdType()) { if (t == Context->getObjCIdType()) {
buf += "1) { "; buf += "1) { ";
Rewrite.ReplaceText(startLoc, lParenLoc-startBuf+1, ReplaceText(startLoc, lParenLoc-startBuf+1, buf.c_str(), buf.size());
buf.c_str(), buf.size());
sawIdTypedCatch = true; sawIdTypedCatch = true;
} else if (const PointerType *pType = t->getAsPointerType()) { } else if (const PointerType *pType = t->getAsPointerType()) {
ObjCInterfaceType *cls; // Should be a pointer to a class. ObjCInterfaceType *cls; // Should be a pointer to a class.
@ -1201,8 +1217,7 @@ Stmt *RewriteTest::RewriteObjCTryStmt(ObjCAtTryStmt *S) {
buf += "objc_exception_match((struct objc_class *)objc_getClass(\""; buf += "objc_exception_match((struct objc_class *)objc_getClass(\"";
buf += cls->getDecl()->getName(); buf += cls->getDecl()->getName();
buf += "\"), (struct objc_object *)_caught)) { "; buf += "\"), (struct objc_object *)_caught)) { ";
Rewrite.ReplaceText(startLoc, lParenLoc-startBuf+1, ReplaceText(startLoc, lParenLoc-startBuf+1, buf.c_str(), buf.size());
buf.c_str(), buf.size());
} }
} }
// Now rewrite the body... // Now rewrite the body...
@ -1217,8 +1232,7 @@ Stmt *RewriteTest::RewriteObjCTryStmt(ObjCAtTryStmt *S) {
buf = " = _caught;"; buf = " = _caught;";
// Here we replace ") {" with "= _caught;" (which initializes and // Here we replace ") {" with "= _caught;" (which initializes and
// declares the @catch parameter). // declares the @catch parameter).
Rewrite.ReplaceText(rParenLoc, bodyBuf-rParenBuf+1, ReplaceText(rParenLoc, bodyBuf-rParenBuf+1, buf.c_str(), buf.size());
buf.c_str(), buf.size());
} else if (!isa<NullStmt>(catchStmt)) { } else if (!isa<NullStmt>(catchStmt)) {
assert(false && "@catch rewrite bug"); assert(false && "@catch rewrite bug");
} }
@ -1243,7 +1257,7 @@ Stmt *RewriteTest::RewriteObjCTryStmt(ObjCAtTryStmt *S) {
assert((*startBuf == '@') && "bogus @finally start"); assert((*startBuf == '@') && "bogus @finally start");
buf = "/* @finally */"; buf = "/* @finally */";
Rewrite.ReplaceText(startLoc, 8, buf.c_str(), buf.size()); ReplaceText(startLoc, 8, buf.c_str(), buf.size());
Stmt *body = finalStmt->getFinallyBody(); Stmt *body = finalStmt->getFinallyBody();
SourceLocation startLoc = body->getLocStart(); SourceLocation startLoc = body->getLocStart();
@ -1294,12 +1308,12 @@ Stmt *RewriteTest::RewriteObjCThrowStmt(ObjCAtThrowStmt *S) {
buf = "objc_exception_throw("; buf = "objc_exception_throw(";
else // add an implicit argument else // add an implicit argument
buf = "objc_exception_throw(_caught"; buf = "objc_exception_throw(_caught";
Rewrite.ReplaceText(startLoc, 6, buf.c_str(), buf.size()); ReplaceText(startLoc, 6, buf.c_str(), buf.size());
const char *semiBuf = strchr(startBuf, ';'); const char *semiBuf = strchr(startBuf, ';');
assert((*semiBuf == ';') && "@throw: can't find ';'"); assert((*semiBuf == ';') && "@throw: can't find ';'");
SourceLocation semiLoc = startLoc.getFileLocWithOffset(semiBuf-startBuf); SourceLocation semiLoc = startLoc.getFileLocWithOffset(semiBuf-startBuf);
buf = ");"; buf = ");";
Rewrite.ReplaceText(semiLoc, 1, buf.c_str(), buf.size()); ReplaceText(semiLoc, 1, buf.c_str(), buf.size());
return 0; return 0;
} }
@ -2077,8 +2091,7 @@ void RewriteTest::SynthesizeObjCInternalStruct(ObjCInterfaceDecl *CDecl,
// have no ivars (thus not synthesized) then no need to synthesize this class. // have no ivars (thus not synthesized) then no need to synthesize this class.
if (NumIvars <= 0 && (!RCDecl || !ObjCSynthesizedStructs.count(RCDecl))) { if (NumIvars <= 0 && (!RCDecl || !ObjCSynthesizedStructs.count(RCDecl))) {
endBuf += Lexer::MeasureTokenLength(LocEnd, *SM); endBuf += Lexer::MeasureTokenLength(LocEnd, *SM);
Rewrite.ReplaceText(LocStart, endBuf-startBuf, ReplaceText(LocStart, endBuf-startBuf, Result.c_str(), Result.size());
Result.c_str(), Result.size());
return; return;
} }
@ -2093,8 +2106,7 @@ void RewriteTest::SynthesizeObjCInternalStruct(ObjCInterfaceDecl *CDecl,
&& "SynthesizeObjCInternalStruct - malformed @interface"); && "SynthesizeObjCInternalStruct - malformed @interface");
// rewrite the original header *without* disturbing the '{' // rewrite the original header *without* disturbing the '{'
Rewrite.ReplaceText(LocStart, cursor-startBuf-1, ReplaceText(LocStart, cursor-startBuf-1, Result.c_str(), Result.size());
Result.c_str(), Result.size());
if (RCDecl && ObjCSynthesizedStructs.count(RCDecl)) { if (RCDecl && ObjCSynthesizedStructs.count(RCDecl)) {
Result = "\n struct "; Result = "\n struct ";
Result += RCDecl->getName(); Result += RCDecl->getName();
@ -2152,8 +2164,7 @@ void RewriteTest::SynthesizeObjCInternalStruct(ObjCInterfaceDecl *CDecl,
// -fms-extensions. If the struct definition were "inlined", we wouldn't // -fms-extensions. If the struct definition were "inlined", we wouldn't
// need to use this switch. That said, I don't want to inline the def. // need to use this switch. That said, I don't want to inline the def.
Result += ";\n};\n"; Result += ";\n};\n";
Rewrite.ReplaceText(LocStart, endBuf-startBuf, ReplaceText(LocStart, endBuf-startBuf, Result.c_str(), Result.size());
Result.c_str(), Result.size());
} }
// Mark this struct as having been generated. // Mark this struct as having been generated.
if (!ObjCSynthesizedStructs.insert(CDecl)) if (!ObjCSynthesizedStructs.insert(CDecl))

16
Rewrite/Rewriter.cpp
Просмотреть файл

@ -204,8 +204,7 @@ RewriteBuffer &Rewriter::getEditBuffer(unsigned FileID) {
} }
/// InsertText - Insert the specified string at the specified location in the /// InsertText - Insert the specified string at the specified location in the
/// original buffer. This method is only valid on rewritable source /// original buffer.
/// locations.
bool Rewriter::InsertText(SourceLocation Loc, bool Rewriter::InsertText(SourceLocation Loc,
const char *StrData, unsigned StrLen) { const char *StrData, unsigned StrLen) {
if (!isRewritable(Loc)) return true; if (!isRewritable(Loc)) return true;
@ -215,26 +214,27 @@ bool Rewriter::InsertText(SourceLocation Loc,
return false; return false;
} }
/// RemoveText - Remove the specified text region. This method is only valid /// RemoveText - Remove the specified text region.
/// on a rewritable source location. bool Rewriter::RemoveText(SourceLocation Start, unsigned Length) {
void Rewriter::RemoveText(SourceLocation Start, unsigned Length) { if (!isRewritable(Start)) return true;
assert(isRewritable(Start) && "Not a rewritable location!");
unsigned FileID; unsigned FileID;
unsigned StartOffs = getLocationOffsetAndFileID(Start, FileID); unsigned StartOffs = getLocationOffsetAndFileID(Start, FileID);
getEditBuffer(FileID).RemoveText(StartOffs, Length); getEditBuffer(FileID).RemoveText(StartOffs, Length);
return false;
} }
/// ReplaceText - This method replaces a range of characters in the input /// ReplaceText - This method replaces a range of characters in the input
/// buffer with a new string. This is effectively a combined "remove/insert" /// buffer with a new string. This is effectively a combined "remove/insert"
/// operation. /// operation.
void Rewriter::ReplaceText(SourceLocation Start, unsigned OrigLength, bool Rewriter::ReplaceText(SourceLocation Start, unsigned OrigLength,
const char *NewStr, unsigned NewLength) { const char *NewStr, unsigned NewLength) {
assert(isRewritable(Start) && "Not a rewritable location!"); if (!isRewritable(Start)) return true;
unsigned StartFileID; unsigned StartFileID;
unsigned StartOffs = getLocationOffsetAndFileID(Start, StartFileID); unsigned StartOffs = getLocationOffsetAndFileID(Start, StartFileID);
getEditBuffer(StartFileID).ReplaceText(StartOffs, OrigLength, getEditBuffer(StartFileID).ReplaceText(StartOffs, OrigLength,
NewStr, NewLength); NewStr, NewLength);
return false;
} }
/// ReplaceStmt - This replaces a Stmt/Expr with another, using the pretty /// ReplaceStmt - This replaces a Stmt/Expr with another, using the pretty

7
include/clang/Rewrite/Rewriter.h
Просмотреть файл

@ -131,14 +131,13 @@ public:
/// location was not rewritable, false otherwise. /// location was not rewritable, false otherwise.
bool InsertText(SourceLocation Loc, const char *StrData, unsigned StrLen); bool InsertText(SourceLocation Loc, const char *StrData, unsigned StrLen);
/// RemoveText - Remove the specified text region. This method is only valid /// RemoveText - Remove the specified text region.
/// on a rewritable source location. bool RemoveText(SourceLocation Start, unsigned Length);
void RemoveText(SourceLocation Start, unsigned Length);
/// ReplaceText - This method replaces a range of characters in the input /// ReplaceText - This method replaces a range of characters in the input
/// buffer with a new string. This is effectively a combined "remove/insert" /// buffer with a new string. This is effectively a combined "remove/insert"
/// operation. /// operation.
void ReplaceText(SourceLocation Start, unsigned OrigLength, bool ReplaceText(SourceLocation Start, unsigned OrigLength,
const char *NewStr, unsigned NewLength); const char *NewStr, unsigned NewLength);
/// ReplaceStmt - This replaces a Stmt/Expr with another, using the pretty /// ReplaceStmt - This replaces a Stmt/Expr with another, using the pretty