ENH: Implement ITK's WriteImage functions

InsightSoftwareConsortium · Mar 11, 2021 · de347bf · de347bf
1 parent ab38077
commit de347bf
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 34 deletions.
diff --git a/src/Filtering/BinaryMathematicalMorphology/ThinImage/Code.cxx b/src/Filtering/BinaryMathematicalMorphology/ThinImage/Code.cxx
@@ -23,8 +23,6 @@
 
 using ImageType = itk::Image<unsigned char, 2>;
 
-static void
-WriteImage(const ImageType::Pointer image, const std::string & fileName);
 static void
 CreateImage(ImageType::Pointer image);
 
@@ -44,7 +42,7 @@ main(int argc, char * argv[])
   else
   {
     CreateImage(image);
-    WriteImage(image, "input.png");
+    itk::WriteImage(image, "input.png");
   }
 
   using BinaryThinningImageFilterType = itk::BinaryThinningImageFilter<ImageType, ImageType>;
@@ -60,7 +58,7 @@ main(int argc, char * argv[])
   rescaler->SetOutputMaximum(255);
   rescaler->Update();
 
-  WriteImage(rescaler->GetOutput(), "Output.png");
+  itk::WriteImage(rescaler->GetOutput(), "Output.png");
 
   return EXIT_SUCCESS;
 }
@@ -93,13 +91,3 @@ CreateImage(ImageType::Pointer image)
     }
   }
 }
-
-void
-WriteImage(const ImageType::Pointer image, const std::string & fileName)
-{
-  using WriterType = itk::ImageFileWriter<ImageType>;
-  WriterType::Pointer writer = WriterType::New();
-  writer->SetFileName(fileName);
-  writer->SetInput(image);
-  writer->Update();
-}
diff --git a/src/Filtering/Convolution/NormalizedCorrelationOfMaskedImage/Code.cxx b/src/Filtering/Convolution/NormalizedCorrelationOfMaskedImage/Code.cxx
@@ -40,26 +40,22 @@ CreateImage(ImageType * const image);
 void
 CreateImageOfSquare(ImageType * const image, const itk::Index<2> & cornerOfSquare);
 
-template <typename TImage>
-void
-WriteImage(const TImage * const image, const std::string & filename);
-
 int
 main(int, char *[])
 {
   // Setup mask
   MaskType::Pointer mask = MaskType::New();
   CreateMask(mask);
 
-  WriteImage(mask.GetPointer(), "mask.png");
+  itk::WriteImage(mask.GetPointer(), "mask.png");
 
   // Setup image1
   ImageType::Pointer image1 = ImageType::New();
   itk::Index<2>      cornerOfSquare1;
   cornerOfSquare1[0] = 3;
   cornerOfSquare1[1] = 8;
   CreateImageOfSquare(image1, cornerOfSquare1);
-  WriteImage(image1.GetPointer(), "image1.png");
+  itk::WriteImage(image1.GetPointer(), "image1.png");
 
   // Setup image2
   itk::Index<2> offset;
@@ -72,7 +68,7 @@ main(int, char *[])
   cornerOfSquare2[1] = cornerOfSquare1[1] + offset[1];
   CreateImageOfSquare(image2, cornerOfSquare2);
 
-  WriteImage(image2.GetPointer(), "image2.png");
+  itk::WriteImage(image2.GetPointer(), "image2.png");
 
   // Create a kernel from an image
   itk::ImageKernelOperator<unsigned char> kernelOperator;
@@ -101,15 +97,15 @@ main(int, char *[])
   correlationFilter->SetTemplate(kernelOperator);
   correlationFilter->Update();
 
-  WriteImage(correlationFilter->GetOutput(), "correlation.mha");
+  itk::WriteImage(correlationFilter->GetOutput(), "correlation.mha");
 
   using RescaleFilterType = itk::RescaleIntensityImageFilter<FloatImageType, ImageType>;
   RescaleFilterType::Pointer rescaleFilter = RescaleFilterType::New();
   rescaleFilter->SetInput(correlationFilter->GetOutput());
   rescaleFilter->SetOutputMinimum(0);
   rescaleFilter->SetOutputMaximum(255);
   rescaleFilter->Update();
-  WriteImage(rescaleFilter->GetOutput(), "correlation.png");
+  itk::WriteImage(rescaleFilter->GetOutput(), "correlation.png");
 
   using MinimumMaximumImageCalculatorType = itk::MinimumMaximumImageCalculator<FloatImageType>;
   MinimumMaximumImageCalculatorType::Pointer minimumMaximumImageCalculatorFilter =
@@ -210,14 +206,3 @@ CreateImageOfSquare(ImageType * const image, const itk::Index<2> & cornerOfSquar
     ++imageIterator;
   }
 }
-
-template <typename TImage>
-void
-WriteImage(const TImage * const image, const std::string & filename)
-{
-  using WriterType = itk::ImageFileWriter<TImage>;
-  typename WriterType::Pointer writer = WriterType::New();
-  writer->SetFileName(filename);
-  writer->SetInput(image);
-  writer->Update();
-}