================ @@ -0,0 +1,129 @@ +//===---------- 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 { + +#define DISPATCH(CLASS) \ + return static_cast<Derived *>(this)->visit##CLASS( \ + static_cast<const CLASS *>(T)) + +template <template <typename> class Ptr, typename Derived> +class SubobjectVisitorBase { + ASTContext &Ctx; + template <typename Class> using ptr_t = typename Ptr<Class>::type; + template <typename Class> + using non_ptr_t = typename std::remove_pointer<ptr_t<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(); + visit(ElTy); + return; + } + + if (ptr_t<RecordDecl> RD = QT->getAsRecordDecl()) { + traverseRecord(RD); + return; + } + + visitGenericType(QT.getCanonicalType().getTypePtr()); + } + + void traverseRecord(ptr_t<RecordDecl> RD) { + if (ptr_t<CXXRecordDecl> CRD = dyn_cast<CXXRecordDecl>(RD)) { + for (non_ptr_t<CXXBaseSpecifier> BS : CRD->bases()) { + if (!static_cast<Derived *>(this)->visitBaseSpecifierPre(BS)) + continue; + visit(BS.getType()); + static_cast<Derived *>(this)->visitBaseSpecifierPost(BS); + } + } + for (ptr_t<FieldDecl> FD : RD->fields()) { + if (!static_cast<Derived *>(this)->visitFieldDeclPre(FD)) + continue; + visit(FD->getType()); + static_cast<Derived *>(this)->visitFieldDeclPost(FD); + } ---------------- Fznamznon wrote:
Yeah, fixed in 74f2460cc0da5379ddec79d1624f0305da600803 https://github.com/llvm/llvm-project/pull/192957 _______________________________________________ cfe-commits mailing list [email protected] https://lists.llvm.org/cgi-bin/mailman/listinfo/cfe-commits
