diff options
| -rw-r--r-- | include/clang/Lex/Lexer.h | 7 | ||||
| -rw-r--r-- | lib/Rewrite/HTMLRewrite.cpp | 2 |
2 files changed, 8 insertions, 1 deletions
diff --git a/include/clang/Lex/Lexer.h b/include/clang/Lex/Lexer.h index 2e8267d794..0ba51a6037 100644 --- a/include/clang/Lex/Lexer.h +++ b/include/clang/Lex/Lexer.h @@ -154,6 +154,13 @@ public: return BufferPtr == BufferEnd; } + /// SetCommentRetentionMode - Change the comment retention mode of the lexer + /// to the specified mode. This is really only useful when lexing in raw + /// mode, because otherwise the lexer needs to manage this. + void SetCommentRetentionState(bool Mode) { + KeepCommentMode = Mode; + } + /// ReadToEndOfLine - Read the rest of the current preprocessor line as an /// uninterpreted string. This switches the lexer out of directive mode. std::string ReadToEndOfLine(); diff --git a/lib/Rewrite/HTMLRewrite.cpp b/lib/Rewrite/HTMLRewrite.cpp index 682fffd82a..fc50eb1151 100644 --- a/lib/Rewrite/HTMLRewrite.cpp +++ b/lib/Rewrite/HTMLRewrite.cpp @@ -237,7 +237,7 @@ void html::SyntaxHighlight(Rewriter &R, unsigned FileID, Preprocessor &PP) { // Inform the preprocessor that we want to retain comments as tokens, so we // can highlight them. - //PP.SetCommentRetentionState(true, false); + L.SetCommentRetentionState(true); // Lex all the tokens in raw mode, to avoid entering #includes or expanding // macros. |