================
@@ -96,24 +96,119 @@ void addRootSignatureMD(llvm::dxbc::RootSignatureVersion
RootSigVer,
RootSignatureValMD->addOperand(MDVals);
}
+// Gived a MemberExpr of a resource or resource array type, find the parent
+// VarDecl of the struct or class instance that contains this resource and
+// build the full resource name based on the member access path.
+//
+// For example, for a member access like "myStructArray[0].memberA",
+// this function will find the VarDecl of "myStructArray" and use the
+// EmbeddedResourceNameBuilder to build the resource name
+// "myStructArray.0.memberA".
+static const VarDecl *findStructResourceParentDeclAndBuildName(
+ const MemberExpr *ME, EmbeddedResourceNameBuilder &NameBuilder) {
+
+ SmallVector<const Expr *> WorkList;
+ const VarDecl *VD = nullptr;
+ const Expr *E = ME;
+
+ for (;;) {
+ if (const auto *DRE = dyn_cast<DeclRefExpr>(E)) {
+ assert(isa<VarDecl>(DRE->getDecl()) &&
+ "member expr base is not a var decl");
+ VD = cast<VarDecl>(DRE->getDecl());
+ NameBuilder.pushName(VD->getName());
+ break;
+ }
+
+ WorkList.push_back(E);
+ if (const auto *ME = dyn_cast<MemberExpr>(E))
+ E = ME->getBase();
+ else if (const auto *ICE = dyn_cast<ImplicitCastExpr>(E))
+ E = ICE->getSubExpr();
+ else if (const auto *ASE = dyn_cast<ArraySubscriptExpr>(E))
+ E = ASE->getBase();
+ else if (isa<CXXThisExpr>(E))
+ // Resource member access on "this" pointer not yet implemented
+ // (llvm/llvm-project#190299)
+ return nullptr;
+ else
+ llvm_unreachable("unexpected expr type in resource member access");
+ }
+
+ while (!WorkList.empty()) {
+ E = WorkList.pop_back_val();
+ if (const auto *ME = dyn_cast<MemberExpr>(E)) {
+ NameBuilder.pushName(
+ ME->getMemberNameInfo().getName().getAsIdentifierInfo()->getName());
+ } else if (const auto *ICE = dyn_cast<ImplicitCastExpr>(E)) {
+ if (ICE->getCastKind() == CK_UncheckedDerivedToBase) {
+ CXXRecordDecl *DerivedRD =
+ ICE->getSubExpr()->getType()->getAsCXXRecordDecl();
+ CXXRecordDecl *BaseRD = ICE->getType()->getAsCXXRecordDecl();
+ NameBuilder.pushBaseNameHierarchy(DerivedRD, BaseRD);
+ }
+ } else if (const auto *ASE = dyn_cast<ArraySubscriptExpr>(E)) {
+ const Expr *IdxExpr = ASE->getIdx();
+ std::optional<llvm::APSInt> Value =
+ IdxExpr->getIntegerConstantExpr(VD->getASTContext());
+ assert(Value &&
+ "expected constant index in struct with resource array access");
+ NameBuilder.pushArrayIndex(Value->getZExtValue());
+ } else {
+ llvm_unreachable("unexpected expr type in resource member access");
+ }
+ }
+ return VD;
+}
+
+// Given a MemberExpr of a resource or resource array type, find the
+// corresponding global resource declaration associated with the owning struct
+// or class instance via HLSLAssociatedResourceDeclAttr.
+static const VarDecl *
+findAssociatedResourceDeclForStruct(ASTContext &AST, const MemberExpr *ME) {
+
+ EmbeddedResourceNameBuilder NameBuilder;
+ const VarDecl *ParentVD =
+ findStructResourceParentDeclAndBuildName(ME, NameBuilder);
+ if (!ParentVD)
+ return nullptr;
+
+ if (!ParentVD->hasGlobalStorage())
+ return nullptr;
+
+ IdentifierInfo *II = NameBuilder.getNameAsIdentifier(AST);
+ for (const Attr *A : ParentVD->getAttrs()) {
+ const auto *ADA = dyn_cast<HLSLAssociatedResourceDeclAttr>(A);
+ if (!ADA)
+ continue;
+ VarDecl *AssocResVD = dyn_cast<VarDecl>(ADA->getResDecl());
----------------
bob80905 wrote:
Should we put this in an if statement?
https://github.com/llvm/llvm-project/pull/187127
_______________________________________________
cfe-commits mailing list
[email protected]
https://lists.llvm.org/cgi-bin/mailman/listinfo/cfe-commits
