/*=========================================================================
 *
 *  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.
 *
 *=========================================================================*/
/*=========================================================================
 *
 *  Portions of this file are subject to the VTK Toolkit Version 3 copyright.
 *
 *  Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
 *
 *  For complete copyright, license and disclaimer of warranty information
 *  please refer to the NOTICE file at the top of the ITK source tree.
 *
 *=========================================================================*/
#ifndef itkThresholdImageFilter_hxx
#define itkThresholdImageFilter_hxx

#include "itkImageScanlineIterator.h"
#include "itkNumericTraits.h"
#include "itkObjectFactory.h"
#include "itkTotalProgressReporter.h"
#include "itkMacro.h"
#include "itkMath.h"

namespace itk
{

template <typename TImage>
ThresholdImageFilter<TImage>::ThresholdImageFilter()
  : m_OutsideValue(PixelType{})
  , m_Lower(NumericTraits<PixelType>::NonpositiveMin())
  , m_Upper(NumericTraits<PixelType>::max())
{
  this->InPlaceOff();
  this->DynamicMultiThreadingOn();
  this->ThreaderUpdateProgressOff();
}

template <typename TImage>
void
ThresholdImageFilter<TImage>::ThresholdAbove(const PixelType & threshold)
{
  if (Math::NotExactlyEquals(m_Upper, threshold) || m_Lower > NumericTraits<PixelType>::NonpositiveMin())
  {
    m_Lower = NumericTraits<PixelType>::NonpositiveMin();
    m_Upper = threshold;
    this->Modified();
  }
}

template <typename TImage>
void
ThresholdImageFilter<TImage>::ThresholdBelow(const PixelType & threshold)
{
  if (Math::NotExactlyEquals(m_Lower, threshold) || m_Upper < NumericTraits<PixelType>::max())
  {
    m_Lower = threshold;
    m_Upper = NumericTraits<PixelType>::max();
    this->Modified();
  }
}

template <typename TImage>
void
ThresholdImageFilter<TImage>::ThresholdOutside(const PixelType & lower, const PixelType & upper)
{
  if (lower > upper)
  {
    itkExceptionMacro("Lower threshold cannot be greater than upper threshold.");
  }

  if (Math::NotExactlyEquals(m_Lower, lower) || Math::NotExactlyEquals(m_Upper, upper))
  {
    m_Lower = lower;
    m_Upper = upper;
    this->Modified();
  }
}

template <typename TImage>
void
ThresholdImageFilter<TImage>::DynamicThreadedGenerateData(const OutputImageRegionType & outputRegionForThread)
{
  // Get the input and output pointers
  InputImagePointer  inputPtr = this->GetInput();
  OutputImagePointer outputPtr = this->GetOutput(0);

  TotalProgressReporter progress(this, outputPtr->GetRequestedRegion().GetNumberOfPixels());

  // Define/declare an iterator that will walk the output region for this
  // thread.
  ImageScanlineConstIterator inIt(inputPtr, outputRegionForThread);
  ImageScanlineIterator      outIt(outputPtr, outputRegionForThread);

  // Walk the regions; threshold each pixel
  while (!outIt.IsAtEnd())
  {
    while (!outIt.IsAtEndOfLine())
    {
      const PixelType value = inIt.Get();
      if (m_Lower <= value && value <= m_Upper)
      {
        // Pixel passes to output unchanged and is replaced by m_OutsideValue in
        // the inverse output image
        outIt.Set(inIt.Get());
      }
      else
      {
        outIt.Set(m_OutsideValue);
      }
      ++inIt;
      ++outIt;
    }
    inIt.NextLine();
    outIt.NextLine();
    progress.Completed(outputRegionForThread.GetSize()[0]);
  }
}

template <typename TImage>
void
ThresholdImageFilter<TImage>::PrintSelf(std::ostream & os, Indent indent) const
{
  Superclass::PrintSelf(os, indent);

  os << indent << "OutsideValue: " << static_cast<typename NumericTraits<PixelType>::PrintType>(m_OutsideValue)
     << std::endl;
  os << indent << "Lower: " << static_cast<typename NumericTraits<PixelType>::PrintType>(m_Lower) << std::endl;
  os << indent << "Upper: " << static_cast<typename NumericTraits<PixelType>::PrintType>(m_Upper) << std::endl;
}
} // end namespace itk

#endif
