如何检查Clang ASTvisitor中的变量声明是否为数组

时间:2016-07-22 01:32:54

标签: c++ clang llvm-clang

我正在尝试确定ASTvisitor中的变量声明是否为数组,如果是数组,我想确定数组的维数。您可以在下面找到我的代码。

bool VisitVarDecl(VarDecl *var)
    {
        if (astContext->getSourceManager().isInMainFile(var->getLocStart())) //checks if the node is in the main = input file.
        {
            FullSourceLoc FullLocation = astContext->getFullLoc(var->getLocStart());
            if((var->hasLocalStorage() || var->isStaticLocal ()))
            {
                if (!var->isDefinedOutsideFunctionOrMethod())
                {
                    if(avoid == 0)
                    {
                        numVariables++;
                        string varName = var->getQualifiedNameAsString();
                        string varType = var->getType().getAsString();
                        const Type *type = var->getType().getTypePtr();
                        if(type->isConstantArrayType())
                        {
                            const ArrayType *Array = type->castAsArrayTypeUnsafe();
                            cout << "Is array of type: "  << Array->getElementType().getAsString() << endl;
                        }
                        REPORT << "[" << FullLocation.getSpellingLineNumber() << "," << FullLocation.getSpellingColumnNumber() << "]Variable Declaration: " << varName << " of type " << varType << "\n";
                        APIs << varType << ";";
                    }
                    else
                    {
                        avoid--;
                        REPORT << "Avoid is: " << avoid << endl;
                    }
                }
            }
        }
        return true;
    }

我不知道我是否正确地做了#34;铸造&#34;从VarDecl到ArrayType。如果你有更好,更安全,更少草率的方式,请接受任何意见。 此外,我现在的主要问题是如何获得数组的维数,甚至是单元格的大小。

谢谢大家。

1 个答案:

答案 0 :(得分:3)

试试这个:

bool VisitVarDecl(VarDecl *D){  
    if (auto t =  dyn_cast_or_null<ConstantArrayType>(D->getType().getTypePtr())) {  
        t->getSize().dump(); // We got the array size as an APInt here  
    }  
    return true;  
}  

最后,这是“一种更好,更安全,更不邋way的方式”:
the-isa-cast-and-dyn-cast-templates