commit 3d3b3603de2e6db14cb6255083ab8cab7c812d56
author Ted Kremenek <kremenek@apple.com> Fri Mar 13 22:53:41 2009 +0000
committer Ted Kremenek <kremenek@apple.com> Fri Mar 13 22:53:41 2009 +0000
tree 09cd2599390ad33eed6bac86dab9b3153f9c4622
parent d51d479473861e45891a2f990796cc2c5e09f587

Add (hidden) TableGen command option '-clang-component' which specifies the
component's warnings to process for '-gen-clang-diags-defs'.

Also, when the component is specified, generate a '#if' prologue at the top of
the generated .def file (to match the current files).


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

utils/TableGen/ClangDiagnosticsEmitter.cpp

diff --git a/utils/TableGen/ClangDiagnosticsEmitter.cpp b/utils/TableGen/ClangDiagnosticsEmitter.cpp
index fa139e7..f9501fb 100644
--- a/utils/TableGen/ClangDiagnosticsEmitter.cpp
+++ b/utils/TableGen/ClangDiagnosticsEmitter.cpp
@@ -22,8 +22,8 @@
 typedef std::vector<Record*> SuperClassVector;
 typedef std::vector<RecordVal> RecordValVector;
 
-static const RecordVal* findRecordVal(const RecordValVector& Vals,
-                                      const std::string &key) {
+static const RecordVal* findRecordVal(const Record& R, const std::string &key) {  
+  const RecordValVector &Vals = R.getValues();
   for (RecordValVector::const_iterator I=Vals.begin(), E=Vals.end(); I!=E; ++I)
     if ((*I).getName() == key)
       return &*I;
@@ -49,6 +49,11 @@
     }
 }
 
+static void EmitAllCaps(std::ostream& OS, const std::string &s) {
+  for (std::string::const_iterator I=s.begin(), E=s.end(); I!=E; ++I)
+    OS << char(toupper(*I));  
+}
+
 static void ProcessDiag(std::ostream& OS, const Record* DiagClass,
                         const Record& R) {
 
@@ -57,12 +62,9 @@
     return;
 
   OS << "DIAG(" << R.getName() << ", ";
+  EmitAllCaps(OS, DiagKind->getName());
   
-  const std::string &s = DiagKind->getName();
-  for (std::string::const_iterator I=s.begin(), E=s.end(); I!=E; ++I)
-    OS << char(toupper(*I));
-  
-  const RecordVal* Text = findRecordVal(R.getValues(), "Text");
+  const RecordVal* Text = findRecordVal(R, "Text");
   assert(Text && "No 'Text' entry in Diagnostic.");
   const StringInit* TextVal = dynamic_cast<const StringInit*>(Text->getValue());
   assert(TextVal && "Value 'Text' must be a string.");
@@ -77,8 +79,30 @@
   const Record* DiagClass = Records.getClass("Diagnostic");
   assert(DiagClass && "No Diagnostic class defined.");  
   
+  // Write the #if guard
+  if (!Component.empty()) {
+    OS << "#ifdef ";
+    EmitAllCaps(OS, Component);
+    OS << "START\n__";
+    EmitAllCaps(OS, Component);
+    OS << "START = DIAG_START_";
+    EmitAllCaps(OS, Component);
+    OS << ",\n#undef ";
+    EmitAllCaps(OS, Component);
+    OS << "START\n#endif\n";
+  }
+  
   for (RecordVector::const_iterator I=Diags.begin(), E=Diags.end(); I!=E; ++I) {
-    // FIXME: Compare the component.
+    if (!Component.empty()) {
+      const RecordVal* V = findRecordVal(**I, "Component");
+      if (!V)
+        continue;
+
+      const StringInit* SV = dynamic_cast<const StringInit*>(V->getValue());
+      if (SV->getValue() != Component)
+        continue;
+    }
+    
     ProcessDiag(OS, DiagClass, **I);
   }
-}
\ No newline at end of file
+}