VuoReplaceGenericTypesWithSpecializedAction.cc

Go to the documentation of this file.

 
#include "VuoReplaceGenericTypesWithSpecializedAction.hh"
#include "VuoPreprocessorCallbacks.hh"
#include "VuoGenericType.hh"
#include "VuoStringUtilities.hh"
 
using namespace clang;
 
VuoReplaceGenericTypesWithSpecializedAction::VuoReplaceGenericTypesWithSpecializedAction(raw_ostream &output,
                                                                                         const map<string, string> &replacements,
                                                                                         bool shouldReplaceGenericTypesInVuoTypeAnnotations,
                                                                                         json_object *specializedModuleDetails,
                                                                                         unique_ptr<VuoPreprocessorCallbacks> preprocessorCallbacks) :
    ASTFrontendAction(),
    output(output),
    replacements(replacements),
    shouldReplaceGenericTypesInVuoTypeAnnotations(shouldReplaceGenericTypesInVuoTypeAnnotations),
    specializedModuleDetails(specializedModuleDetails),
    preprocessorCallbacks(std::move(preprocessorCallbacks))
{
}
 
unique_ptr<ASTConsumer> VuoReplaceGenericTypesWithSpecializedAction::CreateASTConsumer(CompilerInstance &compilerInstance, llvm::StringRef inFile)
{
    rewriter.setSourceMgr(compilerInstance.getSourceManager(), compilerInstance.getLangOpts());
    return unique_ptr<Consumer>(new Consumer(replacements, shouldReplaceGenericTypesInVuoTypeAnnotations, specializedModuleDetails, rewriter));
}
 
bool VuoReplaceGenericTypesWithSpecializedAction::BeginSourceFileAction(CompilerInstance &CI)
{
    CI.getPreprocessor().addPPCallbacks(std::move(preprocessorCallbacks));
    return true;
}
 
void VuoReplaceGenericTypesWithSpecializedAction::EndSourceFileAction()
{
    rewriter.getEditBuffer(rewriter.getSourceMgr().getMainFileID()).write(output);
}
 
VuoReplaceGenericTypesWithSpecializedAction::Consumer::Consumer(const map<string, string> &replacements,
                                                                bool shouldReplaceGenericTypesInVuoTypeAnnotations,
                                                                json_object *specializedModuleDetails,
                                                                clang::Rewriter &rewriter) :
    visitor(replacements, shouldReplaceGenericTypesInVuoTypeAnnotations, specializedModuleDetails, rewriter)
{
}
 
void VuoReplaceGenericTypesWithSpecializedAction::Consumer::HandleTranslationUnit(clang::ASTContext &context)
{
    visitor.TraverseDecl(context.getTranslationUnitDecl());
}
 
VuoReplaceGenericTypesWithSpecializedAction::Visitor::Visitor(const map<string, string> &replacements,
                                                              bool shouldReplaceGenericTypesInVuoTypeAnnotations,
                                                              json_object *specializedModuleDetails,
                                                              clang::Rewriter &rewriter) :
    replacements(replacements),
    shouldReplaceGenericTypesInVuoTypeAnnotations(shouldReplaceGenericTypesInVuoTypeAnnotations),
    specializedModuleDetails(specializedModuleDetails),
    rewriter(rewriter)
{
}
 
bool VuoReplaceGenericTypesWithSpecializedAction::Visitor::VisitFunctionDecl(FunctionDecl *decl)
{
    DeclarationNameInfo nameInfo = decl->getNameInfo();
    string name = nameInfo.getAsString();
 
    if (replaceGenericTypeNamesInIdentifier(name))
        rewriter.ReplaceText(nameInfo.getSourceRange(), name);
 
    return true;
}
 
bool VuoReplaceGenericTypesWithSpecializedAction::Visitor::VisitDeclRefExpr(DeclRefExpr *expr)
{
    DeclarationNameInfo nameInfo = expr->getNameInfo();
    string name = nameInfo.getAsString();
    string originalName = name;
 
    if (replaceGenericTypeNamesInIdentifier(name))
    {
        SourceLocation begin = expr->getLocStart();
        SourceLocation end = begin.getLocWithOffset(originalName.length()-1);
        SourceRange sourceRange(begin, end);
        rewriter.ReplaceText(sourceRange, name);
    }
 
    return true;
}
 
bool VuoReplaceGenericTypesWithSpecializedAction::Visitor::VisitAnnotateAttr(AnnotateAttr *attr)
{
    if (! shouldReplaceGenericTypesInVuoTypeAnnotations)
        return true;
 
    string annotation = attr->getAnnotation().str();
    if (VuoStringUtilities::beginsWith(annotation, "vuoType:"))
    {
        string attribute = rewriter.getRewrittenText(attr->getRange());
        if (replaceGenericTypeNamesInIdentifier(attribute))
            rewriter.ReplaceText(attr->getRange(), attribute);
    }
 
    return true;
}
 
bool VuoReplaceGenericTypesWithSpecializedAction::Visitor::VisitVarDecl(VarDecl *decl)
{
    if (! specializedModuleDetails)
        return true;
 
    if (decl->getNameAsString() == "moduleDetails" && decl->hasInit())
    {
        Expr *expr = decl->getInit();
        while (isa<CastExpr>(expr))
            expr = cast<CastExpr>(expr)->getSubExpr();
 
        clang::StringLiteral *metadataExpr = dyn_cast<clang::StringLiteral>(expr);
        if (metadataExpr)
        {
            json_object *metadataJson = json_tokener_parse(metadataExpr->getString().str().c_str());
            json_object_object_add(metadataJson, "specializedModule", specializedModuleDetails);
            string modifiedMetadata = string("VUO_STRINGIFY(") + json_object_to_json_string_ext(metadataJson, JSON_C_TO_STRING_PLAIN) + ")";
 
            SourceLocation begin = metadataExpr->getLocStart();
            SourceLocation end = metadataExpr->getLocEnd();
            SourceRange sourceRange(begin, end);
            rewriter.ReplaceText(sourceRange, modifiedMetadata);
        }
    }
 
    return true;
}
 
bool VuoReplaceGenericTypesWithSpecializedAction::Visitor::replaceGenericTypeNamesInIdentifier(string &identifier)
{
    return VuoGenericType::replaceGenericTypeNamesInString(identifier, replacements, orderedGenericTypeNames);
}