mitkImageLiveWireContourModelFilter.h

Go to the documentation of this file.

/*============================================================================
 
The Medical Imaging Interaction Toolkit (MITK)
 
Copyright (c) German Cancer Research Center (DKFZ)
All rights reserved.
 
Use of this source code is governed by a 3-clause BSD license that can be
found in the LICENSE file.
 
============================================================================*/
 
#ifndef mitkImageLiveWireContourModelFilter_h
#define mitkImageLiveWireContourModelFilter_h
 
#include "mitkCommon.h"
#include "mitkContourModel.h"
#include "mitkContourModelSource.h"
#include <MitkSegmentationExports.h>
 
#include <mitkImage.h>
#include <mitkImageAccessByItk.h>
#include <mitkImageCast.h>
 
#include <itkShortestPathCostFunctionLiveWire.h>
#include <itkShortestPathImageFilter.h>
 
namespace mitk
{
  class MITKSEGMENTATION_EXPORT ImageLiveWireContourModelFilter : public ContourModelSource
  {
  public:
    mitkClassMacro(ImageLiveWireContourModelFilter, ContourModelSource);
    itkFactorylessNewMacro(Self);
    itkCloneMacro(Self);
 
      typedef ContourModel OutputType;
    typedef OutputType::Pointer OutputTypePointer;
    typedef mitk::Image InputType;
 
    typedef itk::Image<float, 2> InternalImageType;
    typedef itk::ShortestPathImageFilter<InternalImageType, InternalImageType> ShortestPathImageFilterType;
    typedef itk::ShortestPathCostFunctionLiveWire<InternalImageType> CostFunctionType;
    typedef std::vector<itk::Index<2>> ShortestPathType;
 
    itkSetMacro(StartPoint, mitk::Point3D);
    itkGetMacro(StartPoint, mitk::Point3D);
 
    itkSetMacro(EndPoint, mitk::Point3D);
    itkGetMacro(EndPoint, mitk::Point3D);
 
    itkSetMacro(UseDynamicCostMap, bool);
    itkGetMacro(UseDynamicCostMap, bool);
 
    void ClearRepulsivePoints();
 
    void SetRepulsivePoints(const ShortestPathType &points);
 
    void AddRepulsivePoint(const itk::Index<2> &idx);
 
    void RemoveRepulsivePoint(const itk::Index<2> &idx);
 
    virtual void SetInput(const InputType *input);
 
    using Superclass::SetInput;
    virtual void SetInput(unsigned int idx, const InputType *input);
 
    const InputType *GetInput(void);
 
    const InputType *GetInput(unsigned int idx);
 
    virtual OutputType *GetOutput();
 
    virtual void DumpMaskImage();
 
    bool CreateDynamicCostMap(mitk::ContourModel *path = nullptr);
 
    void SetUseCostFunction(bool doUseCostFunction) { m_ShortestPathFilter->SetUseCostFunction(doUseCostFunction); };
 
  protected:
    ImageLiveWireContourModelFilter();
 
    ~ImageLiveWireContourModelFilter() override;
 
    void GenerateOutputInformation() override{};
 
    void GenerateData() override;
 
    void UpdateLiveWire();
 
    mitk::Point3D m_StartPoint;
 
    mitk::Point3D m_EndPoint;
 
    mitk::Point3D m_StartPointInIndex;
 
    mitk::Point3D m_EndPointInIndex;
 
    CostFunctionType::Pointer m_CostFunction;
 
    ShortestPathImageFilterType::Pointer m_ShortestPathFilter;
 
    bool m_UseDynamicCostMap;
 
    unsigned int m_TimeStep;
 
    template <typename TPixel, unsigned int VImageDimension>
    void ItkPreProcessImage(const itk::Image<TPixel, VImageDimension> *inputImage);
 
    template <typename TPixel, unsigned int VImageDimension>
    void CreateDynamicCostMapByITK(const itk::Image<TPixel, VImageDimension> *inputImage,
                                   mitk::ContourModel *path = nullptr);
 
    InternalImageType::Pointer m_InternalImage;
  };
}
 
#endif