Add getPointerLikeOrArrayElementType()

Author: Sirraide

clang/include/clang/AST/Type.h

@@ -2882,6 +2882,10 @@ class alignas(TypeAlignment) Type : public ExtQualsTypeCommonBase {
   ///   - DecayedType
   QualType getPointeeType() const;
 
+  /// Return getElementType() if this is an array type, and getPointeeType()
+  /// otherwise.
+  QualType getPointerLikeOrArrayElementType() const;
+
   /// Return the specified type with any "sugar" removed from the type,
   /// removing any typedefs, typeofs, etc., as well as any qualifiers.
   const Type *getUnqualifiedDesugaredType() const;

clang/lib/AST/Type.cpp

@@ -751,6 +751,12 @@ QualType Type::getPointeeType() const {
   return {};
 }
 
+QualType Type::getPointerLikeOrArrayElementType() const {
+  if (const auto *A = getAs<ArrayType>())
+    return A->getElementType();
+  return getPointeeType();
+}
+
 const RecordType *Type::getAsStructureType() const {
   // If this is directly a structure type, return it.
   if (const auto *RT = dyn_cast<RecordType>(this)) {