itkHighResRelaxStreamlinesFilter.h

Go to the documentation of this file.

/*=============================================================================

  NifTK: A software platform for medical image computing.

  Copyright (c) University College London (UCL). All rights reserved.

  This software is distributed WITHOUT ANY WARRANTY; without even
  the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
  PURPOSE.

  See LICENSE.txt in the top level directory for details.

=============================================================================*/

#ifndef itkHighResRelaxStreamlinesFilter_h
#define itkHighResRelaxStreamlinesFilter_h

#include "itkLagrangianInitializedRelaxStreamlinesFilter.h"
#include "itkHighResLaplacianSolverImageFilter.h"
#include <itkVectorInterpolateImageFunction.h>

namespace itk {
template < class TImageType, typename TScalarType, unsigned int NDimensions> 
class ITK_EXPORT HighResRelaxStreamlinesFilter :
  public LagrangianInitializedRelaxStreamlinesFilter< TImageType, TScalarType, NDimensions>
{
public:

  typedef HighResRelaxStreamlinesFilter                                                      Self;
  typedef LagrangianInitializedRelaxStreamlinesFilter<TImageType, TScalarType, NDimensions>  Superclass;
  typedef SmartPointer<Self>                                                                 Pointer;
  typedef SmartPointer<const Self>                                                           ConstPointer;
  
  itkNewMacro(Self);

  itkTypeMacro(RelaxStreamlinesFilter, LagrangianInitializedRelaxStreamlinesFilter);

  typedef typename Superclass::InputVectorImagePixelType      InputVectorImagePixelType;
  typedef typename Superclass::InputVectorImageType           InputVectorImageType;
  typedef typename Superclass::InputVectorImagePointer        InputVectorImagePointer;
  typedef typename Superclass::InputVectorImageConstPointer   InputVectorImageConstPointer;
  typedef typename Superclass::InputVectorImageIndexType      InputVectorImageIndexType; 
  typedef typename Superclass::InputScalarImagePixelType      InputScalarImagePixelType;
  typedef typename Superclass::InputScalarImageType           InputScalarImageType;
  typedef typename Superclass::InputScalarImagePointType      InputScalarImagePointType;
  typedef typename Superclass::InputScalarImagePointer        InputScalarImagePointer;
  typedef typename Superclass::InputScalarImageIndexType      InputScalarImageIndexType;
  typedef typename Superclass::InputScalarImageConstPointer   InputScalarImageConstPointer;
  typedef typename Superclass::InputScalarImageRegionType     InputScalarImageRegionType;
  typedef typename Superclass::InputScalarImageSpacingType    InputScalarImageSpacingType;
  typedef typename InputScalarImageType::SizeType             InputScalarImageSizeType;
  typedef typename InputScalarImageType::PointType            InputScalarImageOriginType;
  typedef typename Superclass::OutputImageType                OutputImageType;
  typedef typename Superclass::OutputImagePixelType           OutputImagePixelType;
  typedef typename Superclass::OutputImagePointer             OutputImagePointer;
  typedef typename Superclass::OutputImageConstPointer        OutputImageConstPointer;
  typedef typename Superclass::OutputImageIndexType           OutputImageIndexType;
  typedef typename Superclass::OutputImageSpacingType         OutputImageSpacingType;
  typedef typename OutputImageType::RegionType                OutputImageRegionType;
  typedef typename OutputImageType::SizeType                  OutputImageSizeType;
  typedef typename OutputImageType::DirectionType             OutputImageDirectionType;
  typedef typename OutputImageType::PointType                 OutputImageOriginType;  
  typedef typename Superclass::VectorInterpolatorType         VectorInterpolatorType;
  typedef typename Superclass::VectorInterpolatorPointer      VectorInterpolatorPointer;
  typedef typename Superclass::VectorInterpolatorPointType    VectorInterpolatorPointType;
  typedef typename Superclass::ScalarInterpolatorType         ScalarInterpolatorType;
  typedef typename Superclass::ScalarInterpolatorPointer      ScalarInterpolatorPointer;
  typedef typename Superclass::ScalarInterpolatorPointType    ScalarInterpolatorPointType;
  typedef typename HighResLaplacianSolverImageFilter<TImageType, TScalarType>::MapType MapType;
  typedef typename HighResLaplacianSolverImageFilter<TImageType, TScalarType>::IteratorType MapIteratorType;
  typedef typename HighResLaplacianSolverImageFilter<TImageType, TScalarType>::PairType MapPairType;
  typedef typename HighResLaplacianSolverImageFilter<TImageType, TScalarType>::FiniteDifferenceVoxelType FiniteDifferenceVoxelType;
  typedef ContinuousIndex<TScalarType, TImageType::ImageDimension> ContinuousIndexType;
  typedef Point<TScalarType, TImageType::ImageDimension> PointType;
  
  void SetHighResLaplacianMap(MapType *map) { m_LaplacianMap = map; }
  
  itkSetMacro(VoxelMultiplicationFactor, int);
  itkGetMacro(VoxelMultiplicationFactor, int);

protected:
  HighResRelaxStreamlinesFilter();
  ~HighResRelaxStreamlinesFilter() {};
  void PrintSelf(std::ostream& os, Indent indent) const;

  virtual void IntializeBoundaries(
      InputScalarImageType* gmpvImage,
      InputVectorImageType* vectorImage
      );
  
  virtual void SolvePDE(
      int boundaryNumber,
      InputScalarImageSpacingType& virtualSpacing,
      InputScalarImageType* scalarImage,
      InputScalarImageType* gmpvImage,
      InputVectorImageType* vectorImage
      );

private:
  
  HighResRelaxStreamlinesFilter(const Self&); 
  void operator=(const Self&); 

  // The main filter method. Note, single threaded.
  virtual void GenerateData();

  MapType *m_LaplacianMap;
  
  MapType* m_L0L1;

  int m_VoxelMultiplicationFactor;
  
  VectorInterpolatorPointer m_VectorInterpolator;
};

} // end namespace

#ifndef ITK_MANUAL_INSTANTIATION
#include "itkHighResRelaxStreamlinesFilter.txx"
#endif

#endif