VuoReplaceGenericTypesWithSpecializedAction.hh

Go to the documentation of this file.

 
#pragma once
 
#include "clang/AST/ASTConsumer.h"
#include "clang/AST/RecursiveASTVisitor.h"
#include "clang/Rewrite/Core/Rewriter.h"
class VuoPreprocessorCallbacks;
 
class VuoReplaceGenericTypesWithSpecializedAction : public clang::ASTFrontendAction
{
public:
    VuoReplaceGenericTypesWithSpecializedAction(raw_ostream &output, const map<string, string> &replacements, bool shouldReplaceGenericTypesInVuoTypeAnnotations, json_object *specializedModuleDetails, unique_ptr<VuoPreprocessorCallbacks> preprocessorCallbacks);
    unique_ptr<clang::ASTConsumer> CreateASTConsumer(clang::CompilerInstance &compilerInstance, llvm::StringRef inFile);
    bool BeginSourceFileAction(clang::CompilerInstance &CI);
    void EndSourceFileAction();
 
private:
    raw_ostream &output;
    map<string, string> replacements;
    bool shouldReplaceGenericTypesInVuoTypeAnnotations;
    json_object *specializedModuleDetails;
    unique_ptr<VuoPreprocessorCallbacks> preprocessorCallbacks;
    clang::Rewriter rewriter;
 
    class Visitor : public clang::RecursiveASTVisitor<Visitor>
    {
    public:
        Visitor(const map<string, string> &replacements, bool shouldReplaceGenericTypesInVuoTypeAnnotations, json_object *specializedModuleDetails, clang::Rewriter &rewriter);
        bool VisitFunctionDecl(clang::FunctionDecl *decl);
        bool VisitDeclRefExpr(clang::DeclRefExpr *expr);
        bool VisitAnnotateAttr(clang::AnnotateAttr *attr);
        bool VisitVarDecl(clang::VarDecl *decl);
 
    private:
        bool isInModuleSourceFile(clang::SourceRange sourceRange);
        void replaceGenericTypeNamesInTypeSpecifier(clang::QualType qualType, clang::SourceRange sourceRange);
        bool replaceGenericTypeNamesInIdentifier(string &identifier);
        vector<string> orderedGenericTypeNames;
        const map<string, string> replacements;
        bool shouldReplaceGenericTypesInVuoTypeAnnotations;
        json_object *specializedModuleDetails;
        clang::Rewriter &rewriter;
    };
 
    class Consumer : public clang::ASTConsumer
    {
    public:
        Consumer(const map<string, string> &replacements, bool shouldReplaceGenericTypesInVuoTypeAnnotations, json_object *specializedModuleDetails, clang::Rewriter &rewriter);
        void HandleTranslationUnit(clang::ASTContext &context);
 
    private:
        Visitor visitor;
    };
};