Get all mdm attachment keys via iterator.

GothicKit · Aug 27, 2023 · b5d43c3 · b5d43c3
1 parent 222b8cd
commit b5d43c3
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 3 deletions.
diff --git a/include/phoenix/cffi/ModelMesh.h b/include/phoenix/cffi/ModelMesh.h
@@ -24,7 +24,9 @@ PXC_API void pxMdmDestroy(PxModelMesh* mdm);
 
 PXC_API uint32_t pxMdmGetMeshCount(PxModelMesh const* mdm);
 PXC_API PxSoftSkinMesh const* pxMdmGetMesh(PxModelMesh const* mdm, uint32_t i);
-PXC_API PxMultiResolutionMesh const* pxMdmGetAttachment(PxModelMesh const* mdm, char const* name);
+PXC_API uint32_t pxMdmGetAttachmentCount(PxModelMesh const* mdm);
+PXC_API char const* pxMdmGetAttachmentKey(PxModelMesh const* mdm, uint32_t index);
+PXC_API PxMultiResolutionMesh const* pxMdmGetAttachmentValue(PxModelMesh const* mdm, char const* name);
 PXC_API uint32_t pxMdmGetChecksum(PxModelMesh const* mdm);
 
 PXC_API PxMultiResolutionMesh const* pxSsmGetMesh(PxSoftSkinMesh const* ssm);

diff --git a/src/ModelMesh.cc b/src/ModelMesh.cc
@@ -40,11 +40,28 @@ PxSoftSkinMesh const* pxMdmGetMesh(PxModelMesh const* mdm, uint32_t i) {
 	return &mdm->meshes[i];
 }
 
-PxMultiResolutionMesh const* pxMdmGetAttachment(PxModelMesh const* mdm, char const* name) {
+uint32_t pxMdmGetAttachmentCount(PxModelMesh const* mdm) {
+	return mdm->attachments.size();
+}
+
+char const* pxMdmGetAttachmentKey(PxModelMesh const* mdm, uint32_t index) {
 	auto& attachments = mdm->attachments;
+    uint32_t currentIndex = 0;
 
+	for (auto i = attachments.begin(); i != attachments.end(); i++) {
+		if (currentIndex++ == index)
+			return i->first.c_str();
+	}
+
+	return nullptr;
+}
+
+PxMultiResolutionMesh const* pxMdmGetAttachmentValue(PxModelMesh const* mdm, char const* name) {
+	auto& attachments = mdm->attachments;
 	auto rv = attachments.find(name);
-	if (rv == attachments.end()) return nullptr;
+
+	if (rv == attachments.end())
+		return nullptr;
 
 	return &rv->second;
 }