break down EmitCaretDiagnostic to use more primitive calls.

git-svn-id: https://llvm.org/svn/llvm-project/cfe/trunk@64754 91177308-0d34-0410-b5e6-96231b3b80d8

diff --git a/include/clang/Driver/TextDiagnosticPrinter.h b/include/clang/Driver/TextDiagnosticPrinter.h
index 49f456ff17210edec23496349e1a98c9ea90387b..19d1b760303fdd3942313bca4598c481c43fd915 100644 (file)
--- a/include/clang/Driver/TextDiagnosticPrinter.h
+++ b/include/clang/Driver/TextDiagnosticPrinter.h
@@ -46,7 +46,8 @@ public:
                       std::string &CaretLine,
                       const std::string &SourceLine);
 
-  void EmitCaretDiagnostic(FullSourceLoc Loc, const DiagnosticInfo &Info);
+  void EmitCaretDiagnostic(const DiagnosticInfo &Info,
+                           SourceLocation Loc, SourceManager &SM);
   
   virtual void HandleDiagnostic(Diagnostic::Level DiagLevel,
                                 const DiagnosticInfo &Info);

diff --git a/lib/Driver/TextDiagnosticPrinter.cpp b/lib/Driver/TextDiagnosticPrinter.cpp
index fae39ef5fe412b92cd26ce455a4f9c5c2a7c5d40..64214354264e65f35ec3249379cb87530897d912 100644 (file)
--- a/lib/Driver/TextDiagnosticPrinter.cpp
+++ b/lib/Driver/TextDiagnosticPrinter.cpp
@@ -101,18 +101,31 @@ void TextDiagnosticPrinter::HighlightRange(const SourceRange &R,
     CaretLine[i] = '~';
 }
 
-void TextDiagnosticPrinter::EmitCaretDiagnostic(FullSourceLoc ILoc,
-                                                const DiagnosticInfo &Info) {
-  unsigned ColNo = ILoc.getInstantiationColumnNumber();
+void TextDiagnosticPrinter::EmitCaretDiagnostic(const DiagnosticInfo &Info,
+                                                SourceLocation Loc,
+                                                SourceManager &SM) {
+  assert(Loc.isFileID() && "Shouldn't have instantiation locs here");
+  
+  // Decompose the location into a FID/Offset pair.
+  std::pair<FileID, unsigned> LocInfo = SM.getDecomposedLoc(Loc);
+  FileID FID = LocInfo.first;
+  unsigned FileOffset = LocInfo.second;
+  
+  // Get information about the buffer it points into.
+  std::pair<const char*, const char*> BufferInfo = SM.getBufferData(FID);
+  const char *BufStart = BufferInfo.first;
+  const char *BufEnd = BufferInfo.second;
+
+  unsigned ColNo = SM.getColumnNumber(FID, FileOffset);
   
   // Rewind from the current position to the start of the line.
-  const char *TokInstantiationPtr = ILoc.getCharacterData();
-  const char *LineStart = TokInstantiationPtr-ColNo+1; // Column # is 1-based.
+  const char *TokPtr = BufStart+FileOffset;
+  const char *LineStart = TokPtr-ColNo+1; // Column # is 1-based.
+  
   
   // Compute the line end.  Scan forward from the error position to the end of
   // the line.
-  const char *BufEnd = ILoc.getBufferData().second;
-  const char *LineEnd = TokInstantiationPtr;
+  const char *LineEnd = TokPtr;
   while (LineEnd != BufEnd && 
          *LineEnd != '\n' && *LineEnd != '\r')
     ++LineEnd;
@@ -125,10 +138,12 @@ void TextDiagnosticPrinter::EmitCaretDiagnostic(FullSourceLoc ILoc,
   std::string CaretLine(LineEnd-LineStart, ' ');
   
   // Highlight all of the characters covered by Ranges with ~ characters.
-  for (unsigned i = 0; i != Info.getNumRanges(); ++i)
-    HighlightRange(Info.getRange(i), ILoc.getManager(),
-                   ILoc.getInstantiationLineNumber(),
-                   ILoc.getFileID(), CaretLine, SourceLine);
+  if (Info.getNumRanges()) {
+    unsigned LineNo = SM.getLineNumber(FID, FileOffset);
+    
+    for (unsigned i = 0; i != Info.getNumRanges(); ++i)
+      HighlightRange(Info.getRange(i), SM, LineNo, FID, CaretLine, SourceLine);
+  }
   
   // Next, insert the caret itself.
   if (ColNo-1 < CaretLine.size())
@@ -216,8 +231,7 @@ void TextDiagnosticPrinter::HandleDiagnostic(Diagnostic::Level Level,
     // Inspect the actual instantiation point of the diagnostic, we don't care
     // about presumed locations anymore.
     FullSourceLoc ILoc = Info.getLocation().getInstantiationLoc();
-    
-    EmitCaretDiagnostic(ILoc, Info);
+    EmitCaretDiagnostic(Info, ILoc, ILoc.getManager());
   }
   
   OS.flush();