ENH: add a convenience function WriteImage

Modules/Core/Common/test/itkImageDuplicatorTest2.cxx

@@ -37,8 +37,6 @@ itkImageDuplicatorTest2(int argc, char * argv[])
 
   using ReaderType = itk::ImageFileReader<ImageType>;
   ReaderType::Pointer reader = ReaderType::New();
-  using WriterType = itk::ImageFileWriter<ImageType>;
-  WriterType::Pointer writer = WriterType::New();
   using DuplicatorType = itk::ImageDuplicator<ImageType>;
   DuplicatorType::Pointer dup = DuplicatorType::New();
   using AbsType = itk::AbsImageFilter<ImageType, ImageType>;
@@ -69,9 +67,7 @@ itkImageDuplicatorTest2(int argc, char * argv[])
     dup->Update();
     ImageType::ConstPointer dupImage = dup->GetOutput();
 
-    writer->SetInput(dupImage);
-    writer->SetFileName(argv[2]);
-    writer->Update();
+    itk::WriteImage(dupImage, argv[2]);
     std::cout << "Test SUCCESS" << std::endl;
   }
   catch (const itk::ExceptionObject & e)

Modules/Core/Common/test/itkImageRandomIteratorTest2.cxx

@@ -41,12 +41,9 @@ itkImageRandomIteratorTest2(int argc, char * argv[])
   using PixelType = unsigned long;
 
   using ImageType = itk::Image<PixelType, ImageDimension>;
-  using WriterType = itk::ImageFileWriter<ImageType>;
 
   ImageType::Pointer image = ImageType::New();
 
-  WriterType::Pointer writer = WriterType::New();
-
   ImageType::SizeType size;
 
   size[0] = 1000;
@@ -84,9 +81,8 @@ itkImageRandomIteratorTest2(int argc, char * argv[])
     ++counter;
   }
 
-  writer->SetInput(image);
-  writer->SetFileName(argv[1]);
-  writer->Update();
+
+  itk::WriteImage(image, argv[1]);
 
   if (argc > 4)
   {

Modules/Core/Common/test/itkStreamingImageFilterTest3.cxx

@@ -64,13 +64,7 @@ itkStreamingImageFilterTest3(int argc, char * argv[])
   streamer->SetNumberOfStreamDivisions(numberOfStreamDivisions);
   streamer->SetRegionSplitter(splitter);
 
-
-  using WriterType = itk::ImageFileWriter<ImageType>;
-  WriterType::Pointer writer = WriterType::New();
-  writer->SetFileName(outputFilename);
-  writer->SetInput(streamer->GetOutput());
-  writer->Update();
-
+  itk::WriteImage(streamer->GetOutput(), outputFilename);
 
   unsigned int expectedNumberOfStreams =
     splitter->GetNumberOfSplits(streamer->GetOutput()->GetLargestPossibleRegion(), numberOfStreamDivisions);

Modules/Core/Common/test/itkSymmetricSecondRankTensorImageReadTest.cxx

@@ -82,16 +82,9 @@ itkSymmetricSecondRankTensorImageReadTest(int ac, char * av[])
     ++itr;
   }
 
-  using MatrixWriterType = itk::ImageFileWriter<MatrixImageType>;
-
-  MatrixWriterType::Pointer matrixWriter = MatrixWriterType::New();
-
-  matrixWriter->SetInput(matrixImage);
-  matrixWriter->SetFileName(av[1]);
-
   try
   {
-    matrixWriter->Update();
+    itk::WriteImage(matrixImage, av[1]);
   }
   catch (const itk::ExceptionObject & excp)
   {

Modules/Core/Common/test/itkSymmetricSecondRankTensorImageWriteReadTest.cxx

@@ -67,16 +67,9 @@ itkSymmetricSecondRankTensorImageWriteReadTest(int ac, char * av[])
     ++itr;
   }
 
-  using TensorWriterType = itk::ImageFileWriter<TensorImageType>;
-
-  TensorWriterType::Pointer tensorWriter = TensorWriterType::New();
-
-  tensorWriter->SetInput(tensorImageInput);
-  tensorWriter->SetFileName(av[1]);
-
   try
   {
-    tensorWriter->Update();
+    itk::WriteImage(tensorImageInput, av[1]);
   }
   catch (const itk::ExceptionObject & excp)
   {

Modules/Filtering/BiasCorrection/test/itkN4BiasFieldCorrectionImageFilterTest.cxx

@@ -62,24 +62,6 @@ class CommandIterationUpdate : public itk::Command
   }
 };
 
-template <typename TImage>
-void
-WriteImage(const TImage * out, const char * filename)
-{
-  using WriterType = itk::ImageFileWriter<TImage>;
-  typename WriterType::Pointer w = WriterType::New();
-  w->SetInput(out);
-  w->SetFileName(filename);
-  try
-  {
-    w->Update();
-  }
-  catch (const itk::ExceptionObject & error)
-  {
-    std::cerr << error << std::endl;
-  }
-}
-
 template <typename TValue>
 TValue
 Convert(std::string optionString)

Modules/IO/ImageBase/include/itkImageFileWriter.h

@@ -235,6 +235,63 @@ class ITK_TEMPLATE_EXPORT ImageFileWriter : public ProcessObject
   int  m_CompressionLevel{ -1 };
   bool m_UseInputMetaDataDictionary{ true };
 };
+
+
+/** Convenience function for writing an image. */
+template <typename TImage>
+ITK_TEMPLATE_EXPORT void
+WriteImage(const TImage * image, const char * filename, bool compress = false)
+{
+  using WriterType = ImageFileWriter<TImage>;
+  typename WriterType::Pointer writer = WriterType::New();
+  writer->SetInput(image);
+  writer->SetFileName(filename);
+  writer->SetUseCompression(compress);
+  writer->Update();
+}
+template <typename TImage>
+ITK_TEMPLATE_EXPORT void
+WriteImage(const SmartPointer<const TImage> image, const char * filename, bool compress = false)
+{
+  WriteImage(image.GetPointer(), filename, compress);
+}
+template <typename TImage>
+ITK_TEMPLATE_EXPORT void
+WriteImage(TImage * image, const char * filename, bool compress = false)
+{
+  const TImage * constImage = image;
+  WriteImage(constImage, filename, compress);
+}
+template <typename TImage>
+ITK_TEMPLATE_EXPORT void
+WriteImage(SmartPointer<TImage> image, const char * filename, bool compress = false)
+{
+  WriteImage(image.GetPointer(), filename, compress);
+}
+template <typename TImage>
+ITK_TEMPLATE_EXPORT void
+WriteImage(const TImage * image, const std::string & filename, bool compress = false)
+{
+  WriteImage(image, filename.c_str(), compress);
+}
+template <typename TImage>
+ITK_TEMPLATE_EXPORT void
+WriteImage(const SmartPointer<const TImage> image, const std::string & filename, bool compress = false)
+{
+  WriteImage(image, filename.c_str(), compress);
+}
+template <typename TImage>
+ITK_TEMPLATE_EXPORT void
+WriteImage(TImage * image, const std::string & filename, bool compress = false)
+{
+  WriteImage(image, filename.c_str(), compress);
+}
+template <typename TImage>
+ITK_TEMPLATE_EXPORT void
+WriteImage(SmartPointer<TImage> image, const std::string & filename, bool compress = false)
+{
+  WriteImage(image, filename.c_str(), compress);
+}
 } // end namespace itk
 
 #ifndef ITK_MANUAL_INSTANTIATION