itkPhysicsBasedNonRigidRegistrationMethod.h

Go to the documentation of this file.

/*=========================================================================
 *
 *  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 itkPhysicsBasedNonRigidRegistrationMethod_h
#define itkPhysicsBasedNonRigidRegistrationMethod_h


#include "itkMaskFeaturePointSelectionFilter.h"
#include "itkBlockMatchingImageFilter.h"
#include "itkFEMScatteredDataPointSetToImageFilter.h"
#include "itkConceptChecking.h"


namespace itk
{
namespace fem
{

template <typename TFixedImage, typename TMovingImage, typename TMaskImage, typename TMesh, typename TDeformationField>
class ITK_TEMPLATE_EXPORT PhysicsBasedNonRigidRegistrationMethod : public ImageToImageFilter<TMovingImage, TDeformationField>
{
public:
  typedef PhysicsBasedNonRigidRegistrationMethod               Self;
  typedef ImageToImageFilter<TMovingImage, TDeformationField>  Superclass;
  typedef SmartPointer<Self>                                   Pointer;
  typedef SmartPointer<const Self>                             ConstPointer;

  itkNewMacro(Self);

  itkTypeMacro( PhysicsBasedNonRigidRegistrationMethod, ImageToImageFilter );

  typedef TMovingImage                       MovingImageType;
  typedef TFixedImage                        FixedImageType;
  typedef TMaskImage                         MaskImageType;
  typedef TMesh                              MeshType;
  typedef TDeformationField                  DeformationFieldType;

  itkStaticConstMacro(ImageDimension, unsigned int, FixedImageType::ImageDimension);

  typedef ImageRegion< ImageDimension >  ImageRegionType;
  typedef Size< ImageDimension >         ImageSizeType;
  typedef Index< ImageDimension >        ImageIndexType;

  typedef MaskFeaturePointSelectionFilter< MovingImageType, MaskImageType >  FeatureSelectionFilterType;
  typedef BlockMatchingImageFilter< FixedImageType, MovingImageType >        BlockMatchingFilterType;
  typedef FEMScatteredDataPointSetToImageFilter<
    typename BlockMatchingFilterType::DisplacementsType,
    MeshType,
    DeformationFieldType,
    typename BlockMatchingFilterType::SimilaritiesType,
    typename FeatureSelectionFilterType::FeaturePointsType // tensors are optional pixel values of feature points pointset
  >  FEMFilterType;

  itkSetMacro(SelectFraction, double);
  itkGetConstMacro(SelectFraction, double);

  itkSetMacro(NonConnectivity, unsigned int);
  itkGetConstMacro(NonConnectivity, unsigned int);

  itkSetMacro(BlockRadius, ImageSizeType);
  itkGetConstReferenceMacro(BlockRadius, ImageSizeType);

  itkSetMacro(SearchRadius, ImageSizeType);
  itkGetConstReferenceMacro(SearchRadius, ImageSizeType);

  itkSetMacro(ApproximationSteps, unsigned int);
  itkGetMacro(ApproximationSteps, unsigned int);

  itkSetMacro(OutlierRejectionSteps, unsigned int);
  itkGetMacro(OutlierRejectionSteps, unsigned int);

  itkSetInputMacro(FixedImage, FixedImageType);
  itkGetInputMacro(FixedImage, FixedImageType);

  itkSetInputMacro(MovingImage, MovingImageType);
  itkGetInputMacro(MovingImage, MovingImageType);

  itkSetInputMacro(MaskImage, MaskImageType);
  itkGetInputMacro(MaskImage, MaskImageType);

  itkSetInputMacro(Mesh, MeshType);
  itkGetInputMacro(Mesh, MeshType);

  itkGetConstObjectMacro(FEMFilter, FEMFilterType);

#ifdef ITK_USE_CONCEPT_CHECKING
  // Begin concept checking
  /* Currently only the 3D implementation is available due to a narrow
     definition of the filter in the original proposal
     and lack of available resources. */
  itkConceptMacro( FixedImageDimensionShouldBe3,
                   ( Concept::SameDimension< TFixedImage::ImageDimension, 3u > ) );
  itkConceptMacro( MovingImageDimensionShouldBe3,
                   ( Concept::SameDimension< TMovingImage::ImageDimension, 3u > ) );
  itkConceptMacro( MaskImageDimensionShouldBe3,
                   ( Concept::SameDimension< TMaskImage::ImageDimension, 3u > ) );
  itkConceptMacro( MeshDimensionShouldBe3,
                   ( Concept::SameDimension< TMesh::PointType::PointDimension, 3u > ) );
  itkConceptMacro( DeformationFieldImageDimensionShouldBe3,
                   ( Concept::SameDimension< TDeformationField::ImageDimension, 3u > ) );
  // End concept checking
#endif

protected:
  PhysicsBasedNonRigidRegistrationMethod();
  virtual ~PhysicsBasedNonRigidRegistrationMethod() ITK_OVERRIDE;
  virtual void PrintSelf( std::ostream & os, Indent indent ) const ITK_OVERRIDE;
  virtual void GenerateData() ITK_OVERRIDE;

private:
  ITK_DISALLOW_COPY_AND_ASSIGN(PhysicsBasedNonRigidRegistrationMethod);

  double         m_SelectFraction;
  unsigned int   m_NonConnectivity;
  ImageSizeType  m_BlockRadius;
  ImageSizeType  m_SearchRadius;
  unsigned int   m_ApproximationSteps;
  unsigned int   m_OutlierRejectionSteps;

  typename FeatureSelectionFilterType::Pointer m_FeatureSelectionFilter;
  typename BlockMatchingFilterType::Pointer    m_BlockMatchingFilter;
  typename FEMFilterType::Pointer              m_FEMFilter;
};

}
}  // end namespace itk::fem

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

#endif