1 //===-- ModelConsumer.h -----------------------------------------*- C++ -*-===// 2 // 3 // The LLVM Compiler Infrastructure 4 // 5 // This file is distributed under the University of Illinois Open Source 6 // License. See LICENSE.TXT for details. 7 // 8 //===----------------------------------------------------------------------===// 9 /// 10 /// \file 11 /// \brief This file implements clang::ento::ModelConsumer which is an 12 /// ASTConsumer for model files. 13 /// 14 //===----------------------------------------------------------------------===// 15 16 #ifndef LLVM_CLANG_GR_MODELCONSUMER_H 17 #define LLVM_CLANG_GR_MODELCONSUMER_H 18 19 #include "clang/AST/ASTConsumer.h" 20 #include "llvm/ADT/StringMap.h" 21 22 namespace clang { 23 24 class Stmt; 25 26 namespace ento { 27 28 /// \brief ASTConsumer to consume model files' AST. 29 /// 30 /// This consumer collects the bodies of function definitions into a StringMap 31 /// from a model file. 32 class ModelConsumer : public ASTConsumer { 33 public: 34 ModelConsumer(llvm::StringMap<Stmt *> &Bodies); 35 36 bool HandleTopLevelDecl(DeclGroupRef D) override; 37 38 private: 39 llvm::StringMap<Stmt *> &Bodies; 40 }; 41 } 42 } 43 44 #endif 45
