...

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

diff --git a/lib/cpExtensions/Algorithms/CPRImageFilter.hxx b/lib/cpExtensions/Algorithms/CPRImageFilter.hxx
index a29c6565f0fb8a79bf4653f7c621d6e417936bb3..de76e7fd266e755d5327cd80c6fcb578f563f558 100644 (file)
--- a/lib/cpExtensions/Algorithms/CPRImageFilter.hxx
+++ b/lib/cpExtensions/Algorithms/CPRImageFilter.hxx
@@ -1,6 +1,8 @@
 #ifndef __cpExtensions__Algorithms__CPRImageFilter__hxx__
 #define __cpExtensions__Algorithms__CPRImageFilter__hxx__
 
+#include <itkMinimumMaximumImageCalculator.h>
+
 // -------------------------------------------------------------------------
 template< class _TImage, class _TCurve >
 _TCurve* cpExtensions::Algorithms::CPRImageFilter< _TImage, _TCurve >::
@@ -26,7 +28,7 @@ template< class _TImage, class _TCurve >
 void cpExtensions::Algorithms::CPRImageFilter< _TImage, _TCurve >::
 SetCurve( _TCurve* curve )
 {
-  this->itk::ProcessObject::SetInput( 1, curve );
+  this->itk::ProcessObject::SetNthInput( 1, curve );
 }
 
 // -------------------------------------------------------------------------
@@ -68,10 +70,56 @@ template< class _TImage, class _TCurve >
 void cpExtensions::Algorithms::CPRImageFilter< _TImage, _TCurve >::
 GenerateData( )
 {
-  /* TODO
-     double m_SliceRadius
-     typename TInterpolateFunction::Pointer m_Interpolator
-  */
+  typedef itk::MinimumMaximumImageCalculator< _TImage > _TMinMax;
+
+  auto input = this->GetInput( );
+  auto curve = this->GetCurve( );
+  auto output = this->GetOutput( );
+
+  // Compute image intensity range
+  typename _TMinMax::Pointer minmax = _TMinMax::New( );
+  minmax->SetImage( input );
+  minmax->Compute( );
+
+  // Main loop
+  this->m_Join = TJoinFilter::New( );
+  this->m_Slicers.clear( );
+  unsigned long N = curve->GetNumberOfPoints( );
+  double len = 0;
+  typename _TCurve::TPoint p, q;
+  for( unsigned long n = 0; n < N; ++n )
+  {
+    p = curve->GetPoint( n );
+    if( n > 0 )
+      len += p.EuclideanDistanceTo( q );
+    q = p;
+
+    // Prepare slicer
+    typename TSlicer::Pointer slicer = TSlicer::New( );
+    slicer->SetInput( input );
+    slicer->SetDefaultValue( minmax->GetMinimum( ) );
+    slicer->SpacingFromMinimumOn( );
+    if( this->m_Interpolator.IsNotNull( ) )
+      slicer->SetInterpolator( this->m_Interpolator );
+    slicer->SizeFromMaximumOff( );
+    if( this->m_SliceRadius > double( 0 ) )
+    {
+      slicer->SizeFromMinimumOff( );
+      slicer->SetSize( this->m_SliceRadius );
+    }
+    else
+      slicer->SizeFromMinimumOn( );
+    slicer->SetTranslation( p );
+    slicer->SetRotation( curve->GetFrame( n ) );
+    slicer->Update( );
+
+    this->m_Join->SetInput( n, slicer->GetOutput( ) );
+    this->m_Slicers.push_back( slicer );
+
+  } // rof
+  this->m_Join->SetSpacing( len / double( N ) );
+  this->m_Join->Update( );
+  this->GetOutput( )->Graft( this->m_Join->GetOutput( ) );
 }
 
 #endif // __cpExtensions__Algorithms__CPRImageFilter__hxx__