bind gems mesh smoothing to python

ayendiki · Jan 21, 2022 · 6080572 · 6080572
1 parent ce65da0
commit 6080572
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 0 deletions.
diff --git a/python/bindings/gems/pyKvlMesh.cxx b/python/bindings/gems/pyKvlMesh.cxx
@@ -282,6 +282,15 @@ unsigned int KvlMeshCollection::MeshCount() const {
     return meshCollection->GetNumberOfMeshes();
 }
 
+void KvlMeshCollection::Smooth(double sigma) {
+    kvl::AtlasMeshCollection::Pointer atlasMeshCollectionPtr = this->GetMeshCollection();
+    kvl::AtlasMeshSmoother::Pointer  smoother = kvl::AtlasMeshSmoother::New();
+    smoother->SetMeshCollection(atlasMeshCollectionPtr);
+    smoother->SetSigma(sigma);
+    // note: it's very unclear that this actually updates the source mesh collection
+    smoother->GetSmoothedMeshCollection();
+}
+
 py::array_t<uint16_t> KvlMesh::RasterizeMesh(std::vector<size_t> size, int classNumber) {
     // Some typedefs
     typedef kvl::AtlasMeshAlphaDrawer::ImageType  AlphaImageType;

diff --git a/python/bindings/gems/pyKvlMesh.h b/python/bindings/gems/pyKvlMesh.h
@@ -7,6 +7,7 @@
 #include <vector>
 #include "itkObject.h"
 #include "kvlAtlasMeshCollection.h"
+#include "kvlAtlasMeshSmoother.h"
 #include "pyKvlTransform.h"
 
 namespace py = pybind11;
@@ -66,6 +67,7 @@ class KvlMeshCollection {
                    double initialStiffness,
                    unsigned int numberOfClasses, unsigned int numberOfMeshes);
     void Transform(const KvlTransform &transform);
+    void Smooth(double sigma);
 
     // C++ use only
     KvlMeshCollection();

diff --git a/python/bindings/gems/python.cxx b/python/bindings/gems/python.cxx
@@ -107,6 +107,7 @@ PYBIND11_MODULE(gemsbindings, m) {
             .def("construct", &KvlMeshCollection::Construct, py::return_value_policy::take_ownership)
             .def("read", &KvlMeshCollection::Read, py::return_value_policy::take_ownership)
             .def("transform", &KvlMeshCollection::Transform, py::return_value_policy::take_ownership)
+            .def("smooth", &KvlMeshCollection::Smooth, py::return_value_policy::take_ownership)
             .def("write", &KvlMeshCollection::Write, py::return_value_policy::take_ownership)
             ;
      m.def("setGlobalDefaultNumberOfThreads", &setGlobalDefaultNumberOfThreads, "Sets the maximum number of threads for ITK.");