WIP: Fix nanobind memory leaks

Need to add destructor for memory allocated with new in the wrap function
ANTsX · Nov 13, 2024 · 142af47 · 142af47
1 parent 637912b
commit 142af47
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/src/antsImageToImageMetric.h b/src/antsImageToImageMetric.h
@@ -52,8 +52,11 @@ class ANTsImageToImageMetric {
 
 };
 
-
-
+template <typename MetricType>
+void capsuleDestructor(void* ptr) noexcept {
+    auto* metricPtr = static_cast<typename MetricType::Pointer*>(ptr);
+    delete metricPtr;
+}
 
 template <typename MetricType>
 ANTsImageToImageMetric<MetricType> wrap_metric( const typename MetricType::Pointer & itkMetric )
@@ -67,7 +70,7 @@ ANTsImageToImageMetric<MetricType> wrap_metric( const typename MetricType::Point
     antsMetric.dimension         = MetricType::FixedImageDimension;
     antsMetric.metrictype        = itkMetric->GetNameOfClass();
     antsMetric.isVector          = 0;
-    antsMetric.pointer           = nb::capsule(ptr);
+    antsMetric.pointer           = nb::capsule(ptr, capsuleDestructor<MetricType>);
 
     return antsMetric;
 }