Temp

src/core/include/cesium/omniverse/FabricGeometry.h

@@ -39,6 +39,7 @@ class FabricGeometry {
     [[nodiscard]] const FabricGeometryDefinition& getGeometryDefinition() const;
 
     void setMaterial(const omni::fabric::Path& materialPath);
+    void setTextureIndex(uint64_t textureIndex);
 
   private:
     void initialize();

src/core/include/cesium/omniverse/FabricMaterial.h

@@ -20,7 +20,8 @@ class FabricMaterial {
         const omni::fabric::Path& path,
         const FabricMaterialDefinition& materialDefinition,
         const pxr::TfToken& defaultTextureAssetPathToken,
-        long stageId);
+        long stageId,
+        bool useSharedMaterialForTexturedMaterials);
     ~FabricMaterial();
 
     void setMaterial(int64_t tilesetId, const MaterialInfo& materialInfo);
@@ -57,6 +58,7 @@ class FabricMaterial {
     const FabricMaterialDefinition _materialDefinition;
     const pxr::TfToken _defaultTextureAssetPathToken;
     const long _stageId;
+    const bool _useSharedMaterialForTexturedMaterials;
 
     omni::fabric::Path _shaderPath;
     omni::fabric::Path _baseColorTexturePath;

src/core/include/cesium/omniverse/FabricResourceManager.h

@@ -132,6 +132,8 @@ class FabricResourceManager {
 
     bool _debugRandomColors{false};
 
+    bool _useSharedMaterialForTexturedMaterials{true};
+
     std::atomic<uint64_t> _geometryId{0};
     std::atomic<uint64_t> _materialId{0};
     std::atomic<uint64_t> _textureId{0};

src/core/include/cesium/omniverse/FabricTexture.h

@@ -17,19 +17,21 @@ struct ImageCesium;
 namespace cesium::omniverse {
 class FabricTexture {
   public:
-    FabricTexture(const std::string& name);
+    FabricTexture(const std::string& name, uint64_t index);
     ~FabricTexture();
 
     void setImage(const CesiumGltf::ImageCesium& image);
 
     void setActive(bool active);
 
     [[nodiscard]] const pxr::TfToken& getAssetPathToken() const;
+    [[nodiscard]] uint64_t getIndex() const;
 
   private:
     void reset();
 
     std::unique_ptr<omni::ui::DynamicTextureProvider> _texture;
     pxr::TfToken _assetPathToken;
+    uint64_t _index;
 };
 } // namespace cesium::omniverse

src/core/include/cesium/omniverse/Tokens.h

@@ -72,6 +72,7 @@ __pragma(warning(push)) __pragma(warning(disable : 4003))
     ((primvars_displayOpacity, "primvars:displayOpacity")) \
     ((primvars_normals, "primvars:normals")) \
     ((primvars_st, "primvars:st")) \
+    ((primvars_textureIndex, "primvars:textureIndex")) \
     ((primvars_vertexColor, "primvars:vertexColor"))
 
 TF_DECLARE_PUBLIC_TOKENS(UsdTokens, USD_TOKENS);
@@ -140,6 +141,7 @@ const omni::fabric::Type primvars_displayColor(omni::fabric::BaseDataType::eFloa
 const omni::fabric::Type primvars_displayOpacity(omni::fabric::BaseDataType::eFloat, 1, 1, omni::fabric::AttributeRole::eNone);
 const omni::fabric::Type primvars_normals(omni::fabric::BaseDataType::eFloat, 3, 1, omni::fabric::AttributeRole::eNormal);
 const omni::fabric::Type primvars_st(omni::fabric::BaseDataType::eFloat, 2, 1, omni::fabric::AttributeRole::eTexCoord);
+const omni::fabric::Type primvars_textureIndex(omni::fabric::BaseDataType::eInt, 1, 0, omni::fabric::AttributeRole::eNone);
 const omni::fabric::Type primvars_vertexColor(omni::fabric::BaseDataType::eFloat, 3, 1, omni::fabric::AttributeRole::eColor);
 const omni::fabric::Type Shader(omni::fabric::BaseDataType::eTag, 1, 0, omni::fabric::AttributeRole::ePrimTypeName);
 const omni::fabric::Type subdivisionScheme(omni::fabric::BaseDataType::eToken, 1, 0, omni::fabric::AttributeRole::eNone);

src/core/src/FabricGeometry.cpp

@@ -101,6 +101,16 @@ void FabricGeometry::setMaterial(const omni::fabric::Path& materialPath) {
     materialBindingFabric[0] = materialPath;
 }
 
+void FabricGeometry::setTextureIndex(uint64_t textureIndex) {
+    if (stageDestroyed()) {
+        return;
+    }
+
+    auto srw = UsdUtil::getFabricStageReaderWriter();
+    auto textureIndexFabric = srw.getArrayAttributeWr<int>(_path, FabricTokens::primvars_textureIndex);
+    textureIndexFabric[0] = static_cast<int>(textureIndex);
+}
+
 void FabricGeometry::initialize() {
     const auto hasTexcoords = _geometryDefinition.hasTexcoords();
     const auto hasNormals = _geometryDefinition.hasNormals();
@@ -134,6 +144,7 @@ void FabricGeometry::initialize() {
 
     if (hasTexcoords) {
         attributes.addAttribute(FabricTypes::primvars_st, FabricTokens::primvars_st);
+        attributes.addAttribute(FabricTypes::primvars_textureIndex, FabricTokens::primvars_textureIndex);
     }
 
     if (hasNormals) {
@@ -157,6 +168,7 @@ void FabricGeometry::initialize() {
     // Initialize primvars
     size_t primvarsCount = 0;
     size_t primvarIndexSt = 0;
+    size_t primvarTextureIndex = 0;
     size_t primvarIndexNormal = 0;
     size_t primvarIndexVertexColor = 0;
 
@@ -165,6 +177,7 @@ void FabricGeometry::initialize() {
 
     if (hasTexcoords) {
         primvarIndexSt = primvarsCount++;
+        primvarTextureIndex = primvarsCount++;
     }
 
     if (hasNormals) {
@@ -179,6 +192,7 @@ void FabricGeometry::initialize() {
     srw.setArrayAttributeSize(_path, FabricTokens::primvarInterpolations, primvarsCount);
     srw.setArrayAttributeSize(_path, FabricTokens::primvars_displayColor, 1);
     srw.setArrayAttributeSize(_path, FabricTokens::primvars_displayOpacity, 1);
+    srw.setArrayAttributeSize(_path, FabricTokens::primvars_textureIndex, 1);
 
     // clang-format off
     auto primvarsFabric = srw.getArrayAttributeWr<omni::fabric::TokenC>(_path, FabricTokens::primvars);
@@ -194,6 +208,8 @@ void FabricGeometry::initialize() {
     if (hasTexcoords) {
         primvarsFabric[primvarIndexSt] = FabricTokens::primvars_st;
         primvarInterpolationsFabric[primvarIndexSt] = FabricTokens::vertex;
+        primvarsFabric[primvarTextureIndex] = FabricTokens::primvars_textureIndex;
+        primvarInterpolationsFabric[primvarTextureIndex] = FabricTokens::constant;
     }
 
     if (hasNormals) {
@@ -245,6 +261,7 @@ void FabricGeometry::reset() {
 
     if (hasTexcoords) {
         srw.setArrayAttributeSize(_path, FabricTokens::primvars_st, 0);
+        srw.setArrayAttributeSize(_path, FabricTokens::primvars_textureIndex, 0);
     }
 
     if (hasNormals) {

src/core/src/FabricMaterial.cpp

@@ -16,11 +16,13 @@ FabricMaterial::FabricMaterial(
     const omni::fabric::Path& path,
     const FabricMaterialDefinition& materialDefinition,
     const pxr::TfToken& defaultTextureAssetPathToken,
-    long stageId)
+    long stageId,
+    bool useSharedMaterialForTexturedMaterials)
     : _materialPath(path)
     , _materialDefinition(materialDefinition)
     , _defaultTextureAssetPathToken(defaultTextureAssetPathToken)
-    , _stageId(stageId) {
+    , _stageId(stageId)
+    , _useSharedMaterialForTexturedMaterials(useSharedMaterialForTexturedMaterials) {
 
     if (stageDestroyed()) {
         return;

src/core/src/FabricMaterialPool.cpp

@@ -25,7 +25,7 @@ const FabricMaterialDefinition& FabricMaterialPool::getMaterialDefinition() cons
 std::shared_ptr<FabricMaterial> FabricMaterialPool::createObject(uint64_t objectId) {
     const auto pathStr = fmt::format("/fabric_material_pool_{}_object_{}", _poolId, objectId);
     const auto path = omni::fabric::Path(pathStr.c_str());
-    return std::make_shared<FabricMaterial>(path, _materialDefinition, _defaultTextureAssetPathToken, _stageId);
+    return std::make_shared<FabricMaterial>(path, _materialDefinition, _defaultTextureAssetPathToken, _stageId, false);
 }
 
 void FabricMaterialPool::setActive(std::shared_ptr<FabricMaterial> material, bool active) {

src/core/src/FabricPrepareRenderResources.cpp

@@ -190,8 +190,7 @@ void setFabricMeshes(
             geometry->setMaterial(material->getPath());
 
             if (baseColorTexture != nullptr && materialInfo.baseColorTexture.has_value()) {
-                material->setBaseColorTexture(
-                    baseColorTexture->getAssetPathToken(), materialInfo.baseColorTexture.value());
+                geometry->setTextureIndex(baseColorTexture->getIndex());
             }
         } else if (!tilesetMaterialPath.IsEmpty()) {
             geometry->setMaterial(FabricUtil::toFabricPath(tilesetMaterialPath));

src/core/src/FabricResourceManager.cpp

@@ -79,9 +79,9 @@ std::shared_ptr<FabricGeometry> FabricResourceManager::acquireGeometry(
     return geometry;
 }
 
-bool useSharedMaterial(const FabricMaterialDefinition& materialDefinition) {
+bool useSharedMaterial(const FabricMaterialDefinition& materialDefinition, bool useSharedMaterialForTexturedMaterials) {
     if (materialDefinition.hasBaseColorTexture()) {
-        return false;
+        return useSharedMaterialForTexturedMaterials;
     }
 
     return true;
@@ -91,7 +91,8 @@ std::shared_ptr<FabricMaterial>
 FabricResourceManager::createMaterial(const FabricMaterialDefinition& materialDefinition, long stageId) {
     const auto pathStr = fmt::format("/fabric_material_{}", getNextMaterialId());
     const auto path = omni::fabric::Path(pathStr.c_str());
-    return std::make_shared<FabricMaterial>(path, materialDefinition, _defaultTextureAssetPathToken, stageId);
+    return std::make_shared<FabricMaterial>(
+        path, materialDefinition, _defaultTextureAssetPathToken, stageId, _useSharedMaterialForTexturedMaterials);
 }
 
 void FabricResourceManager::removeSharedMaterial(const SharedMaterial& sharedMaterial) {
@@ -168,7 +169,7 @@ std::shared_ptr<FabricMaterial> FabricResourceManager::acquireMaterial(
     int64_t tilesetId) {
     FabricMaterialDefinition materialDefinition(materialInfo, hasImagery, _disableTextures);
 
-    if (useSharedMaterial(materialDefinition)) {
+    if (useSharedMaterial(materialDefinition, _useSharedMaterialForTexturedMaterials)) {
         return acquireSharedMaterial(materialInfo, materialDefinition, stageId, tilesetId);
     }
 
@@ -191,8 +192,9 @@ std::shared_ptr<FabricMaterial> FabricResourceManager::acquireMaterial(
 
 std::shared_ptr<FabricTexture> FabricResourceManager::acquireTexture() {
     if (_disableTexturePool) {
-        const auto name = fmt::format("/fabric_texture_{}", getNextTextureId());
-        return std::make_shared<FabricTexture>(name);
+        const auto id = getNextTextureId();
+        const auto name = fmt::format("/fabric_texture_{}", id);
+        return std::make_shared<FabricTexture>(name, id);
     }
 
     std::scoped_lock<std::mutex> lock(_poolMutex);
@@ -223,7 +225,7 @@ void FabricResourceManager::releaseGeometry(const std::shared_ptr<FabricGeometry
 void FabricResourceManager::releaseMaterial(const std::shared_ptr<FabricMaterial>& material) {
     const auto& materialDefinition = material->getMaterialDefinition();
 
-    if (useSharedMaterial(materialDefinition)) {
+    if (useSharedMaterial(materialDefinition, _useSharedMaterialForTexturedMaterials)) {
         releaseSharedMaterial(material);
         return;
     }

src/core/src/FabricTexture.cpp

@@ -32,9 +32,10 @@ carb::Format getCompressedImageFormat(CesiumGltf::GpuCompressedPixelFormat pixel
 
 } // namespace
 
-FabricTexture::FabricTexture(const std::string& name)
+FabricTexture::FabricTexture(const std::string& name, uint64_t index)
     : _texture(std::make_unique<omni::ui::DynamicTextureProvider>(name))
-    , _assetPathToken(UsdUtil::getDynamicTextureProviderAssetPathToken(name)) {
+    , _assetPathToken(UsdUtil::getDynamicTextureProviderAssetPathToken(name))
+    , _index(index) {
     reset();
 }
 
@@ -50,6 +51,10 @@ const pxr::TfToken& FabricTexture::getAssetPathToken() const {
     return _assetPathToken;
 }
 
+uint64_t FabricTexture::getIndex() const {
+    return _index;
+}
+
 void FabricTexture::reset() {
     const auto bytes = std::array<uint8_t, 4>{{255, 255, 255, 255}};
     const auto size = carb::Uint2{1, 1};

src/core/src/FabricTexturePool.cpp

@@ -12,7 +12,7 @@ FabricTexturePool::FabricTexturePool(uint64_t poolId, uint64_t initialCapacity)
 
 std::shared_ptr<FabricTexture> FabricTexturePool::createObject(uint64_t objectId) {
     const auto name = fmt::format("/fabric_texture_pool_{}_object_{}", _poolId, objectId);
-    return std::make_shared<FabricTexture>(name);
+    return std::make_shared<FabricTexture>(name, objectId);
 }
 
 void FabricTexturePool::setActive(std::shared_ptr<FabricTexture> texture, bool active) {