================ @@ -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; + 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(); + 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 (non_ptr_t<CXXBaseSpecifier &> BS : CRD->bases()) { + if (getDerived().visitBaseSpecifierPre(&BS)) + getDerived().visit(BS.getType()); + getDerived().visitBaseSpecifierPost(&BS); ---------------- Fznamznon wrote:
Sure, we can do that in f025ea78b73d398324600bd8047d5c733668d8bc . Note though, that this now hides that bases() provides access to bases via CXXBaseSpecifier and not the pointer to it which is unusual for clang. https://github.com/llvm/llvm-project/pull/192957 _______________________________________________ cfe-commits mailing list [email protected] https://lists.llvm.org/cgi-bin/mailman/listinfo/cfe-commits
