Remove VertebralBody

[clitk.git] / segmentation / clitkExtractMediastinumFilter.txx

diff --git a/segmentation/clitkExtractMediastinumFilter.txx b/segmentation/clitkExtractMediastinumFilter.txx
index e3f83dc17b6786c500a36c6aa603b2bc217b272e..312fd14ed01d518cddfed3236b5a0ff720526f00 100644 (file)
--- a/segmentation/clitkExtractMediastinumFilter.txx
+++ b/segmentation/clitkExtractMediastinumFilter.txx
@@ -329,6 +329,13 @@ GenerateOutputInformation() {
     this->template StopCurrentStep<MaskImageType>(output);
   }
 
+  //--------------------------------------------------------------------
+  // Remove VertebralBody part
+  this->StartNewStep("[Mediastinum] Remove VertebralBody");  
+  MaskImagePointer VertebralBody = this->GetAFDB()->template GetImage<MaskImageType>("VertebralBody");
+  clitk::AndNot<MaskImageType>(output, VertebralBody, this->GetBackgroundValue());
+  this->template StopCurrentStep<MaskImageType>(output);
+
   //--------------------------------------------------------------------
   // Generic RelativePosition processes
   output = this->ApplyRelativePositionList("Mediastinum", output);