Cast image filter added. ROI filter modified.

[cpPlugins.git] / lib / cpExtensions / Algorithms / SkeletonToImageFilter.hxx

diff --git a/lib/cpExtensions/Algorithms/SkeletonToImageFilter.hxx b/lib/cpExtensions/Algorithms/SkeletonToImageFilter.hxx
index 237fa79ab89560b7fe64aee94e86bcc095ea16f8..1f428c9e654c2c553b1c572569260a876c7b3f4d 100644 (file)
--- a/lib/cpExtensions/Algorithms/SkeletonToImageFilter.hxx
+++ b/lib/cpExtensions/Algorithms/SkeletonToImageFilter.hxx
@@ -13,14 +13,22 @@ TImageBase*
 cpExtensions::Algorithms::SkeletonToImageFilter< _TSkeleton, _TImage >::
 GetTemplateImage( ) const
 {
+  return(
+    dynamic_cast< const TImageBase* >(
+      this->itk::ProcessObject::GetInput( 0 )
+      )
+    );
 }
 
 // -------------------------------------------------------------------------
 template< class _TSkeleton, class _TImage >
 void 
 cpExtensions::Algorithms::SkeletonToImageFilter< _TSkeleton, _TImage >::
-SetTemplateImage( TImageBase* image )
+SetTemplateImage( const TImageBase* image )
 {
+  this->itk::ProcessObject::SetNthInput(
+    0, const_cast< TImageBase* >( image )
+    );
 }
 
 // -------------------------------------------------------------------------
@@ -31,14 +39,22 @@ TSkeleton*
 cpExtensions::Algorithms::SkeletonToImageFilter< _TSkeleton, _TImage >::
 GetSkeleton( ) const
 {
+  return(
+    dynamic_cast< const TSkeleton* >(
+      this->itk::ProcessObject::GetInput( 1 )
+      )
+    );
 }
 
 // -------------------------------------------------------------------------
 template< class _TSkeleton, class _TImage >
 void
 cpExtensions::Algorithms::SkeletonToImageFilter< _TSkeleton, _TImage >::
-SetSkeleton( TSkeleton* skeleton )
+SetSkeleton( const TSkeleton* skeleton )
 {
+  this->itk::ProcessObject::SetNthInput(
+    1, const_cast< TSkeleton* >( skeleton )
+    );
 }
 
 // -------------------------------------------------------------------------