Add a 'kind' field to PathDiagnosticPieces.

llvm-svn: 65862

1 files changed, 7 insertions, 3 deletions

diff --git a/clang/include/clang/Analysis/PathDiagnostic.h b/clang/include/clang/Analysis/PathDiagnostic.h
index a77ff66001a..bdfa88b4ad4 100644
--- a/clang/include/clang/Analysis/PathDiagnostic.h
+++ b/clang/include/clang/Analysis/PathDiagnostic.h
@@ -26,30 +26,34 @@ namespace clang {
 
 class PathDiagnosticPiece {
 public:
+  enum Kind { ControlFlow, Event };
   enum DisplayHint { Above, Below };
 
 private:
   const FullSourceLoc Pos;
   const std::string str;
   std::vector<CodeModificationHint> CodeModificationHints;
+  const Kind kind;
   const DisplayHint Hint;
   std::vector<SourceRange> ranges;
   
 public:
   
   PathDiagnosticPiece(FullSourceLoc pos, const std::string& s,
+                      Kind k = Event,
                       DisplayHint hint = Above);
   
   PathDiagnosticPiece(FullSourceLoc pos, const char* s,
+                      Kind k = Event,
                       DisplayHint hint = Above);
   
   const std::string& getString() const { return str; }
    
   DisplayHint getDisplayHint() const { return Hint; }
   
-  void addRange(SourceRange R) {
-    ranges.push_back(R);
-  }
+  Kind getKind() const { return kind; }
+  
+  void addRange(SourceRange R) { ranges.push_back(R); }
   
   void addRange(SourceLocation B, SourceLocation E) {
     ranges.push_back(SourceRange(B,E));