itkUpdateWhitakerSparseLevelSet.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 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_ASSIGN(UpdateWhitakerSparseLevelSet);

  using Self = UpdateWhitakerSparseLevelSet;
  using Pointer = SmartPointer< Self >;
  using ConstPointer = SmartPointer< const Self >;
  using Superclass = Object;

  itkNewMacro( Self );

  itkTypeMacro( UpdateWhitakerSparseLevelSet, Object );

  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 >;
};
}

#ifndef ITK_MANUAL_INSTANTIATION
#include "itkUpdateWhitakerSparseLevelSet.hxx"
#endif
#endif // itkUpdateWhitakerSparseLevelSet_h