itkSegmentationLevelSetFunction.h

Go to the documentation of this file.

/*=========================================================================
 *
 *  Copyright NumFOCUS
 *
 *  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
 *
 *         https://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 "itkCastImageFilter.h"
 
namespace itk
{
 
template <typename TImageType, typename TFeatureImageType = TImageType>
class ITK_TEMPLATE_EXPORT SegmentationLevelSetFunction : public LevelSetFunction<TImageType>
{
public:
  ITK_DISALLOW_COPY_AND_MOVE(SegmentationLevelSetFunction);

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

  itkOverrideGetNameOfClassMacro(SegmentationLevelSetFunction);

  using typename Superclass::ImageType;
  using typename Superclass::RadiusType;
  using typename Superclass::PixelRealType;
  using FeatureImageType = TFeatureImageType;
  using typename Superclass::FloatOffsetType;
  using typename Superclass::ScalarValueType;
  using typename Superclass::NeighborhoodType;
  using FeatureScalarType = typename FeatureImageType::PixelType;
  using IndexType = typename ImageType::IndexType;
  using typename Superclass::VectorType;
  using 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{};

  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{};
};
} // namespace itk
 
#ifndef ITK_MANUAL_INSTANTIATION
#  include "itkSegmentationLevelSetFunction.hxx"
#endif
 
#endif