//
// Created by erki on 02.03.21.
//

#ifndef C_ANALYZER_OPERATIONFINDERASTCONSUMER_HPP
#define C_ANALYZER_OPERATIONFINDERASTCONSUMER_HPP

#include <clang/AST/ASTConsumer.h>

#include "OperationFinderAstVisitor.hpp"

class OperationFinder;

class OperationFinderAstConsumer : public clang::ASTConsumer
{
public:
  explicit OperationFinderAstConsumer(OperationFinder* op_finder);

  void Initialize(clang::ASTContext& context) override;

  void HandleTranslationUnit(clang::ASTContext& context) override;

private:
  OperationFinderAstVisitor _visitor;
};

#endif //C_ANALYZER_OPERATIONFINDERASTCONSUMER_HPP