ENH: Add .vti read/write regression testing using existing test data

Author: dzenanz

Modules/IO/VTK/test/CMakeLists.txt

@@ -9,6 +9,7 @@ set(
   itkVTKImageIOTest2.cxx
   itkVTKImageIOTest3.cxx
   itkVTIImageIOTest.cxx
+  itkVTIImageIOReadWriteTest.cxx
 )
 
 createtestdriver(ITKIOVTK "${ITKIOVTK-Test_LIBRARIES}" "${ITKIOVTKTests}")
@@ -21,6 +22,54 @@ itk_add_test(
     ${ITK_TEST_OUTPUT_DIR}
 )
 
+itk_add_test(
+  NAME itkVTIImageIOReadWriteTestCTHead1
+  COMMAND
+    ITKIOVTKTestDriver
+    --compare
+    DATA{${ITK_DATA_ROOT}/Input/cthead1.png}
+    ${ITK_TEST_OUTPUT_DIR}/itkVTIImageIOReadWriteTest_CTHead1.vti
+    itkVTIImageIOReadWriteTest
+    DATA{${ITK_DATA_ROOT}/Input/cthead1.png}
+    ${ITK_TEST_OUTPUT_DIR}/itkVTIImageIOReadWriteTest_CTHead1.vti
+)
+
+itk_add_test(
+  NAME itkVTIImageIOReadWriteTestVisibleWomanEyeSlice
+  COMMAND
+    ITKIOVTKTestDriver
+    --compare
+    DATA{${ITK_DATA_ROOT}/Input/VisibleWomanEyeSlice.png}
+    ${ITK_TEST_OUTPUT_DIR}/itkVTIImageIOReadWriteTest_VisibleWomanEyeSlice.vti
+    itkVTIImageIOReadWriteTest
+    DATA{${ITK_DATA_ROOT}/Input/VisibleWomanEyeSlice.png}
+    ${ITK_TEST_OUTPUT_DIR}/itkVTIImageIOReadWriteTest_VisibleWomanEyeSlice.vti
+)
+
+itk_add_test(
+  NAME itkVTIImageIOReadWriteTestHeadMRVolume
+  COMMAND
+    ITKIOVTKTestDriver
+    --compare
+    DATA{${ITK_DATA_ROOT}/Input/HeadMRVolume.mha}
+    ${ITK_TEST_OUTPUT_DIR}/itkVTIImageIOReadWriteTest_HeadMRVolume.vti
+    itkVTIImageIOReadWriteTest
+    DATA{${ITK_DATA_ROOT}/Input/HeadMRVolume.mha}
+    ${ITK_TEST_OUTPUT_DIR}/itkVTIImageIOReadWriteTest_HeadMRVolume.vti
+)
+
+itk_add_test(
+  NAME itkVTIImageIOReadWriteTestVHFColor
+  COMMAND
+    ITKIOVTKTestDriver
+    --compare
+    DATA{${ITK_DATA_ROOT}/Input/VHFColor.mhd,VHFColor.raw}
+    ${ITK_TEST_OUTPUT_DIR}/itkVTIImageIOReadWriteTest_VHFColor.vti
+    itkVTIImageIOReadWriteTest
+    DATA{${ITK_DATA_ROOT}/Input/VHFColor.mhd,VHFColor.raw}
+    ${ITK_TEST_OUTPUT_DIR}/itkVTIImageIOReadWriteTest_VHFColor.vti
+)
+
 itk_add_test(
   NAME itkVTKImageIO2Test
   COMMAND

Modules/IO/VTK/test/itkVTIImageIOReadWriteTest.cxx

@@ -0,0 +1,103 @@
+/*=========================================================================
+ *
+ *  Copyright NumFOCUS
+ *
+ *  Licensed under the Apache License, Version 2.0 (the "License");
+ *  you may not use this file except in compliance with the License.
+ *  You may obtain a copy of the License at
+ *
+ *         https://www.apache.org/licenses/LICENSE-2.0.txt
+ *
+ *  Unless required by applicable law or agreed to in writing, software
+ *  distributed under the License is distributed on an "AS IS" BASIS,
+ *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ *  See the License for the specific language governing permissions and
+ *  limitations under the License.
+ *
+ *=========================================================================*/
+
+#include "itkImageFileReader.h"
+#include "itkImageFileWriter.h"
+#include "itkTestingMacros.h"
+
+#include "itkRGBPixel.h"
+#include "itkVectorImage.h"
+
+namespace
+{
+template <typename TImageType>
+int
+ReadWrite(const std::string & inputImage, const std::string & outputImage)
+{
+  auto image = itk::ReadImage<TImageType>(inputImage);
+  ITK_TRY_EXPECT_NO_EXCEPTION(itk::WriteImage(image, outputImage));
+  return EXIT_SUCCESS;
+}
+
+template <unsigned int Dimension>
+int
+internalMain(const std::string & inputImage, const std::string & outputImage, itk::ImageIOBase::Pointer imageIO)
+{
+  const unsigned int numberOfComponents = imageIO->GetNumberOfComponents();
+  using IOPixelType = itk::IOPixelEnum;
+  const IOPixelType pixelType = imageIO->GetPixelType();
+
+  switch (pixelType)
+  {
+    case IOPixelType::SCALAR:
+      ITK_TEST_EXPECT_EQUAL(numberOfComponents, 1);
+      return ReadWrite<itk::Image<float, Dimension>>(inputImage, outputImage);
+
+    case IOPixelType::RGB:
+      ITK_TEST_EXPECT_EQUAL(numberOfComponents, 3);
+      return ReadWrite<itk::Image<itk::RGBPixel<unsigned char>, Dimension>>(inputImage, outputImage);
+
+    case IOPixelType::VECTOR:
+      return ReadWrite<itk::VectorImage<float, Dimension>>(inputImage, outputImage);
+
+    default:
+      std::cerr << "Test does not support pixel type of " << itk::ImageIOBase::GetPixelTypeAsString(pixelType)
+                << std::endl;
+      return EXIT_FAILURE;
+  }
+}
+
+} // namespace
+
+int
+itkVTIImageIOReadWriteTest(int argc, char * argv[])
+{
+  if (argc < 3)
+  {
+    std::cerr << "Usage: " << itkNameOfTestExecutableMacro(argv);
+    std::cerr << " InputImage OutputImage" << std::endl;
+    return EXIT_FAILURE;
+  }
+  const char * inputImage = argv[1];
+  const char * outputImage = argv[2];
+
+  using ReaderType = itk::ImageFileReader<itk::Image<float, 3>>;
+  auto reader = ReaderType::New();
+  reader->SetFileName(inputImage);
+  ITK_TRY_EXPECT_NO_EXCEPTION(reader->UpdateOutputInformation());
+
+  auto imageIO = reader->GetImageIO();
+  imageIO->SetFileName(inputImage);
+
+  ITK_TRY_EXPECT_NO_EXCEPTION(imageIO->ReadImageInformation());
+
+  std::cout << imageIO << std::endl;
+
+  const unsigned int dimension = imageIO->GetNumberOfDimensions();
+
+  switch (dimension)
+  {
+    case 2:
+      return internalMain<2>(inputImage, outputImage, imageIO);
+    case 3:
+      return internalMain<3>(inputImage, outputImage, imageIO);
+    default:
+      std::cerr << "Test only supports dimensions 2 and 3. Detected dimension " << dimension << std::endl;
+      return EXIT_FAILURE;
+  }
+}