Print connections in Print Fabric stage button

CesiumGS · Jul 12, 2023 · 6402ef2 · 6402ef2
1 parent f060e6b
commit 6402ef2
Showing 1 changed file with 24 additions and 0 deletions.
diff --git a/src/core/src/FabricUtil.cpp b/src/core/src/FabricUtil.cpp
@@ -139,6 +139,19 @@ std::string printAttributeValue(
     }
 }
 
+std::string printConnection(const omni::fabric::Path& primPath, const omni::fabric::Token& attributeName) {
+    auto stageReaderWriter = UsdUtil::getFabricStageReaderWriter();
+    const auto connection = stageReaderWriter.getConnection(primPath, attributeName);
+    if (connection == nullptr) {
+        return NO_DATA_STRING;
+    }
+
+    const auto path = omni::fabric::Path(connection->path).getText();
+    const auto attrName = omni::fabric::Token(connection->attrName).getText();
+
+    return fmt::format("Path: {}, Attribute Name: {}", path, attrName);
+}
+
 std::string printAttributeValue(const omni::fabric::Path& primPath, const omni::fabric::AttrNameAndType& attribute) {
     auto stageReaderWriter = UsdUtil::getFabricStageReaderWriter();
 
@@ -165,6 +178,17 @@ std::string printAttributeValue(const omni::fabric::Path& primPath, const omni::
                 }
                 break;
             }
+            case omni::fabric::BaseDataType::eConnection: {
+                switch (componentCount) {
+                    case 1: {
+                        return printConnection(primPath, name);
+                    }
+                    default: {
+                        break;
+                    }
+                }
+                break;
+            }
             case omni::fabric::BaseDataType::eToken: {
                 switch (componentCount) {
                     case 1: {