mitkPickingTool.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 mitkPickingTool_h_Included
#define mitkPickingTool_h_Included

#include "itkImage.h"
#include "mitkAutoSegmentationTool.h"
#include "mitkCommon.h"
#include "mitkDataStorage.h"
#include "mitkPointSet.h"
#include "mitkSinglePointDataInteractor.h"
#include <MitkSegmentationExports.h>

namespace us
{
  class ModuleResource;
}

namespace mitk
{
  class MITKSEGMENTATION_EXPORT PickingTool : public AutoSegmentationTool
  {
  public:
    mitkClassMacro(PickingTool, AutoSegmentationTool);
    itkFactorylessNewMacro(Self) itkCloneMacro(Self)

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

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

    virtual DataNode::Pointer GetPointSetNode();

    mitk::DataNode *GetReferenceData();
    mitk::DataNode *GetWorkingData();
    mitk::DataStorage *GetDataStorage();

    void ConfirmSegmentation();

  protected:
    PickingTool(); // purposely hidden
    virtual ~PickingTool();

    // Callback for point add event of PointSet
    void OnPointAdded();

    // Observer id
    long m_PointSetAddObserverTag;

    mitk::DataNode::Pointer m_ResultNode;

    // itk regrowing
    template <typename TPixel, unsigned int VImageDimension>
    void StartRegionGrowing(itk::Image<TPixel, VImageDimension> *itkImage,
                            mitk::BaseGeometry *imageGeometry,
                            mitk::PointSet::PointType seedPoint);

    // seed point
    PointSet::Pointer m_PointSet;
    SinglePointDataInteractor::Pointer m_SeedPointInteractor;
    DataNode::Pointer m_PointSetNode;

    DataNode *m_WorkingData;
  };

} // namespace

#endif