itkSegmentationLevelSetFunction.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 itkSegmentationLevelSetFunction_h
#define itkSegmentationLevelSetFunction_h

#include "itkLevelSetFunction.h"
#include "itkLinearInterpolateImageFunction.h"
#include "itkVectorLinearInterpolateImageFunction.h"
#include "itkVectorCastImageFilter.h"

namespace itk
{
template< typename TImageType, typename TFeatureImageType = TImageType >
class ITK_TEMPLATE_EXPORT SegmentationLevelSetFunction:
  public LevelSetFunction< TImageType >
{
public:
  ITK_DISALLOW_COPY_AND_ASSIGN(SegmentationLevelSetFunction);

  using Self = SegmentationLevelSetFunction;
  using Superclass = LevelSetFunction< TImageType >;
  using Pointer = SmartPointer< Self >;
  using ConstPointer = SmartPointer< const Self >;

  itkTypeMacro(SegmentationLevelSetFunction, LevelSetFunction);

  using ImageType = typename Superclass::ImageType;
  using RadiusType = typename Superclass::RadiusType;
  using PixelRealType = typename Superclass::PixelRealType;
  using FeatureImageType = TFeatureImageType;
  using FloatOffsetType = typename Superclass::FloatOffsetType;
  using ScalarValueType = typename Superclass::ScalarValueType;
  using NeighborhoodType = typename Superclass::NeighborhoodType;
  using FeatureScalarType = typename FeatureImageType::PixelType;
  using IndexType = typename ImageType::IndexType;
  using VectorType = typename Superclass::VectorType;
  using GlobalDataStruct = typename Superclass::GlobalDataStruct;

  static constexpr unsigned int ImageDimension = Superclass::ImageDimension;

  using VectorImageType = Image< VectorType, Self::ImageDimension >;

  using InterpolatorType = LinearInterpolateImageFunction< ImageType >;

  using VectorInterpolatorType = VectorLinearInterpolateImageFunction< VectorImageType >;

  using ContinuousIndexType = typename InterpolatorType::ContinuousIndexType;

  virtual const FeatureImageType * GetFeatureImage() const
  { return m_FeatureImage.GetPointer(); }
  virtual void SetFeatureImage(const FeatureImageType *f)
  {    m_FeatureImage = f;  }

  virtual ImageType * GetSpeedImage()
  { return m_SpeedImage.GetPointer(); }
  void SetSpeedImage(ImageType *s);

  virtual VectorImageType * GetAdvectionImage() const
  { return m_AdvectionImage.GetPointer(); }
  void SetAdvectionImage(VectorImageType *s);

  void Initialize(const RadiusType & r) override;

  virtual void CalculateSpeedImage() {}

  virtual void CalculateAdvectionImage() {}

  virtual void AllocateSpeedImage();

  virtual void AllocateAdvectionImage();

  virtual void ReverseExpansionDirection();

protected:
  typename FeatureImageType::ConstPointer m_FeatureImage;

  typename ImageType::Pointer m_SpeedImage;

  typename VectorImageType::Pointer m_AdvectionImage;

  Functor::VectorCast< typename VectorInterpolatorType::OutputType,
                       VectorType > m_VectorCast;

  ScalarValueType PropagationSpeed(const NeighborhoodType &,
                                           const FloatOffsetType &, GlobalDataStruct *gd) const override;

  VectorType AdvectionField(const NeighborhoodType &,
                                    const FloatOffsetType &, GlobalDataStruct *gd) const override;

  ~SegmentationLevelSetFunction() override = default;
  SegmentationLevelSetFunction()
  {
    m_SpeedImage = ImageType::New();
    m_AdvectionImage = VectorImageType::New();
    m_Interpolator = InterpolatorType::New();
    m_VectorInterpolator = VectorInterpolatorType::New();
  }

  typename InterpolatorType::Pointer m_Interpolator;

  typename VectorInterpolatorType::Pointer m_VectorInterpolator;
};
} // end namespace

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

#endif