itkLevelSetFunctionWithRefitTerm.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 itkLevelSetFunctionWithRefitTerm_h
#define itkLevelSetFunctionWithRefitTerm_h

#include "itkLevelSetFunction.h"
#include "itkSparseImage.h"
#include "itkNumericTraits.h"

namespace itk
{
template< typename TImageType, typename TSparseImageType >
class ITK_TEMPLATE_EXPORT LevelSetFunctionWithRefitTerm:
  public LevelSetFunction< TImageType >
{
public:

  typedef LevelSetFunctionWithRefitTerm  Self;
  typedef LevelSetFunction< TImageType > Superclass;
  typedef SmartPointer< Self >           Pointer;
  typedef SmartPointer< const Self >     ConstPointer;

  itkTypeMacro(LevelSetFunctionWithRefitTerm, LevelSetFunction);

  itkNewMacro (Self);

  typedef typename Superclass::ImageType              ImageType;
  typedef typename Superclass::FloatOffsetType        FloatOffsetType;
  typedef typename Superclass::ScalarValueType        ScalarValueType;
  typedef typename Superclass::GlobalDataStruct       GlobalDataStruct;
  typedef typename Superclass::NeighborhoodType       NeighborhoodType;
  typedef typename Superclass::NeighborhoodScalesType NeighborhoodScalesType;
  typedef typename Superclass::TimeStepType           TimeStepType;

  typedef typename NeighborhoodType::SizeValueType    NeighborhoodSizeValueType;

  typedef typename ImageType::IndexType IndexType;

  typedef TSparseImageType SparseImageType;

  typedef typename SparseImageType::NodeType NodeType;

  typedef typename NodeType::NodeDataType NormalVectorType;

  void SetRefitWeight(const ScalarValueType w)
  {
    m_RefitWeight = w;
  }

  void SetOtherPropagationWeight(const ScalarValueType w)
  {
    m_OtherPropagationWeight = w;
  }

  void SetSparseTargetImage(SparseImageType *im)
  { m_SparseTargetImage = im; }

  SparseImageType * GetSparseTargetImage() const
  { return m_SparseTargetImage; }

  virtual TimeStepType ComputeGlobalTimeStep(void *GlobalData) const ITK_OVERRIDE;

protected:
  ScalarValueType m_RefitWeight;

  ScalarValueType m_OtherPropagationWeight;

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

  ScalarValueType ComputeCurvature(const NeighborhoodType &) const;

  virtual ScalarValueType PropagationSpeed(const NeighborhoodType &,
                                           const FloatOffsetType &,
                                           GlobalDataStruct * = 0) const ITK_OVERRIDE;

  virtual ScalarValueType OtherPropagationSpeed(const NeighborhoodType &,
                                                const FloatOffsetType &,
                                                GlobalDataStruct * = 0) const
  {
    return NumericTraits< ScalarValueType >::ZeroValue();
  }

private:
  ITK_DISALLOW_COPY_AND_ASSIGN(LevelSetFunctionWithRefitTerm);

  typename SparseImageType::Pointer m_SparseTargetImage;

  ScalarValueType m_MinVectorNorm;

  static const NeighborhoodSizeValueType    m_NumVertex;
  static const ScalarValueType              m_DimConst;
};
} // end namespace itk

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

#endif