mitkPlanarFigureSegmentationController.h

Go to the documentation of this file.

/*===================================================================

The Medical Imaging Interaction Toolkit (MITK)

Copyright (c) German Cancer Research Center,
Division of Medical and Biological Informatics.
All rights reserved.

This software is distributed WITHOUT ANY WARRANTY; without
even the implied warranty of MERCHANTABILITY or FITNESS FOR
A PARTICULAR PURPOSE.

See LICENSE.txt or http://www.mitk.org for details.

===================================================================*/

#ifndef mitkPlanarFigureSegmentationController_h_Included
#define mitkPlanarFigureSegmentationController_h_Included

#include "mitkCommon.h"
#include "mitkImage.h"
#include <MitkPlanarFigureSegmentationExports.h>

#include "mitkPlanarFigure.h"
#include "mitkReduceContourSetFilter.h"
#include <mitkComputeContourSetNormalsFilter.h>
#include <mitkCreateDistanceImageFromSurfaceFilter.h>
#include <mitkSurface.h>

namespace mitk
{
  class MITKPLANARFIGURESEGMENTATION_EXPORT PlanarFigureSegmentationController : public itk::Object
  {
  public:
    mitkClassMacroItkParent(PlanarFigureSegmentationController, itk::Object);
    itkFactorylessNewMacro(
      PlanarFigureSegmentationController) 
      itkCloneMacro(Self)

        virtual ~PlanarFigureSegmentationController();

    typedef std::vector<PlanarFigure::Pointer> PlanarFigureListType;
    typedef std::vector<Surface::Pointer> SurfaceListType;

    void SetReferenceImage(Image::Pointer referenceImage);

    void AddPlanarFigure(PlanarFigure::Pointer planarFigure);

    void RemovePlanarFigure(mitk::PlanarFigure::Pointer planarFigure);

    Image::Pointer GetInterpolationResult();

    static Surface::Pointer CreateSurfaceFromPlanarFigure(PlanarFigure::Pointer figure);

    PlanarFigureListType GetAllPlanarFigures();

    void SetReferenceDirectionVector(mitk::Vector3D vector);

  protected:
    PlanarFigureSegmentationController();

    void InitializeFilters();

    template <typename TPixel, unsigned int VImageDimension>
    void GetImageBase(itk::Image<TPixel, VImageDimension> *input, itk::ImageBase<3>::Pointer &result);

    PlanarFigureListType m_PlanarFigureList;
    SurfaceListType m_SurfaceList;

    ReduceContourSetFilter::Pointer m_ReduceFilter;

    ComputeContourSetNormalsFilter::Pointer m_NormalsFilter;

    CreateDistanceImageFromSurfaceFilter::Pointer m_DistanceImageCreator;

    Image::Pointer m_ReferenceImage;

    Image::Pointer m_SegmentationAsImage;

    mitk::Vector3D m_ReferenceDirectionVector;
  };

} // namespace

#endif