ENH: Expose getShape, getNumTuples from AttributeMatrix

Signed-off-by: Michael Jackson <[email protected]>
BlueQuartzSoftware · Apr 17, 2024 · c7a64a7 · c7a64a7
1 parent a640bf2
commit c7a64a7
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/Plugins/SimplnxCore/wrapping/python/simplnxpy.cpp b/src/Plugins/SimplnxCore/wrapping/python/simplnxpy.cpp
@@ -835,6 +835,11 @@ PYBIND11_MODULE(simplnx, mod)
 
   py::class_<AttributeMatrix, BaseGroup, std::shared_ptr<AttributeMatrix>> attributeMatrix(mod, "AttributeMatrix");
   attributeMatrix.def("resize_tuples", &AttributeMatrix::resizeTuples, "Resize the tuples with the given shape");
+  attributeMatrix.def("tuple_shape", &AttributeMatrix::getShape, "Returns the Tuple dimensions of the AttributeMatrix");
+  attributeMatrix.def("__len__", &AttributeMatrix::getNumTuples, "Returns the total number of tuples");
+  attributeMatrix.def_property_readonly("size", &AttributeMatrix::getNumTuples, "Returns the total number of tuples");
+
+  baseGroup.def("", &BaseGroup::getSize);
 
   py::class_<IArray, DataObject, std::shared_ptr<IArray>> iArray(mod, "IArray");
   iArray.def_property_readonly("tuple_shape", &IArray::getTupleShape);