be less verbose on debug messages

rizsotto · Oct 10, 2012 · 0d20073 · 0d20073
1 parent a5e8555
commit 0d20073
Showing 1 changed file with 13 additions and 1 deletion.
diff --git a/sources/ScopeAnalysis.cpp b/sources/ScopeAnalysis.cpp
@@ -5,6 +5,7 @@
 
 #include <boost/bind.hpp>
 #include <boost/range.hpp>
+#include <boost/range/adaptor/filtered.hpp>
 #include <boost/range/algorithm/for_each.hpp>
 
 #include <clang/AST/RecursiveASTVisitor.h>
@@ -127,8 +128,19 @@ class UsageRefCollector {
         }
     }
 
+    // helper method not to be so verbose.
+    struct IsItFromMainModule {
+        bool operator()(clang::Decl const * const D) const {
+            clang::SourceManager const & SM = D->getASTContext().getSourceManager();
+            return SM.isFromMainFile(D->getLocation());
+        }
+        bool operator()(ScopeAnalysis::UsageRefsMap::value_type const & Var) const {
+            return this->operator()(Var.first);
+        }
+    };
+
     void Report(char const * const M, clang::DiagnosticsEngine & DE) const {
-        boost::for_each(Results,
+        boost::for_each(Results | boost::adaptors::filtered(IsItFromMainModule()),
             boost::bind(UsageRefCollector::Report, _1, M, boost::ref(DE)));
     }