我正在尝试确定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。如果你有更好,更安全,更少草率的方式,请接受任何意见。 此外,我现在的主要问题是如何获得数组的维数,甚至是单元格的大小。
谢谢大家。
答案 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