improve-abc-curve

zenustech · Aug 8, 2024 · 5a51b04 · 5a51b04
1 parent 38054a5
commit 5a51b04
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/projects/Alembic/ReadAlembic.cpp b/projects/Alembic/ReadAlembic.cpp
@@ -990,6 +990,20 @@ static std::shared_ptr<PrimitiveObject> foundABCCurves(Alembic::AbcGeom::ICurves
             offset += count;
         }
     }
+    if (auto width = mesh.getWidthsParam()) {
+        auto widthsamp =
+            width.getIndexedValue(Alembic::Abc::v12::ISampleSelector((Alembic::AbcCoreAbstract::index_t)sample_index));
+        int index_size = (int)widthsamp.getIndices()->size();
+        prim->userData().set2("index_size", index_size);
+        if (prim->verts.size() == index_size) {
+            auto &width_attr = prim->add_attr<float>("width");
+            for (auto i = 0; i < prim->verts.size(); i++) {
+                auto index = widthsamp.getIndices()->operator[](i);
+                auto value = widthsamp.getVals()->operator[](index);
+                width_attr[i] = value;
+            }
+        }
+    }
     ICompoundProperty arbattrs = mesh.getArbGeomParams();
     read_attributes2(prim, arbattrs, iSS, read_done);
     ICompoundProperty usrData = mesh.getUserProperties();