mitkOtsuTool3D.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 MITKOTSUTOOL3D_H
#define MITKOTSUTOOL3D_H

#include "itkImage.h"
#include "mitkAutoSegmentationTool.h"
#include <MitkSegmentationExports.h>

namespace us
{
  class ModuleResource;
}

namespace mitk
{
  class Image;

  class MITKSEGMENTATION_EXPORT OtsuTool3D : public AutoSegmentationTool
  {
  public:
    mitkClassMacro(OtsuTool3D, AutoSegmentationTool);
    itkFactorylessNewMacro(Self) itkCloneMacro(Self)

      virtual const char *GetName() const override;
    virtual const char **GetXPM() const override;
    us::ModuleResource GetIconResource() const override;

    virtual void Activated() override;
    virtual void Deactivated() override;

    void RunSegmentation(int regions, bool useValley, int numberOfBins);
    void ConfirmSegmentation();
    // void UpdateBinaryPreview(int regionID);
    void UpdateBinaryPreview(std::vector<int> regionIDs);
    void UpdateVolumePreview(bool volumeRendering);
    void ShowMultiLabelResultNode(bool);

    int GetNumberOfBins();

  protected:
    OtsuTool3D();
    virtual ~OtsuTool3D();

    template <typename TPixel, unsigned int VImageDimension>
    void CalculatePreview(itk::Image<TPixel, VImageDimension> *itkImage, std::vector<int> regionIDs);

    itk::SmartPointer<Image> m_OriginalImage;
    // holds the user selected binary segmentation
    mitk::DataNode::Pointer m_BinaryPreviewNode;
    // holds the multilabel result as a preview image
    mitk::DataNode::Pointer m_MultiLabelResultNode;
    // holds the user selected binary segmentation masked original image
    mitk::DataNode::Pointer m_MaskedImagePreviewNode;

  }; // class
} // namespace
#endif