/*=========================================================================
 *
 *  Copyright Insight Software Consortium
 *
 *  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
 *
 *         http://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.
 *
 *=========================================================================*/
#ifndef itkFastMarchingUpwindGradientImageFilterBase_h
#define itkFastMarchingUpwindGradientImageFilterBase_h

#include "itkFastMarchingImageFilterBase.h"
#include "itkImage.h"

namespace itk
{
/** \class FastMarchingUpwindGradientImageFilterBase
 *
 * \brief Generates the upwind gradient field of fast marching arrival times.
 *
 * This filter adds some extra functionality to its base class. While the
 * solution T(x) of the Eikonal equation is being generated by the base class
 * with the fast marching method, the filter generates the upwind gradient
 * vectors of T(x), storing them in an image.
 *
 * Since the Eikonal equation generates the arrival times of a wave travelling
 * at a given speed, the generated gradient vectors can be interpreted as the
 * slowness (1/velocity) vectors of the front (the quantity inside the modulus
 * operator in the Eikonal equation).
 *
 * Gradient vectors are computed using upwind finite differences, that is,
 * information only propagates from points where the wavefront has already
 * passed. This is consistent with how the fast marching method works.
 *
 * \author Luca Antiga Ph.D.  Biomedical Technologies Laboratory,
 *                            Bioengineering Department, Mario Negri Institute, Italy.
 *
 * \ingroup ITKFastMarching
 */
template< typename TInput, typename TOutput >
class FastMarchingUpwindGradientImageFilterBase:
  public FastMarchingImageFilterBase< TInput, TOutput >
{
public:
  /** Standard class typdedefs. */
  typedef FastMarchingUpwindGradientImageFilterBase       Self;
  typedef FastMarchingImageFilterBase< TInput, TOutput >  Superclass;
  typedef SmartPointer< Self >                            Pointer;
  typedef SmartPointer< const Self >                      ConstPointer;
  typedef typename Superclass::Traits                     Traits;

  /** Method for creation through the object factory. */
  itkNewMacro(Self);

  /** Run-time type information (and related methods). */
  itkTypeMacro(FastMarchingUpwindGradientImageFilterBase,
               FastMarchingImageFilterBase);

  /** The dimension of the level set. */
  itkStaticConstMacro(ImageDimension, unsigned int,
                      Superclass::ImageDimension );

  typedef typename Superclass::NodeType           NodeType;
  typedef typename Superclass::OutputImageType    OutputImageType;
  typedef typename Superclass::OutputPixelType    OutputPixelType;
  typedef typename Superclass::OutputSpacingType  OutputSpacingType;

  /** GradientPixel typedef support. */
  typedef CovariantVector< OutputPixelType,
                           ImageDimension > GradientPixelType;

  /** GradientImage typedef support. */
  typedef Image< GradientPixelType,
                 ImageDimension > GradientImageType;

  /** GradientImagePointer typedef support. */
  typedef typename GradientImageType::Pointer GradientImagePointer;

  /** Get the gradient image. */
  GradientImageType* GetGradientImage();

protected:
  FastMarchingUpwindGradientImageFilterBase();
  ~FastMarchingUpwindGradientImageFilterBase(){}

  void PrintSelf(std::ostream & os, Indent indent) const ITK_OVERRIDE;

  virtual void InitializeOutput( OutputImageType* oImage ) ITK_OVERRIDE;

  virtual void UpdateNeighbors( OutputImageType* oImage,
                               const NodeType& iNode ) ITK_OVERRIDE;

  virtual void ComputeGradient(OutputImageType* oImage,
                               const NodeType& iNode );

private:
  FastMarchingUpwindGradientImageFilterBase(const Self &) ITK_DELETE_FUNCTION;
  void operator=(const Self &) ITK_DELETE_FUNCTION;
};

/* this class was made in the case where isotropic and anisotropic fast
    marching would be implemented in this effort. As of now, we focus only
    in the isotropic case. Let's keep it for reference (just as a reminder).
*/
//template< unsigned int VDimension,
//         typename TInputPixel,
//         typename TOutputPixel >
//class IsotropicFastMarchingUpwindGradientImageFilterBase:
//    public FastMarchingUpwindGradientImageFilterBase< VDimension, TInputPixel,
//    TOutputPixel,
//    FastMarchingImageFilterBase< VDimension, TInputPixel, TOutputPixel >
//    >
//  {
//public:
//  typedef FastMarchingImageFilterBase< VDimension, TInputPixel,
//    TOutputPixel > GrandParentClassType;

//  /** Standard class typdedefs. */
//  typedef IsotropicFastMarchingUpwindGradientImageFilterBase             Self;
//  typedef FastMarchingUpwindGradientImageFilterBase< VDimension, TInputPixel,
//    TOutputPixel, GrandParentClassType > Superclass;
//  typedef SmartPointer< Self >                              Pointer;
//  typedef SmartPointer< const Self >                        ConstPointer;

//  /** Method for creation through the object factory. */
//  itkNewMacro(Self);

//  /** Run-time type information (and related methods). */
//  itkTypeMacro(IsotropicFastMarchingUpwindGradientImageFilterBase,
//               FastMarchingUpwindGradientImageFilterBase );

//  /** The dimension of the level set. */
//  itkStaticConstMacro(ImageDimension, unsigned int,
//                      Superclass::ImageDimension );

//  typedef typename Superclass::NodeType        NodeType;
//  typedef typename Superclass::OutputImageType OutputImageType;
//  typedef typename Superclass::OutputPixelType OutputPixelType;
//  typedef typename Superclass::OutputSpacingType OutputSpacingType;

//  /** GradientPixel typedef support. */
//  typedef CovariantVector< OutputPixelType,
//                           ImageDimension > GradientPixelType;

//  /** GradientImage typedef support. */
//  typedef Image< GradientPixelType,
//                 ImageDimension > GradientImageType;

//  /** GradientImagePointer typedef support. */
//  typedef typename GradientImageType::Pointer GradientImagePointer;

//protected:
//  IsotropicFastMarchingUpwindGradientImageFilterBase() : Superclass() {}
//  ~IsotropicFastMarchingUpwindGradientImageFilterBase() {}

//private:
//  void operator = ( const Self& );
//  IsotropicFastMarchingUpwindGradientImageFilterBase( const Self& );
//  };
} // namespace itk

#ifndef ITK_MANUAL_INSTANTIATION
#include "itkFastMarchingUpwindGradientImageFilterBase.hxx"
#endif

#endif
