tools/clang/blink_gc_plugin/CollectVisitor.h

   1 // Copyright 2015 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #ifndef TOOLS_BLINK_GC_PLUGIN_COLLECT_VISITOR_H_
   6 #define TOOLS_BLINK_GC_PLUGIN_COLLECT_VISITOR_H_
   7
   8 #include <vector>
   9
  10 #include "clang/AST/AST.h"
  11 #include "clang/AST/RecursiveASTVisitor.h"
  12
  13 // This visitor collects the entry points for the checker.
  14 class CollectVisitor : public clang::RecursiveASTVisitor<CollectVisitor> {
  15  public:
  16   typedef std::vector<clang::CXXRecordDecl*> RecordVector;
  17   typedef std::vector<clang::CXXMethodDecl*> MethodVector;
  18
  19   CollectVisitor();
  20
  21   RecordVector& record_decls();
  22   MethodVector& trace_decls();
  23
  24   // Collect record declarations, including nested declarations.
  25   bool VisitCXXRecordDecl(clang::CXXRecordDecl* record);
  26
  27   // Collect tracing method definitions, but don't traverse method bodies.
  28   bool VisitCXXMethodDecl(clang::CXXMethodDecl* method);
  29
  30  private:
  31   RecordVector record_decls_;
  32   MethodVector trace_decls_;
  33 };
  34
  35 #endif  // TOOLS_BLINK_GC_PLUGIN_COLLECT_VISITOR_H_