niftkDrawTool.h

Go to the documentation of this file.

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

  NifTK: A software platform for medical image computing.

  Copyright (c) University College London (UCL). 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 in the top level directory for details.

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

#ifndef niftkDrawTool_h
#define niftkDrawTool_h

#include "niftkMIDASExports.h"

#include <mitkPlanarCircle.h>

#include "niftkContourTool.h"
#include "niftkDrawToolEventInterface.h"

namespace niftk
{

class NIFTKMIDAS_EXPORT DrawTool : public ContourTool {

public:

  mitkClassMacro(DrawTool, ContourTool)
  itkNewMacro(DrawTool)

  virtual void InitializeStateMachine();

  virtual void ExecuteOperation(mitk::Operation* operation);

  virtual const char* GetName() const;

  virtual const char** GetXPM() const;

  double GetCursorSize() const;

  void SetCursorSize(double cursorSize);

  mitk::Message1<double> CursorSizeChanged;

  virtual bool StartDrawing(mitk::StateMachineAction* action, mitk::InteractionEvent* event);

  virtual bool KeepDrawing(mitk::StateMachineAction* action, mitk::InteractionEvent* event);

  virtual bool StopDrawing(mitk::StateMachineAction* action, mitk::InteractionEvent* event);

  virtual bool StartErasing(mitk::StateMachineAction* action, mitk::InteractionEvent* event);

  virtual bool KeepErasing(mitk::StateMachineAction* action, mitk::InteractionEvent* event);

  virtual bool StopErasing(mitk::StateMachineAction* action, mitk::InteractionEvent* event);

  virtual void ClearWorkingData();

  virtual void Clean(int sliceNumber, int axisNumber);

protected:

  DrawTool(); // purposely hidden
  virtual ~DrawTool(); // purposely hidden

  virtual void ConnectActionsAndFunctions();

  virtual void Activated();

private:

  template<typename TPixel, unsigned int VImageDimension>
  void ITKCleanContours(
      itk::Image<TPixel, VImageDimension> *itkImage,
      mitk::ContourModelSet& inputContours,
      mitk::ContourModelSet& outputContours,
      int axis,
      int sliceNumber
      );

  static const mitk::OperationType MIDAS_DRAW_TOOL_OP_ERASE_CONTOUR;

  static const mitk::OperationType MIDAS_DRAW_TOOL_OP_CLEAN_CONTOUR;

  bool DeleteFromContour(int workingDataNumber, mitk::StateMachineAction* action, mitk::InteractionEvent* event);

  void SetEraserScopeVisible(bool visible, mitk::BaseRenderer* renderer = 0);

  double m_CursorSize;

  mitk::Point3D m_MostRecentPointInMm;

  DrawToolEventInterface::Pointer m_Interface;

  mitk::PlanarCircle::Pointer m_EraserScope;
  mitk::DataNode::Pointer m_EraserScopeNode;
  bool m_EraserScopeVisible;

};

}

#endif