itkUpdateWhitakerSparseLevelSet.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 itkUpdateWhitakerSparseLevelSet_h
#define itkUpdateWhitakerSparseLevelSet_h
 
#include "itkImage.h"
#include "itkDiscreteLevelSetImage.h"
#include "itkWhitakerSparseLevelSetImage.h"
#include "itkImageRegionIteratorWithIndex.h"
#include "itkShapedNeighborhoodIterator.h"
#include "itkNeighborhoodAlgorithm.h"
#include "itkLabelMapToLabelImageFilter.h"
#include "itkLabelImageToLabelMapFilter.h"
 
namespace itk
{
template <unsigned int VDimension, typename TLevelSetValueType, typename TEquationContainer>
class ITK_TEMPLATE_EXPORT UpdateWhitakerSparseLevelSet : public Object
{
public:
  ITK_DISALLOW_COPY_AND_MOVE(UpdateWhitakerSparseLevelSet);
 
  using Self = UpdateWhitakerSparseLevelSet;
  using Pointer = SmartPointer<Self>;
  using ConstPointer = SmartPointer<const Self>;
  using Superclass = Object;

  itkNewMacro(Self);

  itkOverrideGetNameOfClassMacro(UpdateWhitakerSparseLevelSet);
 
  static constexpr unsigned int ImageDimension = VDimension;
 
  using LevelSetOutputType = TLevelSetValueType;
 
  using LevelSetType = WhitakerSparseLevelSetImage<LevelSetOutputType, ImageDimension>;
  using LevelSetPointer = typename LevelSetType::Pointer;
  using LevelSetInputType = typename LevelSetType::InputType;
  using LevelSetOffsetType = typename LevelSetType::OffsetType;
 
  using LevelSetLabelMapType = typename LevelSetType::LabelMapType;
  using LevelSetLabelMapPointer = typename LevelSetType::LabelMapPointer;
 
  using LevelSetLabelObjectType = typename LevelSetType::LabelObjectType;
  using LevelSetLabelObjectPointer = typename LevelSetType::LabelObjectPointer;
  using LevelSetLabelObjectLengthType = typename LevelSetType::LabelObjectLengthType;
  using LevelSetLabelObjectLineType = typename LevelSetType::LabelObjectLineType;
 
  using LevelSetLayerIdType = typename LevelSetType::LayerIdType;
  using LevelSetLayerType = typename LevelSetType::LayerType;
  using LevelSetLayerIterator = typename LevelSetType::LayerIterator;
  using LevelSetLayerConstIterator = typename LevelSetType::LayerConstIterator;
  using LevelSetOutputRealType = typename LevelSetType::OutputRealType;
 
  using LevelSetLayerMapType = typename LevelSetType::LayerMapType;
  using LevelSetLayerMapIterator = typename LevelSetType::LayerMapIterator;
  using LevelSetLayerMapConstIterator = typename LevelSetType::LayerMapConstIterator;
 
  using EquationContainerType = TEquationContainer;
  using EquationContainerPointer = typename EquationContainerType::Pointer;
 
  using TermContainerType = typename EquationContainerType::TermContainerType;
  using TermContainerPointer = typename EquationContainerType::TermContainerPointer;
 
  using LabelImageType = Image<LevelSetLayerIdType, ImageDimension>;
  using LabelImagePointer = typename LabelImageType::Pointer;
 
  using LabelMapToLabelImageFilterType = LabelMapToLabelImageFilter<LevelSetLabelMapType, LabelImageType>;
  using LabelImageToLabelMapFilterType = LabelImageToLabelMapFilter<LabelImageType, LevelSetLabelMapType>;
 
  itkGetModifiableObjectMacro(OutputLevelSet, LevelSetType);

  void
  Update();

  itkSetObjectMacro(InputLevelSet, LevelSetType);
  itkGetModifiableObjectMacro(InputLevelSet, LevelSetType);

  itkSetMacro(TimeStep, LevelSetOutputType);
  itkGetMacro(TimeStep, LevelSetOutputType);

  itkGetMacro(RMSChangeAccumulator, LevelSetOutputType);

  itkSetObjectMacro(EquationContainer, EquationContainerType);
  itkGetModifiableObjectMacro(EquationContainer, EquationContainerType);

  itkSetMacro(CurrentLevelSetId, IdentifierType);
  itkGetMacro(CurrentLevelSetId, IdentifierType);

  void
  SetUpdate(const LevelSetLayerType & update);
 
protected:
  UpdateWhitakerSparseLevelSet();
  ~UpdateWhitakerSparseLevelSet() override = default;

  void
  UpdateLayerZero();

  void
  UpdateLayerMinus1();

  void
  UpdateLayerPlus1();

  void
  UpdateLayerMinus2();

  void
  UpdateLayerPlus2();

  void
  MovePointIntoZeroLevelSet();

  void
  MovePointFromMinus1();

  void
  MovePointFromPlus1();

  void
  MovePointFromMinus2();

  void
  MovePointFromPlus2();
 
private:
  LevelSetOutputType m_TimeStep{};
  LevelSetOutputType m_RMSChangeAccumulator{};
  IdentifierType     m_CurrentLevelSetId{};
 
  EquationContainerPointer m_EquationContainer{};
 
  LevelSetLayerType m_Update{};
  LevelSetPointer   m_InputLevelSet{};
  LevelSetPointer   m_OutputLevelSet{};
 
  LevelSetPointer   m_TempLevelSet{};
  LevelSetLayerType m_TempPhi{};
 
  LevelSetLayerIdType m_MinStatus{};
  LevelSetLayerIdType m_MaxStatus{};
 
  LabelImagePointer m_InternalImage{};
 
  LevelSetOffsetType m_Offset{};
 
  using NeighborhoodIteratorType = ShapedNeighborhoodIterator<LabelImageType>;
 
  using NodePairType = std::pair<LevelSetInputType, LevelSetOutputType>;
};
} // namespace itk
 
#ifndef ITK_MANUAL_INSTANTIATION
#  include "itkUpdateWhitakerSparseLevelSet.hxx"
#endif
#endif // itkUpdateWhitakerSparseLevelSet_h