Inherits from ImageToImageGenericFilter

[clitk.git] / segmentation / clitkExtractLungFilter.txx

diff --git a/segmentation/clitkExtractLungFilter.txx b/segmentation/clitkExtractLungFilter.txx
index f291739d09cab2f346ca6f2aae46ab72493c0fca..e4aa902ee6cd2f81d91de731506a34b6dad6baab 100644 (file)
--- a/segmentation/clitkExtractLungFilter.txx
+++ b/segmentation/clitkExtractLungFilter.txx
@@ -38,6 +38,7 @@ ExtractLungFilter():
   clitk::FilterBase(),
   itk::ImageToImageFilter<TInputImageType, TMaskImageType>()
 {
+  SetNumberOfSteps(10);
   // Default global options
   this->SetNumberOfRequiredInputs(2);
   SetPatientMaskBackgroundValue(0);
@@ -154,7 +155,6 @@ GenerateOutputInformation()
 { 
   input   = dynamic_cast<const TInputImageType*>(itk::ProcessObject::GetInput(0));
   Superclass::GenerateOutputInformation();
-//   MaskImagePointer output = this->GetOutput(0);
 
   // Get input pointers
   input   = dynamic_cast<const TInputImageType*>(itk::ProcessObject::GetInput(0));
@@ -166,6 +166,9 @@ GenerateOutputInformation()
     return;
   }
   
+  // Set Number of steps
+  SetNumberOfSteps(9);
+  
   //--------------------------------------------------------------------
   //--------------------------------------------------------------------
   StartNewStep("Set background to initial image");