================ @@ -0,0 +1,98 @@ +//===---------- SubobjectVisitor.h - Subobject Visitor ----------*- C++ -*-===// +// +// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. +// See https://llvm.org/LICENSE.txt for license information. +// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception +// +//===----------------------------------------------------------------------===// +// +// This file defines the SubobjectVisitor interface, which recursively +// traverses subobjects within a type. +// +//===----------------------------------------------------------------------===// + +#ifndef LLVM_CLANG_AST_SUBOBJECTVISITOR_H +#define LLVM_CLANG_AST_SUBOBJECTVISITOR_H + +#include "clang/AST/Type.h" + +namespace clang { + +template <template <typename> class Ptr, typename Derived> +class SubobjectVisitorBase { + ASTContext &Ctx; + template <typename Class> using ptr_t = typename Ptr<Class>::type; + +public: + SubobjectVisitorBase(ASTContext &Ctx) : Ctx(Ctx) {} + + /// Return a reference to the derived class. + Derived &getDerived() { return *static_cast<Derived *>(this); } + + void visit(QualType QT) { + // If the type is an array, visit its element type. Separate traversal of + // arrays is not needed because the array will be encountered as a + // FieldDecl. + + if (QT->isArrayType()) { + QualType ElTy = + cast<ConstantArrayType>(Ctx.getAsArrayType(QT))->getElementType(); + getDerived().visit(ElTy); + return; + } + + if (ptr_t<RecordDecl> RD = QT->getAsRecordDecl()) { + getDerived().traverseRecord(RD); + return; + } + } + + void traverseRecord(ptr_t<RecordDecl> RD) { + if (ptr_t<CXXRecordDecl> CRD = dyn_cast<CXXRecordDecl>(RD)) { + for (CXXBaseSpecifier& BS : CRD->bases()) { + if (!getDerived().visitBaseSpecifierPre(&BS)) + continue; + getDerived().visit(BS.getType()); + getDerived().visitBaseSpecifierPost(&BS); + } + } + for (ptr_t<FieldDecl> FD : RD->fields()) { + if (!getDerived().visitFieldDeclPre(FD)) + continue; + getDerived().visit(FD->getType()); + getDerived().visitFieldDeclPost(FD); + } + } + + // Default base class specifier pre-order visitor. + bool visitBaseSpecifierPre(ptr_t<CXXBaseSpecifier> BS) { return true; } + + // Default base class specifier post-order visitor. + void visitBaseSpecifierPost(ptr_t<CXXBaseSpecifier> BS) {} + + // Default field pre-order visitor. + bool visitFieldDeclPre(ptr_t<FieldDecl> FD) { return true; } + + // Default field post-order visitor. + void visitFieldDeclPost(ptr_t<FieldDecl> FD) {} +}; + +template <typename Derived> +class SubobjectVisitor + : public SubobjectVisitorBase<std::add_pointer, Derived> { +public: + SubobjectVisitor(ASTContext &Ctx) + : SubobjectVisitorBase<std::add_pointer, Derived>(Ctx) {} +}; + +template <typename Derived> +class ConstSubobjectVisitor + : public SubobjectVisitorBase<llvm::make_const_ptr, Derived> { +public: + ConstSubobjectVisitor(ASTContext &Ctx) + : SubobjectVisitorBase<std::add_pointer, Derived>(Ctx) {} ---------------- Fznamznon wrote:
Fixed in 46db74989e52a7646bf9714cfa9782b3b5609265 . https://github.com/llvm/llvm-project/pull/192957 _______________________________________________ cfe-commits mailing list [email protected] https://lists.llvm.org/cgi-bin/mailman/listinfo/cfe-commits
