/*=========================================================================
 *
 *  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 "itkSparseFrequencyContainer2.h"


int
itkSparseFrequencyContainer2Test(int, char *[])
{
  std::cout << "SparseFrequencyContainer2 Test \n \n";

  using SparseFrequencyContainer2Type = itk::Statistics::SparseFrequencyContainer2;


  auto container = SparseFrequencyContainer2Type::New();

  using AbsoluteFrequencyType = SparseFrequencyContainer2Type::AbsoluteFrequencyType;

  constexpr unsigned int numberOfBins = 1250;

  container->Initialize(numberOfBins);


  // Test the SetFrequency() / GetFrequency() methods
  {
    std::cout << "Testing Set/Get Frequency methods...";
    for (unsigned int bin = 0; bin < numberOfBins; ++bin)
    {
      // Compute any value as frequency just to test the SetFrequency() method
      const auto frequency = static_cast<AbsoluteFrequencyType>(bin * bin);
      container->SetFrequency(bin, frequency);
    }

    for (unsigned int bin = 0; bin < numberOfBins; ++bin)
    {
      // Test if the values can be read back
      const auto                  frequency = static_cast<AbsoluteFrequencyType>(bin * bin);
      const AbsoluteFrequencyType stored = container->GetFrequency(bin);
      if (stored != frequency)
      {
        std::cout << "Failed !" << std::endl;
        std::cout << "Stored Frequency in bin " << bin << " doesn't match value" << std::endl;
        std::cout << "Value is = " << stored << " value should be " << frequency << std::endl;
        return EXIT_FAILURE;
      }
    }
    std::cout << " PASSED !" << std::endl;

  } // end of SetFrequency() / GetFrequency() test

  // Set all the bins to zero and check the values
  container->SetToZero();
  for (unsigned int bin = 0; bin < numberOfBins; ++bin)
  {
    if (container->GetFrequency(bin) != itk::NumericTraits<AbsoluteFrequencyType>::ZeroValue())
    {
      std::cout << "Failed !" << std::endl;
      std::cout << "Stored Frequency in bin is not zero after SetToZero() method invocation" << std::endl;
      return EXIT_FAILURE;
    }
  }

  // Test the IncreaseFrequency() method
  {
    std::cout << "Testing IncreaseFrequency method...";
    // Try not to depend on previous tests....So, we initialize the histogram again.
    for (unsigned int bin = 0; bin < numberOfBins; ++bin)
    {
      // Compute any value as frequency just to test the SetFrequency() method
      const auto frequency = static_cast<AbsoluteFrequencyType>(bin * bin);
      container->SetFrequency(bin, frequency);
    }

    // Now increment by a number (we use "bin", but any other will do it...)
    for (unsigned int bin = 0; bin < numberOfBins; ++bin)
    {
      // Compute any value as frequency just to test the IncreaseFrequency() method
      const auto frequency = static_cast<AbsoluteFrequencyType>(bin);
      container->IncreaseFrequency(bin, frequency);
    }


    // Test if the values can be read back
    for (unsigned int bin = 0; bin < numberOfBins; ++bin)
    {
      const auto                  frequency = static_cast<AbsoluteFrequencyType>(bin * bin + bin);
      const AbsoluteFrequencyType stored = container->GetFrequency(bin);
      if (stored != frequency)
      {
        std::cout << "Failed !" << std::endl;
        std::cout << "Stored Frequency in bin " << bin << " doesn't match value" << std::endl;
        std::cout << "Value is = " << stored << " value should be " << frequency << std::endl;
        return EXIT_FAILURE;
      }
    }
    std::cout << " PASSED !" << std::endl;
  } // end of SetFrequency() / GetFrequency() test

  std::cout << "Test passed." << std::endl;
  return EXIT_SUCCESS;
}
