itkStatisticalShapePointPenalty.h

Go to the documentation of this file.

/*=========================================================================
 *
 *  Copyright UMC Utrecht and contributors
 *
 *  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 __itkStatisticalShapePointPenalty_h
#define __itkStatisticalShapePointPenalty_h
 
#include "itkSingleValuedPointSetToPointSetMetric.h"
 
#include "itkPoint.h"
#include "itkPointSet.h"
#include "itkImage.h"
#include "itkArray.h"
#include <itkVariableSizeMatrix.h>
 
#include <vnl/vnl_matrix.h>
#include <vnl/vnl_math.h>
#include <vnl/vnl_vector.h>
#include <vnl/algo/vnl_real_eigensystem.h>
#include <vnl/algo/vnl_symmetric_eigensystem.h>
//#include <vnl/algo/vnl_svd.h>
#include <vnl/algo/vnl_svd_economy.h>
 
#include <string>
 
namespace itk
{
template< class TFixedPointSet, class TMovingPointSet >
class StatisticalShapePointPenalty :
  public SingleValuedPointSetToPointSetMetric< TFixedPointSet, TMovingPointSet >
{
public:
 
  typedef StatisticalShapePointPenalty Self;
  typedef SingleValuedPointSetToPointSetMetric<
    TFixedPointSet, TMovingPointSet >               Superclass;
  typedef SmartPointer< Self >       Pointer;
  typedef SmartPointer< const Self > ConstPointer;
 
  itkNewMacro( Self );
 
  itkTypeMacro( StatisticalShapePointPenalty,
    SingleValuedPointSetToPointSetMetric );
 
  typedef typename Superclass::TransformType              TransformType;
  typedef typename Superclass::TransformPointer           TransformPointer;
  typedef typename Superclass::TransformParametersType    TransformParametersType;
  typedef typename Superclass::TransformJacobianType      TransformJacobianType;
  typedef typename Superclass::NonZeroJacobianIndicesType NonZeroJacobianIndicesType;
 
  typedef typename Superclass::MeasureType                MeasureType;
  typedef typename Superclass::DerivativeType             DerivativeType;
  typedef typename Superclass::DerivativeValueType        DerivativeValueType;
  typedef typename Superclass::FixedPointSetType          FixedPointSetType;
  typedef typename Superclass::MovingPointSetType         MovingPointSetType;
  typedef typename Superclass::FixedPointSetConstPointer  FixedPointSetConstPointer;
  typedef typename Superclass::MovingPointSetConstPointer MovingPointSetConstPointer;
 
  typedef typename Superclass::PointIterator     PointIterator;
  typedef typename Superclass::PointDataIterator PointDataIterator;
 
  typedef typename Superclass::InputPointType  InputPointType;
  typedef typename Superclass::OutputPointType OutputPointType;
 
  typedef typename OutputPointType::CoordRepType CoordRepType;
  typedef vnl_vector< CoordRepType >             VnlVectorType;
  typedef vnl_matrix< CoordRepType >             VnlMatrixType;
  //typedef itk::Array<VnlVectorType *> ProposalDerivativeType;
  typedef typename std::vector< VnlVectorType * > ProposalDerivativeType;
  //typedef typename vnl_vector<VnlVectorType *> ProposalDerivativeType; //Cannot be linked
  typedef vnl_svd_economy< CoordRepType > PCACovarianceType;
 
  void Initialize( void ) override;
 
  MeasureType GetValue( const TransformParametersType & parameters ) const override;
 
  void GetDerivative( const TransformParametersType & parameters,
    DerivativeType & Derivative ) const override;
 
  void GetValueAndDerivative( const TransformParametersType & parameters,
    MeasureType & Value, DerivativeType & Derivative ) const override;
 
  itkSetClampMacro( ShrinkageIntensity, MeasureType, 0.0, 1.0 );
  itkGetMacro( ShrinkageIntensity, MeasureType );
 
  itkSetMacro( ShrinkageIntensityNeedsUpdate, bool );
  itkBooleanMacro( ShrinkageIntensityNeedsUpdate );
 
  itkSetClampMacro( BaseVariance, MeasureType,
    -1.0, NumericTraits< MeasureType >::max() );
  itkGetMacro( BaseVariance, MeasureType );
 
  itkSetMacro( BaseVarianceNeedsUpdate, bool );
  itkBooleanMacro( BaseVarianceNeedsUpdate );
 
  itkSetClampMacro( CentroidXVariance, MeasureType,
    -1.0, NumericTraits< MeasureType >::max() );
  itkGetMacro( CentroidXVariance, MeasureType );
 
  itkSetClampMacro( CentroidYVariance, MeasureType,
    -1.0, NumericTraits< MeasureType >::max() );
  itkGetMacro( CentroidYVariance, MeasureType );
 
  itkSetClampMacro( CentroidZVariance, MeasureType,
    -1.0, NumericTraits< MeasureType >::max() );
  itkGetMacro( CentroidZVariance, MeasureType );
 
  itkSetClampMacro( SizeVariance, MeasureType,
    -1.0, NumericTraits< MeasureType >::max() );
  itkGetMacro( SizeVariance, MeasureType );
 
  itkSetMacro( VariancesNeedsUpdate, bool );
  itkBooleanMacro( VariancesNeedsUpdate );
 
  itkSetClampMacro( CutOffValue, MeasureType,
    0.0, NumericTraits< MeasureType >::max() );
  itkGetMacro( CutOffValue, MeasureType );
 
  itkSetClampMacro( CutOffSharpness, MeasureType,
    NumericTraits< MeasureType >::NonpositiveMin(), NumericTraits< MeasureType >::max() );
  itkGetMacro( CutOffSharpness, MeasureType );
 
  itkSetMacro( ShapeModelCalculation, int );
  itkGetConstReferenceMacro( ShapeModelCalculation, int );
 
  itkSetMacro( NormalizedShapeModel, bool );
  itkGetConstReferenceMacro( NormalizedShapeModel, bool );
  itkBooleanMacro( NormalizedShapeModel );
 
  itkSetConstObjectMacro( EigenVectors, vnl_matrix< double > );
  itkSetConstObjectMacro( EigenValues, vnl_vector< double > );
  itkSetConstObjectMacro( MeanVector, vnl_vector< double > );
 
  itkSetConstObjectMacro( CovarianceMatrix, vnl_matrix< double > );
 
protected:
 
  StatisticalShapePointPenalty();
  ~StatisticalShapePointPenalty() override;
 
  void PrintSelf( std::ostream & os, Indent indent ) const override;
 
private:
 
  StatisticalShapePointPenalty( const Self & );  // purposely not implemented
  void operator=( const Self & );                // purposely not implemented
 
  void FillProposalVector( const OutputPointType & fixedPoint,
    const unsigned int vertexindex ) const;
 
  void FillProposalDerivative( const OutputPointType & fixedPoint,
    const unsigned int vertexindex ) const;
 
  void UpdateCentroidAndAlignProposalVector(
    const unsigned int shapeLength ) const;
 
  void UpdateCentroidAndAlignProposalDerivative(
    const unsigned int shapeLength ) const;
 
  void UpdateL2( const unsigned int shapeLength ) const;
 
  void NormalizeProposalVector( const unsigned int shapeLength ) const;
 
  void UpdateL2AndNormalizeProposalDerivative( const unsigned int shapeLength ) const;
 
  void CalculateValue( MeasureType & value, VnlVectorType & differenceVector,
    VnlVectorType & centerrotated, VnlVectorType & eigrot ) const;
 
  void CalculateDerivative( DerivativeType & derivative, const MeasureType & value,
    const VnlVectorType & differenceVector, const VnlVectorType & centerrotated,
    const VnlVectorType & eigrot, const unsigned int shapeLength ) const;
 
  void CalculateCutOffValue( MeasureType & value ) const;
 
  void CalculateCutOffDerivative(
  typename DerivativeType::element_type & derivativeElement,
  const MeasureType &value ) const;
 
  const VnlVectorType * m_MeanVector;
  const VnlMatrixType * m_CovarianceMatrix;
  const VnlMatrixType * m_EigenVectors;
  const VnlVectorType * m_EigenValues;
 
  VnlMatrixType * m_InverseCovarianceMatrix;
 
  double m_CentroidXVariance;
  double m_CentroidXStd;
  double m_CentroidYVariance;
  double m_CentroidYStd;
  double m_CentroidZVariance;
  double m_CentroidZStd;
  double m_SizeVariance;
  double m_SizeStd;
 
  bool m_ShrinkageIntensityNeedsUpdate;
  bool m_BaseVarianceNeedsUpdate;
  bool m_VariancesNeedsUpdate;
 
  VnlVectorType * m_EigenValuesRegularized;
 
  mutable ProposalDerivativeType * m_ProposalDerivative;
  unsigned int                     m_ProposalLength;
  bool                             m_NormalizedShapeModel;
  int                              m_ShapeModelCalculation;
  double                           m_ShrinkageIntensity;
  double                           m_BaseVariance;
  double                           m_BaseStd;
  mutable VnlVectorType            m_ProposalVector;
  mutable VnlVectorType            m_MeanValues;
 
  double m_CutOffValue;
  double m_CutOffSharpness;
 
};
 
} // end namespace itk
 
#ifndef ITK_MANUAL_INSTANTIATION
#include "itkStatisticalShapePointPenalty.hxx"
#endif
 
#endif