itkFastMarchingImageToNodePairContainerAdaptor.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 itkFastMarchingImageToNodePairContainerAdaptor_h
#define itkFastMarchingImageToNodePairContainerAdaptor_h
 
#include "itkObject.h"
#include "itkFastMarchingTraits.h"
 
namespace itk
{
template <typename TInput, typename TOutput, typename TImage>
class ITK_TEMPLATE_EXPORT FastMarchingImageToNodePairContainerAdaptor : public Object
{
public:
  ITK_DISALLOW_COPY_AND_MOVE(FastMarchingImageToNodePairContainerAdaptor);
 
  using Self = FastMarchingImageToNodePairContainerAdaptor;
  using Superclass = Object;
  using Pointer = SmartPointer<Self>;
  using ConstPointer = SmartPointer<const Self>;
 

  itkNewMacro(Self);

  itkOverrideGetNameOfClassMacro(FastMarchingImageToNodePairContainerAdaptor);
 
  using Traits = FastMarchingTraits<TInput, TOutput>;
  using NodePairType = typename Traits::NodePairType;
  using NodePairContainerType = typename Traits::NodePairContainerType;
  using NodePairContainerPointer = typename Traits::NodePairContainerPointer;
  using LabelType = typename Traits::LabelType;
  using OutputPixelType = typename Traits::OutputPixelType;
 
  using ImageType = TImage;
  using ImagePointer = typename ImageType::Pointer;
  using ImageConstPointer = typename ImageType::ConstPointer;
  using ImagePixelType = typename ImageType::PixelType;
 
  static constexpr unsigned int ImageDimension = Traits::ImageDimension;

  void
  SetAliveImage(const ImageType * iImage);
  itkGetConstObjectMacro(AliveImage, ImageType);

  void
  SetTrialImage(const ImageType * iImage);
  itkGetConstObjectMacro(TrialImage, ImageType);

  void
  SetForbiddenImage(const ImageType * iImage);
  itkGetConstObjectMacro(ForbiddenImage, ImageType);
 
  itkSetMacro(IsForbiddenImageBinaryMask, bool);
  itkGetConstMacro(IsForbiddenImageBinaryMask, bool);
  itkBooleanMacro(IsForbiddenImageBinaryMask);

  NodePairContainerType *
  GetAlivePoints();

  NodePairContainerType *
  GetTrialPoints();

  NodePairContainerType *
  GetForbiddenPoints();
 
  itkSetMacro(AliveValue, OutputPixelType);
  itkGetConstMacro(AliveValue, OutputPixelType);
 
  itkSetMacro(TrialValue, OutputPixelType);
  itkGetConstMacro(TrialValue, OutputPixelType);

  void
  Update();
 
protected:
  FastMarchingImageToNodePairContainerAdaptor();

  ~FastMarchingImageToNodePairContainerAdaptor() override = default;
 
  ImageConstPointer m_AliveImage{};
  ImageConstPointer m_TrialImage{};
  ImageConstPointer m_ForbiddenImage{};
 
  NodePairContainerPointer m_AlivePoints{};
  NodePairContainerPointer m_TrialPoints{};
  NodePairContainerPointer m_ForbiddenPoints{};
 
  OutputPixelType m_AliveValue{};
  OutputPixelType m_TrialValue{};
 
  bool m_IsForbiddenImageBinaryMask{ false };
 
  virtual void
  GenerateData();

  void
  SetPointsFromImage(const ImageType * image, const LabelType & iLabel, const OutputPixelType & iValue);
};
} // namespace itk
 
#ifndef ITK_MANUAL_INSTANTIATION
#  include "itkFastMarchingImageToNodePairContainerAdaptor.hxx"
#endif
 
#endif // itkFastMarchingImageToNodePairContainerAdaptor_h