/*=========================================================================
 *
 *  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 <iostream>
#include "itkImageRegionIterator.h"
#include "itkShrinkImageFilter.h"
#include "itkImportImageFilter.h"
#include "itkTestingMacros.h"

int
itkImportImageTest(int, char *[])
{
  // Create a C-array to hold an image
  auto * rawImage = new short[8 * 12];
  for (unsigned int i = 0; i < 8 * 12; ++i)
  {
    rawImage[i] = i;
  }
  // typdefs to simplify the syntax
  constexpr unsigned int Dimension = 2;
  using PixelType = short;

  using ImportImageFilter = itk::ImportImageFilter<PixelType, Dimension>;
  using ShortImage = itk::Image<PixelType, Dimension>;

  // Create an ImportImageFilter filter
  auto basicImport = ImportImageFilter::New();

  ITK_EXERCISE_BASIC_OBJECT_METHODS(basicImport, ImportImageFilter, ImageSource);

  ShortImage::Pointer                    image;
  itk::ImageRegion<Dimension>            region;
  itk::ImageRegion<Dimension>::IndexType index = { { 0, 0 } };
  itk::ImageRegion<Dimension>::SizeType  size = { { 8, 12 } };
  region.SetIndex(index);
  region.SetSize(size);
  // local scope to make sure that imported data is not deleted with ImportImageFilter
  // but with the ImportImageContainer is creates.
  {
    // Create an ImportImageFilter filter
    ImportImageFilter::Pointer import;
    import = ImportImageFilter::New();

    // Test the SetVectorMacros and GetVectorMacros
    const itk::SpacePrecisionType data[2] = { 1.0, 1.0 };
    import->SetSpacing(data);

    const float data2[2] = { 1.0, 1.0 };
    import->SetSpacing(data2);

    const itk::SpacePrecisionType * spacingValue = import->GetSpacing().GetDataPointer();
    std::cout << "import->GetSpacing(): " << spacingValue << std::endl;

    const double data3[2] = { 1.0, 1.0 };
    import->SetOrigin(data3);

    const float data4[2] = { 1.0, 1.0 };
    import->SetOrigin(data4);

    const itk::SpacePrecisionType * originValue = import->GetOrigin().GetDataPointer();
    std::cout << "import->GetOrigin(): " << originValue << std::endl;

    import->SetRegion(region);
    import->SetImportPointer(rawImage, 8 * 12, true);
    import->Update();
    image = import->GetOutput();
  }
  // Create another filter
  itk::ShrinkImageFilter<ImportImageFilter::OutputImageType, ShortImage>::Pointer shrink =
    itk::ShrinkImageFilter<ImportImageFilter::OutputImageType, ShortImage>::New();

  shrink->SetInput(image);
  shrink->SetShrinkFactors(2); // Also tested with factors 3 and 4, with 12x12 image

  ITK_TRY_EXPECT_NO_EXCEPTION(shrink->Update());

  //
  // The rest of this code determines whether the shrink code produced
  // the image we expected.
  //
  ShortImage::RegionType requestedRegion;
  requestedRegion = shrink->GetOutput()->GetRequestedRegion();

  itk::ImageRegionIterator<ShortImage> iterator2(shrink->GetOutput(), requestedRegion);

  bool passed = true;
  for (; !iterator2.IsAtEnd(); ++iterator2)
  {
    std::cout << "Pixel " << iterator2.GetIndex() << " = " << iterator2.Get() << std::endl;
    if (iterator2.Get() !=
        itk::Math::RoundHalfIntegerUp<short>(static_cast<float>(
          (shrink->GetShrinkFactors()[0] * iterator2.GetIndex()[0] + shrink->GetShrinkFactors()[0] / 2) +
          (region.GetSize()[0] *
           ((shrink->GetShrinkFactors()[1] / 2) + (shrink->GetShrinkFactors()[0] * iterator2.GetIndex()[1]))))))
    {
      std::cout << " iterator2.GetIndex() Get() " << iterator2.GetIndex() << ' ' << iterator2.Get() << " compare value "
                << itk::Math::RoundHalfIntegerUp<short>(static_cast<float>(
                     (shrink->GetShrinkFactors()[0] * iterator2.GetIndex()[0] + shrink->GetShrinkFactors()[0] / 2) +
                     (region.GetSize()[0] * ((shrink->GetShrinkFactors()[1] / 2) +
                                             (shrink->GetShrinkFactors()[0] * iterator2.GetIndex()[1])))))
                << '\n';
      passed = false;
    }
  }

  if (passed)
  {
    std::cout << "ImportImageFilter test passed." << std::endl;
    return EXIT_SUCCESS;
  }
  else
  {
    std::cout << "ImportImageFilter test failed." << std::endl;
    return EXIT_FAILURE;
  }
}
