ENH: Add Python wrapping of DrawQuadricImageFilter

RTKConsortium · Nov 12, 2022 · 10429e5 · 10429e5
1 parent cbebbad
commit 10429e5
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/include/rtkDrawQuadricImageFilter.hxx b/include/rtkDrawQuadricImageFilter.hxx
@@ -36,12 +36,12 @@ template <class TInputImage, class TOutputImage>
 void
 DrawQuadricImageFilter<TInputImage, TOutputImage>::BeforeThreadedGenerateData()
 {
-  if (this->GetConvexObject() == nullptr)
-    this->SetConvexObject(QuadricShape::New().GetPointer());
+  if (this->GetConvexShape() == nullptr)
+    this->SetConvexShape(QuadricShape::New().GetPointer());
 
   Superclass::BeforeThreadedGenerateData();
 
-  QuadricShape * qo = dynamic_cast<QuadricShape *>(this->GetConvexObject());
+  auto * qo = dynamic_cast<QuadricShape *>(this->GetModifiableConvexShape());
   if (qo == nullptr)
   {
     itkExceptionMacro("This is not a QuadricShape!");

diff --git a/wrapping/rtkDrawQuadricImageFilter.wrap b/wrapping/rtkDrawQuadricImageFilter.wrap
@@ -0,0 +1,3 @@
+itk_wrap_class("rtk::DrawQuadricImageFilter" POINTER)
+    itk_wrap_image_filter("${WRAP_ITK_REAL}" 2 3)
+itk_end_wrap_class()